2007-04-16 08:12:50 +04:00
/*
common commandline code to ctdb test tools
Copyright ( C ) Andrew Tridgell 2007
2007-05-31 07:50:53 +04:00
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 ,
2007-04-16 08:12:50 +04:00
but WITHOUT ANY WARRANTY ; without even the implied warranty of
2007-05-31 07:50:53 +04:00
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 0213 9 , USA .
2007-04-16 08:12:50 +04:00
*/
# include "includes.h"
# include "lib/events/events.h"
# include "system/filesys.h"
# include "popt.h"
2007-04-20 14:07:47 +04:00
# include "../include/ctdb.h"
# include "../include/ctdb_private.h"
2007-04-16 08:12:50 +04:00
/* Handle common command line options for ctdb test progs
*/
static struct {
2007-04-26 16:27:49 +04:00
const char * socketname ;
2007-04-19 10:27:56 +04:00
int torture ;
2007-05-05 11:18:43 +04:00
const char * events ;
2007-04-16 08:12:50 +04:00
} ctdb_cmdline = {
2007-04-26 16:27:49 +04:00
. socketname = CTDB_PATH ,
2007-04-30 00:42:23 +04:00
. torture = 0 ,
2007-04-16 08:12:50 +04:00
} ;
2007-05-05 11:18:43 +04:00
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 ;
}
}
2007-04-16 08:12:50 +04:00
struct poptOption popt_ctdb_cmdline [ ] = {
2007-05-05 11:18:43 +04:00
{ NULL , 0 , POPT_ARG_CALLBACK , ( void * ) ctdb_cmdline_callback } ,
2007-04-26 16:27:49 +04:00
{ " socket " , 0 , POPT_ARG_STRING , & ctdb_cmdline . socketname , 0 , " local socket name " , " filename " } ,
2007-04-17 16:13:06 +04:00
{ " debug " , ' d ' , POPT_ARG_INT , & LogLevel , 0 , " debug level " } ,
2007-04-19 10:27:56 +04:00
{ " torture " , 0 , POPT_ARG_NONE , & ctdb_cmdline . torture , 0 , " enable nastiness in library " , NULL } ,
2007-05-05 11:18:43 +04:00
{ " events " , 0 , POPT_ARG_STRING , NULL , OPT_EVENTSYSTEM , " event system " , NULL } ,
2007-04-16 08:12:50 +04:00
{ 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 ;
2007-05-10 01:55:46 +04:00
int ret ;
2007-04-16 08:12:50 +04:00
/* initialise ctdb */
ctdb = ctdb_init ( ev ) ;
if ( ctdb = = NULL ) {
printf ( " Failed to init ctdb \n " ) ;
exit ( 1 ) ;
}
2007-04-19 10:27:56 +04:00
if ( ctdb_cmdline . torture ) {
ctdb_set_flags ( ctdb , CTDB_FLAG_TORTURE ) ;
}
2007-04-16 08:12:50 +04:00
2007-04-26 16:27:49 +04:00
/* 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 ) ;
}
2007-04-16 08:12:50 +04:00
return ctdb ;
}
2007-04-20 14:07:47 +04:00
/*
startup a client only ctdb context
*/
2007-04-26 16:27:49 +04:00
struct ctdb_context * ctdb_cmdline_client ( struct event_context * ev )
2007-04-20 14:07:47 +04:00
{
struct ctdb_context * ctdb ;
int ret ;
/* initialise ctdb */
ctdb = ctdb_init ( ev ) ;
if ( ctdb = = NULL ) {
printf ( " Failed to init ctdb \n " ) ;
exit ( 1 ) ;
}
2007-04-26 16:27:49 +04:00
/* 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 ) ;
}
2007-04-20 14:07:47 +04:00
ret = ctdb_socket_connect ( ctdb ) ;
if ( ret ! = 0 ) {
DEBUG ( 0 , ( __location__ " Failed to connect to daemon \n " ) ) ;
talloc_free ( ctdb ) ;
return NULL ;
}
2007-05-23 09:18:30 +04:00
/* 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 " ) ) ;
2007-04-26 21:27:07 +04:00
talloc_free ( ctdb ) ;
return NULL ;
}
2007-04-20 14:07:47 +04:00
return ctdb ;
}