* limitations under the License.
*/
-#include <syslog.h>
+#define _GNU_SOURCE
+#include <stdio.h>
+#include <string.h>
+#include <getopt.h>
#include <setjmp.h>
#include <signal.h>
-#include <getopt.h>
-#include <pwd.h>
+#include <syslog.h>
+#include <fcntl.h>
#include <sys/stat.h>
#include <sys/types.h>
+#include "afb-plugin.h"
+
#include "local-def.h"
-#include "afb-plugins.h"
+#include "afb-hswitch.h"
+#include "afb-api-so.h"
+#include "afb-hsrv.h"
+#include "afb-hreq.h"
+#include "session.h"
+#include "verbose.h"
+#include "utils-upoll.h"
#if !defined(PLUGIN_INSTALL_DIR)
#error "you should define PLUGIN_INSTALL_DIR"
#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_MODE 18
#define SET_READYFD 19
+static struct afb_hsrv *start(AFB_config * config);
+
// Command line structure hold cli --command + help text
typedef struct {
int val; // command number within application
}
// load config from disk and merge with CLI option
-static AFB_error config_set_default (AFB_session * session)
+static void config_set_default (AFB_config * config)
{
- static char cacheTimeout [10];
-
// default HTTP port
- if (session->config->httpdPort == 0) session->config->httpdPort=1234;
+ if (config->httpdPort == 0)
+ config->httpdPort = 1234;
// default Plugin API timeout
- if (session->config->apiTimeout == 0) session->config->apiTimeout=DEFLT_API_TIMEOUT;
+ if (config->apiTimeout == 0)
+ config->apiTimeout = DEFLT_API_TIMEOUT;
// default AUTH_TOKEN
- if (session->config->token == NULL) session->config->token= DEFLT_AUTH_TOKEN;
+ if (config->token == NULL)
+ config->token = DEFLT_AUTH_TOKEN;
// cache timeout default one hour
- if (session->config->cacheTimeout == 0) session->config->cacheTimeout=DEFLT_CACHE_TIMEOUT;
+ if (config->cacheTimeout == 0)
+ 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 (config->cntxTimeout == 0)
+ config->cntxTimeout = DEFLT_CNTX_TIMEOUT;
+
+ if (config->rootdir == NULL) {
+ config->rootdir = getenv("AFBDIR");
+ if (config->rootdir == NULL) {
+ config->rootdir = malloc (512);
+ strncpy (config->rootdir, getenv("HOME"),512);
+ strncat (config->rootdir, "/.AFB",512);
}
// if directory does not exist createit
- mkdir (session->config->rootdir, O_RDWR | S_IRWXU | S_IRGRP);
+ mkdir (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 (config->rootbase == NULL)
+ config->rootbase = "/opa";
- if (session->config->rootapi == NULL) {
- session->config->rootapi = "/api";
- }
+ if (config->rootapi == NULL)
+ config->rootapi = "/api";
- if (session->config->ldpaths == NULL) {
- session->config->ldpaths = PLUGIN_INSTALL_DIR;
- }
+ if (config->ldpaths == NULL)
+ 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 (config->sessiondir == NULL) {
+ config->sessiondir = malloc (512);
+ strncpy (config->sessiondir, config->rootdir, 512);
+ strncat (config->sessiondir, "/sessions",512);
}
// if no config dir create a default path from sessiondir
- if (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);
+ if (config->console == NULL) {
+ config->console = malloc (512);
+ strncpy (config->console, config->sessiondir, 512);
+ strncat (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;
}
switch (optc)
{
case SET_VERBOSE:
- verbose = 1;
+ verbosity++;
break;
case SET_TCP_PORT:
case SET_ROOT_DIR:
if (optarg == 0) goto needValueForOption;
session->config->rootdir = optarg;
- if (verbose) fprintf(stderr, "Forcing Rootdir=%s\n",session->config->rootdir);
+ if (verbosity) fprintf(stderr, "Forcing Rootdir=%s\n",session->config->rootdir);
break;
case SET_ROOT_BASE:
if (optarg == 0) goto needValueForOption;
session->config->rootbase = optarg;
- if (verbose) fprintf(stderr, "Forcing Rootbase=%s\n",session->config->rootbase);
+ if (verbosity) fprintf(stderr, "Forcing Rootbase=%s\n",session->config->rootbase);
break;
case SET_ROOT_API:
if (optarg == 0) goto needValueForOption;
session->config->rootapi = optarg;
- if (verbose) fprintf(stderr, "Forcing Rootapi=%s\n",session->config->rootapi);
+ if (verbosity) fprintf(stderr, "Forcing Rootapi=%s\n",session->config->rootapi);
break;
case SET_ALIAS:
} 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 {
if (!sscanf (optarg, "%d", &session->config->cacheTimeout)) goto notAnInteger;
break;
- case SET_FAKE_MOD:
- if (optarg != 0) goto noValueForOption;
- session->fakemod = 1;
- break;
-
case SET_FORGROUND:
if (optarg != 0) goto noValueForOption;
- session->foreground = 1;
+ session->background = 0;
break;
case SET_BACKGROUND:
}
free(gnuOptions);
- config_set_default (session);
+ config_set_default (session->config);
return;
/*----------------------------------------------------------
| timeout signalQuit
- |
+--------------------------------------------------------- */
-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);
-
- fprintf (stderr, "ERR: Received signal quit\n");
- syslog (LOG_ERR, "Daemon got kill3 & quit [please report bug]");
- exit(1);
+void signalQuit (int signum)
+{
+ fprintf(stderr, "Terminating signal received %s\n", strsignal(signum));
+ exit(1);
}
/*----------------------------------------------------------
- | listenLoop
- | Main listening HTTP loop
+ | Error signals
+ |
+--------------------------------------------------------- */
-static void listenLoop (AFB_session *session) {
- AFB_error err;
-
- // ------ Start httpd server
-
- 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);
+__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);
+}
- // infinite loop
- httpdLoop(session);
+static void install_error_handlers()
+{
+ int i, signals[] = { SIGALRM, SIGSEGV, SIGFPE, 0 };
- fprintf (stderr, "hoops returned from infinite loop [report bug]\n");
+ for (i = 0; signals[i] != 0; i++) {
+ if (signal(signals[i], signalError) == SIG_ERR) {
+ fprintf(stderr, "Signal handler error\n");
+ exit(1);
+ }
+ }
}
-
+
/*----------------------------------------------------------
| daemonize
| set the process in background
int main(int argc, char *argv[]) {
AFB_session *session;
+ struct afb_hsrv *hsrv;
// open syslog if ever needed
openlog("afb-daemon", 0, LOG_DAEMON);
parse_arguments(argc, argv, session);
// ------------------ 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);
}
- initPlugins(session);
- ctxStoreInit(CTX_NBCLIENTS);
+ if (session->config->ldpaths)
+ afb_api_so_add_pathset(session->config->ldpaths);
- // ------------------ Some useful default values -------------------------
- if ((session->background == 0) && (session->foreground == 0)) session->foreground=1;
+ ctxStoreInit(CTX_NBCLIENTS, session->config->cntxTimeout, session->config->token);
+ if (!afb_hreq_init_cookie(session->config->httpdPort, session->config->rootapi, DEFLT_CNTX_TIMEOUT)) {
+ fprintf (stderr, "ERR: initialisation of cookies failed\n");
+ exit (1);
+ }
+
+ install_error_handlers();
// ------------------ clean exit on CTR-C signal ------------------------
if (signal (SIGINT, signalQuit) == SIG_ERR || signal (SIGABRT, signalQuit) == SIG_ERR) {
return 1;
}
-
// let's run this program with a low priority
nice (20);
// 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);
+ if (verbosity) fprintf (stderr, "AFB: notice Init config done\n");
+
+ // --------- 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");
+
}
-#endif
- if (verbose) fprintf (stderr, "AFB: notice Init config done\n");
- // ---- run in foreground mode --------------------
- if (session->foreground) {
+ hsrv = start (session->config);
+ if (hsrv == NULL)
+ exit(1);
- if (verbose) fprintf (stderr,"AFB: notice Foreground mode\n");
+ if (session->readyfd != 0) {
+ static const char readystr[] = "READY=1";
+ write(session->readyfd, readystr, sizeof(readystr) - 1);
+ close(session->readyfd);
+ }
- } // end foreground
+ // infinite loop
+ for(;;)
+ upoll_wait(30000);
- // --------- run in background mode -----------
- if (session->background) {
+ if (verbosity)
+ fprintf (stderr, "hoops returned from infinite loop [report bug]\n");
- if (verbose) printf ("AFB: Entering background mode\n");
+ return 0;
+}
- daemonize(session);
+static int init(struct afb_hsrv *hsrv, AFB_config * config)
+{
+ int idx;
+
+ if (!afb_hsrv_add_handler(hsrv, config->rootapi, afb_hswitch_websocket_switch, NULL, 20))
+ return 0;
+
+ if (!afb_hsrv_add_handler(hsrv, config->rootapi, afb_hswitch_apis, NULL, 10))
+ return 0;
+
+ for (idx = 0; config->aliasdir[idx].url != NULL; idx++)
+ if (!afb_hsrv_add_alias (hsrv, config->aliasdir[idx].url, config->aliasdir[idx].path, 0))
+ return 0;
+
+ if (!afb_hsrv_add_alias(hsrv, "", config->rootdir, -10))
+ return 0;
- // if everything look OK then look forever
- syslog (LOG_ERR, "AFB: Entering infinite loop in background mode");
+ if (!afb_hsrv_add_handler(hsrv, config->rootbase, afb_hswitch_one_page_api_redirect, NULL, -20))
+ return 0;
+ return 1;
+}
+
+static struct afb_hsrv *start(AFB_config * config)
+{
+ int rc;
+ struct afb_hsrv *hsrv;
+
+ hsrv = afb_hsrv_create();
+ if (hsrv == NULL) {
+ fprintf(stderr, "memory allocation failure\n");
+ return NULL;
+ }
- } // end background-foreground
+ if (!afb_hsrv_set_cache_timeout(hsrv, config->cacheTimeout)
+ || !init(hsrv, config)) {
+ printf("Error: initialisation of httpd failed");
+ afb_hsrv_put(hsrv);
+ return NULL;
+ }
+ if (verbosity) {
+ printf("AFB:notice Waiting port=%d rootdir=%s\n", config->httpdPort, config->rootdir);
+ printf("AFB:notice Browser URL= http:/*localhost:%d\n", config->httpdPort);
+ }
- listenLoop(session);
- if (verbose) printf ("\n---- Application Framework Binder Normal End ------\n");
- exit(0);
+ rc = afb_hsrv_start(hsrv, (uint16_t) config->httpdPort, 15);
+ if (!rc) {
+ printf("Error: starting of httpd failed");
+ afb_hsrv_put(hsrv);
+ return NULL;
+ }
+ return hsrv;
}