X-Git-Url: https://gerrit.automotivelinux.org/gerrit/gitweb?a=blobdiff_plain;f=src%2Fmain.c;h=f1ebd969ac2b751175b6dac490374cef4156d933;hb=e2c431bcda7f057b4767c3e8142b9e0ca0d32bc6;hp=f81bfef843a65f9282c8847301f0ac3e2415c18c;hpb=83ce3b29a598141f8f28afde655beb07ec78c312;p=src%2Fapp-framework-binder.git diff --git a/src/main.c b/src/main.c index f81bfef8..f1ebd969 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 * @@ -17,608 +17,453 @@ */ #define _GNU_SOURCE +#define NO_BINDING_VERBOSE_MACRO + +#include #include #include -#include -#include -#include -#include +#include #include #include -#include +#include + +#include +#include -#include "afb-plugin.h" +#include -#include "local-def.h" +#include "afb-config.h" #include "afb-hswitch.h" +#include "afb-apis.h" #include "afb-api-so.h" +#include "afb-api-dbus.h" +#include "afb-api-ws.h" #include "afb-hsrv.h" -#include "session.h" +#include "afb-context.h" +#include "afb-hreq.h" +#include "afb-sig-handler.h" +#include "afb-thread.h" +#include "afb-session.h" #include "verbose.h" -#include "utils-upoll.h" - -#if !defined(PLUGIN_INSTALL_DIR) -#error "you should define PLUGIN_INSTALL_DIR" -#endif - -#define AFB_VERSION "0.1" - -// 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 - -static struct afb_hsrv *start(AFB_config * config); - -// 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"}, +#include "afb-common.h" +#include "afb-hook.h" +#include "sd-fds.h" - {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; +/* + 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 +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 void config_set_default (AFB_config * config) +static int run_start(void *closure, char *value) { - // 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); - } + int (*starter) (const char *value) = closure; + return starter(value) >= 0; } - -/*---------------------------------------------------------- - | printHelp - | print information from long option array - +--------------------------------------------------------- */ - - 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 - -/*--------------------------------------------------------- - | main - | Parse option and launch action - +--------------------------------------------------------- */ - -static void parse_arguments(int argc, char *argv[], AFB_session *session) +static void start_list(struct afb_config_list *list, + int (*starter) (const char *value), const char *message) { - 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: - verbosity++; - 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 (verbosity) fprintf(stderr, "Forcing Rootdir=%s\n",session->config->rootdir); - break; - - case SET_ROOT_BASE: - if (optarg == 0) goto needValueForOption; - session->config->rootbase = optarg; - if (verbosity) fprintf(stderr, "Forcing Rootbase=%s\n",session->config->rootbase); - break; - - case SET_ROOT_API: - if (optarg == 0) goto needValueForOption; - session->config->rootapi = optarg; - if (verbosity) 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 (verbosity) 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_FORGROUND: - if (optarg != 0) goto noValueForOption; - session->background = 0; - 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->config); - 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); + list = run_for_list(list, run_start, starter); + if (list) { + ERROR("can't start %s %s", message, list->value); + exit(1); + } } /*---------------------------------------------------------- - | closeSession - | try to close everything before leaving + | exit_handler + | Handles on exit specific actions +--------------------------------------------------------- */ -static void closeSession (int status, void *data) { - /* AFB_session *session = data; */ +static void exit_handler() +{ + /* TODO: check whether using SIGHUP isn't better */ + if (SELF_PGROUP) + killpg(0, SIGKILL); + else if (childpid > 0) + killpg(childpid, SIGKILL); } /*---------------------------------------------------------- - | timeout signalQuit + | daemonize + | set the process in background +--------------------------------------------------------- */ -void signalQuit (int signum) +static void daemonize() { - fprintf(stderr, "Terminating signal received %s\n", strsignal(signum)); - exit(1); -} + 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); + } + // 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); -/*---------------------------------------------------------- - | Error signals - | +#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 +--------------------------------------------------------- */ -__thread sigjmp_buf *error_handler; -static void signalError(int signum) +static int init_alias(void *closure, char *spec) { - sigset_t sigset; - - // unlock signal to allow a new signal to come - if (error_handler != NULL) { - sigemptyset(&sigset); - sigaddset(&sigset, signum); - sigprocmask(SIG_UNBLOCK, &sigset, 0); - longjmp(*error_handler, signum); + struct afb_hsrv *hsrv = closure; + char *path = strchr(spec, ':'); + + if (path == NULL) { + ERROR("Missing ':' in alias %s. Alias ignored", spec); + return 1; } - if (signum == SIGALRM) - return; - fprintf(stderr, "Unmonitored signal received %s\n", strsignal(signum)); - exit(2); + *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 void install_error_handlers() +static int init_http_server(struct afb_hsrv *hsrv) { - int i, signals[] = { SIGALRM, SIGSEGV, SIGFPE, 0 }; + if (!afb_hsrv_add_handler + (hsrv, config->rootapi, afb_hswitch_websocket_switch, NULL, 20)) + return 0; - for (i = 0; signals[i] != 0; i++) { - if (signal(signals[i], signalError) == SIG_ERR) { - fprintf(stderr, "Signal handler error\n"); - exit(1); - } + if (!afb_hsrv_add_handler + (hsrv, config->rootapi, afb_hswitch_apis, NULL, 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; } + + if (!afb_hsrv_add_handler + (hsrv, config->rootbase, afb_hswitch_one_page_api_redirect, NULL, + -20)) + return 0; + + return 1; } -/*---------------------------------------------------------- - | daemonize - | set the process in background - +--------------------------------------------------------- */ -static void daemonize(AFB_session *session) +static struct afb_hsrv *start_http_server() { - 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); + int rc; + struct afb_hsrv *hsrv; + + if (afb_hreq_init_download_path(config->uploaddir)) { + ERROR("unable to set the upload directory %s", config->uploaddir); + return NULL; } - // if in father process, just leave - if (pid != 0) _exit (0); + hsrv = afb_hsrv_create(); + if (hsrv == NULL) { + ERROR("memory allocation failure"); + return NULL; + } - // son process get all data in standalone mode - printf ("\nAFB: background mode [pid:%d console:%s]\n", getpid(),session->config->console); + if (!afb_hsrv_set_cache_timeout(hsrv, config->cacheTimeout) + || !init_http_server(hsrv)) { + ERROR("initialisation of httpd failed"); + afb_hsrv_put(hsrv); + 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); + NOTICE("Waiting port=%d rootdir=%s", config->httpdPort, config->rootdir); + NOTICE("Browser URL= http:/*localhost:%d", config->httpdPort); -#if 0 - setsid(); // allow father process to fully exit - sleep (2); // allow main to leave and release port -#endif + rc = afb_hsrv_start(hsrv, (uint16_t) config->httpdPort, 15); + if (!rc) { + ERROR("starting of httpd failed"); + afb_hsrv_put(hsrv); + return NULL; + } - fprintf (stderr, "----------------------------\n"); - fprintf (stderr, "INF: main background pid=%d\n", getpid()); - fflush (stderr); + return hsrv; } /*--------------------------------------------------------- - | main - | Parse option and launch action + | execute_command + | +--------------------------------------------------------- */ -int main(int argc, char *argv[]) { - AFB_session *session; - struct afb_hsrv *hsrv; - - // open syslog if ever needed - openlog("afb-daemon", 0, LOG_DAEMON); +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); + } + } +} - // ------------- Build session handler & init config ------- - session = calloc (1, sizeof (AFB_session)); - session->config = calloc (1, sizeof (AFB_config)); +/* +# @@ @ +# @p port +# @t token +*/ - on_exit(closeSession, session); - parse_arguments(argc, argv, session); +#define SUBST_CHAR '@' +#define SUBST_STR "@" - // ------------------ sanity check ---------------------------------------- - if (session->config->httpdPort <= 0) { - fprintf (stderr, "ERR: no port is defined\n"); - exit (1); - } +static int instanciate_command_args() +{ + char *orig, *repl, *sub, *val, port[20]; + int i, rc, r; + size_t s, l; + + rc = snprintf(port, sizeof port, "%d", config->httpdPort); + if (rc < 0 || rc >= (int)(sizeof port)) + return -1; + + for (i = 0 ; (orig = config->exec[i]) ; i++) { + repl = 0; + s = 0; + for(;;) { + sub = strchrnul(orig, SUBST_CHAR); + l = sub - orig; + if (repl) + repl = mempcpy(repl, orig, l); + else + s += l; + if (!*sub) { + /* at end */ + if (repl || orig == config->exec[i]) + break; + repl = malloc(1 + s); + if (!repl) + return -1; + orig = config->exec[i]; + config->exec[i] = repl; + repl[s] = 0; + } else { + r = 2; + switch(sub[1]) { + case 'p': val = port; break; + case 't': val = config->token ? : ""; break; + default: r = 1; + case SUBST_CHAR: val = SUBST_STR; break; + } + orig = &sub[r]; + l = strlen(val); + if (repl) + repl = mempcpy(repl, val, l); + else + s += l; + } + } + } + return 0; +} - if (session->config->ldpaths) - afb_api_so_add_pathset(session->config->ldpaths); +static int execute_command() +{ + struct sigaction siga; - ctxStoreInit(CTX_NBCLIENTS, session->config->cntxTimeout, session->config->token); + /* check whether a command is to execute or not */ + if (!config->exec || !config->exec[0]) + return 0; - install_error_handlers(); + if (SELF_PGROUP) + setpgid(0, 0); - // ------------------ 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; - } + /* install signal handler */ + memset(&siga, 0, sizeof siga); + siga.sa_sigaction = on_sigchld; + siga.sa_flags = SA_SIGINFO; + sigaction(SIGCHLD, &siga, NULL); - // let's run this program with a low priority - nice (20); + /* fork now */ + childpid = fork(); + if (childpid) + return 0; - // ------------------ Finaly Process Commands ----------------------------- - // let's not take the risk to run as ROOT - //if (getuid() == 0) goto errorNoRoot; + /* makes arguments */ + if (instanciate_command_args() >= 0) { + 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; +} - if (verbosity) fprintf (stderr, "AFB: notice Init config done\n"); +/*--------------------------------------------------------- + | main + | Parse option and launch action + +--------------------------------------------------------- */ - // --------- run ----------- - if (session->background) { - // --------- in background mode ----------- - if (verbosity) printf ("AFB: Entering background mode\n"); - daemonize(session); - } else { - // ---- in foreground mode -------------------- - if (verbosity) fprintf (stderr,"AFB: notice Foreground mode\n"); +int main(int argc, char *argv[]) +{ + struct afb_hsrv *hsrv; + struct sd_event *eventloop; - } + LOGAUTH("afb-daemon"); - hsrv = start (session->config); - if (hsrv == NULL) - exit(1); + sd_fds_init(); - if (session->readyfd != 0) { - static const char readystr[] = "READY=1"; - write(session->readyfd, readystr, sizeof(readystr) - 1); - close(session->readyfd); - } + // ------------- Build session handler & init config ------- + config = afb_config_parse_arguments(argc, argv); + atexit(exit_handler); - // infinite loop - for(;;) - upoll_wait(30000); + // ------------------ sanity check ---------------------------------------- + if (config->httpdPort <= 0) { + ERROR("no port is defined"); + exit(1); + } - if (verbosity) - fprintf (stderr, "hoops returned from infinite loop [report bug]\n"); + mkdir(config->workdir, S_IRWXU | S_IRGRP | S_IXGRP); + if (chdir(config->workdir) < 0) { + ERROR("Can't enter working dir %s", config->workdir); + exit(1); + } - return 0; -} + afb_api_so_set_timeout(config->apiTimeout); + start_list(config->dbus_clients, afb_api_dbus_add_client, "the afb-dbus client"); + start_list(config->ws_clients, afb_api_ws_add_client, "the afb-websocket client"); + start_list(config->ldpaths, afb_api_so_add_pathset, "the binding path set"); + start_list(config->so_bindings, afb_api_so_add_binding, "the binding"); -static int init(struct afb_hsrv *hsrv, AFB_config * config) -{ - int idx; + afb_session_init(config->nbSessionMax, config->cntxTimeout, config->token, afb_apis_count()); - if (!afb_hsrv_add_handler(hsrv, config->rootapi, afb_hswitch_websocket_switch, NULL, 20)) - return 0; + start_list(config->dbus_servers, afb_api_dbus_add_server, "the afb-dbus service"); + start_list(config->ws_servers, afb_api_ws_add_server, "the afb-websocket service"); - if (!afb_hsrv_add_handler(hsrv, config->rootapi, afb_hswitch_apis, NULL, 10)) - return 0; + if (!afb_hreq_init_cookie(config->httpdPort, config->rootapi, config->cntxTimeout)) { + ERROR("initialisation of cookies failed"); + exit(1); + } - for (idx = 0; config->aliasdir[idx].url != NULL; idx++) - if (!afb_hsrv_add_alias (hsrv, config->aliasdir[idx].url, config->aliasdir[idx].path, 0)) - return 0; + if (afb_sig_handler_init() < 0) { + ERROR("failed to initialise signal handlers"); + return 1; + } - if (!afb_hsrv_add_alias(hsrv, "", config->rootdir, -10)) - return 0; + // set the root dir + if (afb_common_rootdir_set(config->rootdir) < 0) { + ERROR("failed to set common root directory"); + return 1; + } - if (!afb_hsrv_add_handler(hsrv, config->rootbase, afb_hswitch_one_page_api_redirect, NULL, -20)) - return 0; + if (afb_thread_init(3, 1, 20) < 0) { + ERROR("failed to initialise threading"); + return 1; + } + // let's run this program with a low priority + nice(20); + + // ------------------ Finaly Process Commands ----------------------------- + // let's not take the risk to run as ROOT + //if (getuid() == 0) goto errorNoRoot; + + DEBUG("Init config done"); + + // --------- run ----------- + if (config->background) { + // --------- in background mode ----------- + INFO("entering background mode"); + daemonize(); + } else { + // ---- in foreground mode -------------------- + INFO("entering foreground mode"); + } - return 1; -} + /* ignore any SIGPIPE */ + signal(SIGPIPE, SIG_IGN); -static struct afb_hsrv *start(AFB_config * config) -{ - int rc; - struct afb_hsrv *hsrv; + /* install trace of requests */ + if (config->tracereq) + afb_hook_req_create(NULL, NULL, NULL, config->tracereq, NULL, NULL); - hsrv = afb_hsrv_create(); - if (hsrv == NULL) { - fprintf(stderr, "memory allocation failure\n"); - return NULL; - } + /* start the services */ + if (afb_apis_start_all_services(1) < 0) + exit(1); - if (!afb_hsrv_set_cache_timeout(hsrv, config->cacheTimeout) - || !init(hsrv, config)) { - printf("Error: initialisation of httpd failed"); - afb_hsrv_put(hsrv); - return NULL; + /* start the HTTP server */ + if (!config->noHttpd) { + hsrv = start_http_server(); + if (hsrv == NULL) + exit(1); } - if (verbosity) { - printf("AFB:notice Waiting port=%d rootdir=%s\n", config->httpdPort, config->rootdir); - printf("AFB:notice Browser URL= http:/*localhost:%d\n", config->httpdPort); - } + /* run the command */ + if (execute_command() < 0) + exit(1); - rc = afb_hsrv_start(hsrv, (uint16_t) config->httpdPort, 15); - if (!rc) { - printf("Error: starting of httpd failed"); - afb_hsrv_put(hsrv); - return NULL; + /* signal that ready */ + if (config->readyfd != 0) { + static const char readystr[] = "READY=1"; + write(config->readyfd, readystr, sizeof(readystr) - 1); + close(config->readyfd); } - return hsrv; -} + // infinite loop + eventloop = afb_common_get_event_loop(); + sd_notify(1, "READY=1"); + for (;;) + sd_event_run(eventloop, 30000000); + WARNING("hoops returned from infinite loop [report bug]"); + return 0; +}