ctdb/common/cmdline.c

   1 /*
   2    common commandline code to ctdb test tools
   3
   4    Copyright (C) Andrew Tridgell  2007
   5
   6    This library is free software; you can redistribute it and/or
   7    modify it under the terms of the GNU Lesser General Public
   8    License as published by the Free Software Foundation; either
   9    version 2 of the License, or (at your option) any later version.
  10
  11    This library is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14    Lesser General Public License for more details.
  15
  16    You should have received a copy of the GNU Lesser General Public
  17    License along with this library; if not, write to the Free Software
  18    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19 */
  20
  21 #include "includes.h"
  22 #include "lib/events/events.h"
  23 #include "system/filesys.h"
  24 #include "popt.h"
  25
  26 /* Handle common command line options for ctdb test progs
  27  */
  28
  29 static struct {
  30         const char *nlist;
  31         const char *transport;
  32         const char *myaddress;
  33         int self_connect;
  34         const char *db_dir;
  35 } ctdb_cmdline = {
  36         .nlist = NULL,
  37         .transport = "tcp",
  38         .myaddress = NULL,
  39         .self_connect = 0,
  40         .db_dir = NULL
  41 };
  42
  43
  44 struct poptOption popt_ctdb_cmdline[] = {
  45         { "nlist", 0, POPT_ARG_STRING, &ctdb_cmdline.nlist, 0, "node list file", "filename" },
  46         { "listen", 0, POPT_ARG_STRING, &ctdb_cmdline.myaddress, 0, "address to listen on", "address" },
  47         { "transport", 0, POPT_ARG_STRING, &ctdb_cmdline.transport, 0, "protocol transport", NULL },
  48         { "self-connect", 0, POPT_ARG_NONE, &ctdb_cmdline.self_connect, 0, "enable self connect", "boolean" },
  49         { "debug", 'd', POPT_ARG_INT, &LogLevel, 0, "debug level"},
  50         { "dbdir", 0, POPT_ARG_STRING, &ctdb_cmdline.db_dir, 0, "directory for the tdb files", NULL },
  51         { NULL }
  52 };
  53
  54
  55 /*
  56   startup daemon side of ctdb according to command line options
  57  */
  58 struct ctdb_context *ctdb_cmdline_init(struct event_context *ev)
  59 {
  60         struct ctdb_context *ctdb;
  61         int ret;
  62
  63         if (ctdb_cmdline.nlist == NULL || ctdb_cmdline.myaddress == NULL) {
  64                 printf("You must provide a node list with --nlist and an address with --listen\n");
  65                 exit(1);
  66         }
  67
  68         /* initialise ctdb */
  69         ctdb = ctdb_init(ev);
  70         if (ctdb == NULL) {
  71                 printf("Failed to init ctdb\n");
  72                 exit(1);
  73         }
  74
  75         if (ctdb_cmdline.self_connect) {
  76                 ctdb_set_flags(ctdb, CTDB_FLAG_SELF_CONNECT);
  77         }
  78
  79         ret = ctdb_set_transport(ctdb, ctdb_cmdline.transport);
  80         if (ret == -1) {
  81                 printf("ctdb_set_transport failed - %s\n", ctdb_errstr(ctdb));
  82                 exit(1);
  83         }
  84
  85         /* tell ctdb what address to listen on */
  86         ret = ctdb_set_address(ctdb, ctdb_cmdline.myaddress);
  87         if (ret == -1) {
  88                 printf("ctdb_set_address failed - %s\n", ctdb_errstr(ctdb));
  89                 exit(1);
  90         }
  91
  92         /* tell ctdb what nodes are available */
  93         ret = ctdb_set_nlist(ctdb, ctdb_cmdline.nlist);
  94         if (ret == -1) {
  95                 printf("ctdb_set_nlist failed - %s\n", ctdb_errstr(ctdb));
  96                 exit(1);
  97         }
  98
  99         ret = ctdb_set_tdb_dir(ctdb, ctdb_cmdline.db_dir);
 100         if (ret == -1) {
 101                 printf("ctdb_set_tdb_dir failed - %s\n", ctdb_errstr(ctdb));
 102                 exit(1);
 103         }
 104
 105         return ctdb;
 106 }