mirror of
https://github.com/samba-team/samba.git
synced 2024-12-25 23:21:54 +03:00
4f72a202d9
- fixed a valgrind error on failing to send a control - don't mark node dead when already disconnected - moved node list lock code into common code (This used to be ctdb commit bcc0432d0fea7ef223f82ccee81cf35c18144b1b)
134 lines
3.3 KiB
C
134 lines
3.3 KiB
C
/*
|
|
common commandline code to ctdb test tools
|
|
|
|
Copyright (C) Andrew Tridgell 2007
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
|
*/
|
|
|
|
#include "includes.h"
|
|
#include "lib/events/events.h"
|
|
#include "system/filesys.h"
|
|
#include "popt.h"
|
|
#include "../include/ctdb.h"
|
|
#include "../include/ctdb_private.h"
|
|
|
|
/* Handle common command line options for ctdb test progs
|
|
*/
|
|
|
|
static struct {
|
|
const char *socketname;
|
|
int torture;
|
|
const char *events;
|
|
} ctdb_cmdline = {
|
|
.socketname = CTDB_PATH,
|
|
.torture = 0,
|
|
};
|
|
|
|
enum {OPT_EVENTSYSTEM=1};
|
|
|
|
static void ctdb_cmdline_callback(poptContext con,
|
|
enum poptCallbackReason reason,
|
|
const struct poptOption *opt,
|
|
const char *arg, const void *data)
|
|
{
|
|
switch (opt->val) {
|
|
case OPT_EVENTSYSTEM:
|
|
event_set_default_backend(arg);
|
|
break;
|
|
}
|
|
}
|
|
|
|
|
|
struct poptOption popt_ctdb_cmdline[] = {
|
|
{ NULL, 0, POPT_ARG_CALLBACK, (void *)ctdb_cmdline_callback },
|
|
{ "socket", 0, POPT_ARG_STRING, &ctdb_cmdline.socketname, 0, "local socket name", "filename" },
|
|
{ "debug", 'd', POPT_ARG_INT, &LogLevel, 0, "debug level"},
|
|
{ "torture", 0, POPT_ARG_NONE, &ctdb_cmdline.torture, 0, "enable nastiness in library", NULL },
|
|
{ "events", 0, POPT_ARG_STRING, NULL, OPT_EVENTSYSTEM, "event system", NULL },
|
|
{ NULL }
|
|
};
|
|
|
|
|
|
/*
|
|
startup daemon side of ctdb according to command line options
|
|
*/
|
|
struct ctdb_context *ctdb_cmdline_init(struct event_context *ev)
|
|
{
|
|
struct ctdb_context *ctdb;
|
|
int ret;
|
|
|
|
/* initialise ctdb */
|
|
ctdb = ctdb_init(ev);
|
|
if (ctdb == NULL) {
|
|
printf("Failed to init ctdb\n");
|
|
exit(1);
|
|
}
|
|
|
|
if (ctdb_cmdline.torture) {
|
|
ctdb_set_flags(ctdb, CTDB_FLAG_TORTURE);
|
|
}
|
|
|
|
/* tell ctdb the socket address */
|
|
ret = ctdb_set_socketname(ctdb, ctdb_cmdline.socketname);
|
|
if (ret == -1) {
|
|
printf("ctdb_set_socketname failed - %s\n", ctdb_errstr(ctdb));
|
|
exit(1);
|
|
}
|
|
|
|
return ctdb;
|
|
}
|
|
|
|
|
|
/*
|
|
startup a client only ctdb context
|
|
*/
|
|
struct ctdb_context *ctdb_cmdline_client(struct event_context *ev)
|
|
{
|
|
struct ctdb_context *ctdb;
|
|
int ret;
|
|
|
|
/* initialise ctdb */
|
|
ctdb = ctdb_init(ev);
|
|
if (ctdb == NULL) {
|
|
printf("Failed to init ctdb\n");
|
|
exit(1);
|
|
}
|
|
|
|
/* tell ctdb the socket address */
|
|
ret = ctdb_set_socketname(ctdb, ctdb_cmdline.socketname);
|
|
if (ret == -1) {
|
|
printf("ctdb_set_socketname failed - %s\n", ctdb_errstr(ctdb));
|
|
exit(1);
|
|
}
|
|
|
|
ret = ctdb_socket_connect(ctdb);
|
|
if (ret != 0) {
|
|
DEBUG(0,(__location__ " Failed to connect to daemon\n"));
|
|
talloc_free(ctdb);
|
|
return NULL;
|
|
}
|
|
|
|
/* get our vnn */
|
|
ctdb->vnn = ctdb_ctrl_getvnn(ctdb, timeval_zero(), CTDB_CURRENT_NODE);
|
|
if (ctdb->vnn == (uint32_t)-1) {
|
|
DEBUG(0,(__location__ " Failed to get ctdb vnn\n"));
|
|
talloc_free(ctdb);
|
|
return NULL;
|
|
}
|
|
|
|
return ctdb;
|
|
}
|