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

Annotation of /lbbs/src/section_list_loader.c

Parent Directory Parent Directory | Revision Log Revision Log

Revision 1.24 - (hide annotations)
Tue Jun 17 07:00:32 2025 UTC (9 months ago) by sysadm
Branch: MAIN
Changes since 1.23: +15 -11 lines
Content type: text/x-csrc 
Support locating at first / last topic article by key '=' and '\'
1 sysadm 1.1 /***************************************************************************
2     section_list_loader.c - description
3     -------------------
4     Copyright : (C) 2004-2025 by Leaflet
5     Email : leaflet@leafok.com
6     ***************************************************************************/
7    
8     /***************************************************************************
9     * *
10     * This program is free software; you can redistribute it and/or modify *
11     * it under the terms of the GNU General Public License as published by *
12     * the Free Software Foundation; either version 3 of the License, or *
13     * (at your option) any later version. *
14     * *
15     ***************************************************************************/
16    
17 sysadm 1.22 #define _POSIX_C_SOURCE 200809L
18    
19 sysadm 1.1 #include "section_list_loader.h"
20 sysadm 1.11 #include "article_cache.h"
21 sysadm 1.19 #include "bbs.h"
22 sysadm 1.1 #include "log.h"
23     #include "database.h"
24 sysadm 1.4 #include "menu.h"
25 sysadm 1.1 #include <stdio.h>
26     #include <string.h>
27     #include <errno.h>
28 sysadm 1.4 #include <signal.h>
29 sysadm 1.1 #include <stdlib.h>
30 sysadm 1.22 #include <string.h>
31 sysadm 1.3 #include <strings.h>
32 sysadm 1.4 #include <unistd.h>
33    
34 sysadm 1.5 int section_list_loader_pid;
35     int last_article_op_log_mid;
36 sysadm 1.1
37 sysadm 1.2 int load_section_config_from_db(void)
38 sysadm 1.1 {
39 sysadm 1.23 MYSQL *db = NULL;
40     MYSQL_RES *rs = NULL, *rs2 = NULL;
41 sysadm 1.1 MYSQL_ROW row, row2;
42     char sql[SQL_BUFFER_LEN];
43     int32_t sid;
44 sysadm 1.8 char master_list[(BBS_username_max_len + 1) * 3 + 1];
45 sysadm 1.1 SECTION_LIST *p_section;
46 sysadm 1.23 int ret = 0;
47 sysadm 1.1
48 sysadm 1.2 db = db_open();
49     if (db == NULL)
50     {
51     log_error("db_open() error: %s\n", mysql_error(db));
52 sysadm 1.23 ret = -1;
53     goto cleanup;
54 sysadm 1.2 }
55    
56 sysadm 1.1 snprintf(sql, sizeof(sql),
57 sysadm 1.2 "SELECT section_config.SID, sname, section_config.title, section_config.CID, "
58     "read_user_level, write_user_level, section_config.enable * section_class.enable AS enable "
59     "FROM section_config INNER JOIN section_class ON section_config.CID = section_class.CID "
60 sysadm 1.1 "ORDER BY section_config.SID");
61    
62     if (mysql_query(db, sql) != 0)
63     {
64     log_error("Query section_list error: %s\n", mysql_error(db));
65 sysadm 1.23 ret = -2;
66     goto cleanup;
67 sysadm 1.1 }
68     if ((rs = mysql_store_result(db)) == NULL)
69     {
70     log_error("Get section_list data failed\n");
71 sysadm 1.23 ret = -2;
72     goto cleanup;
73 sysadm 1.1 }
74 sysadm 1.2
75     ret = 0;
76 sysadm 1.1 while ((row = mysql_fetch_row(rs)))
77     {
78     sid = atoi(row[0]);
79    
80     // Query section master
81     snprintf(sql, sizeof(sql),
82     "SELECT username FROM section_master "
83     "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) "
85 sysadm 1.8 "ORDER BY major DESC, begin_dt ASC LIMIT 3",
86 sysadm 1.1 sid);
87    
88     if (mysql_query(db, sql) != 0)
89     {
90     log_error("Query section_master error: %s\n", mysql_error(db));
91 sysadm 1.2 ret = -3;
92     break;
93 sysadm 1.1 }
94     if ((rs2 = mysql_store_result(db)) == NULL)
95     {
96     log_error("Get section_master data failed\n");
97 sysadm 1.2 ret = -3;
98     break;
99 sysadm 1.1 }
100 sysadm 1.8
101     master_list[0] = '\0';
102     while ((row2 = mysql_fetch_row(rs2)))
103 sysadm 1.1 {
104 sysadm 1.8 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 sysadm 1.1 }
107     mysql_free_result(rs2);
108 sysadm 1.23 rs2 = NULL;
109 sysadm 1.1
110     p_section = section_list_find_by_sid(sid);
111    
112     if (p_section == NULL)
113     {
114 sysadm 1.8 p_section = section_list_create(sid, row[1], row[2], master_list);
115 sysadm 1.1 if (p_section == NULL)
116     {
117 sysadm 1.2 log_error("section_list_create() error: load new section sid = %d sname = %s\n", sid, row[1]);
118     ret = -4;
119 sysadm 1.1 break;
120     }
121    
122     // acquire rw lock
123     ret = section_list_rw_lock(p_section);
124     if (ret < 0)
125     {
126     break;
127     }
128     }
129     else
130     {
131     // acquire rw lock
132     ret = section_list_rw_lock(p_section);
133     if (ret < 0)
134     {
135     break;
136     }
137    
138     strncpy(p_section->sname, row[1], sizeof(p_section->sname) - 1);
139     p_section->sname[sizeof(p_section->sname) - 1] = '\0';
140     strncpy(p_section->stitle, row[1], sizeof(p_section->stitle) - 1);
141     p_section->stitle[sizeof(p_section->stitle) - 1] = '\0';
142 sysadm 1.8 strncpy(p_section->master_list, master_list, sizeof(p_section->master_list) - 1);
143     p_section->master_list[sizeof(p_section->master_list) - 1] = '\0';
144 sysadm 1.1 }
145    
146     p_section->class_id = atoi(row[3]);
147     p_section->read_user_level = atoi(row[4]);
148     p_section->write_user_level = atoi(row[5]);
149     p_section->enable = (int8_t)atoi(row[6]);
150    
151     // release rw lock
152     ret = section_list_rw_unlock(p_section);
153     if (ret < 0)
154     {
155     break;
156     }
157     }
158 sysadm 1.23
159     cleanup:
160 sysadm 1.1 mysql_free_result(rs);
161 sysadm 1.2 mysql_close(db);
162    
163     return ret;
164 sysadm 1.1 }
165 sysadm 1.3
166 sysadm 1.9 int append_articles_from_db(int32_t start_aid, int global_lock, int article_count_limit)
167 sysadm 1.3 {
168 sysadm 1.23 MYSQL *db = NULL;
169     MYSQL_RES *rs = NULL;
170 sysadm 1.3 MYSQL_ROW row;
171     char sql[SQL_BUFFER_LEN];
172     ARTICLE article;
173 sysadm 1.5 ARTICLE *p_topic;
174 sysadm 1.3 SECTION_LIST *p_section = NULL;
175     int32_t last_sid = 0;
176 sysadm 1.16 char sub_ip[IP_ADDR_LEN];
177 sysadm 1.9 int article_count = 0;
178 sysadm 1.3 int ret = 0;
179     int i;
180    
181     db = db_open();
182     if (db == NULL)
183     {
184     log_error("db_open() error: %s\n", mysql_error(db));
185 sysadm 1.23 ret = -1;
186     goto cleanup;
187 sysadm 1.3 }
188    
189     snprintf(sql, sizeof(sql),
190 sysadm 1.11 "SELECT bbs.AID, TID, SID, bbs.CID, UID, visible, excerption, ontop, `lock`, "
191     "transship, username, nickname, title, UNIX_TIMESTAMP(sub_dt) AS sub_dt, "
192 sysadm 1.16 "sub_ip, bbs_content.content "
193 sysadm 1.11 "FROM bbs INNER JOIN bbs_content ON bbs.CID = bbs_content.CID "
194     "WHERE bbs.AID >= %d ORDER BY bbs.AID LIMIT %d",
195 sysadm 1.9 start_aid, article_count_limit);
196 sysadm 1.3
197     if (mysql_query(db, sql) != 0)
198     {
199     log_error("Query article list error: %s\n", mysql_error(db));
200 sysadm 1.23 ret = -2;
201     goto cleanup;
202 sysadm 1.3 }
203 sysadm 1.11 if ((rs = mysql_use_result(db)) == NULL)
204 sysadm 1.3 {
205     log_error("Get article list data failed\n");
206 sysadm 1.23 ret = -2;
207     goto cleanup;
208 sysadm 1.3 }
209    
210     // acquire global lock
211 sysadm 1.23 if (global_lock && (ret = section_list_rw_lock(NULL)) < 0)
212 sysadm 1.3 {
213 sysadm 1.23 log_error("section_list_rw_lock(sid = 0) error\n");
214     goto cleanup;
215 sysadm 1.3 }
216    
217     while ((row = mysql_fetch_row(rs)))
218     {
219 sysadm 1.22 memset(&article, 0, sizeof(ARTICLE));
220 sysadm 1.3
221     // copy data of article
222     i = 0;
223    
224     article.aid = atoi(row[i++]);
225     article.tid = atoi(row[i++]);
226     article.sid = atoi(row[i++]);
227     article.cid = atoi(row[i++]);
228     article.uid = atoi(row[i++]);
229     article.visible = (int8_t)atoi(row[i++]);
230     article.excerption = (int8_t)atoi(row[i++]);
231     article.ontop = (int8_t)atoi(row[i++]);
232     article.lock = (int8_t)atoi(row[i++]);
233 sysadm 1.5 article.transship = (int8_t)atoi(row[i++]);
234 sysadm 1.3
235     strncpy(article.username, row[i++], sizeof(article.username) - 1);
236     article.username[sizeof(article.username) - 1] = '\0';
237     strncpy(article.nickname, row[i++], sizeof(article.nickname) - 1);
238     article.nickname[sizeof(article.nickname) - 1] = '\0';
239     strncpy(article.title, row[i++], sizeof(article.title) - 1);
240     article.title[sizeof(article.title) - 1] = '\0';
241    
242     article.sub_dt = atol(row[i++]);
243    
244 sysadm 1.16 strncpy(sub_ip, row[i++], sizeof(sub_ip) - 1);
245     sub_ip[sizeof(sub_ip) - 1] = '\0';
246     ip_mask(sub_ip, 1, '*');
247    
248 sysadm 1.3 // release lock of last section if different from current one
249     if (!global_lock && article.sid != last_sid && last_sid != 0)
250     {
251     if ((ret = section_list_rw_unlock(p_section)) < 0)
252     {
253 sysadm 1.11 log_error("section_list_rw_unlock(sid=%d) error\n", p_section->sid);
254 sysadm 1.3 break;
255     }
256     }
257    
258     if ((p_section = section_list_find_by_sid(article.sid)) == NULL)
259     {
260     log_error("section_list_find_by_sid(%d) error: unknown section, try reloading section config\n", article.sid);
261 sysadm 1.4 ret = ERR_UNKNOWN_SECTION; // Unknown section found
262 sysadm 1.3 break;
263     }
264    
265 sysadm 1.5 if (article.visible != 0 && article.tid != 0)
266     {
267     // Check if topic article is visible
268     p_topic = article_block_find_by_aid(article.tid);
269     if (p_topic == NULL || p_topic->visible == 0)
270     {
271     // log_error("Set article (aid = %d) as invisible due to invisible or non-existing topic head\n", article.aid);
272     article.tid = 0;
273     article.visible = 0;
274     }
275     }
276    
277 sysadm 1.3 // acquire lock of current section if different from last one
278     if (!global_lock && article.sid != last_sid)
279     {
280 sysadm 1.4 if ((ret = section_list_rw_lock(p_section)) < 0)
281 sysadm 1.3 {
282 sysadm 1.11 log_error("section_list_rw_lock(sid=0) error\n");
283 sysadm 1.3 break;
284     }
285     }
286    
287     // append article to section list
288     last_sid = article.sid;
289    
290     if (section_list_append_article(p_section, &article) < 0)
291     {
292 sysadm 1.11 log_error("section_list_append_article(sid=%d, aid=%d) error\n",
293 sysadm 1.3 p_section->sid, article.aid);
294     ret = -3;
295     break;
296     }
297 sysadm 1.9
298     article_count++;
299    
300 sysadm 1.16 if (article_cache_generate(VAR_ARTICLE_CACHE_DIR, &article, p_section, row[i++], sub_ip, 0) < 0)
301 sysadm 1.11 {
302     log_error("article_cache_generate(aid=%d, cid=%d) error\n", article.aid, article.cid);
303     ret = -4;
304     break;
305     }
306 sysadm 1.3 }
307    
308     // release lock of last section
309     if (!global_lock && last_sid != 0)
310     {
311     if ((ret = section_list_rw_unlock(p_section)) < 0)
312     {
313 sysadm 1.11 log_error("section_list_rw_unlock(sid=%d) error\n", p_section->sid);
314 sysadm 1.3 }
315     }
316    
317     // release global lock
318 sysadm 1.23 if (global_lock && (ret = section_list_rw_unlock(NULL)) < 0)
319 sysadm 1.3 {
320 sysadm 1.23 log_error("section_list_rw_unlock(sid=0) error\n");
321 sysadm 1.3 }
322    
323     cleanup:
324     mysql_free_result(rs);
325     mysql_close(db);
326    
327 sysadm 1.9 return (ret < 0 ? ret : article_count);
328 sysadm 1.3 }
329 sysadm 1.4
330 sysadm 1.5 int set_last_article_op_log_from_db(void)
331     {
332 sysadm 1.23 MYSQL *db = NULL;
333     MYSQL_RES *rs = NULL;
334 sysadm 1.5 MYSQL_ROW row;
335     char sql[SQL_BUFFER_LEN];
336 sysadm 1.23 int ret = 0;
337 sysadm 1.5
338     db = db_open();
339     if (db == NULL)
340     {
341     log_error("db_open() error: %s\n", mysql_error(db));
342 sysadm 1.23 ret = -1;
343     goto cleanup;
344 sysadm 1.5 }
345    
346     snprintf(sql, sizeof(sql),
347     "SELECT MID FROM bbs_article_op ORDER BY MID DESC LIMIT 1");
348    
349     if (mysql_query(db, sql) != 0)
350     {
351     log_error("Query article op error: %s\n", mysql_error(db));
352 sysadm 1.23 ret = -2;
353     goto cleanup;
354 sysadm 1.5 }
355     if ((rs = mysql_store_result(db)) == NULL)
356     {
357     log_error("Get article op data failed\n");
358 sysadm 1.23 ret = -2;
359     goto cleanup;
360 sysadm 1.5 }
361    
362     if ((row = mysql_fetch_row(rs)))
363     {
364     last_article_op_log_mid = atoi(row[0]);
365     }
366    
367 sysadm 1.23 cleanup:
368 sysadm 1.5 mysql_free_result(rs);
369     mysql_close(db);
370    
371 sysadm 1.23 return (ret < 0 ? ret : last_article_op_log_mid);
372 sysadm 1.5 }
373    
374 sysadm 1.9 int apply_article_op_log_from_db(int op_count_limit)
375 sysadm 1.5 {
376 sysadm 1.23 MYSQL *db = NULL;
377     MYSQL_RES *rs = NULL, *rs2 = NULL;
378 sysadm 1.5 MYSQL_ROW row, row2;
379     char sql[SQL_BUFFER_LEN];
380     ARTICLE *p_article;
381     SECTION_LIST *p_section = NULL;
382     SECTION_LIST *p_section_dest;
383 sysadm 1.16 char sub_ip[IP_ADDR_LEN];
384 sysadm 1.5 int32_t last_sid = 0;
385     int32_t sid_dest;
386 sysadm 1.9 int op_count = 0;
387 sysadm 1.5 int ret = 0;
388    
389     db = db_open();
390     if (db == NULL)
391     {
392     log_error("db_open() error: %s\n", mysql_error(db));
393 sysadm 1.23 ret = -1;
394     goto cleanup;
395 sysadm 1.5 }
396    
397     snprintf(sql, sizeof(sql),
398     "SELECT MID, AID, type FROM bbs_article_op "
399 sysadm 1.9 "WHERE MID > %d AND type NOT IN ('A') "
400     "ORDER BY MID LIMIT %d",
401     last_article_op_log_mid, op_count_limit);
402 sysadm 1.5
403     if (mysql_query(db, sql) != 0)
404     {
405     log_error("Query article log error: %s\n", mysql_error(db));
406 sysadm 1.23 ret = -2;
407     goto cleanup;
408 sysadm 1.5 }
409     if ((rs = mysql_store_result(db)) == NULL)
410     {
411     log_error("Get article log data failed\n");
412 sysadm 1.23 ret = -2;
413     goto cleanup;
414 sysadm 1.5 }
415    
416     while ((row = mysql_fetch_row(rs)))
417     {
418     p_article = article_block_find_by_aid(atoi(row[1]));
419     if (p_article == NULL) // related article has not been appended yet
420     {
421     ret = -2;
422     break;
423     }
424    
425     // release lock of last section if different from current one
426     if (p_article->sid != last_sid && last_sid != 0)
427     {
428     if ((ret = section_list_rw_unlock(p_section)) < 0)
429     {
430     log_error("section_list_rw_unlock(sid = %d) error\n", p_section->sid);
431     break;
432     }
433     }
434    
435     if ((p_section = section_list_find_by_sid(p_article->sid)) == NULL)
436     {
437     log_error("section_list_find_by_sid(%d) error: unknown section, try reloading section config\n", p_article->sid);
438     ret = ERR_UNKNOWN_SECTION; // Unknown section found
439     break;
440     }
441    
442     // acquire lock of current section if different from last one
443     if (p_article->sid != last_sid)
444     {
445     if ((ret = section_list_rw_lock(p_section)) < 0)
446     {
447     log_error("section_list_rw_lock(sid = 0) error\n");
448     break;
449     }
450     }
451    
452     last_sid = p_article->sid;
453    
454     switch (row[2][0])
455     {
456     case 'A': // Add article
457     log_error("Operation type=A should not be found\n");
458     break;
459     case 'D': // Delete article
460     case 'X': // Delete article by Admin
461 sysadm 1.20 if (section_list_set_article_visible(p_section, atoi(row[1]), 0) < 0)
462 sysadm 1.5 {
463 sysadm 1.20 log_error("section_list_set_article_visible(sid=%d, aid=%d, visible=0) error\n", p_section->sid, atoi(row[1]));
464     }
465     if (section_list_calculate_page(p_section, atoi(row[1])) < 0)
466     {
467     log_error("section_list_calculate_page(aid=%d) error\n", atoi(row[1]));
468 sysadm 1.5 }
469     break;
470     case 'S': // Restore article
471 sysadm 1.20 if (section_list_set_article_visible(p_section, atoi(row[1]), 1) < 0)
472     {
473     log_error("section_list_set_article_visible(sid=%d, aid=%d, visible=1) error\n", p_section->sid, atoi(row[1]));
474     }
475     if (section_list_calculate_page(p_section, atoi(row[1])) < 0)
476     {
477     log_error("section_list_calculate_page(aid=%d) error\n", atoi(row[1]));
478     }
479 sysadm 1.5 break;
480     case 'L': // Lock article
481     p_article->lock = 1;
482     break;
483     case 'U': // Unlock article
484     p_article->lock = 0;
485     break;
486     case 'M': // Modify article
487 sysadm 1.9 snprintf(sql, sizeof(sql),
488 sysadm 1.16 "SELECT bbs.CID, sub_ip, bbs_content.content "
489 sysadm 1.11 "FROM bbs INNER JOIN bbs_content ON bbs.CID = bbs_content.CID "
490     "WHERE bbs.AID = %d",
491 sysadm 1.9 p_article->aid);
492    
493     if (mysql_query(db, sql) != 0)
494     {
495     log_error("Query article error: %s\n", mysql_error(db));
496     ret = -3;
497     break;
498     }
499 sysadm 1.12 if ((rs2 = mysql_use_result(db)) == NULL)
500 sysadm 1.9 {
501     log_error("Get article data failed\n");
502     ret = -3;
503     break;
504     }
505     if ((row2 = mysql_fetch_row(rs2)))
506     {
507     p_article->cid = atoi(row2[0]);
508 sysadm 1.11
509 sysadm 1.16 strncpy(sub_ip, row2[1], sizeof(sub_ip) - 1);
510     sub_ip[sizeof(sub_ip) - 1] = '\0';
511     ip_mask(sub_ip, 1, '*');
512    
513     if (article_cache_generate(VAR_ARTICLE_CACHE_DIR, p_article, p_section, row2[2], sub_ip, 0) < 0)
514 sysadm 1.11 {
515     log_error("article_cache_generate(aid=%d, cid=%d) error\n", p_article->aid, p_article->cid);
516     ret = -4;
517     }
518 sysadm 1.9 }
519     else
520     {
521     p_article->cid = 0;
522     ret = -4;
523     }
524     mysql_free_result(rs2);
525 sysadm 1.23 rs2 = NULL;
526 sysadm 1.21
527     p_article->excerption = 0; // Set excerption = 0 implicitly in case of rare condition
528 sysadm 1.5 break;
529     case 'T': // Move article
530     snprintf(sql, sizeof(sql),
531     "SELECT SID FROM bbs WHERE AID = %d",
532     p_article->aid);
533    
534     if (mysql_query(db, sql) != 0)
535     {
536     log_error("Query article error: %s\n", mysql_error(db));
537     ret = -3;
538     break;
539     }
540     if ((rs2 = mysql_store_result(db)) == NULL)
541     {
542     log_error("Get article data failed\n");
543     ret = -3;
544     break;
545     }
546     if ((row2 = mysql_fetch_row(rs2)))
547     {
548     sid_dest = atoi(row2[0]);
549     }
550     else
551     {
552     sid_dest = 0;
553     ret = -4;
554     }
555     mysql_free_result(rs2);
556 sysadm 1.23 rs2 = NULL;
557 sysadm 1.5
558     if (sid_dest > 0 && sid_dest != p_article->sid)
559     {
560     p_section_dest = section_list_find_by_sid(sid_dest);
561     if (p_section_dest == NULL)
562     {
563     ret = ERR_UNKNOWN_SECTION;
564     break;
565     }
566 sysadm 1.6 // acquire lock of dest section
567     if ((ret = section_list_rw_lock(p_section_dest)) < 0)
568     {
569     log_error("section_list_rw_lock(sid = %d) error\n", p_section_dest);
570     break;
571     }
572 sysadm 1.5 // Move topic
573     if ((ret = section_list_move_topic(p_section, p_section_dest, p_article->aid)) < 0)
574     {
575 sysadm 1.6 log_error("section_list_move_topic(src_sid=%d, dest_sid=%d, aid=%d) error (%d), retry in the next loop\n",
576     p_section->sid, p_section_dest->sid, p_article->aid, ret);
577     }
578     // release lock of dest section
579     if (section_list_rw_unlock(p_section_dest) < 0)
580     {
581     log_error("section_list_rw_unlock(sid = %d) error\n", p_section_dest);
582     ret = -1;
583 sysadm 1.5 }
584     }
585     break;
586     case 'E': // Set article as excerption
587     p_article->excerption = 1;
588     break;
589     case 'O': // Unset article as excerption
590     p_article->excerption = 0;
591     break;
592     case 'F': // Set article on top
593     p_article->ontop = 1;
594     break;
595     case 'V': // Unset article on top
596     p_article->ontop = 0;
597     break;
598     case 'Z': // Set article as trnasship
599     p_article->transship = 1;
600     break;
601     default:
602     // log_error("Operation type=%s unknown, mid=%s\n", row[2], row[0]);
603     break;
604     }
605    
606     if (ret < 0)
607     {
608     break;
609     }
610    
611     // Update MID with last successfully proceeded article_op_log
612     last_article_op_log_mid = atoi(row[0]);
613 sysadm 1.9
614     op_count++;
615 sysadm 1.5 }
616    
617     // release lock of last section
618     if (last_sid != 0)
619     {
620     if ((ret = section_list_rw_unlock(p_section)) < 0)
621     {
622     log_error("section_list_rw_unlock(sid = %d) error\n", p_section->sid);
623     }
624     }
625    
626 sysadm 1.23 cleanup:
627     mysql_free_result(rs2);
628 sysadm 1.5 mysql_free_result(rs);
629     mysql_close(db);
630    
631 sysadm 1.9 return (ret < 0 ? ret : op_count);
632 sysadm 1.5 }
633    
634 sysadm 1.4 int section_list_loader_launch(void)
635     {
636     int pid;
637     int ret;
638     int32_t last_aid;
639     int article_count;
640     int load_count;
641 sysadm 1.5 int last_mid;
642 sysadm 1.4 int i;
643    
644     if (section_list_loader_pid != 0)
645     {
646     log_error("section_list_loader already running, pid = %d\n", section_list_loader_pid);
647     return -2;
648     }
649    
650     pid = fork();
651    
652     if (pid > 0) // Parent process
653     {
654     SYS_child_process_count++;
655     section_list_loader_pid = pid;
656 sysadm 1.18 log_common("Section list loader process (pid = %d) start\n", pid);
657 sysadm 1.4 return 0;
658     }
659     else if (pid < 0) // Error
660     {
661     log_error("fork() error (%d)\n", errno);
662     return -1;
663     }
664    
665     // Child process
666     SYS_child_process_count = 0;
667    
668     // Detach menu in shared memory
669     detach_menu_shm(p_bbs_menu);
670     free(p_bbs_menu);
671     p_bbs_menu = NULL;
672    
673     // Do section data loader periodically
674     while (!SYS_server_exit)
675     {
676     if (SYS_section_list_reload)
677     {
678     SYS_section_list_reload = 0;
679    
680     // Load section config
681     if (load_section_config_from_db() < 0)
682     {
683     log_error("load_section_config_from_db() error\n");
684     }
685     else
686     {
687     log_error("Reload section config successfully\n");
688     }
689     }
690    
691     // Load section articles
692     article_count = article_block_article_count();
693    
694 sysadm 1.9 do
695 sysadm 1.4 {
696 sysadm 1.9 last_aid = article_block_last_aid();
697 sysadm 1.4
698 sysadm 1.9 if ((ret = append_articles_from_db(last_aid + 1, 0, LOAD_ARTICLE_COUNT_LIMIT)) < 0)
699 sysadm 1.4 {
700 sysadm 1.9 log_error("append_articles_from_db(%d, 0, %d) error\n", last_aid + 1, LOAD_ARTICLE_COUNT_LIMIT);
701    
702     if (ret == ERR_UNKNOWN_SECTION)
703     {
704     SYS_section_list_reload = 1; // Force reload section_list
705     }
706 sysadm 1.4 }
707 sysadm 1.9 } while (ret == LOAD_ARTICLE_COUNT_LIMIT);
708 sysadm 1.5
709     load_count = article_block_article_count() - article_count;
710     if (load_count > 0)
711     {
712 sysadm 1.17 log_common("Incrementally load %d articles, last_aid = %d\n", load_count, article_block_last_aid());
713 sysadm 1.5 }
714    
715     if (SYS_section_list_reload)
716     {
717     continue;
718     }
719    
720     // Load article_op log
721     last_mid = last_article_op_log_mid;
722    
723 sysadm 1.9 do
724 sysadm 1.4 {
725 sysadm 1.9 if ((ret = apply_article_op_log_from_db(LOAD_ARTICLE_COUNT_LIMIT)) < 0)
726     {
727     log_error("apply_article_op_log_from_db() error\n");
728 sysadm 1.4
729 sysadm 1.9 if (ret == ERR_UNKNOWN_SECTION)
730     {
731     SYS_section_list_reload = 1; // Force reload section_list
732     }
733 sysadm 1.4 }
734 sysadm 1.9 } while (ret == LOAD_ARTICLE_COUNT_LIMIT);
735 sysadm 1.4
736 sysadm 1.5 if (last_article_op_log_mid > last_mid)
737     {
738 sysadm 1.17 log_common("Proceeded %d article logs, last_mid = %d\n", last_article_op_log_mid - last_mid, last_article_op_log_mid);
739 sysadm 1.5 }
740    
741     if (SYS_section_list_reload)
742     {
743     continue;
744     }
745    
746 sysadm 1.19 for (i = 0; i < BBS_section_list_load_interval && !SYS_server_exit && !SYS_section_list_reload; i++)
747 sysadm 1.5 {
748     sleep(1);
749 sysadm 1.4 }
750     }
751    
752     // Child process exit
753    
754     // Detach data pools shm
755     detach_section_list_shm();
756     detach_article_block_shm();
757     detach_trie_dict_shm();
758    
759 sysadm 1.17 log_common("Section list loader process exit normally\n");
760 sysadm 1.4 log_end();
761    
762     section_list_loader_pid = 0;
763    
764     _exit(0);
765    
766     return 0;
767     }
768    
769     int section_list_loader_reload(void)
770     {
771     if (section_list_loader_pid == 0)
772     {
773     log_error("section_list_loader not running\n");
774     return -2;
775     }
776    
777     if (kill(section_list_loader_pid, SIGHUP) < 0)
778     {
779     log_error("Send SIGTERM signal failed (%d)\n", errno);
780     return -1;
781     }
782    
783     return 0;
784     }
785 sysadm 1.7
786 sysadm 1.10 int query_section_articles(SECTION_LIST *p_section, int page_id, ARTICLE *p_articles[], int *p_article_count, int *p_page_count)
787 sysadm 1.7 {
788     ARTICLE *p_article;
789     ARTICLE *p_next_page_first_article;
790     int ret = 0;
791    
792 sysadm 1.10 if (p_section == NULL || p_articles == NULL || p_article_count == NULL || p_page_count == NULL)
793 sysadm 1.7 {
794     log_error("query_section_articles() NULL pointer error\n");
795     return -1;
796     }
797    
798     // acquire lock of section
799     if ((ret = section_list_rd_lock(p_section)) < 0)
800     {
801     log_error("section_list_rd_lock(sid = %d) error\n", p_section->sid);
802     return -2;
803     }
804    
805 sysadm 1.10 *p_page_count = p_section->page_count;
806    
807 sysadm 1.9 if (p_section->visible_article_count == 0)
808     {
809     *p_article_count = 0;
810     }
811     else if (page_id < 0 || page_id >= p_section->page_count)
812 sysadm 1.7 {
813 sysadm 1.8 log_error("Invalid page_id=%d, not in range [0, %d)\n", page_id, p_section->page_count);
814 sysadm 1.7 ret = -3;
815     }
816     else
817     {
818     ret = page_id;
819     p_article = p_section->p_page_first_article[page_id];
820     p_next_page_first_article =
821     (page_id == p_section->page_count - 1 ? p_section->p_article_head : p_section->p_page_first_article[page_id + 1]);
822     *p_article_count = 0;
823    
824     do
825     {
826     if (p_article->visible)
827     {
828     p_articles[*p_article_count] = p_article;
829     (*p_article_count)++;
830     }
831     p_article = p_article->p_next;
832     } while (p_article != p_next_page_first_article && (*p_article_count) <= BBS_article_limit_per_page);
833 sysadm 1.8
834     if (*p_article_count != (page_id < p_section->page_count - 1 ? BBS_article_limit_per_page : p_section->last_page_visible_article_count))
835     {
836     log_error("Inconsistent visible article count %d detected in section %d page %d\n", *p_article_count, p_section->sid, page_id);
837     }
838 sysadm 1.7 }
839    
840     // release lock of section
841 sysadm 1.8 if (section_list_rd_unlock(p_section) < 0)
842 sysadm 1.7 {
843     log_error("section_list_rd_unlock(sid = %d) error\n", p_section->sid);
844     ret = -2;
845     }
846    
847     return ret;
848     }
849 sysadm 1.14
850 sysadm 1.24 int locate_article_in_section(SECTION_LIST *p_section, const ARTICLE *p_article_cur, int direction, int step,
851 sysadm 1.15 int *p_page_id, int *p_visible_offset, int *p_article_count)
852 sysadm 1.14 {
853 sysadm 1.15 const ARTICLE *p_article = NULL;
854     ARTICLE *p_tmp;
855 sysadm 1.14 int32_t aid = 0;
856 sysadm 1.24 int page_id = 0;
857     int offset = 0;
858 sysadm 1.14 int ret = 0;
859 sysadm 1.15 int i;
860 sysadm 1.14
861 sysadm 1.15 if (p_section == NULL || p_article_cur == NULL || p_page_id == NULL || p_visible_offset == NULL || p_article_count == NULL)
862 sysadm 1.14 {
863     log_error("locate_article_in_section() NULL pointer error\n");
864     return -1;
865     }
866    
867     // acquire lock of section
868     if ((ret = section_list_rd_lock(p_section)) < 0)
869     {
870     log_error("section_list_rd_lock(sid = %d) error\n", p_section->sid);
871     return -2;
872     }
873    
874     if (direction == 0)
875     {
876     aid = p_article_cur->aid;
877     }
878     else if (direction == 1)
879     {
880 sysadm 1.24 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)
881 sysadm 1.14 {
882 sysadm 1.24 if (p_article->visible)
883     {
884     step--;
885     }
886     }
887 sysadm 1.15
888     aid = (p_article->aid > p_article_cur->aid ? p_article->aid : 0);
889 sysadm 1.14 }
890     else if (direction == -1)
891     {
892 sysadm 1.24 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)
893 sysadm 1.14 {
894 sysadm 1.24 if (p_article->visible)
895     {
896     step--;
897     }
898     }
899 sysadm 1.15
900     aid = (p_article->aid < p_article_cur->aid ? p_article->aid : 0);
901 sysadm 1.14 }
902    
903 sysadm 1.15 p_article = NULL;
904    
905 sysadm 1.14 if (aid > 0)
906     {
907 sysadm 1.15 p_article = section_list_find_article_with_offset(p_section, aid, &page_id, &offset, &p_tmp);
908 sysadm 1.14 if (p_article != NULL)
909     {
910 sysadm 1.15 *p_article_count = (page_id == p_section->page_count - 1 ? p_section->last_page_visible_article_count : BBS_article_limit_per_page);
911    
912     p_article = p_section->p_page_first_article[page_id];
913     for (i = 0; i < *p_article_count;)
914     {
915     if (p_article->visible)
916     {
917     if (p_article->aid == aid)
918     {
919     *p_page_id = page_id;
920     *p_visible_offset = i;
921     break;
922     }
923     i++;
924     if (i >= *p_article_count)
925     {
926     log_error("Visible article (aid=%d) not found in page %d\n", aid, page_id);
927     p_article = NULL;
928     }
929     }
930     p_article = p_article->p_next;
931     }
932 sysadm 1.14 }
933     }
934    
935     // release lock of section
936     if (section_list_rd_unlock(p_section) < 0)
937     {
938     log_error("section_list_rd_unlock(sid = %d) error\n", p_section->sid);
939     ret = -2;
940     }
941    
942     return (ret < 0 ? ret : (p_article == NULL ? 0 : 1));
943     }
webmaster@leafok.com
 ViewVC Help
Powered by ViewVC 1.3.0-beta1