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

Diff of /lbbs/src/section_list_loader.c

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

Revision 1.2 by sysadm, Mon May 26 04:47:45 2025 UTC Revision 1.41 by sysadm, Tue Sep 30 13:27:28 2025 UTC
# Line 14  Line 14 
14   *                                                                         *   *                                                                         *
15   ***************************************************************************/   ***************************************************************************/
16    
17  #include "section_list_loader.h"  #include "article_cache.h"
18  #include "log.h"  #include "bbs.h"
19  #include "database.h"  #include "database.h"
20  #include <stdio.h>  #include "ip_mask.h"
21  #include <string.h>  #include "log.h"
22    #include "menu.h"
23    #include "section_list_loader.h"
24  #include <errno.h>  #include <errno.h>
25    #include <signal.h>
26    #include <stdio.h>
27  #include <stdlib.h>  #include <stdlib.h>
28    #include <string.h>
29    #include <unistd.h>
30    
31    int section_list_loader_pid;
32    int last_article_op_log_mid;
33    
34  int load_section_config_from_db(void)  int load_section_config_from_db(int update_gen_ex)
35  {  {
36          MYSQL *db;          MYSQL *db = NULL;
37          MYSQL_RES *rs, *rs2;          MYSQL_RES *rs = NULL, *rs2 = NULL;
38          MYSQL_ROW row, row2;          MYSQL_ROW row, row2;
39          char sql[SQL_BUFFER_LEN];          char sql[SQL_BUFFER_LEN];
40          int32_t sid;          int32_t sid;
41          char master_name[BBS_username_max_len + 1];          char master_list[(BBS_username_max_len + 1) * 3 + 1];
42          SECTION_LIST *p_section;          SECTION_LIST *p_section;
43          int ret;          char ex_menu_conf[FILE_PATH_LEN];
44            MENU_SET ex_menu_set_new;
45            int ret = 0;
46    
47          db = db_open();          db = db_open();
48          if (db == NULL)          if (db == NULL)
49          {          {
50                  log_error("db_open() error: %s\n", mysql_error(db));                  log_error("db_open() error: %s\n", mysql_error(db));
51                  return -2;                  ret = -1;
52                    goto cleanup;
53          }          }
54    
55          snprintf(sql, sizeof(sql),          snprintf(sql, sizeof(sql),
56                           "SELECT section_config.SID, sname, section_config.title, section_config.CID, "                           "SELECT section_config.SID, sname, section_config.title, section_config.CID, "
57                           "read_user_level, write_user_level, section_config.enable * section_class.enable AS enable "                           "read_user_level, write_user_level, section_config.enable * section_class.enable AS enable, "
58                             "UNIX_TIMESTAMP(ex_menu_tm) AS ex_menu_tm "
59                           "FROM section_config INNER JOIN section_class ON section_config.CID = section_class.CID "                           "FROM section_config INNER JOIN section_class ON section_config.CID = section_class.CID "
60                           "ORDER BY section_config.SID");                           "ORDER BY section_config.SID");
61    
62          if (mysql_query(db, sql) != 0)          if (mysql_query(db, sql) != 0)
63          {          {
64                  log_error("Query section_list error: %s\n", mysql_error(db));                  log_error("Query section_list error: %s\n", mysql_error(db));
65                  return -3;                  ret = -2;
66                    goto cleanup;
67          }          }
68          if ((rs = mysql_store_result(db)) == NULL)          if ((rs = mysql_store_result(db)) == NULL)
69          {          {
70                  log_error("Get section_list data failed\n");                  log_error("Get section_list data failed\n");
71                  return -3;                  ret = -2;
72                    goto cleanup;
73          }          }
74    
75          ret = 0;          ret = 0;
# Line 67  int load_section_config_from_db(void) Line 82  int load_section_config_from_db(void)
82                                   "SELECT username FROM section_master "                                   "SELECT username FROM section_master "
83                                   "INNER JOIN user_list ON section_master.UID = user_list.UID "                                   "INNER JOIN user_list ON section_master.UID = user_list.UID "
84                                   "WHERE SID = %d AND section_master.enable AND (NOW() BETWEEN begin_dt AND end_dt) "                                   "WHERE SID = %d AND section_master.enable AND (NOW() BETWEEN begin_dt AND end_dt) "
85                                   "ORDER BY major DESC LIMIT 1",                                   "ORDER BY major DESC, begin_dt ASC LIMIT 3",
86                                   sid);                                   sid);
87    
88                  if (mysql_query(db, sql) != 0)                  if (mysql_query(db, sql) != 0)
# Line 82  int load_section_config_from_db(void) Line 97  int load_section_config_from_db(void)
97                          ret = -3;                          ret = -3;
98                          break;                          break;
99                  }                  }
100                  if ((row2 = mysql_fetch_row(rs2)))  
101                  {                  master_list[0] = '\0';
102                          strncpy(master_name, row2[0], sizeof(master_name) - 1);                  while ((row2 = mysql_fetch_row(rs2)))
                         master_name[sizeof(master_name) - 1] = '\0';  
                 }  
                 else  
103                  {                  {
104                          master_name[0] = '\0';                          strncat(master_list, row2[0], sizeof(master_list) - 1 - strnlen(master_list, sizeof(master_list)));
105                            strncat(master_list, " ", sizeof(master_list) - 1 - strnlen(master_list, sizeof(master_list)));
106                  }                  }
107                  mysql_free_result(rs2);                  mysql_free_result(rs2);
108                    rs2 = NULL;
109    
110                  p_section = section_list_find_by_sid(sid);                  p_section = section_list_find_by_sid(sid);
111    
112                  if (p_section == NULL)                  if (p_section == NULL)
113                  {                  {
114                          p_section = section_list_create(sid, row[1], row[2], "");                          p_section = section_list_create(sid, row[1], row[2], master_list);
115                          if (p_section == NULL)                          if (p_section == NULL)
116                          {                          {
117                                  log_error("section_list_create() error: load new section sid = %d sname = %s\n", sid, row[1]);                                  log_error("section_list_create() error: load new section sid = %d sname = %s\n", sid, row[1]);
# Line 123  int load_section_config_from_db(void) Line 137  int load_section_config_from_db(void)
137    
138                          strncpy(p_section->sname, row[1], sizeof(p_section->sname) - 1);                          strncpy(p_section->sname, row[1], sizeof(p_section->sname) - 1);
139                          p_section->sname[sizeof(p_section->sname) - 1] = '\0';                          p_section->sname[sizeof(p_section->sname) - 1] = '\0';
140                          strncpy(p_section->stitle, row[1], sizeof(p_section->stitle) - 1);                          strncpy(p_section->stitle, row[2], sizeof(p_section->stitle) - 1);
141                          p_section->stitle[sizeof(p_section->stitle) - 1] = '\0';                          p_section->stitle[sizeof(p_section->stitle) - 1] = '\0';
142                          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);
143                          p_section->master_name[sizeof(p_section->master_name) - 1] = '\0';                          p_section->master_list[sizeof(p_section->master_list) - 1] = '\0';
144                  }                  }
145    
146                  p_section->class_id = atoi(row[3]);                  p_section->class_id = atoi(row[3]);
# Line 134  int load_section_config_from_db(void) Line 148  int load_section_config_from_db(void)
148                  p_section->write_user_level = atoi(row[5]);                  p_section->write_user_level = atoi(row[5]);
149                  p_section->enable = (int8_t)atoi(row[6]);                  p_section->enable = (int8_t)atoi(row[6]);
150    
151                    // Update gen_ex menu set
152                    if (update_gen_ex && p_section->enable && atoi(row[7]) > p_section->ex_menu_tm)
153                    {
154                            snprintf(ex_menu_conf, sizeof(ex_menu_conf), "%s/%d", VAR_GEN_EX_MENU_DIR, p_section->sid);
155    
156                            ret = load_menu(&ex_menu_set_new, ex_menu_conf);
157                            if (ret < 0)
158                            {
159                                    unload_menu(&ex_menu_set_new);
160                                    log_error("load_menu(%s) error: %d\n", ex_menu_conf, ret);
161                            }
162                            else
163                            {
164                                    if (p_section->ex_menu_tm > 0)
165                                    {
166                                            unload_menu(&(p_section->ex_menu_set));
167                                    }
168    
169                                    ex_menu_set_new.allow_exit = 1; // Allow exit menu
170                                    memcpy(&(p_section->ex_menu_set), &ex_menu_set_new, sizeof(ex_menu_set_new));
171    
172                                    p_section->ex_menu_tm = atol(row[7]);
173    #ifdef _DEBUG
174                                    log_common("Loaded gen_ex_menu of section %d [%s]\n", p_section->sid, p_section->sname);
175    #endif
176                            }
177                    }
178    
179                  // release rw lock                  // release rw lock
180                  ret = section_list_rw_unlock(p_section);                  ret = section_list_rw_unlock(p_section);
181                  if (ret < 0)                  if (ret < 0)
# Line 141  int load_section_config_from_db(void) Line 183  int load_section_config_from_db(void)
183                          break;                          break;
184                  }                  }
185          }          }
186    
187    cleanup:
188            mysql_free_result(rs2);
189            mysql_free_result(rs);
190            mysql_close(db);
191    
192            return ret;
193    }
194    
195    int append_articles_from_db(int32_t start_aid, int global_lock, int article_count_limit)
196    {
197            MYSQL *db = NULL;
198            MYSQL_RES *rs = NULL;
199            MYSQL_ROW row;
200            char sql[SQL_BUFFER_LEN];
201            ARTICLE article;
202            ARTICLE *p_topic;
203            SECTION_LIST *p_section = NULL;
204            int32_t last_sid = 0;
205            char sub_ip[IP_ADDR_LEN];
206            int article_count = 0;
207            int ret = 0;
208            int i;
209            size_t j;
210    
211            db = db_open();
212            if (db == NULL)
213            {
214                    log_error("db_open() error: %s\n", mysql_error(db));
215                    ret = -1;
216                    goto cleanup;
217            }
218    
219            snprintf(sql, sizeof(sql),
220                             "SELECT bbs.AID, TID, SID, bbs.CID, UID, visible, excerption, ontop, `lock`, "
221                             "transship, username, nickname, title, UNIX_TIMESTAMP(sub_dt) AS sub_dt, "
222                             "sub_ip, bbs_content.content "
223                             "FROM bbs INNER JOIN bbs_content ON bbs.CID = bbs_content.CID "
224                             "WHERE bbs.AID >= %d ORDER BY bbs.AID LIMIT %d",
225                             start_aid, article_count_limit);
226    
227            if (mysql_query(db, sql) != 0)
228            {
229                    log_error("Query article list error: %s\n", mysql_error(db));
230                    ret = -2;
231                    goto cleanup;
232            }
233            if ((rs = mysql_use_result(db)) == NULL)
234            {
235                    log_error("Get article list data failed\n");
236                    ret = -2;
237                    goto cleanup;
238            }
239    
240            // acquire global lock
241            if (global_lock && (ret = section_list_rw_lock(NULL)) < 0)
242            {
243                    log_error("section_list_rw_lock(sid = 0) error\n");
244                    goto cleanup;
245            }
246    
247            while ((row = mysql_fetch_row(rs)))
248            {
249                    memset(&article, 0, sizeof(ARTICLE));
250    
251                    // copy data of article
252                    i = 0;
253    
254                    article.aid = atoi(row[i++]);
255                    article.tid = atoi(row[i++]);
256                    article.sid = atoi(row[i++]);
257                    article.cid = atoi(row[i++]);
258                    article.uid = atoi(row[i++]);
259                    article.visible = (int8_t)atoi(row[i++]);
260                    article.excerption = (int8_t)atoi(row[i++]);
261                    article.ontop = (int8_t)atoi(row[i++]);
262                    article.lock = (int8_t)atoi(row[i++]);
263                    article.transship = (int8_t)atoi(row[i++]);
264    
265                    strncpy(article.username, row[i++], sizeof(article.username) - 1);
266                    article.username[sizeof(article.username) - 1] = '\0';
267                    strncpy(article.nickname, row[i++], sizeof(article.nickname) - 1);
268                    article.nickname[sizeof(article.nickname) - 1] = '\0';
269                    strncpy(article.title, row[i++], sizeof(article.title) - 1);
270                    article.title[sizeof(article.title) - 1] = '\0';
271    
272                    // Rewrite title with "Re: Re: " prefix into "Re: ... "
273                    if (article.tid != 0)
274                    {
275                            for (j = 0; strncmp(article.title + j, "Re: ", strlen("Re: ")) == 0; j += strlen("Re: "))
276                                    ;
277                            if (j >= strlen("Re: Re: "))
278                            {
279                                    memcpy(article.title, "Re: ... ", strlen("Re: ... "));
280                                    memmove(article.title + strlen("Re: ... "), article.title + j, sizeof(article.title) - 1 - j);
281                                    article.title[sizeof(article.title) - 1 - j] = '\0';
282                            }
283                    }
284    
285                    article.sub_dt = atol(row[i++]);
286    
287                    strncpy(sub_ip, row[i++], sizeof(sub_ip) - 1);
288                    sub_ip[sizeof(sub_ip) - 1] = '\0';
289                    ip_mask(sub_ip, 1, '*');
290    
291                    // release lock of last section if different from current one
292                    if (!global_lock && article.sid != last_sid && last_sid != 0)
293                    {
294                            if ((ret = section_list_rw_unlock(p_section)) < 0)
295                            {
296                                    log_error("section_list_rw_unlock(sid=%d) error\n", p_section->sid);
297                                    break;
298                            }
299                    }
300    
301                    if ((p_section = section_list_find_by_sid(article.sid)) == NULL)
302                    {
303                            log_error("section_list_find_by_sid(%d) error: unknown section, try reloading section config\n", article.sid);
304                            ret = ERR_UNKNOWN_SECTION; // Unknown section found
305                            break;
306                    }
307    
308                    if (article.visible != 0 && article.tid != 0)
309                    {
310                            // Check if topic article is visible
311                            p_topic = article_block_find_by_aid(article.tid);
312                            if (p_topic == NULL || p_topic->visible == 0)
313                            {
314                                    // log_error("Set article (aid = %d) as invisible due to invisible or non-existing topic head\n", article.aid);
315                                    article.tid = 0;
316                                    article.visible = 0;
317                            }
318                    }
319    
320                    // acquire lock of current section if different from last one
321                    if (!global_lock && article.sid != last_sid)
322                    {
323                            if ((ret = section_list_rw_lock(p_section)) < 0)
324                            {
325                                    log_error("section_list_rw_lock(sid=0) error\n");
326                                    break;
327                            }
328                    }
329    
330                    // append article to section list
331                    last_sid = article.sid;
332    
333                    if (section_list_append_article(p_section, &article) < 0)
334                    {
335                            log_error("section_list_append_article(sid=%d, aid=%d) error\n",
336                                              p_section->sid, article.aid);
337                            ret = -3;
338                            break;
339                    }
340    
341                    article_count++;
342    
343                    if (article_cache_generate(VAR_ARTICLE_CACHE_DIR, &article, p_section, row[i++], sub_ip, 0) < 0)
344                    {
345                            log_error("article_cache_generate(aid=%d, cid=%d) error\n", article.aid, article.cid);
346                            ret = -4;
347                            break;
348                    }
349            }
350    
351            // release lock of last section
352            if (!global_lock && last_sid != 0)
353            {
354                    if ((ret = section_list_rw_unlock(p_section)) < 0)
355                    {
356                            log_error("section_list_rw_unlock(sid=%d) error\n", p_section->sid);
357                    }
358            }
359    
360            // release global lock
361            if (global_lock && (ret = section_list_rw_unlock(NULL)) < 0)
362            {
363                    log_error("section_list_rw_unlock(sid=0) error\n");
364            }
365    
366    cleanup:
367          mysql_free_result(rs);          mysql_free_result(rs);
368            mysql_close(db);
369    
370            article_cache_cleanup();
371    
372            return (ret < 0 ? ret : article_count);
373    }
374    
375    int set_last_article_op_log_from_db(void)
376    {
377            MYSQL *db = NULL;
378            MYSQL_RES *rs = NULL;
379            MYSQL_ROW row;
380            char sql[SQL_BUFFER_LEN];
381            int ret = 0;
382    
383            db = db_open();
384            if (db == NULL)
385            {
386                    log_error("db_open() error: %s\n", mysql_error(db));
387                    ret = -1;
388                    goto cleanup;
389            }
390    
391            snprintf(sql, sizeof(sql),
392                             "SELECT MID FROM bbs_article_op ORDER BY MID DESC LIMIT 1");
393    
394            if (mysql_query(db, sql) != 0)
395            {
396                    log_error("Query article op error: %s\n", mysql_error(db));
397                    ret = -2;
398                    goto cleanup;
399            }
400            if ((rs = mysql_store_result(db)) == NULL)
401            {
402                    log_error("Get article op data failed\n");
403                    ret = -2;
404                    goto cleanup;
405            }
406    
407            if ((row = mysql_fetch_row(rs)))
408            {
409                    last_article_op_log_mid = atoi(row[0]);
410            }
411    
412    cleanup:
413            mysql_free_result(rs);
414            mysql_close(db);
415    
416            return (ret < 0 ? ret : last_article_op_log_mid);
417    }
418    
419    int apply_article_op_log_from_db(int op_count_limit)
420    {
421            MYSQL *db = NULL;
422            MYSQL_RES *rs = NULL, *rs2 = NULL;
423            MYSQL_ROW row, row2;
424            char sql[SQL_BUFFER_LEN];
425            ARTICLE *p_article;
426            SECTION_LIST *p_section = NULL;
427            SECTION_LIST *p_section_dest;
428            char sub_ip[IP_ADDR_LEN];
429            int32_t last_sid = 0;
430            int32_t sid_dest;
431            int op_count = 0;
432            int ret = 0;
433    
434            db = db_open();
435            if (db == NULL)
436            {
437                    log_error("db_open() error: %s\n", mysql_error(db));
438                    ret = -1;
439                    goto cleanup;
440            }
441    
442            snprintf(sql, sizeof(sql),
443                             "SELECT MID, AID, type FROM bbs_article_op "
444                             "WHERE MID > %d AND type NOT IN ('A') "
445                             "ORDER BY MID LIMIT %d",
446                             last_article_op_log_mid, op_count_limit);
447    
448            if (mysql_query(db, sql) != 0)
449            {
450                    log_error("Query article log error: %s\n", mysql_error(db));
451                    ret = -2;
452                    goto cleanup;
453            }
454            if ((rs = mysql_store_result(db)) == NULL)
455            {
456                    log_error("Get article log data failed\n");
457                    ret = -2;
458                    goto cleanup;
459            }
460    
461            while ((row = mysql_fetch_row(rs)))
462            {
463                    p_article = article_block_find_by_aid(atoi(row[1]));
464                    if (p_article == NULL) // related article has not been appended yet
465                    {
466                            ret = -2;
467                            break;
468                    }
469    
470                    // release lock of last section if different from current one
471                    if (p_article->sid != last_sid && last_sid != 0)
472                    {
473                            if ((ret = section_list_rw_unlock(p_section)) < 0)
474                            {
475                                    log_error("section_list_rw_unlock(sid = %d) error\n", p_section->sid);
476                                    break;
477                            }
478                    }
479    
480                    if ((p_section = section_list_find_by_sid(p_article->sid)) == NULL)
481                    {
482                            log_error("section_list_find_by_sid(%d) error: unknown section, try reloading section config\n", p_article->sid);
483                            ret = ERR_UNKNOWN_SECTION; // Unknown section found
484                            break;
485                    }
486    
487                    // acquire lock of current section if different from last one
488                    if (p_article->sid != last_sid)
489                    {
490                            if ((ret = section_list_rw_lock(p_section)) < 0)
491                            {
492                                    log_error("section_list_rw_lock(sid = 0) error\n");
493                                    break;
494                            }
495                    }
496    
497                    last_sid = p_article->sid;
498    
499                    switch (row[2][0])
500                    {
501                    case 'A': // Add article
502                            log_error("Operation type=A should not be found\n");
503                            break;
504                    case 'D': // Delete article
505                    case 'X': // Delete article by Admin
506                            if (section_list_set_article_visible(p_section, atoi(row[1]), 0) < 0)
507                            {
508                                    log_error("section_list_set_article_visible(sid=%d, aid=%d, visible=0) error\n", p_section->sid, atoi(row[1]));
509                            }
510                            if (section_list_calculate_page(p_section, atoi(row[1])) < 0)
511                            {
512                                    log_error("section_list_calculate_page(aid=%d) error\n", atoi(row[1]));
513                            }
514                            break;
515                    case 'S': // Restore article
516                            if (section_list_set_article_visible(p_section, atoi(row[1]), 1) < 0)
517                            {
518                                    log_error("section_list_set_article_visible(sid=%d, aid=%d, visible=1) error\n", p_section->sid, atoi(row[1]));
519                            }
520                            if (section_list_calculate_page(p_section, atoi(row[1])) < 0)
521                            {
522                                    log_error("section_list_calculate_page(aid=%d) error\n", atoi(row[1]));
523                            }
524                            break;
525                    case 'L': // Lock article
526                            p_article->lock = 1;
527                            break;
528                    case 'U': // Unlock article
529                            p_article->lock = 0;
530                            break;
531                    case 'M': // Modify article
532                            snprintf(sql, sizeof(sql),
533                                             "SELECT bbs.CID, sub_ip, bbs_content.content "
534                                             "FROM bbs INNER JOIN bbs_content ON bbs.CID = bbs_content.CID "
535                                             "WHERE bbs.AID = %d",
536                                             p_article->aid);
537    
538                            if (mysql_query(db, sql) != 0)
539                            {
540                                    log_error("Query article error: %s\n", mysql_error(db));
541                                    ret = -3;
542                                    break;
543                            }
544                            if ((rs2 = mysql_use_result(db)) == NULL)
545                            {
546                                    log_error("Get article data failed\n");
547                                    ret = -3;
548                                    break;
549                            }
550                            if ((row2 = mysql_fetch_row(rs2)))
551                            {
552                                    p_article->cid = atoi(row2[0]);
553    
554                                    strncpy(sub_ip, row2[1], sizeof(sub_ip) - 1);
555                                    sub_ip[sizeof(sub_ip) - 1] = '\0';
556                                    ip_mask(sub_ip, 1, '*');
557    
558                                    if (article_cache_generate(VAR_ARTICLE_CACHE_DIR, p_article, p_section, row2[2], sub_ip, 0) < 0)
559                                    {
560                                            log_error("article_cache_generate(aid=%d, cid=%d) error\n", p_article->aid, p_article->cid);
561                                            ret = -4;
562                                    }
563                            }
564                            else
565                            {
566                                    p_article->cid = 0;
567                                    ret = -4;
568                            }
569                            mysql_free_result(rs2);
570                            rs2 = NULL;
571    
572                            p_article->excerption = 0; // Set excerption = 0 implicitly in case of rare condition
573                            break;
574                    case 'T': // Move article
575                            snprintf(sql, sizeof(sql),
576                                             "SELECT SID FROM bbs WHERE AID = %d",
577                                             p_article->aid);
578    
579                            if (mysql_query(db, sql) != 0)
580                            {
581                                    log_error("Query article error: %s\n", mysql_error(db));
582                                    ret = -3;
583                                    break;
584                            }
585                            if ((rs2 = mysql_store_result(db)) == NULL)
586                            {
587                                    log_error("Get article data failed\n");
588                                    ret = -3;
589                                    break;
590                            }
591                            if ((row2 = mysql_fetch_row(rs2)))
592                            {
593                                    sid_dest = atoi(row2[0]);
594                            }
595                            else
596                            {
597                                    sid_dest = 0;
598                                    ret = -4;
599                            }
600                            mysql_free_result(rs2);
601                            rs2 = NULL;
602    
603                            if (sid_dest > 0 && sid_dest != p_article->sid)
604                            {
605                                    p_section_dest = section_list_find_by_sid(sid_dest);
606                                    if (p_section_dest == NULL)
607                                    {
608                                            ret = ERR_UNKNOWN_SECTION;
609                                            break;
610                                    }
611                                    // acquire lock of dest section
612                                    if ((ret = section_list_rw_lock(p_section_dest)) < 0)
613                                    {
614                                            log_error("section_list_rw_lock(sid = %d) error\n", p_section_dest);
615                                            break;
616                                    }
617                                    // Move topic
618                                    if ((ret = section_list_move_topic(p_section, p_section_dest, p_article->aid)) < 0)
619                                    {
620                                            log_error("section_list_move_topic(src_sid=%d, dest_sid=%d, aid=%d) error (%d), retry in the next loop\n",
621                                                              p_section->sid, p_section_dest->sid, p_article->aid, ret);
622                                    }
623                                    // release lock of dest section
624                                    if (section_list_rw_unlock(p_section_dest) < 0)
625                                    {
626                                            log_error("section_list_rw_unlock(sid = %d) error\n", p_section_dest);
627                                            ret = -1;
628                                    }
629                            }
630                            break;
631                    case 'E': // Set article as excerption
632                            p_article->excerption = 1;
633                            break;
634                    case 'O': // Unset article as excerption
635                            p_article->excerption = 0;
636                            break;
637                    case 'F': // Set article on top
638                    case 'V': // Unset article on top
639                            p_article->ontop = (row[2][0] == 'F' ? 1 : 0);
640                            if (section_list_update_article_ontop(p_section, p_article) < 0)
641                            {
642                                    log_error("section_list_update_article_ontop(sid=%d, aid=%d) error\n",
643                                                      p_section->sid, p_article->aid);
644                            }
645                            break;
646                    case 'Z': // Set article as trnasship
647                            p_article->transship = 1;
648                            break;
649                    default:
650                            // log_error("Operation type=%s unknown, mid=%s\n", row[2], row[0]);
651                            break;
652                    }
653    
654                    if (ret < 0)
655                    {
656                            break;
657                    }
658    
659                    // Update MID with last successfully proceeded article_op_log
660                    last_article_op_log_mid = atoi(row[0]);
661    
662                    op_count++;
663            }
664    
665            // release lock of last section
666            if (last_sid != 0)
667            {
668                    if ((ret = section_list_rw_unlock(p_section)) < 0)
669                    {
670                            log_error("section_list_rw_unlock(sid = %d) error\n", p_section->sid);
671                    }
672            }
673    
674    cleanup:
675            mysql_free_result(rs2);
676            mysql_free_result(rs);
677          mysql_close(db);          mysql_close(db);
678    
679            return (ret < 0 ? ret : op_count);
680    }
681    
682    static void section_list_ex_menu_set_cleanup(void)
683    {
684            int i;
685    
686            for (i = 0; i < p_section_list_pool->section_count; i++)
687            {
688                    if (p_section_list_pool->sections[i].ex_menu_tm > 0)
689                    {
690                            unload_menu(&(p_section_list_pool->sections[i].ex_menu_set));
691                    }
692            }
693    }
694    
695    int section_list_loader_launch(void)
696    {
697            struct sigaction act = {0};
698            int pid;
699            int ret;
700            int32_t last_aid;
701            int article_count;
702            int load_count;
703            int last_mid;
704            int i;
705    
706            if (section_list_loader_pid != 0)
707            {
708                    log_error("section_list_loader already running, pid = %d\n", section_list_loader_pid);
709                    return -2;
710            }
711    
712            pid = fork();
713    
714            if (pid > 0) // Parent process
715            {
716                    SYS_child_process_count++;
717                    section_list_loader_pid = pid;
718                    log_common("Section list loader process (pid = %d) start\n", pid);
719                    return 0;
720            }
721            else if (pid < 0) // Error
722            {
723                    log_error("fork() error (%d)\n", errno);
724                    return -1;
725            }
726    
727            // Child process
728            SYS_child_process_count = 0;
729    
730            // Detach menu in shared memory
731            detach_menu_shm(&bbs_menu);
732    
733            // Set signal handler
734            act.sa_handler = SIG_DFL;
735            if (sigaction(SIGHUP, &act, NULL) == -1)
736            {
737                    log_error("set signal action of SIGHUP error: %d\n", errno);
738            }
739            act.sa_handler = SIG_DFL;
740            if (sigaction(SIGCHLD, &act, NULL) == -1)
741            {
742                    log_error("set signal action of SIGCHLD error: %d\n", errno);
743            }
744    
745            // Do section data loader periodically
746            while (!SYS_server_exit)
747            {
748                    if (SYS_conf_reload)
749                    {
750                            SYS_conf_reload = 0;
751    
752                            // Load section config
753                            if (load_section_config_from_db(0) < 0)
754                            {
755                                    log_error("load_section_config_from_db(0) error\n");
756                            }
757                            else
758                            {
759                                    log_common("Reload section config successfully\n");
760                            }
761                    }
762    
763                    // Load section articles
764                    article_count = article_block_article_count();
765    
766                    do
767                    {
768                            last_aid = article_block_last_aid();
769    
770                            if ((ret = append_articles_from_db(last_aid + 1, 0, LOAD_ARTICLE_COUNT_LIMIT)) < 0)
771                            {
772                                    log_error("append_articles_from_db(%d, 0, %d) error\n", last_aid + 1, LOAD_ARTICLE_COUNT_LIMIT);
773    
774                                    if (ret == ERR_UNKNOWN_SECTION)
775                                    {
776                                            SYS_conf_reload = 1; // Force reload section_list
777                                    }
778                            }
779                    } while (ret == LOAD_ARTICLE_COUNT_LIMIT);
780    
781                    load_count = article_block_article_count() - article_count;
782                    if (load_count > 0)
783                    {
784                            log_common("Incrementally load %d articles, last_aid = %d\n", load_count, article_block_last_aid());
785                    }
786    
787                    if (SYS_conf_reload)
788                    {
789                            continue;
790                    }
791    
792                    // Load article_op log
793                    last_mid = last_article_op_log_mid;
794    
795                    do
796                    {
797                            if ((ret = apply_article_op_log_from_db(LOAD_ARTICLE_COUNT_LIMIT)) < 0)
798                            {
799                                    log_error("apply_article_op_log_from_db() error\n");
800    
801                                    if (ret == ERR_UNKNOWN_SECTION)
802                                    {
803                                            SYS_conf_reload = 1; // Force reload section_list
804                                    }
805                            }
806                    } while (ret == LOAD_ARTICLE_COUNT_LIMIT);
807    
808                    if (last_article_op_log_mid > last_mid)
809                    {
810                            log_common("Proceeded %d article logs, last_mid = %d\n", last_article_op_log_mid - last_mid, last_article_op_log_mid);
811                    }
812    
813                    if (SYS_conf_reload)
814                    {
815                            continue;
816                    }
817    
818                    for (i = 0; i < BBS_section_list_load_interval && !SYS_server_exit && !SYS_conf_reload; i++)
819                    {
820                            sleep(1);
821                    }
822            }
823    
824            // Child process exit
825    
826            article_cache_cleanup();
827    
828            // gen_ex_menu cleanup
829            section_list_ex_menu_set_cleanup();
830    
831            // Detach data pools shm
832            detach_section_list_shm();
833            detach_article_block_shm();
834            detach_trie_dict_shm();
835    
836            log_common("Section list loader process exit normally\n");
837            log_end();
838    
839            section_list_loader_pid = 0;
840    
841            _exit(0);
842    
843            return 0;
844    }
845    
846    int query_section_articles(SECTION_LIST *p_section, int page_id, ARTICLE *p_articles[],
847                                                       int *p_article_count, int *p_page_count, int *p_ontop_start_offset)
848    {
849            ARTICLE *p_article;
850            ARTICLE *p_next_page_first_article;
851            int ret = 0;
852            int i;
853    
854            if (p_section == NULL || p_articles == NULL || p_article_count == NULL || p_page_count == NULL || p_ontop_start_offset == NULL)
855            {
856                    log_error("NULL pointer error\n");
857                    return -1;
858            }
859    
860            // acquire lock of section
861            if ((ret = section_list_rd_lock(p_section)) < 0)
862            {
863                    log_error("section_list_rd_lock(sid = %d) error\n", p_section->sid);
864                    return -2;
865            }
866    
867            *p_page_count = section_list_page_count_with_ontop(p_section);
868            *p_article_count = 0;
869            *p_ontop_start_offset = BBS_article_limit_per_page;
870    
871            if (p_section->visible_article_count == 0)
872            {
873                    // empty section
874            }
875            else if (page_id < 0 || page_id >= *p_page_count)
876            {
877                    log_error("Invalid page_id=%d, not in range [0, %d)\n", page_id, *p_page_count);
878                    ret = -3;
879            }
880            else
881            {
882                    ret = page_id;
883    
884                    if (page_id <= p_section->page_count - 1)
885                    {
886                            p_article = p_section->p_page_first_article[page_id];
887                            p_next_page_first_article =
888                                    (page_id == p_section->page_count - 1 ? p_section->p_article_head : p_section->p_page_first_article[page_id + 1]);
889    
890                            do
891                            {
892                                    if (p_article->visible)
893                                    {
894                                            p_articles[*p_article_count] = p_article;
895                                            (*p_article_count)++;
896                                    }
897                                    p_article = p_article->p_next;
898                            } while (p_article != p_next_page_first_article && (*p_article_count) <= BBS_article_limit_per_page);
899    
900                            if (*p_article_count != (page_id < p_section->page_count - 1 ? BBS_article_limit_per_page : p_section->last_page_visible_article_count))
901                            {
902                                    log_error("Inconsistent visible article count %d detected in section %d page %d\n", *p_article_count, p_section->sid, page_id);
903                            }
904                    }
905    
906                    *p_ontop_start_offset = *p_article_count;
907    
908                    // Append ontop articles
909                    if (page_id >= p_section->page_count - 1)
910                    {
911                            i = (page_id == p_section->page_count - 1
912                                             ? 0
913                                             : (page_id - p_section->page_count + 1) * BBS_article_limit_per_page - p_section->last_page_visible_article_count);
914                            while (*p_article_count < BBS_article_limit_per_page && i < p_section->ontop_article_count)
915                            {
916                                    p_articles[(*p_article_count)++] = p_section->p_ontop_articles[i++];
917                            }
918                    }
919            }
920    
921            // release lock of section
922            if (section_list_rd_unlock(p_section) < 0)
923            {
924                    log_error("section_list_rd_unlock(sid = %d) error\n", p_section->sid);
925                    ret = -2;
926            }
927    
928            return ret;
929    }
930    
931    int locate_article_in_section(SECTION_LIST *p_section, const ARTICLE *p_article_cur, int direction, int step,
932                                                              int *p_page_id, int *p_visible_offset, int *p_article_count)
933    {
934            const ARTICLE *p_article = NULL;
935            const ARTICLE *p_article_last = NULL;
936            ARTICLE *p_tmp;
937            int32_t aid = 0;
938            int page_id = 0;
939            int offset = 0;
940            int ret = 0;
941            int i;
942    
943            if (p_section == NULL || p_article_cur == NULL || p_page_id == NULL || p_visible_offset == NULL || p_article_count == NULL)
944            {
945                    log_error("NULL pointer error\n");
946                    return -1;
947            }
948    
949            // acquire lock of section
950            if ((ret = section_list_rd_lock(p_section)) < 0)
951            {
952                    log_error("section_list_rd_lock(sid = %d) error\n", p_section->sid);
953                    return -2;
954            }
955    
956            if (direction == 0)
957            {
958                    aid = p_article_cur->aid;
959            }
960            else if (direction == 1)
961            {
962                    for (p_article = p_article_cur, p_article_last = p_article_cur;
963                             step > 0 && p_article->p_topic_next->aid > p_article_cur->aid;
964                             p_article = p_article->p_topic_next)
965                    {
966                            if (p_article->visible)
967                            {
968                                    step--;
969                                    p_article_last = p_article;
970                            }
971                    }
972                    if (!p_article->visible)
973                    {
974                            p_article = p_article_last;
975                    }
976    
977                    aid = (p_article->aid > p_article_cur->aid && p_article->visible ? p_article->aid : 0);
978            }
979            else if (direction == -1)
980            {
981                    for (p_article = p_article_cur, p_article_last = p_article_cur;
982                             step > 0 && p_article->p_topic_prior->aid < p_article_cur->aid;
983                             p_article = p_article->p_topic_prior)
984                    {
985                            if (p_article->visible)
986                            {
987                                    step--;
988                                    p_article_last = p_article;
989                            }
990                    }
991                    if (!p_article->visible)
992                    {
993                            p_article = p_article_last;
994                    }
995    
996                    aid = (p_article->aid < p_article_cur->aid && p_article->visible ? p_article->aid : 0);
997            }
998    
999            p_article = NULL;
1000    
1001            if (aid > 0)
1002            {
1003                    p_article = section_list_find_article_with_offset(p_section, aid, &page_id, &offset, &p_tmp);
1004                    if (p_article != NULL)
1005                    {
1006                            *p_article_count = (page_id == p_section->page_count - 1 ? p_section->last_page_visible_article_count : BBS_article_limit_per_page);
1007    
1008                            p_article = p_section->p_page_first_article[page_id];
1009                            for (i = 0; i < *p_article_count && p_article != NULL;)
1010                            {
1011                                    if (p_article->visible)
1012                                    {
1013                                            if (p_article->aid == aid)
1014                                            {
1015                                                    *p_page_id = page_id;
1016                                                    *p_visible_offset = i;
1017                                                    break;
1018                                            }
1019                                            i++;
1020                                            if (i >= *p_article_count)
1021                                            {
1022                                                    log_error("Visible article (aid=%d) not found in page %d\n", aid, page_id);
1023                                                    p_article = NULL;
1024                                                    break;
1025                                            }
1026                                    }
1027                                    p_article = p_article->p_next;
1028                            }
1029    
1030                            // Include ontop articles
1031                            *p_article_count = section_list_page_article_count_with_ontop(p_section, page_id);
1032                    }
1033            }
1034    
1035            // release lock of section
1036            if (section_list_rd_unlock(p_section) < 0)
1037            {
1038                    log_error("section_list_rd_unlock(sid = %d) error\n", p_section->sid);
1039                    ret = -2;
1040            }
1041    
1042            return (ret < 0 ? ret : (p_article == NULL ? 0 : 1));
1043    }
1044    
1045    int get_section_ex_menu_set(SECTION_LIST *p_section, MENU_SET *p_ex_menu_set)
1046    {
1047            int ret = 0;
1048    
1049            if (p_section == NULL || p_ex_menu_set == NULL)
1050            {
1051                    log_error("NULL pointer error\n");
1052                    return -1;
1053            }
1054    
1055            // acquire lock of section
1056            if ((ret = section_list_rd_lock(p_section)) < 0)
1057            {
1058                    log_error("section_list_rd_lock(sid = %d) error\n", p_section->sid);
1059                    return -2;
1060            }
1061    
1062            memcpy(p_ex_menu_set, &(p_section->ex_menu_set), sizeof(p_section->ex_menu_set));
1063    
1064            // release lock of section
1065            if (section_list_rd_unlock(p_section) < 0)
1066            {
1067                    log_error("section_list_rd_unlock(sid = %d) error\n", p_section->sid);
1068                    ret = -2;
1069            }
1070    
1071          return ret;          return ret;
1072  }  }

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