X-Git-Url: https://gerrit.automotivelinux.org/gerrit/gitweb?a=blobdiff_plain;f=src%2Fmain.c;h=bffcb38854419285c8f7ecdee37c980b8e1e9688;hb=4d5ce1b0b3b41b8fb53126a0c67534db7d010a75;hp=48b5a60164a54f3ed4788cc4782a8edb7069a356;hpb=acdf592fee6d7f1970e42e31c23c2a88196051aa;p=src%2Fapp-framework-binder.git diff --git a/src/main.c b/src/main.c index 48b5a601..bffcb388 100644 --- a/src/main.c +++ b/src/main.c @@ -1,5 +1,5 @@ -/* - * Copyright (C) 2015 "IoT.bzh" +/* + * Copyright (C) 2015, 2016, 2017 "IoT.bzh" * Author "Fulup Ar Foll" * Author José Bollo * @@ -16,558 +16,668 @@ * limitations under the License. */ -#include -#include +#define _GNU_SOURCE +#define AFB_BINDING_PRAGMA_NO_VERBOSE_MACRO + +#include +#include +#include #include -#include -#include +#include +#include +#include #include -#include - -#include "local-def.h" -#include "afb-plugins.h" +#include -#if !defined(PLUGIN_INSTALL_DIR) -#error "you should define PLUGIN_INSTALL_DIR" +#if !defined(NO_CALL_PERSONALITY) +#include #endif -#define AFB_VERSION "0.1" - -// Define command line option -#define SET_VERBOSE 1 -#define SET_BACKGROUND 2 -#define SET_FORGROUND 3 -#define SET_FAKE_MOD 4 - -#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 - -// 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"}, - {0, 0, NULL, NULL} - }; - -static AFB_aliasdir aliasdir[MAX_ALIAS]; -static int aliascount = 0; - +#include + +#include + +#include "afb-config.h" +#include "afb-hswitch.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-hreq.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" + +/* + 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 + +struct afb_apiset *main_apiset; + +static struct afb_config *config; +static pid_t childpid; /*---------------------------------------------------------- - | printversion - | print version and copyright + | helpers for handling list of arguments +--------------------------------------------------------- */ -static void printVersion (void) + +/* + * 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 (stderr,"\n----------------------------------------- \n"); - fprintf (stderr,"| AFB [Application Framework Binder] version=%s |\n", AFB_VERSION); - fprintf (stderr,"----------------------------------------- \n"); - fprintf (stderr,"| Copyright(C) 2016 /IoT.bzh [fulup -at- iot.bzh]\n"); - fprintf (stderr,"| AFB comes with ABSOLUTELY NO WARRANTY.\n"); - fprintf (stderr,"| Licence Apache 2\n\n"); - exit (0); + while (list && run(closure, list->value)) + list = list->next; + return list; } -// load config from disk and merge with CLI option -static AFB_error config_set_default (AFB_session * session) +static int run_start(void *closure, char *value) { - static char cacheTimeout [10]; - - // default HTTP port - if (session->config->httpdPort == 0) session->config->httpdPort=1234; - - // default Plugin API timeout - if (session->config->apiTimeout == 0) session->config->apiTimeout=DEFLT_API_TIMEOUT; - - // default AUTH_TOKEN - if (session->config->token == NULL) session->config->token= DEFLT_AUTH_TOKEN; - - // cache timeout default one hour - if (session->config->cacheTimeout == 0) session->config->cacheTimeout=DEFLT_CACHE_TIMEOUT; - - // cache timeout default one hour - if (session->config->cntxTimeout == 0) session->config->cntxTimeout=DEFLT_CNTX_TIMEOUT; - - if (session->config->rootdir == NULL) { - session->config->rootdir = getenv("AFBDIR"); - if (session->config->rootdir == NULL) { - session->config->rootdir = malloc (512); - strncpy (session->config->rootdir, getenv("HOME"),512); - strncat (session->config->rootdir, "/.AFB",512); - } - // if directory does not exist createit - mkdir (session->config->rootdir, O_RDWR | S_IRWXU | S_IRGRP); - } - - // if no Angular/HTML5 rootbase let's try '/' as default - if (session->config->rootbase == NULL) { - session->config->rootbase = "/opa"; - } - - if (session->config->rootapi == NULL) { - session->config->rootapi = "/api"; - } - - if (session->config->ldpaths == NULL) { - session->config->ldpaths = PLUGIN_INSTALL_DIR; - } - - // if no session dir create a default path from rootdir - if (session->config->sessiondir == NULL) { - session->config->sessiondir = malloc (512); - strncpy (session->config->sessiondir, session->config->rootdir, 512); - strncat (session->config->sessiondir, "/sessions",512); - } - - // if no config dir create a default path from sessiondir - if (session->config->console == NULL) { - session->config->console = malloc (512); - strncpy (session->config->console, session->config->sessiondir, 512); - strncat (session->config->console, "/AFB-console.out",512); - } - - // cacheTimeout is an integer but HTTPd wants it as a string - snprintf (cacheTimeout, sizeof (cacheTimeout),"%d", session->config->cacheTimeout); - session->cacheTimeout = cacheTimeout; // httpd uses cacheTimeout string version - - return AFB_SUCCESS; + int (*starter) (const char *value, struct afb_apiset *apiset) = closure; + return starter(value, main_apiset) >= 0; } +static void apiset_start_list(struct afb_config_list *list, + int (*starter) (const char *value, struct afb_apiset *apiset), const char *message) +{ + list = run_for_list(list, run_start, starter); + if (list) { + ERROR("can't start %s %s", message, list->value); + exit(1); + } +} /*---------------------------------------------------------- - | printHelp - | print information from long option array + | exit_handler + | Handles on exit specific actions +--------------------------------------------------------- */ +static void exit_handler() +{ + struct sigaction siga; - static void printHelp(char *name) { - int ind; - char command[20]; - - fprintf (stderr,"%s:\nallowed options\n", name); - for (ind=0; cliOptions [ind].name != NULL;ind++) - { - // display options - if (cliOptions [ind].has_arg == 0 ) - { - fprintf (stderr," --%-15s %s\n", cliOptions [ind].name, cliOptions[ind].help); - } else { - sprintf(command,"%s=xxxx", cliOptions [ind].name); - fprintf (stderr," --%-15s %s\n", command, cliOptions[ind].help); - } - } - fprintf (stderr,"Example:\n %s\\\n --verbose --port=1234 --token='azerty' --ldpaths=build/plugins:/usr/lib64/agl/plugins\n", name); -} // end printHelp + memset(&siga, 0, sizeof siga); + siga.sa_handler = SIG_IGN; + sigaction(SIGTERM, &siga, NULL); -/*--------------------------------------------------------- - | main - | Parse option and launch action - +--------------------------------------------------------- */ + if (SELF_PGROUP) + killpg(0, SIGTERM); + else if (childpid > 0) + killpg(childpid, SIGTERM); +} -static void parse_arguments(int argc, char *argv[], AFB_session *session) +static void on_sigterm(int signum, siginfo_t *info, void *uctx) { - char* programName = argv [0]; - int optionIndex = 0; - int optc, ind; - int nbcmd; - struct option *gnuOptions; - - // ------------- Build session handler & init config ------- - memset(&aliasdir ,0,sizeof(aliasdir)); - session->config->aliasdir = aliasdir; - - // ------------------ Process Command Line ----------------------- - - // if no argument print help and return - if (argc < 2) { - printHelp(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: - verbose = 1; - break; - - case SET_TCP_PORT: - if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%d", &session->config->httpdPort)) goto notAnInteger; - break; - - case SET_APITIMEOUT: - if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%d", &session->config->apiTimeout)) goto notAnInteger; - break; - - case SET_CNTXTIMEOUT: - if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%d", &session->config->cntxTimeout)) goto notAnInteger; - break; - - case SET_ROOT_DIR: - if (optarg == 0) goto needValueForOption; - session->config->rootdir = optarg; - if (verbose) fprintf(stderr, "Forcing Rootdir=%s\n",session->config->rootdir); - break; - - case SET_ROOT_BASE: - if (optarg == 0) goto needValueForOption; - session->config->rootbase = optarg; - if (verbose) fprintf(stderr, "Forcing Rootbase=%s\n",session->config->rootbase); - break; - - case SET_ROOT_API: - if (optarg == 0) goto needValueForOption; - session->config->rootapi = optarg; - if (verbose) fprintf(stderr, "Forcing Rootapi=%s\n",session->config->rootapi); - break; - - case SET_ALIAS: - if (optarg == 0) goto needValueForOption; - if (aliascount < MAX_ALIAS) { - aliasdir[aliascount].url = strsep(&optarg,":"); - if (optarg == NULL) { - fprintf(stderr, "missing ':' in alias %s, ignored\n", aliasdir[aliascount].url); - } else { - aliasdir[aliascount].path = optarg; - aliasdir[aliascount].len = strlen(aliasdir[aliascount].url); - if (verbose) fprintf(stderr, "Alias url=%s path=%s\n", aliasdir[aliascount].url, aliasdir[aliascount].path); - aliascount++; - } - } else { - fprintf(stderr, "Too many aliases [max:%d] %s ignored\n", MAX_ALIAS, optarg); - } - break; - - case SET_AUTH_TOKEN: - if (optarg == 0) goto needValueForOption; - session->config->token = optarg; - break; - - case SET_LDPATH: - if (optarg == 0) goto needValueForOption; - session->config->ldpaths = optarg; - break; - - case SET_SESSION_DIR: - if (optarg == 0) goto needValueForOption; - session->config->sessiondir = optarg; - break; - - case SET_CACHE_TIMEOUT: - if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%d", &session->config->cacheTimeout)) goto notAnInteger; - break; - - case SET_FAKE_MOD: - if (optarg != 0) goto noValueForOption; - session->fakemod = 1; - break; - - case SET_FORGROUND: - if (optarg != 0) goto noValueForOption; - session->foreground = 1; - break; - - case SET_BACKGROUND: - if (optarg != 0) goto noValueForOption; - session->background = 1; - break; - - case SET_MODE: - if (optarg == 0) goto needValueForOption; - if (!strcmp(optarg, "local")) session->config->mode = AFB_MODE_LOCAL; - else if (!strcmp(optarg, "remote")) session->config->mode = AFB_MODE_REMOTE; - else if (!strcmp(optarg, "global")) session->config->mode = AFB_MODE_GLOBAL; - else goto badMode; - break; - - case SET_READYFD: - if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%u", &session->readyfd)) goto notAnInteger; - break; - - case DISPLAY_VERSION: - if (optarg != 0) goto noValueForOption; - printVersion(); - exit(0); - - case DISPLAY_HELP: - default: - printHelp(programName); - exit(0); - } - } - free(gnuOptions); - - config_set_default (session); - return; - - -needValueForOption: - fprintf (stderr,"\nERR: AFB-daemon option [--%s] need a value i.e. --%s=xxx\n\n" - ,gnuOptions[optionIndex].name, gnuOptions[optionIndex].name); - exit (1); - -notAnInteger: - fprintf (stderr,"\nERR: AFB-daemon option [--%s] requirer an interger i.e. --%s=9\n\n" - ,gnuOptions[optionIndex].name, gnuOptions[optionIndex].name); - exit (1); - -noValueForOption: - fprintf (stderr,"\nERR: AFB-daemon option [--%s] don't take value\n\n" - ,gnuOptions[optionIndex].name); - exit (1); - -badMode: - fprintf (stderr,"\nERR: AFB-daemon option [--%s] only accepts local, global or remote.\n\n" - ,gnuOptions[optionIndex].name); - exit (1); + NOTICE("Received SIGTERM"); + exit(0); } -/*---------------------------------------------------------- - | closeSession - | try to close everything before leaving - +--------------------------------------------------------- */ -static void closeSession (int status, void *data) { - /* AFB_session *session = data; */ +static void on_sighup(int signum, siginfo_t *info, void *uctx) +{ + NOTICE("Received SIGHUP"); + /* TODO */ +} + +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); } /*---------------------------------------------------------- - | timeout signalQuit - | + | daemonize + | set the process in background +--------------------------------------------------------- */ -void signalQuit (int signum) { +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(); - sigset_t sigset; + // 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); - // unlock timeout signal to allow a new signal to come - sigemptyset (&sigset); - sigaddset (&sigset, SIGABRT); - sigprocmask (SIG_UNBLOCK, &sigset, 0); +#if 0 + setsid(); // allow father process to fully exit + sleep(2); // allow main to leave and release port +#endif +} + +/*--------------------------------------------------------- + | http server + | Handles the HTTP server + +--------------------------------------------------------- */ +static int init_alias(void *closure, char *spec) +{ + struct afb_hsrv *hsrv = closure; + char *path = strchr(spec, ':'); - fprintf (stderr, "ERR: Received signal quit\n"); - syslog (LOG_ERR, "Daemon got kill3 & quit [please report bug]"); - exit(1); + 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); } +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; + + if (!afb_hsrv_add_handler + (hsrv, config->rootapi, afb_hswitch_apis, main_apiset, 10)) + return 0; + + if (run_for_list(config->aliases, init_alias, hsrv)) + return 0; + + if (config->roothttp != NULL) { + if (!afb_hsrv_add_alias + (hsrv, "", afb_common_rootdir_get_fd(), config->roothttp, + -10, 1)) + return 0; + } -/*---------------------------------------------------------- - | listenLoop - | Main listening HTTP loop - +--------------------------------------------------------- */ -static void listenLoop (AFB_session *session) { - AFB_error err; + if (!afb_hsrv_add_handler + (hsrv, config->rootbase, afb_hswitch_one_page_api_redirect, NULL, + -20)) + return 0; - // ------ Start httpd server + return 1; +} - err = httpdStart (session); - if (err != AFB_SUCCESS) return; +static struct afb_hsrv *start_http_server() +{ + int rc; + struct afb_hsrv *hsrv; - if (session->readyfd != 0) { - static const char readystr[] = "READY=1"; - write(session->readyfd, readystr, sizeof(readystr) - 1); - close(session->readyfd); + if (afb_hreq_init_download_path(config->uploaddir)) { + ERROR("unable to set the upload directory %s", config->uploaddir); + return NULL; } - // infinite loop - httpdLoop(session); + hsrv = afb_hsrv_create(); + if (hsrv == NULL) { + ERROR("memory allocation failure"); + return NULL; + } - fprintf (stderr, "hoops returned from infinite loop [report bug]\n"); + if (!afb_hsrv_set_cache_timeout(hsrv, config->cacheTimeout) + || !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); + + rc = afb_hsrv_start(hsrv, (uint16_t) config->httpdPort, 15); + if (!rc) { + ERROR("starting of httpd failed"); + afb_hsrv_put(hsrv); + return NULL; + } + + return hsrv; } - -/*---------------------------------------------------------- - | daemonize - | set the process in background + +/*--------------------------------------------------------- + | execute_command + | +--------------------------------------------------------- */ -static void daemonize(AFB_session *session) + +static void on_sigchld(int signum, siginfo_t *info, void *uctx) +{ + 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) { - int consoleFD; - int pid; - - // open /dev/console to redirect output messAFBes - consoleFD = open(session->config->console, O_WRONLY | O_APPEND | O_CREAT , 0640); - if (consoleFD < 0) { - fprintf (stderr,"\nERR: AFB-daemon cannot open /dev/console (use --foreground)\n\n"); - exit (1); - } - - // fork process when running background mode - pid = fork (); - - // if fail nothing much to do - if (pid == -1) { - fprintf (stderr,"\nERR: AFB-daemon Failed to fork son process\n\n"); - exit (1); + 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); } - // if in father process, just leave - if (pid != 0) _exit (0); + /* return arg when no change */ + if (!chg) + return arg; - // son process get all data in standalone mode - printf ("\nAFB: background mode [pid:%d console:%s]\n", getpid(),session->config->console); + /* allocates the result */ + resu = malloc((it - arg) + dif + 1); + if (!resu) { + ERROR("out of memory"); + return NULL; + } - // 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); + /* 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; + } -#if 0 - setsid(); // allow father process to fully exit - sleep (2); // allow main to leave and release port -#endif + *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; +} - fprintf (stderr, "----------------------------\n"); - fprintf (stderr, "INF: main background pid=%d\n", getpid()); - fflush (stderr); +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); + return -1; } /*--------------------------------------------------------- - | main - | Parse option and launch action + | startup calls +--------------------------------------------------------- */ -int main(int argc, char *argv[]) { - AFB_session *session; +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); - // open syslog if ever needed - openlog("afb-daemon", 0, LOG_DAEMON); + 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); + } +} - // ------------- Build session handler & init config ------- - session = calloc (1, sizeof (AFB_session)); - session->config = calloc (1, sizeof (AFB_config)); +static void startup_call_current(struct startup_req *sreq); - on_exit(closeSession, session); - parse_arguments(argc, argv, session); +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); + } +} - // ------------------ sanity check ---------------------------------------- - if ((session->background) && (session->foreground)) { - fprintf (stderr, "ERR: cannot select foreground & background at the same time\n"); - exit (1); - } - if (session->config->httpdPort <= 0) { - fprintf (stderr, "ERR: no port is defined\n"); - exit (1); - } +static struct afb_xreq_query_itf startup_xreq_itf = +{ + .reply = startup_call_reply, + .unref = startup_call_unref +}; - initPlugins(session); - ctxStoreInit(CTX_NBCLIENTS); +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); +} - // ------------------ Some useful default values ------------------------- - if ((session->background == 0) && (session->foreground == 0)) session->foreground=1; +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); + } +} - // ------------------ clean exit on CTR-C signal ------------------------ - if (signal (SIGINT, signalQuit) == SIG_ERR || signal (SIGABRT, signalQuit) == SIG_ERR) { - fprintf (stderr, "ERR: main fail to install Signal handler\n"); - return 1; - } +/*--------------------------------------------------------- + | job for starting the daemon + +--------------------------------------------------------- */ +static void start(int signum) +{ + struct afb_hsrv *hsrv; - // let's run this program with a low priority - nice (20); + afb_debug("start-entry"); - // ------------------ Finaly Process Commands ----------------------------- - // let's not take the risk to run as ROOT - //if (getuid() == 0) goto errorNoRoot; + if (signum) { + ERROR("start aborted: received signal %s", strsignal(signum)); + exit(1); + } -#if defined(ALLOWS_SESSION_FILES) - // check session dir and create if it does not exist - if (sessionCheckdir (session) != AFB_SUCCESS) { - fprintf (stderr,"\nERR: AFB-daemon cannot read/write session dir\n\n"); - exit (1); - } -#endif - if (verbose) fprintf (stderr, "AFB: notice Init config done\n"); + // ------------------ sanity check ---------------------------------------- + if (config->httpdPort <= 0) { + ERROR("no port is defined"); + goto error; + } - // ---- run in foreground mode -------------------- - if (session->foreground) { + /* 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; + } - if (verbose) fprintf (stderr,"AFB: notice Foreground mode\n"); + /* 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; + } + + /* 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; + } - } // end foreground + /* run the startup calls */ + afb_debug("start-call"); + run_startup_calls(); - // --------- run in background mode ----------- - if (session->background) { + /* run the command */ + afb_debug("start-exec"); + if (execute_command() < 0) + goto error; - if (verbose) printf ("AFB: Entering background mode\n"); + /* ready */ + sd_notify(1, "READY=1"); + return; +error: + exit(1); +} + +/*--------------------------------------------------------- + | main + | Parse option and launch action + +--------------------------------------------------------- */ - daemonize(session); +int main(int argc, char *argv[]) +{ + afb_debug("main-entry"); - // if everything look OK then look forever - syslog (LOG_ERR, "AFB: Entering infinite loop in background mode"); + // let's run this program with a low priority + nice(20); + sd_fds_init(); - } // end background-foreground + // ------------- 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()); + /* set the daemon environment */ + setup_daemon(); - listenLoop(session); - if (verbose) printf ("\n---- Application Framework Binder Normal End ------\n"); - exit(0); + afb_debug("main-start"); + /* enter job processing */ + jobs_start(3, 0, 50, start); + WARNING("hoops returned from jobs_enter! [report bug]"); + return 1; } -