/[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.18 by sysadm, Thu Oct 23 04:09:13 2025 UTC Revision 1.35 by sysadm, Mon Nov 17 12:16:48 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"
# Line 30  Line 26 
26  #include <sys/sem.h>  #include <sys/sem.h>
27  #include <sys/shm.h>  #include <sys/shm.h>
28    
29  #ifdef _SEM_SEMUN_UNDEFINED  #if defined(_SEM_SEMUN_UNDEFINED) || defined(__MSYS__) || defined(__MINGW32__)
30  union semun  union semun
31  {  {
32          int val;                           /* Value for SETVAL */          int val;                           /* Value for SETVAL */
# Line 39  union semun Line 35  union semun
35          struct seminfo *__buf; /* Buffer for IPC_INFO          struct seminfo *__buf; /* Buffer for IPC_INFO
36                                                            (Linux-specific) */                                                            (Linux-specific) */
37  };  };
38  #endif // #ifdef _SEM_SEMUN_UNDEFINED  #endif // #if defined(_SEM_SEMUN_UNDEFINED)
39    
40  #define USER_LIST_TRY_LOCK_WAIT_TIME 1 // second  enum _user_list_constant_t
41  #define USER_LIST_TRY_LOCK_TIMES 10  {
42            USER_LIST_TRY_LOCK_WAIT_TIME = 1, // second
43            USER_LIST_TRY_LOCK_TIMES = 10,
44    };
45    
46  struct user_list_pool_t  struct user_list_pool_t
47  {  {
# Line 55  struct user_list_pool_t Line 54  struct user_list_pool_t
54          USER_ONLINE_LIST *p_online_current;          USER_ONLINE_LIST *p_online_current;
55          USER_ONLINE_LIST *p_online_new;          USER_ONLINE_LIST *p_online_new;
56          USER_STAT_MAP user_stat_map;          USER_STAT_MAP user_stat_map;
57            int user_login_count;
58  };  };
59  typedef struct user_list_pool_t USER_LIST_POOL;  typedef struct user_list_pool_t USER_LIST_POOL;
60    
# Line 93  static int user_list_rd_lock(int semid); Line 93  static int user_list_rd_lock(int semid);
93  static int user_list_rw_lock(int semid);  static int user_list_rw_lock(int semid);
94    
95  static int user_list_load(MYSQL *db, USER_LIST *p_list);  static int user_list_load(MYSQL *db, USER_LIST *p_list);
96  static int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_list);  static int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_online_list);
97    static int user_login_count_load(MYSQL *db);
98    
99  static int user_info_index_uid_comp(const void *ptr1, const void *ptr2)  static int user_info_index_uid_comp(const void *ptr1, const void *ptr2)
100  {  {
# Line 148  int user_list_load(MYSQL *db, USER_LIST Line 149  int user_list_load(MYSQL *db, USER_LIST
149    
150          snprintf(sql, sizeof(sql),          snprintf(sql, sizeof(sql),
151                           "SELECT user_list.UID AS UID, username, nickname, gender, gender_pub, life, exp, visit_count, "                           "SELECT user_list.UID AS UID, username, nickname, gender, gender_pub, life, exp, visit_count, "
152                           "UNIX_TIMESTAMP(signup_dt), UNIX_TIMESTAMP(last_login_dt), UNIX_TIMESTAMP(birthday), `introduction` "                           "UNIX_TIMESTAMP(signup_dt), UNIX_TIMESTAMP(last_login_dt), UNIX_TIMESTAMP(last_logout_dt), "
153                             "UNIX_TIMESTAMP(birthday), `introduction` "
154                           "FROM user_list INNER JOIN user_pubinfo ON user_list.UID = user_pubinfo.UID "                           "FROM user_list INNER JOIN user_pubinfo ON user_list.UID = user_pubinfo.UID "
155                           "INNER JOIN user_reginfo ON user_list.UID = user_reginfo.UID "                           "INNER JOIN user_reginfo ON user_list.UID = user_reginfo.UID "
156                           "WHERE enable ORDER BY username");                           "WHERE enable ORDER BY username");
# Line 185  int user_list_load(MYSQL *db, USER_LIST Line 187  int user_list_load(MYSQL *db, USER_LIST
187                  p_list->users[i].visit_count = (row[7] == NULL ? 0 : atoi(row[7]));                  p_list->users[i].visit_count = (row[7] == NULL ? 0 : atoi(row[7]));
188                  p_list->users[i].signup_dt = (row[8] == NULL ? 0 : atol(row[8]));                  p_list->users[i].signup_dt = (row[8] == NULL ? 0 : atol(row[8]));
189                  p_list->users[i].last_login_dt = (row[9] == NULL ? 0 : atol(row[9]));                  p_list->users[i].last_login_dt = (row[9] == NULL ? 0 : atol(row[9]));
190                  p_list->users[i].birthday = (row[10] == NULL ? 0 : atol(row[10]));                  p_list->users[i].last_logout_dt = (row[10] == NULL ? 0 : atol(row[10]));
191                  intro_len = strlen((row[11] == NULL ? "" : row[11]));                  p_list->users[i].birthday = (row[11] == NULL ? 0 : atol(row[11]));
192                    intro_len = strlen((row[12] == NULL ? "" : row[12]));
193                  if (intro_len >= sizeof(p_list->user_intro_buf) - 1 - intro_buf_offset)                  if (intro_len >= sizeof(p_list->user_intro_buf) - 1 - intro_buf_offset)
194                  {                  {
195                          log_error("OOM for user introduction: len=%d, i=%d\n", intro_len, i);                          log_error("OOM for user introduction: len=%d, i=%d\n", intro_len, i);
196                          break;                          break;
197                  }                  }
198                  memcpy(p_list->user_intro_buf + intro_buf_offset,                  memcpy(p_list->user_intro_buf + intro_buf_offset,
199                             (row[11] == NULL ? "" : row[11]),                             (row[12] == NULL ? "" : row[12]),
200                             intro_len + 1);                             intro_len + 1);
201                  p_list->users[i].intro = p_list->user_intro_buf + intro_buf_offset;                  p_list->users[i].intro = p_list->user_intro_buf + intro_buf_offset;
202                  intro_buf_offset += (intro_len + 1);                  intro_buf_offset += (intro_len + 1);
# Line 236  cleanup: Line 239  cleanup:
239          return ret;          return ret;
240  }  }
241    
242  int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_list)  int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_online_list)
243  {  {
244          MYSQL_RES *rs = NULL;          MYSQL_RES *rs = NULL;
245          MYSQL_ROW row;          MYSQL_ROW row;
# Line 244  int user_online_list_load(MYSQL *db, USE Line 247  int user_online_list_load(MYSQL *db, USE
247          int ret = 0;          int ret = 0;
248          int i;          int i;
249          int j;          int j;
250            int user_cnt;
251            int guest_cnt;
252    
253          if (db == NULL || p_list == NULL)          if (db == NULL || p_online_list == NULL)
254          {          {
255                  log_error("NULL pointer error\n");                  log_error("NULL pointer error\n");
256                  return -1;                  return -1;
# Line 254  int user_online_list_load(MYSQL *db, USE Line 259  int user_online_list_load(MYSQL *db, USE
259          snprintf(sql, sizeof(sql),          snprintf(sql, sizeof(sql),
260                           "SELECT SID, UID, ip, current_action, UNIX_TIMESTAMP(login_tm), "                           "SELECT SID, UID, ip, current_action, UNIX_TIMESTAMP(login_tm), "
261                           "UNIX_TIMESTAMP(last_tm) FROM user_online "                           "UNIX_TIMESTAMP(last_tm) FROM user_online "
262                           "WHERE last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND) AND UID <> 0 "                           "WHERE last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND) "
263                           "ORDER BY last_tm DESC",                           "ORDER BY last_tm DESC",
264                           BBS_user_off_line);                           BBS_user_off_line);
265    
# Line 273  int user_online_list_load(MYSQL *db, USE Line 278  int user_online_list_load(MYSQL *db, USE
278          }          }
279    
280          i = 0;          i = 0;
281            user_cnt = 0;
282            guest_cnt = 0;
283          while ((row = mysql_fetch_row(rs)))          while ((row = mysql_fetch_row(rs)))
284          {          {
285                  p_list->users[i].id = i;                  if (atoi(row[1]) == 0) // guest
286                  strncpy(p_list->users[i].session_id, row[0], sizeof(p_list->users[i].session_id) - 1);                  {
287                  p_list->users[i].session_id[sizeof(p_list->users[i].session_id) - 1] = '\0';                          guest_cnt++;
288                            continue;
289                    }
290                    else
291                    {
292                            user_cnt++;
293                    }
294    
295                  if ((ret = query_user_info_by_uid(atoi(row[1]), &(p_list->users[i].user_info))) <= 0)                  p_online_list->users[i].id = i;
296                    strncpy(p_online_list->users[i].session_id, row[0], sizeof(p_online_list->users[i].session_id) - 1);
297                    p_online_list->users[i].session_id[sizeof(p_online_list->users[i].session_id) - 1] = '\0';
298    
299                    if ((ret = query_user_info_by_uid(atoi(row[1]), &(p_online_list->users[i].user_info), NULL, 0)) <= 0)
300                  {                  {
301                          log_error("query_user_info_by_uid(%d) error\n", atoi(row[1]));                          log_error("query_user_info_by_uid(%d) error\n", atoi(row[1]));
302                          continue;                          continue;
303                  }                  }
304    
305                  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);
306                  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';
307    
308                  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);
309                  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';
310                  p_list->users[i].current_action_title = NULL;                  p_online_list->users[i].current_action_title = NULL;
311                  if (p_list->users[i].current_action[0] == '\0')                  if (p_online_list->users[i].current_action[0] == '\0')
312                  {                  {
313                          p_list->users[i].current_action_title = "";                          p_online_list->users[i].current_action_title = "";
314                  }                  }
315                  else if (trie_dict_get(p_trie_action_dict, p_list->users[i].current_action, (int64_t *)(&(p_list->users[i].current_action_title))) < 0)                  else if (trie_dict_get(p_trie_action_dict, p_online_list->users[i].current_action, (int64_t *)(&(p_online_list->users[i].current_action_title))) < 0)
316                  {                  {
317                          log_error("trie_dict_get(p_trie_action_dict, %s) error on session_id=%s\n",                          log_error("trie_dict_get(p_trie_action_dict, %s) error on session_id=%s\n",
318                                            p_list->users[i].current_action, p_list->users[i].session_id);                                            p_online_list->users[i].current_action, p_online_list->users[i].session_id);
319                          continue;                          continue;
320                  }                  }
321    
322                  p_list->users[i].login_tm = (row[4] == NULL ? 0 : atol(row[4]));                  p_online_list->users[i].login_tm = (row[4] == NULL ? 0 : atol(row[4]));
323                  p_list->users[i].last_tm = (row[5] == NULL ? 0 : atol(row[5]));                  p_online_list->users[i].last_tm = (row[5] == NULL ? 0 : atol(row[5]));
324    
325                  i++;                  i++;
326                  if (i >= BBS_max_user_online_count)                  if (i >= BBS_max_user_online_count)
# Line 315  int user_online_list_load(MYSQL *db, USE Line 332  int user_online_list_load(MYSQL *db, USE
332          mysql_free_result(rs);          mysql_free_result(rs);
333          rs = NULL;          rs = NULL;
334    
335          if (i > 0)          if (user_cnt > 0)
336          {          {
337                  // Rebuild index                  // Rebuild index
338                  for (j = 0; j < i; j++)                  for (j = 0; j < user_cnt; j++)
339                  {                  {
340                          p_list->index_uid[j].uid = p_list->users[j].user_info.uid;                          p_online_list->index_uid[j].uid = p_online_list->users[j].user_info.uid;
341                          p_list->index_uid[j].id = j;                          p_online_list->index_uid[j].id = j;
342                  }                  }
343    
344                  qsort(p_list->index_uid, (size_t)i, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp);                  qsort(p_online_list->index_uid, (size_t)user_cnt, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp);
   
 #ifdef _DEBUG  
                 log_error("Rebuild index of %d online users\n", i);  
 #endif  
345          }          }
346    
347          p_list->user_count = i;          p_online_list->user_count = user_cnt;
348            p_online_list->guest_count = guest_cnt;
 #ifdef _DEBUG  
         log_error("Loaded %d online users\n", p_list->user_count);  
 #endif  
349    
350  cleanup:  cleanup:
351          mysql_free_result(rs);          mysql_free_result(rs);
# Line 343  cleanup: Line 353  cleanup:
353          return ret;          return ret;
354  }  }
355    
356  int user_list_pool_init(void)  int user_login_count_load(MYSQL *db)
357    {
358            MYSQL_RES *rs = NULL;
359            MYSQL_ROW row;
360            char sql[SQL_BUFFER_LEN];
361    
362            if (db == NULL)
363            {
364                    log_error("NULL pointer error\n");
365                    return -1;
366            }
367    
368            snprintf(sql, sizeof(sql),
369                             "SELECT ID FROM user_login_log ORDER BY ID DESC LIMIT 1");
370            if (mysql_query(db, sql) != 0)
371            {
372                    log_error("Query user_login_log error: %s\n", mysql_error(db));
373                    return -2;
374            }
375            if ((rs = mysql_store_result(db)) == NULL)
376            {
377                    log_error("Get user_login_log data failed\n");
378                    return -2;
379            }
380            if ((row = mysql_fetch_row(rs)))
381            {
382                    p_user_list_pool->user_login_count = atoi(row[0]);
383            }
384            mysql_free_result(rs);
385    
386            return 0;
387    }
388    
389    int user_list_pool_init(const char *filename)
390  {  {
391          int shmid;          int shmid;
392          int semid;          int semid;
# Line 377  int user_list_pool_init(void) Line 420  int user_list_pool_init(void)
420    
421          // Allocate shared memory          // Allocate shared memory
422          proj_id = (int)(time(NULL) % getpid());          proj_id = (int)(time(NULL) % getpid());
423          key = ftok(VAR_USER_LIST_SHM, proj_id);          key = ftok(filename, proj_id);
424          if (key == -1)          if (key == -1)
425          {          {
426                  log_error("ftok(%s %d) error (%d)\n", VAR_USER_LIST_SHM, proj_id, errno);                  log_error("ftok(%s %d) error (%d)\n", filename, proj_id, errno);
427                  return -2;                  return -2;
428          }          }
429    
# Line 459  void user_list_pool_cleanup(void) Line 502  void user_list_pool_cleanup(void)
502                  log_error("shmdt(shmid = %d) error (%d)\n", shmid, errno);                  log_error("shmdt(shmid = %d) error (%d)\n", shmid, errno);
503          }          }
504    
505          if (shmctl(shmid, IPC_RMID, NULL) == -1)          if (shmid != 0 && shmctl(shmid, IPC_RMID, NULL) == -1)
506          {          {
507                  log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", shmid, errno);                  log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", shmid, errno);
508          }          }
# Line 488  int set_user_list_pool_shm_readonly(void Line 531  int set_user_list_pool_shm_readonly(void
531          shmid = p_user_list_pool->shmid;          shmid = p_user_list_pool->shmid;
532    
533          // Remap shared memory in read-only mode          // Remap shared memory in read-only mode
534    #if defined(__MSYS__) || defined(__MINGW32__)
535            if (shmdt(p_user_list_pool) == -1)
536            {
537                    log_error("shmdt(user_list_pool) error (%d)\n", errno);
538                    return -1;
539            }
540            p_shm = shmat(shmid, p_user_list_pool, SHM_RDONLY);
541    #else
542          p_shm = shmat(shmid, p_user_list_pool, SHM_RDONLY | SHM_REMAP);          p_shm = shmat(shmid, p_user_list_pool, SHM_RDONLY | SHM_REMAP);
543    #endif
544          if (p_shm == (void *)-1)          if (p_shm == (void *)-1)
545          {          {
546                  log_error("shmat(user_list_pool shmid = %d) error (%d)\n", shmid, errno);                  log_error("shmat(user_list_pool shmid = %d) error (%d)\n", shmid, errno);
# Line 541  int user_list_pool_reload(int online_use Line 593  int user_list_pool_reload(int online_use
593                          ret = -2;                          ret = -2;
594                          goto cleanup;                          goto cleanup;
595                  }                  }
596    
597                    if (user_login_count_load(db) < 0)
598                    {
599                            log_error("user_login_count_load() error\n");
600                            ret = -2;
601                            goto cleanup;
602                    }
603          }          }
604          else          else
605          {          {
# Line 593  cleanup: Line 652  cleanup:
652  int user_list_try_rd_lock(int semid, int wait_sec)  int user_list_try_rd_lock(int semid, int wait_sec)
653  {  {
654          struct sembuf sops[2];          struct sembuf sops[2];
655    #if !defined(__MSYS__) && !defined(__MINGW32__)
656          struct timespec timeout;          struct timespec timeout;
657    #endif
658          int ret;          int ret;
659    
660          sops[0].sem_num = 1; // w_sem          sops[0].sem_num = 1; // w_sem
# Line 604  int user_list_try_rd_lock(int semid, int Line 665  int user_list_try_rd_lock(int semid, int
665          sops[1].sem_op = 1;                     // lock          sops[1].sem_op = 1;                     // lock
666          sops[1].sem_flg = SEM_UNDO; // undo on terminate          sops[1].sem_flg = SEM_UNDO; // undo on terminate
667    
668    #if defined(__MSYS__) || defined(__MINGW32__)
669            ret = semop(semid, sops, 2);
670    #else
671          timeout.tv_sec = wait_sec;          timeout.tv_sec = wait_sec;
672          timeout.tv_nsec = 0;          timeout.tv_nsec = 0;
673    
674          ret = semtimedop(semid, sops, 2, &timeout);          ret = semtimedop(semid, sops, 2, &timeout);
675    #endif
676          if (ret == -1 && errno != EAGAIN && errno != EINTR)          if (ret == -1 && errno != EAGAIN && errno != EINTR)
677          {          {
678                  log_error("semtimedop(lock read) error %d\n", errno);                  log_error("semop(lock read) error %d\n", errno);
679          }          }
680    
681          return ret;          return ret;
# Line 619  int user_list_try_rd_lock(int semid, int Line 684  int user_list_try_rd_lock(int semid, int
684  int user_list_try_rw_lock(int semid, int wait_sec)  int user_list_try_rw_lock(int semid, int wait_sec)
685  {  {
686          struct sembuf sops[3];          struct sembuf sops[3];
687    #if !defined(__MSYS__) && !defined(__MINGW32__)
688          struct timespec timeout;          struct timespec timeout;
689    #endif
690          int ret;          int ret;
691    
692          sops[0].sem_num = 1; // w_sem          sops[0].sem_num = 1; // w_sem
# Line 634  int user_list_try_rw_lock(int semid, int Line 701  int user_list_try_rw_lock(int semid, int
701          sops[2].sem_op = 0;      // wait until unlocked          sops[2].sem_op = 0;      // wait until unlocked
702          sops[2].sem_flg = 0;          sops[2].sem_flg = 0;
703    
704    #if defined(__MSYS__) || defined(__MINGW32__)
705            ret = semop(semid, sops, 3);
706    #else
707          timeout.tv_sec = wait_sec;          timeout.tv_sec = wait_sec;
708          timeout.tv_nsec = 0;          timeout.tv_nsec = 0;
709    
710          ret = semtimedop(semid, sops, 3, &timeout);          ret = semtimedop(semid, sops, 3, &timeout);
711    #endif
712          if (ret == -1 && errno != EAGAIN && errno != EINTR)          if (ret == -1 && errno != EAGAIN && errno != EINTR)
713          {          {
714                  log_error("semtimedop(lock write) error %d\n", errno);                  log_error("semop(lock write) error %d\n", errno);
715          }          }
716    
717          return ret;          return ret;
# Line 769  int query_user_list(int page_id, USER_IN Line 840  int query_user_list(int page_id, USER_IN
840                  goto cleanup;                  goto cleanup;
841          }          }
842    
843          *p_page_count = p_user_list_pool->p_current->user_count / BBS_user_limit_per_page +          *p_page_count = (p_user_list_pool->p_current->user_count + BBS_user_limit_per_page - 1) /
844                                          (p_user_list_pool->p_current->user_count % BBS_user_limit_per_page == 0 ? 0 : 1);                                          BBS_user_limit_per_page;
845    
846          if (page_id < 0 || page_id >= *p_page_count)          if (page_id < 0 || page_id >= *p_page_count)
847          {          {
# Line 825  int query_user_online_list(int page_id, Line 896  int query_user_online_list(int page_id,
896                  goto cleanup;                  goto cleanup;
897          }          }
898    
899          *p_page_count = p_user_list_pool->p_online_current->user_count / BBS_user_limit_per_page +          *p_page_count = (p_user_list_pool->p_online_current->user_count + BBS_user_limit_per_page - 1) / BBS_user_limit_per_page;
                                         (p_user_list_pool->p_online_current->user_count % BBS_user_limit_per_page == 0 ? 0 : 1);  
900    
901          if (page_id < 0 || page_id >= *p_page_count)          if (page_id < 0 || page_id >= *p_page_count)
902          {          {
# Line 854  cleanup: Line 924  cleanup:
924          return ret;          return ret;
925  }  }
926    
927    int get_user_list_count(int *p_user_cnt)
928    {
929            if (p_user_cnt == NULL)
930            {
931                    log_error("NULL pointer error\n");
932                    return -1;
933            }
934    
935            // acquire lock of user list
936            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
937            {
938                    log_error("user_list_rd_lock() error\n");
939                    return -2;
940            }
941    
942            *p_user_cnt = p_user_list_pool->p_current->user_count;
943    
944            // release lock of user list
945            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
946            {
947                    log_error("user_list_rd_unlock() error\n");
948                    return -2;
949            }
950    
951            return 0;
952    }
953    
954    int get_user_online_list_count(int *p_user_cnt, int *p_guest_cnt)
955    {
956            if (p_user_cnt == NULL || p_guest_cnt == NULL)
957            {
958                    log_error("NULL pointer error\n");
959                    return -1;
960            }
961    
962            // acquire lock of user list
963            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
964            {
965                    log_error("user_list_rd_lock() error\n");
966                    return -2;
967            }
968    
969            *p_user_cnt = p_user_list_pool->p_online_current->user_count;
970            *p_guest_cnt = p_user_list_pool->p_online_current->guest_count;
971    
972            // release lock of user list
973            if (user_list_rd_unlock(p_user_list_pool->semid) < 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_login_count(int *p_login_cnt)
983    {
984            if (p_login_cnt == NULL)
985            {
986                    log_error("NULL pointer error\n");
987                    return -1;
988            }
989    
990            *p_login_cnt = p_user_list_pool->user_login_count;
991    
992            return 0;
993    }
994    
995  int query_user_info(int32_t id, USER_INFO *p_user)  int query_user_info(int32_t id, USER_INFO *p_user)
996  {  {
997          int ret = 0;          int ret = 0;
# Line 887  int query_user_info(int32_t id, USER_INF Line 1025  int query_user_info(int32_t id, USER_INF
1025          return ret;          return ret;
1026  }  }
1027    
1028  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)
1029  {  {
1030          int left;          int left;
1031          int right;          int right;
# Line 916  int query_user_info_by_uid(int32_t uid, Line 1054  int query_user_info_by_uid(int32_t uid,
1054                  mid = (left + right) / 2;                  mid = (left + right) / 2;
1055                  if (uid < p_user_list_pool->p_current->index_uid[mid].uid)                  if (uid < p_user_list_pool->p_current->index_uid[mid].uid)
1056                  {                  {
1057                          right = mid;                          right = mid - 1;
1058                  }                  }
1059                  else if (uid > p_user_list_pool->p_current->index_uid[mid].uid)                  else if (uid > p_user_list_pool->p_current->index_uid[mid].uid)
1060                  {                  {
# Line 934  int query_user_info_by_uid(int32_t uid, Line 1072  int query_user_info_by_uid(int32_t uid,
1072                  id = p_user_list_pool->p_current->index_uid[left].id;                  id = p_user_list_pool->p_current->index_uid[left].id;
1073                  *p_user = p_user_list_pool->p_current->users[id];                  *p_user = p_user_list_pool->p_current->users[id];
1074                  ret = 1;                  ret = 1;
1075    
1076                    if (p_intro_buf != NULL)
1077                    {
1078                            strncpy(p_intro_buf, p_user_list_pool->p_current->users[id].intro, intro_buf_len - 1);
1079                            p_intro_buf[intro_buf_len - 1] = '\0';
1080                            p_user->intro = p_intro_buf;
1081                    }
1082          }          }
1083    
1084          // release lock of user list          // release lock of user list
# Line 946  int query_user_info_by_uid(int32_t uid, Line 1091  int query_user_info_by_uid(int32_t uid,
1091          return ret;          return ret;
1092  }  }
1093    
1094    int query_user_info_by_username(const char *username_prefix, int max_user_cnt,
1095                                                                    int32_t uid_list[], char username_list[][BBS_username_max_len + 1])
1096    {
1097            int left;
1098            int right;
1099            int mid;
1100            int left_save;
1101            int ret = 0;
1102            size_t prefix_len;
1103            int comp;
1104            int i;
1105    
1106            if (username_prefix == NULL || uid_list == NULL || username_list == NULL)
1107            {
1108                    log_error("NULL pointer error\n");
1109                    return -1;
1110            }
1111    
1112            prefix_len = strlen(username_prefix);
1113    
1114            // acquire lock of user list
1115            if (user_list_rd_lock(p_user_list_pool->semid) < 0)
1116            {
1117                    log_error("user_list_rd_lock() error\n");
1118                    return -2;
1119            }
1120    
1121            left = 0;
1122            right = p_user_list_pool->p_current->user_count - 1;
1123    
1124            while (left < right)
1125            {
1126                    mid = (left + right) / 2;
1127                    comp = strncasecmp(username_prefix, p_user_list_pool->p_current->users[mid].username, prefix_len);
1128                    if (comp < 0)
1129                    {
1130                            right = mid - 1;
1131                    }
1132                    else if (comp > 0)
1133                    {
1134                            left = mid + 1;
1135                    }
1136                    else // if (comp == 0)
1137                    {
1138                            left = mid;
1139                            break;
1140                    }
1141            }
1142    
1143            if (strncasecmp(username_prefix, p_user_list_pool->p_current->users[left].username, prefix_len) == 0) // Found
1144            {
1145    #ifdef _DEBUG
1146                    log_error("Debug: match found, pos=%d\n", left);
1147    #endif
1148    
1149                    left_save = left;
1150                    right = left;
1151                    left = 0;
1152    
1153                    while (left < right)
1154                    {
1155                            mid = (left + right) / 2;
1156                            comp = strncasecmp(username_prefix, p_user_list_pool->p_current->users[mid].username, prefix_len);
1157                            if (comp > 0)
1158                            {
1159                                    left = mid + 1;
1160                            }
1161                            else if (comp == 0)
1162                            {
1163                                    right = mid;
1164                            }
1165                            else // if (comp < 0)
1166                            {
1167                                    log_error("Bug: left=%d right=%d mid=%d");
1168                                    ret = -2;
1169                                    goto cleanup;
1170                            }
1171                    }
1172    
1173    #ifdef _DEBUG
1174                    log_error("Debug: first match found, pos=%d\n", right);
1175    #endif
1176    
1177                    left = left_save;
1178                    left_save = right;
1179                    right = p_user_list_pool->p_current->user_count - 1;
1180    
1181                    while (left < right)
1182                    {
1183                            mid = (left + right) / 2 + (left + right) % 2;
1184                            comp = strncasecmp(username_prefix, p_user_list_pool->p_current->users[mid].username, prefix_len);
1185                            if (comp < 0)
1186                            {
1187                                    right = mid - 1;
1188                            }
1189                            else if (comp == 0)
1190                            {
1191                                    left = 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: last match found, pos=%d\n", left);
1203    #endif
1204    
1205                    right = left;
1206                    left = left_save;
1207    
1208                    for (i = 0; i < max_user_cnt && left + i <= right; i++)
1209                    {
1210                            uid_list[i] = p_user_list_pool->p_current->users[left + i].uid;
1211                            strncpy(username_list[i], p_user_list_pool->p_current->users[left + i].username, sizeof(username_list[i]) - 1);
1212                            username_list[i][sizeof(username_list[i]) - 1] = '\0';
1213                    }
1214                    ret = i;
1215            }
1216    
1217    cleanup:
1218            // release lock of user list
1219            if (user_list_rd_unlock(p_user_list_pool->semid) < 0)
1220            {
1221                    log_error("user_list_rd_unlock() error\n");
1222                    ret = -1;
1223            }
1224    
1225            return ret;
1226    }
1227    
1228  int query_user_online_info(int32_t id, USER_ONLINE_INFO *p_user)  int query_user_online_info(int32_t id, USER_ONLINE_INFO *p_user)
1229  {  {
1230          int ret = 0;          int ret = 0;
# Line 987  int query_user_online_info_by_uid(int32_ Line 1266  int query_user_online_info_by_uid(int32_
1266          int32_t id;          int32_t id;
1267          int ret = 0;          int ret = 0;
1268          int i;          int i;
1269            int user_cnt;
1270    
1271          if (p_users == NULL || p_user_cnt == NULL)          if (p_users == NULL || p_user_cnt == NULL)
1272          {          {
# Line 994  int query_user_online_info_by_uid(int32_ Line 1274  int query_user_online_info_by_uid(int32_
1274                  return -1;                  return -1;
1275          }          }
1276    
1277            user_cnt = *p_user_cnt;
1278            *p_user_cnt = 0;
1279    
1280          // acquire lock of user list          // acquire lock of user list
1281          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock(p_user_list_pool->semid) < 0)
1282          {          {
# Line 1009  int query_user_online_info_by_uid(int32_ Line 1292  int query_user_online_info_by_uid(int32_
1292                  mid = (left + right) / 2;                  mid = (left + right) / 2;
1293                  if (uid < p_user_list_pool->p_online_current->index_uid[mid].uid)                  if (uid < p_user_list_pool->p_online_current->index_uid[mid].uid)
1294                  {                  {
1295                          right = mid;                          right = mid - 1;
1296                  }                  }
1297                  else if (uid > p_user_list_pool->p_online_current->index_uid[mid].uid)                  else if (uid > p_user_list_pool->p_online_current->index_uid[mid].uid)
1298                  {                  {
# Line 1041  int query_user_online_info_by_uid(int32_ Line 1324  int query_user_online_info_by_uid(int32_
1324                  }                  }
1325    
1326                  for (i = 0;                  for (i = 0;
1327                           left < p_user_list_pool->p_online_current->user_count && i < *p_user_cnt &&                           left < p_user_list_pool->p_online_current->user_count && i < user_cnt &&
1328                           uid == p_user_list_pool->p_online_current->index_uid[left].uid;                           uid == p_user_list_pool->p_online_current->index_uid[left].uid;
1329                           left++, i++)                           left++, i++)
1330                  {                  {
# Line 1095  int get_user_id_list(int32_t *p_uid_list Line 1378  int get_user_id_list(int32_t *p_uid_list
1378                  mid = (left + right) / 2;                  mid = (left + right) / 2;
1379                  if (start_uid < p_user_list_pool->p_current->index_uid[mid].uid)                  if (start_uid < p_user_list_pool->p_current->index_uid[mid].uid)
1380                  {                  {
1381                          right = mid;                          right = mid - 1;
1382                  }                  }
1383                  else if (start_uid > p_user_list_pool->p_current->index_uid[mid].uid)                  else if (start_uid > p_user_list_pool->p_current->index_uid[mid].uid)
1384                  {                  {
# Line 1129  int user_stat_update(void) Line 1412  int user_stat_update(void)
1412          return user_stat_map_update(&(p_user_list_pool->user_stat_map));          return user_stat_map_update(&(p_user_list_pool->user_stat_map));
1413  }  }
1414    
1415  int user_stat_get_article_cnt(int32_t uid)  int user_article_cnt_inc(int32_t uid, int n)
1416    {
1417            return user_stat_article_cnt_inc(&(p_user_list_pool->user_stat_map), uid, n);
1418    }
1419    
1420    int get_user_article_cnt(int32_t uid)
1421  {  {
1422          const USER_STAT *p_stat;          const USER_STAT *p_stat;
1423          int ret;          int ret;

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