Annotation of embedaddon/strongswan/src/charon-systemd/charon-systemd.c, revision 1.1

1.1     ! misho       1: /*
        !             2:  * Copyright (C) 2006-2018 Tobias Brunner
        !             3:  * Copyright (C) 2005-2014 Martin Willi
        !             4:  * Copyright (C) 2006 Daniel Roethlisberger
        !             5:  * Copyright (C) 2005 Jan Hutter
        !             6:  * HSR Hochschule fuer Technik Rapperswil
        !             7:  * Copyright (C) 2014 revosec AG
        !             8:  *
        !             9:  * This program is free software; you can redistribute it and/or modify it
        !            10:  * under the terms of the GNU General Public License as published by the
        !            11:  * Free Software Foundation; either version 2 of the License, or (at your
        !            12:  * option) any later version.  See <http://www.fsf.org/copyleft/gpl.txt>.
        !            13:  *
        !            14:  * This program is distributed in the hope that it will be useful, but
        !            15:  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
        !            16:  * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
        !            17:  * for more details.
        !            18:  */
        !            19: 
        !            20: #include <signal.h>
        !            21: #include <stdio.h>
        !            22: #include <pthread.h>
        !            23: #include <sys/stat.h>
        !            24: #include <sys/types.h>
        !            25: #include <sys/utsname.h>
        !            26: #include <unistd.h>
        !            27: #include <errno.h>
        !            28: 
        !            29: /* won't make sense from our logging hook */
        !            30: #define SD_JOURNAL_SUPPRESS_LOCATION
        !            31: #include <systemd/sd-daemon.h>
        !            32: #include <systemd/sd-journal.h>
        !            33: 
        !            34: #include <daemon.h>
        !            35: 
        !            36: #include <library.h>
        !            37: #include <utils/backtrace.h>
        !            38: #include <threading/thread.h>
        !            39: #include <threading/rwlock.h>
        !            40: 
        !            41: /**
        !            42:  * Default user and group
        !            43:  */
        !            44: #ifndef IPSEC_USER
        !            45: #define IPSEC_USER NULL
        !            46: #endif
        !            47: 
        !            48: #ifndef IPSEC_GROUP
        !            49: #define IPSEC_GROUP NULL
        !            50: #endif
        !            51: 
        !            52: /**
        !            53:  * hook in library for debugging messages
        !            54:  */
        !            55: extern void (*dbg) (debug_t group, level_t level, char *fmt, ...);
        !            56: 
        !            57: /**
        !            58:  * Logging hook for library logs, using stderr output
        !            59:  */
        !            60: static void dbg_stderr(debug_t group, level_t level, char *fmt, ...)
        !            61: {
        !            62:        va_list args;
        !            63: 
        !            64:        if (level <= 1)
        !            65:        {
        !            66:                va_start(args, fmt);
        !            67:                fprintf(stderr, "00[%N] ", debug_names, group);
        !            68:                vfprintf(stderr, fmt, args);
        !            69:                fprintf(stderr, "\n");
        !            70:                va_end(args);
        !            71:        }
        !            72: }
        !            73: 
        !            74: typedef struct journal_logger_t journal_logger_t;
        !            75: 
        !            76: /**
        !            77:  * Logger implementation using systemd-journal
        !            78:  */
        !            79: struct journal_logger_t {
        !            80: 
        !            81:        /**
        !            82:         * Public interface
        !            83:         */
        !            84:        custom_logger_t public;
        !            85: 
        !            86:        /**
        !            87:         * Configured loglevels
        !            88:         */
        !            89:        level_t levels[DBG_MAX];
        !            90: 
        !            91:        /**
        !            92:         * Lock for levels
        !            93:         */
        !            94:        rwlock_t *lock;
        !            95: };
        !            96: 
        !            97: METHOD(logger_t, vlog, void,
        !            98:        journal_logger_t *this, debug_t group, level_t level, int thread,
        !            99:        ike_sa_t *ike_sa, const char *fmt, va_list args)
        !           100: {
        !           101:        char buf[4096], *msg = buf;
        !           102:        ssize_t len;
        !           103:        va_list copy;
        !           104: 
        !           105:        va_copy(copy, args);
        !           106:        len = vsnprintf(msg, sizeof(buf), fmt, copy);
        !           107:        va_end(copy);
        !           108: 
        !           109:        if (len >= sizeof(buf))
        !           110:        {
        !           111:                len++;
        !           112:                msg = malloc(len);
        !           113:                va_copy(copy, args);
        !           114:                len = vsnprintf(msg, len, fmt, copy);
        !           115:                va_end(copy);
        !           116:        }
        !           117:        if (len > 0)
        !           118:        {
        !           119:                char unique[64] = "", name[256] = "";
        !           120:                int priority;
        !           121: 
        !           122:                if (ike_sa)
        !           123:                {
        !           124:                        snprintf(unique, sizeof(unique), "IKE_SA_UNIQUE_ID=%u",
        !           125:                                         ike_sa->get_unique_id(ike_sa));
        !           126:                        if (ike_sa->get_peer_cfg(ike_sa))
        !           127:                        {
        !           128:                                snprintf(name, sizeof(name), "IKE_SA_NAME=%s",
        !           129:                                                 ike_sa->get_name(ike_sa));
        !           130:                        }
        !           131:                }
        !           132:                switch (level)
        !           133:                {
        !           134:                        case LEVEL_AUDIT:
        !           135:                                priority = LOG_NOTICE;
        !           136:                                break;
        !           137:                        case LEVEL_CTRL:
        !           138:                                priority = LOG_INFO;
        !           139:                                break;
        !           140:                        default:
        !           141:                                priority = LOG_DEBUG;
        !           142:                                break;
        !           143:                }
        !           144:                sd_journal_send(
        !           145:                        "MESSAGE=%s", msg,
        !           146:                        "MESSAGE_ID=57d2708c-d607-43bd-8c39-66bf%.8x",
        !           147:                                chunk_hash_static(chunk_from_str((char*)fmt)),
        !           148:                        "PRIORITY=%d", priority,
        !           149:                        "GROUP=%N", debug_names, group,
        !           150:                        "LEVEL=%d", level,
        !           151:                        "THREAD=%d", thread,
        !           152:                        unique[0] ? unique : NULL,
        !           153:                        name[0] ? name : NULL,
        !           154:                        NULL);
        !           155:        }
        !           156:        if (msg != buf)
        !           157:        {
        !           158:                free(msg);
        !           159:        }
        !           160: }
        !           161: 
        !           162: METHOD(logger_t, get_level, level_t,
        !           163:        journal_logger_t *this, debug_t group)
        !           164: {
        !           165:        level_t level;
        !           166: 
        !           167:        this->lock->read_lock(this->lock);
        !           168:        level = this->levels[group];
        !           169:        this->lock->unlock(this->lock);
        !           170: 
        !           171:        return level;
        !           172: }
        !           173: 
        !           174: METHOD(custom_logger_t, set_level, void,
        !           175:        journal_logger_t *this, debug_t group, level_t level)
        !           176: {
        !           177:        this->lock->write_lock(this->lock);
        !           178:        this->levels[group] = level;
        !           179:        this->lock->unlock(this->lock);
        !           180: }
        !           181: 
        !           182: METHOD(custom_logger_t, logger_destroy, void,
        !           183:        journal_logger_t *this)
        !           184: {
        !           185:        this->lock->destroy(this->lock);
        !           186:        free(this);
        !           187: }
        !           188: 
        !           189: static custom_logger_t *journal_logger_create(const char *name)
        !           190: {
        !           191:        journal_logger_t *this;
        !           192: 
        !           193:        INIT(this,
        !           194:                .public = {
        !           195:                        .logger = {
        !           196:                                .vlog = _vlog,
        !           197:                                .get_level = _get_level,
        !           198:                        },
        !           199:                        .set_level = _set_level,
        !           200:                        .destroy = _logger_destroy,
        !           201:                },
        !           202:                .lock = rwlock_create(RWLOCK_TYPE_DEFAULT),
        !           203:        );
        !           204:        return &this->public;
        !           205: }
        !           206: 
        !           207: /**
        !           208:  * Run the daemon and handle unix signals
        !           209:  */
        !           210: static int run()
        !           211: {
        !           212:        sigset_t set;
        !           213: 
        !           214:        sigemptyset(&set);
        !           215:        sigaddset(&set, SIGHUP);
        !           216:        sigaddset(&set, SIGTERM);
        !           217:        sigprocmask(SIG_BLOCK, &set, NULL);
        !           218: 
        !           219:        sd_notify(0, "READY=1\n");
        !           220: 
        !           221:        while (TRUE)
        !           222:        {
        !           223:                int sig;
        !           224: 
        !           225:                sig = sigwaitinfo(&set, NULL);
        !           226:                if (sig == -1)
        !           227:                {
        !           228:                        if (errno == EINTR)
        !           229:                        {       /* ignore signals we didn't wait for */
        !           230:                                continue;
        !           231:                        }
        !           232:                        DBG1(DBG_DMN, "waiting for signal failed: %s", strerror(errno));
        !           233:                        return SS_RC_INITIALIZATION_FAILED;
        !           234:                }
        !           235:                switch (sig)
        !           236:                {
        !           237:                        case SIGHUP:
        !           238:                        {
        !           239:                                DBG1(DBG_DMN, "signal of type SIGHUP received. Reloading "
        !           240:                                         "configuration");
        !           241:                                if (lib->settings->load_files(lib->settings, lib->conf, FALSE))
        !           242:                                {
        !           243:                                        charon->load_loggers(charon);
        !           244:                                        lib->plugins->reload(lib->plugins, NULL);
        !           245:                                }
        !           246:                                else
        !           247:                                {
        !           248:                                        DBG1(DBG_DMN, "reloading config failed, keeping old");
        !           249:                                }
        !           250:                                break;
        !           251:                        }
        !           252:                        case SIGTERM:
        !           253:                        {
        !           254:                                DBG1(DBG_DMN, "SIGTERM received, shutting down");
        !           255:                                charon->bus->alert(charon->bus, ALERT_SHUTDOWN_SIGNAL, sig);
        !           256:                                return 0;
        !           257:                        }
        !           258:                }
        !           259:        }
        !           260: }
        !           261: 
        !           262: /**
        !           263:  * lookup UID and GID
        !           264:  */
        !           265: static bool lookup_uid_gid()
        !           266: {
        !           267:        char *name;
        !           268: 
        !           269:        name = lib->settings->get_str(lib->settings, "%s.user", IPSEC_USER,
        !           270:                                                                  lib->ns);
        !           271:        if (name && !lib->caps->resolve_uid(lib->caps, name))
        !           272:        {
        !           273:                return FALSE;
        !           274:        }
        !           275:        name = lib->settings->get_str(lib->settings, "%s.group", IPSEC_GROUP,
        !           276:                                                                  lib->ns);
        !           277:        if (name && !lib->caps->resolve_gid(lib->caps, name))
        !           278:        {
        !           279:                return FALSE;
        !           280:        }
        !           281:        return TRUE;
        !           282: }
        !           283: 
        !           284: #ifndef DISABLE_SIGNAL_HANDLER
        !           285: /**
        !           286:  * Handle SIGSEGV/SIGILL signals raised by threads
        !           287:  */
        !           288: static void segv_handler(int signal)
        !           289: {
        !           290:        backtrace_t *backtrace;
        !           291: 
        !           292:        DBG1(DBG_DMN, "thread %u received %d", thread_current_id(), signal);
        !           293:        backtrace = backtrace_create(2);
        !           294:        backtrace->log(backtrace, NULL, TRUE);
        !           295:        backtrace->log(backtrace, stderr, TRUE);
        !           296:        backtrace->destroy(backtrace);
        !           297: 
        !           298:        DBG1(DBG_DMN, "killing ourself, received critical signal");
        !           299:        abort();
        !           300: }
        !           301: #endif /* DISABLE_SIGNAL_HANDLER */
        !           302: 
        !           303: /**
        !           304:  * Add namespace alias
        !           305:  */
        !           306: static void __attribute__ ((constructor))register_namespace()
        !           307: {
        !           308:        /* inherit settings from charon */
        !           309:        library_add_namespace("charon");
        !           310: }
        !           311: 
        !           312: /**
        !           313:  * Register journal logger
        !           314:  */
        !           315: static void __attribute__ ((constructor))register_logger()
        !           316: {
        !           317:        register_custom_logger("journal", journal_logger_create);
        !           318: }
        !           319: 
        !           320: /**
        !           321:  * Main function, starts the daemon.
        !           322:  */
        !           323: int main(int argc, char *argv[])
        !           324: {
        !           325:        struct sigaction action;
        !           326:        struct utsname utsname;
        !           327:        int status = SS_RC_INITIALIZATION_FAILED;
        !           328: 
        !           329:        dbg = dbg_stderr;
        !           330: 
        !           331:        if (uname(&utsname) != 0)
        !           332:        {
        !           333:                memset(&utsname, 0, sizeof(utsname));
        !           334:        }
        !           335: 
        !           336:        sd_notifyf(0, "STATUS=Starting charon-systemd, strongSwan %s, %s %s, %s",
        !           337:                           VERSION, utsname.sysname, utsname.release, utsname.machine);
        !           338: 
        !           339:        atexit(library_deinit);
        !           340:        if (!library_init(NULL, "charon-systemd"))
        !           341:        {
        !           342:                sd_notifyf(0, "STATUS=libstrongswan initialization failed");
        !           343:                return SS_RC_INITIALIZATION_FAILED;
        !           344:        }
        !           345:        if (lib->integrity &&
        !           346:                !lib->integrity->check_file(lib->integrity, "charon-systemd", argv[0]))
        !           347:        {
        !           348:                sd_notifyf(0, "STATUS=integrity check of charon-systemd failed");
        !           349:                return SS_RC_INITIALIZATION_FAILED;
        !           350:        }
        !           351:        if (!libcharon_init())
        !           352:        {
        !           353:                sd_notifyf(0, "STATUS=libcharon initialization failed");
        !           354:                goto error;
        !           355:        }
        !           356:        if (!lookup_uid_gid())
        !           357:        {
        !           358:                sd_notifyf(0, "STATUS=unknown uid/gid");
        !           359:                goto error;
        !           360:        }
        !           361:        /* we registered the journal logger as custom logger, which gets its
        !           362:         * settings from <ns>.customlog.journal, let it fallback to <ns>.journal */
        !           363:        lib->settings->add_fallback(lib->settings, "%s.customlog.journal",
        !           364:                                                                "%s.journal", lib->ns);
        !           365:        /* load the journal logger by default */
        !           366:        lib->settings->set_default_str(lib->settings, "%s.journal.default", "1",
        !           367:                                                                   lib->ns);
        !           368: 
        !           369:        charon->load_loggers(charon);
        !           370: 
        !           371:        if (!charon->initialize(charon,
        !           372:                        lib->settings->get_str(lib->settings, "%s.load", PLUGINS, lib->ns)))
        !           373:        {
        !           374:                sd_notifyf(0, "STATUS=charon initialization failed");
        !           375:                goto error;
        !           376:        }
        !           377:        lib->plugins->status(lib->plugins, LEVEL_CTRL);
        !           378: 
        !           379:        if (!lib->caps->drop(lib->caps))
        !           380:        {
        !           381:                sd_notifyf(0, "STATUS=dropping capabilities failed");
        !           382:                goto error;
        !           383:        }
        !           384: 
        !           385:        /* add handler for fatal signals,
        !           386:         * INT, TERM and HUP are handled by sigwaitinfo() in run() */
        !           387:        action.sa_flags = 0;
        !           388:        sigemptyset(&action.sa_mask);
        !           389:        sigaddset(&action.sa_mask, SIGINT);
        !           390:        sigaddset(&action.sa_mask, SIGTERM);
        !           391:        sigaddset(&action.sa_mask, SIGHUP);
        !           392: 
        !           393:        /* optionally let the external system handle fatal signals */
        !           394: #ifndef DISABLE_SIGNAL_HANDLER
        !           395:        action.sa_handler = segv_handler;
        !           396:        sigaction(SIGSEGV, &action, NULL);
        !           397:        sigaction(SIGILL, &action, NULL);
        !           398:        sigaction(SIGBUS, &action, NULL);
        !           399: #endif /* DISABLE_SIGNAL_HANDLER */
        !           400: 
        !           401:        action.sa_handler = SIG_IGN;
        !           402:        sigaction(SIGPIPE, &action, NULL);
        !           403: 
        !           404:        pthread_sigmask(SIG_SETMASK, &action.sa_mask, NULL);
        !           405: 
        !           406:        charon->start(charon);
        !           407: 
        !           408:        sd_notifyf(0, "STATUS=charon-systemd running, strongSwan %s, %s %s, %s",
        !           409:                           VERSION, utsname.sysname, utsname.release, utsname.machine);
        !           410: 
        !           411:        status = run();
        !           412: 
        !           413: error:
        !           414:        libcharon_deinit();
        !           415:        return status;
        !           416: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>