source/cluster/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 3 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 #include "../include/ctdb.h"
  26 #include "../include/ctdb_private.h"
  27
  28 /* Handle common command line options for ctdb test progs
  29  */
  30
  31 static struct {
  32         const char *nlist;
  33         const char *transport;
  34         const char *myaddress;
  35         int self_connect;
  36         const char *db_dir;
  37         int torture;
  38 } ctdb_cmdline = {
  39         .nlist = NULL,
  40         .transport = "tcp",
  41         .myaddress = NULL,
  42         .self_connect = 0,
  43         .db_dir = NULL,
  44         .torture = 0
  45 };
  46
  47
  48 struct poptOption popt_ctdb_cmdline[] = {
  49         { "nlist", 0, POPT_ARG_STRING, &ctdb_cmdline.nlist, 0, "node list file", "filename" },
  50         { "listen", 0, POPT_ARG_STRING, &ctdb_cmdline.myaddress, 0, "address to listen on", "address" },
  51         { "transport", 0, POPT_ARG_STRING, &ctdb_cmdline.transport, 0, "protocol transport", NULL },
  52         { "self-connect", 0, POPT_ARG_NONE, &ctdb_cmdline.self_connect, 0, "enable self connect", "boolean" },
  53         { "debug", 'd', POPT_ARG_INT, &LogLevel, 0, "debug level"},
  54         { "dbdir", 0, POPT_ARG_STRING, &ctdb_cmdline.db_dir, 0, "directory for the tdb files", NULL },
  55         { "torture", 0, POPT_ARG_NONE, &ctdb_cmdline.torture, 0, "enable nastiness in library", NULL },
  56         { NULL }
  57 };
  58
  59
  60 /*
  61   startup daemon side of ctdb according to command line options
  62  */
  63 struct ctdb_context *ctdb_cmdline_init(struct event_context *ev)
  64 {
  65         struct ctdb_context *ctdb;
  66         int ret;
  67
  68         if (ctdb_cmdline.nlist == NULL || ctdb_cmdline.myaddress == NULL) {
  69                 printf("You must provide a node list with --nlist and an address with --listen\n");
  70                 exit(1);
  71         }
  72
  73         /* initialise ctdb */
  74         ctdb = ctdb_init(ev);
  75         if (ctdb == NULL) {
  76                 printf("Failed to init ctdb\n");
  77                 exit(1);
  78         }
  79
  80         if (ctdb_cmdline.self_connect) {
  81                 ctdb_set_flags(ctdb, CTDB_FLAG_SELF_CONNECT);
  82         }
  83         if (ctdb_cmdline.torture) {
  84                 ctdb_set_flags(ctdb, CTDB_FLAG_TORTURE);
  85         }
  86
  87         ret = ctdb_set_transport(ctdb, ctdb_cmdline.transport);
  88         if (ret == -1) {
  89                 printf("ctdb_set_transport failed - %s\n", ctdb_errstr(ctdb));
  90                 exit(1);
  91         }
  92
  93         /* tell ctdb what address to listen on */
  94         ret = ctdb_set_address(ctdb, ctdb_cmdline.myaddress);
  95         if (ret == -1) {
  96                 printf("ctdb_set_address failed - %s\n", ctdb_errstr(ctdb));
  97                 exit(1);
  98         }
  99
 100         /* tell ctdb what nodes are available */
 101         ret = ctdb_set_nlist(ctdb, ctdb_cmdline.nlist);
 102         if (ret == -1) {
 103                 printf("ctdb_set_nlist failed - %s\n", ctdb_errstr(ctdb));
 104                 exit(1);
 105         }
 106
 107         ret = ctdb_set_tdb_dir(ctdb, ctdb_cmdline.db_dir);
 108         if (ret == -1) {
 109                 printf("ctdb_set_tdb_dir failed - %s\n", ctdb_errstr(ctdb));
 110                 exit(1);
 111         }
 112
 113         return ctdb;
 114 }
 115
 116
 117 /*
 118   startup a client only ctdb context
 119  */
 120 struct ctdb_context *ctdb_cmdline_client(struct event_context *ev, const char *ctdb_socket)
 121 {
 122         struct ctdb_context *ctdb;
 123         int ret;
 124
 125         /* initialise ctdb */
 126         ctdb = ctdb_init(ev);
 127         if (ctdb == NULL) {
 128                 printf("Failed to init ctdb\n");
 129                 exit(1);
 130         }
 131
 132         ctdb->daemon.name = talloc_strdup(ctdb, ctdb_socket);
 133
 134         ret = ctdb_socket_connect(ctdb);
 135         if (ret != 0) {
 136                 DEBUG(0,(__location__ " Failed to connect to daemon\n"));
 137                 talloc_free(ctdb);
 138                 return NULL;
 139         }
 140
 141         return ctdb;
 142 }