1 /* 2 * Copyright (c) 1994 Adam Glass 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. All advertising materials mentioning features or use of this software 14 * must display the following acknowledgement: 15 * This product includes software developed by Adam Glass. 16 * 4. The name of the Author may not be used to endorse or promote products 17 * derived from this software without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY Adam Glass ``AS IS'' AND 20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 22 * ARE DISCLAIMED. IN NO EVENT SHALL Adam Glass BE LIABLE 23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 29 * SUCH DAMAGE. 30 * 31 */ 32 33 #include <sys/cdefs.h> 34 __FBSDID("$FreeBSD$"); 35 36 #include <sys/param.h> 37 #define _WANT_MSG_PROTOTYPES 38 #define _WANT_SEM_PROTOTYPES 39 #define _WANT_SHM_PROTOTYPES 40 #define _KERNEL 41 #include <sys/sem.h> 42 #include <sys/shm.h> 43 #include <sys/msg.h> 44 #undef _KERNEL 45 46 #include <ctype.h> 47 #include <err.h> 48 #include <grp.h> 49 #include <kvm.h> 50 #include <stdio.h> 51 #include <stdlib.h> 52 #include <unistd.h> 53 54 #include "ipc.h" 55 56 static int signaled; 57 static int errflg; 58 static int rmverbose = 0; 59 60 static void 61 usage(void) 62 { 63 64 fprintf(stderr, 65 "usage: ipcrm [-W] [-v[v]]\n" 66 " [-q msqid] [-m shmid] [-s semid]\n" 67 " [-Q msgkey] [-M shmkey] [-S semkey] ...\n"); 68 exit(1); 69 } 70 71 static int 72 msgrm(key_t key, int id) 73 { 74 75 if (key == -1 || id == -1) { 76 struct msqid_kernel *kxmsqids; 77 size_t kxmsqids_len; 78 int num; 79 80 kget(X_MSGINFO, &msginfo, sizeof(msginfo)); 81 kxmsqids_len = sizeof(struct msqid_kernel) * msginfo.msgmni; 82 kxmsqids = malloc(kxmsqids_len); 83 kget(X_MSQIDS, kxmsqids, kxmsqids_len); 84 num = msginfo.msgmni; 85 while (num-- && !signaled) 86 if (kxmsqids[num].u.msg_qbytes != 0) { 87 id = IXSEQ_TO_IPCID(num, 88 kxmsqids[num].u.msg_perm); 89 if (msgctl(id, IPC_RMID, NULL) < 0) { 90 if (rmverbose > 1) 91 warn("msqid(%d): ", id); 92 errflg++; 93 } else 94 if (rmverbose) 95 printf( 96 "Removed %s %d\n", 97 IPC_TO_STRING('Q'), 98 id); 99 } 100 return signaled ? -1 : 0; /* errors maybe handled above */ 101 } 102 103 if (key) { 104 id = msgget(key, 0); 105 if (id == -1) 106 return -1; 107 } 108 109 return msgctl(id, IPC_RMID, NULL); 110 } 111 112 static int 113 shmrm(key_t key, int id) 114 { 115 116 if (key == -1 || id == -1) { 117 struct shmid_kernel *kxshmids; 118 size_t kxshmids_len; 119 int num; 120 121 kget(X_SHMINFO, &shminfo, sizeof(shminfo)); 122 kxshmids_len = sizeof(struct shmid_kernel) * shminfo.shmmni; 123 kxshmids = malloc(kxshmids_len); 124 kget(X_SHMSEGS, kxshmids, kxshmids_len); 125 num = shminfo.shmmni; 126 while (num-- && !signaled) 127 if (kxshmids[num].u.shm_perm.mode & 0x0800) { 128 id = IXSEQ_TO_IPCID(num, 129 kxshmids[num].u.shm_perm); 130 if (shmctl(id, IPC_RMID, NULL) < 0) { 131 if (rmverbose > 1) 132 warn("shmid(%d): ", id); 133 errflg++; 134 } else 135 if (rmverbose) 136 printf( 137 "Removed %s %d\n", 138 IPC_TO_STRING('M'), 139 id); 140 } 141 return signaled ? -1 : 0; /* errors maybe handled above */ 142 } 143 144 if (key) { 145 id = shmget(key, 0, 0); 146 if (id == -1) 147 return -1; 148 } 149 150 return shmctl(id, IPC_RMID, NULL); 151 } 152 153 static int 154 semrm(key_t key, int id) 155 { 156 union semun arg; 157 158 if (key == -1 || id == -1) { 159 struct semid_kernel *kxsema; 160 size_t kxsema_len; 161 int num; 162 163 kget(X_SEMINFO, &seminfo, sizeof(seminfo)); 164 kxsema_len = sizeof(struct semid_kernel) * seminfo.semmni; 165 kxsema = malloc(kxsema_len); 166 kget(X_SEMA, kxsema, kxsema_len); 167 num = seminfo.semmni; 168 while (num-- && !signaled) 169 if ((kxsema[num].u.sem_perm.mode & SEM_ALLOC) != 0) { 170 id = IXSEQ_TO_IPCID(num, 171 kxsema[num].u.sem_perm); 172 if (semctl(id, 0, IPC_RMID, NULL) < 0) { 173 if (rmverbose > 1) 174 warn("semid(%d): ", id); 175 errflg++; 176 } else 177 if (rmverbose) 178 printf( 179 "Removed %s %d\n", 180 IPC_TO_STRING('S'), 181 id); 182 } 183 return signaled ? -1 : 0; /* errors maybe handled above */ 184 } 185 186 if (key) { 187 id = semget(key, 0, 0); 188 if (id == -1) 189 return -1; 190 } 191 192 return semctl(id, 0, IPC_RMID, arg); 193 } 194 195 static void 196 not_configured(int signo __unused) 197 { 198 199 signaled++; 200 } 201 202 int 203 main(int argc, char *argv[]) 204 { 205 int c, result, target_id; 206 key_t target_key; 207 208 while ((c = getopt(argc, argv, "q:m:s:Q:M:S:vWy")) != -1) { 209 210 signaled = 0; 211 switch (c) { 212 case 'v': 213 rmverbose++; 214 break; 215 case 'y': 216 use_sysctl = 0; 217 break; 218 } 219 } 220 221 optind = 1; 222 errflg = 0; 223 signal(SIGSYS, not_configured); 224 while ((c = getopt(argc, argv, "q:m:s:Q:M:S:vWy")) != -1) { 225 226 signaled = 0; 227 switch (c) { 228 case 'q': 229 case 'm': 230 case 's': 231 target_id = atoi(optarg); 232 if (c == 'q') 233 result = msgrm(0, target_id); 234 else if (c == 'm') 235 result = shmrm(0, target_id); 236 else 237 result = semrm(0, target_id); 238 if (result < 0) { 239 errflg++; 240 if (!signaled) 241 warn("%sid(%d): ", 242 IPC_TO_STR(toupper(c)), target_id); 243 else 244 warnx( 245 "%ss are not configured " 246 "in the running kernel", 247 IPC_TO_STRING(toupper(c))); 248 } 249 break; 250 case 'Q': 251 case 'M': 252 case 'S': 253 target_key = atol(optarg); 254 if (target_key == IPC_PRIVATE) { 255 warnx("can't remove private %ss", 256 IPC_TO_STRING(c)); 257 continue; 258 } 259 if (c == 'Q') 260 result = msgrm(target_key, 0); 261 else if (c == 'M') 262 result = shmrm(target_key, 0); 263 else 264 result = semrm(target_key, 0); 265 if (result < 0) { 266 errflg++; 267 if (!signaled) 268 warn("%ss(%ld): ", 269 IPC_TO_STR(c), target_key); 270 else 271 warnx("%ss are not configured " 272 "in the running kernel", 273 IPC_TO_STRING(c)); 274 } 275 break; 276 case 'v': 277 case 'y': 278 /* Handled in other getopt() loop */ 279 break; 280 case 'W': 281 msgrm(-1, 0); 282 shmrm(-1, 0); 283 semrm(-1, 0); 284 break; 285 case ':': 286 fprintf(stderr, 287 "option -%c requires an argument\n", optopt); 288 usage(); 289 case '?': 290 fprintf(stderr, "unrecognized option: -%c\n", optopt); 291 usage(); 292 } 293 } 294 295 if (optind != argc) { 296 fprintf(stderr, "unknown argument: %s\n", argv[optind]); 297 usage(); 298 } 299 exit(errflg); 300 } 301