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

Diff of /lbbs/src/section_list.c

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

Revision 1.15 by sysadm, Sat May 24 03:41:34 2025 UTC Revision 1.40 by sysadm, Mon Oct 13 02:23:27 2025 UTC
# Line 14  Line 14 
14   *                                                                         *   *                                                                         *
15   ***************************************************************************/   ***************************************************************************/
16    
 #include "section_list.h"  
17  #include "log.h"  #include "log.h"
18    #include "section_list.h"
19  #include "trie_dict.h"  #include "trie_dict.h"
20    #include <errno.h>
21    #include <signal.h>
22  #include <stdio.h>  #include <stdio.h>
23    #include <stdlib.h>
24  #include <string.h>  #include <string.h>
 #include <signal.h>  
25  #include <unistd.h>  #include <unistd.h>
26  #include <stdlib.h>  #include <sys/ipc.h>
 #include <errno.h>  
27  #include <sys/param.h>  #include <sys/param.h>
28    #include <sys/sem.h>
29  #include <sys/shm.h>  #include <sys/shm.h>
 #include <sys/ipc.h>  
30    
31  #define ARTICLE_BLOCK_PER_SHM 400                 // sizeof(ARTICLE_BLOCK) * ARTICLE_BLOCK_PER_SHM is the size of each shm segment to allocate  #ifdef _SEM_SEMUN_UNDEFINED
32  #define ARTICLE_BLOCK_SHM_COUNT_LIMIT 256 // limited by length (8-bit) of proj_id in ftok(path, proj_id)  union semun
33    {
34            int val;                           /* Value for SETVAL */
35            struct semid_ds *buf;  /* Buffer for IPC_STAT, IPC_SET */
36            unsigned short *array; /* Array for GETALL, SETALL */
37            struct seminfo *__buf; /* Buffer for IPC_INFO
38                                                              (Linux-specific) */
39    };
40    #endif // #ifdef _SEM_SEMUN_UNDEFINED
41    
42    #define SECTION_TRY_LOCK_WAIT_TIME 1 // second
43    #define SECTION_TRY_LOCK_TIMES 10
44    
45    #define ARTICLE_BLOCK_PER_SHM 1000               // sizeof(ARTICLE_BLOCK) * ARTICLE_BLOCK_PER_SHM is the size of each shm segment to allocate
46    #define ARTICLE_BLOCK_SHM_COUNT_LIMIT 80 // limited by length (8-bit) of proj_id in ftok(path, proj_id)
47  #define ARTICLE_BLOCK_PER_POOL (ARTICLE_BLOCK_PER_SHM * ARTICLE_BLOCK_SHM_COUNT_LIMIT)  #define ARTICLE_BLOCK_PER_POOL (ARTICLE_BLOCK_PER_SHM * ARTICLE_BLOCK_SHM_COUNT_LIMIT)
48    
49  #define CALCULATE_PAGE_THRESHOLD 100 // Adjust to tune performance of move topic  #define CALCULATE_PAGE_THRESHOLD 100 // Adjust to tune performance of moving topic between sections
50    
51  #define SID_STR_LEN 5 // 32-bit + NULL  #define SID_STR_LEN 5 // 32-bit + NULL
52    
53  struct article_block_t  struct article_block_t
54  {  {
55          ARTICLE articles[ARTICLE_PER_BLOCK];          ARTICLE articles[ARTICLE_PER_BLOCK];
56          int32_t article_count;          int article_count;
57          struct article_block_t *p_next_block;          struct article_block_t *p_next_block;
58  };  };
59  typedef struct article_block_t ARTICLE_BLOCK;  typedef struct article_block_t ARTICLE_BLOCK;
60    
 struct article_block_shm_t  
 {  
         int shmid;  
         void *p_shm;  
 };  
 typedef struct article_block_shm_t ARTICLE_BLOCK_SHM;  
   
61  struct article_block_pool_t  struct article_block_pool_t
62  {  {
63          ARTICLE_BLOCK_SHM shm_pool[ARTICLE_BLOCK_SHM_COUNT_LIMIT];          int shmid;
64            struct
65            {
66                    int shmid;
67                    void *p_shm;
68            } shm_pool[ARTICLE_BLOCK_SHM_COUNT_LIMIT];
69          int shm_count;          int shm_count;
70          ARTICLE_BLOCK *p_block_free_list;          ARTICLE_BLOCK *p_block_free_list;
71          ARTICLE_BLOCK *p_block[ARTICLE_BLOCK_PER_POOL];          ARTICLE_BLOCK *p_block[ARTICLE_BLOCK_PER_POOL];
72          int32_t block_count;          int block_count;
73  };  };
74  typedef struct article_block_pool_t ARTICLE_BLOCK_POOL;  typedef struct article_block_pool_t ARTICLE_BLOCK_POOL;
75    
76  static ARTICLE_BLOCK_POOL *p_article_block_pool = NULL;  static ARTICLE_BLOCK_POOL *p_article_block_pool = NULL;
77    
78  static int section_list_pool_shmid;  SECTION_LIST_POOL *p_section_list_pool = NULL;
 static SECTION_LIST *p_section_list_pool = NULL;  
 static int section_list_count = 0;  
 static TRIE_NODE *p_trie_dict_section_by_name = NULL;  
 static TRIE_NODE *p_trie_dict_section_by_sid = NULL;  
79    
80  int article_block_init(const char *filename, int block_count)  int article_block_init(const char *filename, int block_count)
81  {  {
# Line 86  int article_block_init(const char *filen Line 95  int article_block_init(const char *filen
95                  return -1;                  return -1;
96          }          }
97    
98          if (block_count > ARTICLE_BLOCK_PER_POOL)          if (block_count <= 0 || block_count > ARTICLE_BLOCK_PER_POOL)
99          {          {
100                  log_error("article_block_count exceed limit %d\n", ARTICLE_BLOCK_PER_POOL);                  log_error("article_block_count exceed limit %d\n", ARTICLE_BLOCK_PER_POOL);
101                  return -2;                  return -2;
102          }          }
103    
104          p_article_block_pool = calloc(1, sizeof(ARTICLE_BLOCK_POOL));          // Allocate shared memory
105          if (p_article_block_pool == NULL)          proj_id = ARTICLE_BLOCK_SHM_COUNT_LIMIT; // keep different from proj_id used to create block shm
106            key = ftok(filename, proj_id);
107            if (key == -1)
108          {          {
109                  log_error("calloc(ARTICLE_BLOCK_POOL) OOM\n");                  log_error("ftok(%s, %d) error (%d)\n", filename, proj_id, errno);
110                  return -2;                  return -3;
111          }          }
112    
113          // Allocate shared memory          size = sizeof(ARTICLE_BLOCK_POOL);
114            shmid = shmget(key, size, IPC_CREAT | IPC_EXCL | 0600);
115            if (shmid == -1)
116            {
117                    log_error("shmget(article_block_pool_shm, size = %d) error (%d)\n", size, errno);
118                    return -3;
119            }
120            p_shm = shmat(shmid, NULL, 0);
121            if (p_shm == (void *)-1)
122            {
123                    log_error("shmat(shmid = %d) error (%d)\n", shmid, errno);
124                    return -3;
125            }
126    
127            p_article_block_pool = p_shm;
128            p_article_block_pool->shmid = shmid;
129    
130          p_article_block_pool->shm_count = 0;          p_article_block_pool->shm_count = 0;
131          pp_block_next = &(p_article_block_pool->p_block_free_list);          pp_block_next = &(p_article_block_pool->p_block_free_list);
132    
# Line 108  int article_block_init(const char *filen Line 135  int article_block_init(const char *filen
135                  block_count_in_shm = MIN(block_count, ARTICLE_BLOCK_PER_SHM);                  block_count_in_shm = MIN(block_count, ARTICLE_BLOCK_PER_SHM);
136                  block_count -= block_count_in_shm;                  block_count -= block_count_in_shm;
137    
138                  proj_id = getpid() + p_article_block_pool->shm_count;                  proj_id = p_article_block_pool->shm_count;
139                  key = ftok(filename, proj_id);                  key = ftok(filename, proj_id);
140                  if (key == -1)                  if (key == -1)
141                  {                  {
# Line 117  int article_block_init(const char *filen Line 144  int article_block_init(const char *filen
144                          return -3;                          return -3;
145                  }                  }
146    
147                  size = sizeof(shmid) + sizeof(ARTICLE_BLOCK) * (size_t)block_count_in_shm;                  size = sizeof(ARTICLE_BLOCK) * (size_t)block_count_in_shm;
148                  shmid = shmget(key, size, IPC_CREAT | IPC_EXCL | 0600);                  shmid = shmget(key, size, IPC_CREAT | IPC_EXCL | 0600);
149                  if (shmid == -1)                  if (shmid == -1)
150                  {                  {
# Line 161  int article_block_init(const char *filen Line 188  int article_block_init(const char *filen
188    
189  void article_block_cleanup(void)  void article_block_cleanup(void)
190  {  {
191          if (p_article_block_pool != NULL)          int shmid;
192    
193            if (p_article_block_pool == NULL)
194          {          {
195                  for (int i = 0; i < p_article_block_pool->shm_count; i++)                  return;
196            }
197    
198            for (int i = 0; i < p_article_block_pool->shm_count; i++)
199            {
200                    if (shmdt((p_article_block_pool->shm_pool + i)->p_shm) == -1)
201                  {                  {
202                          if (shmdt((p_article_block_pool->shm_pool + i)->p_shm) == -1)                          log_error("shmdt(shmid = %d) error (%d)\n", (p_article_block_pool->shm_pool + i)->shmid, errno);
203                          {                  }
                                 log_error("shmdt(shmid = %d) error (%d)\n", (p_article_block_pool->shm_pool + i)->shmid, errno);  
                         }  
204    
205                          if (shmctl((p_article_block_pool->shm_pool + i)->shmid, IPC_RMID, NULL) == -1)                  if (shmctl((p_article_block_pool->shm_pool + i)->shmid, IPC_RMID, NULL) == -1)
206                          {                  {
207                                  log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", (p_article_block_pool->shm_pool + i)->shmid, errno);                          log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", (p_article_block_pool->shm_pool + i)->shmid, errno);
                         }  
208                  }                  }
209            }
210    
211            shmid = p_article_block_pool->shmid;
212    
213            if (shmdt(p_article_block_pool) == -1)
214            {
215                    log_error("shmdt(shmid = %d) error (%d)\n", shmid, errno);
216            }
217    
218                  free(p_article_block_pool);          if (shmctl(shmid, IPC_RMID, NULL) == -1)
219                  p_article_block_pool = NULL;          {
220                    log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", shmid, errno);
221          }          }
222    
223            p_article_block_pool = NULL;
224    }
225    
226    int set_article_block_shm_readonly(void)
227    {
228            int shmid;
229            void *p_shm;
230            int i;
231    
232            if (p_article_block_pool == NULL)
233            {
234                    log_error("article_block_pool not initialized\n");
235                    return -1;
236            }
237    
238            for (i = 0; i < p_article_block_pool->shm_count; i++)
239            {
240                    shmid = (p_article_block_pool->shm_pool + i)->shmid;
241    
242                    // Remap shared memory in read-only mode
243                    p_shm = shmat(shmid, (p_article_block_pool->shm_pool + i)->p_shm, SHM_RDONLY | SHM_REMAP);
244                    if (p_shm == (void *)-1)
245                    {
246                            log_error("shmat(article_block_pool shmid = %d) error (%d)\n", shmid, errno);
247                            return -2;
248                    }
249            }
250    
251            return 0;
252    }
253    
254    int detach_article_block_shm(void)
255    {
256            int shmid;
257    
258            if (p_article_block_pool == NULL)
259            {
260                    return -1;
261            }
262    
263            for (int i = 0; i < p_article_block_pool->shm_count; i++)
264            {
265                    if ((p_article_block_pool->shm_pool + i)->p_shm != NULL && shmdt((p_article_block_pool->shm_pool + i)->p_shm) == -1)
266                    {
267                            log_error("shmdt(shmid = %d) error (%d)\n", (p_article_block_pool->shm_pool + i)->shmid, errno);
268                            return -2;
269                    }
270            }
271    
272            shmid = p_article_block_pool->shmid;
273    
274            if (shmdt(p_article_block_pool) == -1)
275            {
276                    log_error("shmdt(shmid = %d) error (%d)\n", shmid, errno);
277                    return -3;
278            }
279    
280            p_article_block_pool = NULL;
281    
282            return 0;
283  }  }
284    
285  inline static ARTICLE_BLOCK *pop_free_article_block(void)  inline static ARTICLE_BLOCK *pop_free_article_block(void)
# Line 285  ARTICLE *article_block_find_by_aid(int32 Line 386  ARTICLE *article_block_find_by_aid(int32
386                  }                  }
387          }          }
388    
389          return (p_block->articles + left);          if (aid != p_block->articles[left].aid) // not found
390            {
391                    return NULL;
392            }
393    
394            return (p_block->articles + left); // found
395  }  }
396    
397  ARTICLE *article_block_find_by_index(int index)  ARTICLE *article_block_find_by_index(int index)
# Line 315  ARTICLE *article_block_find_by_index(int Line 421  ARTICLE *article_block_find_by_index(int
421          return (p_block->articles + (index % ARTICLE_PER_BLOCK));          return (p_block->articles + (index % ARTICLE_PER_BLOCK));
422  }  }
423    
424  extern int section_list_pool_init(const char *filename)  extern int section_list_init(const char *filename)
425  {  {
426            int semid;
427          int shmid;          int shmid;
428          int proj_id;          int proj_id;
429          key_t key;          key_t key;
430          size_t size;          size_t size;
431          void *p_shm;          void *p_shm;
432            union semun arg;
433            int i;
434    
435          if (p_section_list_pool == NULL || p_trie_dict_section_by_name == NULL || p_trie_dict_section_by_sid == NULL)          if (p_section_list_pool != NULL)
         {  
                 section_list_pool_cleanup();  
         }  
   
         p_section_list_pool = calloc(BBS_max_section, sizeof(SECTION_LIST));  
         if (p_section_list_pool == NULL)  
436          {          {
437                  log_error("calloc(%d SECTION_LIST) OOM\n", BBS_max_section);                  section_list_cleanup();
                 return -1;  
438          }          }
439    
440          proj_id = (int)(time(NULL) % getpid());          proj_id = (int)(time(NULL) % getpid());
# Line 343  extern int section_list_pool_init(const Line 445  extern int section_list_pool_init(const
445                  return -3;                  return -3;
446          }          }
447    
448          size = sizeof(shmid) + sizeof(SECTION_LIST) * BBS_max_section;          // Allocate shared memory
449            size = sizeof(SECTION_LIST_POOL);
450          shmid = shmget(key, size, IPC_CREAT | IPC_EXCL | 0600);          shmid = shmget(key, size, IPC_CREAT | IPC_EXCL | 0600);
451          if (shmid == -1)          if (shmid == -1)
452          {          {
453                  log_error("shmget(section_list_pool, size = %d) error (%d)\n", size, errno);                  log_error("shmget(section_list_pool_shm, size = %d) error (%d)\n", size, errno);
454                  return -3;                  return -3;
455          }          }
456          p_shm = shmat(shmid, NULL, 0);          p_shm = shmat(shmid, NULL, 0);
# Line 357  extern int section_list_pool_init(const Line 460  extern int section_list_pool_init(const
460                  return -3;                  return -3;
461          }          }
462    
         section_list_pool_shmid = shmid;  
463          p_section_list_pool = p_shm;          p_section_list_pool = p_shm;
464          section_list_count = 0;          p_section_list_pool->shmid = shmid;
465            p_section_list_pool->section_count = 0;
466    
467          p_trie_dict_section_by_name = trie_dict_create();          // Allocate semaphore as section locks
468          if (p_trie_dict_section_by_name == NULL)          size = 2 * (BBS_max_section + 1); // r_sem and w_sem per section, the last pair for all sections
469            semid = semget(key, (int)size, IPC_CREAT | IPC_EXCL | 0600);
470            if (semid == -1)
471            {
472                    log_error("semget(section_list_pool_sem, size = %d) error (%d)\n", size, errno);
473                    return -3;
474            }
475    
476            // Initialize sem value to 0
477            arg.val = 0;
478            for (i = 0; i < size; i++)
479            {
480                    if (semctl(semid, i, SETVAL, arg) == -1)
481                    {
482                            log_error("semctl(section_list_pool_sem, SETVAL) error (%d)\n", errno);
483                            return -3;
484                    }
485            }
486    
487            p_section_list_pool->semid = semid;
488    
489            p_section_list_pool->p_trie_dict_section_by_name = trie_dict_create();
490            if (p_section_list_pool->p_trie_dict_section_by_name == NULL)
491          {          {
492                  log_error("trie_dict_create() OOM\n", BBS_max_section);                  log_error("trie_dict_create() OOM\n", BBS_max_section);
493                  return -2;                  return -2;
494          }          }
495    
496          p_trie_dict_section_by_sid = trie_dict_create();          p_section_list_pool->p_trie_dict_section_by_sid = trie_dict_create();
497          if (p_trie_dict_section_by_sid == NULL)          if (p_section_list_pool->p_trie_dict_section_by_sid == NULL)
498          {          {
499                  log_error("trie_dict_create() OOM\n", BBS_max_section);                  log_error("trie_dict_create() OOM\n", BBS_max_section);
500                  return -2;                  return -2;
# Line 378  extern int section_list_pool_init(const Line 503  extern int section_list_pool_init(const
503          return 0;          return 0;
504  }  }
505    
506    void section_list_cleanup(void)
507    {
508            int shmid;
509    
510            if (p_section_list_pool == NULL)
511            {
512                    return;
513            }
514    
515            if (p_section_list_pool->p_trie_dict_section_by_name != NULL)
516            {
517                    trie_dict_destroy(p_section_list_pool->p_trie_dict_section_by_name);
518                    p_section_list_pool->p_trie_dict_section_by_name = NULL;
519            }
520    
521            if (p_section_list_pool->p_trie_dict_section_by_sid != NULL)
522            {
523                    trie_dict_destroy(p_section_list_pool->p_trie_dict_section_by_sid);
524                    p_section_list_pool->p_trie_dict_section_by_sid = NULL;
525            }
526    
527            shmid = p_section_list_pool->shmid;
528    
529            if (semctl(p_section_list_pool->semid, 0, IPC_RMID) == -1)
530            {
531                    log_error("semctl(semid = %d, IPC_RMID) error (%d)\n", p_section_list_pool->semid, errno);
532            }
533    
534            if (shmdt(p_section_list_pool) == -1)
535            {
536                    log_error("shmdt(shmid = %d) error (%d)\n", shmid, errno);
537            }
538    
539            if (shmctl(shmid, IPC_RMID, NULL) == -1)
540            {
541                    log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", shmid, errno);
542            }
543    
544            p_section_list_pool = NULL;
545    }
546    
547    int set_section_list_shm_readonly(void)
548    {
549            int shmid;
550            void *p_shm;
551    
552            if (p_section_list_pool == NULL)
553            {
554                    log_error("p_section_list_pool not initialized\n");
555                    return -1;
556            }
557    
558            shmid = p_section_list_pool->shmid;
559    
560            // Remap shared memory in read-only mode
561            p_shm = shmat(shmid, p_section_list_pool, SHM_RDONLY | SHM_REMAP);
562            if (p_shm == (void *)-1)
563            {
564                    log_error("shmat(section_list_pool shmid = %d) error (%d)\n", shmid, errno);
565                    return -3;
566            }
567    
568            p_section_list_pool = p_shm;
569    
570            return 0;
571    }
572    
573    int detach_section_list_shm(void)
574    {
575            if (p_section_list_pool != NULL && shmdt(p_section_list_pool) == -1)
576            {
577                    log_error("shmdt(section_list_pool) error (%d)\n", errno);
578                    return -1;
579            }
580    
581            p_section_list_pool = NULL;
582    
583            return 0;
584    }
585    
586  inline static void sid_to_str(int32_t sid, char *p_sid_str)  inline static void sid_to_str(int32_t sid, char *p_sid_str)
587  {  {
588          uint32_t u_sid;          uint32_t u_sid;
# Line 392  inline static void sid_to_str(int32_t si Line 597  inline static void sid_to_str(int32_t si
597          p_sid_str[i] = '\0';          p_sid_str[i] = '\0';
598  }  }
599    
600  SECTION_LIST *section_list_create(int32_t sid, const char *sname, const char *stitle, const char *master_name)  SECTION_LIST *section_list_create(int32_t sid, const char *sname, const char *stitle, const char *master_list)
601  {  {
602          SECTION_LIST *p_section;          SECTION_LIST *p_section;
603          char sid_str[SID_STR_LEN];          char sid_str[SID_STR_LEN];
604    
605          if (p_section_list_pool == NULL || p_trie_dict_section_by_name == NULL || p_trie_dict_section_by_sid == NULL)          if (p_section_list_pool == NULL)
606          {          {
607                  log_error("session_list_pool not initialized\n");                  log_error("session_list_pool not initialized\n");
608                  return NULL;                  return NULL;
609          }          }
610    
611          if (section_list_count >= BBS_max_section)          if (p_section_list_pool->section_count >= BBS_max_section)
612          {          {
613                  log_error("section_list_count exceed limit %d >= %d\n", section_list_count, BBS_max_section);                  log_error("section_count reach limit %d >= %d\n", p_section_list_pool->section_count, BBS_max_section);
614                  return NULL;                  return NULL;
615          }          }
616    
617          sid_to_str(sid, sid_str);          sid_to_str(sid, sid_str);
618    
619          p_section = p_section_list_pool + section_list_count;          p_section = p_section_list_pool->sections + p_section_list_pool->section_count;
620    
621          p_section->sid = sid;          p_section->sid = sid;
622            p_section->ex_menu_tm = 0;
623    
624          strncpy(p_section->sname, sname, sizeof(p_section->sname - 1));          strncpy(p_section->sname, sname, sizeof(p_section->sname) - 1);
625          p_section->sname[sizeof(p_section->sname - 1)] = '\0';          p_section->sname[sizeof(p_section->sname) - 1] = '\0';
626    
627          strncpy(p_section->stitle, stitle, sizeof(p_section->stitle - 1));          strncpy(p_section->stitle, stitle, sizeof(p_section->stitle) - 1);
628          p_section->stitle[sizeof(p_section->stitle - 1)] = '\0';          p_section->stitle[sizeof(p_section->stitle) - 1] = '\0';
629    
630          strncpy(p_section->master_name, master_name, sizeof(p_section->master_name - 1));          strncpy(p_section->master_list, master_list, sizeof(p_section->master_list) - 1);
631          p_section->master_name[sizeof(p_section->master_name - 1)] = '\0';          p_section->master_list[sizeof(p_section->master_list) - 1] = '\0';
632    
633          if (trie_dict_set(p_trie_dict_section_by_name, sname, section_list_count) != 1)          if (trie_dict_set(p_section_list_pool->p_trie_dict_section_by_name, sname, p_section_list_pool->section_count) != 1)
634          {          {
635                  log_error("trie_dict_set(section, %s, %d) error\n", sname, section_list_count);                  log_error("trie_dict_set(section, %s, %d) error\n", sname, p_section_list_pool->section_count);
636                  return NULL;                  return NULL;
637          }          }
638    
639          if (trie_dict_set(p_trie_dict_section_by_sid, sid_str, section_list_count) != 1)          if (trie_dict_set(p_section_list_pool->p_trie_dict_section_by_sid, sid_str, p_section_list_pool->section_count) != 1)
640          {          {
641                  log_error("trie_dict_set(section, %d, %d) error\n", sid, section_list_count);                  log_error("trie_dict_set(section, %d, %d) error\n", sid, p_section_list_pool->section_count);
                 log_std("Debug %x %x %x %x\n", sid_str[0], sid_str[1], sid_str[2], sid_str[3]);  
642                  return NULL;                  return NULL;
643          }          }
644    
645          section_list_reset_articles(p_section);          section_list_reset_articles(p_section);
646    
647          section_list_count++;          p_section_list_pool->section_count++;
648    
649          return p_section;          return p_section;
650  }  }
# Line 455  void section_list_reset_articles(SECTION Line 660  void section_list_reset_articles(SECTION
660    
661          p_section->page_count = 0;          p_section->page_count = 0;
662          p_section->last_page_visible_article_count = 0;          p_section->last_page_visible_article_count = 0;
 }  
   
 void section_list_pool_cleanup(void)  
 {  
         if (p_trie_dict_section_by_name != NULL)  
         {  
                 trie_dict_destroy(p_trie_dict_section_by_name);  
                 p_trie_dict_section_by_name = NULL;  
         }  
   
         if (p_trie_dict_section_by_sid != NULL)  
         {  
                 trie_dict_destroy(p_trie_dict_section_by_sid);  
                 p_trie_dict_section_by_sid = NULL;  
         }  
   
         if (p_section_list_pool != NULL)  
         {  
                 if (shmdt(p_section_list_pool) == -1)  
                 {  
                         log_error("shmdt(shmid = %d) error (%d)\n", section_list_pool_shmid, errno);  
                 }  
                 p_section_list_pool = NULL;  
663    
664                  if (shmctl(section_list_pool_shmid, IPC_RMID, NULL) == -1)          p_section->ontop_article_count = 0;
                 {  
                         log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", section_list_pool_shmid, errno);  
                 }  
         }  
   
         section_list_count = 0;  
665  }  }
666    
667  SECTION_LIST *section_list_find_by_name(const char *sname)  SECTION_LIST *section_list_find_by_name(const char *sname, int64_t *p_index)
668  {  {
669          int64_t index;          int64_t index;
670            int ret;
671    
672          if (p_section_list_pool == NULL || p_trie_dict_section_by_name == NULL)          if (p_section_list_pool == NULL)
673          {          {
674                  log_error("section_list not initialized\n");                  log_error("section_list not initialized\n");
675                  return NULL;                  return NULL;
676          }          }
677    
678          if (trie_dict_get(p_trie_dict_section_by_name, sname, &index) != 1)          ret = trie_dict_get(p_section_list_pool->p_trie_dict_section_by_name, sname, &index);
679            if (ret < 0)
680          {          {
681                  log_error("trie_dict_get(section, %s) error\n", sname);                  log_error("trie_dict_get(section, %s) error\n", sname);
682                  return NULL;                  return NULL;
683          }          }
684            else if (ret == 0)
685            {
686                    return NULL;
687            }
688    
689            if (p_index != NULL)
690            {
691                    *p_index = index;
692            }
693    
694          return (p_section_list_pool + index);          return (p_section_list_pool->sections + index);
695  }  }
696    
697  SECTION_LIST *section_list_find_by_sid(int32_t sid)  SECTION_LIST *section_list_find_by_sid(int32_t sid, int64_t *p_index)
698  {  {
699          int64_t index;          int64_t index;
700            int ret;
701          char sid_str[SID_STR_LEN];          char sid_str[SID_STR_LEN];
702    
703          if (p_section_list_pool == NULL || p_trie_dict_section_by_sid == NULL)          if (p_section_list_pool == NULL)
704          {          {
705                  log_error("section_list not initialized\n");                  log_error("section_list not initialized\n");
706                  return NULL;                  return NULL;
# Line 520  SECTION_LIST *section_list_find_by_sid(i Line 708  SECTION_LIST *section_list_find_by_sid(i
708    
709          sid_to_str(sid, sid_str);          sid_to_str(sid, sid_str);
710    
711          if (trie_dict_get(p_trie_dict_section_by_sid, sid_str, &index) != 1)          ret = trie_dict_get(p_section_list_pool->p_trie_dict_section_by_sid, sid_str, &index);
712            if (ret < 0)
713          {          {
714                  log_error("trie_dict_get(section, %d) error\n", sid);                  log_error("trie_dict_get(section, %d) error\n", sid);
715                  return NULL;                  return NULL;
716          }          }
717            else if (ret == 0)
718            {
719                    return NULL;
720            }
721    
722          return (p_section_list_pool + index);          if (p_index != NULL)
723            {
724                    *p_index = index;
725            }
726    
727            return (p_section_list_pool->sections + index);
728  }  }
729    
730  int section_list_append_article(SECTION_LIST *p_section, const ARTICLE *p_article_src)  int section_list_append_article(SECTION_LIST *p_section, const ARTICLE *p_article_src)
# Line 664  int section_list_append_article(SECTION_ Line 862  int section_list_append_article(SECTION_
862                  p_section->last_page_visible_article_count++;                  p_section->last_page_visible_article_count++;
863          }          }
864    
865            if (p_article->ontop && section_list_update_article_ontop(p_section, p_article) < 0)
866            {
867                    log_error("section_list_update_article_ontop(sid=%d, aid=%d) error\n",
868                                      p_section->sid, p_article->aid);
869                    return -5;
870            }
871    
872          return 0;          return 0;
873  }  }
874    
# Line 675  int section_list_set_article_visible(SEC Line 880  int section_list_set_article_visible(SEC
880    
881          if (p_section == NULL)          if (p_section == NULL)
882          {          {
883                  log_error("section_list_set_article_visible() NULL pointer error\n");                  log_error("NULL pointer error\n");
884                  return -2;                  return -1;
885          }          }
886    
887          p_article = article_block_find_by_aid(aid);          p_article = article_block_find_by_aid(aid);
# Line 687  int section_list_set_article_visible(SEC Line 892  int section_list_set_article_visible(SEC
892    
893          if (p_section->sid != p_article->sid)          if (p_section->sid != p_article->sid)
894          {          {
895                  log_error("section_list_set_article_visible() error: section sid %d != article sid %d\n", p_section->sid, p_article->sid);                  log_error("Inconsistent section sid %d != article sid %d\n", p_section->sid, p_article->sid);
896                  return -2;                  return -2;
897          }          }
898    
# Line 738  int section_list_set_article_visible(SEC Line 943  int section_list_set_article_visible(SEC
943          return affected_count;          return affected_count;
944  }  }
945    
946    int section_list_update_article_ontop(SECTION_LIST *p_section, ARTICLE *p_article)
947    {
948            int i;
949    
950            if (p_section == NULL || p_article == NULL)
951            {
952                    log_error("NULL pointer error\n");
953                    return -1;
954            }
955    
956            if (p_section->sid != p_article->sid)
957            {
958                    log_error("Inconsistent section sid %d != article sid %d\n", p_section->sid, p_article->sid);
959                    return -2;
960            }
961    
962            if (p_article->ontop)
963            {
964                    for (i = 0; i < p_section->ontop_article_count; i++)
965                    {
966                            if (p_section->p_ontop_articles[i]->aid == p_article->aid)
967                            {
968                                    log_error("Inconsistent state found: article %d already ontop in section %d\n", p_article->aid, p_section->sid);
969                                    return 0;
970                            }
971                            else if (p_section->p_ontop_articles[i]->aid > p_article->aid)
972                            {
973                                    break;
974                            }
975                    }
976    
977                    // Remove the oldest one if the array of ontop articles is full
978                    if (p_section->ontop_article_count >= BBS_ontop_article_limit_per_section)
979                    {
980                            if (i == 0) // p_article is the oldest one
981                            {
982                                    return 0;
983                            }
984                            memmove((void *)(p_section->p_ontop_articles),
985                                            (void *)(p_section->p_ontop_articles + 1),
986                                            sizeof(ARTICLE *) * (size_t)(i - 1));
987                            p_section->ontop_article_count--;
988                            i--;
989                    }
990                    else
991                    {
992                            memmove((void *)(p_section->p_ontop_articles + i + 1),
993                                            (void *)(p_section->p_ontop_articles + i),
994                                            sizeof(ARTICLE *) * (size_t)(p_section->ontop_article_count - i));
995                    }
996    
997                    p_section->p_ontop_articles[i] = p_article;
998                    p_section->ontop_article_count++;
999    
1000                    // TODO: debug
1001            }
1002            else // ontop == 0
1003            {
1004                    for (i = 0; i < p_section->ontop_article_count; i++)
1005                    {
1006                            if (p_section->p_ontop_articles[i]->aid == p_article->aid)
1007                            {
1008                                    break;
1009                            }
1010                    }
1011                    if (i == p_section->ontop_article_count) // not found
1012                    {
1013                            log_error("Inconsistent state found: article %d not ontop in section %d\n", p_article->aid, p_section->sid);
1014                            return 0;
1015                    }
1016    
1017                    memmove((void *)(p_section->p_ontop_articles + i),
1018                                    (void *)(p_section->p_ontop_articles + i + 1),
1019                                    sizeof(ARTICLE *) * (size_t)(p_section->ontop_article_count - i - 1));
1020                    p_section->ontop_article_count--;
1021            }
1022    
1023            return 0;
1024    }
1025    
1026    int section_list_page_count_with_ontop(SECTION_LIST *p_section)
1027    {
1028            int page_count;
1029    
1030            if (p_section == NULL)
1031            {
1032                    log_error("NULL pointer error\n");
1033                    return -1;
1034            }
1035    
1036            page_count = p_section->page_count - (p_section->last_page_visible_article_count > 0 ? 1 : 0) +
1037                                     (p_section->last_page_visible_article_count + p_section->ontop_article_count) / BBS_article_limit_per_page +
1038                                     ((p_section->last_page_visible_article_count + p_section->ontop_article_count) % BBS_article_limit_per_page == 0 ? 0 : 1);
1039    
1040            return page_count;
1041    }
1042    
1043    int section_list_page_article_count_with_ontop(SECTION_LIST *p_section, int32_t page_id)
1044    {
1045            if (p_section == NULL)
1046            {
1047                    log_error("NULL pointer error\n");
1048                    return -1;
1049            }
1050    
1051            if (page_id < p_section->page_count - 1)
1052            {
1053                    return BBS_article_limit_per_page;
1054            }
1055            else // if (page_id >= p_section->page_count - 1)
1056            {
1057                    return MAX(0, (p_section->last_page_visible_article_count + p_section->ontop_article_count -
1058                                               BBS_article_limit_per_page * (page_id - p_section->page_count + 1)));
1059            }
1060    }
1061    
1062  ARTICLE *section_list_find_article_with_offset(SECTION_LIST *p_section, int32_t aid, int32_t *p_page, int32_t *p_offset, ARTICLE **pp_next)  ARTICLE *section_list_find_article_with_offset(SECTION_LIST *p_section, int32_t aid, int32_t *p_page, int32_t *p_offset, ARTICLE **pp_next)
1063  {  {
1064          ARTICLE *p_article;          ARTICLE *p_article;
# Line 937  int section_list_calculate_page(SECTION_ Line 1258  int section_list_calculate_page(SECTION_
1258          return 0;          return 0;
1259  }  }
1260    
1261    int32_t article_block_last_aid(void)
1262    {
1263            ARTICLE_BLOCK *p_block = p_article_block_pool->p_block[p_article_block_pool->block_count - 1];
1264            int32_t last_aid = p_block->articles[p_block->article_count - 1].aid;
1265    
1266            return last_aid;
1267    }
1268    
1269    int article_block_article_count(void)
1270    {
1271            int ret;
1272    
1273            if (p_article_block_pool == NULL || p_article_block_pool->block_count <= 0)
1274            {
1275                    return -1;
1276            }
1277    
1278            ret = (p_article_block_pool->block_count - 1) * ARTICLE_PER_BLOCK +
1279                      p_article_block_pool->p_block[p_article_block_pool->block_count - 1]->article_count;
1280    
1281            return ret;
1282    }
1283    
1284  int article_count_of_topic(int32_t aid)  int article_count_of_topic(int32_t aid)
1285  {  {
1286          ARTICLE *p_article;          ARTICLE *p_article;
# Line 977  int section_list_move_topic(SECTION_LIST Line 1321  int section_list_move_topic(SECTION_LIST
1321          int32_t first_inserted_aid_dest;          int32_t first_inserted_aid_dest;
1322          int move_counter;          int move_counter;
1323    
1324          if (p_section_dest == NULL)          if (p_section_src == NULL || p_section_dest == NULL)
1325          {          {
1326                  log_error("section_list_move_topic() NULL pointer error\n");                  log_error("section_list_move_topic() NULL pointer error\n");
1327                  return -1;                  return -1;
1328          }          }
1329    
1330            if (p_section_src->sid == p_section_dest->sid)
1331            {
1332                    log_error("section_list_move_topic() src and dest section are the same\n");
1333                    return -1;
1334            }
1335    
1336          if ((p_article = article_block_find_by_aid(aid)) == NULL)          if ((p_article = article_block_find_by_aid(aid)) == NULL)
1337          {          {
1338                  log_error("section_list_move_topic() error: article %d not found in block\n", aid);                  log_error("article_block_find_by_aid(aid = %d) error: article not found\n", aid);
1339                  return -2;                  return -2;
1340          }          }
1341    
# Line 1007  int section_list_move_topic(SECTION_LIST Line 1357  int section_list_move_topic(SECTION_LIST
1357          move_article_count = article_count_of_topic(aid);          move_article_count = article_count_of_topic(aid);
1358          if (move_article_count <= 0)          if (move_article_count <= 0)
1359          {          {
1360                  log_error("section_list_count_of_topic_articles(aid = %d) <= 0\n", aid);                  log_error("article_count_of_topic(aid = %d) <= 0\n", aid);
1361                  return -2;                  return -2;
1362          }          }
1363    
# Line 1026  int section_list_move_topic(SECTION_LIST Line 1376  int section_list_move_topic(SECTION_LIST
1376          {          {
1377                  if (p_section_src->sid != p_article->sid)                  if (p_section_src->sid != p_article->sid)
1378                  {                  {
1379                          log_error("section_list_move_topic() error: src section sid %d != article %d sid %d\n",                          log_error("section_list_move_topic() warning: src section sid %d != article %d sid %d\n",
1380                                            p_section_src->sid, p_article->aid, p_article->sid);                                            p_section_src->sid, p_article->aid, p_article->sid);
1381                          return -2;                          p_article = p_article->p_topic_next;
1382                            continue;
1383                  }                  }
1384    
1385                  // Remove from bi-directional article list of src section                  // Remove from bi-directional article list of src section
# Line 1054  int section_list_move_topic(SECTION_LIST Line 1405  int section_list_move_topic(SECTION_LIST
1405    
1406                  if (section_list_find_article_with_offset(p_section_dest, p_article->aid, &page, &offset, &p_next) != NULL)                  if (section_list_find_article_with_offset(p_section_dest, p_article->aid, &page, &offset, &p_next) != NULL)
1407                  {                  {
1408                          log_error("section_list_move_topic() error: article %d already in section %d\n", p_article->aid, p_section_dest->sid);                          log_error("section_list_move_topic() warning: article %d already in section %d\n", p_article->aid, p_section_dest->sid);
1409                          return -4;                          p_article = p_article->p_topic_next;
1410                            continue;
1411                  }                  }
1412    
1413                  // Insert into bi-directional article list of dest section                  // Insert into bi-directional article list of dest section
# Line 1133  int section_list_move_topic(SECTION_LIST Line 1485  int section_list_move_topic(SECTION_LIST
1485    
1486          if (p_section_dest->article_count - dest_article_count_old != move_article_count)          if (p_section_dest->article_count - dest_article_count_old != move_article_count)
1487          {          {
1488                  log_error("section_list_move_topic() error: count of moved articles %d != %d\n",                  log_error("section_list_move_topic() warning: count of moved articles %d != %d\n",
1489                                    p_section_dest->article_count - dest_article_count_old, move_article_count);                                    p_section_dest->article_count - dest_article_count_old, move_article_count);
1490          }          }
1491    
# Line 1156  int section_list_move_topic(SECTION_LIST Line 1508  int section_list_move_topic(SECTION_LIST
1508    
1509          return move_article_count;          return move_article_count;
1510  }  }
1511    
1512    int get_section_index(SECTION_LIST *p_section)
1513    {
1514            int index;
1515    
1516            if (p_section_list_pool == NULL)
1517            {
1518                    log_error("get_section_index() error: uninitialized\n");
1519                    return -1;
1520            }
1521    
1522            if (p_section == NULL)
1523            {
1524                    index = BBS_max_section;
1525            }
1526            else
1527            {
1528                    index = (int)(p_section - p_section_list_pool->sections);
1529                    if (index < 0 || index >= BBS_max_section)
1530                    {
1531                            log_error("get_section_index(%d) error: index out of range\n", index);
1532                            return -2;
1533                    }
1534            }
1535    
1536            return index;
1537    }
1538    
1539    int section_list_try_rd_lock(SECTION_LIST *p_section, int wait_sec)
1540    {
1541            int index;
1542            struct sembuf sops[4];
1543            struct timespec timeout;
1544            int ret;
1545    
1546            index = get_section_index(p_section);
1547            if (index < 0)
1548            {
1549                    return -2;
1550            }
1551    
1552            sops[0].sem_num = (unsigned short)(index * 2 + 1); // w_sem of section index
1553            sops[0].sem_op = 0;                                                                // wait until unlocked
1554            sops[0].sem_flg = 0;
1555    
1556            sops[1].sem_num = (unsigned short)(index * 2); // r_sem of section index
1557            sops[1].sem_op = 1;                                                        // lock
1558            sops[1].sem_flg = SEM_UNDO;                                        // undo on terminate
1559    
1560            // Read lock on any specific section will also acquire single read lock on "all section"
1561            // so that write lock on all section only need to acquire single write on on "all section"
1562            // rather than to acquire multiple write locks on all the available sections.
1563            if (index != BBS_max_section)
1564            {
1565                    sops[2].sem_num = BBS_max_section * 2 + 1; // w_sem of all section
1566                    sops[2].sem_op = 0;                                                // wait until unlocked
1567                    sops[2].sem_flg = 0;
1568    
1569                    sops[3].sem_num = BBS_max_section * 2; // r_sem of all section
1570                    sops[3].sem_op = 1;                                        // lock
1571                    sops[3].sem_flg = SEM_UNDO;                        // undo on terminate
1572            }
1573    
1574            timeout.tv_sec = wait_sec;
1575            timeout.tv_nsec = 0;
1576    
1577            ret = semtimedop(p_section_list_pool->semid, sops, (index == BBS_max_section ? 2 : 4), &timeout);
1578            if (ret == -1 && errno != EAGAIN && errno != EINTR)
1579            {
1580                    log_error("semtimedop(index = %d, lock read) error %d\n", index, errno);
1581            }
1582    
1583            return ret;
1584    }
1585    
1586    int section_list_try_rw_lock(SECTION_LIST *p_section, int wait_sec)
1587    {
1588            int index;
1589            struct sembuf sops[3];
1590            struct timespec timeout;
1591            int ret;
1592    
1593            index = get_section_index(p_section);
1594            if (index < 0)
1595            {
1596                    return -2;
1597            }
1598    
1599            sops[0].sem_num = (unsigned short)(index * 2 + 1); // w_sem of section index
1600            sops[0].sem_op = 0;                                                                // wait until unlocked
1601            sops[0].sem_flg = 0;
1602    
1603            sops[1].sem_num = (unsigned short)(index * 2 + 1); // w_sem of section index
1604            sops[1].sem_op = 1;                                                                // lock
1605            sops[1].sem_flg = SEM_UNDO;                                                // undo on terminate
1606    
1607            sops[2].sem_num = (unsigned short)(index * 2); // r_sem of section index
1608            sops[2].sem_op = 0;                                                        // wait until unlocked
1609            sops[2].sem_flg = 0;
1610    
1611            timeout.tv_sec = wait_sec;
1612            timeout.tv_nsec = 0;
1613    
1614            ret = semtimedop(p_section_list_pool->semid, sops, 3, &timeout);
1615            if (ret == -1 && errno != EAGAIN && errno != EINTR)
1616            {
1617                    log_error("semtimedop(index = %d, lock write) error %d\n", index, errno);
1618            }
1619    
1620            return ret;
1621    }
1622    
1623    int section_list_rd_unlock(SECTION_LIST *p_section)
1624    {
1625            int index;
1626            struct sembuf sops[2];
1627            int ret;
1628    
1629            index = get_section_index(p_section);
1630            if (index < 0)
1631            {
1632                    return -2;
1633            }
1634    
1635            sops[0].sem_num = (unsigned short)(index * 2); // r_sem of section index
1636            sops[0].sem_op = -1;                                               // unlock
1637            sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO;           // no wait
1638    
1639            if (index != BBS_max_section)
1640            {
1641                    sops[1].sem_num = BBS_max_section * 2;   // r_sem of all section
1642                    sops[1].sem_op = -1;                                     // unlock
1643                    sops[1].sem_flg = IPC_NOWAIT | SEM_UNDO; // no wait
1644            }
1645    
1646            ret = semop(p_section_list_pool->semid, sops, (index == BBS_max_section ? 1 : 2));
1647            if (ret == -1 && errno != EAGAIN && errno != EINTR)
1648            {
1649                    log_error("semop(index = %d, unlock read) error %d\n", index, errno);
1650            }
1651    
1652            return ret;
1653    }
1654    
1655    int section_list_rw_unlock(SECTION_LIST *p_section)
1656    {
1657            int index;
1658            struct sembuf sops[1];
1659            int ret;
1660    
1661            index = get_section_index(p_section);
1662            if (index < 0)
1663            {
1664                    return -2;
1665            }
1666    
1667            sops[0].sem_num = (unsigned short)(index * 2 + 1); // w_sem of section index
1668            sops[0].sem_op = -1;                                                       // unlock
1669            sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO;                   // no wait
1670    
1671            ret = semop(p_section_list_pool->semid, sops, 1);
1672            if (ret == -1 && errno != EAGAIN && errno != EINTR)
1673            {
1674                    log_error("semop(index = %d, unlock write) error %d\n", index, errno);
1675            }
1676    
1677            return ret;
1678    }
1679    
1680    int section_list_rd_lock(SECTION_LIST *p_section)
1681    {
1682            int timer = 0;
1683            int sid = (p_section == NULL ? 0 : p_section->sid);
1684            int ret = -1;
1685    
1686            while (!SYS_server_exit)
1687            {
1688                    ret = section_list_try_rd_lock(p_section, SECTION_TRY_LOCK_WAIT_TIME);
1689                    if (ret == 0) // success
1690                    {
1691                            break;
1692                    }
1693                    else if (errno == EAGAIN || errno == EINTR) // retry
1694                    {
1695                            timer++;
1696                            if (timer % SECTION_TRY_LOCK_TIMES == 0)
1697                            {
1698                                    log_error("section_list_try_rd_lock() tried %d times on section %d\n", sid, timer);
1699                            }
1700                    }
1701                    else // failed
1702                    {
1703                            log_error("section_list_try_rd_lock() failed on section %d\n", sid);
1704                            break;
1705                    }
1706            }
1707    
1708            return ret;
1709    }
1710    
1711    int section_list_rw_lock(SECTION_LIST *p_section)
1712    {
1713            int timer = 0;
1714            int sid = (p_section == NULL ? 0 : p_section->sid);
1715            int ret = -1;
1716    
1717            while (!SYS_server_exit)
1718            {
1719                    ret = section_list_try_rw_lock(p_section, SECTION_TRY_LOCK_WAIT_TIME);
1720                    if (ret == 0) // success
1721                    {
1722                            break;
1723                    }
1724                    else if (errno == EAGAIN || errno == EINTR) // retry
1725                    {
1726                            timer++;
1727                            if (timer % SECTION_TRY_LOCK_TIMES == 0)
1728                            {
1729                                    log_error("section_list_try_rw_lock() tried %d times on section %d\n", sid, timer);
1730                            }
1731                    }
1732                    else // failed
1733                    {
1734                            log_error("section_list_try_rw_lock() failed on section %d\n", sid);
1735                            break;
1736                    }
1737            }
1738    
1739            return ret;
1740    }

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