/[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.37 by sysadm, Tue Nov 18 14:43:35 2025 UTC Revision 1.45 by sysadm, Wed Dec 3 05:31:11 2025 UTC
# Line 17  Line 17 
17  #include "user_list.h"  #include "user_list.h"
18  #include "user_stat.h"  #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/stat.h>
27    
28    #ifdef HAVE_SYSTEM_V
29  #include <sys/sem.h>  #include <sys/sem.h>
 #include <sys/shm.h>  
30    
31  #if defined(_SEM_SEMUN_UNDEFINED) || defined(__CYGWIN__)  #ifdef _SEM_SEMUN_UNDEFINED
32  union semun  union semun
33  {  {
34          int val;                           /* Value for SETVAL */          int val;                           /* Value for SETVAL */
35          struct semid_ds *buf;  /* Buffer for IPC_STAT, IPC_SET */          struct semid_ds *buf;  /* Buffer for IPC_STAT, IPC_SET */
36          unsigned short *array; /* Array for GETALL, SETALL */          unsigned short *array; /* Array for GETALL, SETALL */
37          struct seminfo *__buf; /* Buffer for IPC_INFO          struct seminfo *__buf; /* Buffer for IPC_INFO
38                                                            (Linux-specific) */                                                          (Linux-specific) */
39  };  };
40  #endif // #if defined(_SEM_SEMUN_UNDEFINED)  #endif // #ifdef _SEM_SEMUN_UNDEFINED
41    
42    #else
43    #include <semaphore.h>
44    #endif
45    
46  enum _user_list_constant_t  enum _user_list_constant_t
47  {  {
48          USER_LIST_TRY_LOCK_WAIT_TIME = 1, // second          USER_LIST_TRY_LOCK_WAIT_TIME = 1, // second
49          USER_LIST_TRY_LOCK_TIMES = 10,          USER_LIST_TRY_LOCK_TIMES = 10,
50            USER_LIST_DEAD_LOCK_TIMEOUT = 15, // second
51  };  };
52    
53  struct user_list_pool_t  struct user_list_pool_t
54  {  {
55          int shmid;          size_t shm_size;
56    #ifndef HAVE_SYSTEM_V
57            sem_t sem;
58            uint16_t read_lock_count;
59            uint16_t write_lock_count;
60    #else
61          int semid;          int semid;
62    #endif
63          USER_LIST user_list[2];          USER_LIST user_list[2];
64          int user_list_index_current;          int user_list_index_current;
65          int user_list_index_new;          int user_list_index_new;
# Line 58  struct user_list_pool_t Line 71  struct user_list_pool_t
71  };  };
72  typedef struct user_list_pool_t USER_LIST_POOL;  typedef struct user_list_pool_t USER_LIST_POOL;
73    
74    static char user_list_shm_name[FILE_NAME_LEN];
75  static USER_LIST_POOL *p_user_list_pool = NULL;  static USER_LIST_POOL *p_user_list_pool = NULL;
76  static TRIE_NODE *p_trie_action_dict = NULL;  static TRIE_NODE *p_trie_action_dict = NULL;
77    
# Line 77  const USER_ACTION_MAP user_action_map[] Line 91  const USER_ACTION_MAP user_action_map[]
91                  {"MENU", "菜单选择"},                  {"MENU", "菜单选择"},
92                  {"POST_ARTICLE", "撰写文章"},                  {"POST_ARTICLE", "撰写文章"},
93                  {"REPLY_ARTICLE", "回复文章"},                  {"REPLY_ARTICLE", "回复文章"},
94                    {"TOP10_MENU", "十大热门"},
95                  {"USER_LIST", "查花名册"},                  {"USER_LIST", "查花名册"},
96                  {"USER_ONLINE", "环顾四周"},                  {"USER_ONLINE", "环顾四周"},
97                  {"VIEW_ARTICLE", "阅读文章"},                  {"VIEW_ARTICLE", "阅读文章"},
# Line 85  const USER_ACTION_MAP user_action_map[] Line 100  const USER_ACTION_MAP user_action_map[]
100    
101  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);
102    
103  static int user_list_try_rd_lock(int semid, int wait_sec);  static int user_list_try_rd_lock(int wait_sec);
104  static int user_list_try_rw_lock(int semid, int wait_sec);  static int user_list_try_rw_lock(int wait_sec);
105  static int user_list_rd_unlock(int semid);  static int user_list_rd_unlock(void);
106  static int user_list_rw_unlock(int semid);  static int user_list_rw_unlock(void);
107  static int user_list_rd_lock(int semid);  static int user_list_rd_lock(void);
108  static int user_list_rw_lock(int semid);  static int user_list_rw_lock(void);
109    #ifndef HAVE_SYSTEM_V
110    static int user_list_reset_lock(void);
111    #endif
112    
113  static int user_list_load(MYSQL *db, USER_LIST *p_list);  static int user_list_load(MYSQL *db, USER_LIST *p_list);
114  static int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_online_list);  static int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_online_list);
# Line 388  int user_login_count_load(MYSQL *db) Line 406  int user_login_count_load(MYSQL *db)
406    
407  int user_list_pool_init(const char *filename)  int user_list_pool_init(const char *filename)
408  {  {
409          int shmid;          char filepath[FILE_PATH_LEN];
410          int semid;          int fd;
         int proj_id;  
         key_t key;  
411          size_t size;          size_t size;
412          void *p_shm;          void *p_shm;
413    #ifdef HAVE_SYSTEM_V
414            int proj_id;
415            key_t key;
416            int semid;
417          union semun arg;          union semun arg;
418    #endif
419          int i;          int i;
420    
421          if (p_user_list_pool != NULL || p_trie_action_dict != NULL)          if (p_user_list_pool != NULL || p_trie_action_dict != NULL)
# Line 419  int user_list_pool_init(const char *file Line 440  int user_list_pool_init(const char *file
440          }          }
441    
442          // Allocate shared memory          // Allocate shared memory
443          proj_id = (int)(time(NULL) % getpid());          size = sizeof(USER_LIST_POOL);
444          key = ftok(filename, proj_id);  
445          if (key == -1)          strncpy(filepath, filename, sizeof(filepath) - 1);
446            filepath[sizeof(filepath) - 1] = '\0';
447            snprintf(user_list_shm_name, sizeof(user_list_shm_name), "/USER_LIST_SHM_%s", basename(filepath));
448    
449            if (shm_unlink(user_list_shm_name) == -1 && errno != ENOENT)
450          {          {
451                  log_error("ftok(%s %d) error (%d)\n", filename, proj_id, errno);                  log_error("shm_unlink(%s) error (%d)\n", user_list_shm_name, errno);
452                  return -2;                  return -2;
453          }          }
454    
455          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)  
456          {          {
457                  log_error("shmget(size = %d) error (%d)\n", size, errno);                  log_error("shm_open(%s) error (%d)\n", user_list_shm_name, errno);
458                  return -3;                  return -2;
459          }          }
460          p_shm = shmat(shmid, NULL, 0);          if (ftruncate(fd, (off_t)size) == -1)
         if (p_shm == (void *)-1)  
461          {          {
462                  log_error("shmat(shmid=%d) error (%d)\n", shmid, errno);                  log_error("ftruncate(size=%d) error (%d)\n", size, errno);
463                  return -3;                  close(fd);
464                    return -2;
465            }
466    
467            p_shm = mmap(NULL, size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0L);
468            if (p_shm == MAP_FAILED)
469            {
470                    log_error("mmap() error (%d)\n", errno);
471                    close(fd);
472                    return -2;
473            }
474    
475            if (close(fd) < 0)
476            {
477                    log_error("close(fd) error (%d)\n", errno);
478                    return -1;
479          }          }
480    
481          p_user_list_pool = p_shm;          p_user_list_pool = p_shm;
482          p_user_list_pool->shmid = shmid;          p_user_list_pool->shm_size = size;
483    
484          // Allocate semaphore as user list pool lock          // Allocate semaphore as user list pool lock
485    #ifndef HAVE_SYSTEM_V
486            if (sem_init(&(p_user_list_pool->sem), 1, 1) == -1)
487            {
488                    log_error("sem_init() error (%d)\n", errno);
489                    return -3;
490            }
491    
492            p_user_list_pool->read_lock_count = 0;
493            p_user_list_pool->write_lock_count = 0;
494    #else
495            proj_id = (int)(time(NULL) % getpid());
496            key = ftok(filename, proj_id);
497            if (key == -1)
498            {
499                    log_error("ftok(%s %d) error (%d)\n", filename, proj_id, errno);
500                    return -2;
501            }
502    
503          size = 2; // r_sem and w_sem          size = 2; // r_sem and w_sem
504          semid = semget(key, (int)size, IPC_CREAT | IPC_EXCL | 0600);          semid = semget(key, (int)size, IPC_CREAT | IPC_EXCL | 0600);
505          if (semid == -1)          if (semid == -1)
# Line 465  int user_list_pool_init(const char *file Line 520  int user_list_pool_init(const char *file
520          }          }
521    
522          p_user_list_pool->semid = semid;          p_user_list_pool->semid = semid;
523    #endif
524    
525          // Set user counts to 0          // Set user counts to 0
526          p_user_list_pool->user_list[0].user_count = 0;          p_user_list_pool->user_list[0].user_count = 0;
# Line 483  int user_list_pool_init(const char *file Line 539  int user_list_pool_init(const char *file
539    
540  void user_list_pool_cleanup(void)  void user_list_pool_cleanup(void)
541  {  {
         int shmid;  
   
542          if (p_user_list_pool == NULL)          if (p_user_list_pool == NULL)
543          {          {
544                  return;                  return;
545          }          }
546    
547          shmid = p_user_list_pool->shmid;  #ifdef HAVE_SYSTEM_V
   
548          if (semctl(p_user_list_pool->semid, 0, IPC_RMID) == -1)          if (semctl(p_user_list_pool->semid, 0, IPC_RMID) == -1)
549          {          {
550                  log_error("semctl(semid = %d, IPC_RMID) error (%d)\n", p_user_list_pool->semid, errno);                  log_error("semctl(semid = %d, IPC_RMID) error (%d)\n", p_user_list_pool->semid, errno);
551          }          }
552    #else
553          if (shmdt(p_user_list_pool) == -1)          if (sem_destroy(&(p_user_list_pool->sem)) == -1)
554          {          {
555                  log_error("shmdt(shmid = %d) error (%d)\n", shmid, errno);                  log_error("sem_destroy() error (%d)\n", errno);
556          }          }
557    #endif
558    
559            detach_user_list_pool_shm();
560    
561          if (shmid != 0 && shmctl(shmid, IPC_RMID, NULL) == -1)          if (shm_unlink(user_list_shm_name) == -1 && errno != ENOENT)
562          {          {
563                  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);
564          }          }
565    
566          p_user_list_pool = NULL;          user_list_shm_name[0] = '\0';
567    
568          if (p_trie_action_dict != NULL)          if (p_trie_action_dict != NULL)
569          {          {
# Line 519  void user_list_pool_cleanup(void) Line 575  void user_list_pool_cleanup(void)
575    
576  int set_user_list_pool_shm_readonly(void)  int set_user_list_pool_shm_readonly(void)
577  {  {
578          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)  
         {  
                 log_error("p_user_list_pool not initialized\n");  
                 return -1;  
         }  
   
         shmid = p_user_list_pool->shmid;  
   
         // Remap shared memory in read-only mode  
 #if defined(__CYGWIN__)  
         if (shmdt(p_user_list_pool) == -1)  
579          {          {
580                  log_error("shmdt(user_list_pool) error (%d)\n", errno);                  log_error("mprotect() error (%d)\n", errno);
581                  return -1;                  return -1;
582          }          }
         p_shm = shmat(shmid, p_user_list_pool, SHM_RDONLY);  
 #else  
         p_shm = shmat(shmid, p_user_list_pool, SHM_RDONLY | SHM_REMAP);  
 #endif  
         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;  
583    
584          return 0;          return 0;
585  }  }
586    
587  int detach_user_list_pool_shm(void)  int detach_user_list_pool_shm(void)
588  {  {
589          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)
590          {          {
591                  log_error("shmdt(user_list_pool) error (%d)\n", errno);                  log_error("munmap() error (%d)\n", errno);
592                  return -1;                  return -1;
593          }          }
594    
# Line 613  int user_list_pool_reload(int online_use Line 645  int user_list_pool_reload(int online_use
645          mysql_close(db);          mysql_close(db);
646          db = NULL;          db = NULL;
647    
648          if (user_list_rw_lock(p_user_list_pool->semid) < 0)          if (user_list_rw_lock() < 0)
649          {          {
650                  log_error("user_list_rw_lock() error\n");                  log_error("user_list_rw_lock() error\n");
651                  ret = -3;                  ret = -3;
# Line 635  int user_list_pool_reload(int online_use Line 667  int user_list_pool_reload(int online_use
667                  p_user_list_pool->user_list_index_new = tmp;                  p_user_list_pool->user_list_index_new = tmp;
668          }          }
669    
670          if (user_list_rw_unlock(p_user_list_pool->semid) < 0)          if (user_list_rw_unlock() < 0)
671          {          {
672                  log_error("user_list_rw_unlock() error\n");                  log_error("user_list_rw_unlock() error\n");
673                  ret = -3;                  ret = -3;
# Line 648  cleanup: Line 680  cleanup:
680          return ret;          return ret;
681  }  }
682    
683  int user_list_try_rd_lock(int semid, int wait_sec)  int user_list_try_rd_lock(int wait_sec)
684  {  {
685    #ifdef HAVE_SYSTEM_V
686          struct sembuf sops[2];          struct sembuf sops[2];
 #if !defined(__CYGWIN__)  
         struct timespec timeout;  
687  #endif  #endif
688          int ret;          struct timespec timeout;
689            int ret = 0;
690    
691            if (p_user_list_pool == NULL)
692            {
693                    log_error("p_user_list_pool not initialized\n");
694                    return -1;
695            }
696    
697            timeout.tv_sec = wait_sec;
698            timeout.tv_nsec = 0;
699    
700    #ifdef HAVE_SYSTEM_V
701          sops[0].sem_num = 1; // w_sem          sops[0].sem_num = 1; // w_sem
702          sops[0].sem_op = 0;      // wait until unlocked          sops[0].sem_op = 0;      // wait until unlocked
703          sops[0].sem_flg = 0;          sops[0].sem_flg = 0;
# Line 664  int user_list_try_rd_lock(int semid, int Line 706  int user_list_try_rd_lock(int semid, int
706          sops[1].sem_op = 1;                     // lock          sops[1].sem_op = 1;                     // lock
707          sops[1].sem_flg = SEM_UNDO; // undo on terminate          sops[1].sem_flg = SEM_UNDO; // undo on terminate
708    
709  #if defined(__CYGWIN__)          ret = semtimedop(p_user_list_pool->semid, sops, 2, &timeout);
         ret = semop(semid, sops, 2);  
 #else  
         timeout.tv_sec = wait_sec;  
         timeout.tv_nsec = 0;  
   
         ret = semtimedop(semid, sops, 2, &timeout);  
 #endif  
710          if (ret == -1 && errno != EAGAIN && errno != EINTR)          if (ret == -1 && errno != EAGAIN && errno != EINTR)
711          {          {
712                  log_error("semop(lock read) error %d\n", errno);                  log_error("semop(lock read) error %d\n", errno);
713          }          }
714    #else
715            if (sem_timedwait(&(p_user_list_pool->sem), &timeout) == -1)
716            {
717                    if (errno != ETIMEDOUT && errno != EAGAIN && errno != EINTR)
718                    {
719                            log_error("sem_timedwait() error %d\n", errno);
720                    }
721                    return -1;
722            }
723    
724            if (p_user_list_pool->write_lock_count == 0)
725            {
726                    p_user_list_pool->read_lock_count++;
727            }
728            else
729            {
730                    errno = EAGAIN;
731                    ret = -1;
732            }
733    
734            if (sem_post(&(p_user_list_pool->sem)) == -1)
735            {
736                    log_error("sem_post() error %d\n", errno);
737                    return -1;
738            }
739    #endif
740    
741          return ret;          return ret;
742  }  }
743    
744  int user_list_try_rw_lock(int semid, int wait_sec)  int user_list_try_rw_lock(int wait_sec)
745  {  {
746    #ifdef HAVE_SYSTEM_V
747          struct sembuf sops[3];          struct sembuf sops[3];
 #if !defined(__CYGWIN__)  
         struct timespec timeout;  
748  #endif  #endif
749          int ret;          struct timespec timeout;
750            int ret = 0;
751    
752            if (p_user_list_pool == NULL)
753            {
754                    log_error("p_user_list_pool not initialized\n");
755                    return -1;
756            }
757    
758            timeout.tv_sec = wait_sec;
759            timeout.tv_nsec = 0;
760    
761    #ifdef HAVE_SYSTEM_V
762          sops[0].sem_num = 1; // w_sem          sops[0].sem_num = 1; // w_sem
763          sops[0].sem_op = 0;      // wait until unlocked          sops[0].sem_op = 0;      // wait until unlocked
764          sops[0].sem_flg = 0;          sops[0].sem_flg = 0;
# Line 700  int user_list_try_rw_lock(int semid, int Line 771  int user_list_try_rw_lock(int semid, int
771          sops[2].sem_op = 0;      // wait until unlocked          sops[2].sem_op = 0;      // wait until unlocked
772          sops[2].sem_flg = 0;          sops[2].sem_flg = 0;
773    
774  #if defined(__CYGWIN__)          ret = semtimedop(p_user_list_pool->semid, sops, 3, &timeout);
         ret = semop(semid, sops, 3);  
 #else  
         timeout.tv_sec = wait_sec;  
         timeout.tv_nsec = 0;  
   
         ret = semtimedop(semid, sops, 3, &timeout);  
 #endif  
775          if (ret == -1 && errno != EAGAIN && errno != EINTR)          if (ret == -1 && errno != EAGAIN && errno != EINTR)
776          {          {
777                  log_error("semop(lock write) error %d\n", errno);                  log_error("semop(lock write) error %d\n", errno);
778          }          }
779    #else
780            if (sem_timedwait(&(p_user_list_pool->sem), &timeout) == -1)
781            {
782                    if (errno != ETIMEDOUT && errno != EAGAIN && errno != EINTR)
783                    {
784                            log_error("sem_timedwait() error %d\n", errno);
785                    }
786                    return -1;
787            }
788    
789            if (p_user_list_pool->read_lock_count == 0 && p_user_list_pool->write_lock_count == 0)
790            {
791                    p_user_list_pool->write_lock_count++;
792            }
793            else
794            {
795                    errno = EAGAIN;
796                    ret = -1;
797            }
798    
799            if (sem_post(&(p_user_list_pool->sem)) == -1)
800            {
801                    log_error("sem_post() error %d\n", errno);
802                    return -1;
803            }
804    #endif
805    
806          return ret;          return ret;
807  }  }
808    
809  int user_list_rd_unlock(int semid)  int user_list_rd_unlock(void)
810  {  {
811    #ifdef HAVE_SYSTEM_V
812          struct sembuf sops[2];          struct sembuf sops[2];
813          int ret;  #endif
814            int ret = 0;
815    
816            if (p_user_list_pool == NULL)
817            {
818                    log_error("p_user_list_pool not initialized\n");
819                    return -1;
820            }
821    
822    #ifdef HAVE_SYSTEM_V
823          sops[0].sem_num = 0;                                     // r_sem          sops[0].sem_num = 0;                                     // r_sem
824          sops[0].sem_op = -1;                                     // unlock          sops[0].sem_op = -1;                                     // unlock
825          sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO; // no wait          sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO; // no wait
826    
827          ret = semop(semid, sops, 1);          ret = semop(p_user_list_pool->semid, sops, 1);
828          if (ret == -1 && errno != EAGAIN && errno != EINTR)          if (ret == -1 && errno != EAGAIN && errno != EINTR)
829          {          {
830                  log_error("semop(unlock read) error %d\n", errno);                  log_error("semop(unlock read) error %d\n", errno);
831          }          }
832    #else
833            if (sem_wait(&(p_user_list_pool->sem)) == -1)
834            {
835                    if (errno != ETIMEDOUT && errno != EAGAIN && errno != EINTR)
836                    {
837                            log_error("sem_wait() error %d\n", errno);
838                    }
839                    return -1;
840            }
841    
842            if (p_user_list_pool->read_lock_count > 0)
843            {
844                    p_user_list_pool->read_lock_count--;
845            }
846            else
847            {
848                    log_error("read_lock_count already 0\n");
849            }
850    
851            if (sem_post(&(p_user_list_pool->sem)) == -1)
852            {
853                    log_error("sem_post() error %d\n", errno);
854                    return -1;
855            }
856    #endif
857    
858          return ret;          return ret;
859  }  }
860    
861  int user_list_rw_unlock(int semid)  int user_list_rw_unlock(void)
862  {  {
863    #ifdef HAVE_SYSTEM_V
864          struct sembuf sops[1];          struct sembuf sops[1];
865          int ret;  #endif
866            int ret = 0;
867    
868            if (p_user_list_pool == NULL)
869            {
870                    log_error("p_user_list_pool not initialized\n");
871                    return -1;
872            }
873    
874    #ifdef HAVE_SYSTEM_V
875          sops[0].sem_num = 1;                                     // w_sem          sops[0].sem_num = 1;                                     // w_sem
876          sops[0].sem_op = -1;                                     // unlock          sops[0].sem_op = -1;                                     // unlock
877          sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO; // no wait          sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO; // no wait
878    
879          ret = semop(semid, sops, 1);          ret = semop(p_user_list_pool->semid, sops, 1);
880          if (ret == -1 && errno != EAGAIN && errno != EINTR)          if (ret == -1 && errno != EAGAIN && errno != EINTR)
881          {          {
882                  log_error("semop(unlock write) error %d\n", errno);                  log_error("semop(unlock write) error %d\n", errno);
883          }          }
884    #else
885            if (sem_wait(&(p_user_list_pool->sem)) == -1)
886            {
887                    if (errno != ETIMEDOUT && errno != EAGAIN && errno != EINTR)
888                    {
889                            log_error("sem_wait() error %d\n", errno);
890                    }
891                    return -1;
892            }
893    
894            if (p_user_list_pool->write_lock_count > 0)
895            {
896                    p_user_list_pool->write_lock_count--;
897            }
898            else
899            {
900                    log_error("write_lock_count already 0\n");
901            }
902    
903            if (sem_post(&(p_user_list_pool->sem)) == -1)
904            {
905                    log_error("sem_post() error %d\n", errno);
906                    return -1;
907            }
908    #endif
909    
910          return ret;          return ret;
911  }  }
912    
913  int user_list_rd_lock(int semid)  int user_list_rd_lock(void)
914  {  {
915          int timer = 0;          int timer = 0;
916          int ret = -1;          int ret = -1;
917            time_t tm_first_failure = 0;
918    
919            if (p_user_list_pool == NULL)
920            {
921                    log_error("p_user_list_pool not initialized\n");
922                    return -1;
923            }
924    
925          while (!SYS_server_exit)          while (!SYS_server_exit)
926          {          {
927                  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);
928                  if (ret == 0) // success                  if (ret == 0) // success
929                  {                  {
930                          break;                          break;
# Line 770  int user_list_rd_lock(int semid) Line 935  int user_list_rd_lock(int semid)
935                          if (timer % USER_LIST_TRY_LOCK_TIMES == 0)                          if (timer % USER_LIST_TRY_LOCK_TIMES == 0)
936                          {                          {
937                                  log_error("user_list_try_rd_lock() tried %d times\n", timer);                                  log_error("user_list_try_rd_lock() tried %d times\n", timer);
938    
939                                    if (time(NULL) - tm_first_failure >= USER_LIST_DEAD_LOCK_TIMEOUT)
940                                    {
941                                            log_error("Unable to acquire rw_lock for %d seconds\n", time(NULL) - tm_first_failure);
942    #ifndef HAVE_SYSTEM_V
943                                            user_list_reset_lock();
944                                            log_error("Reset POSIX semaphore to resolve dead lock\n");
945    #endif
946                                            break;
947                                    }
948                          }                          }
949                            usleep(100 * 1000); // 0.1 second
950                  }                  }
951                  else // failed                  else // failed
952                  {                  {
# Line 782  int user_list_rd_lock(int semid) Line 958  int user_list_rd_lock(int semid)
958          return ret;          return ret;
959  }  }
960    
961  int user_list_rw_lock(int semid)  int user_list_rw_lock(void)
962  {  {
963          int timer = 0;          int timer = 0;
964          int ret = -1;          int ret = -1;
965            time_t tm_first_failure = 0;
966    
967            if (p_user_list_pool == NULL)
968            {
969                    log_error("p_user_list_pool not initialized\n");
970                    return -1;
971            }
972    
973          while (!SYS_server_exit)          while (!SYS_server_exit)
974          {          {
975                  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);
976                  if (ret == 0) // success                  if (ret == 0) // success
977                  {                  {
978                          break;                          break;
# Line 800  int user_list_rw_lock(int semid) Line 983  int user_list_rw_lock(int semid)
983                          if (timer % USER_LIST_TRY_LOCK_TIMES == 0)                          if (timer % USER_LIST_TRY_LOCK_TIMES == 0)
984                          {                          {
985                                  log_error("user_list_try_rw_lock() tried %d times\n", timer);                                  log_error("user_list_try_rw_lock() tried %d times\n", timer);
986    
987                                    if (time(NULL) - tm_first_failure >= USER_LIST_DEAD_LOCK_TIMEOUT)
988                                    {
989                                            log_error("Unable to acquire rw_lock for %d seconds\n", time(NULL) - tm_first_failure);
990    #ifndef HAVE_SYSTEM_V
991                                            user_list_reset_lock();
992                                            log_error("Reset POSIX semaphore to resolve dead lock\n");
993    #endif
994                                            break;
995                                    }
996                          }                          }
997                            usleep(100 * 1000); // 0.1 second
998                  }                  }
999                  else // failed                  else // failed
1000                  {                  {
# Line 812  int user_list_rw_lock(int semid) Line 1006  int user_list_rw_lock(int semid)
1006          return ret;          return ret;
1007  }  }
1008    
1009    #ifndef HAVE_SYSTEM_V
1010    int user_list_reset_lock(void)
1011    {
1012            if (p_user_list_pool == NULL)
1013            {
1014                    log_error("p_user_list_pool not initialized\n");
1015                    return -1;
1016            }
1017    
1018            if (sem_destroy(&(p_user_list_pool->sem)) == -1)
1019            {
1020                    log_error("sem_destroy() error (%d)\n", errno);
1021            }
1022    
1023            p_user_list_pool->read_lock_count = 0;
1024            p_user_list_pool->write_lock_count = 0;
1025    
1026            if (sem_init(&(p_user_list_pool->sem), 1, 1) == -1)
1027            {
1028                    log_error("sem_init() error (%d)\n", errno);
1029            }
1030    
1031            return 0;
1032    }
1033    #endif
1034    
1035  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)
1036  {  {
1037          int ret = 0;          int ret = 0;
# Line 826  int query_user_list(int page_id, USER_IN Line 1046  int query_user_list(int page_id, USER_IN
1046          *p_page_count = 0;          *p_page_count = 0;
1047    
1048          // acquire lock of user list          // acquire lock of user list
1049          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock() < 0)
1050          {          {
1051                  log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
1052                  return -2;                  return -2;
# Line 859  int query_user_list(int page_id, USER_IN Line 1079  int query_user_list(int page_id, USER_IN
1079    
1080  cleanup:  cleanup:
1081          // release lock of user list          // release lock of user list
1082          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock() < 0)
1083          {          {
1084                  log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
1085                  ret = -1;                  ret = -1;
# Line 882  int query_user_online_list(int page_id, Line 1102  int query_user_online_list(int page_id,
1102          *p_page_count = 0;          *p_page_count = 0;
1103    
1104          // acquire lock of user list          // acquire lock of user list
1105          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock() < 0)
1106          {          {
1107                  log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
1108                  return -2;                  return -2;
# Line 914  int query_user_online_list(int page_id, Line 1134  int query_user_online_list(int page_id,
1134    
1135  cleanup:  cleanup:
1136          // release lock of user list          // release lock of user list
1137          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock() < 0)
1138          {          {
1139                  log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
1140                  ret = -1;                  ret = -1;
# Line 932  int get_user_list_count(int *p_user_cnt) Line 1152  int get_user_list_count(int *p_user_cnt)
1152          }          }
1153    
1154          // acquire lock of user list          // acquire lock of user list
1155          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock() < 0)
1156          {          {
1157                  log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
1158                  return -2;                  return -2;
# Line 941  int get_user_list_count(int *p_user_cnt) Line 1161  int get_user_list_count(int *p_user_cnt)
1161          *p_user_cnt = p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].user_count;          *p_user_cnt = p_user_list_pool->user_list[p_user_list_pool->user_list_index_current].user_count;
1162    
1163          // release lock of user list          // release lock of user list
1164          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock() < 0)
1165          {          {
1166                  log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
1167                  return -2;                  return -2;
# Line 959  int get_user_online_list_count(int *p_us Line 1179  int get_user_online_list_count(int *p_us
1179          }          }
1180    
1181          // acquire lock of user list          // acquire lock of user list
1182          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock() < 0)
1183          {          {
1184                  log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
1185                  return -2;                  return -2;
# Line 969  int get_user_online_list_count(int *p_us Line 1189  int get_user_online_list_count(int *p_us
1189          *p_guest_cnt = p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].guest_count;          *p_guest_cnt = p_user_list_pool->user_online_list[p_user_list_pool->user_online_list_index_current].guest_count;
1190    
1191          // release lock of user list          // release lock of user list
1192          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock() < 0)
1193          {          {
1194                  log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
1195                  return -2;                  return -2;
# Line 1002  int query_user_info(int32_t id, USER_INF Line 1222  int query_user_info(int32_t id, USER_INF
1222          }          }
1223    
1224          // acquire lock of user list          // acquire lock of user list
1225          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock() < 0)
1226          {          {
1227                  log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
1228                  return -2;                  return -2;
# Line 1015  int query_user_info(int32_t id, USER_INF Line 1235  int query_user_info(int32_t id, USER_INF
1235          }          }
1236    
1237          // release lock of user list          // release lock of user list
1238          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock() < 0)
1239          {          {
1240                  log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
1241                  ret = -1;                  ret = -1;
# Line 1039  int query_user_info_by_uid(int32_t uid, Line 1259  int query_user_info_by_uid(int32_t uid,
1259          }          }
1260    
1261          // acquire lock of user list          // acquire lock of user list
1262          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock() < 0)
1263          {          {
1264                  log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
1265                  return -2;                  return -2;
# Line 1081  int query_user_info_by_uid(int32_t uid, Line 1301  int query_user_info_by_uid(int32_t uid,
1301          }          }
1302    
1303          // release lock of user list          // release lock of user list
1304          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock() < 0)
1305          {          {
1306                  log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
1307                  ret = -1;                  ret = -1;
# Line 1111  int query_user_info_by_username(const ch Line 1331  int query_user_info_by_username(const ch
1331          prefix_len = strlen(username_prefix);          prefix_len = strlen(username_prefix);
1332    
1333          // acquire lock of user list          // acquire lock of user list
1334          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock() < 0)
1335          {          {
1336                  log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
1337                  return -2;                  return -2;
# Line 1215  int query_user_info_by_username(const ch Line 1435  int query_user_info_by_username(const ch
1435    
1436  cleanup:  cleanup:
1437          // release lock of user list          // release lock of user list
1438          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock() < 0)
1439          {          {
1440                  log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
1441                  ret = -1;                  ret = -1;
# Line 1235  int query_user_online_info(int32_t id, U Line 1455  int query_user_online_info(int32_t id, U
1455          }          }
1456    
1457          // acquire lock of user list          // acquire lock of user list
1458          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock() < 0)
1459          {          {
1460                  log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
1461                  return -2;                  return -2;
# Line 1248  int query_user_online_info(int32_t id, U Line 1468  int query_user_online_info(int32_t id, U
1468          }          }
1469    
1470          // release lock of user list          // release lock of user list
1471          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock() < 0)
1472          {          {
1473                  log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
1474                  ret = -1;                  ret = -1;
# Line 1277  int query_user_online_info_by_uid(int32_ Line 1497  int query_user_online_info_by_uid(int32_
1497          *p_user_cnt = 0;          *p_user_cnt = 0;
1498    
1499          // acquire lock of user list          // acquire lock of user list
1500          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock() < 0)
1501          {          {
1502                  log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
1503                  return -2;                  return -2;
# Line 1339  int query_user_online_info_by_uid(int32_ Line 1559  int query_user_online_info_by_uid(int32_
1559          }          }
1560    
1561          // release lock of user list          // release lock of user list
1562          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock() < 0)
1563          {          {
1564                  log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
1565                  ret = -1;                  ret = -1;
# Line 1363  int get_user_id_list(int32_t *p_uid_list Line 1583  int get_user_id_list(int32_t *p_uid_list
1583          }          }
1584    
1585          // acquire lock of user list          // acquire lock of user list
1586          if (user_list_rd_lock(p_user_list_pool->semid) < 0)          if (user_list_rd_lock() < 0)
1587          {          {
1588                  log_error("user_list_rd_lock() error\n");                  log_error("user_list_rd_lock() error\n");
1589                  return -2;                  return -2;
# Line 1397  int get_user_id_list(int32_t *p_uid_list Line 1617  int get_user_id_list(int32_t *p_uid_list
1617          *p_user_cnt = i;          *p_user_cnt = i;
1618    
1619          // release lock of user list          // release lock of user list
1620          if (user_list_rd_unlock(p_user_list_pool->semid) < 0)          if (user_list_rd_unlock() < 0)
1621          {          {
1622                  log_error("user_list_rd_unlock() error\n");                  log_error("user_list_rd_unlock() error\n");
1623                  ret = -1;                  ret = -1;

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