X-Git-Url: https://gerrit.automotivelinux.org/gerrit/gitweb?a=blobdiff_plain;f=src%2Fmain.c;h=aa4b84c353c34e2e994971e3aff2ddfb8960e101;hb=8449480d29d2e03d9094ddce82557b251deb6567;hp=af90eaec35fa6cb751a952c859ab44fa664f2e38;hpb=b0d938b609b645b0353e95b7e5c6313f20811da9;p=src%2Fapp-framework-binder.git diff --git a/src/main.c b/src/main.c index af90eaec..aa4b84c3 100644 --- a/src/main.c +++ b/src/main.c @@ -1,88 +1,71 @@ /* * Copyright (C) 2015 "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 - */ +#define _GNU_SOURCE +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include "afb-plugin.h" #include "local-def.h" +#include "afb-apis.h" +#include "afb-hsrv.h" +#include "session.h" +#include "verbose.h" +#include "utils-upoll.h" -#include -#include -#include -#include -#include +#if !defined(PLUGIN_INSTALL_DIR) +#error "you should define PLUGIN_INSTALL_DIR" +#endif #define AFB_VERSION "0.1" -static sigjmp_buf exitPoint; // context save for set/longjmp +// Define command line option +#define SET_VERBOSE 1 +#define SET_BACKGROUND 2 +#define SET_FORGROUND 3 -/*---------------------------------------------------------- - | printversion - | print version and copyright - +--------------------------------------------------------- */ - static void printVersion (void) { +#define SET_TCP_PORT 5 +#define SET_ROOT_DIR 6 +#define SET_ROOT_BASE 7 +#define SET_ROOT_API 8 +#define SET_ALIAS 9 - fprintf (stderr,"\n----------------------------------------- \n"); - fprintf (stderr,"| AFB [Application Framework Binder] version=%s |\n", AFB_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 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 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 +#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 { @@ -99,25 +82,18 @@ static AFB_options cliOptions [] = { {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_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_CACHE_TIMEOUT,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]"}, @@ -131,24 +107,81 @@ static AFB_options cliOptions [] = { }; static AFB_aliasdir aliasdir[MAX_ALIAS]; -static int aliascount=0; +static int aliascount = 0; + /*---------------------------------------------------------- - | timeout signalQuit - | + | printversion + | print version and copyright +--------------------------------------------------------- */ -void signalQuit (int signum) { - - sigset_t sigset; - - // unlock timeout signal to allow a new signal to come - sigemptyset (&sigset); - sigaddset (&sigset, SIGABRT); - sigprocmask (SIG_UNBLOCK, &sigset, 0); +static void printVersion (void) +{ + 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); +} - fprintf (stderr, "%s ERR:Received signal quit\n",configTime()); - syslog (LOG_ERR, "Daemon got kill3 & quit [please report bug]"); - longjmp (exitPoint, signum); +// load config from disk and merge with CLI option +static void config_set_default (AFB_session * session) +{ + // 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); + } } @@ -176,125 +209,29 @@ void signalQuit (int signum) { fprintf (stderr,"Example:\n %s\\\n --verbose --port=1234 --token='azerty' --ldpaths=build/plugins:/usr/lib64/agl/plugins\n", name); } // end printHelp -/*---------------------------------------------------------- - | writePidFile - | write a file in /var/run/AFB with pid - +--------------------------------------------------------- */ -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; -} - -/*---------------------------------------------------------- - | readPidFile - | read file in /var/run/AFB with pid - +--------------------------------------------------------- */ -static int readPidFile (AFB_config *config) { - int pid; - FILE *file; - int status; - - if (config->pidfile == NULL) return -1; - - // 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; - } - - // write pid in file and close - status = fscanf (file, "%d\n", &pid); - fclose (file); - - // never kill pid 0 - if (status != 1) return -1; - - return (pid); -} - -/*---------------------------------------------------------- - | closeSession - | try to close everything before leaving - +--------------------------------------------------------- */ -static void closeSession (AFB_session *session) { - - -} - -/*---------------------------------------------------------- - | listenLoop - | Main listening HTTP loop - +--------------------------------------------------------- */ -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; - } - - // ------ Start httpd server - if (session->config->httpdPort > 0) { - - err = httpdStart (session); - if (err != AFB_SUCCESS) return; - - if (session->readyfd != 0) { - static const char readystr[] = "READY=1"; - write(session->readyfd, readystr, sizeof(readystr) - 1); - close(session->readyfd); - } - - // infinite loop - httpdLoop(session); - - fprintf (stderr, "hoops returned from infinite loop [report bug]\n"); - } -} - /*--------------------------------------------------------- | main | Parse option and launch action +--------------------------------------------------------- */ -int main(int argc, char *argv[]) { - AFB_session *session; +static void parse_arguments(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 + int optc, ind; + int nbcmd; + struct option *gnuOptions; // ------------- 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 *gnuOptions; + session->config->aliasdir = aliasdir; // ------------------ Process Command Line ----------------------- // if no argument print help and return if (argc < 2) { printHelp(programName); - return (-1); + exit(1); } // build GNU getopt info from cliOptions @@ -314,43 +251,43 @@ int main(int argc, char *argv[]) { switch (optc) { case SET_VERBOSE: - verbose = 1; + verbosity++; break; case SET_TCP_PORT: if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%d", &cliconfig.httpdPort)) goto notAnInteger; + if (!sscanf (optarg, "%d", &session->config->httpdPort)) goto notAnInteger; break; case SET_APITIMEOUT: if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%d", &cliconfig.apiTimeout)) goto notAnInteger; + if (!sscanf (optarg, "%d", &session->config->apiTimeout)) goto notAnInteger; break; case SET_CNTXTIMEOUT: if (optarg == 0) goto needValueForOption; - if (!sscanf (optarg, "%d", &cliconfig.cntxTimeout)) goto notAnInteger; + if (!sscanf (optarg, "%d", &session->config->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); + 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; - cliconfig.rootbase = optarg; - if (verbose) fprintf(stderr, "Forcing Rootbase=%s\n",cliconfig.rootbase); + 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; - cliconfig.rootapi = optarg; - if (verbose) fprintf(stderr, "Forcing Rootapi=%s\n",cliconfig.rootapi); + session->config->rootapi = optarg; + if (verbosity) fprintf(stderr, "Forcing Rootapi=%s\n",session->config->rootapi); break; - case SET_ROOT_ALIAS: + case SET_ALIAS: if (optarg == 0) goto needValueForOption; if (aliascount < MAX_ALIAS) { aliasdir[aliascount].url = strsep(&optarg,":"); @@ -359,7 +296,7 @@ int main(int argc, char *argv[]) { } 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); + if (verbosity) fprintf(stderr, "Alias url=%s path=%s\n", aliasdir[aliascount].url, aliasdir[aliascount].path); aliascount++; } } else { @@ -369,58 +306,27 @@ int main(int argc, char *argv[]) { case SET_AUTH_TOKEN: if (optarg == 0) goto needValueForOption; - cliconfig.token = optarg; + session->config->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; + session->config->ldpaths = 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; + session->config->sessiondir = optarg; break; - case SET_USERID: + case SET_CACHE_TIMEOUT: if (optarg == 0) goto needValueForOption; - cliconfig.setuid = optarg; - break; - - case SET_FAKE_MOD: - if (optarg != 0) goto noValueForOption; - session->fakemod = 1; + if (!sscanf (optarg, "%d", &session->config->cacheTimeout)) goto notAnInteger; break; case SET_FORGROUND: if (optarg != 0) goto noValueForOption; - session->foreground = 1; + session->background = 0; break; case SET_BACKGROUND: @@ -428,21 +334,11 @@ int main(int argc, char *argv[]) { 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; + 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; @@ -454,227 +350,216 @@ int main(int argc, char *argv[]) { case DISPLAY_VERSION: if (optarg != 0) goto noValueForOption; printVersion(); - goto normalExit; + exit(0); case DISPLAY_HELP: default: printHelp(programName); - goto normalExit; - + exit(0); } } + free(gnuOptions); - // 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); - } + config_set_default (session); + return; - // ------------------ 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; +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); - // 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"); +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); +} - // ---- run in foreground mode -------------------- - if (session->foreground) { +/*---------------------------------------------------------- + | closeSession + | try to close everything before leaving + +--------------------------------------------------------- */ +static void closeSession (int status, void *data) { + /* AFB_session *session = data; */ +} - if (verbose) fprintf (stderr,"AFB:notice Foreground mode\n"); +/*---------------------------------------------------------- + | timeout signalQuit + +--------------------------------------------------------- */ +void signalQuit (int signum) +{ + fprintf(stderr, "Terminating signal received %s\n", strsignal(signum)); + exit(1); +} - // 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 +/*---------------------------------------------------------- + | Error signals + | + +--------------------------------------------------------- */ +__thread sigjmp_buf *error_handler; +static void signalError(int signum) +{ + 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); + } + if (signum == SIGALRM) + return; + fprintf(stderr, "Unmonitored signal received %s\n", strsignal(signum)); + exit(2); +} +static void install_error_handlers() +{ + int i, signals[] = { SIGALRM, SIGSEGV, SIGFPE, 0 }; - // --------- run in background mode ----------- - if (session->background) { + for (i = 0; signals[i] != 0; i++) { + if (signal(signals[i], signalError) == SIG_ERR) { + fprintf(stderr, "Signal handler error\n"); + exit(1); + } + } +} - // if (status != 0) goto errorCommand; - if (verbose) printf ("AFB: Entering background mode\n"); +/*---------------------------------------------------------- + | daemonize + | set the process in background + +--------------------------------------------------------- */ +static void daemonize(AFB_session *session) +{ + 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) goto errConsole; + 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 (); - // son process get all data in standalone mode - if (pid == 0) { + // if fail nothing much to do + if (pid == -1) { + fprintf (stderr,"\nERR: AFB-daemon Failed to fork son process\n\n"); + exit (1); + } - 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); + // if in father process, just leave + if (pid != 0) _exit (0); - // 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); + // son process get all data in standalone mode + printf ("\nAFB: background mode [pid:%d console:%s]\n", getpid(),session->config->console); - setsid(); // allow father process to fully exit - sleep (2); // allow main to leave and release port + // 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 fprintf (stderr, "----------------------------\n"); - fprintf (stderr, "%s INF:main background pid=%d\n", configTime(), getpid()); + fprintf (stderr, "INF: main background pid=%d\n", getpid()); fflush (stderr); +} - // if everything look OK then look forever - syslog (LOG_ERR, "AFB: Entering infinite loop in background mode"); +/*--------------------------------------------------------- + | main + | Parse option and launch action + +--------------------------------------------------------- */ - // should normally never return from this loop - listenLoop(session); - syslog (LOG_ERR, "AFB:FAIL background infinite loop exited check [%s]\n", session->config->console); +int main(int argc, char *argv[]) { + int rc; + AFB_session *session; - goto exitInitLoop; - } + // open syslog if ever needed + openlog("afb-daemon", 0, LOG_DAEMON); - // if fail nothing much to do - if (pid == -1) goto errorFork; + // ------------- Build session handler & init config ------- + session = calloc (1, sizeof (AFB_session)); + session->config = calloc (1, sizeof (AFB_config)); - // fork worked and we are in father process - status = writePidFile (session->config, pid); - if (status == -1) goto errorPidFile; + on_exit(closeSession, session); + parse_arguments(argc, argv, session); - // we are in father process, we don't need this one - _exit (0); + // ------------------ sanity check ---------------------------------------- + if (session->config->httpdPort <= 0) { + fprintf (stderr, "ERR: no port is defined\n"); + exit (1); + } - } // end background-foreground + if (session->config->ldpaths) + afb_apis_add_pathset(session->config->ldpaths); -normalExit: - closeSession (session); // try to close everything before leaving - if (verbose) printf ("\n---- Application Framework Binder Normal End ------\n"); - exit(0); + ctxStoreInit(CTX_NBCLIENTS, session->config->cntxTimeout, afb_apis_count(), session->config->token); -// ------------- 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); + install_error_handlers(); -errorPidFile: - fprintf (stderr,"\nERR:AFB-daemon Failed to write pid file [%s]\n\n", session->config->pidfile); - exit (-1); + // ------------------ 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; + } -errorFork: - fprintf (stderr,"\nERR:AFB-daemon Failed to fork son process\n\n"); - exit (-1); + // let's run this program with a low priority + nice (20); -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); + // ------------------ Finaly Process Commands ----------------------------- + // let's not take the risk to run as ROOT + //if (getuid() == 0) goto errorNoRoot; -noValueForOption: - fprintf (stderr,"\nERR:AFB-daemon option [--%s] don't take value\n\n" - ,gnuOptions[optionIndex].name); - exit (-1); + if (verbosity) fprintf (stderr, "AFB: notice Init config done\n"); -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 ----------- + 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"); -badMode: - fprintf (stderr,"\nERR:AFB-daemon option [--%s] only accepts local, global or remote.\n\n" - ,gnuOptions[optionIndex].name); - exit (-1); + } + + rc = afb_hsrv_start (session); + if (!rc) + exit(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); + if (session->readyfd != 0) { + static const char readystr[] = "READY=1"; + write(session->readyfd, readystr, sizeof(readystr) - 1); + close(session->readyfd); + } -errConsole: - fprintf (stderr,"\nERR:AFB-daemon cannot open /dev/console (use --foreground)\n\n"); - exit (-1); + // infinite loop + for(;;) + upoll_wait(30000); -errSessiondir: - fprintf (stderr,"\nERR:AFB-daemon cannot read/write session dir\n\n"); - exit (-1); + if (verbosity) + fprintf (stderr, "hoops returned from infinite loop [report bug]\n"); -exitInitLoop: - // try to unlink pid file if any - if (session->background && session->config->pidfile != NULL) unlink (session->config->pidfile); - exit (-1); + return 0; +} -} /* END AFB-daemon() */