src/charon-tkm/src/charon-tkm.c

   1 /*
   2  * Copyright (C) 2012 Tobias Brunner
   3  * Copyright (C) 2012 Reto Buerki
   4  * Copyright (C) 2012 Adrian-Ken Rueegsegger
   5  * Hochschule fuer Technik Rapperswil
   6  *
   7  * This program is free software; you can redistribute it and/or modify it
   8  * under the terms of the GNU General Public License as published by the
   9  * Free Software Foundation; either version 2 of the License, or (at your
  10  * option) any later version.  See <http://www.fsf.org/copyleft/gpl.txt>.
  11  *
  12  * This program is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  14  * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  15  * for more details.
  16  */
  17
  18 #define _GNU_SOURCE
  19
  20 #include <stdio.h>
  21 #include <syslog.h>
  22 #include <signal.h>
  23 #include <sys/stat.h>
  24 #include <sys/types.h>
  25 #include <unistd.h>
  26 #include <libgen.h>
  27
  28 #include <hydra.h>
  29 #include <daemon.h>
  30 #include <library.h>
  31 #include <utils/backtrace.h>
  32 #include <threading/thread.h>
  33 #include <sa/keymat.h>
  34 #include <credentials/credential_manager.h>
  35
  36 #include "tkm.h"
  37 #include "tkm_nonceg.h"
  38 #include "tkm_diffie_hellman.h"
  39 #include "tkm_keymat.h"
  40 #include "tkm_listener.h"
  41 #include "tkm_kernel_ipsec.h"
  42 #include "tkm_public_key.h"
  43 #include "tkm_cred.h"
  44 #include "tkm_encoder.h"
  45
  46 /**
  47  * TKM bus listener for IKE authorize events.
  48  */
  49 static tkm_listener_t *listener;
  50
  51 /**
  52  * PID file, in which charon-tkm stores its process id
  53  */
  54 static char *pidfile_name = NULL;
  55
  56 /**
  57  * Global reference to PID file (required to truncate, if undeletable)
  58  */
  59 static FILE *pidfile = NULL;
  60
  61 /**
  62  * Hook in library for debugging messages
  63  */
  64 extern void (*dbg) (debug_t group, level_t level, char *fmt, ...);
  65
  66 /**
  67  * Simple logging hook for library logs, using syslog output
  68  */
  69 static void dbg_syslog(debug_t group, level_t level, char *fmt, ...)
  70 {
  71         if (level <= 1)
  72         {
  73                 char buffer[8192];
  74                 va_list args;
  75
  76                 va_start(args, fmt);
  77                 /* write in memory buffer first */
  78                 vsnprintf(buffer, sizeof(buffer), fmt, args);
  79                 syslog(LOG_DAEMON|LOG_INFO, "00[%s] %s", debug_names->names[group],
  80                                 buffer);
  81                 va_end(args);
  82         }
  83 }
  84
  85 /**
  86  * Run the daemon and handle unix signals
  87  */
  88 static void run()
  89 {
  90         sigset_t set;
  91
  92         /* handle SIGINT and SIGTERM in this handler */
  93         sigemptyset(&set);
  94         sigaddset(&set, SIGINT);
  95         sigaddset(&set, SIGTERM);
  96         sigprocmask(SIG_BLOCK, &set, NULL);
  97
  98         while (TRUE)
  99         {
 100                 int sig;
 101                 int error;
 102
 103                 error = sigwait(&set, &sig);
 104                 if (error)
 105                 {
 106                         DBG1(DBG_DMN, "error %d while waiting for a signal", error);
 107                         return;
 108                 }
 109                 switch (sig)
 110                 {
 111                         case SIGINT:
 112                         {
 113                                 DBG1(DBG_DMN, "signal of type SIGINT received. Shutting down");
 114                                 charon->bus->alert(charon->bus, ALERT_SHUTDOWN_SIGNAL, sig);
 115                                 return;
 116                         }
 117                         case SIGTERM:
 118                         {
 119                                 DBG1(DBG_DMN, "signal of type SIGTERM received. Shutting down");
 120                                 charon->bus->alert(charon->bus, ALERT_SHUTDOWN_SIGNAL, sig);
 121                                 return;
 122                         }
 123                         default:
 124                         {
 125                                 DBG1(DBG_DMN, "unknown signal %d received. Ignored", sig);
 126                                 break;
 127                         }
 128                 }
 129         }
 130 }
 131
 132 /**
 133  * Handle SIGSEGV/SIGILL signals raised by threads
 134  */
 135 static void segv_handler(int signal)
 136 {
 137         backtrace_t *backtrace;
 138
 139         DBG1(DBG_DMN, "thread %u received %d", thread_current_id(), signal);
 140         backtrace = backtrace_create(2);
 141         backtrace->log(backtrace, stderr, TRUE);
 142         backtrace->destroy(backtrace);
 143
 144         DBG1(DBG_DMN, "killing ourself, received critical signal");
 145         abort();
 146 }
 147
 148 /**
 149  * Lookup UID and GID
 150  */
 151 static bool lookup_uid_gid()
 152 {
 153 #ifdef IPSEC_USER
 154         if (!lib->caps->resolve_uid(lib->caps, IPSEC_USER))
 155         {
 156                 return FALSE;
 157         }
 158 #endif
 159 #ifdef IPSEC_GROUP
 160         if (!lib->caps->resolve_gid(lib->caps, IPSEC_GROUP))
 161         {
 162                 return FALSE;
 163         }
 164 #endif
 165         return TRUE;
 166 }
 167
 168 /**
 169  * Check/create PID file, return TRUE if already running
 170  */
 171 static bool check_pidfile()
 172 {
 173         struct stat stb;
 174
 175         if (stat(pidfile_name, &stb) == 0)
 176         {
 177                 pidfile = fopen(pidfile_name, "r");
 178                 if (pidfile)
 179                 {
 180                         char buf[64];
 181                         pid_t pid = 0;
 182
 183                         memset(buf, 0, sizeof(buf));
 184                         if (fread(buf, 1, sizeof(buf), pidfile))
 185                         {
 186                                 buf[sizeof(buf) - 1] = '\0';
 187                                 pid = atoi(buf);
 188                         }
 189                         fclose(pidfile);
 190                         if (pid && kill(pid, 0) == 0)
 191                         {       /* such a process is running */
 192                                 return TRUE;
 193                         }
 194                 }
 195                 DBG1(DBG_DMN, "removing pidfile '%s', process not running", pidfile_name);
 196                 unlink(pidfile_name);
 197         }
 198
 199         /* create new pidfile */
 200         pidfile = fopen(pidfile_name, "w");
 201         if (pidfile)
 202         {
 203                 ignore_result(fchown(fileno(pidfile),
 204                                                          lib->caps->get_uid(lib->caps),
 205                                                          lib->caps->get_gid(lib->caps)));
 206                 fprintf(pidfile, "%d\n", getpid());
 207                 fflush(pidfile);
 208         }
 209         return FALSE;
 210 }
 211
 212 /**
 213  * Delete/truncate the PID file
 214  */
 215 static void unlink_pidfile()
 216 {
 217         /* because unlinking the PID file may fail, we truncate it to ensure the
 218          * daemon can be properly restarted.  one probable cause for this is the
 219          * combination of not running as root and the effective user lacking
 220          * permissions on the parent dir(s) of the PID file */
 221         if (pidfile)
 222         {
 223                 ignore_result(ftruncate(fileno(pidfile), 0));
 224                 fclose(pidfile);
 225         }
 226         unlink(pidfile_name);
 227 }
 228 /**
 229  * Main function, starts TKM backend.
 230  */
 231 int main(int argc, char *argv[])
 232 {
 233         char *dmn_name;
 234         if (argc > 0 && strlen(argv[0]) > 0)
 235         {
 236                 dmn_name = basename(argv[0]);
 237         }
 238         else
 239         {
 240                 dmn_name = "charon-tkm";
 241         }
 242
 243         /* TKM credential set */
 244         tkm_cred_t *creds;
 245
 246         struct sigaction action;
 247         int status = SS_RC_INITIALIZATION_FAILED;
 248
 249         /* logging for library during initialization, as we have no bus yet */
 250         dbg = dbg_syslog;
 251
 252         /* initialize library */
 253         if (!library_init(NULL, dmn_name))
 254         {
 255                 library_deinit();
 256                 exit(status);
 257         }
 258
 259         if (!libhydra_init())
 260         {
 261                 dbg_syslog(DBG_DMN, 1, "initialization failed - aborting %s", dmn_name);
 262                 libhydra_deinit();
 263                 library_deinit();
 264                 exit(status);
 265         }
 266
 267         if (!libcharon_init())
 268         {
 269                 dbg_syslog(DBG_DMN, 1, "initialization failed - aborting %s", dmn_name);
 270                 goto deinit;
 271         }
 272
 273         if (!lookup_uid_gid())
 274         {
 275                 dbg_syslog(DBG_DMN, 1, "invalid uid/gid - aborting %s", dmn_name);
 276                 goto deinit;
 277         }
 278
 279         /* make sure we log to the DAEMON facility by default */
 280         lib->settings->set_int(lib->settings, "%s.syslog.daemon.default",
 281                         lib->settings->get_int(lib->settings, "%s.syslog.daemon.default", 1,
 282                                                                    dmn_name), dmn_name);
 283         charon->load_loggers(charon, NULL, FALSE);
 284
 285         DBG1(DBG_DMN, "Starting charon with TKM backend (strongSwan "VERSION")");
 286
 287         /* register TKM specific plugins */
 288         static plugin_feature_t features[] = {
 289                 PLUGIN_REGISTER(NONCE_GEN, tkm_nonceg_create),
 290                         PLUGIN_PROVIDE(NONCE_GEN),
 291                 PLUGIN_REGISTER(PUBKEY, tkm_public_key_load, TRUE),
 292                         PLUGIN_PROVIDE(PUBKEY, KEY_RSA),
 293                         PLUGIN_PROVIDE(PUBKEY_VERIFY, SIGN_RSA_EMSA_PKCS1_SHA1),
 294                         PLUGIN_PROVIDE(PUBKEY_VERIFY, SIGN_RSA_EMSA_PKCS1_SHA256),
 295                 PLUGIN_CALLBACK(kernel_ipsec_register, tkm_kernel_ipsec_create),
 296                         PLUGIN_PROVIDE(CUSTOM, "kernel-ipsec"),
 297         };
 298         lib->plugins->add_static_features(lib->plugins, "tkm-backend", features,
 299                         countof(features), TRUE, NULL, NULL);
 300
 301         if (!register_dh_mapping())
 302         {
 303                 DBG1(DBG_DMN, "no DH group mapping defined - aborting %s", dmn_name);
 304                 goto deinit;
 305         }
 306
 307         /* register TKM keymat variant */
 308         keymat_register_constructor(IKEV2, (keymat_constructor_t)tkm_keymat_create);
 309
 310         /* initialize daemon */
 311         if (!charon->initialize(charon, PLUGINS))
 312         {
 313                 DBG1(DBG_DMN, "initialization failed - aborting %s", dmn_name);
 314                 goto deinit;
 315         }
 316         lib->plugins->status(lib->plugins, LEVEL_CTRL);
 317
 318         /* set global pidfile name depending on daemon name */
 319         if (asprintf(&pidfile_name, IPSEC_PIDDIR"/%s.pid", dmn_name) < 0)
 320         {
 321                 DBG1(DBG_DMN, "unable to set pidfile name - aborting %s", dmn_name);
 322                 goto deinit;
 323         };
 324
 325         if (check_pidfile())
 326         {
 327                 DBG1(DBG_DMN, "%s already running (\"%s\" exists)", dmn_name,
 328                          pidfile_name);
 329                 goto deinit;
 330         }
 331
 332         if (!lib->caps->drop(lib->caps))
 333         {
 334                 DBG1(DBG_DMN, "capability dropping failed - aborting %s", dmn_name);
 335                 goto deinit;
 336         }
 337
 338         /* initialize TKM client */
 339         if (!tkm_init())
 340         {
 341                 DBG1(DBG_DMN, "init of TKM client failed - aborting %s", dmn_name);
 342                 goto deinit;
 343         }
 344
 345         /* register TKM authorization hook */
 346         listener = tkm_listener_create();
 347         charon->bus->add_listener(charon->bus, &listener->listener);
 348
 349         /* register TKM credential set */
 350         creds = tkm_cred_create();
 351         lib->credmgr->add_set(lib->credmgr, (credential_set_t*)creds);
 352
 353         /* register TKM credential encoder */
 354         lib->encoding->add_encoder(lib->encoding, tkm_encoder_encode);
 355
 356         /* add handler for SEGV and ILL,
 357          * INT and TERM are handled by sigwait() in run() */
 358         action.sa_handler = segv_handler;
 359         action.sa_flags = 0;
 360         sigemptyset(&action.sa_mask);
 361         sigaddset(&action.sa_mask, SIGINT);
 362         sigaddset(&action.sa_mask, SIGTERM);
 363         sigaction(SIGSEGV, &action, NULL);
 364         sigaction(SIGILL, &action, NULL);
 365         sigaction(SIGBUS, &action, NULL);
 366         action.sa_handler = SIG_IGN;
 367         sigaction(SIGPIPE, &action, NULL);
 368
 369         pthread_sigmask(SIG_SETMASK, &action.sa_mask, NULL);
 370
 371         /* start daemon (i.e. the threads in the thread-pool) */
 372         charon->start(charon);
 373
 374         /* main thread goes to run loop */
 375         run();
 376
 377         unlink_pidfile();
 378         status = 0;
 379         charon->bus->remove_listener(charon->bus, &listener->listener);
 380         listener->destroy(listener);
 381         creds->destroy(creds);
 382         lib->encoding->remove_encoder(lib->encoding, tkm_encoder_encode);
 383
 384 deinit:
 385         destroy_dh_mapping();
 386         libcharon_deinit();
 387         libhydra_deinit();
 388         library_deinit();
 389         tkm_deinit();
 390         return status;
 391 }