util-linux/ipcrm.c

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