X-Git-Url: https://gerrit.automotivelinux.org/gerrit/gitweb?a=blobdiff_plain;f=src%2Fmain.c;h=0522e70841fb2615cfce318704c0fdb9a045bb87;hb=1164734afd826ca8cef0a624aa4f4c59ecb1e228;hp=78d7f79e62b1e25c9a00510bc3ac60ae676f0e1f;hpb=1edf05100332c85ee28abc0754fcb825648157c4;p=src%2Fapp-framework-binder.git diff --git a/src/main.c b/src/main.c index 78d7f79e..0522e708 100644 --- a/src/main.c +++ b/src/main.c @@ -1,671 +1,645 @@ -/* - * Copyright (C) 2015 "IoT.bzh" +/* + * Copyright (C) 2015, 2016, 2017 "IoT.bzh" * Author "Fulup Ar Foll" + * Author José Bollo * - * This program is free software: you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation, either version 3 of the License, or - * (at your option) any later version. + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. + * http://www.apache.org/licenses/LICENSE-2.0 * - * You should have received a copy of the GNU General Public License - * along with this program. If not, see . + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. */ -/* - * File: main.c - * Author: "Fulup Ar Foll" - * - * Created on 05 December 2015, 15:38 - */ - -#include "local-def.h" +#define _GNU_SOURCE +#define NO_BINDING_VERBOSE_MACRO -#include -#include +#include +#include #include -#include -#include - -static sigjmp_buf exitPoint; // context save for set/longjmp +#include +#include +#include +#include +#include + +#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" + +/* + 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) { - - fprintf (stderr,"\n----------------------------------------- \n"); - fprintf (stderr,"| AFB [Application Framework Binder] version=%s |\n", AJQ_VERSION); - fprintf (stderr,"----------------------------------------- \n"); - fprintf (stderr,"| Copyright(C) 2015 Fulup Ar Foll /IoT.bzh [fulup -at- iot.bzh]\n"); - fprintf (stderr,"| AFB comes with ABSOLUTELY NO WARRANTY.\n"); - fprintf (stderr,"| Licence [what ever makes you happy] until you fix bugs by yourself :)\n\n"); - exit (0); - } // end printVersion - - -// Define command line option -#define SET_VERBOSE 101 -#define SET_BACKGROUND 105 -#define SET_FORGROUND 106 -#define KILL_PREV_EXIT 107 -#define KILL_PREV_REST 108 -#define SET_FAKE_MOD 109 - -#define SET_TCP_PORT 120 -#define SET_ROOT_DIR 121 -#define SET_ROOT_BASE 122 -#define SET_ROOT_API 123 -#define SET_ROOT_ALIAS 124 - -#define SET_CACHE_TO 130 -#define SET_USERID 131 -#define SET_PID_FILE 132 -#define SET_SESSION_DIR 133 -#define SET_CONFIG_FILE 134 -#define SET_CONFIG_SAVE 135 -#define SET_CONFIG_EXIT 138 - -#define SET_AUTH_TOKEN 141 -#define SET_LDPATH 142 -#define SET_APITIMEOUT 143 -#define SET_CNTXTIMEOUT 144 - -#define DISPLAY_VERSION 150 -#define DISPLAY_HELP 151 - -#define SET_MODE 160 -#define SET_READYFD 161 - - -// 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"}, - {KILL_PREV_EXIT ,0,"kill" , "Kill active process if any and exit"}, - {KILL_PREV_REST ,0,"restart" , "Kill active process if any and restart"}, - - {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_ROOT_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_TO ,1,"cache-eol" , "Client cache end of live [default 3600s]"}, - - {SET_USERID ,1,"setuid" , "Change user id [default don't change]"}, - {SET_PID_FILE ,1,"pidfile" , "PID file path [default none]"}, - {SET_SESSION_DIR ,1,"sessiondir" , "Sessions file path [default rootdir/sessions]"}, - {SET_CONFIG_FILE ,1,"config" , "Config Filename [default rootdir/sessions/configs/default.AFB]"}, - {SET_CONFIG_SAVE ,0,"save" , "Save config on disk [default no]"}, - {SET_CONFIG_EXIT ,0,"saveonly" , "Save config on disk and then exit"}, - - {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, 0} - }; - -static AFB_aliasdir aliasdir[MAX_ALIAS]; -static int aliascount=0; + +/* + * 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) +{ + while (list && run(closure, list->value)) + list = list->next; + return list; +} + +static int run_start(void *closure, char *value) +{ + 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); + } +} /*---------------------------------------------------------- - | timeout signalQuit - | + | exit_handler + | Handles on exit specific actions +--------------------------------------------------------- */ -void signalQuit (int signum) { - - sigset_t sigset; +static void exit_handler() +{ + struct sigaction siga; + + 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); + exit(0); +} - // unlock timeout signal to allow a new signal to come - sigemptyset (&sigset); - sigaddset (&sigset, SIGABRT); - sigprocmask (SIG_UNBLOCK, &sigset, 0); +static void on_sigterm(int signum, siginfo_t *info, void *uctx) +{ + NOTICE("Received SIGTERM"); + exit(0); +} - fprintf (stderr, "%s ERR:Received signal quit\n",configTime()); - syslog (LOG_ERR, "Daemon got kill3 & quit [please report bug]"); - longjmp (exitPoint, signum); +static void on_sighup(int signum, siginfo_t *info, void *uctx) +{ + NOTICE("Received SIGHUP"); + /* TODO */ } +static void setup_daemon() +{ + struct sigaction siga; -/*---------------------------------------------------------- - | printHelp - | print information from long option array - +--------------------------------------------------------- */ + /* install signal handlers */ + memset(&siga, 0, sizeof siga); + siga.sa_flags = SA_SIGINFO; - 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 + 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); +} /*---------------------------------------------------------- - | writePidFile - | write a file in /var/run/AFB with pid + | daemonize + | set the process in background +--------------------------------------------------------- */ -static int writePidFile (AFB_config *config, int pid) { - FILE *file; - - // if no pid file configure just return - if (config->pidfile == NULL) return 0; - - // open pid file in write mode - file = fopen(config->pidfile,"w"); - if (file == NULL) { - fprintf (stderr,"%s ERR:writePidFile fail to open [%s]\n",configTime(), config->pidfile); - return -1; - } - - // write pid in file and close - fprintf (file, "%d\n", pid); - fclose (file); - return 0; +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); + } + // 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 +#endif } -/*---------------------------------------------------------- - | readPidFile - | read file in /var/run/AFB with pid +/*--------------------------------------------------------- + | http server + | Handles the HTTP server +--------------------------------------------------------- */ -static int readPidFile (AFB_config *config) { - int pid; - FILE *file; - int status; +static int init_alias(void *closure, char *spec) +{ + struct afb_hsrv *hsrv = closure; + char *path = strchr(spec, ':'); + + 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 (config->pidfile == NULL) return -1; +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; - // open pid file in write mode - file = fopen(config->pidfile,"r"); - if (file == NULL) { - fprintf (stderr,"%s ERR:readPidFile fail to open [%s]\n",configTime(), config->pidfile); - return -1; - } + if (!afb_hsrv_add_handler + (hsrv, config->rootapi, afb_hswitch_apis, main_apiset, 10)) + return 0; - // write pid in file and close - status = fscanf (file, "%d\n", &pid); - fclose (file); + if (run_for_list(config->aliases, init_alias, hsrv)) + return 0; - // never kill pid 0 - if (status != 1) return -1; + 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 (pid); + return 1; } -/*---------------------------------------------------------- - | closeSession - | try to close everything before leaving - +--------------------------------------------------------- */ -static void closeSession (AFB_session *session) { +static struct afb_hsrv *start_http_server() +{ + 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; + } + hsrv = afb_hsrv_create(); + if (hsrv == NULL) { + ERROR("memory allocation failure"); + return NULL; + } + + 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; } -/*---------------------------------------------------------- - | listenLoop - | Main listening HTTP loop +/*--------------------------------------------------------- + | execute_command + | +--------------------------------------------------------- */ -static void listenLoop (AFB_session *session) { - AFB_error err; - if (signal (SIGABRT, signalQuit) == SIG_ERR) { - fprintf (stderr, "%s ERR: main fail to install Signal handler\n", configTime()); - return; - } +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) +{ + 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); + } - // ------ Start httpd server - if (session->config->httpdPort > 0) { + /* return arg when no change */ + if (!chg) + return arg; - err = httpdStart (session); - if (err != AFB_SUCCESS) return; + /* allocates the result */ + resu = malloc((it - arg) + dif + 1); + if (!resu) { + ERROR("out of memory"); + return NULL; + } - if (session->readyfd != 0) { - static const char readystr[] = "READY=1"; - write(session->readyfd, readystr, sizeof(readystr) - 1); - close(session->readyfd); + /* 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; + case SUBST_CHAR: *wr++ = *it; + } + arg = ++it; } - // infinite loop - httpdLoop(session); + *wr = 0; + return resu; +} - fprintf (stderr, "hoops returned from infinite loop [report bug]\n"); - } +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]; + 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 */ + if (instanciate_command_args(port, config->token) >= 0 + && instanciate_environ(port, config->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; - char* programName = argv [0]; - int optionIndex = 0; - int optc, ind, consoleFD; - int pid, nbcmd, status; - AFB_config cliconfig; // temp structure to store CLI option before file config upload - - // ------------- Build session handler & init config ------- - session = configInit (); - memset(&cliconfig,0,sizeof(cliconfig)); - memset(&aliasdir ,0,sizeof(aliasdir)); - cliconfig.aliasdir = aliasdir; - - // GNU CLI getopts nterface. - struct option ggcOption; - struct option *gnuOptions; - - // ------------------ Process Command Line ----------------------- - - // if no argument print help and return - if (argc < 2) { - printHelp(programName); - return (-1); - } - - // build GNU getopt info from cliOptions - nbcmd = sizeof (cliOptions) / sizeof (AFB_options); - gnuOptions = malloc (sizeof (ggcOption) * 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", &cliconfig.httpdPort)) goto notAnInteger; - break; - - case SET_APITIMEOUT: - if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%d", &cliconfig.apiTimeout)) goto notAnInteger; - break; - - case SET_CNTXTIMEOUT: - if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%d", &cliconfig.cntxTimeout)) goto notAnInteger; - break; - - case SET_ROOT_DIR: - if (optarg == 0) goto needValueForOption; - cliconfig.rootdir = optarg; - if (verbose) fprintf(stderr, "Forcing Rootdir=%s\n",cliconfig.rootdir); - break; - - case SET_ROOT_BASE: - if (optarg == 0) goto needValueForOption; - cliconfig.rootbase = optarg; - if (verbose) fprintf(stderr, "Forcing Rootbase=%s\n",cliconfig.rootbase); - break; - - case SET_ROOT_API: - if (optarg == 0) goto needValueForOption; - cliconfig.rootapi = optarg; - if (verbose) fprintf(stderr, "Forcing Rootapi=%s\n",cliconfig.rootapi); - break; - - case SET_ROOT_ALIAS: - if (optarg == 0) goto needValueForOption; - if (aliascount < MAX_ALIAS) { - aliasdir[aliascount].url = strsep(&optarg,":"); - aliasdir[aliascount].path = strsep(&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:%s] %s ignored\n", optarg, MAX_ALIAS-1); - } - break; - - case SET_AUTH_TOKEN: - if (optarg == 0) goto needValueForOption; - cliconfig.token = optarg; - break; - - case SET_LDPATH: - if (optarg == 0) goto needValueForOption; - cliconfig.ldpaths = optarg; - break; - - case SET_PID_FILE: - if (optarg == 0) goto needValueForOption; - cliconfig.pidfile = optarg; - break; - - case SET_SESSION_DIR: - if (optarg == 0) goto needValueForOption; - cliconfig.sessiondir = optarg; - break; - - case SET_CONFIG_FILE: - if (optarg == 0) goto needValueForOption; - cliconfig.configfile = optarg; - break; - - case SET_CACHE_TO: - if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%d", &cliconfig.cacheTimeout)) goto notAnInteger; - break; - - case SET_CONFIG_EXIT: - if (optarg != 0) goto noValueForOption; - session->configsave = 1; - session->forceexit = 1; - break; - - case SET_CONFIG_SAVE: - if (optarg != 0) goto noValueForOption; - session->configsave = 1; - break; - - case SET_USERID: - if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%s", &cliconfig.setuid)) 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 KILL_PREV_REST: - if (optarg != 0) goto noValueForOption; - session->killPrevious = 1; - break; - - case KILL_PREV_EXIT: - if (optarg != 0) goto noValueForOption; - session->killPrevious = 2; - break; - - case SET_MODE: - if (optarg == 0) goto needValueForOption; - if (!strcmp(optarg, "local")) cliconfig.mode = AFB_MODE_LOCAL; - else if (!strcmp(optarg, "remote")) cliconfig.mode = AFB_MODE_REMOTE; - else if (!strcmp(optarg, "global")) cliconfig.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(); - goto normalExit; - - case DISPLAY_HELP: - default: - printHelp(programName); - goto normalExit; - - } - } - - // if exist merge config file with CLI arguments - configLoadFile (session, &cliconfig); - initPlugins(session); - - // ------------------ sanity check ---------------------------------------- - if ((session->background) && (session->foreground)) { - fprintf (stderr, "%s ERR: cannot select foreground & background at the same time\n",configTime()); - exit (-1); - } - - // ------------------ Some useful default values ------------------------- - if ((session->background == 0) && (session->foreground == 0)) session->foreground=1; - - // open syslog if ever needed - openlog("AFB-log", 0, LOG_DAEMON); - - // -------------- Try to kill any previous process if asked --------------------- - if (session->killPrevious) { - pid = readPidFile (session->config); // enforce commandline option - switch (pid) { - case -1: - fprintf (stderr, "%s ERR:main --kill ignored no PID file [%s]\n",configTime(), session->config->pidfile); - break; - case 0: - fprintf (stderr, "%s ERR:main --kill ignored no active AFB process\n",configTime()); - break; - default: - status = kill (pid,SIGINT ); - if (status == 0) { - if (verbose) printf ("%s INF:main signal INTR sent to pid:%d \n", configTime(), pid); - } else { - // try kill -9 - status = kill (pid,9); - if (status != 0) fprintf (stderr, "%s ERR:main failled to killed pid=%d \n",configTime(), pid); - } - } // end switch pid - - if (session->killPrevious >= 2) goto normalExit; - } // end killPrevious - - - // ------------------ clean exit on CTR-C signal ------------------------ - if (signal (SIGINT, signalQuit) == SIG_ERR) { - fprintf (stderr, "%s Quit Signal received.",configTime()); - return (-1); - } - - // save exitPoint context when returning from longjmp closeSession and exit - status = setjmp (exitPoint); // return !+ when coming from longjmp - if (status != 0) { - if (verbose) printf ("INF:main returning from longjump after signal [%d]\n", status); - closeSession (session); - goto exitOnSignal; - } - - // let's run this program with a low priority - status=nice (20); - - // ------------------ Finaly Process Commands ----------------------------- - // if --save then store config on disk upfront - if (session->configsave) configStoreFile (session); - if (session->forceexit) exit (0); - - if (session->config->setuid) { - int err; - struct passwd *passwd; - passwd=getpwnam(session->config->setuid); - - if (passwd == NULL) goto errorSetuid; - - err = setuid(passwd->pw_uid); - if (err) goto errorSetuid; - } - - // let's not take the risk to run as ROOT - //if (getuid() == 0) goto errorNoRoot; - - // check session dir and create if it does not exist - if (sessionCheckdir (session) != AFB_SUCCESS) goto errSessiondir; - if (verbose) fprintf (stderr, "AFB:notice Init config done\n"); - - - - // ---- run in foreground mode -------------------- - if (session->foreground) { - - if (verbose) fprintf (stderr,"AFB:notice Foreground mode\n"); - - // write a pid file for --kill-previous and --raise-debug option - status = writePidFile (session->config, getpid()); - if (status == -1) goto errorPidFile; - - // enter listening loop in foreground - listenLoop(session); - goto exitInitLoop; - } // end foreground - - - // --------- run in background mode ----------- - if (session->background) { - - // if (status != 0) goto errorCommand; - if (verbose) printf ("AFB: Entering background mode\n"); - - // open /dev/console to redirect output messAFBes - consoleFD = open(session->config->console, O_WRONLY | O_APPEND | O_CREAT , 0640); - if (consoleFD < 0) goto errConsole; - - // fork process when running background mode - pid = fork (); - - // son process get all data in standalone mode - if (pid == 0) { - - printf ("\nAFB: background mode [pid:%d console:%s]\n", getpid(),session->config->console); - if (verbose) printf ("AFB:info use '%s --restart --rootdir=%s # [--pidfile=%s] to restart daemon\n", programName,session->config->rootdir, session->config->pidfile); - - // redirect default I/O on console - close (2); status=dup(consoleFD); // redirect stderr - close (1); status=dup(consoleFD); // redirect stdout - close (0); // no need for stdin - close (consoleFD); - - setsid(); // allow father process to fully exit - sleep (2); // allow main to leave and release port - - fprintf (stderr, "----------------------------\n"); - fprintf (stderr, "%s INF:main background pid=%d\n", configTime(), getpid()); - fflush (stderr); - - // if everything look OK then look forever - syslog (LOG_ERR, "AFB: Entering infinite loop in background mode"); +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 iserror, struct json_object *obj) +{ + struct startup_req *sreq = CONTAINER_OF_XREQ(struct startup_req, xreq); + + if (!iserror) + 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 +}; + +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); +} - // should normally never return from this loop - listenLoop(session); - syslog (LOG_ERR, "AFB:FAIL background infinite loop exited check [%s]\n", session->config->console); +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); + } +} - goto exitInitLoop; - } +/*--------------------------------------------------------- + | job for starting the daemon + +--------------------------------------------------------- */ - // if fail nothing much to do - if (pid == -1) goto errorFork; +static void start() +{ + struct afb_hsrv *hsrv; - // fork worked and we are in father process - status = writePidFile (session->config, pid); - if (status == -1) goto errorPidFile; + // ------------------ sanity check ---------------------------------------- + if (config->httpdPort <= 0) { + ERROR("no port is defined"); + goto error; + } - // we are in father process, we don't need this one - _exit (0); + /* 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; + } - } // end background-foreground + /* 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; + } -normalExit: - closeSession (session); // try to close everything before leaving - if (verbose) printf ("\n---- Application Framework Binder Normal End ------\n"); - exit(0); + /* 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); -// ------------- Fatal ERROR display error and quit ------------- -errorSetuid: - fprintf (stderr,"\nERR:AFB-daemon Failed to change UID to username=[%s]\n\n", session->config->setuid); - exit (-1); - -//errorNoRoot: -// fprintf (stderr,"\nERR:AFB-daemon Not allow to run as root [use --seteuid=username option]\n\n"); -// exit (-1); + /* load bindings */ + 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, "the binding path set"); + apiset_start_list(config->so_bindings, afb_api_so_add_binding, "the binding"); -errorPidFile: - fprintf (stderr,"\nERR:AFB-daemon Failed to write pid file [%s]\n\n", session->config->pidfile); - exit (-1); + 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"); -errorFork: - fprintf (stderr,"\nERR:AFB-daemon Failed to fork son process\n\n"); - exit (-1); + DEBUG("Init config done"); -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); + /* start the services */ + if (afb_apiset_start_all_services(main_apiset, 1) < 0) + goto error; -noValueForOption: - fprintf (stderr,"\nERR:AFB-daemon option [--%s] don't take value\n\n" - ,gnuOptions[optionIndex].name); - exit (-1); + /* start the HTTP server */ + if (!config->noHttpd) { + hsrv = start_http_server(); + if (hsrv == NULL) + goto error; + } -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); + /* run the command */ + if (execute_command() < 0) + goto error; -badMode: - fprintf (stderr,"\nERR:AFB-daemon option [--%s] only accepts local, global or remote.\n\n" - ,gnuOptions[optionIndex].name); - exit (-1); + /* ready */ + sd_notify(1, "READY=1"); -exitOnSignal: - fprintf (stderr,"\n%s INF:AFB-daemon pid=%d received exit signal (Hopefully crtl-C or --kill-previous !!!)\n\n" - ,configTime(), getpid()); - exit (-1); + /* run the startup calls */ + run_startup_calls(); + return; +error: + exit(1); +} -errConsole: - fprintf (stderr,"\nERR:AFB-daemon cannot open /dev/console (use --foreground)\n\n"); - exit (-1); +/*--------------------------------------------------------- + | main + | Parse option and launch action + +--------------------------------------------------------- */ -errSessiondir: - fprintf (stderr,"\nERR:AFB-daemon cannot read/write session dir\n\n"); - exit (-1); +int main(int argc, char *argv[]) +{ + // 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); + INFO("running with pid %d", getpid()); + + // --------- run ----------- + if (config->background) { + // --------- in background mode ----------- + INFO("entering background mode"); + daemonize(); + } else { + // ---- in foreground mode -------------------- + INFO("entering foreground mode"); + } -errSoundCard: - fprintf (stderr,"\nERR:AFB-daemon fail to probe sound cards\n\n"); - exit (-1); + /* set the daemon environment */ + setup_daemon(); -exitInitLoop: - // try to unlink pid file if any - if (session->background && session->config->pidfile != NULL) unlink (session->config->pidfile); - exit (-1); - -}; /* END AFB-daemon() */ + /* enter job processing */ + jobs_start(3, 0, 50, start); + WARNING("hoops returned from jobs_enter! [report bug]"); + return 1; +}