source4/cluster/ctdb/tests/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         int daemon_mode;
  35 } ctdb_cmdline = {
  36         .nlist = NULL,
  37         .transport = "tcp",
  38         .myaddress = NULL,
  39         .self_connect = 0,
  40         .daemon_mode = 0
  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         { "daemon", 0, POPT_ARG_NONE, &ctdb_cmdline.daemon_mode, 0, "spawn a ctdb daemon", "boolean" },
  50         { NULL }
  51 };
  52
  53
  54 /*
  55   startup daemon side of ctdb according to command line options
  56  */
  57 struct ctdb_context *ctdb_cmdline_init(struct event_context *ev)
  58 {
  59         struct ctdb_context *ctdb;
  60         int ret;
  61
  62         if (ctdb_cmdline.nlist == NULL || ctdb_cmdline.myaddress == NULL) {
  63                 printf("You must provide a node list with --nlist and an address with --listen\n");
  64                 exit(1);
  65         }
  66
  67         /* initialise ctdb */
  68         ctdb = ctdb_init(ev);
  69         if (ctdb == NULL) {
  70                 printf("Failed to init ctdb\n");
  71                 exit(1);
  72         }
  73
  74         if (ctdb_cmdline.self_connect) {
  75                 ctdb_set_flags(ctdb, CTDB_FLAG_SELF_CONNECT);
  76         }
  77         if (ctdb_cmdline.daemon_mode) {
  78                 ctdb_set_flags(ctdb, CTDB_FLAG_DAEMON_MODE);
  79         }
  80
  81         ret = ctdb_set_transport(ctdb, ctdb_cmdline.transport);
  82         if (ret == -1) {
  83                 printf("ctdb_set_transport failed - %s\n", ctdb_errstr(ctdb));
  84                 exit(1);
  85         }
  86
  87         /* tell ctdb what address to listen on */
  88         ret = ctdb_set_address(ctdb, ctdb_cmdline.myaddress);
  89         if (ret == -1) {
  90                 printf("ctdb_set_address failed - %s\n", ctdb_errstr(ctdb));
  91                 exit(1);
  92         }
  93
  94         /* tell ctdb what nodes are available */
  95         ret = ctdb_set_nlist(ctdb, ctdb_cmdline.nlist);
  96         if (ret == -1) {
  97                 printf("ctdb_set_nlist failed - %s\n", ctdb_errstr(ctdb));
  98                 exit(1);
  99         }
 100
 101         return ctdb;
 102 }