[3320] | 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 source tree.
|
---|
| 9 | */
|
---|
| 10 |
|
---|
| 11 | #include "libbb.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 *array;
|
---|
| 28 | struct seminfo *__buf;
|
---|
| 29 | };
|
---|
| 30 | #endif
|
---|
| 31 |
|
---|
| 32 | #define IPCRM_LEGACY 1
|
---|
| 33 |
|
---|
| 34 |
|
---|
| 35 | #if IPCRM_LEGACY
|
---|
| 36 |
|
---|
| 37 | typedef enum type_id {
|
---|
| 38 | SHM,
|
---|
| 39 | SEM,
|
---|
| 40 | MSG
|
---|
| 41 | } type_id;
|
---|
| 42 |
|
---|
| 43 | static int remove_ids(type_id type, char **argv)
|
---|
| 44 | {
|
---|
| 45 | unsigned long id;
|
---|
| 46 | int nb_errors = 0;
|
---|
| 47 | union semun arg;
|
---|
| 48 |
|
---|
| 49 | arg.val = 0;
|
---|
| 50 |
|
---|
| 51 | while (argv[0]) {
|
---|
| 52 | id = bb_strtoul(argv[0], NULL, 10);
|
---|
| 53 | if (errno || id > INT_MAX) {
|
---|
| 54 | bb_error_msg("invalid id: %s", argv[0]);
|
---|
| 55 | nb_errors++;
|
---|
| 56 | } else {
|
---|
| 57 | int ret = 0;
|
---|
| 58 | if (type == SEM)
|
---|
| 59 | ret = semctl(id, 0, IPC_RMID, arg);
|
---|
| 60 | else if (type == MSG)
|
---|
| 61 | ret = msgctl(id, IPC_RMID, NULL);
|
---|
| 62 | else if (type == SHM)
|
---|
| 63 | ret = shmctl(id, IPC_RMID, NULL);
|
---|
| 64 |
|
---|
| 65 | if (ret) {
|
---|
| 66 | bb_perror_msg("can't remove id %s", argv[0]);
|
---|
| 67 | nb_errors++;
|
---|
| 68 | }
|
---|
| 69 | }
|
---|
| 70 | argv++;
|
---|
| 71 | }
|
---|
| 72 |
|
---|
| 73 | return nb_errors;
|
---|
| 74 | }
|
---|
| 75 | #endif /* IPCRM_LEGACY */
|
---|
| 76 |
|
---|
| 77 |
|
---|
| 78 | int ipcrm_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
|
---|
| 79 | int ipcrm_main(int argc, char **argv)
|
---|
| 80 | {
|
---|
| 81 | int c;
|
---|
| 82 | int error = 0;
|
---|
| 83 |
|
---|
| 84 | /* if the command is executed without parameters, do nothing */
|
---|
| 85 | if (argc == 1)
|
---|
| 86 | return 0;
|
---|
| 87 | #if IPCRM_LEGACY
|
---|
| 88 | /* check to see if the command is being invoked in the old way if so
|
---|
| 89 | then run the old code. Valid commands are msg, shm, sem. */
|
---|
| 90 | {
|
---|
| 91 | type_id what = 0; /* silence gcc */
|
---|
| 92 | char w;
|
---|
| 93 |
|
---|
| 94 | w = argv[1][0];
|
---|
| 95 | if ( ((w == 'm' && argv[1][1] == 's' && argv[1][2] == 'g')
|
---|
| 96 | || (argv[1][0] == 's'
|
---|
| 97 | && ((w = argv[1][1]) == 'h' || w == 'e')
|
---|
| 98 | && argv[1][2] == 'm')
|
---|
| 99 | ) && argv[1][3] == '\0'
|
---|
| 100 | ) {
|
---|
| 101 | if (argc < 3)
|
---|
| 102 | bb_show_usage();
|
---|
| 103 |
|
---|
| 104 | if (w == 'h')
|
---|
| 105 | what = SHM;
|
---|
| 106 | else if (w == 'm')
|
---|
| 107 | what = MSG;
|
---|
| 108 | else if (w == 'e')
|
---|
| 109 | what = SEM;
|
---|
| 110 |
|
---|
| 111 | if (remove_ids(what, &argv[2]))
|
---|
| 112 | fflush_stdout_and_exit(EXIT_FAILURE);
|
---|
| 113 | printf("resource(s) deleted\n");
|
---|
| 114 | return 0;
|
---|
| 115 | }
|
---|
| 116 | }
|
---|
| 117 | #endif /* IPCRM_LEGACY */
|
---|
| 118 |
|
---|
| 119 | /* process new syntax to conform with SYSV ipcrm */
|
---|
| 120 | while ((c = getopt(argc, argv, "q:m:s:Q:M:S:h?")) != -1) {
|
---|
| 121 | int result;
|
---|
| 122 | int id = 0;
|
---|
| 123 | int iskey = isupper(c);
|
---|
| 124 |
|
---|
| 125 | /* needed to delete semaphores */
|
---|
| 126 | union semun arg;
|
---|
| 127 |
|
---|
| 128 | arg.val = 0;
|
---|
| 129 |
|
---|
| 130 | if ((c == '?') || (c == 'h')) {
|
---|
| 131 | bb_show_usage();
|
---|
| 132 | }
|
---|
| 133 |
|
---|
| 134 | /* we don't need case information any more */
|
---|
| 135 | c = tolower(c);
|
---|
| 136 |
|
---|
| 137 | /* make sure the option is in range: allowed are q, m, s */
|
---|
| 138 | if (c != 'q' && c != 'm' && c != 's') {
|
---|
| 139 | bb_show_usage();
|
---|
| 140 | }
|
---|
| 141 |
|
---|
| 142 | if (iskey) {
|
---|
| 143 | /* keys are in hex or decimal */
|
---|
| 144 | key_t key = xstrtoul(optarg, 0);
|
---|
| 145 |
|
---|
| 146 | if (key == IPC_PRIVATE) {
|
---|
| 147 | error++;
|
---|
| 148 | bb_error_msg("illegal key (%s)", optarg);
|
---|
| 149 | continue;
|
---|
| 150 | }
|
---|
| 151 |
|
---|
| 152 | /* convert key to id */
|
---|
| 153 | id = ((c == 'q') ? msgget(key, 0) :
|
---|
| 154 | (c == 'm') ? shmget(key, 0, 0) : semget(key, 0, 0));
|
---|
| 155 |
|
---|
| 156 | if (id < 0) {
|
---|
| 157 | const char *errmsg;
|
---|
| 158 |
|
---|
| 159 | error++;
|
---|
| 160 | switch (errno) {
|
---|
| 161 | case EACCES:
|
---|
| 162 | errmsg = "permission denied for";
|
---|
| 163 | break;
|
---|
| 164 | case EIDRM:
|
---|
| 165 | errmsg = "already removed";
|
---|
| 166 | break;
|
---|
| 167 | case ENOENT:
|
---|
| 168 | errmsg = "invalid";
|
---|
| 169 | break;
|
---|
| 170 | default:
|
---|
| 171 | errmsg = "unknown error in";
|
---|
| 172 | break;
|
---|
| 173 | }
|
---|
| 174 | bb_error_msg("%s %s (%s)", errmsg, "key", optarg);
|
---|
| 175 | continue;
|
---|
| 176 | }
|
---|
| 177 | } else {
|
---|
| 178 | /* ids are in decimal */
|
---|
| 179 | id = xatoul(optarg);
|
---|
| 180 | }
|
---|
| 181 |
|
---|
| 182 | result = ((c == 'q') ? msgctl(id, IPC_RMID, NULL) :
|
---|
| 183 | (c == 'm') ? shmctl(id, IPC_RMID, NULL) :
|
---|
| 184 | semctl(id, 0, IPC_RMID, arg));
|
---|
| 185 |
|
---|
| 186 | if (result) {
|
---|
| 187 | const char *errmsg;
|
---|
| 188 | const char *const what = iskey ? "key" : "id";
|
---|
| 189 |
|
---|
| 190 | error++;
|
---|
| 191 | switch (errno) {
|
---|
| 192 | case EACCES:
|
---|
| 193 | case EPERM:
|
---|
| 194 | errmsg = "permission denied for";
|
---|
| 195 | break;
|
---|
| 196 | case EINVAL:
|
---|
| 197 | errmsg = "invalid";
|
---|
| 198 | break;
|
---|
| 199 | case EIDRM:
|
---|
| 200 | errmsg = "already removed";
|
---|
| 201 | break;
|
---|
| 202 | default:
|
---|
| 203 | errmsg = "unknown error in";
|
---|
| 204 | break;
|
---|
| 205 | }
|
---|
| 206 | bb_error_msg("%s %s (%s)", errmsg, what, optarg);
|
---|
| 207 | continue;
|
---|
| 208 | }
|
---|
| 209 | }
|
---|
| 210 |
|
---|
| 211 | /* print usage if we still have some arguments left over */
|
---|
| 212 | if (optind != argc) {
|
---|
| 213 | bb_show_usage();
|
---|
| 214 | }
|
---|
| 215 |
|
---|
| 216 | /* exit value reflects the number of errors encountered */
|
---|
| 217 | return error;
|
---|
| 218 | }
|
---|