/*
* Copyright (C) 2015 "IoT.bzh"
* Author "Fulup Ar Foll"
+ * Author José Bollo <jose.bollo@iot.bzh>
*
- * 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 <http://www.gnu.org/licenses/>.
+ * 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"
-
#include <syslog.h>
#include <setjmp.h>
#include <signal.h>
#include <getopt.h>
+#include <pwd.h>
+#include <pthread.h>
+#include <sys/stat.h>
+#include <sys/types.h>
-static sigjmp_buf exitPoint; // context save for set/longjmp
+#include "local-def.h"
+#include "afb-apis.h"
+#include "session.h"
-/*----------------------------------------------------------
- | printversion
- | print version and copyright
- +--------------------------------------------------------- */
- 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
+#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 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_cardid 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_SMACK 140
- #define SET_PLUGINS 141
- #define SET_APITIMEOUT 142
-
- #define DISPLAY_VERSION 150
- #define DISPLAY_HELP 151
+#define SET_VERBOSE 1
+#define SET_BACKGROUND 2
+#define SET_FORGROUND 3
+#define SET_FAKE_MOD 4
+
+#define SET_TCP_PORT 5
+#define SET_ROOT_DIR 6
+#define SET_ROOT_BASE 7
+#define SET_ROOT_API 8
+#define SET_ALIAS 9
+
+#define SET_CACHE_TIMEOUT 10
+#define SET_SESSION_DIR 11
+
+#define SET_AUTH_TOKEN 12
+#define SET_LDPATH 13
+#define SET_APITIMEOUT 14
+#define SET_CNTXTIMEOUT 15
+
+#define DISPLAY_VERSION 16
+#define DISPLAY_HELP 17
+
+#define SET_MODE 18
+#define SET_READYFD 19
+
+// Command line structure hold cli --command + help text
+typedef struct {
+ int val; // command number within application
+ int has_arg; // command number within application
+ char *name; // command as used in --xxxx cli
+ char *help; // help text
+} AFB_options;
// Supported option
{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_CACHE_TO ,1,"cache-eol" , "Client cache end of live [default 3600s]"},
- {SET_cardid ,1,"setuid" , "Change user id [default don't change]"},
- {SET_PID_FILE ,1,"pidfile" , "PID file path [default none]"},
+ {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_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_SMACK ,1,"smack" , "Set Smack Label [default demo]"},
- {SET_PLUGINS ,1,"mods" , "Enable module [default all]"},
+ {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"},
- {0, 0, 0}
+
+ {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;
+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 AFB_error config_set_default (AFB_session * session)
+{
+ static char cacheTimeout [10];
+
+ // default HTTP port
+ if (session->config->httpdPort == 0) session->config->httpdPort=1234;
+
+ // default Plugin API timeout
+ if (session->config->apiTimeout == 0) session->config->apiTimeout=DEFLT_API_TIMEOUT;
+
+ // default AUTH_TOKEN
+ if (session->config->token == NULL) session->config->token= DEFLT_AUTH_TOKEN;
+
+ // cache timeout default one hour
+ if (session->config->cacheTimeout == 0) session->config->cacheTimeout=DEFLT_CACHE_TIMEOUT;
+
+ // cache timeout default one hour
+ if (session->config->cntxTimeout == 0) session->config->cntxTimeout=DEFLT_CNTX_TIMEOUT;
+
+ if (session->config->rootdir == NULL) {
+ session->config->rootdir = getenv("AFBDIR");
+ if (session->config->rootdir == NULL) {
+ session->config->rootdir = malloc (512);
+ strncpy (session->config->rootdir, getenv("HOME"),512);
+ strncat (session->config->rootdir, "/.AFB",512);
+ }
+ // if directory does not exist createit
+ mkdir (session->config->rootdir, O_RDWR | S_IRWXU | S_IRGRP);
+ }
+
+ // if no Angular/HTML5 rootbase let's try '/' as default
+ if (session->config->rootbase == NULL) {
+ session->config->rootbase = "/opa";
+ }
+
+ if (session->config->rootapi == NULL) {
+ session->config->rootapi = "/api";
+ }
+
+ if (session->config->ldpaths == NULL) {
+ session->config->ldpaths = PLUGIN_INSTALL_DIR;
+ }
+
+ // if no session dir create a default path from rootdir
+ if (session->config->sessiondir == NULL) {
+ session->config->sessiondir = malloc (512);
+ strncpy (session->config->sessiondir, session->config->rootdir, 512);
+ strncat (session->config->sessiondir, "/sessions",512);
+ }
+
+ // if no config dir create a default path from sessiondir
+ if (session->config->console == NULL) {
+ session->config->console = malloc (512);
+ strncpy (session->config->console, session->config->sessiondir, 512);
+ strncat (session->config->console, "/AFB-console.out",512);
+ }
+
+ // cacheTimeout is an integer but HTTPd wants it as a string
+ snprintf (cacheTimeout, sizeof (cacheTimeout),"%d", session->config->cacheTimeout);
+ session->cacheTimeout = cacheTimeout; // httpd uses cacheTimeout string version
+
+ return AFB_SUCCESS;
}
fprintf (stderr," --%-15s %s\n", command, cliOptions[ind].help);
}
}
- fprintf (stderr,"Example:\n %s\\\n --verbose --port=1234 --smack=xxxx --mods=alsa:dbus\n", name);
+ 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;
-
- // 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 ggcOption;
- 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
nbcmd = sizeof (cliOptions) / sizeof (AFB_options);
- gnuOptions = malloc (sizeof (ggcOption) * nbcmd);
+ 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;
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", &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 (verbose) 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 (verbose) 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 (verbose) 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,":");
- 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++;
+ if (optarg == NULL) {
+ fprintf(stderr, "missing ':' in alias %s, ignored\n", aliasdir[aliascount].url);
+ } else {
+ aliasdir[aliascount].path = optarg;
+ aliasdir[aliascount].len = strlen(aliasdir[aliascount].url);
+ if (verbose) fprintf(stderr, "Alias url=%s path=%s\n", aliasdir[aliascount].url, aliasdir[aliascount].path);
+ aliascount++;
+ }
} else {
- fprintf(stderr, "Too many aliases [max:%s] %s ignored\n", optarg, MAX_ALIAS-1);
+ fprintf(stderr, "Too many aliases [max:%d] %s ignored\n", MAX_ALIAS, optarg);
}
break;
- case SET_SMACK:
+ case SET_AUTH_TOKEN:
if (optarg == 0) goto needValueForOption;
- fprintf (stderr, "Not Implemented yet\n");
- cliconfig.smack = optarg;
+ session->config->token = optarg;
break;
- case SET_PLUGINS:
+ case SET_LDPATH:
if (optarg == 0) goto needValueForOption;
- fprintf (stderr, "Not Implemented yet\n");
- cliconfig.plugins = 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;
+ session->config->sessiondir = optarg;
break;
- case SET_CACHE_TO:
+ case SET_CACHE_TIMEOUT:
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_cardid:
- if (optarg == 0) goto needValueForOption;
- if (!sscanf (optarg, "%d", &cliconfig.setuid)) goto notAnInteger;
+ if (!sscanf (optarg, "%d", &session->config->cacheTimeout)) goto notAnInteger;
break;
case SET_FAKE_MOD:
session->background = 1;
break;
- case KILL_PREV_REST:
- if (optarg != 0) goto noValueForOption;
- session->killPrevious = 1;
+ 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 KILL_PREV_EXIT:
- if (optarg != 0) goto noValueForOption;
- session->killPrevious = 2;
+ 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;
+ exit(0);
case DISPLAY_HELP:
default:
printHelp(programName);
- goto normalExit;
-
- }
- }
- // Create session config
- configInit (/* session & config are initialized globally */);
-
- // if exist merge config file with CLI arguments
- configLoadFile (session, &cliconfig);
-
- // ------------------ sanity check ----------------------------------------
- if ((session->background) && (session->foreground)) {
- fprintf (stderr, "%s ERR: cannot select foreground & background at the same time\n",configTime());
- exit (-1);
+ exit(0);
+ }
}
+ free(gnuOptions);
+
+ 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("AGB-log", 0, LOG_DAEMON);
-
- // -------------- Try to kill any previsou 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
+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);
- // ------------------ clean exit on CTR-C signal ------------------------
- if (signal (SIGINT, signalQuit) == SIG_ERR) {
- fprintf (stderr, "%s Quit Signal received.",configTime());
- return (-1);
- }
+noValueForOption:
+ fprintf (stderr,"\nERR: AFB-daemon option [--%s] don't take value\n\n"
+ ,gnuOptions[optionIndex].name);
+ exit (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;
- }
+badMode:
+ fprintf (stderr,"\nERR: AFB-daemon option [--%s] only accepts local, global or remote.\n\n"
+ ,gnuOptions[optionIndex].name);
+ exit (1);
+}
- // let's run this program with a low priority
- status=nice (20);
+/*----------------------------------------------------------
+ | closeSession
+ | try to close everything before leaving
+ +--------------------------------------------------------- */
+static void closeSession (int status, void *data) {
+ /* AFB_session *session = data; */
+}
- // ------------------ Finaly Process Commands -----------------------------
- // if --save then store config on disk upfront
- if (session->configsave) configStoreFile (session);
- if (session->forceexit) exit (0);
+/*----------------------------------------------------------
+ | timeout signalQuit
+ |
+ +--------------------------------------------------------- */
+void signalQuit (int signum) {
- if (session->config->setuid) {
- int err;
+ sigset_t sigset;
- err = setuid(session->config->setuid);
- if (err) fprintf (stderr, "Fail to change program cardid error=%s", strerror(err));
- }
+ // unlock timeout signal to allow a new signal to come
+ sigemptyset (&sigset);
+ sigaddset (&sigset, SIGABRT);
+ sigprocmask (SIG_UNBLOCK, &sigset, 0);
- // let's not take the risk to run as ROOT
- if (getuid() == 0) status=setuid(65534); // run as nobody
+ fprintf (stderr, "ERR: Received signal quit\n");
+ syslog (LOG_ERR, "Daemon got kill3 & quit [please report bug]");
+ 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");
+/*----------------------------------------------------------
+ | Error signals
+ |
+ +--------------------------------------------------------- */
+__thread sigjmp_buf *error_handler;
+static void signalError(int signum)
+{
+ sigset_t sigset;
+
+ // unlock signal to allow a new signal to come
+ sigemptyset(&sigset);
+ sigaddset(&sigset, signum);
+ sigprocmask(SIG_UNBLOCK, &sigset, 0);
+ if (error_handler != NULL) {
+ longjmp(*error_handler, signum);
+ }
+}
+static void install_error_handlers()
+{
+ int i, signals[] = { SIGALRM, SIGSEGV, SIGFPE, 0 };
- // ---- run in foreground mode --------------------
- if (session->foreground) {
+ for (i = 0; signals[i] != 0; i++) {
+ if (signal(signals[i], signalError) == SIG_ERR) {
+ fprintf(stderr, "Signal handler error\n");
+ exit(1);
+ }
+ }
+}
- if (verbose) fprintf (stderr,"AFB:notice Foreground mode\n");
+/*----------------------------------------------------------
+ | listenLoop
+ | Main listening HTTP loop
+ +--------------------------------------------------------- */
+static void listenLoop (AFB_session *session) {
+ AFB_error err;
- // write a pid file for --kill-previous and --raise-debug option
- status = writePidFile (session->config, getpid());
- if (status == -1) goto errorPidFile;
+ // ------ Start httpd server
- // enter listening loop in foreground
- listenLoop(session);
- goto exitInitLoop;
- } // end foreground
+ 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);
+ }
- // --------- run in background mode -----------
- if (session->background) {
+ // infinite loop
+ httpdLoop(session);
- // if (status != 0) goto errorCommand;
- if (verbose) printf ("AFB: Entering background mode\n");
+ fprintf (stderr, "hoops returned from infinite loop [report bug]\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[]) {
+ 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->background) && (session->foreground)) {
+ fprintf (stderr, "ERR: cannot select foreground & background at the same time\n");
+ exit (1);
+ }
+ if (session->config->httpdPort <= 0) {
+ fprintf (stderr, "ERR: no port is defined\n");
+ exit (1);
+ }
- } // 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);
+
+ install_error_handlers();
-// ------------- Fatal ERROR display error and quit -------------
-errorPidFile:
- fprintf (stderr,"\nERR:main Failled to write pid file [%s]\n\n", session->config->pidfile);
- exit (-1);
+ // ------------------ Some useful default values -------------------------
+ if ((session->background == 0) && (session->foreground == 0)) session->foreground=1;
-errorFork:
- fprintf (stderr,"\nERR:main Failled to fork son process\n\n");
- 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;
+ }
-needValueForOption:
- fprintf (stderr,"\nERR:main option [--%s] need a value i.e. --%s=xxx\n\n"
- ,gnuOptions[optionIndex].name, gnuOptions[optionIndex].name);
- exit (-1);
-noValueForOption:
- fprintf (stderr,"\nERR:main option [--%s] don't take value\n\n"
- ,gnuOptions[optionIndex].name);
- exit (-1);
+ // let's run this program with a low priority
+ nice (20);
-notAnInteger:
- fprintf (stderr,"\nERR:main option [--%s] requirer an interger i.e. --%s=9\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;
+
+#if defined(ALLOWS_SESSION_FILES)
+ // check session dir and create if it does not exist
+ if (sessionCheckdir (session) != AFB_SUCCESS) {
+ fprintf (stderr,"\nERR: AFB-daemon cannot read/write session dir\n\n");
+ exit (1);
+ }
+#endif
+ if (verbose) fprintf (stderr, "AFB: notice Init config done\n");
-exitOnSignal:
- fprintf (stderr,"\n%s INF:main pid=%d received exit signal (Hopefully crtl-C or --kill-previous !!!)\n\n"
- ,configTime(), getpid());
- exit (-1);
+ // ---- run in foreground mode --------------------
+ if (session->foreground) {
-errConsole:
- fprintf (stderr,"\nERR:cannot open /dev/console (use --foreground)\n\n");
- exit (-1);
+ if (verbose) fprintf (stderr,"AFB: notice Foreground mode\n");
-errSessiondir:
- fprintf (stderr,"\nERR:cannot read/write session dir\n\n");
- exit (-1);
+ } // end foreground
-errSoundCard:
- fprintf (stderr,"\nERR:fail to probe sound cards\n\n");
- exit (-1);
+ // --------- run in background mode -----------
+ if (session->background) {
+
+ if (verbose) printf ("AFB: Entering background mode\n");
+
+ daemonize(session);
+
+ // if everything look OK then look forever
+ syslog (LOG_ERR, "AFB: Entering infinite loop in background mode");
-exitInitLoop:
- // try to unlink pid file if any
- if (session->background && session->config->pidfile != NULL) unlink (session->config->pidfile);
- exit (-1);
-}; /* END main() */
+ } // end background-foreground
+
+
+ listenLoop(session);
+ if (verbose) printf ("\n---- Application Framework Binder Normal End ------\n");
+ exit(0);
+
+}
+
Code Review / src / app-framework-binder.git / blobdiff