gclib2/modules/IPC/ipc.cxx

   1 /*
   2  * (c) Oleg Puchinin 2006
   3  * graycardinalster@gmail.com
   4  *
   5  */
   6
   7 #include <sys/types.h>
   8 #include <sys/ipc.h>
   9 #include <sys/sem.h>
  10 #include <sys/msg.h>
  11
  12 /// \file ipc.c Межпроцессорные функции.
  13
  14 /*! \page IPCModule Межпроцессорное взаимодействие.
  15  * \see ipc.c
  16  */
  17
  18 inline void __semset (struct sembuf * action, int num, int op)
  19 {
  20         action->sem_flg = SEM_UNDO;
  21         action->sem_num = num;
  22         action->sem_op = op;
  23 }
  24
  25 enum {
  26         hold_read = 0,
  27         hold_write = 1,
  28 };
  29
  30 inline void __sem_zero (int sem, int num) // = 0
  31 {
  32         semctl (sem, num, SETVAL, 0);
  33 }
  34
  35 int __sem_try_key (int *KEY, int count) // = 1
  36 {
  37         int key = 9900;
  38         int id;
  39
  40         while (key < 10000) {
  41                 id = semget (key, count, (0600|IPC_CREAT));
  42                 if (id < 0) {
  43                         key++;
  44                         continue;
  45                 }
  46
  47                 if (KEY)
  48                         *KEY = key;
  49
  50                 return id;
  51         }
  52
  53         return -1;
  54 }
  55
  56 int __sem_init_one (int key, int count) // = 1
  57 {
  58         int id;
  59         id = semget (key, 1, (0600|IPC_CREAT));
  60         if (id > 0)
  61                 __sem_zero (id, 0);
  62         return id;
  63 }
  64
  65 /*! \brief Инициализировать семафор.
  66  * \param KEY - указатель на номер ключа. Если не задан - выбирается рандомный и записывается в *KEY.
  67  * \return Код (id) семафора, который используется во всех остальных функциях.
  68  */
  69 int sem_init (int * KEY)
  70 {
  71         int key = 9900;
  72         int id;
  73
  74         if (KEY && *KEY)
  75                 id = __sem_init_one (*KEY, 1);
  76         else
  77                 id = __sem_try_key (&key, 1);
  78
  79         if (id < 0)
  80                 return id;
  81
  82         __sem_zero (id, 0);
  83         if (KEY)
  84                 *KEY = key;
  85
  86         return id;
  87 }
  88
  89 /// Получить идентификатор существующего семафора.
  90 int sem_get (int key)
  91 {
  92         return semget (key, 1, 0600);
  93 }
  94
  95 /*! \brief Заблокировать семафор.
  96  * \param sem - код (id) семафора.
  97  */
  98 int down (int sem)
  99 {
 100         struct sembuf actions[2];
 101         __semset (&actions[0], 0, 0);
 102         __semset (&actions[1], 0, +1);
 103         return semop (sem, actions, 2);
 104 }
 105
 106 /*! \brief Разблокировать семафор.
 107  * \param sem - код (id) семафора.
 108  */
 109 int up (int sem)
 110 {
 111         struct sembuf actions[1];
 112         __semset (&actions[0], 0, -1);
 113         return semop (sem, actions, 1);
 114 }
 115
 116 /*! \brief Инициализировать семафор (чтение/запись).
 117  * \param KEY - указатель на номер ключа. Если не задан - выбирается рандомный и записывается в *KEY.
 118  * \return Код (id) семафора, который используется во всех остальных функциях.
 119  */
 120 int sem_init_rw (int * KEY)
 121 {
 122         int id;
 123
 124         if (KEY && *KEY)
 125                 id = __sem_init_one (*KEY, 2);
 126         else
 127                 id = __sem_try_key (KEY, 2);
 128
 129         if (id < 0)
 130                 return id;
 131
 132         __sem_zero (id, hold_read);
 133         __sem_zero (id, hold_write);
 134
 135         return id;
 136 }
 137
 138 /*! \brief Заблокировать семафор (чтение).
 139  * \param sem - код (id) семафора.
 140  */
 141 int down_read (int sem)
 142 {
 143         struct sembuf actions[2];
 144         __semset (&actions[0], hold_write, 0);
 145         __semset (&actions[1], hold_read, +1);
 146         return semop (sem, actions, 2);
 147 }
 148
 149 /*! \brief Разблокировать семафор (чтение).
 150  * \param sem - код (id) семафора.
 151  */
 152 int up_read (int sem)
 153 {
 154         struct sembuf actions[1];
 155         __semset (&actions[0], hold_read, -1);
 156         return semop (sem, actions, 1);
 157 }
 158
 159 /*! \brief Заблокировать семафор (запись).
 160  * \param sem - код (id) семафора.
 161  */
 162 int down_write (int sem)
 163 {
 164         struct sembuf actions[3];
 165         __semset (&actions[0], hold_read, 0);
 166         __semset (&actions[1], hold_write, 0);
 167         __semset (&actions[2], hold_write, +1);
 168         return semop (sem, actions, 3);
 169 }
 170
 171 /*! \brief Разлокировать семафор (запись).
 172  * \param sem - код (id) семафора.
 173  */
 174 int up_write (int sem)
 175 {
 176         struct sembuf actions[2];
 177         __semset (&actions[0], hold_write, -1);
 178         return semop (sem, actions, 1);
 179 }
 180
 181 int __msg_try_key (int *KEY)
 182 {
 183         int key = 9900;
 184         int id;
 185
 186         if (KEY && *KEY != 0) {
 187                 id = msgget (key, (0600|IPC_CREAT|IPC_EXCL));
 188                 return id;
 189         }
 190
 191         while (key < 10000) {
 192                 id = msgget (key, (0600|IPC_CREAT|IPC_EXCL));
 193                 if (id < 0) {
 194                         ++key;
 195                         continue;
 196                 }
 197
 198                 if (KEY)
 199                         *KEY = key;
 200
 201                 return id;
 202         }
 203
 204         return -1;
 205 }
 206
 207 /*! \brief Инициализировать пул сообщений.
 208  * \param KEY - указатель на номер ключа. Если не задан - выбирается рандомный и записывается в *KEY.
 209  * \return Код (id) пула сообщений, который используется во всех остальных функциях.
 210  */
 211 int msg_init (int *KEY)
 212 {
 213         return __msg_try_key (KEY);
 214 }
 215