sbin/hammer2/main.c

   1 /*
   2  * Copyright (c) 2011-2012 The DragonFly Project.  All rights reserved.
   3  *
   4  * This code is derived from software contributed to The DragonFly Project
   5  * by Matthew Dillon <dillon@dragonflybsd.org>
   6  * by Venkatesh Srinivas <vsrinivas@dragonflybsd.org>
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  *
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in
  16  *    the documentation and/or other materials provided with the
  17  *    distribution.
  18  * 3. Neither the name of The DragonFly Project nor the names of its
  19  *    contributors may be used to endorse or promote products derived
  20  *    from this software without specific, prior written permission.
  21  *
  22  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  23  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  24  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  25  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
  26  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  27  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
  28  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  29  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  30  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  31  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  32  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  33  * SUCH DAMAGE.
  34  */
  35
  36 #include "hammer2.h"
  37
  38 static void usage(int code);
  39
  40 int DebugOpt;
  41 int NormalExit = 1;     /* if set to 0 main() has to pthread_exit() */
  42
  43 int
  44 main(int ac, char **av)
  45 {
  46         const char *sel_path = NULL;
  47         const char *uuid_str = NULL;
  48         int pfs_type = HAMMER2_PFSTYPE_NONE;
  49         int quick_opt = 0;
  50         int all_opt = 0;
  51         int ecode = 0;
  52         int ch;
  53
  54         /*
  55          * Core options
  56          */
  57         while ((ch = getopt(ac, av, "aqs:t:u:")) != -1) {
  58                 switch(ch) {
  59                 case 'a':
  60                         all_opt = 1;
  61                         break;
  62                 case 'q':
  63                         /*
  64                          * Quick mode - do not block verifying certain
  65                          * operations such as (connect).
  66                          */
  67                         quick_opt = 1;
  68                         break;
  69                 case 's':
  70                         sel_path = optarg;
  71                         break;
  72                 case 't':
  73                         /*
  74                          * set node type for mkpfs
  75                          */
  76                         if (strcasecmp(optarg, "ADMIN") == 0) {
  77                                 pfs_type = HAMMER2_PFSTYPE_ADMIN;
  78                         } else if (strcasecmp(optarg, "CACHE") == 0) {
  79                                 pfs_type = HAMMER2_PFSTYPE_CACHE;
  80                         } else if (strcasecmp(optarg, "COPY") == 0) {
  81                                 pfs_type = HAMMER2_PFSTYPE_COPY;
  82                         } else if (strcasecmp(optarg, "SLAVE") == 0) {
  83                                 pfs_type = HAMMER2_PFSTYPE_SLAVE;
  84                         } else if (strcasecmp(optarg, "SOFT_SLAVE") == 0) {
  85                                 pfs_type = HAMMER2_PFSTYPE_SOFT_SLAVE;
  86                         } else if (strcasecmp(optarg, "SOFT_MASTER") == 0) {
  87                                 pfs_type = HAMMER2_PFSTYPE_SOFT_MASTER;
  88                         } else if (strcasecmp(optarg, "MASTER") == 0) {
  89                                 pfs_type = HAMMER2_PFSTYPE_MASTER;
  90                         } else {
  91                                 fprintf(stderr, "-t: Unrecognized node type\n");
  92                                 usage(1);
  93                         }
  94                         break;
  95                 case 'u':
  96                         /*
  97                          * set uuid for mkpfs, else one will be generated
  98                          * (required for all except the MASTER node_type)
  99                          */
 100                         uuid_str = optarg;
 101                         break;
 102                 case 'd':
 103                         DebugOpt = 1;
 104                         break;
 105                 default:
 106                         fprintf(stderr, "Unknown option: %c\n", ch);
 107                         usage(1);
 108                         /* not reached */
 109                         break;
 110                 }
 111         }
 112
 113         /*
 114          * Adjust, then process the command
 115          */
 116         ac -= optind;
 117         av += optind;
 118         if (ac < 1) {
 119                 fprintf(stderr, "Missing command\n");
 120                 usage(1);
 121                 /* not reached */
 122         }
 123
 124         if (strcmp(av[0], "connect") == 0) {
 125                 /*
 126                  * Add cluster connection
 127                  */
 128                 if (ac < 2) {
 129                         fprintf(stderr, "connect: missing argument\n");
 130                         usage(1);
 131                 }
 132                 ecode = cmd_remote_connect(sel_path, av[1]);
 133         } else if (strcmp(av[0], "disconnect") == 0) {
 134                 /*
 135                  * Remove cluster connection
 136                  */
 137                 if (ac < 2) {
 138                         fprintf(stderr, "disconnect: missing argument\n");
 139                         usage(1);
 140                 }
 141                 ecode = cmd_remote_disconnect(sel_path, av[1]);
 142         } else if (strcmp(av[0], "status") == 0) {
 143                 /*
 144                  * Get status of PFS and its connections (-a for all PFSs)
 145                  */
 146                 ecode = cmd_remote_status(sel_path, all_opt);
 147         } else if (strcmp(av[0], "mkpfs") == 0) {
 148                 /*
 149                  * Create new PFS using pfs_type
 150                  */
 151         } else if (strcmp(av[0], "snapshot") == 0) {
 152                 /*
 153                  * Create snapshot with optional pfs_type and optional
 154                  * label override.
 155                  */
 156         } else if (strcmp(av[0], "helper") == 0) {
 157                 /*
 158                  * Typically run as a daemon, this multi-threaded helper
 159                  * subsystem manages socket communications for the
 160                  * filesystem.
 161                  */
 162                 ecode = cmd_helper(sel_path);
 163         } else {
 164                 fprintf(stderr, "Unrecognized command: %s\n", av[0]);
 165                 usage(1);
 166         }
 167
 168         /*
 169          * In DebugMode we may wind up starting several pthreads in the
 170          * original process, in which case we have to let them run and
 171          * not actually exit.
 172          */
 173         if (NormalExit) {
 174                 return (ecode);
 175         } else {
 176                 pthread_exit(NULL);
 177                 _exit(2);       /* NOT REACHED */
 178         }
 179 }
 180
 181 static
 182 void
 183 usage(int code)
 184 {
 185         fprintf(stderr,
 186                 "hammer2 [-s path] command...\n"
 187                 "    -s path            Select filesystem\n"
 188         );
 189         exit(code);
 190 }