123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616 |
- #define _GNU_SOURCE
- #include <stdlib.h>
- #include <stdint.h>
- #include <stdio.h>
- #include <string.h>
- #include <assert.h>
- #include <signal.h>
- #include <syslog.h>
- #include <unistd.h>
- #include <errno.h>
- #include <sys/types.h>
- #include <sys/stat.h>
- #include <fcntl.h>
- #include <sys/socket.h>
- #include <sys/un.h>
- #include <sys/fsuid.h>
- #include <sys/wait.h>
- #include <poll.h>
- #include <time.h>
- #include <faux/faux.h>
- #include <faux/str.h>
- #include <faux/argv.h>
- #include <faux/ini.h>
- #include <faux/log.h>
- #include <faux/sched.h>
- #include <faux/sysdb.h>
- #include <faux/net.h>
- #include <faux/list.h>
- #include <faux/conv.h>
- #include <faux/file.h>
- #include <faux/eloop.h>
- #include <faux/error.h>
- #include <klish/ktp.h>
- #include <klish/ktp_session.h>
- #include <klish/kscheme.h>
- #include <klish/ischeme.h>
- #include <klish/kcontext.h>
- #include <klish/ksession.h>
- #include <klish/ksession_parse.h>
- #include <klish/kdb.h>
- #include <klish/kpargv.h>
- #include "private.h"
- bool_t daemonize(const char *pidfile);
- bool_t kentry_entrys_is_empty(const kentry_t *entry);
- static int create_listen_unix_sock(const char *path);
- static kscheme_t *load_all_dbs(const char *dbs,
- faux_ini_t *global_config, faux_error_t *error);
- static bool_t clear_scheme(kscheme_t *scheme, faux_error_t *error);
- static void signal_handler_empty(int signo);
- static bool_t stop_loop_ev(faux_eloop_t *eloop, faux_eloop_type_e type,
- void *associated_data, void *user_data);
- static bool_t refresh_config_ev(faux_eloop_t *eloop, faux_eloop_type_e type,
- void *associated_data, void *user_data);
- static bool_t listen_socket_ev(faux_eloop_t *eloop, faux_eloop_type_e type,
- void *associated_data, void *user_data);
- static bool_t wait_for_child_ev(faux_eloop_t *eloop, faux_eloop_type_e type,
- void *associated_data, void *user_data);
- int main(int argc, char **argv)
- {
- int retval = -1;
- struct options *opts = NULL;
- int pidfd = -1;
- int logoptions = 0;
- faux_eloop_t *eloop = NULL;
- int listen_unix_sock = -1;
- ktpd_session_t *ktpd_session = NULL;
- kscheme_t *scheme = NULL;
- faux_error_t *error = faux_error_new();
- faux_ini_t *config = NULL;
- int client_fd = -1;
- struct sigaction sig_act = {};
- sigset_t sig_set = {};
-
- opts = opts_init();
- if (opts_parse(argc, argv, opts))
- goto err;
-
- logoptions = LOG_CONS;
- if (opts->foreground)
- logoptions |= LOG_PERROR;
- openlog(LOG_NAME, logoptions, opts->log_facility);
- if (!opts->verbose)
- setlogmask(LOG_UPTO(LOG_INFO));
-
- syslog(LOG_DEBUG, "Parse config file: %s\n", opts->cfgfile);
- if (!access(opts->cfgfile, R_OK)) {
- if (!(config = config_parse(opts->cfgfile, opts)))
- goto err;
- } else if (opts->cfgfile_userdefined) {
-
- fprintf(stderr, "Error: Can't find config file %s\n",
- opts->cfgfile);
- goto err;
- }
-
- opts_show(opts);
- syslog(LOG_INFO, "Start daemon.\n");
-
- if (!opts->foreground && !daemonize(opts->pidfile))
- goto err;
-
- if (!(scheme = load_all_dbs(opts->dbs, config, error))) {
- fprintf(stderr, "Scheme errors:\n");
- goto err;
- }
-
- syslog(LOG_DEBUG, "Create listen UNIX socket: %s\n", opts->unix_socket_path);
- listen_unix_sock = create_listen_unix_sock(opts->unix_socket_path);
- if (listen_unix_sock < 0)
- goto err;
- syslog(LOG_DEBUG, "Listen socket %d", listen_unix_sock);
-
- eloop = faux_eloop_new(NULL);
-
- faux_eloop_add_signal(eloop, SIGINT, stop_loop_ev, NULL);
- faux_eloop_add_signal(eloop, SIGTERM, stop_loop_ev, NULL);
- faux_eloop_add_signal(eloop, SIGQUIT, stop_loop_ev, NULL);
- faux_eloop_add_signal(eloop, SIGHUP, refresh_config_ev, opts);
- faux_eloop_add_signal(eloop, SIGCHLD, wait_for_child_ev, NULL);
-
- faux_eloop_add_fd(eloop, listen_unix_sock, POLLIN,
- listen_socket_ev, &client_fd);
-
-
- faux_eloop_loop(eloop);
- faux_eloop_free(eloop);
- retval = 0;
- err:
-
- if (faux_error_len(error) > 0)
- faux_error_show(error);
- faux_error_free(error);
-
- if (listen_unix_sock >= 0)
- close(listen_unix_sock);
-
- if (client_fd < 0) {
-
- clear_scheme(scheme, error);
-
- opts_free(opts);
- faux_ini_free(config);
-
- if (pidfd >= 0) {
- if (unlink(opts->pidfile) < 0) {
- syslog(LOG_ERR, "Can't remove pid-file %s: %s\n",
- opts->pidfile, strerror(errno));
- }
- }
- syslog(LOG_INFO, "Stop daemon.\n");
- return retval;
- }
-
- retval = -1;
- eloop = NULL;
-
- openlog(LOG_SERVICE_NAME, logoptions, opts->log_facility);
- if (!opts->verbose)
- setlogmask(LOG_UPTO(LOG_INFO));
-
- eloop = faux_eloop_new(NULL);
-
-
- ktpd_session = ktpd_session_new(client_fd, scheme, NULL, eloop);
- if (!ktpd_session) {
- syslog(LOG_ERR, "Can't create KTPd session\n");
- goto err_client;
- }
- syslog(LOG_DEBUG, "New connection %d\n", client_fd);
-
- faux_eloop_add_signal(eloop, SIGINT, stop_loop_ev, NULL);
- faux_eloop_add_signal(eloop, SIGTERM, stop_loop_ev, NULL);
- faux_eloop_add_signal(eloop, SIGQUIT, stop_loop_ev, NULL);
-
-
- sigemptyset(&sig_set);
- sig_act.sa_flags = 0;
- sig_act.sa_mask = sig_set;
- sig_act.sa_handler = &signal_handler_empty;
- sigaction(SIGPIPE, &sig_act, NULL);
-
- faux_eloop_loop(eloop);
- retval = 0;
- err_client:
- ktpd_session_free(ktpd_session);
- faux_eloop_free(eloop);
- syslog(LOG_DEBUG, "Close connection %d", client_fd);
- close(client_fd);
-
- clear_scheme(scheme, error);
-
- opts_free(opts);
- faux_ini_free(config);
- return retval;
- }
- bool_t daemonize(const char *pidfile)
- {
-
- syslog(LOG_DEBUG, "Daemonize\n");
- if (!faux_daemon(0, 0, pidfile, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH)) {
- syslog(LOG_ERR, "Can't daemonize\n");
- return BOOL_FALSE;
- }
- syslog(LOG_DEBUG, "PID file: %s\n", pidfile);
- return BOOL_TRUE;
- }
- static bool_t load_db(kscheme_t *scheme, const char *db_name,
- faux_ini_t *config, faux_error_t *error)
- {
- kdb_t *db = NULL;
- const char *sofile = NULL;
- assert(scheme);
- if (!scheme)
- return BOOL_FALSE;
- assert(db_name);
- if (!db_name)
- return BOOL_FALSE;
-
- if (config)
- sofile = faux_ini_find(config, "so");
- db = kdb_new(db_name, sofile);
- assert(db);
- if (!db) {
- faux_ini_free(config);
- return BOOL_FALSE;
- }
-
- kdb_set_ini(db, config);
- kdb_set_error(db, error);
-
- if (!kdb_load_plugin(db)) {
- faux_error_sprintf(error,
- "DB \"%s\": Can't load DB plugin", db_name);
- kdb_free(db);
- return BOOL_FALSE;
- }
-
- if ((kdb_major(db) != KDB_MAJOR) ||
- (kdb_minor(db) != KDB_MINOR)) {
- faux_error_sprintf(error,
- "DB \"%s\": Plugin's API version is %u.%u, need %u.%u",
- db_name,
- kdb_major(db), kdb_minor(db),
- KDB_MAJOR, KDB_MINOR);
- kdb_free(db);
- return BOOL_FALSE;
- }
-
- if (kdb_has_init_fn(db) && !kdb_init(db)) {
- faux_error_sprintf(error,
- "DB \"%s\": Can't init DB plugin", db_name);
- kdb_free(db);
- return BOOL_FALSE;
- }
-
- if (!kdb_has_load_fn(db) || !kdb_load_scheme(db, scheme)) {
- faux_error_sprintf(error,
- "DB \"%s\": Can't load scheme from DB plugin", db_name);
- kdb_fini(db);
- kdb_free(db);
- return BOOL_FALSE;
- }
-
- if (kdb_has_fini_fn(db) && !kdb_fini(db)) {
- faux_error_sprintf(error,
- "DB \"%s\": Can't fini DB plugin", db_name);
- kdb_free(db);
- return BOOL_FALSE;
- }
- kdb_free(db);
- return BOOL_TRUE;
- }
- static kscheme_t *load_all_dbs(const char *dbs,
- faux_ini_t *global_config, faux_error_t *error)
- {
- kscheme_t *scheme = NULL;
- faux_argv_t *dbs_argv = NULL;
- faux_argv_node_t *iter = NULL;
- const char *db_name = NULL;
- bool_t retcode = BOOL_TRUE;
- kcontext_t *context = NULL;
- assert(dbs);
- if (!dbs)
- return NULL;
- scheme = kscheme_new();
- assert(scheme);
- if (!scheme)
- return NULL;
- dbs_argv = faux_argv_new();
- assert(dbs_argv);
- if (!dbs_argv) {
- kscheme_free(scheme);
- return NULL;
- }
- if (faux_argv_parse(dbs_argv, dbs) <= 0) {
- kscheme_free(scheme);
- faux_argv_free(dbs_argv);
- return NULL;
- }
-
- iter = faux_argv_iter(dbs_argv);
- while ((db_name = faux_argv_each(&iter))) {
- faux_ini_t *config = NULL;
- if (global_config) {
- char *prefix = NULL;
- prefix = faux_str_mcat(&prefix, "DB.", db_name, ".", NULL);
- config = faux_ini_extract_subini(global_config, prefix);
- faux_str_free(prefix);
- }
- if (!load_db(scheme, db_name, config, error))
- retcode = BOOL_FALSE;
- }
- faux_argv_free(dbs_argv);
-
- if (!retcode) {
- kscheme_free(scheme);
- return NULL;
- }
-
- context = kcontext_new(KCONTEXT_TYPE_PLUGIN_INIT);
- kcontext_set_scheme(context, scheme);
- retcode = kscheme_prepare(scheme, context, error);
- kcontext_free(context);
- if (!retcode) {
- kscheme_free(scheme);
- faux_error_sprintf(error, "Scheme preparing errors.\n");
- return NULL;
- }
-
- return scheme;
- }
- static bool_t clear_scheme(kscheme_t *scheme, faux_error_t *error)
- {
- kcontext_t *context = NULL;
- if (!scheme)
- return BOOL_TRUE;
- context = kcontext_new(KCONTEXT_TYPE_PLUGIN_FINI);
- kcontext_set_scheme(context, scheme);
- kscheme_fini(scheme, context, error);
- kcontext_free(context);
- kscheme_free(scheme);
- return BOOL_TRUE;
- }
- static int create_listen_unix_sock(const char *path)
- {
- int sock = -1;
- int opt = 1;
- struct sockaddr_un laddr = {};
- assert(path);
- if (!path)
- return -1;
- if ((sock = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) {
- syslog(LOG_ERR, "Can't create socket: %s\n", strerror(errno));
- goto err;
- }
- if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt))) {
- syslog(LOG_ERR, "Can't set socket options: %s\n", strerror(errno));
- goto err;
- }
-
- unlink(path);
- laddr.sun_family = AF_UNIX;
- strncpy(laddr.sun_path, path, USOCK_PATH_MAX);
- laddr.sun_path[USOCK_PATH_MAX - 1] = '\0';
- if (bind(sock, (struct sockaddr *)&laddr, sizeof(laddr))) {
- syslog(LOG_ERR, "Can't bind socket %s: %s\n", path, strerror(errno));
- goto err;
- }
- if (listen(sock, 128)) {
- unlink(path);
- syslog(LOG_ERR, "Can't listen on socket %s: %s\n", path, strerror(errno));
- goto err;
- }
- return sock;
- err:
- if (sock >= 0)
- close(sock);
- return -1;
- }
- static bool_t stop_loop_ev(faux_eloop_t *eloop, faux_eloop_type_e type,
- void *associated_data, void *user_data)
- {
-
- eloop = eloop;
- type = type;
- associated_data = associated_data;
- user_data = user_data;
- return BOOL_FALSE;
- }
- static bool_t wait_for_child_ev(faux_eloop_t *eloop, faux_eloop_type_e type,
- void *associated_data, void *user_data)
- {
- int wstatus = 0;
- pid_t child_pid = -1;
-
- while ((child_pid = waitpid(-1, &wstatus, WNOHANG)) > 0) {
- if (WIFSIGNALED(wstatus)) {
- syslog(LOG_ERR, "Service process %d was terminated "
- "by signal: %d",
- child_pid, WTERMSIG(wstatus));
- } else {
- syslog(LOG_ERR, "Service process %d was terminated: %d",
- child_pid, WEXITSTATUS(wstatus));
- }
- }
-
- eloop = eloop;
- type = type;
- associated_data = associated_data;
- user_data = user_data;
- return BOOL_TRUE;
- }
- static bool_t refresh_config_ev(faux_eloop_t *eloop, faux_eloop_type_e type,
- void *associated_data, void *user_data)
- {
- struct options *opts = (struct options *)user_data;
- faux_ini_t *ini = NULL;
- if (access(opts->cfgfile, R_OK) == 0) {
- syslog(LOG_DEBUG, "Re-reading config file \"%s\"\n", opts->cfgfile);
- if (!(ini = config_parse(opts->cfgfile, opts)))
- syslog(LOG_ERR, "Error while config file parsing.\n");
- } else if (opts->cfgfile_userdefined) {
- syslog(LOG_ERR, "Can't find config file \"%s\"\n", opts->cfgfile);
- }
- faux_ini_free(ini);
-
- eloop = eloop;
- type = type;
- associated_data = associated_data;
- return BOOL_TRUE;
- }
- static bool_t listen_socket_ev(faux_eloop_t *eloop, faux_eloop_type_e type,
- void *associated_data, void *user_data)
- {
- int new_conn = -1;
- faux_eloop_info_fd_t *info = (faux_eloop_info_fd_t *)associated_data;
- pid_t child_pid = -1;
- assert(user_data);
- new_conn = accept(info->fd, NULL, NULL);
- if (new_conn < 0) {
- syslog(LOG_ERR, "Can't accept() new connection");
- return BOOL_TRUE;
- }
-
- child_pid = fork();
- if (child_pid < 0) {
- close(new_conn);
- syslog(LOG_ERR, "Can't fork service process for client");
- return BOOL_TRUE;
- }
-
- if (child_pid > 0) {
- close(new_conn);
- syslog(LOG_ERR, "Service process for client was forked: %d",
- child_pid);
- return BOOL_TRUE;
- }
-
-
- *((int *)user_data) = new_conn;
- type = type;
- eloop = eloop;
-
-
- return BOOL_FALSE;
- }
- static void signal_handler_empty(int signo)
- {
- signo = signo;
- }
|