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 } ctdb_cmdline = {
  35         .nlist = NULL,
  36         .transport = "tcp",
  37         .myaddress = NULL,
  38         .self_connect = 0,
  39 };
  40
  41
  42 struct poptOption popt_ctdb_cmdline[] = {
  43         { "nlist", 0, POPT_ARG_STRING, &ctdb_cmdline.nlist, 0, "node list file", "filename" },
  44         { "listen", 0, POPT_ARG_STRING, &ctdb_cmdline.myaddress, 0, "address to listen on", "address" },
  45         { "transport", 0, POPT_ARG_STRING, &ctdb_cmdline.transport, 0, "protocol transport", NULL },
  46         { "self-connect", 0, POPT_ARG_NONE, &ctdb_cmdline.self_connect, 0, "enable self connect", "boolean" },
  47         { "debug", 'd', POPT_ARG_INT, &LogLevel, 0, "debug level"},
  48         { NULL }
  49 };
  50
  51
  52 /*
  53   startup daemon side of ctdb according to command line options
  54  */
  55 struct ctdb_context *ctdb_cmdline_init(struct event_context *ev)
  56 {
  57         struct ctdb_context *ctdb;
  58         int ret;
  59
  60         if (ctdb_cmdline.nlist == NULL || ctdb_cmdline.myaddress == NULL) {
  61                 printf("You must provide a node list with --nlist and an address with --listen\n");
  62                 exit(1);
  63         }
  64
  65         /* initialise ctdb */
  66         ctdb = ctdb_init(ev);
  67         if (ctdb == NULL) {
  68                 printf("Failed to init ctdb\n");
  69                 exit(1);
  70         }
  71
  72         if (ctdb_cmdline.self_connect) {
  73                 ctdb_set_flags(ctdb, CTDB_FLAG_SELF_CONNECT);
  74         }
  75
  76         ret = ctdb_set_transport(ctdb, ctdb_cmdline.transport);
  77         if (ret == -1) {
  78                 printf("ctdb_set_transport failed - %s\n", ctdb_errstr(ctdb));
  79                 exit(1);
  80         }
  81
  82         /* tell ctdb what address to listen on */
  83         ret = ctdb_set_address(ctdb, ctdb_cmdline.myaddress);
  84         if (ret == -1) {
  85                 printf("ctdb_set_address failed - %s\n", ctdb_errstr(ctdb));
  86                 exit(1);
  87         }
  88
  89         /* tell ctdb what nodes are available */
  90         ret = ctdb_set_nlist(ctdb, ctdb_cmdline.nlist);
  91         if (ret == -1) {
  92                 printf("ctdb_set_nlist failed - %s\n", ctdb_errstr(ctdb));
  93                 exit(1);
  94         }
  95
  96         return ctdb;
  97 }