gvpe/src/vped.C

/*
    vped.C -- the main file for vped
    Copyright (C) 1998-2002 Ivo Timmermans <ivo@o2w.nl>
                  2000-2002 Guus Sliepen <guus@sliepen.eu.org>
                  2003      Marc Lehmannn <pcg@goof.com>
 
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.
 
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.
 
    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/

#include "config.h"

#include <cstdio>
#include <cstring>

#include <errno.h>
#include <fcntl.h>
#include <getopt.h>
#include <signal.h>
#include <sys/types.h>
#include <unistd.h>
#include <signal.h>
#include <termios.h>

#include <sys/mman.h>

#include <openssl/rand.h>

#include "gettext.h"
#include "pidfile.h"

#include "conf.h"
#include "slog.h"
#include "util.h"
#include "protocol.h"

vpn network;

/* If nonzero, display usage information and exit. */
static int show_help;

/* If nonzero, print the version on standard output and exit.  */
static int show_version;

#if HAVE_MLOCKALL
/* If nonzero, disable swapping for this process. */
static int do_mlock = 0;
#endif

/* If zero, don't detach from the terminal. */
static int do_detach = 1;

static struct option const long_options[] =
    {
      {"config", required_argument, NULL, 'c'},
      {"help", no_argument, &show_help, 1},
      {"version", no_argument, &show_version, 1},
      {"no-detach", no_argument, &do_detach, 0},
      {"log-level", required_argument, NULL, 'l'},
#if HAVE_MLOCKALL
      {"mlock", no_argument, &do_mlock, 1},
#endif
      {NULL, 0, NULL, 0}
    };

static void
usage (int status)
{
  if (status != 0)
    fprintf (stderr, _("Try `%s --help\' for more information.\n"), get_identity ());
  else
    {
      printf (_("Usage: %s [option]... NODENAME\n\n"), get_identity ());
      printf (_
              ("  -c, --config=DIR           Read configuration options from DIR.\n"
               "  -D, --no-detach            Don't fork and detach.\n"
               "  -l, --log-level=LEVEL      Set logging level (info, notice, warn are common).\n"
               "  -L, --mlock                Lock tinc into main memory.\n"
               "      --help                 Display this help and exit.\n"
               "      --version              Output version information and exit.\n\n"));
      printf (_("Report bugs to <vpe@plan9.de>.\n"));
    }

  exit (status);
}

void
parse_options (int argc, char **argv, char **envp)
{
  int r;
  int option_index = 0;

  while ((r = getopt_long (argc, argv, "-c:DLl:", long_options, &option_index)) != EOF)
    {
      switch (r)
        {
        case 0:         /* long option */
          break;

        case 1:         /* this node name */
          thisnode = strdup (optarg);
          break;

        case 'c':               /* config file */
          confbase = strdup (optarg);
          break;

        case 'D':               /* no detach */
          do_detach = 0;
          break;

#if HAVE_MLOCKALL
        case 'L':               /* lock into memory */
          do_mlock = 1;
          break;
#endif

        case 'l':               /* inc debug level */
          {
            loglevel l = string_to_loglevel (optarg);

            if (l != L_NONE)
              set_loglevel (l);
            else
              slog (L_WARN, "'%s': %s", optarg, UNKNOWN_LOGLEVEL);
          }
          break;

        case '?':
          usage (1);

        default:
          break;
        }
    }
}

/*
  Close network connections, and terminate neatly
*/
void cleanup_and_exit(int c)
{
  network.shutdown_all ();

  if (pidfilename)
    remove_pid (pidfilename);

  slog (L_INFO, _("terminating with exit code %d"), c);

  exit (c);
}

/*
  Signal handlers.
*/
RETSIGTYPE
sigterm_handler (int a)
{
  network.events |= vpn::EVENT_SHUTDOWN;
}

RETSIGTYPE
sighup_handler (int a)
{
  network.events |= vpn::EVENT_RECONNECT;
}

RETSIGTYPE
sigusr1_handler (int a)
{
}

RETSIGTYPE
sigusr2_handler (int a)
{
}

void
setup_signals (void)
{
  struct sigaction act;

  sigfillset (&act.sa_mask);
  act.sa_flags = 0;

  act.sa_handler = sighup_handler;  sigaction (SIGHUP , &act, NULL);
  act.sa_handler = sigusr1_handler; sigaction (SIGUSR1, &act, NULL);
  act.sa_handler = sigusr2_handler; sigaction (SIGUSR2, &act, NULL);
  act.sa_handler = SIG_IGN;         sigaction (SIGCHLD, &act, NULL);
  act.sa_flags = SA_RESETHAND;
  act.sa_handler = sigterm_handler; sigaction (SIGINT , &act, NULL);
  act.sa_handler = sigterm_handler; sigaction (SIGTERM, &act, NULL);
}

int
main (int argc, char **argv, char **envp)
{
  set_loglevel (L_INFO);
  set_identity (argv[0]);
  log_to (LOGTO_SYSLOG | LOGTO_STDERR);

  setlocale (LC_ALL, "");
  bindtextdomain (PACKAGE, LOCALEDIR);
  textdomain (PACKAGE);

  parse_options (argc, argv, envp);

  if (show_version)
    {
      printf (_("%s version %s (built %s %s, protocol %d:%d)\n"), get_identity (),
              VERSION, __DATE__, __TIME__, PROTOCOL_MAJOR, PROTOCOL_MINOR);
      printf (_
              ("Copyright (C) 2003 Marc Lehmann <vpe@plan9.de> and others.\n"
               "See the AUTHORS file for a complete list.\n\n"
               "tinc comes with ABSOLUTELY NO WARRANTY.  This is free software,\n"
               "and you are welcome to redistribute it under certain conditions;\n"
               "see the file COPYING for details.\n"));

      return 0;
    }

  if (show_help)
    usage (0);

  log_to (LOGTO_SYSLOG | LOGTO_STDERR);

  /* Lock all pages into memory if requested */

#if HAVE_MLOCKALL
  if (do_mlock)
    if (mlockall (MCL_CURRENT | MCL_FUTURE))
      slog (L_ERR, _("system call `%s' failed: %s"), "mlockall", strerror (errno));
#endif

  make_names ();
  conf.read_config (true);

  RAND_load_file ("/dev/urandom", 1024);

  //OpenSSL_add_all_algorithms ();

  if (!THISNODE)
    {
      slog (L_ERR, _("current node not set, or node '%s' not found in configfile, use the -n switch when starting vped."),
            thisnode ? thisnode : "<unset>");
      exit (1);
    }

  if (detach (do_detach))
    exit (0);

  setup_signals ();

  if (!network.setup ())
    {
      network.main_loop ();
      cleanup_and_exit (1);
    }

  slog (L_ERR, _("unable to setup network, unrecoverable error, exiting."));
  cleanup_and_exit (1);
}

Revision:	1.1
Committed:	Sat Mar 1 15:53:03 2003 UTC (21 years, 2 months ago) by pcg
Content type:	text/plain
Branch:	MAIN
Log Message:	* empty log message *
#	User	Rev	Content
1	pcg	1.1	/*
2			vped.C -- the main file for vped
3			Copyright (C) 1998-2002 Ivo Timmermans <ivo@o2w.nl>
4			2000-2002 Guus Sliepen <guus@sliepen.eu.org>
5			2003 Marc Lehmannn <pcg@goof.com>
6
7			This program is free software; you can redistribute it and/or modify
8			it under the terms of the GNU General Public License as published by
9			the Free Software Foundation; either version 2 of the License, or
10			(at your option) any later version.
11
12			This program is distributed in the hope that it will be useful,
13			but WITHOUT ANY WARRANTY; without even the implied warranty of
14			MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15			GNU General Public License for more details.
16
17			You should have received a copy of the GNU General Public License
18			along with this program; if not, write to the Free Software
19			Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20			*/
21
22			#include "config.h"
23
24			#include <cstdio>
25			#include <cstring>
26
27			#include <errno.h>
28			#include <fcntl.h>
29			#include <getopt.h>
30			#include <signal.h>
31			#include <sys/types.h>
32			#include <unistd.h>
33			#include <signal.h>
34			#include <termios.h>
35
36			#include <sys/mman.h>
37
38			#include <openssl/rand.h>
39
40			#include "gettext.h"
41			#include "pidfile.h"
42
43			#include "conf.h"
44			#include "slog.h"
45			#include "util.h"
46			#include "protocol.h"
47
48			vpn network;
49
50			/* If nonzero, display usage information and exit. */
51			static int show_help;
52
53			/* If nonzero, print the version on standard output and exit. */
54			static int show_version;
55
56			#if HAVE_MLOCKALL
57			/* If nonzero, disable swapping for this process. */
58			static int do_mlock = 0;
59			#endif
60
61			/* If zero, don't detach from the terminal. */
62			static int do_detach = 1;
63
64			static struct option const long_options[] =
65			{
66			{"config", required_argument, NULL, 'c'},
67			{"help", no_argument, &show_help, 1},
68			{"version", no_argument, &show_version, 1},
69			{"no-detach", no_argument, &do_detach, 0},
70			{"log-level", required_argument, NULL, 'l'},
71			#if HAVE_MLOCKALL
72			{"mlock", no_argument, &do_mlock, 1},
73			#endif
74			{NULL, 0, NULL, 0}
75			};
76
77			static void
78			usage (int status)
79			{
80			if (status != 0)
81			fprintf (stderr, _("Try `%s --help\' for more information.\n"), get_identity ());
82			else
83			{
84			printf (_("Usage: %s [option]... NODENAME\n\n"), get_identity ());
85			printf (_
86			(" -c, --config=DIR Read configuration options from DIR.\n"
87			" -D, --no-detach Don't fork and detach.\n"
88			" -l, --log-level=LEVEL Set logging level (info, notice, warn are common).\n"
89			" -L, --mlock Lock tinc into main memory.\n"
90			" --help Display this help and exit.\n"
91			" --version Output version information and exit.\n\n"));
92			printf (_("Report bugs to <vpe@plan9.de>.\n"));
93			}
94
95			exit (status);
96			}
97
98			void
99			parse_options (int argc, char argv, char envp)
100			{
101			int r;
102			int option_index = 0;
103
104			while ((r = getopt_long (argc, argv, "-c:DLl:", long_options, &option_index)) != EOF)
105			{
106			switch (r)
107			{
108			case 0: /* long option */
109			break;
110
111			case 1: /* this node name */
112			thisnode = strdup (optarg);
113			break;
114
115			case 'c': /* config file */
116			confbase = strdup (optarg);
117			break;
118
119			case 'D': /* no detach */
120			do_detach = 0;
121			break;
122
123			#if HAVE_MLOCKALL
124			case 'L': /* lock into memory */
125			do_mlock = 1;
126			break;
127			#endif
128
129			case 'l': /* inc debug level */
130			{
131			loglevel l = string_to_loglevel (optarg);
132
133			if (l != L_NONE)
134			set_loglevel (l);
135			else
136			slog (L_WARN, "'%s': %s", optarg, UNKNOWN_LOGLEVEL);
137			}
138			break;
139
140			case '?':
141			usage (1);
142
143			default:
144			break;
145			}
146			}
147			}
148
149			/*
150			Close network connections, and terminate neatly
151			*/
152			void cleanup_and_exit(int c)
153			{
154			network.shutdown_all ();
155
156			if (pidfilename)
157			remove_pid (pidfilename);
158
159			slog (L_INFO, _("terminating with exit code %d"), c);
160
161			exit (c);
162			}
163
164			/*
165			Signal handlers.
166			*/
167			RETSIGTYPE
168			sigterm_handler (int a)
169			{
170			network.events \|= vpn::EVENT_SHUTDOWN;
171			}
172
173			RETSIGTYPE
174			sighup_handler (int a)
175			{
176			network.events \|= vpn::EVENT_RECONNECT;
177			}
178
179			RETSIGTYPE
180			sigusr1_handler (int a)
181			{
182			}
183
184			RETSIGTYPE
185			sigusr2_handler (int a)
186			{
187			}
188
189			void
190			setup_signals (void)
191			{
192			struct sigaction act;
193
194			sigfillset (&act.sa_mask);
195			act.sa_flags = 0;
196
197			act.sa_handler = sighup_handler; sigaction (SIGHUP , &act, NULL);
198			act.sa_handler = sigusr1_handler; sigaction (SIGUSR1, &act, NULL);
199			act.sa_handler = sigusr2_handler; sigaction (SIGUSR2, &act, NULL);
200			act.sa_handler = SIG_IGN; sigaction (SIGCHLD, &act, NULL);
201			act.sa_flags = SA_RESETHAND;
202			act.sa_handler = sigterm_handler; sigaction (SIGINT , &act, NULL);
203			act.sa_handler = sigterm_handler; sigaction (SIGTERM, &act, NULL);
204			}
205
206			int
207			main (int argc, char argv, char envp)
208			{
209			set_loglevel (L_INFO);
210			set_identity (argv[0]);
211			log_to (LOGTO_SYSLOG \| LOGTO_STDERR);
212
213			setlocale (LC_ALL, "");
214			bindtextdomain (PACKAGE, LOCALEDIR);
215			textdomain (PACKAGE);
216
217			parse_options (argc, argv, envp);
218
219			if (show_version)
220			{
221			printf (_("%s version %s (built %s %s, protocol %d:%d)\n"), get_identity (),
222			VERSION, __DATE__, __TIME__, PROTOCOL_MAJOR, PROTOCOL_MINOR);
223			printf (_
224			("Copyright (C) 2003 Marc Lehmann <vpe@plan9.de> and others.\n"
225			"See the AUTHORS file for a complete list.\n\n"
226			"tinc comes with ABSOLUTELY NO WARRANTY. This is free software,\n"
227			"and you are welcome to redistribute it under certain conditions;\n"
228			"see the file COPYING for details.\n"));
229
230			return 0;
231			}
232
233			if (show_help)
234			usage (0);
235
236			log_to (LOGTO_SYSLOG \| LOGTO_STDERR);
237
238			/* Lock all pages into memory if requested */
239
240			#if HAVE_MLOCKALL
241			if (do_mlock)
242			if (mlockall (MCL_CURRENT \| MCL_FUTURE))
243			slog (L_ERR, _("system call `%s' failed: %s"), "mlockall", strerror (errno));
244			#endif
245
246			make_names ();
247			conf.read_config (true);
248
249			RAND_load_file ("/dev/urandom", 1024);
250
251			//OpenSSL_add_all_algorithms ();
252
253			if (!THISNODE)
254			{
255			slog (L_ERR, _("current node not set, or node '%s' not found in configfile, use the -n switch when starting vped."),
256			thisnode ? thisnode : "<unset>");
257			exit (1);
258			}
259
260			if (detach (do_detach))
261			exit (0);
262
263			setup_signals ();
264
265			if (!network.setup ())
266			{
267			network.main_loop ();
268			cleanup_and_exit (1);
269			}
270
271			slog (L_ERR, _("unable to setup network, unrecoverable error, exiting."));
272			cleanup_and_exit (1);
273			}
274