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 tarball for details.
   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 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 int ipcrm_main(int argc, char **argv)
  78 {
  79         int c;
  80         int error = 0;
  81
  82         /* if the command is executed without parameters, do nothing */
  83         if (argc == 1)
  84                 return 0;
  85 #ifndef CONFIG_IPCRM_DROP_LEGACY
  86         /* check to see if the command is being invoked in the old way if so
  87            then run the old code. Valid commands are msg, shm, sem. */
  88         {
  89                 type_id what = 0; /* silence gcc */
  90                 char w;
  91
  92                 w=argv[1][0];
  93                 if ( ((w == 'm' && argv[1][1] == 's' && argv[1][2] == 'g')
  94                        || (argv[1][0] == 's'
  95                            && ((w=argv[1][1]) == 'h' || w == 'e')
  96                            && argv[1][2] == 'm')
  97                      ) && argv[1][3] == '\0'
  98                 ) {
  99
 100                         if (argc < 3)
 101                                 bb_show_usage();
 102
 103                         if (w == 'h')
 104                                 what = SHM;
 105                         else if (w == 'm')
 106                                 what = MSG;
 107                         else if (w == 'e')
 108                                 what = SEM;
 109
 110                         if (remove_ids(what, argc-2, &argv[2]))
 111                                 fflush_stdout_and_exit(1);
 112                         printf("resource(s) deleted\n");
 113                         return 0;
 114                 }
 115         }
 116 #endif /* #ifndef CONFIG_IPCRM_DROP_LEGACY */
 117
 118         /* process new syntax to conform with SYSV ipcrm */
 119         while ((c = getopt(argc, argv, "q:m:s:Q:M:S:h?")) != -1) {
 120                 int result;
 121                 int id = 0;
 122                 int iskey = (isupper)(c);
 123
 124                 /* needed to delete semaphores */
 125                 union semun arg;
 126
 127                 arg.val = 0;
 128
 129                 if ((c == '?') || (c == 'h')) {
 130                         bb_show_usage();
 131                 }
 132
 133                 /* we don't need case information any more */
 134                 c = tolower(c);
 135
 136                 /* make sure the option is in range: allowed are q, m, s */
 137                 if (c != 'q' && c != 'm' && c != 's') {
 138                         bb_show_usage();
 139                 }
 140
 141                 if (iskey) {
 142                         /* keys are in hex or decimal */
 143                         key_t key = xstrtoul(optarg, 0);
 144
 145                         if (key == IPC_PRIVATE) {
 146                                 error++;
 147                                 bb_error_msg("illegal key (%s)", optarg);
 148                                 continue;
 149                         }
 150
 151                         /* convert key to id */
 152                         id = ((c == 'q') ? msgget(key, 0) :
 153                                   (c == 'm') ? shmget(key, 0, 0) : semget(key, 0, 0));
 154
 155                         if (id < 0) {
 156                                 const char *errmsg;
 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, "key", 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                         const 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 }