sys-utils/ipcmk.c

   1 /*
   2  *  ipcmk.c - used to create ad-hoc IPC segments
   3  *
   4  *  Copyright (C) 2008 Hayden A. James (hayden.james@gmail.com)
   5  *  Copyright (C) 2008 Karel Zak <kzak@redhat.com>
   6  *
   7  *  This program is free software; you can redistribute it and/or modify
   8  *  it under the terms of the GNU General Public License as published by
   9  *  the Free Software Foundation; either version 2 of the License, or
  10  *  (at your option) any later version.
  11  *
  12  *  This program is distributed in the hope that it will be useful,
  13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  *  GNU General Public License for more details.
  16  *
  17  *  You should have received a copy of the GNU General Public License
  18  *  along with this program; if not, write to the Free Software
  19  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  20  */
  21
  22 #include <stdlib.h>
  23 #include <stdio.h>
  24 #include <string.h>
  25 #include <errno.h>
  26 #include <time.h>
  27
  28 #include <unistd.h>
  29 #include <sys/types.h>
  30 #include <sys/ipc.h>
  31 #include <sys/shm.h>
  32 #include <sys/sem.h>
  33 #include <sys/msg.h>
  34
  35 #include "nls.h"
  36 #include "c.h"
  37
  38 static const char *progname;
  39
  40 key_t createKey(void)
  41 {
  42         srandom( time( NULL ) );
  43         return random();
  44 }
  45
  46 int createShm(size_t size, int permission)
  47 {
  48         int result = -1;
  49         int shmid;
  50         key_t key = createKey();
  51
  52         if (-1 != (shmid = shmget(key, size, permission | IPC_CREAT)))
  53                 result = shmid;
  54
  55         return result;
  56 }
  57
  58 int createMsg(int permission)
  59 {
  60         int result = -1;
  61         int msgid;
  62         key_t key = createKey();
  63
  64         if (-1 != (msgid = msgget(key, permission | IPC_CREAT)))
  65                 result = msgid;
  66
  67         return result;
  68 }
  69
  70 int createSem(int nsems, int permission)
  71 {
  72         int result = -1;
  73         int semid;
  74         key_t key = createKey();
  75
  76         if (-1 != (semid = semget(key, nsems, permission | IPC_CREAT)))
  77                 result = semid;
  78
  79         return result;
  80 }
  81
  82 void usage(int rc)
  83 {
  84         FILE *out = rc == EXIT_FAILURE ? stderr : stdout;
  85
  86         fputs(_("\nUsage:\n"), out);
  87         fprintf(out,
  88              _(" %s [options]\n"), progname);
  89
  90         fputs(_("\nOptions:\n"), out);
  91         fputs(_(" -M <size>     create shared memory segment of size <size>\n"
  92                 " -S <nsems>    create semaphore array with <nsems> elements\n"
  93                 " -Q            create message queue\n"
  94                 " -p <mode>     permission for the resource (default is 0644)\n"), out);
  95
  96         fputs(_("\nFor more information see ipcmk(1).\n"), out);
  97
  98         exit(rc);
  99 }
 100
 101 int main(int argc, char **argv)
 102 {
 103         int permission = 0644;
 104         int opt;
 105         size_t size = 0;
 106         int nsems = 0;
 107         int doShm = 0, doMsg = 0, doSem = 0;
 108
 109         progname = program_invocation_short_name;
 110         if (!progname)
 111                 progname = "ipcmk";
 112
 113         setlocale(LC_ALL, "");
 114         bindtextdomain(PACKAGE, LOCALEDIR);
 115         textdomain(PACKAGE);
 116
 117         while((opt = getopt(argc, argv, "hM:QS:p:")) != -1) {
 118                 switch(opt) {
 119                 case 'M':
 120                         size = atoi(optarg);
 121                         doShm = 1;
 122                         break;
 123                 case 'Q':
 124                         doMsg = 1;
 125                         break;
 126                 case 'S':
 127                         nsems = atoi(optarg);
 128                         doSem = 1;
 129                         break;
 130                 case 'p':
 131                         permission = strtoul(optarg, NULL, 8);
 132                         break;
 133                 case 'h':
 134                         usage(EXIT_SUCCESS);
 135                         break;
 136                 default:
 137                         doShm = doMsg = doSem = 0;
 138                         break;
 139                 }
 140         }
 141
 142         if(!doShm && !doMsg && !doSem)
 143                 usage(EXIT_FAILURE);
 144
 145         if (doShm) {
 146                 int shmid;
 147                 if (-1 == (shmid = createShm(size, permission)))
 148                         err(EXIT_FAILURE, _("create share memory failed"));
 149                 else
 150                         printf(_("Shared memory id: %d\n"), shmid);
 151         }
 152
 153         if (doMsg) {
 154                 int msgid;
 155                 if (-1 == (msgid = createMsg(permission)))
 156                         err(EXIT_FAILURE, _("create message queue failed"));
 157                 else
 158                         printf(_("Message queue id: %d\n"), msgid);
 159         }
 160
 161         if (doSem) {
 162                 int semid;
 163                 if (-1 == (semid = createSem(nsems, permission)))
 164                         err(EXIT_FAILURE, _("create semaphore failed"));
 165                 else
 166                         printf(_("Semaphore id: %d\n"), semid);
 167         }
 168
 169         return EXIT_SUCCESS;
 170 }