cmdline.c

打魔兽战网的都知道他是什么

/*
 * Copyright (C) 2005           Dizzy       
 * Copyright (C) 2005           Olaf Freyer ([e-mail])
 *
 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */
#include "common/setup_before.h"
#include "setup.h"

#ifdef HAVE_STDDEF_H
# include <stddef.h>
#else
# ifndef NULL
#  define NULL ((void *)0)
# endif
#endif
#include <stdio.h>
#ifdef HAVE_STRING_H
# include <string.h>
#else
# ifdef HAVE_STRINGS_H
#  include <strings.h>
# endif
# ifdef HAVE_MEMORY_H
#  include <memory.h>
# endif
#endif
#include "compat/memset.h"
#ifdef WIN32
# include "win32/service.h"
#endif
#include "common/conf.h"
#include "common/xalloc.h"
#include "version.h"
#include "cmdline.h"
#include "compat/strcasecmp.h"
#include "common/eventlog.h"
#include "common/setup_after.h"

static struct {
#ifdef DO_DAEMONIZE
	unsigned foreground;
#endif
	const char *preffile;
	const char *logfile;
	unsigned debug;
} cmdline_config;


static unsigned exitflag;
static const char *progname;


static int conf_set_preffile(const char *valstr);
static int conf_setdef_preffile(void);

static int conf_set_logfile(const char * valstr);
static int conf_setdef_logfile(void);

static int conf_set_help(const char * valstr);
static int conf_setdef_help(void);

static int conf_set_version(const char * valstr);
static int conf_setdef_version(void);

static int conf_set_foreground(const char * valstr);
static int conf_setdef_foreground(void);

static int conf_set_debug(const char * valstr);
static int conf_setdef_debug(void);

#ifdef WIN32
static int conf_set_service(char const * valstr);
static int conf_setdef_service(void);

static int conf_set_servaction(const char * valstr);
static int conf_setdef_servaction(void);
#endif


static t_conf_entry conftab[]={
	{ "c",          conf_set_preffile,      NULL, conf_setdef_preffile},
	{ "config",     conf_set_preffile,      NULL, conf_setdef_preffile},
	{ "l",          conf_set_logfile,       NULL, conf_setdef_logfile   },
	{ "log",        conf_set_logfile,       NULL, conf_setdef_logfile   },
	{ "h",          conf_set_help,          NULL, conf_setdef_help      },
	{ "help",       conf_set_help,          NULL, conf_setdef_help      },
	{ "usage",      conf_set_help,          NULL, conf_setdef_help      },
	{ "v",          conf_set_version,       NULL, conf_setdef_version   },
	{ "version",    conf_set_version,       NULL, conf_setdef_version   },
#ifdef DO_DAEMONIZE
	{ "f",	        conf_set_foreground,    NULL, conf_setdef_foreground},
	{ "foreground", conf_set_foreground,    NULL, conf_setdef_foreground},
#endif
	{ "D",          conf_set_debug,         NULL, conf_setdef_debug     },
	{ "debug",      conf_set_debug,         NULL, conf_setdef_debug     },
#ifdef WIN32
	{ "service",    conf_set_service,       NULL, conf_setdef_service   },
	{ "s",		conf_set_servaction,	NULL, conf_setdef_servaction},
#endif
	{ NULL,         NULL,                   NULL, NULL                  }
};


extern int cmdline_load(int argc, char** argv)
{
	int res;

	if (argc<1 || !argv || !argv[0]) {
		fprintf(stderr,"bad arguments\n");
		return -1;
	}

	exitflag = 0;
	progname = argv[0];

	res = conf_load_cmdline(argc, argv, conftab);
	if (res < 0) return -1;
	return exitflag ? 0 : 1;
}

extern void cmdline_unload(void)
{
	conf_unload(conftab);
}



static void usage(void)
{
	fprintf(stderr,
		"Usage: %s [<options>]\n"
		"    -c FILE, --config=FILE   use FILE as configuration file (default is " D2DBS_DEFAULT_CONF_FILE ")\n"
		"    -l FILE, --log=FILE      set log to FILE\n"
#ifdef DO_DAEMONIZE
		"    -f, --foreground:        don't daemonize\n"
#endif
		"    -D, --debug:             run in debug mode (run in foreground and log to stdout)\n"
		"    -h, --help, --usage      show this information and exit\n"
		"    -v, --version:           print version number and exit\n"
#ifdef WIN32
		"    Running as service functions:\n"
		"    --service                run as service\n"
		"    -s install               install service\n"
		"    -s uninstall             uninstall service\n"
#endif	    
		"\n"
		"Notes:\n"
		"	1.You should always use absolute path here for all FILE names\n\n",
		progname);
}

#ifdef DO_DAEMONIZE
extern int cmdline_get_foreground(void)
{
	return cmdline_config.foreground;
}

static int conf_set_foreground(const char *valstr)
{
	return conf_set_bool(&cmdline_config.foreground, valstr, 0);
}

static int conf_setdef_foreground(void)
{
	return conf_set_bool(&cmdline_config.foreground, NULL, 0);
}
#endif


extern const char* cmdline_get_preffile(void)
{
	return cmdline_config.preffile;
}

static int conf_set_preffile(const char *valstr)
{
	return conf_set_str(&cmdline_config.preffile, valstr, NULL);
}

static int conf_setdef_preffile(void)
{
	return conf_set_str(&cmdline_config.preffile, NULL, D2DBS_DEFAULT_CONF_FILE);
}


extern const char* cmdline_get_logfile(void)
{
	return cmdline_config.logfile;
}

static int conf_set_logfile(const char *valstr)
{
	return conf_set_str(&cmdline_config.logfile, valstr, NULL);
}

static int conf_setdef_logfile(void)
{
	return conf_set_str(&cmdline_config.logfile, NULL, NULL);
}


static int conf_set_debug(const char *valstr)
{
	conf_set_bool(&cmdline_config.debug, valstr, 0);
	if (cmdline_config.debug) eventlog_set_debugmode(1);
#ifdef DO_DAEMONIZE
	cmdline_config.foreground = 1;
#endif
	return 0;
}

static int conf_setdef_debug(void)
{
	return conf_set_bool(&cmdline_config.debug, NULL, 0);
}


static int conf_set_help(const char *valstr)
{
	unsigned tmp = 0;

	conf_set_bool(&tmp, valstr, 0);
	if (tmp) {
		usage();
		exitflag = 1;
	}

	return 0;
}

static int conf_setdef_help(void)
{
	return 0;
}


static int conf_set_version(const char *valstr)
{
	unsigned tmp = 0;

	conf_set_bool(&tmp, valstr, 0);
	if (tmp) {
		printf(D2DBS_VERSION"\n");
		exitflag = 1;
	}

	return 0;
}

static int conf_setdef_version(void)
{
	return 0;
}


#ifdef WIN32
static int conf_set_service(const char *valstr)
{
	unsigned tmp = 0;

	conf_set_bool(&tmp, valstr, 0);
	if (tmp) {
		Win32_ServiceRun();
		exitflag = 1;
	}

	return 0;
}

static int conf_setdef_service(void)
{
	return 0;
}


static int conf_set_servaction(const char *valstr)
{
	const char* tmp = NULL;

	conf_set_str(&tmp, valstr, NULL);

	if (tmp) {
		if (!strcasecmp(tmp, "install")) {
			fprintf(stderr, "Installing service");
			Win32_ServiceInstall();
		} else if (!strcasecmp(tmp, "uninstall")) {
			fprintf(stderr, "Uninstalling service");
			Win32_ServiceUninstall();
		} else {
			fprintf(stderr, "Unknown service action '%s'\n", tmp);
		}

		exitflag = 1;
		xfree((void *)tmp);
	}

	return 0;
}

static int conf_setdef_servaction(void)
{
	return 0;
}

#endif