chiark / gitweb /
server/: Split peer and admin initialization into smaller pieces.
[tripe] / server / tripe.c
CommitLineData
410c8acf 1/* -*-c-*-
410c8acf 2 *
3 * Main program
4 *
5 * (c) 2001 Straylight/Edgeware
6 */
7
e04c2d50 8/*----- Licensing notice --------------------------------------------------*
410c8acf 9 *
10 * This file is part of Trivial IP Encryption (TrIPE).
11 *
11ad66c2
MW
12 * TrIPE is free software: you can redistribute it and/or modify it under
13 * the terms of the GNU General Public License as published by the Free
14 * Software Foundation; either version 3 of the License, or (at your
15 * option) any later version.
e04c2d50 16 *
11ad66c2
MW
17 * TrIPE is distributed in the hope that it will be useful, but WITHOUT
18 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
19 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
20 * for more details.
e04c2d50 21 *
410c8acf 22 * You should have received a copy of the GNU General Public License
11ad66c2 23 * along with TrIPE. If not, see <https://www.gnu.org/licenses/>.
410c8acf 24 */
25
410c8acf 26/*----- Header files ------------------------------------------------------*/
27
28#include "tripe.h"
29
30/*----- Global variables --------------------------------------------------*/
31
32sel_state sel;
410c8acf 33
34/*----- Static variables --------------------------------------------------*/
35
36static sel_timer it;
37#define T_INTERVAL MIN(1)
38
39/*----- Main code ---------------------------------------------------------*/
40
41/* --- @interval@ --- *
42 *
43 * Arguments: @struct timeval *tv@ = time when called
44 * @void *v@ = boring pointer
45 *
46 * Returns: ---
47 *
48 * Use: Called periodically to do housekeeping tasks.
49 */
50
42da2a58 51static void interval(struct timeval *tv, void *v)
410c8acf 52{
53 struct timeval tvv;
54 T( trace(T_PEER, "peer: interval timer"); )
b5c45da1 55 rand_seed(RAND_GLOBAL, MAXHASHSZ);
410c8acf 56 p_interval();
57 tvv = *tv;
58 tvv.tv_sec += T_INTERVAL;
59 sel_addtimer(&sel, &it, &tvv, interval, v);
60}
61
62/* --- @main@ --- *
63 *
64 * Arguments: @int argc@ = number of command line arguments
65 * @char *argv[]@ = vector of arguments
66 *
67 * Returns: Zero if OK, nonzero on error.
68 *
69 * Use: Main program. Provides a simple VPN.
70 */
71
72static void usage(FILE *fp)
73{
46dde080 74 pquis(fp, "Usage: $ [-DF] [-d DIR] [-b ADDR] [-p PORT] [-n TUNNEL]\n\
a9279e37 75 [-U USER] [-G GROUP] [-a SOCKET] [-m MODE] [-T TRACE-OPTS]\n\
2d752320 76 [-k PRIV-KEYRING] [-K PUB-KEYRING] [-t KEY-TAG]\n");
410c8acf 77}
78
f98df549 79static void version(FILE *fp) { pquis(fp, "$, version " VERSION "\n"); }
410c8acf 80
81static void help(FILE *fp)
82{
83 version(fp);
84 fputc('\n', fp);
85 usage(fp);
86 fputs("\n\
87Options:\n\
88\n\
89-h, --help Display this help text.\n\
90-v, --version Display version number.\n\
91-u, --usage Display pointless usage message.\n\
42da2a58 92 --tunnels Display IP tunnel drivers and exit.\n\
410c8acf 93\n\
47828bd9
MW
94-4, --ipv4 Transport over IPv4 only.\n\
95-6, --ipv6 Transport over IPv6 only.\n\
410c8acf 96-D, --daemon Run in the background.\n\
46dde080 97-F, --foreground Quit when stdin reports end-of-file.\n\
ef4a1ab7 98-d, --directory=DIR Switch to directory DIR [default " CONFIGDIR "].\n\
d13e5724 99-b, --bind-address=ADDR Bind UDP socket to this IP ADDR.\n\
165efde7
MW
100-p, --port=PORT Select UDP port to listen to "
101 "[default " STR(TRIPE_PORT) "].\n\
42da2a58 102-n, --tunnel=TUNNEL Seelect default tunnel driver.\n\
d13e5724 103-U, --setuid=USER Set uid to USER after initialization.\n\
104-G, --setgid=GROUP Set gid to GROUP after initialization.\n\
410c8acf 105-k, --priv-keyring=FILE Get private key from FILE.\n\
106-K, --pub-keyring=FILE Get public keys from FILE.\n\
107-t, --tag=KEYTAG Use private key labelled TAG.\n\
108-a, --admin-socket=FILE Use FILE as the adminstration socket.\n\
a9279e37 109-m, --admin-perms=MODE Permissions to set on admin socket [default 600].\n\
c5a2ea9b 110" T( "\
410c8acf 111-T, --trace=OPTIONS Turn on tracing options.\n\
c5a2ea9b 112" ) "\
410c8acf 113", fp);
114}
115
116int main(int argc, char *argv[])
117{
118 const char *kr_priv = "keyring", *kr_pub = "keyring.pub";
fc5f4823 119 const char *tag_priv = 0;
ef4a1ab7 120 const char *csock = SOCKETDIR "/tripesock";
a9279e37 121 int csockmode = 0600;
ef4a1ab7 122 const char *dir = CONFIGDIR;
410c8acf 123 const char *p;
89640f3f
MW
124 const char *bindhost = 0, *bindsvc = STR(TRIPE_PORT);
125 struct addrinfo aihint = { 0 }, *ailist;
410c8acf 126 unsigned f = 0;
42da2a58 127 int i;
89640f3f 128 int err, selerr = 0;
46dde080 129 unsigned af;
42da2a58 130 struct timeval tv;
410c8acf 131 uid_t u = -1;
132 gid_t g = -1;
133
134#define f_bogus 1u
135#define f_daemon 2u
46dde080 136#define f_foreground 4u
410c8acf 137
138 ego(argv[0]);
d360f042 139 T( trace_on(stderr, 0); )
410c8acf 140
141 if ((p = getenv("TRIPEDIR")) != 0)
142 dir = p;
797cf76b
MW
143 if ((p = getenv("TRIPESOCK")) != 0)
144 csock = p;
42da2a58 145 tun_default = tunnels[0];
47828bd9 146 aihint.ai_family = AF_UNSPEC;
410c8acf 147
148 for (;;) {
149 static const struct option opts[] = {
150 { "help", 0, 0, 'h' },
151 { "version", 0, 0, 'v' },
152 { "usage", 0, 0, 'u' },
42da2a58 153 { "tunnels", 0, 0, '0' },
410c8acf 154
47828bd9
MW
155 { "ipv4", 0, 0, '4' },
156 { "ipv6", 0, 0, '6' },
410c8acf 157 { "daemon", 0, 0, 'D' },
46dde080 158 { "foreground", 0, 0, 'F' },
410c8acf 159 { "uid", OPTF_ARGREQ, 0, 'U' },
160 { "setuid", OPTF_ARGREQ, 0, 'U' },
161 { "gid", OPTF_ARGREQ, 0, 'G' },
162 { "setgid", OPTF_ARGREQ, 0, 'G' },
767b36e2 163 { "bind-address", OPTF_ARGREQ, 0, 'b' },
42da2a58 164 { "tunnel", OPTF_ARGREQ, 0, 'n' },
410c8acf 165 { "port", OPTF_ARGREQ, 0, 'p' },
166 { "directory", OPTF_ARGREQ, 0, 'd' },
167 { "priv-keyring", OPTF_ARGREQ, 0, 'k' },
168 { "pub-keyring", OPTF_ARGREQ, 0, 'K' },
169 { "tag", OPTF_ARGREQ, 0, 't' },
170 { "admin-socket", OPTF_ARGREQ, 0, 'a' },
a9279e37 171 { "admin-perms", OPTF_ARGREQ, 0, 'm' },
410c8acf 172#ifndef NTRACE
173 { "trace", OPTF_ARGREQ, 0, 'T' },
174#endif
175
176 { 0, 0, 0, 0 }
177 };
178
47828bd9 179 i = mdwopt(argc, argv, "hvu46DFU:G:b:n:p:d:k:K:t:a:m:" T("T:"),
42da2a58 180 opts, 0, 0, 0);
410c8acf 181 if (i < 0)
182 break;
183 switch (i) {
184 case 'h':
185 help(stdout);
186 exit(0);
187 case 'v':
188 version(stdout);
189 exit(0);
190 case 'u':
191 usage(stdout);
192 exit(0);
193
47828bd9
MW
194 case '4':
195 aihint.ai_family = AF_INET;
196 break;
197 case '6':
198 aihint.ai_family = AF_INET6;
199 break;
410c8acf 200 case 'D':
201 f |= f_daemon;
202 break;
52b86648
MW
203 case 'U':
204 u = u_getuser(optarg, &g);
205 break;
206 case 'G':
207 g = u_getgroup(optarg);
208 break;
46dde080
MW
209 case 'F':
210 f |= f_foreground;
211 break;
767b36e2 212
89640f3f
MW
213 case 'b':
214 bindhost = optarg;
215 break;
216 case 'p':
217 bindsvc = optarg;
218 break;
42da2a58 219 case 'n': {
220 int i;
221 for (i = 0;; i++) {
222 if (!tunnels[i])
223 die(EXIT_FAILURE, "unknown tunnel `%s'", optarg);
224 if (mystrieq(optarg, tunnels[i]->name))
225 break;
226 }
227 tun_default = tunnels[i];
228 } break;
410c8acf 229 case 'd':
230 dir = optarg;
231 break;
232 case 'k':
233 kr_priv = optarg;
234 break;
235 case 'K':
236 kr_pub = optarg;
237 break;
238 case 'a':
239 csock = optarg;
240 break;
a9279e37
MW
241 case 'm': {
242 char *p;
243 csockmode = strtol(optarg, &p, 8);
244 if (*p) die(EXIT_FAILURE, "bad permissions: `%s'", optarg);
245 } break;
410c8acf 246 case 't':
247 tag_priv = optarg;
248 break;
249#ifndef NTRACE
250 case 'T':
251 tr_flags = traceopt(tr_opts, optarg, tr_flags, 0);
252 trace_level(tr_flags);
253 break;
254#endif
ef4a1ab7 255 case '0': {
42da2a58 256 int i;
257 for (i = 0; tunnels[i]; i++)
258 puts(tunnels[i]->name);
ef4a1ab7 259 exit(0);
260 } break;
410c8acf 261 default:
262 f |= f_bogus;
263 break;
264 }
265 }
266
267 if (optind < argc || (f & f_bogus)) {
268 usage(stderr);
269 exit(EXIT_FAILURE);
270 }
46dde080
MW
271 if (!(~f & (f_daemon | f_foreground)))
272 die(EXIT_FAILURE, "foreground operation for a daemon is silly");
410c8acf 273
89640f3f
MW
274 aihint.ai_protocol = IPPROTO_UDP;
275 aihint.ai_socktype = SOCK_DGRAM;
276 aihint.ai_flags = AI_PASSIVE | AI_ADDRCONFIG;
277 if ((err = getaddrinfo(bindhost, bindsvc, &aihint, &ailist)) != 0) {
278 die(EXIT_FAILURE, "couldn't resolve hostname %c%s%c, port `%s': %s",
279 bindhost ? '`' : '<',
280 bindhost ? bindhost : "nil",
281 bindhost ? '\'' : '>',
282 bindsvc, gai_strerror(err));
283 }
284
410c8acf 285 if (chdir(dir)) {
286 die(EXIT_FAILURE, "can't set current directory to `%s': %s",
287 dir, strerror(errno));
288 }
289
290 sel_init(&sel);
291 sig_init(&sel);
292 rand_noisesrc(RAND_GLOBAL, &noise_source);
b5c45da1 293 rand_seed(RAND_GLOBAL, MAXHASHSZ);
410c8acf 294 signal(SIGPIPE, SIG_IGN);
42da2a58 295 for (i = 0; tunnels[i]; i++)
296 tunnels[i]->init();
7737eb87
MW
297 p_init();
298 p_bind(ailist); freeaddrinfo(ailist);
3cdc3f3a 299 if (!(f & f_daemon)) {
46dde080 300 af = AF_WARN;
3cdc3f3a 301#ifndef NTRACE
46dde080 302 af |= AF_TRACE;
3cdc3f3a 303#endif
46dde080
MW
304 if (f & f_foreground)
305 af |= AF_FOREGROUND;
306 a_create(STDIN_FILENO, STDOUT_FILENO, af);
3cdc3f3a 307 }
97d410cd
MW
308
309 for (i = 0; tunnels[i]; i++) {
310 if (tunnels[i]->flags&TUNF_PRIVOPEN) {
311 ps_split(f & f_daemon);
312 break;
313 }
314 }
315
7737eb87
MW
316 a_init();
317 a_signals();
318 a_listen(csock, u, g, csockmode);
319 a_switcherr();
52b86648 320 u_setugid(u, g);
410c8acf 321 km_init(kr_priv, kr_pub, tag_priv);
8362ac1c 322 kx_init();
410c8acf 323 if (f & f_daemon) {
19dd2531 324 if (daemonize())
410c8acf 325 die(EXIT_FAILURE, "couldn't become a daemon: %s", strerror(errno));
326 a_daemon();
327 }
328
42da2a58 329 tv.tv_sec = time(0) + T_INTERVAL;
330 tv.tv_usec = 0;
331 sel_addtimer(&sel, &it, &tv, interval, 0);
410c8acf 332
42da2a58 333 for (;;) {
c511e1f9 334 a_preselect();
42da2a58 335 if (!sel_select(&sel))
336 selerr = 0;
337 else if (errno != EINTR && errno != EAGAIN) {
f43df819 338 a_warn("SERVER", "select-error", "?ERRNO", A_END);
42da2a58 339 selerr++;
340 if (selerr > 8) {
f43df819 341 a_warn("ABORT", "repeated-select-errors", A_END);
42da2a58 342 abort();
410c8acf 343 }
344 }
345 }
346
347 return (0);
348}
349
350/*----- That's all, folks -------------------------------------------------*/