/*--------------------------------------------------------- */
int main(int argc, char **argv)
{
int running;
int result = -1;
clish_shell_t *shell = NULL;
/* Command line options */
const char *socket_path = KONFD_SOCKET_PATH;
bool_t lockless = BOOL_FALSE;
bool_t stop_on_error = BOOL_FALSE;
bool_t interactive = BOOL_TRUE;
bool_t quiet = BOOL_FALSE;
bool_t utf8 = BOOL_FALSE;
bool_t bit8 = BOOL_FALSE;
bool_t log = BOOL_FALSE;
int log_facility = LOG_LOCAL0;
bool_t dryrun = BOOL_FALSE;
bool_t dryrun_config = BOOL_FALSE;
const char *xml_path = getenv("CLISH_PATH");
const char *view = getenv("CLISH_VIEW");
const char *viewid = getenv("CLISH_VIEWID");
const char *xslt_file = NULL;
FILE *outfd = stdout;
bool_t istimeout = BOOL_FALSE;
int timeout = 0;
bool_t cmd = BOOL_FALSE; /* -c option */
lub_list_t *cmds; /* Commands defined by -c */
lub_list_node_t *iter;
const char *histfile = "~/.clish_history";
char *histfile_expanded = NULL;
unsigned int histsize = 50;
clish_sym_t *sym = NULL;
/* Signal vars */
struct sigaction sigpipe_act;
sigset_t sigpipe_set;
static const char *shortopts = "hvs:ledx:w:i:bqu8oO:kt:c:f:z:p:";
#ifdef HAVE_GETOPT_LONG
static const struct option longopts[] = {
{"help", 0, NULL, 'h'},
{"version", 0, NULL, 'v'},
{"socket", 1, NULL, 's'},
{"lockless", 0, NULL, 'l'},
{"stop-on-error", 0, NULL, 'e'},
{"dry-run", 0, NULL, 'd'},
{"xml-path", 1, NULL, 'x'},
{"view", 1, NULL, 'w'},
{"viewid", 1, NULL, 'i'},
{"background", 0, NULL, 'b'},
{"quiet", 0, NULL, 'q'},
{"utf8", 0, NULL, 'u'},
{"8bit", 0, NULL, '8'},
{"log", 0, NULL, 'o'},
{"facility", 1, NULL, 'O'},
{"check", 0, NULL, 'k'},
{"timeout", 1, NULL, 't'},
{"command", 1, NULL, 'c'},
{"histfile", 1, NULL, 'f'},
{"histsize", 1, NULL, 'z'},
{"xslt", 1, NULL, 'p'},
{NULL, 0, NULL, 0}
};
#endif
/* Ignore SIGPIPE */
sigemptyset(&sigpipe_set);
sigaddset(&sigpipe_set, SIGPIPE);
sigpipe_act.sa_flags = 0;
sigpipe_act.sa_mask = sigpipe_set;
sigpipe_act.sa_handler = &sighandler;
sigaction(SIGPIPE, &sigpipe_act, NULL);
#if HAVE_LOCALE_H
/* Set current locale */
setlocale(LC_ALL, "");
#endif
/* Var initialization */
cmds = lub_list_new(NULL);
/* Parse command line options */
while(1) {
int opt;
#ifdef HAVE_GETOPT_LONG
opt = getopt_long(argc, argv, shortopts, longopts, NULL);
#else
opt = getopt(argc, argv, shortopts);
#endif
if (-1 == opt)
break;
switch (opt) {
case 's':
socket_path = optarg;
break;
case 'l':
lockless = BOOL_TRUE;
break;
case 'e':
stop_on_error = BOOL_TRUE;
break;
case 'b':
interactive = BOOL_FALSE;
break;
case 'q':
quiet = BOOL_TRUE;
break;
case 'u':
utf8 = BOOL_TRUE;
break;
case '8':
bit8 = BOOL_TRUE;
break;
case 'o':
log = BOOL_TRUE;
break;
case 'O':
if (lub_log_facility(optarg, &log_facility)) {
fprintf(stderr, "Error: Illegal syslog facility %s.\n", optarg);
help(-1, argv[0]);
goto end;
}
break;
case 'd':
dryrun = BOOL_TRUE;
break;
case 'x':
xml_path = optarg;
break;
case 'w':
view = optarg;
break;
case 'i':
viewid = optarg;
break;
case 'k':
lockless = BOOL_TRUE;
dryrun = BOOL_TRUE;
dryrun_config = BOOL_TRUE;
break;
case 't':
istimeout = BOOL_TRUE;
timeout = atoi(optarg);
break;
case 'c': {
char *str;
cmd = BOOL_TRUE;
quiet = BOOL_TRUE;
str = strdup(optarg);
lub_list_add(cmds, str);
}
break;
case 'f':
if (!strcmp(optarg, "/dev/null"))
histfile = NULL;
else
histfile = optarg;
break;
case 'z': {
int itmp = 0;
itmp = atoi(optarg);
if (itmp <= 0) {
fprintf(stderr, "Error: Illegal histsize option value.\n");
help(-1, argv[0]);
goto end;
}
histsize = itmp;
}
break;
case 'p':
#ifdef HAVE_LIB_LIBXSLT
xslt_file = optarg;
#else
fprintf(stderr, "Error: The klish was built without XSLT support.\n");
goto end;
#endif
break;
case 'h':
help(0, argv[0]);
exit(0);
break;
case 'v':
version(VERSION);
exit(0);
break;
default:
help(-1, argv[0]);
goto end;
break;
}
}
/* Validate command line options */
if (utf8 && bit8) {
fprintf(stderr, "Error: The -u and -8 options can't be used together.\n");
goto end;
}
/* Create shell instance */
if (quiet) {
FILE *tmpfd = NULL;
if ((tmpfd = fopen("/dev/null", "w")))
outfd = tmpfd;
}
shell = clish_shell_new(NULL, outfd, stop_on_error);
if (!shell) {
fprintf(stderr, "Error: Can't run clish.\n");
goto end;
}
/* Load the XML files */
clish_xmldoc_start();
if (clish_shell_load_scheme(shell, xml_path, xslt_file))
goto end;
/* Set communication to the konfd */
clish_shell__set_socket(shell, socket_path);
/* Set lockless mode */
if (lockless)
clish_shell__set_lockfile(shell, NULL);
/* Set interactive mode */
if (!interactive)
clish_shell__set_interactive(shell, interactive);
/* Set startup view */
if (view)
clish_shell__set_startup_view(shell, view);
/* Set startup viewid */
if (viewid)
clish_shell__set_startup_viewid(shell, viewid);
/* Set UTF-8 or 8-bit mode */
if (utf8 || bit8)
clish_shell__set_utf8(shell, utf8);
else {
#if HAVE_LANGINFO_CODESET
/* Autodetect encoding */
if (!strcmp(nl_langinfo(CODESET), "UTF-8"))
clish_shell__set_utf8(shell, BOOL_TRUE);
#else
/* The default is 8-bit if locale is not supported */
clish_shell__set_utf8(shell, BOOL_FALSE);
#endif
}
/* Set logging */
if (log) {
clish_shell__set_log(shell, log);
clish_shell__set_facility(shell, log_facility);
}
/* Set dry-run */
if (dryrun)
clish_shell__set_dryrun(shell, dryrun);
/* Set idle timeout */
if (istimeout)
clish_shell__set_timeout(shell, timeout);
/* Set history settings */
clish_shell__stifle_history(shell, histsize);
if (histfile)
histfile_expanded = lub_system_tilde_expand(histfile);
if (histfile_expanded)
clish_shell__restore_history(shell, histfile_expanded);
/* Load plugins, link aliases and check access rights */
if (clish_shell_prepare(shell) < 0)
goto end;
/* Dryrun config and log hooks */
if (dryrun_config) {
if ((sym = clish_shell_get_hook(shell, CLISH_SYM_TYPE_CONFIG)))
clish_sym__set_permanent(sym, BOOL_FALSE);
if ((sym = clish_shell_get_hook(shell, CLISH_SYM_TYPE_LOG)))
clish_sym__set_permanent(sym, BOOL_FALSE);
}
#ifdef DEBUG
clish_shell_dump(shell);
#endif
/* Set source of command stream: files or interactive tty */
if(optind < argc) {
int i;
/* Run the commands from the files */
for (i = argc - 1; i >= optind; i--)
clish_shell_push_file(shell, argv[i], stop_on_error);
} else {
/* The interactive shell */
int tmpfd = dup(fileno(stdin));
#ifdef FD_CLOEXEC
fcntl(tmpfd, F_SETFD, fcntl(tmpfd, F_GETFD) | FD_CLOEXEC);
#endif
clish_shell_push_fd(shell, fdopen(tmpfd, "r"), stop_on_error);
}
/* Execute startup */
running = clish_shell_startup(shell);
if (running) {
fprintf(stderr, "Error: Can't startup clish.\n");
goto end;
}
if (cmd) {
/* Iterate cmds */
for(iter = lub_list__get_head(cmds);
iter; iter = lub_list_node__get_next(iter)) {
char *str = (char *)lub_list_node__get_data(iter);
result = clish_shell_forceline(shell, str, NULL);
if (stop_on_error && result)
break;
}
} else {
/* Main loop */
result = clish_shell_loop(shell);
}
end:
/* Cleanup */
if (shell) {
if (histfile_expanded) {
clish_shell__save_history(shell, histfile_expanded);
free(histfile_expanded);
}
clish_shell_delete(shell);
}
if (quiet && (outfd != stdout))
fclose(outfd);
/* Delete each cmds element */
while ((iter = lub_list__get_head(cmds))) {
lub_list_del(cmds, iter);
free(lub_list_node__get_data(iter));
lub_list_node_free(iter);
}
lub_list_free(cmds);
/* Stop XML engine */
clish_xmldoc_stop();
return result;
}
/* Parse command line options */
static int opts_parse(int argc, char *argv[], struct options *opts)
{
static const char *shortopts = "hp:dO:t:vri:I:s:x:";
#ifdef HAVE_GETOPT_H
static const struct option longopts[] = {
{"help", 0, NULL, 'h'},
{"pid", 1, NULL, 'p'},
{"debug", 0, NULL, 'd'},
{"facility", 1, NULL, 'O'},
{"threshold", 1, NULL, 't'},
{"verbose", 0, NULL, 'v'},
{"ht", 0, NULL, 'r'},
{"short-interval", 1, NULL, 'i'},
{"long-interval", 1, NULL, 'i'},
{"strategy", 1, NULL, 's'},
{"pxm", 1, NULL, 'x'},
{NULL, 0, NULL, 0}
};
#endif
optind = 1;
while(1) {
int opt;
#ifdef HAVE_GETOPT_H
opt = getopt_long(argc, argv, shortopts, longopts, NULL);
#else
opt = getopt(argc, argv, shortopts);
#endif
if (-1 == opt)
break;
switch (opt) {
case 'p':
if (opts->pidfile)
free(opts->pidfile);
opts->pidfile = strdup(optarg);
break;
case 'x':
if (opts->pxm)
free(opts->pxm);
opts->pxm = strdup(optarg);
break;
case 'd':
opts->debug = 1;
break;
case 'v':
opts->verbose = 1;
break;
case 'r':
opts->ht = 1;
break;
case 'O':
if (lub_log_facility(optarg, &(opts->log_facility))) {
fprintf(stderr, "Error: Illegal syslog facility %s.\n", optarg);
help(-1, argv[0]);
exit(-1);
}
break;
case 't':
{
char *endptr;
float thresh;
thresh = strtof(optarg, &endptr);
if (endptr == optarg)
thresh = opts->threshold;
opts->threshold = thresh;
if (thresh > 100.00) {
fprintf(stderr, "Error: Illegal threshold value %s.\n", optarg);
help(-1, argv[0]);
exit(-1);
}
}
break;
case 'i':
{
char *endptr;
unsigned long int val;
val = strtoul(optarg, &endptr, 10);
if (endptr != optarg)
opts->short_interval = val;
}
break;
case 'I':
{
char *endptr;
unsigned long int val;
val = strtoul(optarg, &endptr, 10);
if (endptr != optarg)
opts->long_interval = val;
}
break;
case 's':
if (!strcmp(optarg, "max"))
opts->strategy = BIRQ_CHOOSE_MAX;
else if (!strcmp(optarg, "min"))
opts->strategy = BIRQ_CHOOSE_MIN;
else if (!strcmp(optarg, "rnd"))
opts->strategy = BIRQ_CHOOSE_RND;
else {
fprintf(stderr, "Error: Illegal strategy value %s.\n", optarg);
help(-1, argv[0]);
exit(-1);
}
break;
case 'h':
help(0, argv[0]);
exit(0);
break;
default:
help(-1, argv[0]);
exit(-1);
break;
}
}
return 0;
}
