sem.c
上传用户:lgb322
上传日期:2013-02-24
资源大小:30529k
文件大小:26k
源码类别:

嵌入式Linux

开发平台:

Unix_Linux

  1. /*
  2.  * linux/ipc/sem.c
  3.  * Copyright (C) 1992 Krishna Balasubramanian
  4.  * Copyright (C) 1995 Eric Schenk, Bruno Haible
  5.  *
  6.  * IMPLEMENTATION NOTES ON CODE REWRITE (Eric Schenk, January 1995):
  7.  * This code underwent a massive rewrite in order to solve some problems
  8.  * with the original code. In particular the original code failed to
  9.  * wake up processes that were waiting for semval to go to 0 if the
  10.  * value went to 0 and was then incremented rapidly enough. In solving
  11.  * this problem I have also modified the implementation so that it
  12.  * processes pending operations in a FIFO manner, thus give a guarantee
  13.  * that processes waiting for a lock on the semaphore won't starve
  14.  * unless another locking process fails to unlock.
  15.  * In addition the following two changes in behavior have been introduced:
  16.  * - The original implementation of semop returned the value
  17.  *   last semaphore element examined on success. This does not
  18.  *   match the manual page specifications, and effectively
  19.  *   allows the user to read the semaphore even if they do not
  20.  *   have read permissions. The implementation now returns 0
  21.  *   on success as stated in the manual page.
  22.  * - There is some confusion over whether the set of undo adjustments
  23.  *   to be performed at exit should be done in an atomic manner.
  24.  *   That is, if we are attempting to decrement the semval should we queue
  25.  *   up and wait until we can do so legally?
  26.  *   The original implementation attempted to do this.
  27.  *   The current implementation does not do so. This is because I don't
  28.  *   think it is the right thing (TM) to do, and because I couldn't
  29.  *   see a clean way to get the old behavior with the new design.
  30.  *   The POSIX standard and SVID should be consulted to determine
  31.  *   what behavior is mandated.
  32.  *
  33.  * Further notes on refinement (Christoph Rohland, December 1998):
  34.  * - The POSIX standard says, that the undo adjustments simply should
  35.  *   redo. So the current implementation is o.K.
  36.  * - The previous code had two flaws:
  37.  *   1) It actively gave the semaphore to the next waiting process
  38.  *      sleeping on the semaphore. Since this process did not have the
  39.  *      cpu this led to many unnecessary context switches and bad
  40.  *      performance. Now we only check which process should be able to
  41.  *      get the semaphore and if this process wants to reduce some
  42.  *      semaphore value we simply wake it up without doing the
  43.  *      operation. So it has to try to get it later. Thus e.g. the
  44.  *      running process may reacquire the semaphore during the current
  45.  *      time slice. If it only waits for zero or increases the semaphore,
  46.  *      we do the operation in advance and wake it up.
  47.  *   2) It did not wake up all zero waiting processes. We try to do
  48.  *      better but only get the semops right which only wait for zero or
  49.  *      increase. If there are decrement operations in the operations
  50.  *      array we do the same as before.
  51.  *
  52.  * /proc/sysvipc/sem support (c) 1999 Dragos Acostachioaie <dragos@iname.com>
  53.  *
  54.  * SMP-threaded, sysctl's added
  55.  * (c) 1999 Manfred Spraul <manfreds@colorfullife.com>
  56.  * Enforced range limit on SEM_UNDO
  57.  * (c) 2001 Red Hat Inc <alan@redhat.com>
  58.  */
  59. #include <linux/config.h>
  60. #include <linux/slab.h>
  61. #include <linux/spinlock.h>
  62. #include <linux/init.h>
  63. #include <linux/proc_fs.h>
  64. #include <asm/uaccess.h>
  65. #include "util.h"
  66. #define sem_lock(id) ((struct sem_array*)ipc_lock(&sem_ids,id))
  67. #define sem_unlock(id) ipc_unlock(&sem_ids,id)
  68. #define sem_rmid(id) ((struct sem_array*)ipc_rmid(&sem_ids,id))
  69. #define sem_checkid(sma, semid)
  70. ipc_checkid(&sem_ids,&sma->sem_perm,semid)
  71. #define sem_buildid(id, seq) 
  72. ipc_buildid(&sem_ids, id, seq)
  73. static struct ipc_ids sem_ids;
  74. static int newary (key_t, int, int);
  75. static void freeary (int id);
  76. #ifdef CONFIG_PROC_FS
  77. static int sysvipc_sem_read_proc(char *buffer, char **start, off_t offset, int length, int *eof, void *data);
  78. #endif
  79. #define SEMMSL_FAST 256 /* 512 bytes on stack */
  80. #define SEMOPM_FAST 64  /* ~ 372 bytes on stack */
  81. /*
  82.  * linked list protection:
  83.  * sem_undo.id_next,
  84.  * sem_array.sem_pending{,last},
  85.  * sem_array.sem_undo: sem_lock() for read/write
  86.  * sem_undo.proc_next: only "current" is allowed to read/write that field.
  87.  *
  88.  */
  89. int sem_ctls[4] = {SEMMSL, SEMMNS, SEMOPM, SEMMNI};
  90. #define sc_semmsl (sem_ctls[0])
  91. #define sc_semmns (sem_ctls[1])
  92. #define sc_semopm (sem_ctls[2])
  93. #define sc_semmni (sem_ctls[3])
  94. static int used_sems;
  95. void __init sem_init (void)
  96. {
  97. used_sems = 0;
  98. ipc_init_ids(&sem_ids,sc_semmni);
  99. #ifdef CONFIG_PROC_FS
  100. create_proc_read_entry("sysvipc/sem", 0, 0, sysvipc_sem_read_proc, NULL);
  101. #endif
  102. }
  103. static int newary (key_t key, int nsems, int semflg)
  104. {
  105. int id;
  106. struct sem_array *sma;
  107. int size;
  108. if (!nsems)
  109. return -EINVAL;
  110. if (used_sems + nsems > sc_semmns)
  111. return -ENOSPC;
  112. size = sizeof (*sma) + nsems * sizeof (struct sem);
  113. sma = (struct sem_array *) ipc_alloc(size);
  114. if (!sma) {
  115. return -ENOMEM;
  116. }
  117. memset (sma, 0, size);
  118. id = ipc_addid(&sem_ids, &sma->sem_perm, sc_semmni);
  119. if(id == -1) {
  120. ipc_free(sma, size);
  121. return -ENOSPC;
  122. }
  123. used_sems += nsems;
  124. sma->sem_perm.mode = (semflg & S_IRWXUGO);
  125. sma->sem_perm.key = key;
  126. sma->sem_base = (struct sem *) &sma[1];
  127. /* sma->sem_pending = NULL; */
  128. sma->sem_pending_last = &sma->sem_pending;
  129. /* sma->undo = NULL; */
  130. sma->sem_nsems = nsems;
  131. sma->sem_ctime = CURRENT_TIME;
  132. sem_unlock(id);
  133. return sem_buildid(id, sma->sem_perm.seq);
  134. }
  135. asmlinkage long sys_semget (key_t key, int nsems, int semflg)
  136. {
  137. int id, err = -EINVAL;
  138. struct sem_array *sma;
  139. if (nsems < 0 || nsems > sc_semmsl)
  140. return -EINVAL;
  141. down(&sem_ids.sem);
  142. if (key == IPC_PRIVATE) {
  143. err = newary(key, nsems, semflg);
  144. } else if ((id = ipc_findkey(&sem_ids, key)) == -1) {  /* key not used */
  145. if (!(semflg & IPC_CREAT))
  146. err = -ENOENT;
  147. else
  148. err = newary(key, nsems, semflg);
  149. } else if (semflg & IPC_CREAT && semflg & IPC_EXCL) {
  150. err = -EEXIST;
  151. } else {
  152. sma = sem_lock(id);
  153. if(sma==NULL)
  154. BUG();
  155. if (nsems > sma->sem_nsems)
  156. err = -EINVAL;
  157. else if (ipcperms(&sma->sem_perm, semflg))
  158. err = -EACCES;
  159. else
  160. err = sem_buildid(id, sma->sem_perm.seq);
  161. sem_unlock(id);
  162. }
  163. up(&sem_ids.sem);
  164. return err;
  165. }
  166. /* doesn't acquire the sem_lock on error! */
  167. static int sem_revalidate(int semid, struct sem_array* sma, int nsems, short flg)
  168. {
  169. struct sem_array* smanew;
  170. smanew = sem_lock(semid);
  171. if(smanew==NULL)
  172. return -EIDRM;
  173. if(smanew != sma || sem_checkid(sma,semid) || sma->sem_nsems != nsems) {
  174. sem_unlock(semid);
  175. return -EIDRM;
  176. }
  177. if (ipcperms(&sma->sem_perm, flg)) {
  178. sem_unlock(semid);
  179. return -EACCES;
  180. }
  181. return 0;
  182. }
  183. /* Manage the doubly linked list sma->sem_pending as a FIFO:
  184.  * insert new queue elements at the tail sma->sem_pending_last.
  185.  */
  186. static inline void append_to_queue (struct sem_array * sma,
  187.     struct sem_queue * q)
  188. {
  189. *(q->prev = sma->sem_pending_last) = q;
  190. *(sma->sem_pending_last = &q->next) = NULL;
  191. }
  192. static inline void prepend_to_queue (struct sem_array * sma,
  193.      struct sem_queue * q)
  194. {
  195. q->next = sma->sem_pending;
  196. *(q->prev = &sma->sem_pending) = q;
  197. if (q->next)
  198. q->next->prev = &q->next;
  199. else /* sma->sem_pending_last == &sma->sem_pending */
  200. sma->sem_pending_last = &q->next;
  201. }
  202. static inline void remove_from_queue (struct sem_array * sma,
  203.       struct sem_queue * q)
  204. {
  205. *(q->prev) = q->next;
  206. if (q->next)
  207. q->next->prev = q->prev;
  208. else /* sma->sem_pending_last == &q->next */
  209. sma->sem_pending_last = q->prev;
  210. q->prev = NULL; /* mark as removed */
  211. }
  212. /*
  213.  * Determine whether a sequence of semaphore operations would succeed
  214.  * all at once. Return 0 if yes, 1 if need to sleep, else return error code.
  215.  */
  216. static int try_atomic_semop (struct sem_array * sma, struct sembuf * sops,
  217.      int nsops, struct sem_undo *un, int pid,
  218.      int do_undo)
  219. {
  220. int result, sem_op;
  221. struct sembuf *sop;
  222. struct sem * curr;
  223. for (sop = sops; sop < sops + nsops; sop++) {
  224. curr = sma->sem_base + sop->sem_num;
  225. sem_op = sop->sem_op;
  226. if (!sem_op && curr->semval)
  227. goto would_block;
  228. curr->sempid = (curr->sempid << 16) | pid;
  229. curr->semval += sem_op;
  230. if (sop->sem_flg & SEM_UNDO)
  231. {
  232. int undo = un->semadj[sop->sem_num] - sem_op;
  233. /*
  234.    * Exceeding the undo range is an error.
  235.  */
  236. if (undo < (-SEMAEM - 1) || undo > SEMAEM)
  237. {
  238. /* Don't undo the undo */
  239. sop->sem_flg &= ~SEM_UNDO;
  240. goto out_of_range;
  241. }
  242. un->semadj[sop->sem_num] = undo;
  243. }
  244. if (curr->semval < 0)
  245. goto would_block;
  246. if (curr->semval > SEMVMX)
  247. goto out_of_range;
  248. }
  249. if (do_undo)
  250. {
  251. sop--;
  252. result = 0;
  253. goto undo;
  254. }
  255. sma->sem_otime = CURRENT_TIME;
  256. return 0;
  257. out_of_range:
  258. result = -ERANGE;
  259. goto undo;
  260. would_block:
  261. if (sop->sem_flg & IPC_NOWAIT)
  262. result = -EAGAIN;
  263. else
  264. result = 1;
  265. undo:
  266. while (sop >= sops) {
  267. curr = sma->sem_base + sop->sem_num;
  268. curr->semval -= sop->sem_op;
  269. curr->sempid >>= 16;
  270. if (sop->sem_flg & SEM_UNDO)
  271. un->semadj[sop->sem_num] += sop->sem_op;
  272. sop--;
  273. }
  274. return result;
  275. }
  276. /* Go through the pending queue for the indicated semaphore
  277.  * looking for tasks that can be completed.
  278.  */
  279. static void update_queue (struct sem_array * sma)
  280. {
  281. int error;
  282. struct sem_queue * q;
  283. for (q = sma->sem_pending; q; q = q->next) {
  284. if (q->status == 1)
  285. continue; /* this one was woken up before */
  286. error = try_atomic_semop(sma, q->sops, q->nsops,
  287.  q->undo, q->pid, q->alter);
  288. /* Does q->sleeper still need to sleep? */
  289. if (error <= 0) {
  290. /* Found one, wake it up */
  291. wake_up_process(q->sleeper);
  292. if (error == 0 && q->alter) {
  293. /* if q-> alter let it self try */
  294. q->status = 1;
  295. return;
  296. }
  297. q->status = error;
  298. remove_from_queue(sma,q);
  299. }
  300. }
  301. }
  302. /* The following counts are associated to each semaphore:
  303.  *   semncnt        number of tasks waiting on semval being nonzero
  304.  *   semzcnt        number of tasks waiting on semval being zero
  305.  * This model assumes that a task waits on exactly one semaphore.
  306.  * Since semaphore operations are to be performed atomically, tasks actually
  307.  * wait on a whole sequence of semaphores simultaneously.
  308.  * The counts we return here are a rough approximation, but still
  309.  * warrant that semncnt+semzcnt>0 if the task is on the pending queue.
  310.  */
  311. static int count_semncnt (struct sem_array * sma, ushort semnum)
  312. {
  313. int semncnt;
  314. struct sem_queue * q;
  315. semncnt = 0;
  316. for (q = sma->sem_pending; q; q = q->next) {
  317. struct sembuf * sops = q->sops;
  318. int nsops = q->nsops;
  319. int i;
  320. for (i = 0; i < nsops; i++)
  321. if (sops[i].sem_num == semnum
  322.     && (sops[i].sem_op < 0)
  323.     && !(sops[i].sem_flg & IPC_NOWAIT))
  324. semncnt++;
  325. }
  326. return semncnt;
  327. }
  328. static int count_semzcnt (struct sem_array * sma, ushort semnum)
  329. {
  330. int semzcnt;
  331. struct sem_queue * q;
  332. semzcnt = 0;
  333. for (q = sma->sem_pending; q; q = q->next) {
  334. struct sembuf * sops = q->sops;
  335. int nsops = q->nsops;
  336. int i;
  337. for (i = 0; i < nsops; i++)
  338. if (sops[i].sem_num == semnum
  339.     && (sops[i].sem_op == 0)
  340.     && !(sops[i].sem_flg & IPC_NOWAIT))
  341. semzcnt++;
  342. }
  343. return semzcnt;
  344. }
  345. /* Free a semaphore set. */
  346. static void freeary (int id)
  347. {
  348. struct sem_array *sma;
  349. struct sem_undo *un;
  350. struct sem_queue *q;
  351. int size;
  352. sma = sem_rmid(id);
  353. /* Invalidate the existing undo structures for this semaphore set.
  354.  * (They will be freed without any further action in sem_exit()
  355.  * or during the next semop.)
  356.  */
  357. for (un = sma->undo; un; un = un->id_next)
  358. un->semid = -1;
  359. /* Wake up all pending processes and let them fail with EIDRM. */
  360. for (q = sma->sem_pending; q; q = q->next) {
  361. q->status = -EIDRM;
  362. q->prev = NULL;
  363. wake_up_process(q->sleeper); /* doesn't sleep */
  364. }
  365. sem_unlock(id);
  366. used_sems -= sma->sem_nsems;
  367. size = sizeof (*sma) + sma->sem_nsems * sizeof (struct sem);
  368. ipc_free(sma, size);
  369. }
  370. static unsigned long copy_semid_to_user(void *buf, struct semid64_ds *in, int version)
  371. {
  372. switch(version) {
  373. case IPC_64:
  374. return copy_to_user(buf, in, sizeof(*in));
  375. case IPC_OLD:
  376.     {
  377. struct semid_ds out;
  378. ipc64_perm_to_ipc_perm(&in->sem_perm, &out.sem_perm);
  379. out.sem_otime = in->sem_otime;
  380. out.sem_ctime = in->sem_ctime;
  381. out.sem_nsems = in->sem_nsems;
  382. return copy_to_user(buf, &out, sizeof(out));
  383.     }
  384. default:
  385. return -EINVAL;
  386. }
  387. }
  388. int semctl_nolock(int semid, int semnum, int cmd, int version, union semun arg)
  389. {
  390. int err = -EINVAL;
  391. switch(cmd) {
  392. case IPC_INFO:
  393. case SEM_INFO:
  394. {
  395. struct seminfo seminfo;
  396. int max_id;
  397. memset(&seminfo,0,sizeof(seminfo));
  398. seminfo.semmni = sc_semmni;
  399. seminfo.semmns = sc_semmns;
  400. seminfo.semmsl = sc_semmsl;
  401. seminfo.semopm = sc_semopm;
  402. seminfo.semvmx = SEMVMX;
  403. seminfo.semmnu = SEMMNU;
  404. seminfo.semmap = SEMMAP;
  405. seminfo.semume = SEMUME;
  406. down(&sem_ids.sem);
  407. if (cmd == SEM_INFO) {
  408. seminfo.semusz = sem_ids.in_use;
  409. seminfo.semaem = used_sems;
  410. } else {
  411. seminfo.semusz = SEMUSZ;
  412. seminfo.semaem = SEMAEM;
  413. }
  414. max_id = sem_ids.max_id;
  415. up(&sem_ids.sem);
  416. if (copy_to_user (arg.__buf, &seminfo, sizeof(struct seminfo))) 
  417. return -EFAULT;
  418. return (max_id < 0) ? 0: max_id;
  419. }
  420. case SEM_STAT:
  421. {
  422. struct sem_array *sma;
  423. struct semid64_ds tbuf;
  424. int id;
  425. if(semid >= sem_ids.size)
  426. return -EINVAL;
  427. memset(&tbuf,0,sizeof(tbuf));
  428. sma = sem_lock(semid);
  429. if(sma == NULL)
  430. return -EINVAL;
  431. err = -EACCES;
  432. if (ipcperms (&sma->sem_perm, S_IRUGO))
  433. goto out_unlock;
  434. id = sem_buildid(semid, sma->sem_perm.seq);
  435. kernel_to_ipc64_perm(&sma->sem_perm, &tbuf.sem_perm);
  436. tbuf.sem_otime  = sma->sem_otime;
  437. tbuf.sem_ctime  = sma->sem_ctime;
  438. tbuf.sem_nsems  = sma->sem_nsems;
  439. sem_unlock(semid);
  440. if (copy_semid_to_user (arg.buf, &tbuf, version))
  441. return -EFAULT;
  442. return id;
  443. }
  444. default:
  445. return -EINVAL;
  446. }
  447. return err;
  448. out_unlock:
  449. sem_unlock(semid);
  450. return err;
  451. }
  452. int semctl_main(int semid, int semnum, int cmd, int version, union semun arg)
  453. {
  454. struct sem_array *sma;
  455. struct sem* curr;
  456. int err;
  457. ushort fast_sem_io[SEMMSL_FAST];
  458. ushort* sem_io = fast_sem_io;
  459. int nsems;
  460. sma = sem_lock(semid);
  461. if(sma==NULL)
  462. return -EINVAL;
  463. nsems = sma->sem_nsems;
  464. err=-EIDRM;
  465. if (sem_checkid(sma,semid))
  466. goto out_unlock;
  467. err = -EACCES;
  468. if (ipcperms (&sma->sem_perm, (cmd==SETVAL||cmd==SETALL)?S_IWUGO:S_IRUGO))
  469. goto out_unlock;
  470. switch (cmd) {
  471. case GETALL:
  472. {
  473. ushort *array = arg.array;
  474. int i;
  475. if(nsems > SEMMSL_FAST) {
  476. sem_unlock(semid);
  477. sem_io = ipc_alloc(sizeof(ushort)*nsems);
  478. if(sem_io == NULL)
  479. return -ENOMEM;
  480. err = sem_revalidate(semid, sma, nsems, S_IRUGO);
  481. if(err)
  482. goto out_free;
  483. }
  484. for (i = 0; i < sma->sem_nsems; i++)
  485. sem_io[i] = sma->sem_base[i].semval;
  486. sem_unlock(semid);
  487. err = 0;
  488. if(copy_to_user(array, sem_io, nsems*sizeof(ushort)))
  489. err = -EFAULT;
  490. goto out_free;
  491. }
  492. case SETALL:
  493. {
  494. int i;
  495. struct sem_undo *un;
  496. sem_unlock(semid);
  497. if(nsems > SEMMSL_FAST) {
  498. sem_io = ipc_alloc(sizeof(ushort)*nsems);
  499. if(sem_io == NULL)
  500. return -ENOMEM;
  501. }
  502. if (copy_from_user (sem_io, arg.array, nsems*sizeof(ushort))) {
  503. err = -EFAULT;
  504. goto out_free;
  505. }
  506. for (i = 0; i < nsems; i++) {
  507. if (sem_io[i] > SEMVMX) {
  508. err = -ERANGE;
  509. goto out_free;
  510. }
  511. }
  512. err = sem_revalidate(semid, sma, nsems, S_IWUGO);
  513. if(err)
  514. goto out_free;
  515. for (i = 0; i < nsems; i++)
  516. sma->sem_base[i].semval = sem_io[i];
  517. for (un = sma->undo; un; un = un->id_next)
  518. for (i = 0; i < nsems; i++)
  519. un->semadj[i] = 0;
  520. sma->sem_ctime = CURRENT_TIME;
  521. /* maybe some queued-up processes were waiting for this */
  522. update_queue(sma);
  523. err = 0;
  524. goto out_unlock;
  525. }
  526. case IPC_STAT:
  527. {
  528. struct semid64_ds tbuf;
  529. memset(&tbuf,0,sizeof(tbuf));
  530. kernel_to_ipc64_perm(&sma->sem_perm, &tbuf.sem_perm);
  531. tbuf.sem_otime  = sma->sem_otime;
  532. tbuf.sem_ctime  = sma->sem_ctime;
  533. tbuf.sem_nsems  = sma->sem_nsems;
  534. sem_unlock(semid);
  535. if (copy_semid_to_user (arg.buf, &tbuf, version))
  536. return -EFAULT;
  537. return 0;
  538. }
  539. /* GETVAL, GETPID, GETNCTN, GETZCNT, SETVAL: fall-through */
  540. }
  541. err = -EINVAL;
  542. if(semnum < 0 || semnum >= nsems)
  543. goto out_unlock;
  544. curr = &sma->sem_base[semnum];
  545. switch (cmd) {
  546. case GETVAL:
  547. err = curr->semval;
  548. goto out_unlock;
  549. case GETPID:
  550. err = curr->sempid & 0xffff;
  551. goto out_unlock;
  552. case GETNCNT:
  553. err = count_semncnt(sma,semnum);
  554. goto out_unlock;
  555. case GETZCNT:
  556. err = count_semzcnt(sma,semnum);
  557. goto out_unlock;
  558. case SETVAL:
  559. {
  560. int val = arg.val;
  561. struct sem_undo *un;
  562. err = -ERANGE;
  563. if (val > SEMVMX || val < 0)
  564. goto out_unlock;
  565. for (un = sma->undo; un; un = un->id_next)
  566. un->semadj[semnum] = 0;
  567. curr->semval = val;
  568. sma->sem_ctime = CURRENT_TIME;
  569. /* maybe some queued-up processes were waiting for this */
  570. update_queue(sma);
  571. err = 0;
  572. goto out_unlock;
  573. }
  574. }
  575. out_unlock:
  576. sem_unlock(semid);
  577. out_free:
  578. if(sem_io != fast_sem_io)
  579. ipc_free(sem_io, sizeof(ushort)*nsems);
  580. return err;
  581. }
  582. struct sem_setbuf {
  583. uid_t uid;
  584. gid_t gid;
  585. mode_t mode;
  586. };
  587. static inline unsigned long copy_semid_from_user(struct sem_setbuf *out, void *buf, int version)
  588. {
  589. switch(version) {
  590. case IPC_64:
  591.     {
  592. struct semid64_ds tbuf;
  593. if(copy_from_user(&tbuf, buf, sizeof(tbuf)))
  594. return -EFAULT;
  595. out->uid = tbuf.sem_perm.uid;
  596. out->gid = tbuf.sem_perm.gid;
  597. out->mode = tbuf.sem_perm.mode;
  598. return 0;
  599.     }
  600. case IPC_OLD:
  601.     {
  602. struct semid_ds tbuf_old;
  603. if(copy_from_user(&tbuf_old, buf, sizeof(tbuf_old)))
  604. return -EFAULT;
  605. out->uid = tbuf_old.sem_perm.uid;
  606. out->gid = tbuf_old.sem_perm.gid;
  607. out->mode = tbuf_old.sem_perm.mode;
  608. return 0;
  609.     }
  610. default:
  611. return -EINVAL;
  612. }
  613. }
  614. int semctl_down(int semid, int semnum, int cmd, int version, union semun arg)
  615. {
  616. struct sem_array *sma;
  617. int err;
  618. struct sem_setbuf setbuf;
  619. struct kern_ipc_perm *ipcp;
  620. if(cmd == IPC_SET) {
  621. if(copy_semid_from_user (&setbuf, arg.buf, version))
  622. return -EFAULT;
  623. }
  624. sma = sem_lock(semid);
  625. if(sma==NULL)
  626. return -EINVAL;
  627. if (sem_checkid(sma,semid)) {
  628. err=-EIDRM;
  629. goto out_unlock;
  630. }
  631. ipcp = &sma->sem_perm;
  632. if (current->euid != ipcp->cuid && 
  633.     current->euid != ipcp->uid && !capable(CAP_SYS_ADMIN)) {
  634.      err=-EPERM;
  635. goto out_unlock;
  636. }
  637. switch(cmd){
  638. case IPC_RMID:
  639. freeary(semid);
  640. err = 0;
  641. break;
  642. case IPC_SET:
  643. ipcp->uid = setbuf.uid;
  644. ipcp->gid = setbuf.gid;
  645. ipcp->mode = (ipcp->mode & ~S_IRWXUGO)
  646. | (setbuf.mode & S_IRWXUGO);
  647. sma->sem_ctime = CURRENT_TIME;
  648. sem_unlock(semid);
  649. err = 0;
  650. break;
  651. default:
  652. sem_unlock(semid);
  653. err = -EINVAL;
  654. break;
  655. }
  656. return err;
  657. out_unlock:
  658. sem_unlock(semid);
  659. return err;
  660. }
  661. asmlinkage long sys_semctl (int semid, int semnum, int cmd, union semun arg)
  662. {
  663. int err = -EINVAL;
  664. int version;
  665. if (semid < 0)
  666. return -EINVAL;
  667. version = ipc_parse_version(&cmd);
  668. switch(cmd) {
  669. case IPC_INFO:
  670. case SEM_INFO:
  671. case SEM_STAT:
  672. err = semctl_nolock(semid,semnum,cmd,version,arg);
  673. return err;
  674. case GETALL:
  675. case GETVAL:
  676. case GETPID:
  677. case GETNCNT:
  678. case GETZCNT:
  679. case IPC_STAT:
  680. case SETVAL:
  681. case SETALL:
  682. err = semctl_main(semid,semnum,cmd,version,arg);
  683. return err;
  684. case IPC_RMID:
  685. case IPC_SET:
  686. down(&sem_ids.sem);
  687. err = semctl_down(semid,semnum,cmd,version,arg);
  688. up(&sem_ids.sem);
  689. return err;
  690. default:
  691. return -EINVAL;
  692. }
  693. }
  694. static struct sem_undo* freeundos(struct sem_array *sma, struct sem_undo* un)
  695. {
  696. struct sem_undo* u;
  697. struct sem_undo** up;
  698. for(up = &current->semundo;(u=*up);up=&u->proc_next) {
  699. if(un==u) {
  700. un=u->proc_next;
  701. *up=un;
  702. kfree(u);
  703. return un;
  704. }
  705. }
  706. printk ("freeundos undo list error id=%dn", un->semid);
  707. return un->proc_next;
  708. }
  709. /* returns without sem_lock on error! */
  710. static int alloc_undo(struct sem_array *sma, struct sem_undo** unp, int semid, int alter)
  711. {
  712. int size, nsems, error;
  713. struct sem_undo *un;
  714. nsems = sma->sem_nsems;
  715. size = sizeof(struct sem_undo) + sizeof(short)*nsems;
  716. sem_unlock(semid);
  717. un = (struct sem_undo *) kmalloc(size, GFP_KERNEL);
  718. if (!un)
  719. return -ENOMEM;
  720. memset(un, 0, size);
  721. error = sem_revalidate(semid, sma, nsems, alter ? S_IWUGO : S_IRUGO);
  722. if(error) {
  723. kfree(un);
  724. return error;
  725. }
  726. un->semadj = (short *) &un[1];
  727. un->semid = semid;
  728. un->proc_next = current->semundo;
  729. current->semundo = un;
  730. un->id_next = sma->undo;
  731. sma->undo = un;
  732. *unp = un;
  733. return 0;
  734. }
  735. asmlinkage long sys_semop (int semid, struct sembuf *tsops, unsigned nsops)
  736. {
  737. int error = -EINVAL;
  738. struct sem_array *sma;
  739. struct sembuf fast_sops[SEMOPM_FAST];
  740. struct sembuf* sops = fast_sops, *sop;
  741. struct sem_undo *un;
  742. int undos = 0, decrease = 0, alter = 0;
  743. struct sem_queue queue;
  744. if (nsops < 1 || semid < 0)
  745. return -EINVAL;
  746. if (nsops > sc_semopm)
  747. return -E2BIG;
  748. if(nsops > SEMOPM_FAST) {
  749. sops = kmalloc(sizeof(*sops)*nsops,GFP_KERNEL);
  750. if(sops==NULL)
  751. return -ENOMEM;
  752. }
  753. if (copy_from_user (sops, tsops, nsops * sizeof(*tsops))) {
  754. error=-EFAULT;
  755. goto out_free;
  756. }
  757. sma = sem_lock(semid);
  758. error=-EINVAL;
  759. if(sma==NULL)
  760. goto out_free;
  761. error = -EIDRM;
  762. if (sem_checkid(sma,semid))
  763. goto out_unlock_free;
  764. error = -EFBIG;
  765. for (sop = sops; sop < sops + nsops; sop++) {
  766. if (sop->sem_num >= sma->sem_nsems)
  767. goto out_unlock_free;
  768. if (sop->sem_flg & SEM_UNDO)
  769. undos++;
  770. if (sop->sem_op < 0)
  771. decrease = 1;
  772. if (sop->sem_op > 0)
  773. alter = 1;
  774. }
  775. alter |= decrease;
  776. error = -EACCES;
  777. if (ipcperms(&sma->sem_perm, alter ? S_IWUGO : S_IRUGO))
  778. goto out_unlock_free;
  779. if (undos) {
  780. /* Make sure we have an undo structure
  781.  * for this process and this semaphore set.
  782.  */
  783. un=current->semundo;
  784. while(un != NULL) {
  785. if(un->semid==semid)
  786. break;
  787. if(un->semid==-1)
  788. un=freeundos(sma,un);
  789.  else
  790. un=un->proc_next;
  791. }
  792. if (!un) {
  793. error = alloc_undo(sma,&un,semid,alter);
  794. if(error)
  795. goto out_free;
  796. }
  797. } else
  798. un = NULL;
  799. error = try_atomic_semop (sma, sops, nsops, un, current->pid, 0);
  800. if (error <= 0)
  801. goto update;
  802. /* We need to sleep on this operation, so we put the current
  803.  * task into the pending queue and go to sleep.
  804.  */
  805. queue.sma = sma;
  806. queue.sops = sops;
  807. queue.nsops = nsops;
  808. queue.undo = un;
  809. queue.pid = current->pid;
  810. queue.alter = decrease;
  811. queue.id = semid;
  812. if (alter)
  813. append_to_queue(sma ,&queue);
  814. else
  815. prepend_to_queue(sma ,&queue);
  816. current->semsleeping = &queue;
  817. for (;;) {
  818. struct sem_array* tmp;
  819. queue.status = -EINTR;
  820. queue.sleeper = current;
  821. current->state = TASK_INTERRUPTIBLE;
  822. sem_unlock(semid);
  823. schedule();
  824. tmp = sem_lock(semid);
  825. if(tmp==NULL) {
  826. if(queue.prev != NULL)
  827. BUG();
  828. current->semsleeping = NULL;
  829. error = -EIDRM;
  830. goto out_free;
  831. }
  832. /*
  833.  * If queue.status == 1 we where woken up and
  834.  * have to retry else we simply return.
  835.  * If an interrupt occurred we have to clean up the
  836.  * queue
  837.  *
  838.  */
  839. if (queue.status == 1)
  840. {
  841. error = try_atomic_semop (sma, sops, nsops, un,
  842.   current->pid,0);
  843. if (error <= 0) 
  844. break;
  845. } else {
  846. error = queue.status;
  847. if (queue.prev) /* got Interrupt */
  848. break;
  849. /* Everything done by update_queue */
  850. current->semsleeping = NULL;
  851. goto out_unlock_free;
  852. }
  853. }
  854. current->semsleeping = NULL;
  855. remove_from_queue(sma,&queue);
  856. update:
  857. if (alter)
  858. update_queue (sma);
  859. out_unlock_free:
  860. sem_unlock(semid);
  861. out_free:
  862. if(sops != fast_sops)
  863. kfree(sops);
  864. return error;
  865. }
  866. /*
  867.  * add semadj values to semaphores, free undo structures.
  868.  * undo structures are not freed when semaphore arrays are destroyed
  869.  * so some of them may be out of date.
  870.  * IMPLEMENTATION NOTE: There is some confusion over whether the
  871.  * set of adjustments that needs to be done should be done in an atomic
  872.  * manner or not. That is, if we are attempting to decrement the semval
  873.  * should we queue up and wait until we can do so legally?
  874.  * The original implementation attempted to do this (queue and wait).
  875.  * The current implementation does not do so. The POSIX standard
  876.  * and SVID should be consulted to determine what behavior is mandated.
  877.  */
  878. void sem_exit (void)
  879. {
  880. struct sem_queue *q;
  881. struct sem_undo *u, *un = NULL, **up, **unp;
  882. struct sem_array *sma;
  883. int nsems, i;
  884. /* If the current process was sleeping for a semaphore,
  885.  * remove it from the queue.
  886.  */
  887. if ((q = current->semsleeping)) {
  888. int semid = q->id;
  889. sma = sem_lock(semid);
  890. current->semsleeping = NULL;
  891. if (q->prev) {
  892. if(sma==NULL)
  893. BUG();
  894. remove_from_queue(q->sma,q);
  895. }
  896. if(sma!=NULL)
  897. sem_unlock(semid);
  898. }
  899. for (up = &current->semundo; (u = *up); *up = u->proc_next, kfree(u)) {
  900. int semid = u->semid;
  901. if(semid == -1)
  902. continue;
  903. sma = sem_lock(semid);
  904. if (sma == NULL)
  905. continue;
  906. if (u->semid == -1)
  907. goto next_entry;
  908. if (sem_checkid(sma,u->semid))
  909. goto next_entry;
  910. /* remove u from the sma->undo list */
  911. for (unp = &sma->undo; (un = *unp); unp = &un->id_next) {
  912. if (u == un)
  913. goto found;
  914. }
  915. printk ("sem_exit undo list error id=%dn", u->semid);
  916. goto next_entry;
  917. found:
  918. *unp = un->id_next;
  919. /* perform adjustments registered in u */
  920. nsems = sma->sem_nsems;
  921. for (i = 0; i < nsems; i++) {
  922. struct sem * sem = &sma->sem_base[i];
  923. sem->semval += u->semadj[i];
  924. if (sem->semval < 0)
  925. sem->semval = 0; /* shouldn't happen */
  926. sem->sempid = current->pid;
  927. }
  928. sma->sem_otime = CURRENT_TIME;
  929. /* maybe some queued-up processes were waiting for this */
  930. update_queue(sma);
  931. next_entry:
  932. sem_unlock(semid);
  933. }
  934. current->semundo = NULL;
  935. }
  936. #ifdef CONFIG_PROC_FS
  937. static int sysvipc_sem_read_proc(char *buffer, char **start, off_t offset, int length, int *eof, void *data)
  938. {
  939. off_t pos = 0;
  940. off_t begin = 0;
  941. int i, len = 0;
  942. len += sprintf(buffer, "       key      semid perms      nsems   uid   gid  cuid  cgid      otime      ctimen");
  943. down(&sem_ids.sem);
  944. for(i = 0; i <= sem_ids.max_id; i++) {
  945. struct sem_array *sma;
  946. sma = sem_lock(i);
  947. if(sma) {
  948. len += sprintf(buffer + len, "%10d %10d  %4o %10lu %5u %5u %5u %5u %10lu %10lun",
  949. sma->sem_perm.key,
  950. sem_buildid(i,sma->sem_perm.seq),
  951. sma->sem_perm.mode,
  952. sma->sem_nsems,
  953. sma->sem_perm.uid,
  954. sma->sem_perm.gid,
  955. sma->sem_perm.cuid,
  956. sma->sem_perm.cgid,
  957. sma->sem_otime,
  958. sma->sem_ctime);
  959. sem_unlock(i);
  960. pos += len;
  961. if(pos < offset) {
  962. len = 0;
  963.      begin = pos;
  964. }
  965. if(pos > offset + length)
  966. goto done;
  967. }
  968. }
  969. *eof = 1;
  970. done:
  971. up(&sem_ids.sem);
  972. *start = buffer + (offset - begin);
  973. len -= (offset - begin);
  974. if(len > length)
  975. len = length;
  976. if(len < 0)
  977. len = 0;
  978. return len;
  979. }
  980. #endif