www.tinc-vpn.org Git - tinc/blob - src/tincd.c

   1 /*
   2     tincd.c -- the main file for tincd
   3     Copyright (C) 1998-2005 Ivo Timmermans
   4                   2000-2009 Guus Sliepen <guus@tinc-vpn.org>
   5                   2008      Max Rijevski <maksuf@gmail.com>
   6                   2009      Michael Tokarev <mjt@tls.msk.ru>
   7
   8     This program is free software; you can redistribute it and/or modify
   9     it under the terms of the GNU General Public License as published by
  10     the Free Software Foundation; either version 2 of the License, or
  11     (at your option) any later version.
  12
  13     This program is distributed in the hope that it will be useful,
  14     but WITHOUT ANY WARRANTY; without even the implied warranty of
  15     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16     GNU General Public License for more details.
  17
  18     You should have received a copy of the GNU General Public License along
  19     with this program; if not, write to the Free Software Foundation, Inc.,
  20     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  21 */
  22
  23 #include "system.h"
  24
  25 /* Darwin (MacOS/X) needs the following definition... */
  26 #ifndef _P1003_1B_VISIBLE
  27 #define _P1003_1B_VISIBLE
  28 #endif
  29
  30 #ifdef HAVE_SYS_MMAN_H
  31 #include <sys/mman.h>
  32 #endif
  33
  34 #include LZO1X_H
  35
  36 #ifndef HAVE_MINGW
  37 #include <pwd.h>
  38 #include <grp.h>
  39 #include <time.h>
  40 #endif
  41
  42 #include <getopt.h>
  43
  44 #include "conf.h"
  45 #include "control.h"
  46 #include "crypto.h"
  47 #include "device.h"
  48 #include "logger.h"
  49 #include "net.h"
  50 #include "netutl.h"
  51 #include "process.h"
  52 #include "protocol.h"
  53 #include "utils.h"
  54 #include "xalloc.h"
  55
  56 /* The name this program was run with. */
  57 char *program_name = NULL;
  58
  59 /* If nonzero, display usage information and exit. */
  60 bool show_help = false;
  61
  62 /* If nonzero, print the version on standard output and exit.  */
  63 bool show_version = false;
  64
  65 /* If nonzero, use null ciphers and skip all key exchanges. */
  66 bool bypass_security = false;
  67
  68 /* If nonzero, disable swapping for this process. */
  69 bool do_mlock = false;
  70
  71 /* If nonzero, chroot to netdir after startup. */
  72 static bool do_chroot = false;
  73
  74 /* If !NULL, do setuid to given user after startup */
  75 static const char *switchuser = NULL;
  76
  77 /* If nonzero, write log entries to a separate file. */
  78 bool use_logfile = false;
  79
  80 char *identname = NULL;                         /* program name for syslog */
  81 char *logfilename = NULL;                       /* log file location */
  82 char *controlcookiename = NULL;
  83 char **g_argv;                                  /* a copy of the cmdline arguments */
  84
  85 static int status;
  86
  87 static struct option const long_options[] = {
  88         {"config", required_argument, NULL, 'c'},
  89         {"net", required_argument, NULL, 'n'},
  90         {"help", no_argument, NULL, 1},
  91         {"version", no_argument, NULL, 2},
  92         {"no-detach", no_argument, NULL, 'D'},
  93         {"debug", optional_argument, NULL, 'd'},
  94         {"bypass-security", no_argument, NULL, 3},
  95         {"mlock", no_argument, NULL, 'L'},
  96         {"chroot", no_argument, NULL, 'R'},
  97         {"user", required_argument, NULL, 'U'},
  98         {"logfile", optional_argument, NULL, 4},
  99         {"controlcookie", required_argument, NULL, 5},
 100         {NULL, 0, NULL, 0}
 101 };
 102
 103 #ifdef HAVE_MINGW
 104 static struct WSAData wsa_state;
 105 CRITICAL_SECTION mutex;
 106 #endif
 107
 108 static void usage(bool status) {
 109         if(status)
 110                 fprintf(stderr, "Try `%s --help\' for more information.\n",
 111                                 program_name);
 112         else {
 113                 printf("Usage: %s [option]...\n\n", program_name);
 114                 printf( "  -c, --config=DIR              Read configuration options from DIR.\n"
 115                                 "  -D, --no-detach               Don't fork and detach.\n"
 116                                 "  -d, --debug[=LEVEL]           Increase debug level or set it to LEVEL.\n"
 117                                 "  -n, --net=NETNAME             Connect to net NETNAME.\n"
 118                                 "  -L, --mlock                   Lock tinc into main memory.\n"
 119                                 "      --logfile[=FILENAME]      Write log entries to a logfile.\n"
 120                                 "      --controlcookie=FILENAME  Write control socket cookie to FILENAME.\n"
 121                                 "      --bypass-security         Disables meta protocol security, for debugging.\n"
 122                                 "  -R, --chroot                  chroot to NET dir at startup.\n"
 123                                 "  -U, --user=USER               setuid to given USER at startup.\n"                            "      --help                    Display this help and exit.\n"
 124                                 "      --version                 Output version information and exit.\n\n");
 125                 printf("Report bugs to tinc@tinc-vpn.org.\n");
 126         }
 127 }
 128
 129 static bool parse_options(int argc, char **argv) {
 130         int r;
 131         int option_index = 0;
 132
 133         while((r = getopt_long(argc, argv, "c:DLd::n:RU:", long_options, &option_index)) != EOF) {
 134                 switch (r) {
 135                         case 0:                         /* long option */
 136                                 break;
 137
 138                         case 'c':                               /* config file */
 139                                 confbase = xstrdup(optarg);
 140                                 break;
 141
 142                         case 'D':                               /* no detach */
 143                                 do_detach = false;
 144                                 break;
 145
 146                         case 'L':                               /* no detach */
 147 #ifndef HAVE_MLOCKALL
 148                                 logger(LOG_ERR, "%s not supported on this platform", "mlockall()");
 149                                 return false;
 150 #else
 151                                 do_mlock = true;
 152                                 break;
 153 #endif
 154
 155                         case 'd':                               /* inc debug level */
 156                                 if(optarg)
 157                                         debug_level = atoi(optarg);
 158                                 else
 159                                         debug_level++;
 160                                 break;
 161
 162                         case 'n':                               /* net name given */
 163                                 netname = xstrdup(optarg);
 164                                 break;
 165
 166                         case 'R':                               /* chroot to NETNAME dir */
 167                                 do_chroot = true;
 168                                 break;
 169
 170                         case 'U':                               /* setuid to USER */
 171                                 switchuser = optarg;
 172                                 break;
 173
 174                         case 1:                                 /* show help */
 175                                 show_help = true;
 176                                 break;
 177
 178                         case 2:                                 /* show version */
 179                                 show_version = true;
 180                                 break;
 181
 182                         case 3:                                 /* bypass security */
 183                                 bypass_security = true;
 184                                 break;
 185
 186                         case 4:                                 /* write log entries to a file */
 187                                 use_logfile = true;
 188                                 if(optarg)
 189                                         logfilename = xstrdup(optarg);
 190                                 break;
 191
 192                         case 5:                                 /* open control socket here */
 193                                 controlcookiename = xstrdup(optarg);
 194                                 break;
 195
 196                         case '?':
 197                                 usage(true);
 198                                 return false;
 199
 200                         default:
 201                                 break;
 202                 }
 203         }
 204
 205         return true;
 206 }
 207
 208 /*
 209   Set all files and paths according to netname
 210 */
 211 static void make_names(void) {
 212 #ifdef HAVE_MINGW
 213         HKEY key;
 214         char installdir[1024] = "";
 215         long len = sizeof installdir;
 216 #endif
 217
 218         if(netname)
 219                 xasprintf(&identname, "tinc.%s", netname);
 220         else
 221                 identname = xstrdup("tinc");
 222
 223 #ifdef HAVE_MINGW
 224         if(!RegOpenKeyEx(HKEY_LOCAL_MACHINE, "SOFTWARE\\tinc", 0, KEY_READ, &key)) {
 225                 if(!RegQueryValueEx(key, NULL, 0, 0, installdir, &len)) {
 226                         if(!logfilename)
 227                                 xasprintf(&logfilename, "%s/log/%s.log", identname);
 228                         if(!confbase) {
 229                                 if(netname)
 230                                         xasprintf(&confbase, "%s/%s", installdir, netname);
 231                                 else
 232                                         xasprintf(&confbase, "%s", installdir);
 233                         }
 234                         if(!controlcookiename)
 235                                 xasprintf(&controlcookiename, "%s/cookie", confbase);
 236                 }
 237                 RegCloseKey(key);
 238                 if(*installdir)
 239                         return;
 240         }
 241 #endif
 242
 243         if(!logfilename)
 244                 xasprintf(&logfilename, LOCALSTATEDIR "/log/%s.log", identname);
 245
 246         if(!controlcookiename)
 247                 xasprintf(&controlcookiename, LOCALSTATEDIR "/run/%s.cookie", identname);
 248
 249         if(netname) {
 250                 if(!confbase)
 251                         xasprintf(&confbase, CONFDIR "/tinc/%s", netname);
 252                 else
 253                         logger(LOG_INFO, "Both netname and configuration directory given, using the latter...");
 254         } else {
 255                 if(!confbase)
 256                         xasprintf(&confbase, CONFDIR "/tinc");
 257         }
 258 }
 259
 260 static void free_names() {
 261         if (identname) free(identname);
 262         if (netname) free(netname);
 263         if (controlcookiename) free(controlcookiename);
 264         if (logfilename) free(logfilename);
 265         if (confbase) free(confbase);
 266 }
 267
 268 static bool drop_privs() {
 269 #ifdef HAVE_MINGW
 270         if (switchuser) {
 271                 logger(LOG_ERR, "%s not supported on this platform", "-U");
 272                 return false;
 273         }
 274         if (do_chroot) {
 275                 logger(LOG_ERR, "%s not supported on this platform", "-R");
 276                 return false;
 277         }
 278 #else
 279         uid_t uid = 0;
 280         if (switchuser) {
 281                 struct passwd *pw = getpwnam(switchuser);
 282                 if (!pw) {
 283                         logger(LOG_ERR, "unknown user `%s'", switchuser);
 284                         return false;
 285                 }
 286                 uid = pw->pw_uid;
 287                 if (initgroups(switchuser, pw->pw_gid) != 0 ||
 288                     setgid(pw->pw_gid) != 0) {
 289                         logger(LOG_ERR, "System call `%s' failed: %s",
 290                                "initgroups", strerror(errno));
 291                         return false;
 292                 }
 293                 endgrent();
 294                 endpwent();
 295         }
 296         if (do_chroot) {
 297                 tzset();        /* for proper timestamps in logs */
 298                 if (chroot(confbase) != 0 || chdir("/") != 0) {
 299                         logger(LOG_ERR, "System call `%s' failed: %s",
 300                                "chroot", strerror(errno));
 301                         return false;
 302                 }
 303                 free(confbase);
 304                 confbase = xstrdup("");
 305         }
 306         if (switchuser)
 307                 if (setuid(uid) != 0) {
 308                         logger(LOG_ERR, "System call `%s' failed: %s",
 309                                "setuid", strerror(errno));
 310                         return false;
 311                 }
 312 #endif
 313         return true;
 314 }
 315
 316 #ifdef HAVE_MINGW
 317 # define setpriority(level) SetPriorityClass(GetCurrentProcess(), level)
 318 #else
 319 # define NORMAL_PRIORITY_CLASS 0
 320 # define BELOW_NORMAL_PRIORITY_CLASS 10
 321 # define HIGH_PRIORITY_CLASS -10
 322 # define setpriority(level) nice(level)
 323 #endif
 324
 325 int main(int argc, char **argv) {
 326         program_name = argv[0];
 327
 328         if(!parse_options(argc, argv))
 329                 return 1;
 330
 331         make_names();
 332
 333         if(show_version) {
 334                 printf("%s version %s (built %s %s, protocol %d)\n", PACKAGE,
 335                            VERSION, __DATE__, __TIME__, PROT_CURRENT);
 336                 printf("Copyright (C) 1998-2009 Ivo Timmermans, Guus Sliepen and others.\n"
 337                                 "See the AUTHORS file for a complete list.\n\n"
 338                                 "tinc comes with ABSOLUTELY NO WARRANTY.  This is free software,\n"
 339                                 "and you are welcome to redistribute it under certain conditions;\n"
 340                                 "see the file COPYING for details.\n");
 341
 342                 return 0;
 343         }
 344
 345         if(show_help) {
 346                 usage(false);
 347                 return 0;
 348         }
 349
 350 #ifdef HAVE_MINGW
 351         if(WSAStartup(MAKEWORD(2, 2), &wsa_state)) {
 352                 logger(LOG_ERR, "System call `%s' failed: %s", "WSAStartup", winerror(GetLastError()));
 353                 return 1;
 354         }
 355 #endif
 356
 357         openlogger("tinc", use_logfile?LOGMODE_FILE:LOGMODE_STDERR);
 358
 359         if(!event_init()) {
 360                 logger(LOG_ERR, "Error initializing libevent!");
 361                 return 1;
 362         }
 363
 364         g_argv = argv;
 365
 366         init_configuration(&config_tree);
 367
 368         /* Slllluuuuuuurrrrp! */
 369
 370         srand(time(NULL));
 371         crypto_init();
 372
 373         if(!read_server_config())
 374                 return 1;
 375
 376         if(lzo_init() != LZO_E_OK) {
 377                 logger(LOG_ERR, "Error initializing LZO compressor!");
 378                 return 1;
 379         }
 380
 381 #ifdef HAVE_MINGW
 382         if(!do_detach || !init_service())
 383                 return main2(argc, argv);
 384         else
 385                 return 1;
 386 }
 387
 388 int main2(int argc, char **argv) {
 389         InitializeCriticalSection(&mutex);
 390         EnterCriticalSection(&mutex);
 391 #endif
 392
 393         if(!detach())
 394                 return 1;
 395
 396 #ifdef HAVE_MLOCKALL
 397         /* Lock all pages into memory if requested.
 398          * This has to be done after daemon()/fork() so it works for child.
 399          * No need to do that in parent as it's very short-lived. */
 400         if(do_mlock && mlockall(MCL_CURRENT | MCL_FUTURE) != 0) {
 401                 logger(LOG_ERR, "System call `%s' failed: %s", "mlockall",
 402                    strerror(errno));
 403                 return 1;
 404         }
 405 #endif
 406
 407         /* Setup sockets and open device. */
 408
 409         if(!setup_network())
 410                 goto end;
 411
 412         if(!init_control())
 413                 return 1;
 414
 415         /* Initiate all outgoing connections. */
 416
 417         try_outgoing_connections();
 418
 419         /* Change process priority */
 420
 421         char *priority = 0;
 422
 423         if(get_config_string(lookup_config(config_tree, "ProcessPriority"), &priority)) {
 424                 if(!strcasecmp(priority, "Normal"))
 425                         setpriority(NORMAL_PRIORITY_CLASS);
 426                 else if(!strcasecmp(priority, "Low"))
 427                         setpriority(BELOW_NORMAL_PRIORITY_CLASS);
 428                 else if(!strcasecmp(priority, "High"))
 429                         setpriority(HIGH_PRIORITY_CLASS);
 430                 else {
 431                         logger(LOG_ERR, "Invalid priority `%s`!", priority);
 432                         goto end;
 433                 }
 434         }
 435
 436         /* drop privileges */
 437         if (!drop_privs())
 438                 goto end;
 439
 440         /* Start main loop. It only exits when tinc is killed. */
 441
 442         status = main_loop();
 443
 444         /* Shutdown properly. */
 445
 446         ifdebug(CONNECTIONS)
 447                 dump_device_stats();
 448
 449         close_network_connections();
 450
 451 end:
 452         logger(LOG_NOTICE, "Terminating");
 453
 454         exit_control();
 455
 456         crypto_exit();
 457
 458         exit_configuration(&config_tree);
 459         free_names();
 460
 461         return status;
 462 }