1/* ipcrm.c - remove msg que, sem or shared memory
2 *
3 * Copyright 2014 Ashwini Kumar <ak.ashwini1981@gmail.com>
4 *
5 * See http://pubs.opengroup.org/onlinepubs/9699919799/utilities/ipcrm.html
6
7
8USE_IPCRM(NEWTOY(ipcrm, "m*M*s*S*q*Q*", TOYFLAG_USR|TOYFLAG_BIN))
9
10config IPCRM
11  bool "ipcrm"
12  default n
13  help
14    usage: ipcrm [ [-q msqid] [-m shmid] [-s semid]
15              [-Q msgkey] [-M shmkey] [-S semkey] ... ]
16
17    -mM Remove memory segment after last detach
18    -qQ Remove message queue
19    -sS Remove semaphore
20*/
21
22#define FOR_ipcrm
23#include "toys.h"
24#include <sys/ipc.h>
25#include <sys/shm.h>
26#include <sys/sem.h>
27#include <sys/msg.h>
28
29GLOBALS(
30  struct arg_list *qkey;
31  struct arg_list *qid;
32  struct arg_list *skey;
33  struct arg_list *sid;
34  struct arg_list *mkey;
35  struct arg_list *mid;
36)
37
38static void do_ipcrm(int key, int ipc, char *name)
39{
40  char *c;
41  int id, ret = 0;
42
43  id = strtol(name, &c, 0);
44  if (*c) {
45    error_msg("invalid number :%s", name);
46    return;
47  }
48
49  if (key) {
50    if (id == IPC_PRIVATE) {
51      error_msg("illegal key (%s)", name);
52      return;
53    }
54    id = ((ipc == 1)?shmget(id, 0, 0) :
55         (ipc == 2)? msgget(id, 0): semget(id, 0, 0));
56    if (id < 0) {
57      perror_msg("key (%s)", name);
58      return;
59    }
60  }
61
62  if (ipc == 1) ret = shmctl(id, IPC_RMID, NULL);
63  else if (ipc == 2) ret = msgctl(id, IPC_RMID, NULL);
64  else if (ipc == 3) ret = semctl(id, 0, IPC_RMID, NULL);
65
66  if (ret < 0) perror_msg("%s (%s)", ((key)? "key": "id"), name);
67}
68
69void ipcrm_main(void)
70{
71  ++toys.argv;
72  if (toys.optc && (!strcmp(*toys.argv, "shm") ||
73        !strcmp(*toys.argv, "sem") || !strcmp(*toys.argv, "msg"))) {
74    int t = (toys.argv[0][1] == 'h')? 1 : (toys.argv[0][1] == 's')? 2:3;
75
76    while (*(++toys.argv)) do_ipcrm(0, t, *toys.argv);
77  } else {
78    struct arg_list *tmp;
79
80    for (tmp = TT.mkey; tmp; tmp = tmp->next) do_ipcrm(1, 1, tmp->arg);
81    for (tmp = TT.mid; tmp; tmp = tmp->next) do_ipcrm(0, 1, tmp->arg);
82    for (tmp = TT.qkey; tmp; tmp = tmp->next) do_ipcrm(1, 2, tmp->arg);
83    for (tmp = TT.qid; tmp; tmp = tmp->next) do_ipcrm(0, 2, tmp->arg);
84    for (tmp = TT.skey; tmp; tmp = tmp->next) do_ipcrm(1, 3, tmp->arg);
85    for (tmp = TT.sid; tmp; tmp = tmp->next) do_ipcrm(0, 3, tmp->arg);
86    if (toys.optc) help_exit("unknown argument: %s", *toys.optargs);
87  }
88}
89