LXR toybox/toys/pending/ipcrm.c

   1/* ipcrm.c - remove msg que, sem or shared memory
   2 *
   3 * Copyright 2014 Ashwini Kumar <ak.ashwini1981@gmail.com>
   4 *
   5 * See http://pubs.opengroup.org/onlinepubs/9699919799/utilities/ipcrm.html
   6
   7
   8USE_IPCRM(NEWTOY(ipcrm, "m*M*s*S*q*Q*", TOYFLAG_USR|TOYFLAG_BIN))
   9
  10config IPCRM
  11  bool "ipcrm"
  12  default n
  13  help
  14    usage: ipcrm [ [-q msqid] [-m shmid] [-s semid]
  15              [-Q msgkey] [-M shmkey] [-S semkey] ... ]
  16
  17    -mM Remove memory segment after last detach
  18    -qQ Remove message queue
  19    -sS Remove semaphore
  20*/
  21
  22#define FOR_ipcrm
  23#include "toys.h"
  24#include <sys/ipc.h>
  25#include <sys/shm.h>
  26#include <sys/sem.h>
  27#include <sys/msg.h>
  28
  29GLOBALS(
  30  struct arg_list *qkey;
  31  struct arg_list *qid;
  32  struct arg_list *skey;
  33  struct arg_list *sid;
  34  struct arg_list *mkey;
  35  struct arg_list *mid;
  36)
  37
  38static void do_ipcrm(int key, int ipc, char *name)
  39{
  40  char *c;
  41  int id, ret = 0;
  42
  43  id = strtol(name, &c, 0);
  44  if (*c) {
  45    error_msg("invalid number :%s", name);
  46    return;
  47  }
  48
  49  if (key) {
  50    if (id == IPC_PRIVATE) {
  51      error_msg("illegal key (%s)", name);
  52      return;
  53    }
  54    id = ((ipc == 1)?shmget(id, 0, 0) :
  55         (ipc == 2)? msgget(id, 0): semget(id, 0, 0));
  56    if (id < 0) {
  57      perror_msg("key (%s)", name);
  58      return;
  59    }
  60  }
  61
  62  if (ipc == 1) ret = shmctl(id, IPC_RMID, NULL);
  63  else if (ipc == 2) ret = msgctl(id, IPC_RMID, NULL);
  64  else if (ipc == 3) ret = semctl(id, 0, IPC_RMID, NULL);
  65
  66  if (ret < 0) perror_msg("%s (%s)", ((key)? "key": "id"), name);
  67}
  68
  69void ipcrm_main(void)
  70{
  71  ++toys.argv;
  72  if (toys.optc && (!strcmp(*toys.argv, "shm") ||
  73        !strcmp(*toys.argv, "sem") || !strcmp(*toys.argv, "msg"))) {
  74    int t = (toys.argv[0][1] == 'h')? 1 : (toys.argv[0][1] == 's')? 2:3;
  75
  76    while (*(++toys.argv)) do_ipcrm(0, t, *toys.argv); 
  77  } else {
  78    struct arg_list *tmp;
  79
  80    for (tmp = TT.mkey; tmp; tmp = tmp->next) do_ipcrm(1, 1, tmp->arg);
  81    for (tmp = TT.mid; tmp; tmp = tmp->next) do_ipcrm(0, 1, tmp->arg);
  82    for (tmp = TT.qkey; tmp; tmp = tmp->next) do_ipcrm(1, 2, tmp->arg);
  83    for (tmp = TT.qid; tmp; tmp = tmp->next) do_ipcrm(0, 2, tmp->arg);
  84    for (tmp = TT.skey; tmp; tmp = tmp->next) do_ipcrm(1, 3, tmp->arg);
  85    for (tmp = TT.sid; tmp; tmp = tmp->next) do_ipcrm(0, 3, tmp->arg);
  86    if (toys.optc) help_exit("unknown argument: %s", *toys.optargs);
  87  }
  88}
  89