/[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.4 by sysadm, Tue Oct 21 12:35:03 2025 UTC Revision 1.6 by sysadm, Wed Oct 22 04:48:53 2025 UTC
# Line 1  Line 1 
1  /***************************************************************************  /***************************************************************************
2                           user_list.c  -  description                                                   user_list.c  -  description
3                               -------------------                                                           -------------------
4      Copyright            : (C) 2004-2025 by Leaflet          Copyright            : (C) 2004-2025 by Leaflet
5      Email                : leaflet@leafok.com          Email                : leaflet@leafok.com
6   ***************************************************************************/   ***************************************************************************/
7    
8  /***************************************************************************  /***************************************************************************
# Line 17  Line 17 
17  #include "common.h"  #include "common.h"
18  #include "database.h"  #include "database.h"
19  #include "log.h"  #include "log.h"
20    #include "trie_dict.h"
21  #include "user_list.h"  #include "user_list.h"
22  #include <errno.h>  #include <errno.h>
23  #include <stdlib.h>  #include <stdlib.h>
# Line 31  Line 32 
32  #ifdef _SEM_SEMUN_UNDEFINED  #ifdef _SEM_SEMUN_UNDEFINED
33  union semun  union semun
34  {  {
35      int val;               /* Value for SETVAL */          int val;                           /* Value for SETVAL */
36      struct semid_ds *buf;  /* Buffer for IPC_STAT, IPC_SET */          struct semid_ds *buf;  /* Buffer for IPC_STAT, IPC_SET */
37      unsigned short *array; /* Array for GETALL, SETALL */          unsigned short *array; /* Array for GETALL, SETALL */
38      struct seminfo *__buf; /* Buffer for IPC_INFO          struct seminfo *__buf; /* Buffer for IPC_INFO
39                                (Linux-specific) */                                                            (Linux-specific) */
40  };  };
41  #endif // #ifdef _SEM_SEMUN_UNDEFINED  #endif // #ifdef _SEM_SEMUN_UNDEFINED
42    
# Line 44  union semun Line 45  union semun
45    
46  struct user_list_pool_t  struct user_list_pool_t
47  {  {
48      int shmid;          int shmid;
49      int semid;          int semid;
50      USER_LIST user_list[2];          USER_LIST user_list[2];
51      USER_LIST *p_current;          USER_LIST *p_current;
52      USER_LIST *p_new;          USER_LIST *p_new;
53            USER_ONLINE_LIST user_online_list[2];
54            USER_ONLINE_LIST *p_online_current;
55            USER_ONLINE_LIST *p_online_new;
56  };  };
57  typedef struct user_list_pool_t USER_LIST_POOL;  typedef struct user_list_pool_t USER_LIST_POOL;
58    
59  static USER_LIST_POOL *p_user_list_pool = NULL;  static USER_LIST_POOL *p_user_list_pool = NULL;
60    static TRIE_NODE *p_trie_action_dict = NULL;
61    
62    typedef struct user_action_map_t
63    {
64            char name[BBS_current_action_max_len + 1];
65            char title[BBS_current_action_max_len + 1];
66    } USER_ACTION_MAP;
67    
68    const USER_ACTION_MAP user_action_map[] =
69            {
70                    {"ARTICLE_FAVOR", "浏览收藏"},
71                    {"BBS_NET", "站点穿梭"},
72                    {"CHICKEN", "电子小鸡"},
73                    {"EDIT_ARTICLE", "修改文章"},
74                    {"MENU", "菜单选择"},
75                    {"POST_ARTICLE", "撰写文章"},
76                    {"REPLY_ARTICLE", "回复文章"},
77                    {"USER_LIST", "查花名册"},
78                    {"USER_ONLINE", "环顾四周"},
79                    {"VIEW_ARTICLE", "阅读文章"},
80                    {"VIEW_FILE", "查看文档"}};
81    
82    const int user_action_map_size = 11;
83    
84    static int user_list_try_rd_lock(int semid, int wait_sec);
85    static int user_list_try_rw_lock(int semid, int wait_sec);
86    static int user_list_rd_unlock(int semid);
87    static int user_list_rw_unlock(int semid);
88    static int user_list_rd_lock(int semid);
89    static int user_list_rw_lock(int semid);
90    
91    static int user_list_load(MYSQL *db, USER_LIST *p_list);
92    static int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_list);
93    
94  int user_list_load(MYSQL *db, USER_LIST *p_list)  int user_list_load(MYSQL *db, USER_LIST *p_list)
95  {  {
96      MYSQL_RES *rs = NULL;          MYSQL_RES *rs = NULL;
97      MYSQL_ROW row;          MYSQL_ROW row;
98      char sql[SQL_BUFFER_LEN];          char sql[SQL_BUFFER_LEN];
99      int ret = 0;          int ret = 0;
100      int i;          int i;
101    
102      if (db == NULL || p_list == NULL)          if (db == NULL || p_list == NULL)
103      {          {
104          log_error("NULL pointer error\n");                  log_error("NULL pointer error\n");
105          return -1;                  return -1;
106      }          }
107    
108      snprintf(sql, sizeof(sql),          snprintf(sql, sizeof(sql),
109               "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, "
110               "UNIX_TIMESTAMP(signup_dt), UNIX_TIMESTAMP(last_login_dt), UNIX_TIMESTAMP(birthday) "                           "UNIX_TIMESTAMP(signup_dt), UNIX_TIMESTAMP(last_login_dt), UNIX_TIMESTAMP(birthday) "
111               "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 "
112               "INNER JOIN user_reginfo ON user_list.UID = user_reginfo.UID "                           "INNER JOIN user_reginfo ON user_list.UID = user_reginfo.UID "
113               "WHERE enable ORDER BY UID");                           "WHERE enable ORDER BY UID");
114    
115      if (mysql_query(db, sql) != 0)          if (mysql_query(db, sql) != 0)
116      {          {
117          log_error("Query user info error: %s\n", mysql_error(db));                  log_error("Query user info error: %s\n", mysql_error(db));
118          ret = -1;                  ret = -1;
119          goto cleanup;                  goto cleanup;
120      }          }
121    
122      if ((rs = mysql_use_result(db)) == NULL)          if ((rs = mysql_use_result(db)) == NULL)
123      {          {
124          log_error("Get user info data failed\n");                  log_error("Get user info data failed\n");
125          ret = -1;                  ret = -1;
126          goto cleanup;                  goto cleanup;
127      }          }
128    
129      i = 0;          i = 0;
130      while ((row = mysql_fetch_row(rs)))          while ((row = mysql_fetch_row(rs)))
131      {          {
132          p_list->users[i].uid = atoi(row[0]);                  p_list->users[i].uid = atoi(row[0]);
133          strncpy(p_list->users[i].username, row[1], sizeof(p_list->users[i].username) - 1);                  strncpy(p_list->users[i].username, row[1], sizeof(p_list->users[i].username) - 1);
134          p_list->users[i].username[sizeof(p_list->users[i].username) - 1] = '\0';                  p_list->users[i].username[sizeof(p_list->users[i].username) - 1] = '\0';
135          strncpy(p_list->users[i].nickname, row[2], sizeof(p_list->users[i].nickname) - 1);                  strncpy(p_list->users[i].nickname, row[2], sizeof(p_list->users[i].nickname) - 1);
136          p_list->users[i].nickname[sizeof(p_list->users[i].nickname) - 1] = '\0';                  p_list->users[i].nickname[sizeof(p_list->users[i].nickname) - 1] = '\0';
137          p_list->users[i].gender = row[3][0];                  p_list->users[i].gender = row[3][0];
138          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]));
139          p_list->users[i].life = (row[5] == NULL ? 0 : atoi(row[5]));                  p_list->users[i].life = (row[5] == NULL ? 0 : atoi(row[5]));
140          p_list->users[i].exp = (row[6] == NULL ? 0 : atoi(row[6]));                  p_list->users[i].exp = (row[6] == NULL ? 0 : atoi(row[6]));
141          p_list->users[i].signup_dt = (row[7] == NULL ? 0 : atol(row[7]));                  p_list->users[i].signup_dt = (row[7] == NULL ? 0 : atol(row[7]));
142          p_list->users[i].last_login_dt = (row[8] == NULL ? 0 : atol(row[8]));                  p_list->users[i].last_login_dt = (row[8] == NULL ? 0 : atol(row[8]));
143          p_list->users[i].birthday = (row[9] == NULL ? 0 : atol(row[9]));                  p_list->users[i].birthday = (row[9] == NULL ? 0 : atol(row[9]));
144    
145          i++;                  i++;
146          if (i >= BBS_max_user_count)                  if (i >= BBS_max_user_count)
147          {                  {
148              log_error("Too many users, exceed limit %d\n", BBS_max_user_count);                          log_error("Too many users, exceed limit %d\n", BBS_max_user_count);
149              break;                          break;
150          }                  }
151      }          }
152      mysql_free_result(rs);          mysql_free_result(rs);
153      rs = NULL;          rs = NULL;
154    
155      p_list->user_count = i;          p_list->user_count = i;
156    
157  #ifdef _DEBUG  #ifdef _DEBUG
158      log_error("Loaded %d users\n", p_list->user_count);          log_error("Loaded %d users\n", p_list->user_count);
159  #endif  #endif
160    
161  cleanup:  cleanup:
162      mysql_free_result(rs);          mysql_free_result(rs);
   
     return ret;  
 }  
   
 int user_list_pool_init(void)  
 {  
     int shmid;  
     int semid;  
     int proj_id;  
     key_t key;  
     size_t size;  
     void *p_shm;  
     union semun arg;  
     int i;  
   
     if (p_user_list_pool != NULL)  
     {  
         log_error("p_user_list_pool already initialized\n");  
         return -1;  
     }  
   
     // Allocate shared memory  
     proj_id = (int)(time(NULL) % getpid());  
     key = ftok(VAR_USER_LIST_SHM, proj_id);  
     if (key == -1)  
     {  
         log_error("ftok(%s %d) error (%d)\n", VAR_USER_LIST_SHM, proj_id, errno);  
         return -2;  
     }  
   
     size = sizeof(USER_LIST_POOL);  
     shmid = shmget(key, size, IPC_CREAT | IPC_EXCL | 0600);  
     if (shmid == -1)  
     {  
         log_error("shmget(size = %d) error (%d)\n", size, errno);  
         return -3;  
     }  
     p_shm = shmat(shmid, NULL, 0);  
     if (p_shm == (void *)-1)  
     {  
         log_error("shmat(shmid=%d) error (%d)\n", shmid, errno);  
         return -3;  
     }  
   
     p_user_list_pool = p_shm;  
     p_user_list_pool->shmid = shmid;  
   
     // Allocate semaphore as user list pool lock  
     size = 2; // r_sem and w_sem  
     semid = semget(key, (int)size, IPC_CREAT | IPC_EXCL | 0600);  
     if (semid == -1)  
     {  
         log_error("semget(user_list_pool_sem, size = %d) error (%d)\n", size, errno);  
         return -3;  
     }  
   
     // Initialize sem value to 0  
     arg.val = 0;  
     for (i = 0; i < size; i++)  
     {  
         if (semctl(semid, i, SETVAL, arg) == -1)  
         {  
             log_error("semctl(user_list_pool_sem, SETVAL) error (%d)\n", errno);  
             return -3;  
         }  
     }  
   
     p_user_list_pool->semid = semid;  
   
     // Set user counts to 0  
     p_user_list_pool->user_list[0].user_count = 0;  
     p_user_list_pool->user_list[1].user_count = 0;  
   
     p_user_list_pool->p_current = &(p_user_list_pool->user_list[0]);  
     p_user_list_pool->p_new = &(p_user_list_pool->user_list[1]);  
   
     return 0;  
 }  
   
 void user_list_pool_cleanup(void)  
 {  
     int shmid;  
   
     if (p_user_list_pool == NULL)  
     {  
         return;  
     }  
   
     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);  
     }  
   
     if (shmdt(p_user_list_pool) == -1)  
     {  
         log_error("shmdt(shmid = %d) error (%d)\n", shmid, errno);  
     }  
   
     if (shmctl(shmid, IPC_RMID, NULL) == -1)  
     {  
         log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", shmid, errno);  
     }  
163    
164      p_user_list_pool = NULL;          return ret;
165  }  }
166    
167  int set_user_list_pool_shm_readonly(void)  int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_list)
 {  
     int shmid;  
     void *p_shm;  
   
     if (p_user_list_pool == NULL)  
     {  
         log_error("p_user_list_pool not initialized\n");  
         return -1;  
     }  
   
     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;  
   
     return 0;  
 }  
   
 int detach_user_list_pool_shm(void)  
168  {  {
169      if (p_user_list_pool != NULL && shmdt(p_user_list_pool) == -1)          MYSQL_RES *rs = NULL;
170      {          MYSQL_ROW row;
171          log_error("shmdt(user_list_pool) error (%d)\n", errno);          char sql[SQL_BUFFER_LEN];
172          return -1;          int ret = 0;
173      }          int i;
174    
175      p_user_list_pool = NULL;          if (db == NULL || p_list == NULL)
176            {
177      return 0;                  log_error("NULL pointer error\n");
178  }                  return -1;
179            }
180  int user_list_pool_reload(void)  
181  {          snprintf(sql, sizeof(sql),
182      MYSQL *db = NULL;                           "SELECT SID, UID, ip, current_action, UNIX_TIMESTAMP(login_tm), "
183      USER_LIST *p_tmp;                           "UNIX_TIMESTAMP(last_tm) FROM user_online "
184                             "WHERE last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND) "
185      if (p_user_list_pool == NULL)                           "ORDER BY last_tm DESC",
186      {                           BBS_user_off_line);
187          log_error("p_user_list_pool not initialized\n");  
188          return -1;          if (mysql_query(db, sql) != 0)
189      }          {
190                    log_error("Query user online error: %s\n", mysql_error(db));
191      db = db_open();                  ret = -1;
192      if (db == NULL)                  goto cleanup;
193      {          }
194          log_error("db_open() error: %s\n", mysql_error(db));  
195          return -1;          if ((rs = mysql_use_result(db)) == NULL)
196      }          {
197                    log_error("Get user online data failed\n");
198                    ret = -1;
199                    goto cleanup;
200            }
201    
202            i = 0;
203            while ((row = mysql_fetch_row(rs)))
204            {
205                    strncpy(p_list->users[i].session_id, row[0], sizeof(p_list->users[i].session_id) - 1);
206    
207                    p_list->users[i].session_id[sizeof(p_list->users[i].session_id) - 1] = '\0';
208                    if (query_user_info(atoi(row[1]), &(p_list->users[i].user_info)) <= 0)
209                    {
210                            log_error("query_user_info(%d) error\n", atoi(row[1]));
211                            continue;
212                    }
213    
214                    strncpy(p_list->users[i].ip, row[2], sizeof(p_list->users[i].ip) - 1);
215                    p_list->users[i].ip[sizeof(p_list->users[i].ip) - 1] = '\0';
216    
217                    strncpy(p_list->users[i].current_action, row[3], sizeof(p_list->users[i].current_action) - 1);
218                    p_list->users[i].current_action[sizeof(p_list->users[i].current_action) - 1] = '\0';
219                    p_list->users[i].current_action_title = NULL;
220                    if (p_list->users[i].current_action[0] == '\0')
221                    {
222                            p_list->users[i].current_action_title = "Web浏览";
223                    }
224                    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)
225                    {
226                            log_error("trie_dict_get(p_trie_action_dict, %s) error on session_id=%s\n",
227                                              p_list->users[i].current_action, p_list->users[i].session_id);
228                            continue;
229                    }
230    
231                    p_list->users[i].login_tm = (row[4] == NULL ? 0 : atol(row[4]));
232                    p_list->users[i].last_tm = (row[5] == NULL ? 0 : atol(row[5]));
233    
234                    i++;
235                    if (i >= BBS_max_user_online_count)
236                    {
237                            log_error("Too many online users, exceed limit %d\n", BBS_max_user_online_count);
238                            break;
239                    }
240            }
241            mysql_free_result(rs);
242            rs = NULL;
243    
244      if (user_list_load(db, p_user_list_pool->p_new) < 0)          p_list->user_count = i;
     {  
         log_error("user_list_rw_lock() error\n");  
         return -2;  
     }  
245    
246      mysql_close(db);  #ifdef _DEBUG
247            log_error("Loaded %d users\n", p_list->user_count);
248      if (user_list_rw_lock() < 0)  #endif
     {  
         log_error("user_list_rw_lock() error\n");  
         return -3;  
     }  
   
     // Swap p_current and p_new  
     p_tmp = p_user_list_pool->p_current;  
     p_user_list_pool->p_current = p_user_list_pool->p_new;  
     p_user_list_pool->p_new = p_tmp;  
249    
250      if (user_list_rw_unlock() < 0)  cleanup:
251      {          mysql_free_result(rs);
         log_error("user_list_rw_unlock() error\n");  
         return -3;  
     }  
252    
253      return 0;          return ret;
254  }  }
255    
256  int user_list_try_rd_lock(int wait_sec)  int user_list_pool_init(void)
257  {  {
258      struct sembuf sops[2];          int shmid;
259      struct timespec timeout;          int semid;
260      int ret;          int proj_id;
261            key_t key;
262      sops[0].sem_num = 1; // w_sem          size_t size;
263      sops[0].sem_op = 0;  // wait until unlocked          void *p_shm;
264      sops[0].sem_flg = 0;          union semun arg;
265            int i;
266      sops[1].sem_num = 0;        // r_sem  
267      sops[1].sem_op = 1;         // lock          if (p_user_list_pool != NULL || p_trie_action_dict != NULL)
268      sops[1].sem_flg = SEM_UNDO; // undo on terminate          {
269                    log_error("p_user_list_pool already initialized\n");
270                    return -1;
271            }
272    
273            p_trie_action_dict = trie_dict_create();
274            if (p_trie_action_dict == NULL)
275            {
276                    log_error("trie_dict_create() error\n");
277                    return -1;
278            }
279    
280            for (i = 0; i < user_action_map_size; i++)
281            {
282                    if (trie_dict_set(p_trie_action_dict, user_action_map[i].name, (int64_t)(user_action_map[i].title)) < 0)
283                    {
284                            log_error("trie_dict_set(p_trie_action_dict, %s) error\n", user_action_map[i].name);
285                    }
286            }
287    
288            // Allocate shared memory
289            proj_id = (int)(time(NULL) % getpid());
290            key = ftok(VAR_USER_LIST_SHM, proj_id);
291            if (key == -1)
292            {
293                    log_error("ftok(%s %d) error (%d)\n", VAR_USER_LIST_SHM, proj_id, errno);
294                    return -2;
295            }
296    
297            size = sizeof(USER_LIST_POOL);
298            shmid = shmget(key, size, IPC_CREAT | IPC_EXCL | 0600);
299            if (shmid == -1)
300            {
301                    log_error("shmget(size = %d) error (%d)\n", size, errno);
302                    return -3;
303            }
304            p_shm = shmat(shmid, NULL, 0);
305            if (p_shm == (void *)-1)
306            {
307                    log_error("shmat(shmid=%d) error (%d)\n", shmid, errno);
308                    return -3;
309            }
310    
311            p_user_list_pool = p_shm;
312            p_user_list_pool->shmid = shmid;
313    
314            // Allocate semaphore as user list pool lock
315            size = 2; // r_sem and w_sem
316            semid = semget(key, (int)size, IPC_CREAT | IPC_EXCL | 0600);
317            if (semid == -1)
318            {
319                    log_error("semget(user_list_pool_sem, size = %d) error (%d)\n", size, errno);
320                    return -3;
321            }
322    
323            // Initialize sem value to 0
324            arg.val = 0;
325            for (i = 0; i < size; i++)
326            {
327                    if (semctl(semid, i, SETVAL, arg) == -1)
328                    {
329                            log_error("semctl(user_list_pool_sem, SETVAL) error (%d)\n", errno);
330                            return -3;
331                    }
332            }
333    
334            p_user_list_pool->semid = semid;
335    
336            // Set user counts to 0
337            p_user_list_pool->user_list[0].user_count = 0;
338            p_user_list_pool->user_list[1].user_count = 0;
339    
340      timeout.tv_sec = wait_sec;          p_user_list_pool->p_current = &(p_user_list_pool->user_list[0]);
341      timeout.tv_nsec = 0;          p_user_list_pool->p_new = &(p_user_list_pool->user_list[1]);
342    
343      ret = semtimedop(p_user_list_pool->semid, sops, 2, &timeout);          p_user_list_pool->p_online_current = &(p_user_list_pool->user_online_list[0]);
344      if (ret == -1 && errno != EAGAIN && errno != EINTR)          p_user_list_pool->p_online_new = &(p_user_list_pool->user_online_list[1]);
     {  
         log_error("semtimedop(lock read) error %d\n", errno);  
     }  
345    
346      return ret;          return 0;
347  }  }
348    
349  int user_list_try_rw_lock(int wait_sec)  void user_list_pool_cleanup(void)
 {  
     struct sembuf sops[3];  
     struct timespec timeout;  
     int ret;  
   
     sops[0].sem_num = 1; // w_sem  
     sops[0].sem_op = 0;  // wait until unlocked  
     sops[0].sem_flg = 0;  
   
     sops[1].sem_num = 1;        // w_sem  
     sops[1].sem_op = 1;         // lock  
     sops[1].sem_flg = SEM_UNDO; // undo on terminate  
   
     sops[2].sem_num = 0; // r_sem  
     sops[2].sem_op = 0;  // wait until unlocked  
     sops[2].sem_flg = 0;  
   
     timeout.tv_sec = wait_sec;  
     timeout.tv_nsec = 0;  
   
     ret = semtimedop(p_user_list_pool->semid, sops, 3, &timeout);  
     if (ret == -1 && errno != EAGAIN && errno != EINTR)  
     {  
         log_error("semtimedop(lock write) error %d\n", errno);  
     }  
   
     return ret;  
 }  
   
 int user_list_rd_unlock(void)  
350  {  {
351      struct sembuf sops[2];          int shmid;
     int ret;  
   
     sops[0].sem_num = 0;                     // r_sem  
     sops[0].sem_op = -1;                     // unlock  
     sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO; // no wait  
352    
353      ret = semop(p_user_list_pool->semid, sops, 1);          if (p_user_list_pool == NULL)
354      if (ret == -1 && errno != EAGAIN && errno != EINTR)          {
355      {                  return;
356          log_error("semop(unlock read) error %d\n", errno);          }
357      }  
358            shmid = p_user_list_pool->shmid;
359    
360            if (semctl(p_user_list_pool->semid, 0, IPC_RMID) == -1)
361            {
362                    log_error("semctl(semid = %d, IPC_RMID) error (%d)\n", p_user_list_pool->semid, errno);
363            }
364    
365            if (shmdt(p_user_list_pool) == -1)
366            {
367                    log_error("shmdt(shmid = %d) error (%d)\n", shmid, errno);
368            }
369    
370            if (shmctl(shmid, IPC_RMID, NULL) == -1)
371            {
372                    log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", shmid, errno);
373            }
374    
375            p_user_list_pool = NULL;
376    
377            if (p_trie_action_dict != NULL)
378            {
379                    trie_dict_destroy(p_trie_action_dict);
380    
381      return ret;                  p_trie_action_dict = NULL;
382            }
383  }  }
384    
385  int user_list_rw_unlock(void)  int set_user_list_pool_shm_readonly(void)
386  {  {
387      struct sembuf sops[1];          int shmid;
388      int ret;          void *p_shm;
   
     sops[0].sem_num = 1;                     // w_sem  
     sops[0].sem_op = -1;                     // unlock  
     sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO; // no wait  
   
     ret = semop(p_user_list_pool->semid, sops, 1);  
     if (ret == -1 && errno != EAGAIN && errno != EINTR)  
     {  
         log_error("semop(unlock write) error %d\n", errno);  
     }  
389    
390      return ret;          if (p_user_list_pool == NULL)
391  }          {
392                    log_error("p_user_list_pool not initialized\n");
393  int user_list_rd_lock(void)                  return -1;
394  {          }
395      int timer = 0;  
396      int ret = -1;          shmid = p_user_list_pool->shmid;
397    
398            // Remap shared memory in read-only mode
399            p_shm = shmat(shmid, p_user_list_pool, SHM_RDONLY | SHM_REMAP);
400            if (p_shm == (void *)-1)
401            {
402                    log_error("shmat(user_list_pool shmid = %d) error (%d)\n", shmid, errno);
403                    return -3;
404            }
405    
406      while (!SYS_server_exit)          p_user_list_pool = p_shm;
     {  
         ret = user_list_try_rd_lock(USER_LIST_TRY_LOCK_WAIT_TIME);  
         if (ret == 0) // success  
         {  
             break;  
         }  
         else if (errno == EAGAIN || errno == EINTR) // retry  
         {  
             timer++;  
             if (timer % USER_LIST_TRY_LOCK_TIMES == 0)  
             {  
                 log_error("user_list_try_rd_lock() tried %d times\n", timer);  
             }  
         }  
         else // failed  
         {  
             log_error("user_list_try_rd_lock() failed\n");  
             break;  
         }  
     }  
407    
408      return ret;          return 0;
409  }  }
410    
411  int user_list_rw_lock(void)  int detach_user_list_pool_shm(void)
412  {  {
413      int timer = 0;          if (p_user_list_pool != NULL && shmdt(p_user_list_pool) == -1)
414      int ret = -1;          {
415                    log_error("shmdt(user_list_pool) error (%d)\n", errno);
416                    return -1;
417            }
418    
419            p_user_list_pool = NULL;
420    
421            return 0;
422    }
423    
424    int user_list_pool_reload(int online_user)
425    {
426            MYSQL *db = NULL;
427            USER_LIST *p_tmp;
428            USER_ONLINE_LIST *p_online_tmp;
429    
430            if (p_user_list_pool == NULL)
431            {
432                    log_error("p_user_list_pool not initialized\n");
433                    return -1;
434            }
435    
436            db = db_open();
437            if (db == NULL)
438            {
439                    log_error("db_open() error: %s\n", mysql_error(db));
440                    return -1;
441            }
442    
443            if (online_user)
444            {
445                    if (user_online_list_load(db, p_user_list_pool->p_online_new) < 0)
446                    {
447                            log_error("user_online_list_load() error\n");
448                            return -2;
449                    }
450            }
451            else
452            {
453                    if (user_list_load(db, p_user_list_pool->p_new) < 0)
454                    {
455                            log_error("user_list_load() error\n");
456                            return -2;
457                    }
458            }
459    
460            mysql_close(db);
461    
462            if (user_list_rw_lock(p_user_list_pool->semid) < 0)
463            {
464                    log_error("user_list_rw_lock() error\n");
465                    return -3;
466            }
467    
468            if (online_user)
469            {
470                    // Swap p_online_current and p_online_new
471                    p_online_tmp = p_user_list_pool->p_online_current;
472                    p_user_list_pool->p_online_current = p_user_list_pool->p_online_new;
473                    p_user_list_pool->p_online_new = p_online_tmp;
474            }
475            else
476            {
477                    // Swap p_current and p_new
478                    p_tmp = p_user_list_pool->p_current;
479                    p_user_list_pool->p_current = p_user_list_pool->p_new;
480                    p_user_list_pool->p_new = p_tmp;
481            }
482    
483            if (user_list_rw_unlock(p_user_list_pool->semid) < 0)
484            {
485                    log_error("user_list_rw_unlock() error\n");
486                    return -3;
487            }
488    
489            return 0;
490    }
491    
492    int user_list_try_rd_lock(int semid, int wait_sec)
493    {
494            struct sembuf sops[2];
495            struct timespec timeout;
496            int ret;
497    
498            sops[0].sem_num = 1; // w_sem
499            sops[0].sem_op = 0;      // wait until unlocked
500            sops[0].sem_flg = 0;
501    
502            sops[1].sem_num = 0;            // r_sem
503            sops[1].sem_op = 1;                     // lock
504            sops[1].sem_flg = SEM_UNDO; // undo on terminate
505    
506            timeout.tv_sec = wait_sec;
507            timeout.tv_nsec = 0;
508    
509            ret = semtimedop(semid, sops, 2, &timeout);
510            if (ret == -1 && errno != EAGAIN && errno != EINTR)
511            {
512                    log_error("semtimedop(lock read) error %d\n", errno);
513            }
514    
515            return ret;
516    }
517    
518    int user_list_try_rw_lock(int semid, int wait_sec)
519    {
520            struct sembuf sops[3];
521            struct timespec timeout;
522            int ret;
523    
524            sops[0].sem_num = 1; // w_sem
525            sops[0].sem_op = 0;      // wait until unlocked
526            sops[0].sem_flg = 0;
527    
528            sops[1].sem_num = 1;            // w_sem
529            sops[1].sem_op = 1;                     // lock
530            sops[1].sem_flg = SEM_UNDO; // undo on terminate
531    
532            sops[2].sem_num = 0; // r_sem
533            sops[2].sem_op = 0;      // wait until unlocked
534            sops[2].sem_flg = 0;
535    
536            timeout.tv_sec = wait_sec;
537            timeout.tv_nsec = 0;
538    
539            ret = semtimedop(semid, sops, 3, &timeout);
540            if (ret == -1 && errno != EAGAIN && errno != EINTR)
541            {
542                    log_error("semtimedop(lock write) error %d\n", errno);
543            }
544    
545            return ret;
546    }
547    
548    int user_list_rd_unlock(int semid)
549    {
550            struct sembuf sops[2];
551            int ret;
552    
553            sops[0].sem_num = 0;                                     // r_sem
554            sops[0].sem_op = -1;                                     // unlock
555            sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO; // no wait
556    
557            ret = semop(semid, sops, 1);
558            if (ret == -1 && errno != EAGAIN && errno != EINTR)
559            {
560                    log_error("semop(unlock read) error %d\n", errno);
561            }
562    
563            return ret;
564    }
565    
566    int user_list_rw_unlock(int semid)
567    {
568            struct sembuf sops[1];
569            int ret;
570    
571            sops[0].sem_num = 1;                                     // w_sem
572            sops[0].sem_op = -1;                                     // unlock
573            sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO; // no wait
574    
575            ret = semop(semid, sops, 1);
576            if (ret == -1 && errno != EAGAIN && errno != EINTR)
577            {
578                    log_error("semop(unlock write) error %d\n", errno);
579            }
580    
581            return ret;
582    }
583    
584    int user_list_rd_lock(int semid)
585    {
586            int timer = 0;
587            int ret = -1;
588    
589            while (!SYS_server_exit)
590            {
591                    ret = user_list_try_rd_lock(semid, USER_LIST_TRY_LOCK_WAIT_TIME);
592                    if (ret == 0) // success
593                    {
594                            break;
595                    }
596                    else if (errno == EAGAIN || errno == EINTR) // retry
597                    {
598                            timer++;
599                            if (timer % USER_LIST_TRY_LOCK_TIMES == 0)
600                            {
601                                    log_error("user_list_try_rd_lock() tried %d times\n", timer);
602                            }
603                    }
604                    else // failed
605                    {
606                            log_error("user_list_try_rd_lock() failed\n");
607                            break;
608                    }
609            }
610    
611            return ret;
612    }
613    
614    int user_list_rw_lock(int semid)
615    {
616            int timer = 0;
617            int ret = -1;
618    
619            while (!SYS_server_exit)
620            {
621                    ret = user_list_try_rw_lock(semid, USER_LIST_TRY_LOCK_WAIT_TIME);
622                    if (ret == 0) // success
623                    {
624                            break;
625                    }
626                    else if (errno == EAGAIN || errno == EINTR) // retry
627                    {
628                            timer++;
629                            if (timer % USER_LIST_TRY_LOCK_TIMES == 0)
630                            {
631                                    log_error("user_list_try_rw_lock() tried %d times\n", timer);
632                            }
633                    }
634                    else // failed
635                    {
636                            log_error("user_list_try_rw_lock() failed\n");
637                            break;
638                    }
639            }
640    
641      while (!SYS_server_exit)          return ret;
     {  
         ret = user_list_try_rw_lock(USER_LIST_TRY_LOCK_WAIT_TIME);  
         if (ret == 0) // success  
         {  
             break;  
         }  
         else if (errno == EAGAIN || errno == EINTR) // retry  
         {  
             timer++;  
             if (timer % USER_LIST_TRY_LOCK_TIMES == 0)  
             {  
                 log_error("user_list_try_rw_lock() tried %d times\n", timer);  
             }  
         }  
         else // failed  
         {  
             log_error("user_list_try_rw_lock() failed\n");  
             break;  
         }  
     }  
   
     return ret;  
642  }  }
643    
644  int query_user_list(int page_id, USER_INFO *p_users, int *p_user_count, int *p_page_count)  int query_user_list(int page_id, USER_INFO *p_users, int *p_user_count, int *p_page_count)
645  {  {
646      int ret = 0;          int ret = 0;
647    
648            if (p_users == NULL || p_user_count == NULL || p_page_count == NULL)
649            {
650                    log_error("NULL pointer error\n");
651                    return -1;
652            }
653    
654            // acquire lock of user list
655            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
656            {
657                    log_error("user_list_rd_lock() error\n");
658                    return -2;
659            }
660    
661            if (p_user_list_pool->p_current->user_count == 0)
662            {
663                    // empty list
664                    ret = 0;
665                    goto cleanup;
666            }
667    
668            *p_page_count = p_user_list_pool->p_current->user_count / BBS_user_limit_per_page +
669                                            (p_user_list_pool->p_current->user_count % BBS_user_limit_per_page == 0 ? 0 : 1);
670    
671            if (page_id < 0 || page_id >= *p_page_count)
672            {
673                    log_error("Invalid page_id = %d, not in range [0, %d)\n", page_id, *p_page_count);
674                    ret = -3;
675                    goto cleanup;
676            }
677    
678            *p_user_count = MIN(BBS_user_limit_per_page,
679                                                    p_user_list_pool->p_current->user_count -
680                                                            page_id * BBS_user_limit_per_page);
681    
682            memcpy(p_users,
683                       p_user_list_pool->p_current->users + page_id * BBS_user_limit_per_page,
684                       sizeof(USER_INFO) * (size_t)(*p_user_count));
685    
686      if (p_users == NULL || p_user_count == NULL || p_page_count == NULL)  cleanup:
687      {          // release lock of user list
688          log_error("NULL pointer error\n");          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
689          return -1;          {
690      }                  log_error("user_list_rd_unlock() error\n");
691                    ret = -1;
692      // acquire lock of user list          }
693      if (user_list_rd_lock() < 0)  
694      {          return ret;
695          log_error("user_list_rd_lock() error\n");  }
696          return -2;  
697      }  int query_user_online_list(int page_id, USER_ONLINE_INFO *p_online_users, int *p_user_count, int *p_page_count)
698    {
699      if (p_user_list_pool->p_current->user_count == 0)          int ret = 0;
700      {  
701          // empty list          if (p_online_users == NULL || p_user_count == NULL || p_page_count == NULL)
702          ret = 0;          {
703          goto cleanup;                  log_error("NULL pointer error\n");
704      }                  return -1;
705            }
706      *p_page_count = p_user_list_pool->p_current->user_count / BBS_user_limit_per_page +  
707                      (p_user_list_pool->p_current->user_count % BBS_user_limit_per_page == 0 ? 0 : 1);          // acquire lock of user list
708            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
709      if (page_id < 0 || page_id >= *p_page_count)          {
710      {                  log_error("user_list_rd_lock() error\n");
711          log_error("Invalid page_id = %d, not in range [0, %d)\n", page_id, *p_page_count);                  return -2;
712          ret = -3;          }
713          goto cleanup;  
714      }          if (p_user_list_pool->p_online_current->user_count == 0)
715            {
716      *p_user_count = MIN(BBS_user_limit_per_page,                  // empty list
717                          p_user_list_pool->p_current->user_count -                  ret = 0;
718                              page_id * BBS_user_limit_per_page);                  goto cleanup;
719            }
720      memcpy(p_users,  
721             p_user_list_pool->p_current->users + page_id * BBS_user_limit_per_page,          *p_page_count = p_user_list_pool->p_online_current->user_count / BBS_user_limit_per_page +
722             sizeof(USER_INFO) * (size_t)(*p_user_count));                                          (p_user_list_pool->p_online_current->user_count % BBS_user_limit_per_page == 0 ? 0 : 1);
723    
724            if (page_id < 0 || page_id >= *p_page_count)
725            {
726                    log_error("Invalid page_id = %d, not in range [0, %d)\n", page_id, *p_page_count);
727                    ret = -3;
728                    goto cleanup;
729            }
730    
731            *p_user_count = MIN(BBS_user_limit_per_page,
732                                                    p_user_list_pool->p_online_current->user_count -
733                                                            page_id * BBS_user_limit_per_page);
734    
735            memcpy(p_online_users,
736                       p_user_list_pool->p_online_current->users + page_id * BBS_user_limit_per_page,
737                       sizeof(USER_ONLINE_INFO) * (size_t)(*p_user_count));
738    
739  cleanup:  cleanup:
740      // release lock of user list          // release lock of user list
741      if (user_list_rd_unlock() < 0)          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
742      {          {
743          log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
744          ret = -1;                  ret = -1;
745      }          }
746    
747      return ret;          return ret;
748  }  }
749    
750  int query_user_info(int32_t uid, USER_INFO *p_user)  int query_user_info(int32_t uid, USER_INFO *p_user)
751  {  {
752      int left;          int left;
753      int right;          int right;
754      int mid;          int mid;
755      int ret = 0;          int ret = 0;
756    
757      if (p_user == NULL)          if (p_user == NULL)
758      {          {
759          log_error("NULL pointer error\n");                  log_error("NULL pointer error\n");
760          return -1;                  return -1;
761      }          }
762    
763      // acquire lock of user list          // acquire lock of user list
764      if (user_list_rd_lock() < 0)          if (user_list_rd_lock(p_user_list_pool->semid) < 0)
765      {          {
766          log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
767          return -2;                  return -2;
768      }          }
769    
770      left = 0;          left = 0;
771      right = p_user_list_pool->p_current->user_count - 1;          right = p_user_list_pool->p_current->user_count - 1;
772    
773      while (left < right)          while (left < right)
774      {          {
775          mid = (left + right) / 2;                  mid = (left + right) / 2;
776          if (uid < p_user_list_pool->p_current->users[mid].uid)                  if (uid < p_user_list_pool->p_current->users[mid].uid)
777          {                  {
778              right = mid;                          right = mid;
779          }                  }
780          else if (uid > p_user_list_pool->p_current->users[mid].uid)                  else if (uid > p_user_list_pool->p_current->users[mid].uid)
781          {                  {
782              left = mid + 1;                          left = mid + 1;
783          }                  }
784          else // if (uid == p_user_list_pool->p_current->users[mid].uid)                  else // if (uid == p_user_list_pool->p_current->users[mid].uid)
785          {                  {
786              left = mid;                          left = mid;
787              break;                          break;
788          }                  }
789      }          }
790    
791      if (uid == p_user_list_pool->p_current->users[left].uid) // Found          if (uid == p_user_list_pool->p_current->users[left].uid) // Found
792      {          {
793          *p_user = p_user_list_pool->p_current->users[left];                  *p_user = p_user_list_pool->p_current->users[left];
794          ret = 1;                  ret = 1;
795      }          }
796    
797      // release lock of user list          // release lock of user list
798      if (user_list_rd_unlock() < 0)          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
799      {          {
800          log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
801          ret = -1;                  ret = -1;
802      }          }
803    
804      return ret;          return ret;
805  }  }

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