static void dump_services( int fd )
{
unsigned u ;
/*
* Dump the current configuration (services + defaults)
*/
Sprint( fd, "Services + defaults:\n" ) ;
sc_dump( DEFAULTS( ps ), fd, 0, TRUE ) ;
for ( u = 0 ; u < pset_count( SERVICES( ps ) ) ; u++ )
svc_dump( SP( pset_pointer( SERVICES( ps ), u ) ), fd ) ;
}
void svc_dump( const struct service *sp, int fd )
{
tabprint( fd, 0, "Service = %s\n", SC_NAME( SVC_CONF( sp ) ) ) ;
tabprint( fd, 1, "State = %s\n",
nv_get_name( service_states, (int) SVC_STATE(sp) ) ) ;
sc_dump( SVC_CONF( sp ), fd, 1, FALSE ) ;
if ( SVC_IS_ACTIVE(sp) )
{
tabprint( fd, 1, "running servers = %d\n", SVC_RUNNING_SERVERS(sp) ) ;
tabprint( fd, 1, "retry servers = %d\n", SVC_RETRIES(sp) ) ;
tabprint( fd, 1, "attempts = %d\n", SVC_ATTEMPTS(sp) ) ;
tabprint( fd, 1, "service fd = %d\n", SVC_FD(sp) ) ;
}
Sputchar( fd, '\n' ) ;
}
int
do_configure(config_object_t *config, const char *config_file)
{
FILE *fp = NULL;
char message[80];
char tmp_filename[4096];
int tmp_fd = -1;
if (sc_parse(config, config_file) != 0) {
printf("Parsing of %s failed.\n", config_file);
if (yesno("Start from scratch", 0) == 0) {
return 1;
}
}
plugin_path_configure(config);
listener_configure(config);
backend_configure(config);
printf("\nConfiguration complete.\n\n");
printf("=== Begin Configuration ===\n");
sc_dump(config, stdout);
printf("=== End Configuration ===\n");
snprintf(message, sizeof(message), "Replace %s with the above",
config_file);
if (yesno(message, 0) == 0) {
return 1;
}
snprintf(tmp_filename, sizeof(tmp_filename),
"%s.XXXXXX", config_file);
tmp_fd = mkstemp(tmp_filename);
if (tmp_fd < 0) {
perror("fopen");
printf("Failed to write configuration file!\n");
return 1;
}
fp = fdopen(tmp_fd, "w+");
if (fp == NULL)
goto out_fail;
sc_dump(config, fp);
if (rename(tmp_filename, config_file) < 0) {
perror("rename");
goto out_fail;
}
fclose(fp);
close(tmp_fd);
return 0;
out_fail:
if (fp)
fclose(fp);
if (tmp_fd >= 0)
close(tmp_fd);
if (strlen(tmp_filename))
unlink(tmp_filename);
printf("Failed to write configuration file!\n");
return 1;
}
int
main(int argc, char **argv)
{
char val[4096];
char listener_name[80];
char backend_name[80];
const char *config_file = DEFAULT_CONFIG_FILE;
config_object_t *config = NULL;
map_object_t *map = NULL;
const listener_plugin_t *lp;
const backend_plugin_t *p;
listener_context_t listener_ctx = NULL;
backend_context_t backend_ctx = NULL;
int debug_set = 0, foreground = 0, wait_for_init = 0;
int opt, configure = 0;
config = sc_init();
map = map_init();
if (!config || !map) {
perror("malloc");
return -1;
}
while ((opt = getopt(argc, argv, "Ff:d:cwlh")) != EOF) {
switch(opt) {
case 'F':
printf("Background mode disabled\n");
foreground = 1;
break;
case 'f':
printf("Using %s\n", optarg);
config_file = optarg;
break;
case 'd':
debug_set = atoi(optarg);
break;
case 'c':
configure = 1;
break;
case 'w':
wait_for_init = 1;
break;
case 'l':
plugin_dump();
return 0;
case 'h':
case '?':
usage();
return 0;
default:
return -1;
}
}
if (configure) {
return do_configure(config, config_file);
}
if (sc_parse(config, config_file) != 0) {
printf("Failed to parse %s\n", config_file);
return -1;
}
if (debug_set) {
snprintf(val, sizeof(val), "%d", debug_set);
sc_set(config, "fence_virtd/@debug", val);
} else {
if (sc_get(config, "fence_virtd/@debug", val, sizeof(val))==0)
debug_set = atoi(val);
}
dset(debug_set);
if (!foreground) {
if (sc_get(config, "fence_virtd/@foreground",
val, sizeof(val)) == 0)
foreground = atoi(val);
}
if (!wait_for_init) {
if (sc_get(config, "fence_virtd/@wait_for_init",
val, sizeof(val)) == 0)
wait_for_init = atoi(val);
if (!wait_for_init) {
/* XXX compat */
if (sc_get(config, "fence_virtd/@wait_for_backend",
val, sizeof(val)) == 0)
wait_for_init = atoi(val);
}
}
if (dget() > 3)
sc_dump(config, stdout);
if (sc_get(config, "fence_virtd/@backend", backend_name,
sizeof(backend_name))) {
printf("Failed to determine backend.\n");
printf("%s\n", val);
return -1;
}
dbg_printf(1, "Backend plugin: %s\n", backend_name);
if (sc_get(config, "fence_virtd/@listener", listener_name,
sizeof(listener_name))) {
printf("Failed to determine backend.\n");
printf("%s\n", val);
return -1;
}
dbg_printf(1, "Listener plugin: %s\n", listener_name);
#ifdef _MODULE
if (sc_get(config, "fence_virtd/@module_path", val,
sizeof(val))) {
#ifdef MODULE_PATH
snprintf(val, sizeof(val), MODULE_PATH);
#else
printf("Failed to determine module path.\n");
return -1;
#endif
}
dbg_printf(1, "Searching %s for plugins...\n", val);
opt = plugin_search(val);
if (opt > 0) {
dbg_printf(1, "%d plugins found\n", opt);
} else {
printf("No plugins found\n");
return 1;
}
#endif
if (dget() > 3)
plugin_dump();
lp = plugin_find_listener(listener_name);
if (!lp) {
printf("Could not find listener \"%s\"\n", listener_name);
return 1;
}
p = plugin_find_backend(backend_name);
if (!p) {
printf("Could not find backend \"%s\"\n", backend_name);
return 1;
}
daemon_init(basename(argv[0]), foreground);
signal(SIGINT, exit_handler);
signal(SIGTERM, exit_handler);
signal(SIGQUIT, exit_handler);
syslog(LOG_NOTICE, "fence_virtd starting. Listener: %s Backend: %s", backend_name, listener_name);
while (p->init(&backend_ctx, config) < 0) {
if (!wait_for_init) {
if (foreground) {
printf("Backend plugin %s failed to initialize\n",
backend_name);
}
syslog(LOG_ERR,
"Backend plugin %s failed to initialize\n",
backend_name);
return 1;
}
sleep(5);
}
if (map_load(map, config) < 0) {
syslog(LOG_WARNING, "Failed to load static maps\n");
}
/* only client we have now is mcast (fence_xvm behavior) */
while (lp->init(&listener_ctx, p->callbacks, config, map,
backend_ctx) != 0) {
if (!wait_for_init) {
if (foreground) {
printf("Listener plugin %s failed to initialize\n",
listener_name);
}
syslog(LOG_ERR,
"Listener plugin %s failed to initialize\n",
listener_name);
return 1;
}
sleep(5);
}
while (run && lp->dispatch(listener_ctx, NULL) >= 0);
syslog(LOG_NOTICE, "fence_virtd shutting down");
map_release(map);
sc_release(config);
lp->cleanup(listener_ctx);
p->cleanup(backend_ctx);
daemon_cleanup();
return 0;
}
