/[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.5 by sysadm, Wed May 21 09:18:17 2025 UTC Revision 1.8 by sysadm, Thu May 22 11:10:19 2025 UTC
# Line 23  Line 23 
23  #include <unistd.h>  #include <unistd.h>
24  #include <stdlib.h>  #include <stdlib.h>
25  #include <errno.h>  #include <errno.h>
26    #include <sys/param.h>
27  #include <sys/shm.h>  #include <sys/shm.h>
28  #include <sys/ipc.h>  #include <sys/ipc.h>
29    
30  // ARTICLE_BLOCK_PER_SHM * ARTICLE_BLOCK_SHM_COUNT_LIMIT should be  #define ARTICLE_BLOCK_PER_SHM 400                 // sizeof(ARTICLE_BLOCK) * ARTICLE_BLOCK_PER_SHM is the size of each shm segment to allocate
 // no less than BBS_article_block_limit_per_section * BBS_max_section,  
 // in order to allocate enough memory for blocks  
 #define ARTICLE_BLOCK_PER_SHM 400 // sizeof(ARTICLE_BLOCK) * ARTICLE_BLOCK_PER_SHM is the size of each shm segment to allocate  
31  #define ARTICLE_BLOCK_SHM_COUNT_LIMIT 256 // limited by length (8-bit) of proj_id in ftok(path, proj_id)  #define ARTICLE_BLOCK_SHM_COUNT_LIMIT 256 // limited by length (8-bit) of proj_id in ftok(path, proj_id)
32    #define ARTICLE_BLOCK_PER_POOL (ARTICLE_BLOCK_PER_SHM * ARTICLE_BLOCK_SHM_COUNT_LIMIT)
33    
34    struct article_block_t
35    {
36            ARTICLE articles[ARTICLE_PER_BLOCK];
37            int32_t article_count;
38            struct article_block_t *p_next_block;
39    };
40    typedef struct article_block_t ARTICLE_BLOCK;
41    
42  struct article_block_shm_t  struct article_block_shm_t
43  {  {
# Line 39  struct article_block_shm_t Line 46  struct article_block_shm_t
46  };  };
47  typedef struct article_block_shm_t ARTICLE_BLOCK_SHM;  typedef struct article_block_shm_t ARTICLE_BLOCK_SHM;
48    
49  static ARTICLE_BLOCK_SHM *p_article_block_shm_pool;  struct article_block_pool_t
50  static int article_block_shm_count;  {
51  static ARTICLE_BLOCK *p_article_block_free_list;          ARTICLE_BLOCK_SHM shm_pool[ARTICLE_BLOCK_SHM_COUNT_LIMIT];
52            int shm_count;
53  static SECTION_DATA *p_section_data_pool;          ARTICLE_BLOCK *p_block_free_list;
54  static int section_data_count;          ARTICLE_BLOCK *p_block[ARTICLE_BLOCK_PER_POOL];
55  static TRIE_NODE *p_trie_dict_section_data;          int32_t block_count;
56    };
57    typedef struct article_block_pool_t ARTICLE_BLOCK_POOL;
58    
59    static ARTICLE_BLOCK_POOL *p_article_block_pool = NULL;
60    
61  int section_data_pool_init(const char *filename, int article_block_count)  static SECTION_LIST *p_section_list_pool = NULL;
62    static int section_list_count = 0;
63    static TRIE_NODE *p_trie_dict_section_list = NULL;
64    
65    int article_block_init(const char *filename, int block_count)
66  {  {
67          int shmid;          int shmid;
68          int proj_id;          int proj_id;
# Line 55  int section_data_pool_init(const char *f Line 70  int section_data_pool_init(const char *f
70          size_t size;          size_t size;
71          void *p_shm;          void *p_shm;
72          int i;          int i;
73          int article_block_count_in_shm;          int block_count_in_shm;
74          ARTICLE_BLOCK *p_article_block_in_shm;          ARTICLE_BLOCK *p_block_in_shm;
75          ARTICLE_BLOCK **pp_article_block_next;          ARTICLE_BLOCK **pp_block_next;
   
         if (p_article_block_shm_pool != NULL ||  
                 p_article_block_free_list != NULL ||  
                 p_section_data_pool != NULL ||  
                 p_trie_dict_section_data != NULL)  
         {  
                 log_error("section_data_pool already initialized\n");  
                 return -1;  
         }  
   
         if (article_block_count > ARTICLE_BLOCK_PER_SHM * ARTICLE_BLOCK_SHM_COUNT_LIMIT)  
         {  
                 log_error("article_block_count exceed limit %d\n", ARTICLE_BLOCK_PER_SHM * ARTICLE_BLOCK_SHM_COUNT_LIMIT);  
                 return -2;  
         }  
76    
77          p_article_block_shm_pool = calloc((size_t)article_block_count / ARTICLE_BLOCK_PER_SHM + 1, sizeof(ARTICLE_BLOCK_SHM));          if (p_article_block_pool != NULL)
         if (p_article_block_shm_pool == NULL)  
78          {          {
79                  log_error("calloc(%d ARTICLE_BLOCK_SHM) OOM\n", article_block_count / ARTICLE_BLOCK_PER_SHM + 1);                  log_error("article_block_pool already initialized\n");
80                  return -2;                  return -1;
81          }          }
82    
83          p_section_data_pool = calloc(BBS_max_section, sizeof(SECTION_DATA));          if (block_count > ARTICLE_BLOCK_PER_POOL)
         if (p_section_data_pool == NULL)  
84          {          {
85                  log_error("calloc(%d SECTION_DATA) OOM\n", BBS_max_section);                  log_error("article_block_count exceed limit %d\n", ARTICLE_BLOCK_PER_POOL);
86                  return -2;                  return -2;
87          }          }
         section_data_count = 0;  
88    
89          p_trie_dict_section_data = trie_dict_create();          p_article_block_pool = calloc(1, sizeof(ARTICLE_BLOCK_POOL));
90          if (p_trie_dict_section_data == NULL)          if (p_article_block_pool == NULL)
91          {          {
92                  log_error("trie_dict_create() OOM\n", BBS_max_section);                  log_error("calloc(ARTICLE_BLOCK_POOL) OOM\n");
93                  return -2;                  return -2;
94          }          }
95    
96          // Allocate shared memory          // Allocate shared memory
97          article_block_shm_count = 0;          p_article_block_pool->shm_count = 0;
98          pp_article_block_next = &p_article_block_free_list;          pp_block_next = &(p_article_block_pool->p_block_free_list);
99    
100          while (article_block_count > 0)          while (block_count > 0)
101          {          {
102                  article_block_count_in_shm =                  block_count_in_shm = MIN(block_count, ARTICLE_BLOCK_PER_SHM);
103                          (article_block_count < ARTICLE_BLOCK_PER_SHM ? article_block_count : ARTICLE_BLOCK_PER_SHM);                  block_count -= block_count_in_shm;
                 article_block_count -= article_block_count_in_shm;  
104    
105                  proj_id = getpid() + article_block_shm_count;                  proj_id = getpid() + p_article_block_pool->shm_count;
106                  key = ftok(filename, proj_id);                  key = ftok(filename, proj_id);
107                  if (key == -1)                  if (key == -1)
108                  {                  {
109                          log_error("ftok(%s, %d) error (%d)\n", filename, proj_id, errno);                          log_error("ftok(%s, %d) error (%d)\n", filename, proj_id, errno);
110                          section_data_pool_cleanup();                          article_block_cleanup();
111                          return -3;                          return -3;
112                  }                  }
113    
114                  size = sizeof(shmid) + sizeof(ARTICLE_BLOCK) * (size_t)article_block_count_in_shm;                  size = sizeof(shmid) + sizeof(ARTICLE_BLOCK) * (size_t)block_count_in_shm;
115                  shmid = shmget(key, size, IPC_CREAT | IPC_EXCL | 0600);                  shmid = shmget(key, size, IPC_CREAT | IPC_EXCL | 0600);
116                  if (shmid == -1)                  if (shmid == -1)
117                  {                  {
118                          log_error("shmget(shm_index = %d, size = %d) error (%d)\n", article_block_shm_count, size, errno);                          log_error("shmget(shm_index = %d, size = %d) error (%d)\n", p_article_block_pool->shm_count, size, errno);
119                          section_data_pool_cleanup();                          article_block_cleanup();
120                          return -3;                          return -3;
121                  }                  }
122                  p_shm = shmat(shmid, NULL, 0);                  p_shm = shmat(shmid, NULL, 0);
123                  if (p_shm == (void *)-1)                  if (p_shm == (void *)-1)
124                  {                  {
125                          log_error("shmat(shmid = %d) error (%d)\n", shmid, errno);                          log_error("shmat(shmid = %d) error (%d)\n", shmid, errno);
126                          section_data_pool_cleanup();                          article_block_cleanup();
127                          return -3;                          return -3;
128                  }                  }
129    
130                  (p_article_block_shm_pool + article_block_shm_count)->shmid = shmid;                  (p_article_block_pool->shm_pool + p_article_block_pool->shm_count)->shmid = shmid;
131                  (p_article_block_shm_pool + article_block_shm_count)->p_shm = p_shm;                  (p_article_block_pool->shm_pool + p_article_block_pool->shm_count)->p_shm = p_shm;
132                  article_block_shm_count++;                  p_article_block_pool->shm_count++;
133    
134                  p_article_block_in_shm = p_shm;                  p_block_in_shm = p_shm;
135                  *pp_article_block_next = p_article_block_in_shm;                  *pp_block_next = p_block_in_shm;
136    
137                  for (i = 0; i < article_block_count_in_shm; i++)                  for (i = 0; i < block_count_in_shm; i++)
138                  {                  {
139                          if (i < article_block_count_in_shm - 1)                          if (i < block_count_in_shm - 1)
140                          {                          {
141                                  (p_article_block_in_shm + i)->p_next_block = (p_article_block_in_shm + i + 1);                                  (p_block_in_shm + i)->p_next_block = (p_block_in_shm + i + 1);
142                          }                          }
143                          else                          else
144                          {                          {
145                                  (p_article_block_in_shm + i)->p_next_block = NULL;                                  (p_block_in_shm + i)->p_next_block = NULL;
146                                  pp_article_block_next = &((p_article_block_in_shm + i)->p_next_block);                                  pp_block_next = &((p_block_in_shm + i)->p_next_block);
147                          }                          }
148                  }                  }
149          }          }
150    
151            p_article_block_pool->block_count = 0;
152    
153          return 0;          return 0;
154  }  }
155    
156  void section_data_pool_cleanup(void)  void article_block_cleanup(void)
157  {  {
158          int i;          if (p_article_block_pool != NULL)
159            {
160                    for (int i = 0; i < p_article_block_pool->shm_count; i++)
161                    {
162                            if (shmdt((p_article_block_pool->shm_pool + i)->p_shm) == -1)
163                            {
164                                    log_error("shmdt(shmid = %d) error (%d)\n", (p_article_block_pool->shm_pool + i)->shmid, errno);
165                            }
166    
167                            if (shmctl((p_article_block_pool->shm_pool + i)->shmid, IPC_RMID, NULL) == -1)
168                            {
169                                    log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", (p_article_block_pool->shm_pool + i)->shmid, errno);
170                            }
171                    }
172    
173                    free(p_article_block_pool);
174                    p_article_block_pool = NULL;
175            }
176    }
177    
178    inline static ARTICLE_BLOCK *pop_free_article_block(void)
179    {
180            ARTICLE_BLOCK *p_block = NULL;
181    
182          if (p_trie_dict_section_data != NULL)          if (p_article_block_pool->p_block_free_list != NULL)
183          {          {
184                  trie_dict_destroy(p_trie_dict_section_data);                  p_block = p_article_block_pool->p_block_free_list;
185                  p_trie_dict_section_data = NULL;                  p_article_block_pool->p_block_free_list = p_block->p_next_block;
186                  section_data_count = 0;                  p_block->p_next_block = NULL;
187                    p_block->article_count = 0;
188          }          }
189    
190          if (p_section_data_pool != NULL)          return p_block;
191    }
192    
193    inline static void push_free_article_block(ARTICLE_BLOCK *p_block)
194    {
195            p_block->p_next_block = p_article_block_pool->p_block_free_list;
196            p_article_block_pool->p_block_free_list = p_block;
197    }
198    
199    int article_block_reset(void)
200    {
201            ARTICLE_BLOCK *p_block;
202    
203            if (p_article_block_pool == NULL)
204          {          {
205                  free(p_section_data_pool);                  log_error("article_block_pool not initialized\n");
206                  p_section_data_pool = NULL;                  return -1;
207          }          }
208    
209          if (p_article_block_free_list != NULL)          while (p_article_block_pool->block_count > 0)
210          {          {
211                  p_article_block_free_list = NULL;                  p_article_block_pool->block_count--;
212                    p_block = p_article_block_pool->p_block[p_article_block_pool->block_count];
213                    push_free_article_block(p_block);
214          }          }
215    
216          if (p_article_block_shm_pool != NULL)          return 0;
217    }
218    
219    ARTICLE *article_block_find_by_aid(int32_t aid)
220    {
221            ARTICLE_BLOCK *p_block;
222            int left;
223            int right;
224            int mid;
225    
226            if (p_article_block_pool == NULL)
227          {          {
228                  for (i = 0; i < article_block_shm_count; i++)                  log_error("article_block_pool not initialized\n");
229                    return NULL;
230            }
231    
232            if (p_article_block_pool->block_count == 0) // empty
233            {
234                    return NULL;
235            }
236    
237            left = 0;
238            right = p_article_block_pool->block_count;
239    
240            // aid in the range [ head aid of blocks[left], tail aid of blocks[right - 1] ]
241            while (left < right - 1)
242            {
243                    // get block offset no less than mid value of left and right block offsets
244                    mid = (left + right) / 2 + (right - left) % 2;
245    
246                    if (mid >= p_article_block_pool->block_count)
247                  {                  {
248                          if (shmdt((p_article_block_shm_pool + i)->p_shm) == -1)                          log_error("block(mid = %d) is out of boundary\n", mid);
249                          {                          return NULL;
250                                  log_error("shmdt(shmid = %d) error (%d)\n", (p_article_block_shm_pool + i)->shmid, errno);                  }
                         }  
251    
252                          if (shmctl((p_article_block_shm_pool + i)->shmid, IPC_RMID, NULL) == -1)                  if (aid < p_article_block_pool->p_block[mid]->articles[0].aid)
253                          {                  {
254                                  log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", (p_article_block_shm_pool + i)->shmid, errno);                          right = mid;
                         }  
255                  }                  }
256                    else
257                    {
258                            left = mid;
259                    }
260            }
261    
262            p_block = p_article_block_pool->p_block[left];
263    
264                  p_article_block_shm_pool = NULL;          left = 0;
265                  article_block_shm_count = 0;          right = p_block->article_count - 1;
266    
267            // aid in the range [ aid of articles[left], aid of articles[right] ]
268            while (left < right)
269            {
270                    mid = (left + right) / 2;
271    
272                    if (aid <= p_block->articles[mid].aid)
273                    {
274                            right = mid;
275                    }
276                    else
277                    {
278                            left = mid + 1;
279                    }
280          }          }
281    
282            return (p_block->articles + left);
283  }  }
284    
285  inline static ARTICLE_BLOCK *pop_free_article_block(void)  ARTICLE *article_block_find_by_index(int index)
286  {  {
287          ARTICLE_BLOCK *p_article_block = NULL;          ARTICLE_BLOCK *p_block;
288    
289          if (p_article_block_free_list != NULL)          if (p_article_block_pool == NULL)
290          {          {
291                  p_article_block = p_article_block_free_list;                  log_error("article_block_pool not initialized\n");
292                  p_article_block_free_list = p_article_block_free_list->p_next_block;                  return NULL;
293          }          }
294    
295          return p_article_block;          if (index < 0 || index / ARTICLE_PER_BLOCK >= p_article_block_pool->block_count)
296  }          {
297                    log_error("section_data_find_article_by_index(%d) is out of boundary of block [0, %d)\n", index, p_article_block_pool->block_count);
298                    return NULL;
299            }
300    
301  inline static void push_free_article_block(ARTICLE_BLOCK *p_article_block)          p_block = p_article_block_pool->p_block[index / ARTICLE_PER_BLOCK];
302  {  
303          p_article_block->p_next_block = p_article_block_free_list;          if (index % ARTICLE_PER_BLOCK >= p_block->article_count)
304          p_article_block_free_list = p_article_block;          {
305                    log_error("section_data_find_article_by_index(%d) is out of boundary of article [0, %d)\n", index, p_block->article_count);
306                    return NULL;
307            }
308    
309            return (p_block->articles + (index % ARTICLE_PER_BLOCK));
310  }  }
311    
312  SECTION_DATA *section_data_create(const char *sname, const char *stitle, const char *master_name)  SECTION_LIST *section_list_create(const char *sname, const char *stitle, const char *master_name)
313  {  {
314          SECTION_DATA *p_section;          SECTION_LIST *p_section;
         int index;  
315    
316          if (p_section_data_pool == NULL || p_trie_dict_section_data == NULL)          if (p_section_list_pool == NULL)
317          {          {
318                  log_error("section_data not initialized\n");                  p_section_list_pool = calloc(BBS_max_section, sizeof(SECTION_LIST));
319                  return NULL;                  if (p_section_list_pool == NULL)
320                    {
321                            log_error("calloc(%d SECTION_LIST) OOM\n", BBS_max_section);
322                            return NULL;
323                    }
324    
325                    section_list_count = 0;
326            }
327    
328            if (p_trie_dict_section_list == NULL)
329            {
330                    p_trie_dict_section_list = trie_dict_create();
331                    if (p_trie_dict_section_list == NULL)
332                    {
333                            log_error("trie_dict_create() OOM\n", BBS_max_section);
334                            return NULL;
335                    }
336          }          }
337    
338          if (section_data_count >= BBS_max_section)          if (section_list_count >= BBS_max_section)
339          {          {
340                  log_error("section_data_count exceed limit %d\n", BBS_max_section);                  log_error("section_list_count exceed limit %d\n", BBS_max_section);
341                  return NULL;                  return NULL;
342          }          }
343    
344          index = section_data_count;          p_section = p_section_list_pool + section_list_count;
         p_section = p_section_data_pool + index;  
345    
346          strncpy(p_section->sname, sname, sizeof(p_section->sname - 1));          strncpy(p_section->sname, sname, sizeof(p_section->sname - 1));
347          p_section->sname[sizeof(p_section->sname - 1)] = '\0';          p_section->sname[sizeof(p_section->sname - 1)] = '\0';
# Line 245  SECTION_DATA *section_data_create(const Line 352  SECTION_DATA *section_data_create(const
352          strncpy(p_section->master_name, master_name, sizeof(p_section->master_name - 1));          strncpy(p_section->master_name, master_name, sizeof(p_section->master_name - 1));
353          p_section->master_name[sizeof(p_section->master_name - 1)] = '\0';          p_section->master_name[sizeof(p_section->master_name - 1)] = '\0';
354    
355          p_section->p_head_block = NULL;          if (trie_dict_set(p_trie_dict_section_list, sname, section_list_count) != 1)
         p_section->p_tail_block = NULL;  
         p_section->block_count = 0;  
         p_section->article_count = 0;  
         p_section->delete_count = 0;  
   
         if (trie_dict_set(p_trie_dict_section_data, sname, index) != 1)  
356          {          {
357                  log_error("trie_dict_set(section_data, %s, %d) error\n", sname, index);                  log_error("trie_dict_set(section_data, %s, %d) error\n", sname, section_list_count);
358                  return NULL;                  return NULL;
359          }          }
360    
361          section_data_count++;          section_list_reset_articles(p_section);
362    
363            section_list_count++;
364    
365          return p_section;          return p_section;
366  }  }
367    
368  int section_data_free_block(SECTION_DATA *p_section)  void section_list_reset_articles(SECTION_LIST *p_section)
369  {  {
370          ARTICLE_BLOCK *p_block;          p_section->article_count = 0;
371            p_section->topic_count = 0;
372            p_section->visible_article_count = 0;
373            p_section->visible_topic_count = 0;
374            p_section->p_article_head = NULL;
375            p_section->p_article_tail = NULL;
376    
377          if (p_section == NULL)          p_section->page_count = 0;
378          {          p_section->last_page_visible_article_count = 0;
379                  log_error("section_data_free_block() NULL pointer error\n");  }
                 return -1;  
         }  
380    
381          if (p_section_data_pool == NULL)  void section_list_cleanup(void)
382    {
383            if (p_trie_dict_section_list != NULL)
384          {          {
385                  log_error("section_data not initialized\n");                  trie_dict_destroy(p_trie_dict_section_list);
386                  return -1;                  p_trie_dict_section_list = NULL;
387          }          }
388    
389          while (p_section->p_head_block != NULL)          if (p_section_list_pool != NULL)
390          {          {
391                  p_block = p_section->p_head_block;                  free(p_section_list_pool);
392                  p_section->p_head_block = p_block->p_next_block;                  p_section_list_pool = NULL;
                 push_free_article_block(p_block);  
393          }          }
394    
395          p_section->p_tail_block = NULL;          section_list_count = 0;
         p_section->block_count = 0;  
         p_section->article_count = 0;  
         p_section->delete_count = 0;  
   
         return 0;  
396  }  }
397    
398  SECTION_DATA *section_data_find_section_by_name(const char *sname)  SECTION_LIST *section_list_find_by_name(const char *sname)
399  {  {
400          int64_t index;          int64_t index;
401    
402          if (p_section_data_pool == NULL || p_trie_dict_section_data == NULL)          if (p_section_list_pool == NULL || p_trie_dict_section_list == NULL)
403          {          {
404                  log_error("section_data not initialized\n");                  log_error("section_list not initialized\n");
405                  return NULL;                  return NULL;
406          }          }
407    
408          if (trie_dict_get(p_trie_dict_section_data, sname, &index) != 1)          if (trie_dict_get(p_trie_dict_section_list, sname, &index) != 1)
409          {          {
410                  log_error("trie_dict_get(section_data, %s) error\n", sname);                  log_error("trie_dict_get(section_data, %s) error\n", sname);
411                  return NULL;                  return NULL;
412          }          }
413    
414          return (p_section_data_pool + index);          return (p_section_list_pool + index);
415  }  }
416    
417  int section_data_append_article(SECTION_DATA *p_section, const ARTICLE *p_article)  int section_list_append_article(SECTION_LIST *p_section, const ARTICLE *p_article_src)
418  {  {
419          ARTICLE_BLOCK *p_block;          ARTICLE_BLOCK *p_block;
420          int32_t last_aid = 0;          int32_t last_aid = 0;
421            ARTICLE *p_article;
422          ARTICLE *p_topic_head;          ARTICLE *p_topic_head;
423          ARTICLE *p_topic_tail;          ARTICLE *p_topic_tail;
424    
425          if (p_section == NULL || p_article == NULL)          if (p_section == NULL || p_article_src == NULL)
426          {          {
427                  log_error("section_data_append_article() NULL pointer error\n");                  log_error("section_list_append_article() NULL pointer error\n");
428                  return -1;                  return -1;
429          }          }
430    
431          if (p_section_data_pool == NULL)          if (p_article_block_pool == NULL)
432          {          {
433                  log_error("section_data not initialized\n");                  log_error("article_block_pool not initialized\n");
434                  return -1;                  return -1;
435          }          }
436    
437          if (p_section->p_tail_block == NULL || p_section->p_tail_block->article_count >= BBS_article_limit_per_block)          if (p_article_block_pool->block_count == 0 ||
438                    p_article_block_pool->p_block[p_article_block_pool->block_count - 1]->article_count >= ARTICLE_PER_BLOCK)
439          {          {
                 if (p_section->block_count >= BBS_article_block_limit_per_section)  
                 {  
                         log_error("section block count %d reach limit\n", p_section->block_count);  
                         return -2;  
                 }  
   
440                  if ((p_block = pop_free_article_block()) == NULL)                  if ((p_block = pop_free_article_block()) == NULL)
441                  {                  {
442                          log_error("pop_free_article_block() error\n");                          log_error("pop_free_article_block() error\n");
443                          return -2;                          return -2;
444                  }                  }
445    
446                  p_block->article_count = 0;                  if (p_article_block_pool->block_count > 0)
                 p_block->p_next_block = NULL;  
   
                 if (p_section->p_tail_block == NULL)  
447                  {                  {
448                          p_section->p_head_block = p_block;                          last_aid = p_article_block_pool->p_block[p_article_block_pool->block_count - 1]->articles[ARTICLE_PER_BLOCK - 1].aid;
                         last_aid = 0;  
449                  }                  }
450                  else  
451                  {                  p_article_block_pool->p_block[p_article_block_pool->block_count] = p_block;
452                          p_section->p_tail_block->p_next_block = p_block;                  p_article_block_pool->block_count++;
                         last_aid = p_section->p_tail_block->articles[BBS_article_limit_per_block - 1].aid;  
                 }  
                 p_section->p_tail_block = p_block;  
                 p_section->p_block[p_section->block_count] = p_block;  
                 p_section->block_count++;  
453          }          }
454          else          else
455          {          {
456                  p_block = p_section->p_tail_block;                  p_block = p_article_block_pool->p_block[p_article_block_pool->block_count - 1];
457                  last_aid = p_block->articles[p_block->article_count - 1].aid;                  last_aid = p_block->articles[p_block->article_count - 1].aid;
458          }          }
459    
460          // AID of articles should be strictly ascending          // AID of articles should be strictly ascending
461          if (p_article->aid <= last_aid)          if (p_article_src->aid <= last_aid)
462          {          {
463                  log_error("section_data_append_article(aid=%d) error: last_aid=%d\n", p_article->aid, last_aid);                  log_error("section_data_append_article(aid=%d) error: last_aid=%d\n", p_article_src->aid, last_aid);
464                  return -3;                  return -3;
465          }          }
466    
467          if (p_block->article_count == 0)          p_article = (p_block->articles + p_block->article_count);
468            p_block->article_count++;
469            p_section->article_count++;
470    
471            // Copy article data
472            *p_article = *p_article_src;
473    
474            if (p_article->visible)
475          {          {
476                  p_section->block_head_aid[p_section->block_count - 1] = p_article->aid;                  p_section->visible_article_count++;
477          }          }
478    
479            // Link appended article as tail node of topic bi-directional list
480          if (p_article->tid != 0)          if (p_article->tid != 0)
481          {          {
482                  p_topic_head = section_data_find_article_by_aid(p_section, p_article->tid);                  p_topic_head = article_block_find_by_aid(p_article->tid);
483                  if (p_topic_head == NULL)                  if (p_topic_head == NULL)
484                  {                  {
485                          log_error("search head of topic (aid=%d) error\n", p_article->tid);                          log_error("search head of topic (aid=%d) error\n", p_article->tid);
486                          return -4;                          return -4;
487                  }                  }
488    
489                  p_topic_tail = section_data_find_article_by_aid(p_section, p_topic_head->prior_aid);                  p_topic_tail = p_topic_head->p_topic_prior;
490                  if (p_topic_tail == NULL)                  if (p_topic_tail == NULL)
491                  {                  {
492                          log_error("search tail of topic (aid=%d) error\n", p_topic_head->prior_aid);                          log_error("tail of topic (aid=%d) is NULL\n", p_article->tid);
493                          return -4;                          return -4;
494                  }                  }
495          }          }
496          else          else
497          {          {
498                  p_topic_head = &(p_block->articles[p_block->article_count]);                  p_section->topic_count++;
499                  p_topic_tail = p_topic_head;  
500                    if (p_article->visible)
501                    {
502                            p_section->visible_topic_count++;
503                    }
504    
505                    p_topic_head = p_article;
506                    p_topic_tail = p_article;
507          }          }
508    
509          // Copy article data          p_article->p_topic_prior = p_topic_tail;
510          p_block->articles[p_block->article_count] = *p_article;          p_article->p_topic_next = p_topic_head;
511            p_topic_head->p_topic_prior = p_article;
512            p_topic_tail->p_topic_next = p_article;
513    
514          // Link appended article as tail node of topic bi-directional list;          // Link appended article as tail node of article bi-directional list
515          p_block->articles[p_block->article_count].prior_aid = p_topic_tail->aid;          if (p_section->p_article_head == NULL)
516          p_block->articles[p_block->article_count].next_aid = p_topic_head->aid;          {
517          p_topic_head->prior_aid = p_article->aid;                  p_section->p_article_head = p_article;
518          p_topic_tail->next_aid = p_article->aid;                  p_section->p_article_tail = p_article;
519            }
520            p_article->p_prior = p_section->p_article_tail;
521            p_article->p_next = p_section->p_article_head;
522            p_section->p_article_head->p_prior = p_article;
523            p_section->p_article_tail->p_next = p_article;
524            p_section->p_article_tail = p_article;
525    
526          p_block->article_count++;          // Update page
527          p_section->article_count++;          if (p_article->visible && p_section->last_page_visible_article_count % BBS_article_limit_per_page == 0)
528            {
529                    p_section->p_page_first_article[p_section->page_count] = p_article;
530                    p_section->page_count++;
531                    p_section->last_page_visible_article_count = 0;
532            }
533            p_section->last_page_visible_article_count++;
534    
535          return 0;          return 0;
536  }  }
537    
538  ARTICLE *section_data_find_article_by_aid(SECTION_DATA *p_section, int32_t aid)  int section_list_set_article_visible(SECTION_LIST *p_section, int32_t aid, int8_t visible)
539    {
540            ARTICLE *p_article;
541            ARTICLE *p_reply;
542            int affected_count = 0;
543    
544            if (p_section == NULL)
545            {
546                    log_error("section_list_set_article_visible() NULL pointer error\n");
547                    return -2;
548            }
549    
550            p_article = article_block_find_by_aid(aid);
551            if (p_article == NULL)
552            {
553                    return -1; // Not found
554            }
555    
556            if (p_article->visible == visible)
557            {
558                    return 0; // Already set
559            }
560    
561            if (visible == 0) // 1 -> 0
562            {
563                    p_section->visible_article_count--;
564    
565                    if (p_article->tid == 0)
566                    {
567                            p_section->visible_topic_count--;
568    
569                            // Set related visible replies to invisible
570                            for (p_reply = p_article->p_topic_next; p_reply->tid != 0; p_reply = p_reply->p_topic_next)
571                            {
572                                    if (p_reply->tid != aid)
573                                    {
574                                            log_error("Inconsistent tid = %d found in reply %d of topic %d\n", p_reply->tid, p_reply->aid, aid);
575                                            continue;
576                                    }
577    
578                                    if (p_reply->visible == 1)
579                                    {
580                                            p_reply->visible = 0;
581                                            p_section->visible_article_count--;
582                                            affected_count++;
583                                    }
584                            }
585                    }
586            }
587            else // 0 -> 1
588            {
589                    p_section->visible_article_count++;
590    
591                    if (p_article->tid == 0)
592                    {
593                            p_section->visible_topic_count++;
594                    }
595            }
596    
597            p_article->visible = visible;
598            affected_count++;
599    
600            return affected_count;
601    }
602    
603    ARTICLE *section_list_find_article_with_offset(SECTION_LIST *p_section, int32_t aid, int32_t *p_page, int32_t *p_offset)
604  {  {
605          ARTICLE *p_article;          ARTICLE *p_article;
         ARTICLE_BLOCK *p_block;  
606          int left;          int left;
607          int right;          int right;
608          int mid;          int mid;
609    
610            *p_page = -1;
611            *p_offset = -1;
612    
613          if (p_section == NULL)          if (p_section == NULL)
614          {          {
615                  log_error("section_data_find_article_by_aid() NULL pointer error\n");                  log_error("section_list_find_article_with_offset() NULL pointer error\n");
616                  return NULL;                  return NULL;
617          }          }
618    
619          if (p_section->block_count == 0) // empty section          if (p_section->article_count == 0) // empty
620          {          {
621                    *p_page = 0;
622                    *p_offset = 0;
623                  return NULL;                  return NULL;
624          }          }
625    
626          left = 0;          left = 0;
627          right = p_section->block_count;          right = p_section->page_count;
628    
629          // aid in the range [ head aid of blocks[left], tail aid of blocks[right - 1] ]          // aid in the range [ head aid of pages[left], tail aid of pages[right - 1] ]
630          while (left < right - 1)          while (left < right - 1)
631          {          {
632                  // get block offset no less than mid value of left and right block offsets                  // get page id no less than mid value of left page id and right page id
633                  mid = (left + right) / 2 + (right - left) % 2;                  mid = (left + right) / 2 + (right - left) % 2;
634                    
635                  if (mid >= BBS_article_block_limit_per_section)                  if (mid >= p_section->page_count)
636                  {                  {
637                          log_error("block_m(%d) is out of boundary\n", mid);                          log_error("page id (mid = %d) is out of boundary\n", mid);
638                          return NULL;                          return NULL;
639                  }                  }
640    
641                  if (aid < p_section->block_head_aid[mid])                  if (aid < p_section->p_page_first_article[mid]->aid)
642                  {                  {
643                          right = mid;                          right = mid;
644                  }                  }
# Line 461  ARTICLE *section_data_find_article_by_ai Line 648  ARTICLE *section_data_find_article_by_ai
648                  }                  }
649          }          }
650    
651          p_block = p_section->p_block[left];          *p_page = left;
652    
653            p_article = p_section->p_page_first_article[*p_page];
654    
655            // p_section->p_page_first_article[*p_page]->aid <= aid < p_section->p_page_first_article[*p_page + 1]->aid
656            right = (*p_page == p_section->page_count - 1 ? INT32_MAX : p_section->p_page_first_article[*p_page + 1]->aid);
657    
658            // left will be the offset of article found or offset to insert
659          left = 0;          left = 0;
         right = p_block->article_count - 1;  
660    
661          // aid in the range [ aid of articles[left], aid of articles[right] ]          while (1)
         while (left < right)  
662          {          {
663                  mid = (left + right) / 2;                  if (aid == p_article->aid) // found
   
                 if (aid <= p_block->articles[mid].aid)  
664                  {                  {
665                          right = mid;                          break;
666                  }                  }
667                  else                  else if (aid < p_article->aid) // not exist
668                  {                  {
669                          left = mid + 1;                          p_article = NULL;
670                            break;
671                  }                  }
         }  
   
         p_article = &(p_block->articles[left]);  
   
         return p_article;  
 }  
672    
673  ARTICLE *section_data_find_article_by_index(SECTION_DATA *p_section, int index)                  // aid > p_article->aid
674  {                  p_article = p_article->p_next;
675          ARTICLE *p_article;                  left++;
         ARTICLE_BLOCK *p_block;  
676    
677          if (p_section == NULL)                  // over last article in the page
678          {                  if (p_article == p_section->p_article_head || p_article->aid >= right)
679                  log_error("section_data_find_article_by_index() NULL pointer error\n");                  {
680                  return NULL;                          p_article = NULL;
681          }                          break;
682                    }
         if (index < 0 || index >= p_section->article_count)  
         {  
                 log_error("section_data_find_article_by_index(%d) is out of boundary [0, %d)\n", index, p_section->article_count);  
                 return NULL;  
683          }          }
684    
685          p_block = p_section->p_block[index / BBS_article_limit_per_block];          *p_offset = left;
         p_article = &(p_block->articles[index % BBS_article_limit_per_block]);  
686    
687          return p_article;          return p_article;
688  }  }
689    
690  int section_data_mark_del_article(SECTION_DATA *p_section, int32_t aid)  int section_list_calculate_page(SECTION_LIST *p_section, int32_t start_aid)
691  {  {
692          ARTICLE *p_article;          // ARTICLE *p_article;
693    
694          if (p_section == NULL)          if (p_section == NULL)
695          {          {
696                  log_error("section_data_mark_del_article() NULL pointer error\n");                  log_error("section_list_calculate_page() NULL pointer error\n");
697                  return -2;                  return -1;
         }  
   
         p_article = section_data_find_article_by_aid(p_section, aid);  
         if (p_article == NULL)  
         {  
                 return -1; // Not found  
         }  
   
         if (p_article->visible == 0)  
         {  
                 return 0; // Already deleted  
698          }          }
699    
700          p_article->visible = 0;          return 0;
         p_section->delete_count++;  
   
         return 1;  
701  }  }

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