/[LeafOK_CVS]/lbbs/src/user_list.c
ViewVC logotype

Diff of /lbbs/src/user_list.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

Revision 1.8 by sysadm, Wed Oct 22 07:16:48 2025 UTC Revision 1.34 by sysadm, Mon Nov 17 11:56:11 2025 UTC
# Line 1  Line 1 
1  /***************************************************************************  /* SPDX-License-Identifier: GPL-3.0-or-later */
2                                                   user_list.c  -  description  /*
3                                                           -------------------   * user_list
4          Copyright            : (C) 2004-2025 by Leaflet   *   - data model and basic operations of (online) user list
5          Email                : leaflet@leafok.com   *
6   ***************************************************************************/   * Copyright (C) 2004-2025  Leaflet <leaflet@leafok.com>
7     */
8  /***************************************************************************  
9   *                                                                         *  #ifdef HAVE_CONFIG_H
10   *   This program is free software; you can redistribute it and/or modify  *  #include "config.h"
11   *   it under the terms of the GNU General Public License as published by  *  #endif
  *   the Free Software Foundation; either version 3 of the License, or     *  
  *   (at your option) any later version.                                   *  
  *                                                                         *  
  ***************************************************************************/  
12    
13  #include "common.h"  #include "common.h"
14  #include "database.h"  #include "database.h"
15  #include "log.h"  #include "log.h"
16  #include "trie_dict.h"  #include "trie_dict.h"
17  #include "user_list.h"  #include "user_list.h"
18    #include "user_stat.h"
19  #include <errno.h>  #include <errno.h>
20  #include <stdlib.h>  #include <stdlib.h>
21  #include <string.h>  #include <string.h>
# Line 29  Line 26 
26  #include <sys/sem.h>  #include <sys/sem.h>
27  #include <sys/shm.h>  #include <sys/shm.h>
28    
29  #ifdef _SEM_SEMUN_UNDEFINED  #if defined(_SEM_SEMUN_UNDEFINED) || defined(__MSYS__) || defined(__MINGW32__)
30  union semun  union semun
31  {  {
32          int val;                           /* Value for SETVAL */          int val;                           /* Value for SETVAL */
# Line 38  union semun Line 35  union semun
35          struct seminfo *__buf; /* Buffer for IPC_INFO          struct seminfo *__buf; /* Buffer for IPC_INFO
36                                                            (Linux-specific) */                                                            (Linux-specific) */
37  };  };
38  #endif // #ifdef _SEM_SEMUN_UNDEFINED  #endif // #if defined(_SEM_SEMUN_UNDEFINED)
39    
40  #define USER_LIST_TRY_LOCK_WAIT_TIME 1 // second  enum _user_list_constant_t
41  #define USER_LIST_TRY_LOCK_TIMES 10  {
42            USER_LIST_TRY_LOCK_WAIT_TIME = 1, // second
43            USER_LIST_TRY_LOCK_TIMES = 10,
44    };
45    
46  struct user_list_pool_t  struct user_list_pool_t
47  {  {
# Line 53  struct user_list_pool_t Line 53  struct user_list_pool_t
53          USER_ONLINE_LIST user_online_list[2];          USER_ONLINE_LIST user_online_list[2];
54          USER_ONLINE_LIST *p_online_current;          USER_ONLINE_LIST *p_online_current;
55          USER_ONLINE_LIST *p_online_new;          USER_ONLINE_LIST *p_online_new;
56            USER_STAT_MAP user_stat_map;
57            int user_login_count;
58  };  };
59  typedef struct user_list_pool_t USER_LIST_POOL;  typedef struct user_list_pool_t USER_LIST_POOL;
60    
# Line 91  static int user_list_rd_lock(int semid); Line 93  static int user_list_rd_lock(int semid);
93  static int user_list_rw_lock(int semid);  static int user_list_rw_lock(int semid);
94    
95  static int user_list_load(MYSQL *db, USER_LIST *p_list);  static int user_list_load(MYSQL *db, USER_LIST *p_list);
96  static int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_list);  static int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_online_list);
97    static int user_login_count_load(MYSQL *db);
98    
99  static int user_info_index_uid_comp(const void *ptr1, const void *ptr2)  static int user_info_index_uid_comp(const void *ptr1, const void *ptr2)
100  {  {
# Line 106  static int user_info_index_uid_comp(cons Line 109  static int user_info_index_uid_comp(cons
109          {          {
110                  return 1;                  return 1;
111          }          }
112            else if (p1->id < p2->id)
113            {
114                    return -1;
115            }
116            else if (p1->id > p2->id)
117            {
118                    return 1;
119            }
120          return 0;          return 0;
121  }  }
122    
# Line 116  int user_list_load(MYSQL *db, USER_LIST Line 127  int user_list_load(MYSQL *db, USER_LIST
127          char sql[SQL_BUFFER_LEN];          char sql[SQL_BUFFER_LEN];
128          int ret = 0;          int ret = 0;
129          int i;          int i;
130            int j;
131            int32_t last_uid;
132            size_t intro_buf_offset;
133            size_t intro_len;
134    
135          if (db == NULL || p_list == NULL)          if (db == NULL || p_list == NULL)
136          {          {
# Line 123  int user_list_load(MYSQL *db, USER_LIST Line 138  int user_list_load(MYSQL *db, USER_LIST
138                  return -1;                  return -1;
139          }          }
140    
141            if (p_list->user_count > 0)
142            {
143                    last_uid = p_list->users[p_list->user_count - 1].uid;
144            }
145            else
146            {
147                    last_uid = -1;
148            }
149    
150          snprintf(sql, sizeof(sql),          snprintf(sql, sizeof(sql),
151                           "SELECT user_list.UID AS UID, username, nickname, gender, gender_pub, life, exp, "                           "SELECT user_list.UID AS UID, username, nickname, gender, gender_pub, life, exp, visit_count, "
152                           "UNIX_TIMESTAMP(signup_dt), UNIX_TIMESTAMP(last_login_dt), UNIX_TIMESTAMP(birthday) "                           "UNIX_TIMESTAMP(signup_dt), UNIX_TIMESTAMP(last_login_dt), UNIX_TIMESTAMP(last_logout_dt), "
153                             "UNIX_TIMESTAMP(birthday), `introduction` "
154                           "FROM user_list INNER JOIN user_pubinfo ON user_list.UID = user_pubinfo.UID "                           "FROM user_list INNER JOIN user_pubinfo ON user_list.UID = user_pubinfo.UID "
155                           "INNER JOIN user_reginfo ON user_list.UID = user_reginfo.UID "                           "INNER JOIN user_reginfo ON user_list.UID = user_reginfo.UID "
156                           "WHERE enable ORDER BY username");                           "WHERE enable ORDER BY username");
# Line 144  int user_list_load(MYSQL *db, USER_LIST Line 169  int user_list_load(MYSQL *db, USER_LIST
169                  goto cleanup;                  goto cleanup;
170          }          }
171    
172            intro_buf_offset = 0;
173          i = 0;          i = 0;
174          while ((row = mysql_fetch_row(rs)))          while ((row = mysql_fetch_row(rs)))
175          {          {
# Line 158  int user_list_load(MYSQL *db, USER_LIST Line 184  int user_list_load(MYSQL *db, USER_LIST
184                  p_list->users[i].gender_pub = (int8_t)(row[4] == NULL ? 0 : atoi(row[4]));                  p_list->users[i].gender_pub = (int8_t)(row[4] == NULL ? 0 : atoi(row[4]));
185                  p_list->users[i].life = (row[5] == NULL ? 0 : atoi(row[5]));                  p_list->users[i].life = (row[5] == NULL ? 0 : atoi(row[5]));
186                  p_list->users[i].exp = (row[6] == NULL ? 0 : atoi(row[6]));                  p_list->users[i].exp = (row[6] == NULL ? 0 : atoi(row[6]));
187                  p_list->users[i].signup_dt = (row[7] == NULL ? 0 : atol(row[7]));                  p_list->users[i].visit_count = (row[7] == NULL ? 0 : atoi(row[7]));
188                  p_list->users[i].last_login_dt = (row[8] == NULL ? 0 : atol(row[8]));                  p_list->users[i].signup_dt = (row[8] == NULL ? 0 : atol(row[8]));
189                  p_list->users[i].birthday = (row[9] == NULL ? 0 : atol(row[9]));                  p_list->users[i].last_login_dt = (row[9] == NULL ? 0 : atol(row[9]));
190                    p_list->users[i].last_logout_dt = (row[10] == NULL ? 0 : atol(row[10]));
191                  // index                  p_list->users[i].birthday = (row[11] == NULL ? 0 : atol(row[11]));
192                  p_list->index_uid[i].uid = p_list->users[i].uid;                  intro_len = strlen((row[12] == NULL ? "" : row[12]));
193                  p_list->index_uid[i].id = i;                  if (intro_len >= sizeof(p_list->user_intro_buf) - 1 - intro_buf_offset)
194                    {
195                            log_error("OOM for user introduction: len=%d, i=%d\n", intro_len, i);
196                            break;
197                    }
198                    memcpy(p_list->user_intro_buf + intro_buf_offset,
199                               (row[12] == NULL ? "" : row[12]),
200                               intro_len + 1);
201                    p_list->users[i].intro = p_list->user_intro_buf + intro_buf_offset;
202                    intro_buf_offset += (intro_len + 1);
203    
204                  i++;                  i++;
205                  if (i >= BBS_max_user_count)                  if (i >= BBS_max_user_count)
# Line 176  int user_list_load(MYSQL *db, USER_LIST Line 211  int user_list_load(MYSQL *db, USER_LIST
211          mysql_free_result(rs);          mysql_free_result(rs);
212          rs = NULL;          rs = NULL;
213    
214          p_list->user_count = i;          if (i != p_list->user_count || p_list->users[i - 1].uid != last_uid) // Count of users changed
215            {
216                    // Rebuild index
217                    for (j = 0; j < i; j++)
218                    {
219                            p_list->index_uid[j].uid = p_list->users[j].uid;
220                            p_list->index_uid[j].id = j;
221                    }
222    
223          // Sort index                  qsort(p_list->index_uid, (size_t)i, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp);
224          qsort(p_list->index_uid, (size_t)i, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp);  
225    #ifdef _DEBUG
226                    log_error("Rebuild index of %d users, last_uid=%d\n", i, p_list->users[i - 1].uid);
227    #endif
228            }
229    
230            p_list->user_count = i;
231    
232  #ifdef _DEBUG  #ifdef _DEBUG
233          log_error("Loaded %d users\n", p_list->user_count);          log_error("Loaded %d users\n", p_list->user_count);
# Line 191  cleanup: Line 239  cleanup:
239          return ret;          return ret;
240  }  }
241    
242  int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_list)  int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_online_list)
243  {  {
244          MYSQL_RES *rs = NULL;          MYSQL_RES *rs = NULL;
245          MYSQL_ROW row;          MYSQL_ROW row;
246          char sql[SQL_BUFFER_LEN];          char sql[SQL_BUFFER_LEN];
247          int ret = 0;          int ret = 0;
248          int i;          int i;
249            int j;
250            int user_cnt;
251            int guest_cnt;
252    
253          if (db == NULL || p_list == NULL)          if (db == NULL || p_online_list == NULL)
254          {          {
255                  log_error("NULL pointer error\n");                  log_error("NULL pointer error\n");
256                  return -1;                  return -1;
# Line 227  int user_online_list_load(MYSQL *db, USE Line 278  int user_online_list_load(MYSQL *db, USE
278          }          }
279    
280          i = 0;          i = 0;
281            user_cnt = 0;
282            guest_cnt = 0;
283          while ((row = mysql_fetch_row(rs)))          while ((row = mysql_fetch_row(rs)))
284          {          {
285                  p_list->users[i].id = i;                  if (atoi(row[1]) == 0) // guest
                 strncpy(p_list->users[i].session_id, row[0], sizeof(p_list->users[i].session_id) - 1);  
                 p_list->users[i].session_id[sizeof(p_list->users[i].session_id) - 1] = '\0';  
   
                 if ((ret = query_user_info_by_uid(atoi(row[1]), &(p_list->users[i].user_info))) < 0)  
286                  {                  {
287                          log_error("query_user_info(%d) error\n", atoi(row[1]));                          guest_cnt++;
288                          continue;                          continue;
289                  }                  }
290                  else if (ret == 0) // Guest                  else
291                  {                  {
292                          p_list->users[i].user_info.id = -1;                          user_cnt++;
293                          p_list->users[i].user_info.uid = 0;                  }
294                          strncpy(p_list->users[i].user_info.username, "guest", sizeof(p_list->users[i].user_info.username) - 1);  
295                          p_list->users[i].user_info.username[sizeof(p_list->users[i].user_info.username) - 1] = '\0';                  p_online_list->users[i].id = i;
296                          strncpy(p_list->users[i].user_info.nickname, "Guest", sizeof(p_list->users[i].user_info.nickname) - 1);                  strncpy(p_online_list->users[i].session_id, row[0], sizeof(p_online_list->users[i].session_id) - 1);
297                          p_list->users[i].user_info.nickname[sizeof(p_list->users[i].user_info.nickname) - 1] = '\0';                  p_online_list->users[i].session_id[sizeof(p_online_list->users[i].session_id) - 1] = '\0';
298                          p_list->users[i].user_info.gender = 'M';  
299                          p_list->users[i].user_info.gender_pub = 0;                  if ((ret = query_user_info_by_uid(atoi(row[1]), &(p_online_list->users[i].user_info), NULL, 0)) <= 0)
300                          p_list->users[i].user_info.life = 150;                  {
301                          p_list->users[i].user_info.exp = 0;                          log_error("query_user_info_by_uid(%d) error\n", atoi(row[1]));
302                          p_list->users[i].user_info.signup_dt = 0;                          continue;
303                          p_list->users[i].user_info.last_login_dt = 0;                  }
304                          p_list->users[i].user_info.birthday = 0;  
305                  }                  strncpy(p_online_list->users[i].ip, row[2], sizeof(p_online_list->users[i].ip) - 1);
306                    p_online_list->users[i].ip[sizeof(p_online_list->users[i].ip) - 1] = '\0';
307                  strncpy(p_list->users[i].ip, row[2], sizeof(p_list->users[i].ip) - 1);  
308                  p_list->users[i].ip[sizeof(p_list->users[i].ip) - 1] = '\0';                  strncpy(p_online_list->users[i].current_action, row[3], sizeof(p_online_list->users[i].current_action) - 1);
309                    p_online_list->users[i].current_action[sizeof(p_online_list->users[i].current_action) - 1] = '\0';
310                  strncpy(p_list->users[i].current_action, row[3], sizeof(p_list->users[i].current_action) - 1);                  p_online_list->users[i].current_action_title = NULL;
311                  p_list->users[i].current_action[sizeof(p_list->users[i].current_action) - 1] = '\0';                  if (p_online_list->users[i].current_action[0] == '\0')
                 p_list->users[i].current_action_title = NULL;  
                 if (p_list->users[i].current_action[0] == '\0')  
312                  {                  {
313                          p_list->users[i].current_action_title = "";                          p_online_list->users[i].current_action_title = "";
314                  }                  }
315                  else if (trie_dict_get(p_trie_action_dict, p_list->users[i].current_action, (int64_t *)(&(p_list->users[i].current_action_title))) < 0)                  else if (trie_dict_get(p_trie_action_dict, p_online_list->users[i].current_action, (int64_t *)(&(p_online_list->users[i].current_action_title))) < 0)
316                  {                  {
317                          log_error("trie_dict_get(p_trie_action_dict, %s) error on session_id=%s\n",                          log_error("trie_dict_get(p_trie_action_dict, %s) error on session_id=%s\n",
318                                            p_list->users[i].current_action, p_list->users[i].session_id);                                            p_online_list->users[i].current_action, p_online_list->users[i].session_id);
319                          continue;                          continue;
320                  }                  }
321    
322                  p_list->users[i].login_tm = (row[4] == NULL ? 0 : atol(row[4]));                  p_online_list->users[i].login_tm = (row[4] == NULL ? 0 : atol(row[4]));
323                  p_list->users[i].last_tm = (row[5] == NULL ? 0 : atol(row[5]));                  p_online_list->users[i].last_tm = (row[5] == NULL ? 0 : atol(row[5]));
324    
325                  i++;                  i++;
326                  if (i >= BBS_max_user_online_count)                  if (i >= BBS_max_user_online_count)
# Line 285  int user_online_list_load(MYSQL *db, USE Line 332  int user_online_list_load(MYSQL *db, USE
332          mysql_free_result(rs);          mysql_free_result(rs);
333          rs = NULL;          rs = NULL;
334    
335          p_list->user_count = i;          if (user_cnt > 0)
336            {
337                    // Rebuild index
338                    for (j = 0; j < user_cnt; j++)
339                    {
340                            p_online_list->index_uid[j].uid = p_online_list->users[j].user_info.uid;
341                            p_online_list->index_uid[j].id = j;
342                    }
343    
344  #ifdef _DEBUG                  qsort(p_online_list->index_uid, (size_t)user_cnt, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp);
345          log_error("Loaded %d users\n", p_list->user_count);          }
346  #endif  
347            p_online_list->user_count = user_cnt;
348            p_online_list->guest_count = guest_cnt;
349    
350  cleanup:  cleanup:
351          mysql_free_result(rs);          mysql_free_result(rs);
# Line 297  cleanup: Line 353  cleanup:
353          return ret;          return ret;
354  }  }
355    
356  int user_list_pool_init(void)  int user_login_count_load(MYSQL *db)
357    {
358            MYSQL_RES *rs = NULL;
359            MYSQL_ROW row;
360            char sql[SQL_BUFFER_LEN];
361    
362            if (db == NULL)
363            {
364                    log_error("NULL pointer error\n");
365                    return -1;
366            }
367    
368            snprintf(sql, sizeof(sql),
369                             "SELECT ID FROM user_login_log ORDER BY ID DESC LIMIT 1");
370            if (mysql_query(db, sql) != 0)
371            {
372                    log_error("Query user_login_log error: %s\n", mysql_error(db));
373                    return -2;
374            }
375            if ((rs = mysql_store_result(db)) == NULL)
376            {
377                    log_error("Get user_login_log data failed\n");
378                    return -2;
379            }
380            if ((row = mysql_fetch_row(rs)))
381            {
382                    p_user_list_pool->user_login_count = atoi(row[0]);
383            }
384            mysql_free_result(rs);
385    
386            return 0;
387    }
388    
389    int user_list_pool_init(const char *filename)
390  {  {
391          int shmid;          int shmid;
392          int semid;          int semid;
# Line 331  int user_list_pool_init(void) Line 420  int user_list_pool_init(void)
420    
421          // Allocate shared memory          // Allocate shared memory
422          proj_id = (int)(time(NULL) % getpid());          proj_id = (int)(time(NULL) % getpid());
423          key = ftok(VAR_USER_LIST_SHM, proj_id);          key = ftok(filename, proj_id);
424          if (key == -1)          if (key == -1)
425          {          {
426                  log_error("ftok(%s %d) error (%d)\n", VAR_USER_LIST_SHM, proj_id, errno);                  log_error("ftok(%s %d) error (%d)\n", filename, proj_id, errno);
427                  return -2;                  return -2;
428          }          }
429    
# Line 387  int user_list_pool_init(void) Line 476  int user_list_pool_init(void)
476          p_user_list_pool->p_online_current = &(p_user_list_pool->user_online_list[0]);          p_user_list_pool->p_online_current = &(p_user_list_pool->user_online_list[0]);
477          p_user_list_pool->p_online_new = &(p_user_list_pool->user_online_list[1]);          p_user_list_pool->p_online_new = &(p_user_list_pool->user_online_list[1]);
478    
479            user_stat_map_init(&(p_user_list_pool->user_stat_map));
480    
481          return 0;          return 0;
482  }  }
483    
# Line 411  void user_list_pool_cleanup(void) Line 502  void user_list_pool_cleanup(void)
502                  log_error("shmdt(shmid = %d) error (%d)\n", shmid, errno);                  log_error("shmdt(shmid = %d) error (%d)\n", shmid, errno);
503          }          }
504    
505          if (shmctl(shmid, IPC_RMID, NULL) == -1)          if (shmid != 0 && shmctl(shmid, IPC_RMID, NULL) == -1)
506          {          {
507                  log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", shmid, errno);                  log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", shmid, errno);
508          }          }
# Line 470  int user_list_pool_reload(int online_use Line 561  int user_list_pool_reload(int online_use
561          MYSQL *db = NULL;          MYSQL *db = NULL;
562          USER_LIST *p_tmp;          USER_LIST *p_tmp;
563          USER_ONLINE_LIST *p_online_tmp;          USER_ONLINE_LIST *p_online_tmp;
564            int ret = 0;
565    
566          if (p_user_list_pool == NULL)          if (p_user_list_pool == NULL)
567          {          {
# Line 489  int user_list_pool_reload(int online_use Line 581  int user_list_pool_reload(int online_use
581                  if (user_online_list_load(db, p_user_list_pool->p_online_new) < 0)                  if (user_online_list_load(db, p_user_list_pool->p_online_new) < 0)
582                  {                  {
583                          log_error("user_online_list_load() error\n");                          log_error("user_online_list_load() error\n");
584                          return -2;                          ret = -2;
585                            goto cleanup;
586                    }
587    
588                    if (user_login_count_load(db) < 0)
589                    {
590                            log_error("user_login_count_load() error\n");
591                            ret = -2;
592                            goto cleanup;
593                  }                  }
594          }          }
595          else          else
# Line 497  int user_list_pool_reload(int online_use Line 597  int user_list_pool_reload(int online_use
597                  if (user_list_load(db, p_user_list_pool->p_new) < 0)                  if (user_list_load(db, p_user_list_pool->p_new) < 0)
598                  {                  {
599                          log_error("user_list_load() error\n");                          log_error("user_list_load() error\n");
600                          return -2;                          ret = -2;
601                            goto cleanup;
602                  }                  }
603          }          }
604    
605          mysql_close(db);          mysql_close(db);
606            db = NULL;
607    
608          if (user_list_rw_lock(p_user_list_pool->semid) < 0)          if (user_list_rw_lock(p_user_list_pool->semid) < 0)
609          {          {
610                  log_error("user_list_rw_lock() error\n");                  log_error("user_list_rw_lock() error\n");
611                  return -3;                  ret = -3;
612                    goto cleanup;
613          }          }
614    
615          if (online_user)          if (online_user)
# Line 527  int user_list_pool_reload(int online_use Line 630  int user_list_pool_reload(int online_use
630          if (user_list_rw_unlock(p_user_list_pool->semid) < 0)          if (user_list_rw_unlock(p_user_list_pool->semid) < 0)
631          {          {
632                  log_error("user_list_rw_unlock() error\n");                  log_error("user_list_rw_unlock() error\n");
633                  return -3;                  ret = -3;
634                    goto cleanup;
635          }          }
636    
637          return 0;  cleanup:
638            mysql_close(db);
639    
640            return ret;
641  }  }
642    
643  int user_list_try_rd_lock(int semid, int wait_sec)  int user_list_try_rd_lock(int semid, int wait_sec)
644  {  {
645          struct sembuf sops[2];          struct sembuf sops[2];
646    #if !defined(__MSYS__) && !defined(__MINGW32__)
647          struct timespec timeout;          struct timespec timeout;
648    #endif
649          int ret;          int ret;
650    
651          sops[0].sem_num = 1; // w_sem          sops[0].sem_num = 1; // w_sem
# Line 547  int user_list_try_rd_lock(int semid, int Line 656  int user_list_try_rd_lock(int semid, int
656          sops[1].sem_op = 1;                     // lock          sops[1].sem_op = 1;                     // lock
657          sops[1].sem_flg = SEM_UNDO; // undo on terminate          sops[1].sem_flg = SEM_UNDO; // undo on terminate
658    
659    #if defined(__MSYS__) || defined(__MINGW32__)
660            ret = semop(semid, sops, 2);
661    #else
662          timeout.tv_sec = wait_sec;          timeout.tv_sec = wait_sec;
663          timeout.tv_nsec = 0;          timeout.tv_nsec = 0;
664    
665          ret = semtimedop(semid, sops, 2, &timeout);          ret = semtimedop(semid, sops, 2, &timeout);
666    #endif
667          if (ret == -1 && errno != EAGAIN && errno != EINTR)          if (ret == -1 && errno != EAGAIN && errno != EINTR)
668          {          {
669                  log_error("semtimedop(lock read) error %d\n", errno);                  log_error("semop(lock read) error %d\n", errno);
670          }          }
671    
672          return ret;          return ret;
# Line 562  int user_list_try_rd_lock(int semid, int Line 675  int user_list_try_rd_lock(int semid, int
675  int user_list_try_rw_lock(int semid, int wait_sec)  int user_list_try_rw_lock(int semid, int wait_sec)
676  {  {
677          struct sembuf sops[3];          struct sembuf sops[3];
678    #if !defined(__MSYS__) && !defined(__MINGW32__)
679          struct timespec timeout;          struct timespec timeout;
680    #endif
681          int ret;          int ret;
682    
683          sops[0].sem_num = 1; // w_sem          sops[0].sem_num = 1; // w_sem
# Line 577  int user_list_try_rw_lock(int semid, int Line 692  int user_list_try_rw_lock(int semid, int
692          sops[2].sem_op = 0;      // wait until unlocked          sops[2].sem_op = 0;      // wait until unlocked
693          sops[2].sem_flg = 0;          sops[2].sem_flg = 0;
694    
695    #if defined(__MSYS__) || defined(__MINGW32__)
696            ret = semop(semid, sops, 3);
697    #else
698          timeout.tv_sec = wait_sec;          timeout.tv_sec = wait_sec;
699          timeout.tv_nsec = 0;          timeout.tv_nsec = 0;
700    
701          ret = semtimedop(semid, sops, 3, &timeout);          ret = semtimedop(semid, sops, 3, &timeout);
702    #endif
703          if (ret == -1 && errno != EAGAIN && errno != EINTR)          if (ret == -1 && errno != EAGAIN && errno != EINTR)
704          {          {
705                  log_error("semtimedop(lock write) error %d\n", errno);                  log_error("semop(lock write) error %d\n", errno);
706          }          }
707    
708          return ret;          return ret;
# Line 695  int query_user_list(int page_id, USER_IN Line 814  int query_user_list(int page_id, USER_IN
814                  return -1;                  return -1;
815          }          }
816    
817            *p_user_count = 0;
818            *p_page_count = 0;
819    
820          // acquire lock of user list          // acquire lock of user list
821          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock(p_user_list_pool->semid) < 0)
822          {          {
# Line 709  int query_user_list(int page_id, USER_IN Line 831  int query_user_list(int page_id, USER_IN
831                  goto cleanup;                  goto cleanup;
832          }          }
833    
834          *p_page_count = p_user_list_pool->p_current->user_count / BBS_user_limit_per_page +          *p_page_count = (p_user_list_pool->p_current->user_count + BBS_user_limit_per_page - 1) /
835                                          (p_user_list_pool->p_current->user_count % BBS_user_limit_per_page == 0 ? 0 : 1);                                          BBS_user_limit_per_page;
836    
837          if (page_id < 0 || page_id >= *p_page_count)          if (page_id < 0 || page_id >= *p_page_count)
838          {          {
# Line 748  int query_user_online_list(int page_id, Line 870  int query_user_online_list(int page_id,
870                  return -1;                  return -1;
871          }          }
872    
873            *p_user_count = 0;
874            *p_page_count = 0;
875    
876          // acquire lock of user list          // acquire lock of user list
877          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock(p_user_list_pool->semid) < 0)
878          {          {
# Line 762  int query_user_online_list(int page_id, Line 887  int query_user_online_list(int page_id,
887                  goto cleanup;                  goto cleanup;
888          }          }
889    
890          *p_page_count = p_user_list_pool->p_online_current->user_count / BBS_user_limit_per_page +          *p_page_count = (p_user_list_pool->p_online_current->user_count + BBS_user_limit_per_page - 1) / BBS_user_limit_per_page;
                                         (p_user_list_pool->p_online_current->user_count % BBS_user_limit_per_page == 0 ? 0 : 1);  
891    
892          if (page_id < 0 || page_id >= *p_page_count)          if (page_id < 0 || page_id >= *p_page_count)
893          {          {
# Line 791  cleanup: Line 915  cleanup:
915          return ret;          return ret;
916  }  }
917    
918    int get_user_list_count(int *p_user_cnt)
919    {
920            if (p_user_cnt == NULL)
921            {
922                    log_error("NULL pointer error\n");
923                    return -1;
924            }
925    
926            // acquire lock of user list
927            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
928            {
929                    log_error("user_list_rd_lock() error\n");
930                    return -2;
931            }
932    
933            *p_user_cnt = p_user_list_pool->p_current->user_count;
934    
935            // release lock of user list
936            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
937            {
938                    log_error("user_list_rd_unlock() error\n");
939                    return -2;
940            }
941    
942            return 0;
943    }
944    
945    int get_user_online_list_count(int *p_user_cnt, int *p_guest_cnt)
946    {
947            if (p_user_cnt == NULL || p_guest_cnt == NULL)
948            {
949                    log_error("NULL pointer error\n");
950                    return -1;
951            }
952    
953            // acquire lock of user list
954            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
955            {
956                    log_error("user_list_rd_lock() error\n");
957                    return -2;
958            }
959    
960            *p_user_cnt = p_user_list_pool->p_online_current->user_count;
961            *p_guest_cnt = p_user_list_pool->p_online_current->guest_count;
962    
963            // release lock of user list
964            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
965            {
966                    log_error("user_list_rd_unlock() error\n");
967                    return -2;
968            }
969    
970            return 0;
971    }
972    
973    int get_user_login_count(int *p_login_cnt)
974    {
975            if (p_login_cnt == NULL)
976            {
977                    log_error("NULL pointer error\n");
978                    return -1;
979            }
980    
981            *p_login_cnt = p_user_list_pool->user_login_count;
982    
983            return 0;
984    }
985    
986  int query_user_info(int32_t id, USER_INFO *p_user)  int query_user_info(int32_t id, USER_INFO *p_user)
987  {  {
988          int ret = 0;          int ret = 0;
# Line 824  int query_user_info(int32_t id, USER_INF Line 1016  int query_user_info(int32_t id, USER_INF
1016          return ret;          return ret;
1017  }  }
1018    
1019  int query_user_info_by_uid(int32_t uid, USER_INFO *p_user)  int query_user_info_by_uid(int32_t uid, USER_INFO *p_user, char *p_intro_buf, size_t intro_buf_len)
1020  {  {
1021          int left;          int left;
1022          int right;          int right;
# Line 853  int query_user_info_by_uid(int32_t uid, Line 1045  int query_user_info_by_uid(int32_t uid,
1045                  mid = (left + right) / 2;                  mid = (left + right) / 2;
1046                  if (uid < p_user_list_pool->p_current->index_uid[mid].uid)                  if (uid < p_user_list_pool->p_current->index_uid[mid].uid)
1047                  {                  {
1048                          right = mid;                          right = mid - 1;
1049                  }                  }
1050                  else if (uid > p_user_list_pool->p_current->index_uid[mid].uid)                  else if (uid > p_user_list_pool->p_current->index_uid[mid].uid)
1051                  {                  {
# Line 869  int query_user_info_by_uid(int32_t uid, Line 1061  int query_user_info_by_uid(int32_t uid,
1061          if (uid == p_user_list_pool->p_current->index_uid[left].uid) // Found          if (uid == p_user_list_pool->p_current->index_uid[left].uid) // Found
1062          {          {
1063                  id = p_user_list_pool->p_current->index_uid[left].id;                  id = p_user_list_pool->p_current->index_uid[left].id;
1064                  if ((ret = query_user_info(id, p_user)) <= 0)                  *p_user = p_user_list_pool->p_current->users[id];
1065                    ret = 1;
1066    
1067                    if (p_intro_buf != NULL)
1068                  {                  {
1069                          log_error("query_user_info(id=%d) error: %d\n", id, ret);                          strncpy(p_intro_buf, p_user_list_pool->p_current->users[id].intro, intro_buf_len - 1);
1070                            p_intro_buf[intro_buf_len - 1] = '\0';
1071                            p_user->intro = p_intro_buf;
1072                  }                  }
1073                  else          }
1074    
1075            // release lock of user list
1076            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
1077            {
1078                    log_error("user_list_rd_unlock() error\n");
1079                    ret = -1;
1080            }
1081    
1082            return ret;
1083    }
1084    
1085    int query_user_info_by_username(const char *username_prefix, int max_user_cnt,
1086                                                                    int32_t uid_list[], char username_list[][BBS_username_max_len + 1])
1087    {
1088            int left;
1089            int right;
1090            int mid;
1091            int left_save;
1092            int ret = 0;
1093            size_t prefix_len;
1094            int comp;
1095            int i;
1096    
1097            if (username_prefix == NULL || uid_list == NULL || username_list == NULL)
1098            {
1099                    log_error("NULL pointer error\n");
1100                    return -1;
1101            }
1102    
1103            prefix_len = strlen(username_prefix);
1104    
1105            // acquire lock of user list
1106            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
1107            {
1108                    log_error("user_list_rd_lock() error\n");
1109                    return -2;
1110            }
1111    
1112            left = 0;
1113            right = p_user_list_pool->p_current->user_count - 1;
1114    
1115            while (left < right)
1116            {
1117                    mid = (left + right) / 2;
1118                    comp = strncasecmp(username_prefix, p_user_list_pool->p_current->users[mid].username, prefix_len);
1119                    if (comp < 0)
1120                  {                  {
1121                          ret = 1;                          right = mid - 1;
1122                    }
1123                    else if (comp > 0)
1124                    {
1125                            left = mid + 1;
1126                    }
1127                    else // if (comp == 0)
1128                    {
1129                            left = mid;
1130                            break;
1131                    }
1132            }
1133    
1134            if (strncasecmp(username_prefix, p_user_list_pool->p_current->users[left].username, prefix_len) == 0) // Found
1135            {
1136    #ifdef _DEBUG
1137                    log_error("Debug: match found, pos=%d\n", left);
1138    #endif
1139    
1140                    left_save = left;
1141                    right = left;
1142                    left = 0;
1143    
1144                    while (left < right)
1145                    {
1146                            mid = (left + right) / 2;
1147                            comp = strncasecmp(username_prefix, p_user_list_pool->p_current->users[mid].username, prefix_len);
1148                            if (comp > 0)
1149                            {
1150                                    left = mid + 1;
1151                            }
1152                            else if (comp == 0)
1153                            {
1154                                    right = mid;
1155                            }
1156                            else // if (comp < 0)
1157                            {
1158                                    log_error("Bug: left=%d right=%d mid=%d");
1159                                    ret = -2;
1160                                    goto cleanup;
1161                            }
1162                    }
1163    
1164    #ifdef _DEBUG
1165                    log_error("Debug: first match found, pos=%d\n", right);
1166    #endif
1167    
1168                    left = left_save;
1169                    left_save = right;
1170                    right = p_user_list_pool->p_current->user_count - 1;
1171    
1172                    while (left < right)
1173                    {
1174                            mid = (left + right) / 2 + (left + right) % 2;
1175                            comp = strncasecmp(username_prefix, p_user_list_pool->p_current->users[mid].username, prefix_len);
1176                            if (comp < 0)
1177                            {
1178                                    right = mid - 1;
1179                            }
1180                            else if (comp == 0)
1181                            {
1182                                    left = mid;
1183                            }
1184                            else // if (comp > 0)
1185                            {
1186                                    log_error("Bug: left=%d right=%d mid=%d");
1187                                    ret = -2;
1188                                    goto cleanup;
1189                            }
1190                  }                  }
1191    
1192    #ifdef _DEBUG
1193                    log_error("Debug: last match found, pos=%d\n", left);
1194    #endif
1195    
1196                    right = left;
1197                    left = left_save;
1198    
1199                    for (i = 0; i < max_user_cnt && left + i <= right; i++)
1200                    {
1201                            uid_list[i] = p_user_list_pool->p_current->users[left + i].uid;
1202                            strncpy(username_list[i], p_user_list_pool->p_current->users[left + i].username, sizeof(username_list[i]) - 1);
1203                            username_list[i][sizeof(username_list[i]) - 1] = '\0';
1204                    }
1205                    ret = i;
1206          }          }
1207    
1208    cleanup:
1209          // release lock of user list          // release lock of user list
1210          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
1211          {          {
# Line 921  int query_user_online_info(int32_t id, U Line 1248  int query_user_online_info(int32_t id, U
1248    
1249          return ret;          return ret;
1250  }  }
1251    
1252    int query_user_online_info_by_uid(int32_t uid, USER_ONLINE_INFO *p_users, int *p_user_cnt, int start_id)
1253    {
1254            int left;
1255            int right;
1256            int mid;
1257            int32_t id;
1258            int ret = 0;
1259            int i;
1260            int user_cnt;
1261    
1262            if (p_users == NULL || p_user_cnt == NULL)
1263            {
1264                    log_error("NULL pointer error\n");
1265                    return -1;
1266            }
1267    
1268            user_cnt = *p_user_cnt;
1269            *p_user_cnt = 0;
1270    
1271            // acquire lock of user list
1272            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
1273            {
1274                    log_error("user_list_rd_lock() error\n");
1275                    return -2;
1276            }
1277    
1278            left = start_id;
1279            right = p_user_list_pool->p_online_current->user_count - 1;
1280    
1281            while (left < right)
1282            {
1283                    mid = (left + right) / 2;
1284                    if (uid < p_user_list_pool->p_online_current->index_uid[mid].uid)
1285                    {
1286                            right = mid - 1;
1287                    }
1288                    else if (uid > p_user_list_pool->p_online_current->index_uid[mid].uid)
1289                    {
1290                            left = mid + 1;
1291                    }
1292                    else // if (uid == p_user_list_pool->p_online_current->index_uid[mid].uid)
1293                    {
1294                            left = mid;
1295                            break;
1296                    }
1297            }
1298    
1299            if (uid == p_user_list_pool->p_online_current->index_uid[left].uid)
1300            {
1301                    right = left;
1302                    left = start_id;
1303    
1304                    while (left < right)
1305                    {
1306                            mid = (left + right) / 2;
1307                            if (uid - 1 < p_user_list_pool->p_online_current->index_uid[mid].uid)
1308                            {
1309                                    right = mid;
1310                            }
1311                            else // if (uid - 1 >= p_user_list_pool->p_online_current->index_uid[mid].uid)
1312                            {
1313                                    left = mid + 1;
1314                            }
1315                    }
1316    
1317                    for (i = 0;
1318                             left < p_user_list_pool->p_online_current->user_count && i < user_cnt &&
1319                             uid == p_user_list_pool->p_online_current->index_uid[left].uid;
1320                             left++, i++)
1321                    {
1322                            id = p_user_list_pool->p_online_current->index_uid[left].id;
1323                            p_users[i] = p_user_list_pool->p_online_current->users[id];
1324                    }
1325    
1326                    if (i > 0)
1327                    {
1328                            *p_user_cnt = i;
1329                            ret = 1;
1330                    }
1331            }
1332    
1333            // release lock of user list
1334            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
1335            {
1336                    log_error("user_list_rd_unlock() error\n");
1337                    ret = -1;
1338            }
1339    
1340            return ret;
1341    }
1342    
1343    int get_user_id_list(int32_t *p_uid_list, int *p_user_cnt, int start_uid)
1344    {
1345            int left;
1346            int right;
1347            int mid;
1348            int ret = 0;
1349            int i;
1350    
1351            if (p_uid_list == NULL || p_user_cnt == NULL)
1352            {
1353                    log_error("NULL pointer error\n");
1354                    return -1;
1355            }
1356    
1357            // acquire lock of user list
1358            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
1359            {
1360                    log_error("user_list_rd_lock() error\n");
1361                    return -2;
1362            }
1363    
1364            left = 0;
1365            right = p_user_list_pool->p_current->user_count - 1;
1366    
1367            while (left < right)
1368            {
1369                    mid = (left + right) / 2;
1370                    if (start_uid < p_user_list_pool->p_current->index_uid[mid].uid)
1371                    {
1372                            right = mid - 1;
1373                    }
1374                    else if (start_uid > p_user_list_pool->p_current->index_uid[mid].uid)
1375                    {
1376                            left = mid + 1;
1377                    }
1378                    else // if (start_uid == p_user_list_pool->p_current->index_uid[mid].uid)
1379                    {
1380                            left = mid;
1381                            break;
1382                    }
1383            }
1384    
1385            for (i = 0; i < *p_user_cnt && left + i < p_user_list_pool->p_current->user_count; i++)
1386            {
1387                    p_uid_list[i] = p_user_list_pool->p_current->index_uid[left + i].uid;
1388            }
1389            *p_user_cnt = i;
1390    
1391            // release lock of user list
1392            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
1393            {
1394                    log_error("user_list_rd_unlock() error\n");
1395                    ret = -1;
1396            }
1397    
1398            return ret;
1399    }
1400    
1401    int user_stat_update(void)
1402    {
1403            return user_stat_map_update(&(p_user_list_pool->user_stat_map));
1404    }
1405    
1406    int user_article_cnt_inc(int32_t uid, int n)
1407    {
1408            return user_stat_article_cnt_inc(&(p_user_list_pool->user_stat_map), uid, n);
1409    }
1410    
1411    int get_user_article_cnt(int32_t uid)
1412    {
1413            const USER_STAT *p_stat;
1414            int ret;
1415    
1416            ret = user_stat_get(&(p_user_list_pool->user_stat_map), uid, &p_stat);
1417            if (ret < 0)
1418            {
1419                    log_error("user_stat_get(uid=%d) error: %d\n", uid, ret);
1420                    return -1;
1421            }
1422            else if (ret == 0) // user not found
1423            {
1424                    return -1;
1425            }
1426    
1427            return p_stat->article_count;
1428    }

Legend:
Removed lines/characters  
Changed lines/characters
  Added lines/characters
webmaster@leafok.com
 ViewVC Help
Powered by ViewVC 1.3.0-beta1