/[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.9 by sysadm, Wed Oct 22 07:39:02 2025 UTC Revision 1.39 by sysadm, Thu Nov 20 01:54:18 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 <fcntl.h>
21  #include <stdlib.h>  #include <stdlib.h>
22  #include <string.h>  #include <string.h>
23  #include <time.h>  #include <time.h>
 #include <sys/ipc.h>  
24  #include <sys/mman.h>  #include <sys/mman.h>
25  #include <sys/param.h>  #include <sys/param.h>
26  #include <sys/sem.h>  #include <sys/sem.h>
27  #include <sys/shm.h>  #include <sys/stat.h>
28    
29  #ifdef _SEM_SEMUN_UNDEFINED  #if defined(_SEM_SEMUN_UNDEFINED) || defined(__CYGWIN__)
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  {  {
48          int shmid;          size_t shm_size;
49          int semid;          int semid;
50          USER_LIST user_list[2];          USER_LIST user_list[2];
51          USER_LIST *p_current;          int user_list_index_current;
52          USER_LIST *p_new;          int user_list_index_new;
53          USER_ONLINE_LIST user_online_list[2];          USER_ONLINE_LIST user_online_list[2];
54          USER_ONLINE_LIST *p_online_current;          int user_online_list_index_current;
55          USER_ONLINE_LIST *p_online_new;          int user_online_list_index_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    
61    static char user_list_shm_name[FILE_PATH_LEN];
62  static USER_LIST_POOL *p_user_list_pool = NULL;  static USER_LIST_POOL *p_user_list_pool = NULL;
63  static TRIE_NODE *p_trie_action_dict = NULL;  static TRIE_NODE *p_trie_action_dict = NULL;
64    
# Line 83  const USER_ACTION_MAP user_action_map[] Line 86  const USER_ACTION_MAP user_action_map[]
86    
87  const int user_action_map_size = sizeof(user_action_map) / sizeof(USER_ACTION_MAP);  const int user_action_map_size = sizeof(user_action_map) / sizeof(USER_ACTION_MAP);
88    
89  static int user_list_try_rd_lock(int semid, int wait_sec);  static int user_list_try_rd_lock(int wait_sec);
90  static int user_list_try_rw_lock(int semid, int wait_sec);  static int user_list_try_rw_lock(int wait_sec);
91  static int user_list_rd_unlock(int semid);  static int user_list_rd_unlock(void);
92  static int user_list_rw_unlock(int semid);  static int user_list_rw_unlock(void);
93  static int user_list_rd_lock(int semid);  static int user_list_rd_lock(void);
94  static int user_list_rw_lock(int semid);  static int user_list_rw_lock(void);
95    
96  static int user_list_load(MYSQL *db, USER_LIST *p_list);  static int user_list_load(MYSQL *db, USER_LIST *p_list);
97  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);
98    static int user_login_count_load(MYSQL *db);
99    
100  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)
101  {  {
# Line 106  static int user_info_index_uid_comp(cons Line 110  static int user_info_index_uid_comp(cons
110          {          {
111                  return 1;                  return 1;
112          }          }
113            else if (p1->id < p2->id)
114            {
115                    return -1;
116            }
117            else if (p1->id > p2->id)
118            {
119                    return 1;
120            }
121          return 0;          return 0;
122  }  }
123    
# Line 116  int user_list_load(MYSQL *db, USER_LIST Line 128  int user_list_load(MYSQL *db, USER_LIST
128          char sql[SQL_BUFFER_LEN];          char sql[SQL_BUFFER_LEN];
129          int ret = 0;          int ret = 0;
130          int i;          int i;
131            int j;
132            int32_t last_uid;
133            size_t intro_buf_offset;
134            size_t intro_len;
135    
136          if (db == NULL || p_list == NULL)          if (db == NULL || p_list == NULL)
137          {          {
# Line 123  int user_list_load(MYSQL *db, USER_LIST Line 139  int user_list_load(MYSQL *db, USER_LIST
139                  return -1;                  return -1;
140          }          }
141    
142            if (p_list->user_count > 0)
143            {
144                    last_uid = p_list->users[p_list->user_count - 1].uid;
145            }
146            else
147            {
148                    last_uid = -1;
149            }
150    
151          snprintf(sql, sizeof(sql),          snprintf(sql, sizeof(sql),
152                           "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, "
153                           "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), "
154                             "UNIX_TIMESTAMP(birthday), `introduction` "
155                           "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 "
156                           "INNER JOIN user_reginfo ON user_list.UID = user_reginfo.UID "                           "INNER JOIN user_reginfo ON user_list.UID = user_reginfo.UID "
157                           "WHERE enable ORDER BY username");                           "WHERE enable ORDER BY username");
# Line 144  int user_list_load(MYSQL *db, USER_LIST Line 170  int user_list_load(MYSQL *db, USER_LIST
170                  goto cleanup;                  goto cleanup;
171          }          }
172    
173            intro_buf_offset = 0;
174          i = 0;          i = 0;
175          while ((row = mysql_fetch_row(rs)))          while ((row = mysql_fetch_row(rs)))
176          {          {
# Line 158  int user_list_load(MYSQL *db, USER_LIST Line 185  int user_list_load(MYSQL *db, USER_LIST
185                  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]));
186                  p_list->users[i].life = (row[5] == NULL ? 0 : atoi(row[5]));                  p_list->users[i].life = (row[5] == NULL ? 0 : atoi(row[5]));
187                  p_list->users[i].exp = (row[6] == NULL ? 0 : atoi(row[6]));                  p_list->users[i].exp = (row[6] == NULL ? 0 : atoi(row[6]));
188                  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]));
189                  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]));
190                  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]));
191                    p_list->users[i].last_logout_dt = (row[10] == NULL ? 0 : atol(row[10]));
192                  // index                  p_list->users[i].birthday = (row[11] == NULL ? 0 : atol(row[11]));
193                  p_list->index_uid[i].uid = p_list->users[i].uid;                  intro_len = strlen((row[12] == NULL ? "" : row[12]));
194                  p_list->index_uid[i].id = i;                  if (intro_len >= sizeof(p_list->user_intro_buf) - 1 - intro_buf_offset)
195                    {
196                            log_error("OOM for user introduction: len=%d, i=%d\n", intro_len, i);
197                            break;
198                    }
199                    memcpy(p_list->user_intro_buf + intro_buf_offset,
200                               (row[12] == NULL ? "" : row[12]),
201                               intro_len + 1);
202                    p_list->users[i].intro = p_list->user_intro_buf + intro_buf_offset;
203                    intro_buf_offset += (intro_len + 1);
204    
205                  i++;                  i++;
206                  if (i >= BBS_max_user_count)                  if (i >= BBS_max_user_count)
# Line 176  int user_list_load(MYSQL *db, USER_LIST Line 212  int user_list_load(MYSQL *db, USER_LIST
212          mysql_free_result(rs);          mysql_free_result(rs);
213          rs = NULL;          rs = NULL;
214    
215          p_list->user_count = i;          if (i != p_list->user_count || p_list->users[i - 1].uid != last_uid) // Count of users changed
216            {
217                    // Rebuild index
218                    for (j = 0; j < i; j++)
219                    {
220                            p_list->index_uid[j].uid = p_list->users[j].uid;
221                            p_list->index_uid[j].id = j;
222                    }
223    
224          // Sort index                  qsort(p_list->index_uid, (size_t)i, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp);
225          qsort(p_list->index_uid, (size_t)i, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp);  
226    #ifdef _DEBUG
227                    log_error("Rebuild index of %d users, last_uid=%d\n", i, p_list->users[i - 1].uid);
228    #endif
229            }
230    
231            p_list->user_count = i;
232    
233  #ifdef _DEBUG  #ifdef _DEBUG
234          log_error("Loaded %d users\n", p_list->user_count);          log_error("Loaded %d users\n", p_list->user_count);
# Line 191  cleanup: Line 240  cleanup:
240          return ret;          return ret;
241  }  }
242    
243  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)
244  {  {
245          MYSQL_RES *rs = NULL;          MYSQL_RES *rs = NULL;
246          MYSQL_ROW row;          MYSQL_ROW row;
247          char sql[SQL_BUFFER_LEN];          char sql[SQL_BUFFER_LEN];
248          int ret = 0;          int ret = 0;
249          int i;          int i;
250            int j;
251            int user_cnt;
252            int guest_cnt;
253    
254          if (db == NULL || p_list == NULL)          if (db == NULL || p_online_list == NULL)
255          {          {
256                  log_error("NULL pointer error\n");                  log_error("NULL pointer error\n");
257                  return -1;                  return -1;
# Line 208  int user_online_list_load(MYSQL *db, USE Line 260  int user_online_list_load(MYSQL *db, USE
260          snprintf(sql, sizeof(sql),          snprintf(sql, sizeof(sql),
261                           "SELECT SID, UID, ip, current_action, UNIX_TIMESTAMP(login_tm), "                           "SELECT SID, UID, ip, current_action, UNIX_TIMESTAMP(login_tm), "
262                           "UNIX_TIMESTAMP(last_tm) FROM user_online "                           "UNIX_TIMESTAMP(last_tm) FROM user_online "
263                           "WHERE last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND) AND UID <> 0 "                           "WHERE last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND) "
264                           "ORDER BY last_tm DESC",                           "ORDER BY last_tm DESC",
265                           BBS_user_off_line);                           BBS_user_off_line);
266    
# Line 227  int user_online_list_load(MYSQL *db, USE Line 279  int user_online_list_load(MYSQL *db, USE
279          }          }
280    
281          i = 0;          i = 0;
282            user_cnt = 0;
283            guest_cnt = 0;
284          while ((row = mysql_fetch_row(rs)))          while ((row = mysql_fetch_row(rs)))
285          {          {
286                  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)  
287                  {                  {
288                          log_error("query_user_info(%d) error\n", atoi(row[1]));                          guest_cnt++;
289                          continue;                          continue;
290                  }                  }
291                  else if (ret == 0) // skip Guest                  else
292                    {
293                            user_cnt++;
294                    }
295    
296                    p_online_list->users[i].id = i;
297                    strncpy(p_online_list->users[i].session_id, row[0], sizeof(p_online_list->users[i].session_id) - 1);
298                    p_online_list->users[i].session_id[sizeof(p_online_list->users[i].session_id) - 1] = '\0';
299    
300                    if ((ret = query_user_info_by_uid(atoi(row[1]), &(p_online_list->users[i].user_info), NULL, 0)) <= 0)
301                  {                  {
302                            log_error("query_user_info_by_uid(%d) error\n", atoi(row[1]));
303                          continue;                          continue;
304                  }                  }
305    
306                  strncpy(p_list->users[i].ip, row[2], sizeof(p_list->users[i].ip) - 1);                  strncpy(p_online_list->users[i].ip, row[2], sizeof(p_online_list->users[i].ip) - 1);
307                  p_list->users[i].ip[sizeof(p_list->users[i].ip) - 1] = '\0';                  p_online_list->users[i].ip[sizeof(p_online_list->users[i].ip) - 1] = '\0';
308    
309                  strncpy(p_list->users[i].current_action, row[3], sizeof(p_list->users[i].current_action) - 1);                  strncpy(p_online_list->users[i].current_action, row[3], sizeof(p_online_list->users[i].current_action) - 1);
310                  p_list->users[i].current_action[sizeof(p_list->users[i].current_action) - 1] = '\0';                  p_online_list->users[i].current_action[sizeof(p_online_list->users[i].current_action) - 1] = '\0';
311                  p_list->users[i].current_action_title = NULL;                  p_online_list->users[i].current_action_title = NULL;
312                  if (p_list->users[i].current_action[0] == '\0')                  if (p_online_list->users[i].current_action[0] == '\0')
313                  {                  {
314                          p_list->users[i].current_action_title = "";                          p_online_list->users[i].current_action_title = "";
315                  }                  }
316                  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)
317                  {                  {
318                          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",
319                                            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);
320                          continue;                          continue;
321                  }                  }
322    
323                  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]));
324                  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]));
325    
326                  i++;                  i++;
327                  if (i >= BBS_max_user_online_count)                  if (i >= BBS_max_user_online_count)
# Line 273  int user_online_list_load(MYSQL *db, USE Line 333  int user_online_list_load(MYSQL *db, USE
333          mysql_free_result(rs);          mysql_free_result(rs);
334          rs = NULL;          rs = NULL;
335    
336          p_list->user_count = i;          if (user_cnt > 0)
337            {
338                    // Rebuild index
339                    for (j = 0; j < user_cnt; j++)
340                    {
341                            p_online_list->index_uid[j].uid = p_online_list->users[j].user_info.uid;
342                            p_online_list->index_uid[j].id = j;
343                    }
344    
345  #ifdef _DEBUG                  qsort(p_online_list->index_uid, (size_t)user_cnt, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp);
346          log_error("Loaded %d users\n", p_list->user_count);          }
347  #endif  
348            p_online_list->user_count = user_cnt;
349            p_online_list->guest_count = guest_cnt;
350    
351  cleanup:  cleanup:
352          mysql_free_result(rs);          mysql_free_result(rs);
# Line 285  cleanup: Line 354  cleanup:
354          return ret;          return ret;
355  }  }
356    
357  int user_list_pool_init(void)  int user_login_count_load(MYSQL *db)
358  {  {
359          int shmid;          MYSQL_RES *rs = NULL;
360          int semid;          MYSQL_ROW row;
361          int proj_id;          char sql[SQL_BUFFER_LEN];
362          key_t key;  
363            if (db == NULL)
364            {
365                    log_error("NULL pointer error\n");
366                    return -1;
367            }
368    
369            snprintf(sql, sizeof(sql),
370                             "SELECT ID FROM user_login_log ORDER BY ID DESC LIMIT 1");
371            if (mysql_query(db, sql) != 0)
372            {
373                    log_error("Query user_login_log error: %s\n", mysql_error(db));
374                    return -2;
375            }
376            if ((rs = mysql_store_result(db)) == NULL)
377            {
378                    log_error("Get user_login_log data failed\n");
379                    return -2;
380            }
381            if ((row = mysql_fetch_row(rs)))
382            {
383                    p_user_list_pool->user_login_count = atoi(row[0]);
384            }
385            mysql_free_result(rs);
386    
387            return 0;
388    }
389    
390    int user_list_pool_init(const char *filename)
391    {
392            char filepath[FILE_PATH_LEN];
393            int fd;
394          size_t size;          size_t size;
395          void *p_shm;          void *p_shm;
396            int proj_id;
397            key_t key;
398            int semid;
399          union semun arg;          union semun arg;
400          int i;          int i;
401    
# Line 318  int user_list_pool_init(void) Line 421  int user_list_pool_init(void)
421          }          }
422    
423          // Allocate shared memory          // Allocate shared memory
424          proj_id = (int)(time(NULL) % getpid());          size = sizeof(USER_LIST_POOL);
425          key = ftok(VAR_USER_LIST_SHM, proj_id);  
426          if (key == -1)          strncpy(filepath, filename, sizeof(filepath) - 1);
427            filepath[sizeof(filepath) - 1] = '\0';
428            snprintf(user_list_shm_name, sizeof(user_list_shm_name), "/USER_LIST_SHM_%s", basename(filepath));
429    
430            if (shm_unlink(user_list_shm_name) == -1 && errno != ENOENT)
431          {          {
432                  log_error("ftok(%s %d) error (%d)\n", VAR_USER_LIST_SHM, proj_id, errno);                  log_error("shm_unlink(%s) error (%d)\n", user_list_shm_name, errno);
433                  return -2;                  return -2;
434          }          }
435    
436          size = sizeof(USER_LIST_POOL);          if ((fd = shm_open(user_list_shm_name, O_CREAT | O_EXCL | O_RDWR, 0600)) == -1)
         shmid = shmget(key, size, IPC_CREAT | IPC_EXCL | 0600);  
         if (shmid == -1)  
437          {          {
438                  log_error("shmget(size = %d) error (%d)\n", size, errno);                  log_error("shm_open(%s) error (%d)\n", user_list_shm_name, errno);
439                  return -3;                  return -2;
440          }          }
441          p_shm = shmat(shmid, NULL, 0);          if (ftruncate(fd, (off_t)size) == -1)
         if (p_shm == (void *)-1)  
442          {          {
443                  log_error("shmat(shmid=%d) error (%d)\n", shmid, errno);                  log_error("ftruncate(size=%d) error (%d)\n", size, errno);
444                  return -3;                  close(fd);
445                    return -2;
446            }
447    
448            p_shm = mmap(NULL, size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0L);
449            if (p_shm == MAP_FAILED)
450            {
451                    log_error("mmap() error (%d)\n", errno);
452                    close(fd);
453                    return -2;
454            }
455    
456            if (close(fd) < 0)
457            {
458                    log_error("close(fd) error (%d)\n", errno);
459                    return -1;
460          }          }
461    
462          p_user_list_pool = p_shm;          p_user_list_pool = p_shm;
463          p_user_list_pool->shmid = shmid;          p_user_list_pool->shm_size = size;
464    
465          // Allocate semaphore as user list pool lock          // Allocate semaphore as user list pool lock
466            proj_id = (int)(time(NULL) % getpid());
467            key = ftok(filename, proj_id);
468            if (key == -1)
469            {
470                    log_error("ftok(%s %d) error (%d)\n", filename, proj_id, errno);
471                    return -2;
472            }
473    
474          size = 2; // r_sem and w_sem          size = 2; // r_sem and w_sem
475          semid = semget(key, (int)size, IPC_CREAT | IPC_EXCL | 0600);          semid = semget(key, (int)size, IPC_CREAT | IPC_EXCL | 0600);
476          if (semid == -1)          if (semid == -1)
# Line 369  int user_list_pool_init(void) Line 496  int user_list_pool_init(void)
496          p_user_list_pool->user_list[0].user_count = 0;          p_user_list_pool->user_list[0].user_count = 0;
497          p_user_list_pool->user_list[1].user_count = 0;          p_user_list_pool->user_list[1].user_count = 0;
498    
499          p_user_list_pool->p_current = &(p_user_list_pool->user_list[0]);          p_user_list_pool->user_list_index_current = 0;
500          p_user_list_pool->p_new = &(p_user_list_pool->user_list[1]);          p_user_list_pool->user_list_index_new = 1;
501    
502            p_user_list_pool->user_online_list_index_current = 0;
503            p_user_list_pool->user_online_list_index_new = 1;
504    
505          p_user_list_pool->p_online_current = &(p_user_list_pool->user_online_list[0]);          user_stat_map_init(&(p_user_list_pool->user_stat_map));
         p_user_list_pool->p_online_new = &(p_user_list_pool->user_online_list[1]);  
506    
507          return 0;          return 0;
508  }  }
509    
510  void user_list_pool_cleanup(void)  int user_list_pool_cleanup(void)
511  {  {
         int shmid;  
   
512          if (p_user_list_pool == NULL)          if (p_user_list_pool == NULL)
513          {          {
514                  return;                  return -1;
         }  
   
         shmid = p_user_list_pool->shmid;  
   
         if (semctl(p_user_list_pool->semid, 0, IPC_RMID) == -1)  
         {  
                 log_error("semctl(semid = %d, IPC_RMID) error (%d)\n", p_user_list_pool->semid, errno);  
515          }          }
516    
517          if (shmdt(p_user_list_pool) == -1)          detach_user_list_pool_shm();
         {  
                 log_error("shmdt(shmid = %d) error (%d)\n", shmid, errno);  
         }  
518    
519          if (shmctl(shmid, IPC_RMID, NULL) == -1)          if (shm_unlink(user_list_shm_name) == -1 && errno != ENOENT)
520          {          {
521                  log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", shmid, errno);                  log_error("shm_unlink(%s) error (%d)\n", user_list_shm_name, errno);
522                    return -2;
523          }          }
524    
525          p_user_list_pool = NULL;          user_list_shm_name[0] = '\0';
526    
527          if (p_trie_action_dict != NULL)          if (p_trie_action_dict != NULL)
528          {          {
# Line 412  void user_list_pool_cleanup(void) Line 530  void user_list_pool_cleanup(void)
530    
531                  p_trie_action_dict = NULL;                  p_trie_action_dict = NULL;
532          }          }
533    
534            return 0;
535  }  }
536    
537  int set_user_list_pool_shm_readonly(void)  int set_user_list_pool_shm_readonly(void)
538  {  {
539          int shmid;          if (p_user_list_pool != NULL && mprotect(p_user_list_pool, p_user_list_pool->shm_size, PROT_READ) < 0)
         void *p_shm;  
   
         if (p_user_list_pool == NULL)  
540          {          {
541                  log_error("p_user_list_pool not initialized\n");                  log_error("mprotect() error (%d)\n", errno);
542                  return -1;                  return -1;
543          }          }
544    
         shmid = p_user_list_pool->shmid;  
   
         // Remap shared memory in read-only mode  
         p_shm = shmat(shmid, p_user_list_pool, SHM_RDONLY | SHM_REMAP);  
         if (p_shm == (void *)-1)  
         {  
                 log_error("shmat(user_list_pool shmid = %d) error (%d)\n", shmid, errno);  
                 return -3;  
         }  
   
         p_user_list_pool = p_shm;  
   
545          return 0;          return 0;
546  }  }
547    
548  int detach_user_list_pool_shm(void)  int detach_user_list_pool_shm(void)
549  {  {
550          if (p_user_list_pool != NULL && shmdt(p_user_list_pool) == -1)          if (p_user_list_pool != NULL && munmap(p_user_list_pool, p_user_list_pool->shm_size) < 0)
551          {          {
552                  log_error("shmdt(user_list_pool) error (%d)\n", errno);                  log_error("munmap() error (%d)\n", errno);
553                  return -1;                  return -1;
554          }          }
555    
# Line 456  int detach_user_list_pool_shm(void) Line 561  int detach_user_list_pool_shm(void)
561  int user_list_pool_reload(int online_user)  int user_list_pool_reload(int online_user)
562  {  {
563          MYSQL *db = NULL;          MYSQL *db = NULL;
564          USER_LIST *p_tmp;          int tmp;
565          USER_ONLINE_LIST *p_online_tmp;          int ret = 0;
566    
567          if (p_user_list_pool == NULL)          if (p_user_list_pool == NULL)
568          {          {
# Line 474  int user_list_pool_reload(int online_use Line 579  int user_list_pool_reload(int online_use
579    
580          if (online_user)          if (online_user)
581          {          {
582                  if (user_online_list_load(db, p_user_list_pool->p_online_new) < 0)                  if (user_online_list_load(db, &(p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_new])) < 0)
583                  {                  {
584                          log_error("user_online_list_load() error\n");                          log_error("user_online_list_load() error\n");
585                          return -2;                          ret = -2;
586                            goto cleanup;
587                    }
588    
589                    if (user_login_count_load(db) < 0)
590                    {
591                            log_error("user_login_count_load() error\n");
592                            ret = -2;
593                            goto cleanup;
594                  }                  }
595          }          }
596          else          else
597          {          {
598                  if (user_list_load(db, p_user_list_pool->p_new) < 0)                  if (user_list_load(db, &(p_user_list_pool->user_list[p_user_list_pool->user_list_index_new])) < 0)
599                  {                  {
600                          log_error("user_list_load() error\n");                          log_error("user_list_load() error\n");
601                          return -2;                          ret = -2;
602                            goto cleanup;
603                  }                  }
604          }          }
605    
606          mysql_close(db);          mysql_close(db);
607            db = NULL;
608    
609          if (user_list_rw_lock(p_user_list_pool->semid) < 0)          if (user_list_rw_lock() < 0)
610          {          {
611                  log_error("user_list_rw_lock() error\n");                  log_error("user_list_rw_lock() error\n");
612                  return -3;                  ret = -3;
613                    goto cleanup;
614          }          }
615    
616          if (online_user)          if (online_user)
617          {          {
618                  // Swap p_online_current and p_online_new                  // Swap p_online_current and p_online_new
619                  p_online_tmp = p_user_list_pool->p_online_current;                  tmp = p_user_list_pool->user_online_list_index_current;
620                  p_user_list_pool->p_online_current = p_user_list_pool->p_online_new;                  p_user_list_pool->user_online_list_index_current = p_user_list_pool->user_online_list_index_new;
621                  p_user_list_pool->p_online_new = p_online_tmp;                  p_user_list_pool->user_online_list_index_new = tmp;
622          }          }
623          else          else
624          {          {
625                  // Swap p_current and p_new                  // Swap index_current and index_new
626                  p_tmp = p_user_list_pool->p_current;                  tmp = p_user_list_pool->user_list_index_current;
627                  p_user_list_pool->p_current = p_user_list_pool->p_new;                  p_user_list_pool->user_list_index_current = p_user_list_pool->user_list_index_new;
628                  p_user_list_pool->p_new = p_tmp;                  p_user_list_pool->user_list_index_new = tmp;
629          }          }
630    
631          if (user_list_rw_unlock(p_user_list_pool->semid) < 0)          if (user_list_rw_unlock() < 0)
632          {          {
633                  log_error("user_list_rw_unlock() error\n");                  log_error("user_list_rw_unlock() error\n");
634                  return -3;                  ret = -3;
635                    goto cleanup;
636          }          }
637    
638          return 0;  cleanup:
639            mysql_close(db);
640    
641            return ret;
642  }  }
643    
644  int user_list_try_rd_lock(int semid, int wait_sec)  int user_list_try_rd_lock(int wait_sec)
645  {  {
646          struct sembuf sops[2];          struct sembuf sops[2];
647    #ifndef __CYGWIN__
648          struct timespec timeout;          struct timespec timeout;
649    #endif
650          int ret;          int ret;
651    
652            if (p_user_list_pool == NULL)
653            {
654                    log_error("p_user_list_pool not initialized\n");
655                    return -1;
656            }
657    
658          sops[0].sem_num = 1; // w_sem          sops[0].sem_num = 1; // w_sem
659          sops[0].sem_op = 0;      // wait until unlocked          sops[0].sem_op = 0;      // wait until unlocked
660          sops[0].sem_flg = 0;          sops[0].sem_flg = 0;
# Line 535  int user_list_try_rd_lock(int semid, int Line 663  int user_list_try_rd_lock(int semid, int
663          sops[1].sem_op = 1;                     // lock          sops[1].sem_op = 1;                     // lock
664          sops[1].sem_flg = SEM_UNDO; // undo on terminate          sops[1].sem_flg = SEM_UNDO; // undo on terminate
665    
666    #ifdef __CYGWIN__
667            ret = semop(p_user_list_pool->semid, sops, 2);
668    #else
669          timeout.tv_sec = wait_sec;          timeout.tv_sec = wait_sec;
670          timeout.tv_nsec = 0;          timeout.tv_nsec = 0;
671    
672          ret = semtimedop(semid, sops, 2, &timeout);          ret = semtimedop(p_user_list_pool->semid, sops, 2, &timeout);
673    #endif
674          if (ret == -1 && errno != EAGAIN && errno != EINTR)          if (ret == -1 && errno != EAGAIN && errno != EINTR)
675          {          {
676                  log_error("semtimedop(lock read) error %d\n", errno);                  log_error("semop(lock read) error %d\n", errno);
677          }          }
678    
679          return ret;          return ret;
680  }  }
681    
682  int user_list_try_rw_lock(int semid, int wait_sec)  int user_list_try_rw_lock(int wait_sec)
683  {  {
684          struct sembuf sops[3];          struct sembuf sops[3];
685    #ifndef __CYGWIN__
686          struct timespec timeout;          struct timespec timeout;
687    #endif
688          int ret;          int ret;
689    
690            if (p_user_list_pool == NULL)
691            {
692                    log_error("p_user_list_pool not initialized\n");
693                    return -1;
694            }
695    
696          sops[0].sem_num = 1; // w_sem          sops[0].sem_num = 1; // w_sem
697          sops[0].sem_op = 0;      // wait until unlocked          sops[0].sem_op = 0;      // wait until unlocked
698          sops[0].sem_flg = 0;          sops[0].sem_flg = 0;
# Line 565  int user_list_try_rw_lock(int semid, int Line 705  int user_list_try_rw_lock(int semid, int
705          sops[2].sem_op = 0;      // wait until unlocked          sops[2].sem_op = 0;      // wait until unlocked
706          sops[2].sem_flg = 0;          sops[2].sem_flg = 0;
707    
708    #ifdef __CYGWIN__
709            ret = semop(p_user_list_pool->semid, sops, 3);
710    #else
711          timeout.tv_sec = wait_sec;          timeout.tv_sec = wait_sec;
712          timeout.tv_nsec = 0;          timeout.tv_nsec = 0;
713    
714          ret = semtimedop(semid, sops, 3, &timeout);          ret = semtimedop(p_user_list_pool->semid, sops, 3, &timeout);
715    #endif
716          if (ret == -1 && errno != EAGAIN && errno != EINTR)          if (ret == -1 && errno != EAGAIN && errno != EINTR)
717          {          {
718                  log_error("semtimedop(lock write) error %d\n", errno);                  log_error("semop(lock write) error %d\n", errno);
719          }          }
720    
721          return ret;          return ret;
722  }  }
723    
724  int user_list_rd_unlock(int semid)  int user_list_rd_unlock(void)
725  {  {
726          struct sembuf sops[2];          struct sembuf sops[2];
727          int ret;          int ret;
728    
729            if (p_user_list_pool == NULL)
730            {
731                    log_error("p_user_list_pool not initialized\n");
732                    return -1;
733            }
734    
735          sops[0].sem_num = 0;                                     // r_sem          sops[0].sem_num = 0;                                     // r_sem
736          sops[0].sem_op = -1;                                     // unlock          sops[0].sem_op = -1;                                     // unlock
737          sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO; // no wait          sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO; // no wait
738    
739          ret = semop(semid, sops, 1);          ret = semop(p_user_list_pool->semid, sops, 1);
740          if (ret == -1 && errno != EAGAIN && errno != EINTR)          if (ret == -1 && errno != EAGAIN && errno != EINTR)
741          {          {
742                  log_error("semop(unlock read) error %d\n", errno);                  log_error("semop(unlock read) error %d\n", errno);
# Line 595  int user_list_rd_unlock(int semid) Line 745  int user_list_rd_unlock(int semid)
745          return ret;          return ret;
746  }  }
747    
748  int user_list_rw_unlock(int semid)  int user_list_rw_unlock(void)
749  {  {
750          struct sembuf sops[1];          struct sembuf sops[1];
751          int ret;          int ret;
752    
753            if (p_user_list_pool == NULL)
754            {
755                    log_error("p_user_list_pool not initialized\n");
756                    return -1;
757            }
758    
759          sops[0].sem_num = 1;                                     // w_sem          sops[0].sem_num = 1;                                     // w_sem
760          sops[0].sem_op = -1;                                     // unlock          sops[0].sem_op = -1;                                     // unlock
761          sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO; // no wait          sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO; // no wait
762    
763          ret = semop(semid, sops, 1);          ret = semop(p_user_list_pool->semid, sops, 1);
764          if (ret == -1 && errno != EAGAIN && errno != EINTR)          if (ret == -1 && errno != EAGAIN && errno != EINTR)
765          {          {
766                  log_error("semop(unlock write) error %d\n", errno);                  log_error("semop(unlock write) error %d\n", errno);
# Line 613  int user_list_rw_unlock(int semid) Line 769  int user_list_rw_unlock(int semid)
769          return ret;          return ret;
770  }  }
771    
772  int user_list_rd_lock(int semid)  int user_list_rd_lock(void)
773  {  {
774          int timer = 0;          int timer = 0;
775          int ret = -1;          int ret = -1;
776    
777            if (p_user_list_pool == NULL)
778            {
779                    log_error("p_user_list_pool not initialized\n");
780                    return -1;
781            }
782    
783          while (!SYS_server_exit)          while (!SYS_server_exit)
784          {          {
785                  ret = user_list_try_rd_lock(semid, USER_LIST_TRY_LOCK_WAIT_TIME);                  ret = user_list_try_rd_lock(USER_LIST_TRY_LOCK_WAIT_TIME);
786                  if (ret == 0) // success                  if (ret == 0) // success
787                  {                  {
788                          break;                          break;
# Line 643  int user_list_rd_lock(int semid) Line 805  int user_list_rd_lock(int semid)
805          return ret;          return ret;
806  }  }
807    
808  int user_list_rw_lock(int semid)  int user_list_rw_lock(void)
809  {  {
810          int timer = 0;          int timer = 0;
811          int ret = -1;          int ret = -1;
812    
813            if (p_user_list_pool == NULL)
814            {
815                    log_error("p_user_list_pool not initialized\n");
816                    return -1;
817            }
818    
819          while (!SYS_server_exit)          while (!SYS_server_exit)
820          {          {
821                  ret = user_list_try_rw_lock(semid, USER_LIST_TRY_LOCK_WAIT_TIME);                  ret = user_list_try_rw_lock(USER_LIST_TRY_LOCK_WAIT_TIME);
822                  if (ret == 0) // success                  if (ret == 0) // success
823                  {                  {
824                          break;                          break;
# Line 683  int query_user_list(int page_id, USER_IN Line 851  int query_user_list(int page_id, USER_IN
851                  return -1;                  return -1;
852          }          }
853    
854            *p_user_count = 0;
855            *p_page_count = 0;
856    
857          // acquire lock of user list          // acquire lock of user list
858          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock() < 0)
859          {          {
860                  log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
861                  return -2;                  return -2;
862          }          }
863    
864          if (p_user_list_pool->p_current->user_count == 0)          if (p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].user_count == 0)
865          {          {
866                  // empty list                  // empty list
867                  ret = 0;                  ret = 0;
868                  goto cleanup;                  goto cleanup;
869          }          }
870    
871          *p_page_count = p_user_list_pool->p_current->user_count / BBS_user_limit_per_page +          *p_page_count = (p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].user_count + BBS_user_limit_per_page - 1) /
872                                          (p_user_list_pool->p_current->user_count % BBS_user_limit_per_page == 0 ? 0 : 1);                                          BBS_user_limit_per_page;
873    
874          if (page_id < 0 || page_id >= *p_page_count)          if (page_id < 0 || page_id >= *p_page_count)
875          {          {
# Line 708  int query_user_list(int page_id, USER_IN Line 879  int query_user_list(int page_id, USER_IN
879          }          }
880    
881          *p_user_count = MIN(BBS_user_limit_per_page,          *p_user_count = MIN(BBS_user_limit_per_page,
882                                                  p_user_list_pool->p_current->user_count -                                                  p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].user_count -
883                                                          page_id * BBS_user_limit_per_page);                                                          page_id * BBS_user_limit_per_page);
884    
885          memcpy(p_users,          memcpy(p_users,
886                     p_user_list_pool->p_current->users + page_id * BBS_user_limit_per_page,                     p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].users + page_id * BBS_user_limit_per_page,
887                     sizeof(USER_INFO) * (size_t)(*p_user_count));                     sizeof(USER_INFO) * (size_t)(*p_user_count));
888    
889  cleanup:  cleanup:
890          // release lock of user list          // release lock of user list
891          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock() < 0)
892          {          {
893                  log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
894                  ret = -1;                  ret = -1;
# Line 736  int query_user_online_list(int page_id, Line 907  int query_user_online_list(int page_id,
907                  return -1;                  return -1;
908          }          }
909    
910            *p_user_count = 0;
911            *p_page_count = 0;
912    
913          // acquire lock of user list          // acquire lock of user list
914          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock() < 0)
915          {          {
916                  log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
917                  return -2;                  return -2;
918          }          }
919    
920          if (p_user_list_pool->p_online_current->user_count == 0)          if (p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].user_count == 0)
921          {          {
922                  // empty list                  // empty list
923                  ret = 0;                  ret = 0;
924                  goto cleanup;                  goto cleanup;
925          }          }
926    
927          *p_page_count = p_user_list_pool->p_online_current->user_count / BBS_user_limit_per_page +          *p_page_count = (p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_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);  
928    
929          if (page_id < 0 || page_id >= *p_page_count)          if (page_id < 0 || page_id >= *p_page_count)
930          {          {
# Line 761  int query_user_online_list(int page_id, Line 934  int query_user_online_list(int page_id,
934          }          }
935    
936          *p_user_count = MIN(BBS_user_limit_per_page,          *p_user_count = MIN(BBS_user_limit_per_page,
937                                                  p_user_list_pool->p_online_current->user_count -                                                  p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].user_count -
938                                                          page_id * BBS_user_limit_per_page);                                                          page_id * BBS_user_limit_per_page);
939    
940          memcpy(p_online_users,          memcpy(p_online_users,
941                     p_user_list_pool->p_online_current->users + page_id * BBS_user_limit_per_page,                     p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].users + page_id * BBS_user_limit_per_page,
942                     sizeof(USER_ONLINE_INFO) * (size_t)(*p_user_count));                     sizeof(USER_ONLINE_INFO) * (size_t)(*p_user_count));
943    
944  cleanup:  cleanup:
945          // release lock of user list          // release lock of user list
946          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock() < 0)
947          {          {
948                  log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
949                  ret = -1;                  ret = -1;
# Line 779  cleanup: Line 952  cleanup:
952          return ret;          return ret;
953  }  }
954    
955    int get_user_list_count(int *p_user_cnt)
956    {
957            if (p_user_cnt == NULL)
958            {
959                    log_error("NULL pointer error\n");
960                    return -1;
961            }
962    
963            // acquire lock of user list
964            if (user_list_rd_lock() < 0)
965            {
966                    log_error("user_list_rd_lock() error\n");
967                    return -2;
968            }
969    
970            *p_user_cnt = p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].user_count;
971    
972            // release lock of user list
973            if (user_list_rd_unlock() < 0)
974            {
975                    log_error("user_list_rd_unlock() error\n");
976                    return -2;
977            }
978    
979            return 0;
980    }
981    
982    int get_user_online_list_count(int *p_user_cnt, int *p_guest_cnt)
983    {
984            if (p_user_cnt == NULL || p_guest_cnt == NULL)
985            {
986                    log_error("NULL pointer error\n");
987                    return -1;
988            }
989    
990            // acquire lock of user list
991            if (user_list_rd_lock() < 0)
992            {
993                    log_error("user_list_rd_lock() error\n");
994                    return -2;
995            }
996    
997            *p_user_cnt = p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].user_count;
998            *p_guest_cnt = p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].guest_count;
999    
1000            // release lock of user list
1001            if (user_list_rd_unlock() < 0)
1002            {
1003                    log_error("user_list_rd_unlock() error\n");
1004                    return -2;
1005            }
1006    
1007            return 0;
1008    }
1009    
1010    int get_user_login_count(int *p_login_cnt)
1011    {
1012            if (p_login_cnt == NULL)
1013            {
1014                    log_error("NULL pointer error\n");
1015                    return -1;
1016            }
1017    
1018            *p_login_cnt = p_user_list_pool->user_login_count;
1019    
1020            return 0;
1021    }
1022    
1023  int query_user_info(int32_t id, USER_INFO *p_user)  int query_user_info(int32_t id, USER_INFO *p_user)
1024  {  {
1025          int ret = 0;          int ret = 0;
# Line 790  int query_user_info(int32_t id, USER_INF Line 1031  int query_user_info(int32_t id, USER_INF
1031          }          }
1032    
1033          // acquire lock of user list          // acquire lock of user list
1034          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock() < 0)
1035          {          {
1036                  log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
1037                  return -2;                  return -2;
1038          }          }
1039    
1040          if (id >= 0 && id < p_user_list_pool->p_current->user_count) // Found          if (id >= 0 && id < p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].user_count) // Found
1041          {          {
1042                  *p_user = p_user_list_pool->p_current->users[id];                  *p_user = p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].users[id];
1043                  ret = 1;                  ret = 1;
1044          }          }
1045    
1046          // release lock of user list          // release lock of user list
1047          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock() < 0)
1048          {          {
1049                  log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
1050                  ret = -1;                  ret = -1;
# Line 812  int query_user_info(int32_t id, USER_INF Line 1053  int query_user_info(int32_t id, USER_INF
1053          return ret;          return ret;
1054  }  }
1055    
1056  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)
1057  {  {
1058          int left;          int left;
1059          int right;          int right;
# Line 827  int query_user_info_by_uid(int32_t uid, Line 1068  int query_user_info_by_uid(int32_t uid,
1068          }          }
1069    
1070          // acquire lock of user list          // acquire lock of user list
1071          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock() < 0)
1072          {          {
1073                  log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
1074                  return -2;                  return -2;
1075          }          }
1076    
1077          left = 0;          left = 0;
1078          right = p_user_list_pool->p_current->user_count - 1;          right = p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].user_count - 1;
1079    
1080          while (left < right)          while (left < right)
1081          {          {
1082                  mid = (left + right) / 2;                  mid = (left + right) / 2;
1083                  if (uid < p_user_list_pool->p_current->index_uid[mid].uid)                  if (uid < p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].index_uid[mid].uid)
1084                  {                  {
1085                          right = mid;                          right = mid - 1;
1086                  }                  }
1087                  else if (uid > p_user_list_pool->p_current->index_uid[mid].uid)                  else if (uid > p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].index_uid[mid].uid)
1088                  {                  {
1089                          left = mid + 1;                          left = mid + 1;
1090                  }                  }
1091                  else // if (uid == p_user_list_pool->p_current->index_uid[mid].uid)                  else // if (uid == p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].index_uid[mid].uid)
1092                  {                  {
1093                          left = mid;                          left = mid;
1094                          break;                          break;
1095                  }                  }
1096          }          }
1097    
1098          if (uid == p_user_list_pool->p_current->index_uid[left].uid) // Found          if (uid == p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].index_uid[left].uid) // Found
1099          {          {
1100                  id = p_user_list_pool->p_current->index_uid[left].id;                  id = p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].index_uid[left].id;
1101                  if ((ret = query_user_info(id, p_user)) <= 0)                  *p_user = p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].users[id];
1102                    ret = 1;
1103    
1104                    if (p_intro_buf != NULL)
1105                  {                  {
1106                          log_error("query_user_info(id=%d) error: %d\n", id, ret);                          strncpy(p_intro_buf, p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].users[id].intro, intro_buf_len - 1);
1107                            p_intro_buf[intro_buf_len - 1] = '\0';
1108                            p_user->intro = p_intro_buf;
1109                  }                  }
1110                  else          }
1111    
1112            // release lock of user list
1113            if (user_list_rd_unlock() < 0)
1114            {
1115                    log_error("user_list_rd_unlock() error\n");
1116                    ret = -1;
1117            }
1118    
1119            return ret;
1120    }
1121    
1122    int query_user_info_by_username(const char *username_prefix, int max_user_cnt,
1123                                                                    int32_t uid_list[], char username_list[][BBS_username_max_len + 1])
1124    {
1125            int left;
1126            int right;
1127            int mid;
1128            int left_save;
1129            int ret = 0;
1130            size_t prefix_len;
1131            int comp;
1132            int i;
1133    
1134            if (username_prefix == NULL || uid_list == NULL || username_list == NULL)
1135            {
1136                    log_error("NULL pointer error\n");
1137                    return -1;
1138            }
1139    
1140            prefix_len = strlen(username_prefix);
1141    
1142            // acquire lock of user list
1143            if (user_list_rd_lock() < 0)
1144            {
1145                    log_error("user_list_rd_lock() error\n");
1146                    return -2;
1147            }
1148    
1149            left = 0;
1150            right = p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].user_count - 1;
1151    
1152            while (left < right)
1153            {
1154                    mid = (left + right) / 2;
1155                    comp = strncasecmp(username_prefix, p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].users[mid].username, prefix_len);
1156                    if (comp < 0)
1157                  {                  {
1158                          ret = 1;                          right = mid - 1;
1159                    }
1160                    else if (comp > 0)
1161                    {
1162                            left = mid + 1;
1163                    }
1164                    else // if (comp == 0)
1165                    {
1166                            left = mid;
1167                            break;
1168                  }                  }
1169          }          }
1170    
1171            if (strncasecmp(username_prefix, p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].users[left].username, prefix_len) == 0) // Found
1172            {
1173    #ifdef _DEBUG
1174                    log_error("Debug: match found, pos=%d\n", left);
1175    #endif
1176    
1177                    left_save = left;
1178                    right = left;
1179                    left = 0;
1180    
1181                    while (left < right)
1182                    {
1183                            mid = (left + right) / 2;
1184                            comp = strncasecmp(username_prefix, p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].users[mid].username, prefix_len);
1185                            if (comp > 0)
1186                            {
1187                                    left = mid + 1;
1188                            }
1189                            else if (comp == 0)
1190                            {
1191                                    right = mid;
1192                            }
1193                            else // if (comp < 0)
1194                            {
1195                                    log_error("Bug: left=%d right=%d mid=%d");
1196                                    ret = -2;
1197                                    goto cleanup;
1198                            }
1199                    }
1200    
1201    #ifdef _DEBUG
1202                    log_error("Debug: first match found, pos=%d\n", right);
1203    #endif
1204    
1205                    left = left_save;
1206                    left_save = right;
1207                    right = p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].user_count - 1;
1208    
1209                    while (left < right)
1210                    {
1211                            mid = (left + right) / 2 + (left + right) % 2;
1212                            comp = strncasecmp(username_prefix, p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].users[mid].username, prefix_len);
1213                            if (comp < 0)
1214                            {
1215                                    right = mid - 1;
1216                            }
1217                            else if (comp == 0)
1218                            {
1219                                    left = mid;
1220                            }
1221                            else // if (comp > 0)
1222                            {
1223                                    log_error("Bug: left=%d right=%d mid=%d");
1224                                    ret = -2;
1225                                    goto cleanup;
1226                            }
1227                    }
1228    
1229    #ifdef _DEBUG
1230                    log_error("Debug: last match found, pos=%d\n", left);
1231    #endif
1232    
1233                    right = left;
1234                    left = left_save;
1235    
1236                    for (i = 0; i < max_user_cnt && left + i <= right; i++)
1237                    {
1238                            uid_list[i] = p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].users[left + i].uid;
1239                            strncpy(username_list[i], p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].users[left + i].username, sizeof(username_list[i]) - 1);
1240                            username_list[i][sizeof(username_list[i]) - 1] = '\0';
1241                    }
1242                    ret = i;
1243            }
1244    
1245    cleanup:
1246          // release lock of user list          // release lock of user list
1247          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock() < 0)
1248          {          {
1249                  log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
1250                  ret = -1;                  ret = -1;
# Line 888  int query_user_online_info(int32_t id, U Line 1264  int query_user_online_info(int32_t id, U
1264          }          }
1265    
1266          // acquire lock of user list          // acquire lock of user list
1267          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock() < 0)
1268          {          {
1269                  log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
1270                  return -2;                  return -2;
1271          }          }
1272    
1273          if (id >= 0 && id < p_user_list_pool->p_online_current->user_count) // Found          if (id >= 0 && id < p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].user_count) // Found
1274          {          {
1275                  *p_user = p_user_list_pool->p_online_current->users[id];                  *p_user = p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].users[id];
1276                  ret = 1;                  ret = 1;
1277          }          }
1278    
1279          // release lock of user list          // release lock of user list
1280          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock() < 0)
1281            {
1282                    log_error("user_list_rd_unlock() error\n");
1283                    ret = -1;
1284            }
1285    
1286            return ret;
1287    }
1288    
1289    int query_user_online_info_by_uid(int32_t uid, USER_ONLINE_INFO *p_users, int *p_user_cnt, int start_id)
1290    {
1291            int left;
1292            int right;
1293            int mid;
1294            int32_t id;
1295            int ret = 0;
1296            int i;
1297            int user_cnt;
1298    
1299            if (p_users == NULL || p_user_cnt == NULL)
1300            {
1301                    log_error("NULL pointer error\n");
1302                    return -1;
1303            }
1304    
1305            user_cnt = *p_user_cnt;
1306            *p_user_cnt = 0;
1307    
1308            // acquire lock of user list
1309            if (user_list_rd_lock() < 0)
1310            {
1311                    log_error("user_list_rd_lock() error\n");
1312                    return -2;
1313            }
1314    
1315            left = start_id;
1316            right = p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].user_count - 1;
1317    
1318            while (left < right)
1319            {
1320                    mid = (left + right) / 2;
1321                    if (uid < p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].index_uid[mid].uid)
1322                    {
1323                            right = mid - 1;
1324                    }
1325                    else if (uid > p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].index_uid[mid].uid)
1326                    {
1327                            left = mid + 1;
1328                    }
1329                    else // if (uid == p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].index_uid[mid].uid)
1330                    {
1331                            left = mid;
1332                            break;
1333                    }
1334            }
1335    
1336            if (uid == p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].index_uid[left].uid)
1337            {
1338                    right = left;
1339                    left = start_id;
1340    
1341                    while (left < right)
1342                    {
1343                            mid = (left + right) / 2;
1344                            if (uid - 1 < p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].index_uid[mid].uid)
1345                            {
1346                                    right = mid;
1347                            }
1348                            else // if (uid - 1 >= p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].index_uid[mid].uid)
1349                            {
1350                                    left = mid + 1;
1351                            }
1352                    }
1353    
1354                    for (i = 0;
1355                             left < p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].user_count && i < user_cnt &&
1356                             uid == p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].index_uid[left].uid;
1357                             left++, i++)
1358                    {
1359                            id = p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].index_uid[left].id;
1360                            p_users[i] = p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].users[id];
1361                    }
1362    
1363                    if (i > 0)
1364                    {
1365                            *p_user_cnt = i;
1366                            ret = 1;
1367                    }
1368            }
1369    
1370            // release lock of user list
1371            if (user_list_rd_unlock() < 0)
1372            {
1373                    log_error("user_list_rd_unlock() error\n");
1374                    ret = -1;
1375            }
1376    
1377            return ret;
1378    }
1379    
1380    int get_user_id_list(int32_t *p_uid_list, int *p_user_cnt, int start_uid)
1381    {
1382            int left;
1383            int right;
1384            int mid;
1385            int ret = 0;
1386            int i;
1387    
1388            if (p_uid_list == NULL || p_user_cnt == NULL)
1389            {
1390                    log_error("NULL pointer error\n");
1391                    return -1;
1392            }
1393    
1394            // acquire lock of user list
1395            if (user_list_rd_lock() < 0)
1396            {
1397                    log_error("user_list_rd_lock() error\n");
1398                    return -2;
1399            }
1400    
1401            left = 0;
1402            right = p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].user_count - 1;
1403    
1404            while (left < right)
1405            {
1406                    mid = (left + right) / 2;
1407                    if (start_uid < p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].index_uid[mid].uid)
1408                    {
1409                            right = mid - 1;
1410                    }
1411                    else if (start_uid > p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].index_uid[mid].uid)
1412                    {
1413                            left = mid + 1;
1414                    }
1415                    else // if (start_uid == p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].index_uid[mid].uid)
1416                    {
1417                            left = mid;
1418                            break;
1419                    }
1420            }
1421    
1422            for (i = 0; i < *p_user_cnt && left + i < p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].user_count; i++)
1423            {
1424                    p_uid_list[i] = p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].index_uid[left + i].uid;
1425            }
1426            *p_user_cnt = i;
1427    
1428            // release lock of user list
1429            if (user_list_rd_unlock() < 0)
1430          {          {
1431                  log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
1432                  ret = -1;                  ret = -1;
# Line 909  int query_user_online_info(int32_t id, U Line 1434  int query_user_online_info(int32_t id, U
1434    
1435          return ret;          return ret;
1436  }  }
1437    
1438    int user_stat_update(void)
1439    {
1440            return user_stat_map_update(&(p_user_list_pool->user_stat_map));
1441    }
1442    
1443    int user_article_cnt_inc(int32_t uid, int n)
1444    {
1445            return user_stat_article_cnt_inc(&(p_user_list_pool->user_stat_map), uid, n);
1446    }
1447    
1448    int get_user_article_cnt(int32_t uid)
1449    {
1450            const USER_STAT *p_stat;
1451            int ret;
1452    
1453            ret = user_stat_get(&(p_user_list_pool->user_stat_map), uid, &p_stat);
1454            if (ret < 0)
1455            {
1456                    log_error("user_stat_get(uid=%d) error: %d\n", uid, ret);
1457                    return -1;
1458            }
1459            else if (ret == 0) // user not found
1460            {
1461                    return -1;
1462            }
1463    
1464            return p_stat->article_count;
1465    }

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