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


Legend:
Removed lines/characters  
Changed lines/characters
  Added lines/characters

webmaster@leafok.com
ViewVC Help
Powered by ViewVC 1.3.0-beta1