1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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