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

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