diff options
Diffstat (limited to 'util-linux/ipcrm.c')
-rw-r--r-- | util-linux/ipcrm.c | 217 |
1 files changed, 217 insertions, 0 deletions
diff --git a/util-linux/ipcrm.c b/util-linux/ipcrm.c new file mode 100644 index 000000000..507e58fe3 --- /dev/null +++ b/util-linux/ipcrm.c | |||
@@ -0,0 +1,217 @@ | |||
1 | /* vi: set sw=4 ts=4: */ | ||
2 | /* | ||
3 | * ipcrm.c - utility to allow removal of IPC objects and data structures. | ||
4 | * | ||
5 | * 01 Sept 2004 - Rodney Radford <rradford@mindspring.com> | ||
6 | * Adapted for busybox from util-linux-2.12a. | ||
7 | * | ||
8 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. | ||
9 | */ | ||
10 | |||
11 | #include "busybox.h" | ||
12 | |||
13 | /* X/OPEN tells us to use <sys/{types,ipc,sem}.h> for semctl() */ | ||
14 | /* X/OPEN tells us to use <sys/{types,ipc,msg}.h> for msgctl() */ | ||
15 | #include <sys/ipc.h> | ||
16 | #include <sys/shm.h> | ||
17 | #include <sys/msg.h> | ||
18 | #include <sys/sem.h> | ||
19 | |||
20 | #if defined (__GNU_LIBRARY__) && !defined(_SEM_SEMUN_UNDEFINED) | ||
21 | /* union semun is defined by including <sys/sem.h> */ | ||
22 | #else | ||
23 | /* according to X/OPEN we have to define it ourselves */ | ||
24 | union semun { | ||
25 | int val; | ||
26 | struct semid_ds *buf; | ||
27 | unsigned short int *array; | ||
28 | struct seminfo *__buf; | ||
29 | }; | ||
30 | #endif | ||
31 | |||
32 | #ifndef CONFIG_IPCRM_DROP_LEGACY | ||
33 | |||
34 | typedef enum type_id { | ||
35 | SHM, | ||
36 | SEM, | ||
37 | MSG | ||
38 | } type_id; | ||
39 | |||
40 | static int remove_ids(type_id type, int argc, char **argv) | ||
41 | { | ||
42 | unsigned long id; | ||
43 | int ret = 0; /* silence gcc */ | ||
44 | int nb_errors = 0; | ||
45 | union semun arg; | ||
46 | |||
47 | arg.val = 0; | ||
48 | |||
49 | while (argc) { | ||
50 | id = bb_strtoul(argv[0], NULL, 10); | ||
51 | if (errno || id > INT_MAX) { | ||
52 | bb_error_msg("invalid id: %s", argv[0]); | ||
53 | nb_errors++; | ||
54 | } else { | ||
55 | if (type == SEM) | ||
56 | ret = semctl(id, 0, IPC_RMID, arg); | ||
57 | else if (type == MSG) | ||
58 | ret = msgctl(id, IPC_RMID, NULL); | ||
59 | else if (type == SHM) | ||
60 | ret = shmctl(id, IPC_RMID, NULL); | ||
61 | |||
62 | if (ret) { | ||
63 | bb_perror_msg("cannot remove id %s", argv[0]); | ||
64 | nb_errors++; | ||
65 | } | ||
66 | } | ||
67 | argc--; | ||
68 | argv++; | ||
69 | } | ||
70 | |||
71 | return nb_errors; | ||
72 | } | ||
73 | #endif /* #ifndef CONFIG_IPCRM_DROP_LEGACY */ | ||
74 | |||
75 | |||
76 | int ipcrm_main(int argc, char **argv) | ||
77 | { | ||
78 | int c; | ||
79 | int error = 0; | ||
80 | |||
81 | /* if the command is executed without parameters, do nothing */ | ||
82 | if (argc == 1) | ||
83 | return 0; | ||
84 | #ifndef CONFIG_IPCRM_DROP_LEGACY | ||
85 | /* check to see if the command is being invoked in the old way if so | ||
86 | then run the old code. Valid commands are msg, shm, sem. */ | ||
87 | { | ||
88 | type_id what = 0; /* silence gcc */ | ||
89 | char w; | ||
90 | |||
91 | w=argv[1][0]; | ||
92 | if ( ((w == 'm' && argv[1][1] == 's' && argv[1][2] == 'g') | ||
93 | || (argv[1][0] == 's' | ||
94 | && ((w=argv[1][1]) == 'h' || w == 'e') | ||
95 | && argv[1][2] == 'm') | ||
96 | ) && argv[1][3] == '\0' | ||
97 | ) { | ||
98 | |||
99 | if (argc < 3) | ||
100 | bb_show_usage(); | ||
101 | |||
102 | if (w == 'h') | ||
103 | what = SHM; | ||
104 | else if (w == 'm') | ||
105 | what = MSG; | ||
106 | else if (w == 'e') | ||
107 | what = SEM; | ||
108 | |||
109 | if (remove_ids(what, argc-2, &argv[2])) | ||
110 | fflush_stdout_and_exit(1); | ||
111 | printf("resource(s) deleted\n"); | ||
112 | return 0; | ||
113 | } | ||
114 | } | ||
115 | #endif /* #ifndef CONFIG_IPCRM_DROP_LEGACY */ | ||
116 | |||
117 | /* process new syntax to conform with SYSV ipcrm */ | ||
118 | while ((c = getopt(argc, argv, "q:m:s:Q:M:S:h?")) != -1) { | ||
119 | int result; | ||
120 | int id = 0; | ||
121 | int iskey = (isupper)(c); | ||
122 | |||
123 | /* needed to delete semaphores */ | ||
124 | union semun arg; | ||
125 | |||
126 | arg.val = 0; | ||
127 | |||
128 | if ((c == '?') || (c == 'h')) { | ||
129 | bb_show_usage(); | ||
130 | } | ||
131 | |||
132 | /* we don't need case information any more */ | ||
133 | c = tolower(c); | ||
134 | |||
135 | /* make sure the option is in range: allowed are q, m, s */ | ||
136 | if (c != 'q' && c != 'm' && c != 's') { | ||
137 | bb_show_usage(); | ||
138 | } | ||
139 | |||
140 | if (iskey) { | ||
141 | /* keys are in hex or decimal */ | ||
142 | key_t key = xstrtoul(optarg, 0); | ||
143 | |||
144 | if (key == IPC_PRIVATE) { | ||
145 | error++; | ||
146 | bb_error_msg("illegal key (%s)", optarg); | ||
147 | continue; | ||
148 | } | ||
149 | |||
150 | /* convert key to id */ | ||
151 | id = ((c == 'q') ? msgget(key, 0) : | ||
152 | (c == 'm') ? shmget(key, 0, 0) : semget(key, 0, 0)); | ||
153 | |||
154 | if (id < 0) { | ||
155 | char *errmsg; | ||
156 | const char * const what = "key"; | ||
157 | |||
158 | error++; | ||
159 | switch (errno) { | ||
160 | case EACCES: | ||
161 | errmsg = "permission denied for"; | ||
162 | break; | ||
163 | case EIDRM: | ||
164 | errmsg = "already removed"; | ||
165 | break; | ||
166 | case ENOENT: | ||
167 | errmsg = "invalid"; | ||
168 | break; | ||
169 | default: | ||
170 | errmsg = "unknown error in"; | ||
171 | break; | ||
172 | } | ||
173 | bb_error_msg("%s %s (%s)", errmsg, what, optarg); | ||
174 | continue; | ||
175 | } | ||
176 | } else { | ||
177 | /* ids are in decimal */ | ||
178 | id = xatoul(optarg); | ||
179 | } | ||
180 | |||
181 | result = ((c == 'q') ? msgctl(id, IPC_RMID, NULL) : | ||
182 | (c == 'm') ? shmctl(id, IPC_RMID, NULL) : | ||
183 | semctl(id, 0, IPC_RMID, arg)); | ||
184 | |||
185 | if (result) { | ||
186 | char *errmsg; | ||
187 | const char * const what = iskey ? "key" : "id"; | ||
188 | |||
189 | error++; | ||
190 | switch (errno) { | ||
191 | case EACCES: | ||
192 | case EPERM: | ||
193 | errmsg = "permission denied for"; | ||
194 | break; | ||
195 | case EINVAL: | ||
196 | errmsg = "invalid"; | ||
197 | break; | ||
198 | case EIDRM: | ||
199 | errmsg = "already removed"; | ||
200 | break; | ||
201 | default: | ||
202 | errmsg = "unknown error in"; | ||
203 | break; | ||
204 | } | ||
205 | bb_error_msg("%s %s (%s)", errmsg, what, optarg); | ||
206 | continue; | ||
207 | } | ||
208 | } | ||
209 | |||
210 | /* print usage if we still have some arguments left over */ | ||
211 | if (optind != argc) { | ||
212 | bb_show_usage(); | ||
213 | } | ||
214 | |||
215 | /* exit value reflects the number of errors encountered */ | ||
216 | return error; | ||
217 | } | ||