]> git.ipfire.org Git - thirdparty/util-linux.git/blobdiff - sys-utils/ipcmk.c
misc: consolidate version printing and close_stdout()
[thirdparty/util-linux.git] / sys-utils / ipcmk.c
index 5999f0f1c928908fffd3b40be6e97bb4a4c1e128..8aadffbbbacd69b3a8378eb24a80f202b0e0b2dd 100644 (file)
  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  *  GNU General Public License for more details.
  *
- *  You should have received a copy of the GNU General Public License
- *  along with this program; if not, write to the Free Software
- *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  */
 
-#include <stdlib.h>
-#include <stdio.h>
-#include <string.h>
 #include <errno.h>
-#include <time.h>
-
-#include <unistd.h>
-#include <sys/types.h>
+#include <getopt.h>
+#include <stddef.h>
+#include <stdio.h>
+#include <stdlib.h>
 #include <sys/ipc.h>
-#include <sys/shm.h>
-#include <sys/sem.h>
 #include <sys/msg.h>
+#include <sys/sem.h>
+#include <sys/shm.h>
+#include <sys/time.h>
 
-#include "nls.h"
 #include "c.h"
+#include "nls.h"
+#include "randutils.h"
+#include "strutils.h"
+#include "closestream.h"
 
-static const char *progname;
-
-key_t createKey(void)
-{
-       srandom( time( NULL ) );
-       return random();
-}
-
-int createShm(size_t size, int permission)
+static int create_shm(size_t size, int permission)
 {
-       int result = -1;
-       int shmid;
-       key_t key = createKey();
+       key_t key;
 
-       if (-1 != (shmid = shmget(key, size, permission | IPC_CREAT)))
-               result = shmid;
-
-       return result;
+       random_get_bytes(&key, sizeof(key));
+       return shmget(key, size, permission | IPC_CREAT);
 }
 
-int createMsg(int permission)
+static int create_msg(int permission)
 {
-       int result = -1;
-       int msgid;
-       key_t key = createKey();
-
-       if (-1 != (msgid = msgget(key, permission | IPC_CREAT)))
-               result = msgid;
+       key_t key;
 
-       return result;
+       random_get_bytes(&key, sizeof(key));
+       return msgget(key, permission | IPC_CREAT);
 }
 
-int createSem(int nsems, int permission)
+static int create_sem(int nsems, int permission)
 {
-       int result = -1;
-       int semid;
-       key_t key = createKey();
-
-       if (-1 != (semid = semget(key, nsems, permission | IPC_CREAT)))
-               result = semid;
+       key_t key;
 
-       return result;
+       random_get_bytes(&key, sizeof(key));
+       return semget(key, nsems, permission | IPC_CREAT);
 }
 
-void usage(int rc)
+static void __attribute__((__noreturn__)) usage(void)
 {
-       FILE *out = rc == EXIT_FAILURE ? stderr : stdout;
+       FILE *out = stdout;
+       fputs(USAGE_HEADER, out);
+       fprintf(out, _(" %s [options]\n"), program_invocation_short_name);
 
-       fputs(_("\nUsage:\n"), out);
-       fprintf(out,
-            _(" %s [options]\n"), progname);
+       fputs(USAGE_SEPARATOR, out);
+       fputs(_("Create various IPC resources.\n"), out);
 
-       fputs(_("\nOptions:\n"), out);
-       fputs(_(" -M <size>     create shared memory segment of size <size>\n"
-               " -S <nsems>    create semaphore array with <nsems> elements\n"
-               " -Q            create message queue\n"
-               " -p <mode>     permission for the resource (default is 0644)\n"), out);
+       fputs(USAGE_OPTIONS, out);
+       fputs(_(" -M, --shmem <size>       create shared memory segment of size <size>\n"), out);
+       fputs(_(" -S, --semaphore <number> create semaphore array with <number> elements\n"), out);
+       fputs(_(" -Q, --queue              create message queue\n"), out);
+       fputs(_(" -p, --mode <mode>        permission for the resource (default is 0644)\n"), out);
 
-       fputs(_("\nFor more information see ipcmk(1).\n"), out);
+       fputs(USAGE_SEPARATOR, out);
+       printf(USAGE_HELP_OPTIONS(26));
+       printf(USAGE_MAN_TAIL("ipcmk(1)"));
 
-       exit(rc);
+       exit(EXIT_SUCCESS);
 }
 
 int main(int argc, char **argv)
@@ -104,63 +88,71 @@ int main(int argc, char **argv)
        int opt;
        size_t size = 0;
        int nsems = 0;
-       int doShm = 0, doMsg = 0, doSem = 0;
-
-       progname = program_invocation_short_name;
-       if (!progname)
-               progname = "ipcmk";
+       int ask_shm = 0, ask_msg = 0, ask_sem = 0;
+       static const struct option longopts[] = {
+               {"shmem", required_argument, NULL, 'M'},
+               {"semaphore", required_argument, NULL, 'S'},
+               {"queue", no_argument, NULL, 'Q'},
+               {"mode", required_argument, NULL, 'p'},
+               {"version", no_argument, NULL, 'V'},
+               {"help", no_argument, NULL, 'h'},
+               {NULL, 0, NULL, 0}
+       };
 
        setlocale(LC_ALL, "");
        bindtextdomain(PACKAGE, LOCALEDIR);
        textdomain(PACKAGE);
+       close_stdout_atexit();
 
-       while((opt = getopt(argc, argv, "hM:QS:p:")) != -1) {
+       while((opt = getopt_long(argc, argv, "hM:QS:p:Vh", longopts, NULL)) != -1) {
                switch(opt) {
                case 'M':
-                       size = atoi(optarg);
-                       doShm = 1;
+                       size = strtosize_or_err(optarg, _("failed to parse size"));
+                       ask_shm = 1;
                        break;
                case 'Q':
-                       doMsg = 1;
+                       ask_msg = 1;
                        break;
                case 'S':
-                       nsems = atoi(optarg);
-                       doSem = 1;
+                       nsems = strtos32_or_err(optarg, _("failed to parse elements"));
+                       ask_sem = 1;
                        break;
                case 'p':
                        permission = strtoul(optarg, NULL, 8);
                        break;
+
                case 'h':
-                       usage(EXIT_SUCCESS);
-                       break;
+                       usage();
+               case 'V':
+                       print_version(EXIT_SUCCESS);
                default:
-                       doShm = doMsg = doSem = 0;
-                       break;
+                       errtryhelp(EXIT_FAILURE);
                }
        }
 
-       if(!doShm && !doMsg && !doSem)
-               usage(EXIT_FAILURE);
-
-       if (doShm) {
+       if(!ask_shm && !ask_msg && !ask_sem) {
+               warnx(_("bad usage"));
+               errtryhelp(EXIT_FAILURE);
+       }
+       if (ask_shm) {
                int shmid;
-               if (-1 == (shmid = createShm(size, permission)))
+               if (-1 == (shmid = create_shm(size, permission)))
                        err(EXIT_FAILURE, _("create share memory failed"));
                else
                        printf(_("Shared memory id: %d\n"), shmid);
        }
 
-       if (doMsg) {
+       if (ask_msg) {
                int msgid;
-               if (-1 == (msgid = createMsg(permission)))
+               if (-1 == (msgid = create_msg(permission)))
                        err(EXIT_FAILURE, _("create message queue failed"));
                else
                        printf(_("Message queue id: %d\n"), msgid);
        }
 
-       if (doSem) {
+       if (ask_sem) {
                int semid;
-               if (-1 == (semid = createSem(nsems, permission)))
+               if (-1 == (semid = create_sem(nsems, permission)))
                        err(EXIT_FAILURE, _("create semaphore failed"));
                else
                        printf(_("Semaphore id: %d\n"), semid);