X-Git-Url: https://gerrit.automotivelinux.org/gerrit/gitweb?a=blobdiff_plain;f=src%2Fmain.c;h=bffcb38854419285c8f7ecdee37c980b8e1e9688;hb=4d5ce1b0b3b41b8fb53126a0c67534db7d010a75;hp=dbfbac5d455b96e661b94b4803b591530589dafa;hpb=06382af9092babedbf56aa1c00b3bd7cb0b86cda;p=src%2Fapp-framework-binder.git diff --git a/src/main.c b/src/main.c index dbfbac5d..bffcb388 100644 --- a/src/main.c +++ b/src/main.c @@ -1,5 +1,5 @@ -/* - * Copyright (C) 2015, 2016 "IoT.bzh" +/* + * Copyright (C) 2015, 2016, 2017 "IoT.bzh" * Author "Fulup Ar Foll" * Author José Bollo * @@ -17,450 +17,185 @@ */ #define _GNU_SOURCE +#define AFB_BINDING_PRAGMA_NO_VERBOSE_MACRO #include #include +#include +#include #include #include #include #include -#include +#include -#include +#if !defined(NO_CALL_PERSONALITY) +#include +#endif + +#include + +#include #include "afb-config.h" #include "afb-hswitch.h" -#include "afb-apis.h" +#include "afb-apiset.h" #include "afb-api-so.h" #include "afb-api-dbus.h" +#include "afb-api-ws.h" #include "afb-hsrv.h" -#include "afb-context.h" #include "afb-hreq.h" -#include "afb-sig-handler.h" -#include "session.h" +#include "afb-xreq.h" +#include "jobs.h" +#include "afb-session.h" #include "verbose.h" #include "afb-common.h" +#include "afb-monitor.h" +#include "afb-hook.h" +#include "sd-fds.h" +#include "afb-debug.h" +#include "process-name.h" -#include +/* + if SELF_PGROUP == 0 the launched command is the group leader + if SELF_PGROUP != 0 afb-daemon is the group leader +*/ +#define SELF_PGROUP 1 -#if !defined(PLUGIN_INSTALL_DIR) -#error "you should define PLUGIN_INSTALL_DIR" -#endif +struct afb_apiset *main_apiset; -#define AFB_VERSION "0.4" - -// Define command line option -#define SET_VERBOSE 1 -#define SET_BACKGROUND 2 -#define SET_FORGROUND 3 - -#define SET_TCP_PORT 5 -#define SET_ROOT_DIR 6 -#define SET_ROOT_BASE 7 -#define SET_ROOT_API 8 -#define SET_ALIAS 9 - -#define SET_CACHE_TIMEOUT 10 -#define SET_SESSION_DIR 11 - -#define SET_AUTH_TOKEN 12 -#define SET_LDPATH 13 -#define SET_APITIMEOUT 14 -#define SET_CNTXTIMEOUT 15 - -#define DISPLAY_VERSION 16 -#define DISPLAY_HELP 17 - -#define SET_MODE 18 -#define SET_READYFD 19 - -#define DBUS_CLIENT 20 -#define DBUS_SERVICE 21 -#define SO_PLUGIN 22 - -// Command line structure hold cli --command + help text -typedef struct { - int val; // command number within application - int has_arg; // command number within application - char *name; // command as used in --xxxx cli - char *help; // help text -} AFB_options; - - -// Supported option -static AFB_options cliOptions [] = { - {SET_VERBOSE ,0,"verbose" , "Verbose Mode"}, - - {SET_FORGROUND ,0,"foreground" , "Get all in foreground mode"}, - {SET_BACKGROUND ,0,"daemon" , "Get all in background mode"}, - - {SET_TCP_PORT ,1,"port" , "HTTP listening TCP port [default 1234]"}, - {SET_ROOT_DIR ,1,"rootdir" , "HTTP Root Directory [default $HOME/.AFB]"}, - {SET_ROOT_BASE ,1,"rootbase" , "Angular Base Root URL [default /opa]"}, - {SET_ROOT_API ,1,"rootapi" , "HTML Root API URL [default /api]"}, - {SET_ALIAS ,1,"alias" , "Muliple url map outside of rootdir [eg: --alias=/icons:/usr/share/icons]"}, - - {SET_APITIMEOUT ,1,"apitimeout" , "Plugin API timeout in seconds [default 10]"}, - {SET_CNTXTIMEOUT ,1,"cntxtimeout" , "Client Session Context Timeout [default 900]"}, - {SET_CACHE_TIMEOUT,1,"cache-eol" , "Client cache end of live [default 3600s]"}, - - {SET_SESSION_DIR ,1,"sessiondir" , "Sessions file path [default rootdir/sessions]"}, - - {SET_LDPATH ,1,"ldpaths" , "Load Plugins from dir1:dir2:... [default = PLUGIN_INSTALL_DIR"}, - {SET_AUTH_TOKEN ,1,"token" , "Initial Secret [default=no-session, --token="" for session without authentication]"}, - - {DISPLAY_VERSION ,0,"version" , "Display version and copyright"}, - {DISPLAY_HELP ,0,"help" , "Display this help"}, - - {SET_MODE ,1,"mode" , "set the mode: either local, remote or global"}, - {SET_READYFD ,1,"readyfd" , "set the #fd to signal when ready"}, - - {DBUS_CLIENT ,1,"dbus-client" , "bind to an afb service through dbus"}, - {DBUS_SERVICE ,1,"dbus-server" , "provides an afb service through dbus"}, - {SO_PLUGIN ,1,"plugin" , "load the plugin of path"}, - - {0, 0, NULL, NULL} - }; +static struct afb_config *config; +static pid_t childpid; /*---------------------------------------------------------- - | printversion - | print version and copyright + | helpers for handling list of arguments +--------------------------------------------------------- */ -static void printVersion (FILE *file) + +/* + * Calls the callback 'run' for each value of the 'list' + * until the callback returns 0 or the end of the list is reached. + * Returns either NULL if the end of the list is reached or a pointer + * to the item whose value made 'run' return 0. + * 'closure' is used for passing user data. + */ +static struct afb_config_list *run_for_list(struct afb_config_list *list, + int (*run) (void *closure, char *value), + void *closure) { - fprintf(file, "\n----------------------------------------- \n"); - fprintf(file, " AFB [Application Framework Binder] version=%s |\n", AFB_VERSION); - fprintf(file, " \n"); - fprintf(file, " Copyright (C) 2015, 2016 \"IoT.bzh\" [fulup -at- iot.bzh]\n"); - fprintf(file, " AFB comes with ABSOLUTELY NO WARRANTY.\n"); - fprintf(file, " Licence Apache 2\n\n"); - exit (0); + while (list && run(closure, list->value)) + list = list->next; + return list; } -/*---------------------------------------------------------- - | printHelp - | print information from long option array - +--------------------------------------------------------- */ - -static void printHelp(FILE *file, const char *name) +static int run_start(void *closure, char *value) { - int ind; - char command[50]; - - fprintf (file, "%s:\nallowed options\n", name); - for (ind=0; cliOptions [ind].name != NULL;ind++) - { - // display options - if (cliOptions [ind].has_arg == 0 ) - { - fprintf (file, " --%-15s %s\n", cliOptions [ind].name, cliOptions[ind].help); - } else { - sprintf(command, "%s=xxxx", cliOptions [ind].name); - fprintf (file, " --%-15s %s\n", command, cliOptions[ind].help); - } - } - fprintf (file, "Example:\n %s\\\n --verbose --port=1234 --token='azerty' --ldpaths=build/plugins:/usr/lib64/agl/plugins\n", name); + int (*starter) (const char *value, struct afb_apiset *apiset) = closure; + return starter(value, main_apiset) >= 0; } -// load config from disk and merge with CLI option -static void config_set_default (struct afb_config * config) +static void apiset_start_list(struct afb_config_list *list, + int (*starter) (const char *value, struct afb_apiset *apiset), const char *message) { - // default HTTP port - if (config->httpdPort == 0) - config->httpdPort = 1234; - - // default Plugin API timeout - if (config->apiTimeout == 0) - config->apiTimeout = DEFLT_API_TIMEOUT; - - // default AUTH_TOKEN - if (config->token == NULL) - config->token = DEFLT_AUTH_TOKEN; - - // cache timeout default one hour - if (config->cacheTimeout == 0) - config->cacheTimeout = DEFLT_CACHE_TIMEOUT; - - // cache timeout default one hour - if (config->cntxTimeout == 0) - config->cntxTimeout = DEFLT_CNTX_TIMEOUT; - - if (config->rootdir == NULL) { - config->rootdir = getenv("AFBDIR"); - if (config->rootdir == NULL) { - config->rootdir = malloc (512); - strncpy (config->rootdir, getenv("HOME"),512); - strncat (config->rootdir, "/.AFB",512); - } - // if directory does not exist createit - mkdir (config->rootdir, O_RDWR | S_IRWXU | S_IRGRP); - } - - // if no Angular/HTML5 rootbase let's try '/' as default - if (config->rootbase == NULL) - config->rootbase = "/opa"; - - if (config->rootapi == NULL) - config->rootapi = "/api"; - - if (config->ldpaths == NULL) - config->ldpaths = PLUGIN_INSTALL_DIR; - - // if no session dir create a default path from rootdir - if (config->sessiondir == NULL) { - config->sessiondir = malloc (512); - strncpy (config->sessiondir, config->rootdir, 512); - strncat (config->sessiondir, "/sessions",512); - } - - // if no config dir create a default path from sessiondir - if (config->console == NULL) { - config->console = malloc (512); - strncpy (config->console, config->sessiondir, 512); - strncat (config->console, "/AFB-console.out",512); - } + list = run_for_list(list, run_start, starter); + if (list) { + ERROR("can't start %s %s", message, list->value); + exit(1); + } } - -/*--------------------------------------------------------- - | main - | Parse option and launch action +/*---------------------------------------------------------- + | exit_handler + | Handles on exit specific actions +--------------------------------------------------------- */ +static void exit_handler() +{ + struct sigaction siga; -static void add_item(struct afb_config *config, int kind, char *value) + memset(&siga, 0, sizeof siga); + siga.sa_handler = SIG_IGN; + sigaction(SIGTERM, &siga, NULL); + + if (SELF_PGROUP) + killpg(0, SIGTERM); + else if (childpid > 0) + killpg(childpid, SIGTERM); +} + +static void on_sigterm(int signum, siginfo_t *info, void *uctx) { - struct afb_config_item *item = malloc(sizeof *item); - if (item == NULL) { - ERROR("out of memory"); - exit(1); - } - item->kind = kind; - item->value = value; - item->previous = config->items; - config->items = item; + NOTICE("Received SIGTERM"); + exit(0); } -static void parse_arguments(int argc, char *argv[], struct afb_config *config) +static void on_sighup(int signum, siginfo_t *info, void *uctx) { - char* programName = argv [0]; - int optionIndex = 0; - int optc, ind; - int nbcmd; - struct option *gnuOptions; - - // ------------------ Process Command Line ----------------------- - - // if no argument print help and return - if (argc < 2) { - printHelp(stderr, programName); - exit(1); - } - - // build GNU getopt info from cliOptions - nbcmd = sizeof (cliOptions) / sizeof (AFB_options); - gnuOptions = malloc (sizeof (*gnuOptions) * (unsigned)nbcmd); - for (ind=0; ind < nbcmd;ind++) { - gnuOptions [ind].name = cliOptions[ind].name; - gnuOptions [ind].has_arg = cliOptions[ind].has_arg; - gnuOptions [ind].flag = 0; - gnuOptions [ind].val = cliOptions[ind].val; - } - - // get all options from command line - while ((optc = getopt_long (argc, argv, "vsp?", gnuOptions, &optionIndex)) - != EOF) - { - switch (optc) - { - case SET_VERBOSE: - verbosity++; - break; - - case SET_TCP_PORT: - if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%d", &config->httpdPort)) goto notAnInteger; - break; - - case SET_APITIMEOUT: - if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%d", &config->apiTimeout)) goto notAnInteger; - break; - - case SET_CNTXTIMEOUT: - if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%d", &config->cntxTimeout)) goto notAnInteger; - break; - - case SET_ROOT_DIR: - if (optarg == 0) goto needValueForOption; - config->rootdir = optarg; - INFO("Forcing Rootdir=%s",config->rootdir); - break; - - case SET_ROOT_BASE: - if (optarg == 0) goto needValueForOption; - config->rootbase = optarg; - INFO("Forcing Rootbase=%s",config->rootbase); - break; - - case SET_ROOT_API: - if (optarg == 0) goto needValueForOption; - config->rootapi = optarg; - INFO("Forcing Rootapi=%s",config->rootapi); - break; - - case SET_ALIAS: - if (optarg == 0) goto needValueForOption; - if ((unsigned)config->aliascount < sizeof (config->aliasdir) / sizeof (config->aliasdir[0])) { - config->aliasdir[config->aliascount].url = strsep(&optarg,":"); - if (optarg == NULL) { - ERROR("missing ':' in alias %s, ignored", config->aliasdir[config->aliascount].url); - } else { - config->aliasdir[config->aliascount].path = optarg; - INFO("Alias url=%s path=%s", config->aliasdir[config->aliascount].url, config->aliasdir[config->aliascount].path); - config->aliascount++; - } - } else { - ERROR("Too many aliases [max:%d] %s ignored", MAX_ALIAS, optarg); - } - break; - - case SET_AUTH_TOKEN: - if (optarg == 0) goto needValueForOption; - config->token = optarg; - break; - - case SET_LDPATH: - if (optarg == 0) goto needValueForOption; - config->ldpaths = optarg; - break; - - case SET_SESSION_DIR: - if (optarg == 0) goto needValueForOption; - config->sessiondir = optarg; - break; - - case SET_CACHE_TIMEOUT: - if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%d", &config->cacheTimeout)) goto notAnInteger; - break; - - case SET_FORGROUND: - if (optarg != 0) goto noValueForOption; - config->background = 0; - break; - - case SET_BACKGROUND: - if (optarg != 0) goto noValueForOption; - config->background = 1; - break; - - case SET_MODE: - if (optarg == 0) goto needValueForOption; - if (!strcmp(optarg, "local")) config->mode = AFB_MODE_LOCAL; - else if (!strcmp(optarg, "remote")) config->mode = AFB_MODE_REMOTE; - else if (!strcmp(optarg, "global")) config->mode = AFB_MODE_GLOBAL; - else goto badMode; - break; - - case SET_READYFD: - if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%u", &config->readyfd)) goto notAnInteger; - break; - - case DBUS_CLIENT: - case DBUS_SERVICE: - case SO_PLUGIN: - if (optarg == 0) goto needValueForOption; - add_item(config, optc, optarg); - break; - - case DISPLAY_VERSION: - if (optarg != 0) goto noValueForOption; - printVersion(stdout); - break; - - case DISPLAY_HELP: - default: - printHelp(stdout, programName); - exit(0); - } - } - free(gnuOptions); - - config_set_default (config); - return; - - -needValueForOption: - ERROR("AFB-daemon option [--%s] need a value i.e. --%s=xxx" - ,gnuOptions[optionIndex].name, gnuOptions[optionIndex].name); - exit (1); - -notAnInteger: - ERROR("AFB-daemon option [--%s] requirer an interger i.e. --%s=9" - ,gnuOptions[optionIndex].name, gnuOptions[optionIndex].name); - exit (1); - -noValueForOption: - ERROR("AFB-daemon option [--%s] don't take value" - ,gnuOptions[optionIndex].name); - exit (1); - -badMode: - ERROR("AFB-daemon option [--%s] only accepts local, global or remote." - ,gnuOptions[optionIndex].name); - exit (1); + NOTICE("Received SIGHUP"); + /* TODO */ } -/*---------------------------------------------------------- - | closeSession - | try to close everything before leaving - +--------------------------------------------------------- */ -static void closeSession (int status, void *data) { - /* struct afb_config *config = data; */ +static void setup_daemon() +{ + struct sigaction siga; + + /* install signal handlers */ + memset(&siga, 0, sizeof siga); + siga.sa_flags = SA_SIGINFO; + + siga.sa_sigaction = on_sigterm; + sigaction(SIGTERM, &siga, NULL); + + siga.sa_sigaction = on_sighup; + sigaction(SIGHUP, &siga, NULL); + + /* handle groups */ + atexit(exit_handler); + + /* ignore any SIGPIPE */ + signal(SIGPIPE, SIG_IGN); } /*---------------------------------------------------------- | daemonize | set the process in background +--------------------------------------------------------- */ -static void daemonize(struct afb_config *config) +static void daemonize() { - int consoleFD; - int pid; - - // open /dev/console to redirect output messAFBes - consoleFD = open(config->console, O_WRONLY | O_APPEND | O_CREAT , 0640); - if (consoleFD < 0) { - ERROR("AFB-daemon cannot open /dev/console (use --foreground)"); - exit (1); - } - - // fork process when running background mode - pid = fork (); - - // if fail nothing much to do - if (pid == -1) { - ERROR("AFB-daemon Failed to fork son process"); - exit (1); - } + int consoleFD; + int pid; - // if in father process, just leave - if (pid != 0) _exit (0); - - // son process get all data in standalone mode - NOTICE("background mode [pid:%d console:%s]", getpid(),config->console); + // open /dev/console to redirect output messAFBes + consoleFD = open(config->console, O_WRONLY | O_APPEND | O_CREAT, 0640); + if (consoleFD < 0) { + ERROR("AFB-daemon cannot open /dev/console (use --foreground)"); + exit(1); + } + // fork process when running background mode + pid = fork(); - // redirect default I/O on console - close (2); dup(consoleFD); // redirect stderr - close (1); dup(consoleFD); // redirect stdout - close (0); // no need for stdin - close (consoleFD); + // if fail nothing much to do + if (pid == -1) { + ERROR("AFB-daemon Failed to fork son process"); + exit(1); + } + // if in father process, just leave + if (pid != 0) + _exit(0); + + // son process get all data in standalone mode + NOTICE("background mode [pid:%d console:%s]", getpid(), + config->console); + + // redirect default I/O on console + close(2); + dup(consoleFD); // redirect stderr + close(1); + dup(consoleFD); // redirect stdout + close(0); // no need for stdin + close(consoleFD); #if 0 - setsid(); // allow father process to fully exit - sleep (2); // allow main to leave and release port + setsid(); // allow father process to fully exit + sleep(2); // allow main to leave and release port #endif } @@ -468,36 +203,56 @@ static void daemonize(struct afb_config *config) | http server | Handles the HTTP server +--------------------------------------------------------- */ -static int init_http_server(struct afb_hsrv *hsrv, struct afb_config * config) +static int init_alias(void *closure, char *spec) { - int idx; + struct afb_hsrv *hsrv = closure; + char *path = strchr(spec, ':'); - if (!afb_hsrv_add_handler(hsrv, config->rootapi, afb_hswitch_websocket_switch, NULL, 20)) - return 0; + if (path == NULL) { + ERROR("Missing ':' in alias %s. Alias ignored", spec); + return 1; + } + *path++ = 0; + INFO("Alias for url=%s to path=%s", spec, path); + return afb_hsrv_add_alias(hsrv, spec, afb_common_rootdir_get_fd(), path, + 0, 0); +} - if (!afb_hsrv_add_handler(hsrv, config->rootapi, afb_hswitch_apis, NULL, 10)) +static int init_http_server(struct afb_hsrv *hsrv) +{ + if (!afb_hsrv_add_handler + (hsrv, config->rootapi, afb_hswitch_websocket_switch, main_apiset, 20)) return 0; - for (idx = 0; idx < config->aliascount; idx++) - if (!afb_hsrv_add_alias (hsrv, config->aliasdir[idx].url, config->aliasdir[idx].path, 0)) - return 0; + if (!afb_hsrv_add_handler + (hsrv, config->rootapi, afb_hswitch_apis, main_apiset, 10)) + return 0; - if (!afb_hsrv_add_alias(hsrv, "", config->rootdir, -10)) + if (run_for_list(config->aliases, init_alias, hsrv)) return 0; - if (!afb_hsrv_add_handler(hsrv, config->rootbase, afb_hswitch_one_page_api_redirect, NULL, -20)) + if (config->roothttp != NULL) { + if (!afb_hsrv_add_alias + (hsrv, "", afb_common_rootdir_get_fd(), config->roothttp, + -10, 1)) + return 0; + } + + if (!afb_hsrv_add_handler + (hsrv, config->rootbase, afb_hswitch_one_page_api_redirect, NULL, + -20)) return 0; return 1; } -static struct afb_hsrv *start_http_server(struct afb_config * config) +static struct afb_hsrv *start_http_server() { int rc; struct afb_hsrv *hsrv; - if (afb_hreq_init_download_path("/tmp")) { /* TODO: sessiondir? */ - ERROR("unable to set the tmp directory"); + if (afb_hreq_init_download_path(config->uploaddir)) { + ERROR("unable to set the upload directory %s", config->uploaddir); return NULL; } @@ -508,14 +263,14 @@ static struct afb_hsrv *start_http_server(struct afb_config * config) } if (!afb_hsrv_set_cache_timeout(hsrv, config->cacheTimeout) - || !init_http_server(hsrv, config)) { + || !init_http_server(hsrv)) { ERROR("initialisation of httpd failed"); afb_hsrv_put(hsrv); return NULL; } NOTICE("Waiting port=%d rootdir=%s", config->httpdPort, config->rootdir); - NOTICE("Browser URL= http:/*localhost:%d", config->httpdPort); + NOTICE("Browser URL= http://localhost:%d", config->httpdPort); rc = afb_hsrv_start(hsrv, (uint16_t) config->httpdPort, 15); if (!rc) { @@ -527,116 +282,402 @@ static struct afb_hsrv *start_http_server(struct afb_config * config) return hsrv; } -static void start_items(struct afb_config_item *item) +/*--------------------------------------------------------- + | execute_command + | + +--------------------------------------------------------- */ + +static void on_sigchld(int signum, siginfo_t *info, void *uctx) { - if (item != NULL) { - /* keeps the order */ - start_items(item->previous); - switch(item->kind) { - case DBUS_CLIENT: - if (afb_api_dbus_add_client(item->value) < 0) { - ERROR("can't start the afb-dbus client of path %s",item->value); - exit(1); - } - break; - case DBUS_SERVICE: - if (afb_api_dbus_add_server(item->value) < 0) { - ERROR("can't start the afb-dbus service of path %s",item->value); - exit(1); - } - break; - case SO_PLUGIN: - if (afb_api_so_add_plugin(item->value) < 0) { - ERROR("can't start the plugin of path %s",item->value); + if (info->si_pid == childpid) { + switch (info->si_code) { + case CLD_EXITED: + case CLD_KILLED: + case CLD_DUMPED: + childpid = 0; + if (!SELF_PGROUP) + killpg(info->si_pid, SIGKILL); + waitpid(info->si_pid, NULL, 0); + exit(0); + } + } +} + +/* +# @@ @ +# @p port +# @t token +*/ + +#define SUBST_CHAR '@' +#define SUBST_STR "@" + +static char *instanciate_string(char *arg, const char *port, const char *token) +{ + char *resu, *it, *wr; + int chg, dif; + + /* get the changes */ + chg = 0; + dif = 0; + it = strchrnul(arg, SUBST_CHAR); + while (*it) { + switch(*++it) { + case 'p': chg++; dif += (int)strlen(port) - 2; break; + case 't': chg++; dif += (int)strlen(token) - 2; break; + case SUBST_CHAR: it++; chg++; dif--; break; + default: break; + } + it = strchrnul(it, SUBST_CHAR); + } + + /* return arg when no change */ + if (!chg) + return arg; + + /* allocates the result */ + resu = malloc((it - arg) + dif + 1); + if (!resu) { + ERROR("out of memory"); + return NULL; + } + + /* instanciate the arguments */ + wr = resu; + for (;;) { + it = strchrnul(arg, SUBST_CHAR); + wr = mempcpy(wr, arg, it - arg); + if (!*it) + break; + switch(*++it) { + case 'p': wr = stpcpy(wr, port); break; + case 't': wr = stpcpy(wr, token); break; + default: *wr++ = SUBST_CHAR; /*@fallthrough@*/ + case SUBST_CHAR: *wr++ = *it; + } + arg = ++it; + } + + *wr = 0; + return resu; +} + +static int instanciate_environ(const char *port, const char *token) +{ + extern char **environ; + char *repl; + int i; + + /* instanciate the environment */ + for (i = 0 ; environ[i] ; i++) { + repl = instanciate_string(environ[i], port, token); + if (!repl) + return -1; + environ[i] = repl; + } + return 0; +} + +static int instanciate_command_args(const char *port, const char *token) +{ + char *repl; + int i; + + /* instanciate the arguments */ + for (i = 0 ; config->exec[i] ; i++) { + repl = instanciate_string(config->exec[i], port, token); + if (!repl) + return -1; + config->exec[i] = repl; + } + return 0; +} + +static int execute_command() +{ + struct sigaction siga; + char port[20]; + const char *token; + int rc; + + /* check whether a command is to execute or not */ + if (!config->exec || !config->exec[0]) + return 0; + + if (SELF_PGROUP) + setpgid(0, 0); + + /* install signal handler */ + memset(&siga, 0, sizeof siga); + siga.sa_sigaction = on_sigchld; + siga.sa_flags = SA_SIGINFO; + sigaction(SIGCHLD, &siga, NULL); + + /* fork now */ + childpid = fork(); + if (childpid) + return 0; + + /* compute the string for port */ + if (config->httpdPort) + rc = snprintf(port, sizeof port, "%d", config->httpdPort); + else + rc = snprintf(port, sizeof port, "%cp", SUBST_CHAR); + if (rc < 0 || rc >= (int)(sizeof port)) { + ERROR("port->txt failed"); + } + else { + /* instanciate arguments and environment */ + token = afb_session_initial_token(); + if (instanciate_command_args(port, token) >= 0 + && instanciate_environ(port, token) >= 0) { + /* run */ + if (!SELF_PGROUP) + setpgid(0, 0); + execv(config->exec[0], config->exec); + ERROR("can't launch %s: %m", config->exec[0]); + } + } exit(1); - } - break; - default: - ERROR("unexpected internal error"); - exit(1); - } - /* frre the item */ - free(item); - } + return -1; } /*--------------------------------------------------------- - | main - | Parse option and launch action + | startup calls +--------------------------------------------------------- */ -int main(int argc, char *argv[]) { - struct afb_hsrv *hsrv; - struct afb_config *config; - struct sd_event *eventloop; +struct startup_req +{ + struct afb_xreq xreq; + char *api; + char *verb; + struct afb_config_list *current; + struct afb_session *session; +}; + +static void startup_call_reply(struct afb_xreq *xreq, int status, struct json_object *obj) +{ + struct startup_req *sreq = CONTAINER_OF_XREQ(struct startup_req, xreq); + + if (status >= 0) + NOTICE("startup call %s returned %s", sreq->current->value, json_object_get_string(obj)); + else { + ERROR("startup call %s ERROR! %s", sreq->current->value, json_object_get_string(obj)); + exit(1); + } +} + +static void startup_call_current(struct startup_req *sreq); + +static void startup_call_unref(struct afb_xreq *xreq) +{ + struct startup_req *sreq = CONTAINER_OF_XREQ(struct startup_req, xreq); + + free(sreq->api); + free(sreq->verb); + json_object_put(sreq->xreq.json); + sreq->current = sreq->current->next; + if (sreq->current) + startup_call_current(sreq); + else { + afb_session_close(sreq->session); + afb_session_unref(sreq->session); + free(sreq); + } +} + +static struct afb_xreq_query_itf startup_xreq_itf = +{ + .reply = startup_call_reply, + .unref = startup_call_unref +}; - LOGAUTH("afb-daemon"); +static void startup_call_current(struct startup_req *sreq) +{ + char *api, *verb, *json; + + api = sreq->current->value; + verb = strchr(api, '/'); + if (verb) { + json = strchr(verb, ':'); + if (json) { + afb_xreq_init(&sreq->xreq, &startup_xreq_itf); + afb_context_init(&sreq->xreq.context, sreq->session, NULL); + sreq->xreq.context.validated = 1; + sreq->api = strndup(api, verb - api); + sreq->verb = strndup(verb + 1, json - verb - 1); + sreq->xreq.api = sreq->api; + sreq->xreq.verb = sreq->verb; + sreq->xreq.json = json_tokener_parse(json + 1); + if (sreq->api && sreq->verb && sreq->xreq.json) { + afb_xreq_process(&sreq->xreq, main_apiset); + return; + } + } + } + ERROR("Bad call specification %s", sreq->current->value); + exit(1); +} - // ------------- Build session handler & init config ------- - config = calloc (1, sizeof (struct afb_config)); +static void run_startup_calls() +{ + struct afb_config_list *list; + struct startup_req *sreq; + + list = config->calls; + if (list) { + sreq = calloc(1, sizeof *sreq); + sreq->session = afb_session_create("startup", 3600); + sreq->current = list; + startup_call_current(sreq); + } +} - on_exit(closeSession, config); - parse_arguments(argc, argv, config); +/*--------------------------------------------------------- + | job for starting the daemon + +--------------------------------------------------------- */ - // ------------------ sanity check ---------------------------------------- - if (config->httpdPort <= 0) { - ERROR("no port is defined"); - exit (1); - } +static void start(int signum) +{ + struct afb_hsrv *hsrv; - if (config->ldpaths) - afb_api_so_add_pathset(config->ldpaths); + afb_debug("start-entry"); - start_items(config->items); - config->items = NULL; + if (signum) { + ERROR("start aborted: received signal %s", strsignal(signum)); + exit(1); + } - ctxStoreInit(CTX_NBCLIENTS, config->cntxTimeout, config->token, afb_apis_count()); - if (!afb_hreq_init_cookie(config->httpdPort, config->rootapi, DEFLT_CNTX_TIMEOUT)) { - ERROR("initialisation of cookies failed"); - exit (1); - } + // ------------------ sanity check ---------------------------------------- + if (config->httpdPort <= 0) { + ERROR("no port is defined"); + goto error; + } - if (afb_sig_handler_init() < 0) { - ERROR("main fail to initialise signal handlers"); - return 1; - } + /* set the directories */ + mkdir(config->workdir, S_IRWXU | S_IRGRP | S_IXGRP); + if (chdir(config->workdir) < 0) { + ERROR("Can't enter working dir %s", config->workdir); + goto error; + } + if (afb_common_rootdir_set(config->rootdir) < 0) { + ERROR("failed to set common root directory"); + goto error; + } - // let's run this program with a low priority - nice (20); + /* configure the daemon */ + afb_session_init(config->nbSessionMax, config->cntxTimeout, config->token); + if (!afb_hreq_init_cookie(config->httpdPort, config->rootapi, config->cntxTimeout)) { + ERROR("initialisation of cookies failed"); + goto error; + } + main_apiset = afb_apiset_create("main", config->apiTimeout); + if (!main_apiset) { + ERROR("can't create main api set"); + goto error; + } + if (afb_monitor_init() < 0) { + ERROR("failed to setup monitor"); + goto error; + } - // ------------------ Finaly Process Commands ----------------------------- - // let's not take the risk to run as ROOT - //if (getuid() == 0) goto errorNoRoot; + /* install hooks */ + if (config->tracereq) + afb_hook_create_xreq(NULL, NULL, NULL, config->tracereq, NULL, NULL); + if (config->traceditf) + afb_hook_create_ditf(NULL, config->traceditf, NULL, NULL); + if (config->tracesvc) + afb_hook_create_svc(NULL, config->tracesvc, NULL, NULL); + if (config->traceevt) + afb_hook_create_evt(NULL, config->traceevt, NULL, NULL); + + /* load bindings */ + afb_debug("start-load"); + apiset_start_list(config->so_bindings, afb_api_so_add_binding, "the binding"); + apiset_start_list(config->dbus_clients, afb_api_dbus_add_client, "the afb-dbus client"); + apiset_start_list(config->ws_clients, afb_api_ws_add_client, "the afb-websocket client"); + apiset_start_list(config->ldpaths, afb_api_so_add_pathset_fails, "the binding path set"); + apiset_start_list(config->weak_ldpaths, afb_api_so_add_pathset_nofails, "the weak binding path set"); + + apiset_start_list(config->dbus_servers, afb_api_dbus_add_server, "the afb-dbus service"); + apiset_start_list(config->ws_servers, afb_api_ws_add_server, "the afb-websocket service"); + + DEBUG("Init config done"); + + /* start the services */ + afb_debug("start-start"); +#if !defined(NO_CALL_PERSONALITY) + personality((unsigned long)-1L); +#endif + if (afb_apiset_start_all_services(main_apiset, 1) < 0) + goto error; + + /* start the HTTP server */ + afb_debug("start-http"); + if (!config->noHttpd) { + hsrv = start_http_server(); + if (hsrv == NULL) + goto error; + } - DEBUG("Init config done"); + /* run the startup calls */ + afb_debug("start-call"); + run_startup_calls(); - // --------- run ----------- - if (config->background) { - // --------- in background mode ----------- - INFO("entering background mode"); - daemonize(config); - } else { - // ---- in foreground mode -------------------- - INFO("entering foreground mode"); - } + /* run the command */ + afb_debug("start-exec"); + if (execute_command() < 0) + goto error; - hsrv = start_http_server(config); - if (hsrv == NULL) + /* ready */ + sd_notify(1, "READY=1"); + return; +error: exit(1); +} - if (config->readyfd != 0) { - static const char readystr[] = "READY=1"; - write(config->readyfd, readystr, sizeof(readystr) - 1); - close(config->readyfd); - } +/*--------------------------------------------------------- + | main + | Parse option and launch action + +--------------------------------------------------------- */ - // infinite loop - eventloop = afb_common_get_event_loop(); - for(;;) - sd_event_run(eventloop, 30000000); +int main(int argc, char *argv[]) +{ + afb_debug("main-entry"); + + // let's run this program with a low priority + nice(20); + + sd_fds_init(); + + // ------------- Build session handler & init config ------- + config = afb_config_parse_arguments(argc, argv); + if (config->name) { + verbose_set_name(config->name, 0); + process_name_set_name(config->name); + process_name_replace_cmdline(argv, config->name); + } + afb_debug("main-args"); + + // --------- run ----------- + if (config->background) { + // --------- in background mode ----------- + INFO("entering background mode"); + daemonize(); + } else { + // ---- in foreground mode -------------------- + INFO("entering foreground mode"); + } + INFO("running with pid %d", getpid()); - WARNING("hoops returned from infinite loop [report bug]"); + /* set the daemon environment */ + setup_daemon(); - return 0; + afb_debug("main-start"); + + /* enter job processing */ + jobs_start(3, 0, 50, start); + WARNING("hoops returned from jobs_enter! [report bug]"); + return 1; }