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

Diff of /lbbs/src/section_list.c

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

Revision 1.14 by sysadm, Sat May 24 03:32:32 2025 UTC Revision 1.17 by sysadm, Sat May 24 13:52:44 2025 UTC
# Line 14  Line 14 
14   *                                                                         *   *                                                                         *
15   ***************************************************************************/   ***************************************************************************/
16    
17    #define _GNU_SOURCE
18    
19  #include "section_list.h"  #include "section_list.h"
20  #include "log.h"  #include "log.h"
21  #include "trie_dict.h"  #include "trie_dict.h"
# Line 24  Line 26 
26  #include <stdlib.h>  #include <stdlib.h>
27  #include <errno.h>  #include <errno.h>
28  #include <sys/param.h>  #include <sys/param.h>
29    #include <sys/sem.h>
30  #include <sys/shm.h>  #include <sys/shm.h>
31  #include <sys/ipc.h>  #include <sys/ipc.h>
32    
33    #ifdef _SEM_SEMUN_UNDEFINED
34    union semun
35    {
36            int val;                           /* Value for SETVAL */
37            struct semid_ds *buf;  /* Buffer for IPC_STAT, IPC_SET */
38            unsigned short *array; /* Array for GETALL, SETALL */
39            struct seminfo *__buf; /* Buffer for IPC_INFO
40                                                              (Linux-specific) */
41    };
42    #endif // #ifdef _SEM_SEMUN_UNDEFINED
43    
44  #define ARTICLE_BLOCK_PER_SHM 400                 // sizeof(ARTICLE_BLOCK) * ARTICLE_BLOCK_PER_SHM is the size of each shm segment to allocate  #define ARTICLE_BLOCK_PER_SHM 400                 // sizeof(ARTICLE_BLOCK) * ARTICLE_BLOCK_PER_SHM is the size of each shm segment to allocate
45  #define ARTICLE_BLOCK_SHM_COUNT_LIMIT 256 // limited by length (8-bit) of proj_id in ftok(path, proj_id)  #define ARTICLE_BLOCK_SHM_COUNT_LIMIT 256 // limited by length (8-bit) of proj_id in ftok(path, proj_id)
46  #define ARTICLE_BLOCK_PER_POOL (ARTICLE_BLOCK_PER_SHM * ARTICLE_BLOCK_SHM_COUNT_LIMIT)  #define ARTICLE_BLOCK_PER_POOL (ARTICLE_BLOCK_PER_SHM * ARTICLE_BLOCK_SHM_COUNT_LIMIT)
# Line 63  typedef struct article_block_pool_t ARTI Line 77  typedef struct article_block_pool_t ARTI
77  static ARTICLE_BLOCK_POOL *p_article_block_pool = NULL;  static ARTICLE_BLOCK_POOL *p_article_block_pool = NULL;
78    
79  static int section_list_pool_shmid;  static int section_list_pool_shmid;
80    static int section_list_pool_semid;
81  static SECTION_LIST *p_section_list_pool = NULL;  static SECTION_LIST *p_section_list_pool = NULL;
82  static int section_list_count = 0;  static int section_list_count = 0;
83  static TRIE_NODE *p_trie_dict_section_by_name = NULL;  static TRIE_NODE *p_trie_dict_section_by_name = NULL;
# Line 317  ARTICLE *article_block_find_by_index(int Line 332  ARTICLE *article_block_find_by_index(int
332    
333  extern int section_list_pool_init(const char *filename)  extern int section_list_pool_init(const char *filename)
334  {  {
335            int semid;
336          int shmid;          int shmid;
337          int proj_id;          int proj_id;
338          key_t key;          key_t key;
339          size_t size;          size_t size;
340          void *p_shm;          void *p_shm;
341            union semun arg;
342            int i;
343    
344          if (p_section_list_pool == NULL || p_trie_dict_section_by_name == NULL || p_trie_dict_section_by_sid == NULL)          if (p_section_list_pool == NULL || p_trie_dict_section_by_name == NULL || p_trie_dict_section_by_sid == NULL)
345          {          {
# Line 343  extern int section_list_pool_init(const Line 361  extern int section_list_pool_init(const
361                  return -3;                  return -3;
362          }          }
363    
364            size = 2 * (BBS_max_section + 1); // r_sem and w_sem per section, the last pair for all sections
365            semid = semget(key, (int)size, IPC_CREAT | IPC_EXCL | 0600);
366            if (semid == -1)
367            {
368                    log_error("semget(section_list_pool_sem, size = %d) error (%d)\n", size, errno);
369                    return -3;
370            }
371    
372            // Initialize sem value to 0
373            arg.val = 0;
374            for (i = 0; i < size; i++)
375            {
376                    if (semctl(semid, i, SETVAL, arg) == -1)
377                    {
378                            log_error("semctl(section_list_pool_sem, SETVAL) error (%d)\n", errno);
379                            return -3;
380                    }
381            }
382    
383            section_list_pool_semid = semid;
384    
385          size = sizeof(shmid) + sizeof(SECTION_LIST) * BBS_max_section;          size = sizeof(shmid) + sizeof(SECTION_LIST) * BBS_max_section;
386          shmid = shmget(key, size, IPC_CREAT | IPC_EXCL | 0600);          shmid = shmget(key, size, IPC_CREAT | IPC_EXCL | 0600);
387          if (shmid == -1)          if (shmid == -1)
388          {          {
389                  log_error("shmget(section_list_pool, size = %d) error (%d)\n", size, errno);                  log_error("shmget(section_list_pool_shm, size = %d) error (%d)\n", size, errno);
390                  return -3;                  return -3;
391          }          }
392          p_shm = shmat(shmid, NULL, 0);          p_shm = shmat(shmid, NULL, 0);
# Line 483  void section_list_pool_cleanup(void) Line 522  void section_list_pool_cleanup(void)
522                  {                  {
523                          log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", section_list_pool_shmid, errno);                          log_error("shmctl(shmid = %d, IPC_RMID) error (%d)\n", section_list_pool_shmid, errno);
524                  }                  }
525    
526                    if (semctl(section_list_pool_semid, 0, IPC_RMID) == -1)
527                    {
528                            log_error("semctl(semid = %d, IPC_RMID) error (%d)\n", section_list_pool_semid, errno);
529                    }
530          }          }
531    
532          section_list_count = 0;          section_list_count = 0;
# Line 952  int article_count_of_topic(int32_t aid) Line 996  int article_count_of_topic(int32_t aid)
996    
997          do          do
998          {          {
999                    if (p_article->tid != 0 && p_article->tid != aid)
1000                    {
1001                            log_error("article_count_of_topic(%d) error: article %d not linked to the topic\n", aid, p_article->aid);
1002                            break;
1003                    }
1004    
1005                  article_count++;                  article_count++;
1006                  p_article = p_article->p_topic_next;                  p_article = p_article->p_topic_next;
1007          } while (p_article->aid != aid);          } while (p_article->aid != aid);
# Line 1150  int section_list_move_topic(SECTION_LIST Line 1200  int section_list_move_topic(SECTION_LIST
1200    
1201          return move_article_count;          return move_article_count;
1202  }  }
1203    
1204    int get_section_index(SECTION_LIST *p_section)
1205    {
1206            int index;
1207    
1208            if (p_section_list_pool == NULL)
1209            {
1210                    log_error("get_section_index() error: uninitialized\n");
1211                    return -1;
1212            }
1213    
1214            if (p_section == NULL)
1215            {
1216                    index = BBS_max_section;
1217            }
1218            else
1219            {
1220                    index = (int)(p_section - p_section_list_pool);
1221                    if (index < 0 || index >= BBS_max_section)
1222                    {
1223                            log_error("get_section_index(%d) error: index out of range\n", index);
1224                            return -2;
1225                    }
1226            }
1227    
1228            return index;
1229    }
1230    
1231    int section_list_try_rd_lock(SECTION_LIST *p_section, int wait_sec)
1232    {
1233            int index;
1234            struct sembuf sops[4];
1235            struct timespec timeout;
1236            int ret;
1237    
1238            index = get_section_index(p_section);
1239            if (index < 0)
1240            {
1241                    return -2;
1242            }
1243    
1244            sops[0].sem_num = (unsigned short)(index * 2 + 1); // w_sem of section index
1245            sops[0].sem_op = 0;                                                                // wait until unlocked
1246            sops[0].sem_flg = 0;
1247    
1248            sops[1].sem_num = (unsigned short)(index * 2); // r_sem of section index
1249            sops[1].sem_op = 1;                                                        // lock
1250            sops[1].sem_flg = SEM_UNDO;                                        // undo on terminate
1251    
1252            // Read lock on any specific section will also acquire single read lock on "all section"
1253            // so that write lock on all section only need to acquire single write on on "all section"
1254            // rather than to acquire multiple write locks on all the available sections.
1255            if (index == BBS_max_section)
1256            {
1257                    sops[2].sem_num = BBS_max_section * 2 + 1; // w_sem of all section
1258                    sops[2].sem_op = 0;                                                // wait until unlocked
1259                    sops[2].sem_flg = 0;
1260    
1261                    sops[3].sem_num = BBS_max_section * 2; // r_sem of all section
1262                    sops[3].sem_op = 1;                                        // lock
1263                    sops[3].sem_flg = SEM_UNDO;                        // undo on terminate
1264            }
1265    
1266            timeout.tv_sec = wait_sec;
1267            timeout.tv_nsec = 0;
1268    
1269            ret = semtimedop(section_list_pool_semid, sops, (index == BBS_max_section ? 4 : 2), &timeout);
1270            if (ret == -1 && errno != EAGAIN && errno != EINTR)
1271            {
1272                    log_error("semtimedop(index = %d, lock read) error %d\n", index, errno);
1273            }
1274    
1275            return ret;
1276    }
1277    
1278    int section_list_try_rw_lock(SECTION_LIST *p_section, int wait_sec)
1279    {
1280            int index;
1281            struct sembuf sops[3];
1282            struct timespec timeout;
1283            int ret;
1284    
1285            index = get_section_index(p_section);
1286            if (index < 0)
1287            {
1288                    return -2;
1289            }
1290    
1291            sops[0].sem_num = (unsigned short)(index * 2 + 1); // w_sem of section index
1292            sops[0].sem_op = 0;                                                                // wait until unlocked
1293            sops[0].sem_flg = 0;
1294    
1295            sops[1].sem_num = (unsigned short)(index * 2 + 1); // w_sem of section index
1296            sops[1].sem_op = 1;                                                                // lock
1297            sops[1].sem_flg = SEM_UNDO;                                                // undo on terminate
1298    
1299            sops[2].sem_num = (unsigned short)(index * 2); // r_sem of section index
1300            sops[2].sem_op = 0;                                                        // wait until unlocked
1301            sops[2].sem_flg = 0;
1302    
1303            timeout.tv_sec = wait_sec;
1304            timeout.tv_nsec = 0;
1305    
1306            ret = semtimedop(section_list_pool_semid, sops, 3, &timeout);
1307            if (ret == -1 && errno != EAGAIN && errno != EINTR)
1308            {
1309                    log_error("semtimedop(index = %d, lock write) error %d\n", index, errno);
1310            }
1311    
1312            return ret;
1313    }
1314    
1315    int section_list_rd_unlock(SECTION_LIST *p_section)
1316    {
1317            int index;
1318            struct sembuf sops[2];
1319            int ret;
1320    
1321            index = get_section_index(p_section);
1322            if (index < 0)
1323            {
1324                    return -2;
1325            }
1326    
1327            sops[0].sem_num = (unsigned short)(index * 2); // r_sem of section index
1328            sops[0].sem_op = -1;                                               // unlock
1329            sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO;           // no wait
1330    
1331            // The same reason as section_list_try_rd_lock()
1332            if (index == BBS_max_section)
1333            {
1334                    sops[1].sem_num = BBS_max_section * 2;   // r_sem of all section
1335                    sops[1].sem_op = -1;                                     // unlock
1336                    sops[1].sem_flg = IPC_NOWAIT | SEM_UNDO; // no wait
1337            }
1338    
1339            ret = semop(section_list_pool_semid, sops, (index == BBS_max_section ? 2 : 1));
1340            if (ret == -1 && errno != EAGAIN && errno != EINTR)
1341            {
1342                    log_error("semop(index = %d, unlock read) error %d\n", index, errno);
1343            }
1344    
1345            return ret;
1346    }
1347    
1348    int section_list_rw_unlock(SECTION_LIST *p_section)
1349    {
1350            int index;
1351            struct sembuf sops[1];
1352            int ret;
1353    
1354            index = get_section_index(p_section);
1355            if (index < 0)
1356            {
1357                    return -2;
1358            }
1359    
1360            sops[0].sem_num = (unsigned short)(index * 2 + 1); // w_sem of section index
1361            sops[0].sem_op = -1;                                                       // unlock
1362            sops[0].sem_flg = IPC_NOWAIT | SEM_UNDO;                   // no wait
1363    
1364            ret = semop(section_list_pool_semid, sops, 1);
1365            if (ret == -1 && errno != EAGAIN && errno != EINTR)
1366            {
1367                    log_error("semop(index = %d, unlock write) error %d\n", index, errno);
1368            }
1369    
1370            return ret;
1371    }

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