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#include <sys/cdefs.h>
33#ifndef lint
34__used static const char rcsid[] =
35	"$Id: ipcrm.c,v 1.3 2005/02/03 07:31:33 josborne Exp $";
36#endif /* not lint */
37
38#include <ctype.h>
39#include <err.h>
40#include <signal.h>
41#include <stdio.h>
42#include <stdlib.h>
43#include <unistd.h>
44#include <sys/types.h>
45#include <sys/ipc.h>
46#include <sys/msg.h>
47#include <sys/sem.h>
48#include <sys/shm.h>
49
50#define IPC_TO_STR(x) (x == 'Q' ? "msq" : (x == 'M' ? "shm" : "sem"))
51#define IPC_TO_STRING(x) (x == 'Q' ? "message queue" : \
52	(x == 'M' ? "shared memory segment" : "semaphore"))
53
54int signaled;
55
56static void usage(void)
57{
58	fprintf(stderr, "%s\n%s\n",
59		"usage: ipcrm [-q msqid] [-m shmid] [-s semid]",
60		"             [-Q msgkey] [-M shmkey] [-S semkey] ...");
61	exit(1);
62}
63
64static int msgrm(key_t key, int id)
65{
66    if (key) {
67	id = msgget(key, 0);
68	if (id == -1)
69	    return -1;
70    }
71    return msgctl(id, IPC_RMID, NULL);
72}
73
74static int shmrm(key_t key, int id)
75{
76    if (key) {
77	id = shmget(key, 0, 0);
78	if (id == -1)
79	    return -1;
80    }
81    return shmctl(id, IPC_RMID, NULL);
82}
83
84static int semrm(key_t key, int id)
85{
86    union semun arg;
87
88    if (key) {
89	id = semget(key, 0, 0);
90	if (id == -1)
91	    return -1;
92    }
93    return semctl(id, 0, IPC_RMID, arg);
94}
95
96static void not_configured(__unused int unused)
97{
98    signaled++;
99}
100
101int main(argc, argv)
102    int argc;
103    char *argv[];
104
105{
106    int c, result, errflg, target_id;
107    key_t target_key;
108    char *en;
109
110    errflg = 0;
111    signal(SIGSYS, not_configured);
112    while ((c = getopt(argc, argv, ":q:m:s:Q:M:S:")) != -1) {
113
114	signaled = 0;
115	switch (c) {
116	case 'q':
117	case 'm':
118	case 's':
119	    target_id = (int)strtol(optarg, &en, 0);
120	    if (*en) {
121		warnx("%s: '%s' is not a number",
122		    IPC_TO_STRING(toupper(c)), optarg);
123		continue;
124	    }
125	    if (c == 'q')
126		result = msgrm(0, target_id);
127	    else if (c == 'm')
128		result = shmrm(0, target_id);
129	    else
130		result = semrm(0, target_id);
131	    if (result < 0) {
132		errflg++;
133		if (!signaled)
134		    warn("%sid(%d): ", IPC_TO_STR(toupper(c)), target_id);
135		else
136		    warnx("%ss are not configured in the running kernel",
137			  IPC_TO_STRING(toupper(c)));
138	    }
139	    break;
140	case 'Q':
141	case 'M':
142	case 'S':
143	    target_key = (key_t)strtol(optarg, &en, 0);
144	    if (*en) {
145		warnx("%s: '%s' is not a number", IPC_TO_STRING(c), optarg);
146		continue;
147	    }
148	    if (target_key == IPC_PRIVATE) {
149		warnx("can't remove private %ss", IPC_TO_STRING(c));
150		continue;
151	    }
152	    if (c == 'Q')
153		result = msgrm(target_key, 0);
154	    else if (c == 'M')
155		result = shmrm(target_key, 0);
156	    else
157		result = semrm(target_key, 0);
158	    if (result < 0) {
159		errflg++;
160		if (!signaled)
161		    warn("%s key(%d): ", IPC_TO_STRING(c), target_key);
162		else
163		    warnx("%ss are not configured in the running kernel",
164			  IPC_TO_STRING(c));
165	    }
166	    break;
167	case ':':
168	    fprintf(stderr, "option -%c requires an argument\n", optopt);
169	    usage();
170	case '?':
171	    fprintf(stderr, "unrecognized option: -%c\n", optopt);
172	    usage();
173	}
174    }
175
176    if (optind != argc) {
177	    fprintf(stderr, "unknown argument: %s\n", argv[optind]);
178	    usage();
179    }
180    exit(errflg);
181}
182