daemon.c 27.5 KB
Newer Older
1
#include "cache.h"
2
#include "pkt-line.h"
3
#include "exec_cmd.h"
4
#include "interpolate.h"
P
Petr Baudis 已提交
5

6 7
#include <syslog.h>

8 9 10 11
#ifndef HOST_NAME_MAX
#define HOST_NAME_MAX 256
#endif

12 13 14 15
#ifndef NI_MAXSERV
#define NI_MAXSERV 32
#endif

16
static int log_syslog;
P
Petr Baudis 已提交
17
static int verbose;
18
static int reuseaddr;
19
static int child_handler_pipe[2];
P
Petr Baudis 已提交
20

21
static const char daemon_usage[] =
S
Stephan Beyer 已提交
22
"git daemon [--verbose] [--syslog] [--export-all]\n"
P
Petr Baudis 已提交
23
"           [--timeout=n] [--init-timeout=n] [--strict-paths]\n"
J
Jens Axboe 已提交
24 25
"           [--base-path=path] [--base-path-relaxed]\n"
"           [--user-path | --user-path=path]\n"
26
"           [--interpolated-path=path]\n"
27
"           [--reuseaddr] [--detach] [--pid-file=file]\n"
28
"           [--[enable|disable|allow-override|forbid-override]=service]\n"
29 30 31
"           [--inetd | [--listen=host_or_ipaddr] [--port=n]\n"
"                      [--user=user [--group=group]]\n"
"           [directory...]";
32 33

/* List of acceptable pathname prefixes */
34 35
static char **ok_paths;
static int strict_paths;
36 37

/* If this is set, git-daemon-export-ok is not required */
38
static int export_all_trees;
P
Petr Baudis 已提交
39

P
Petr Baudis 已提交
40
/* Take all paths relative to this one if non-NULL */
41
static char *base_path;
42
static char *interpolated_path;
J
Jens Axboe 已提交
43
static int base_path_relaxed;
44 45 46

/* Flag indicating client sent extra args. */
static int saw_extended_args;
P
Petr Baudis 已提交
47

48 49 50 51
/* If defined, ~user notation is allowed and the string is inserted
 * after ~user/.  E.g. a request to git://host/~alice/frotz would
 * go to /home/alice/pub_git/frotz with --user-path=pub_git.
 */
52
static const char *user_path;
53

54
/* Timeout, and initial timeout */
55 56
static unsigned int timeout;
static unsigned int init_timeout;
P
Petr Baudis 已提交
57

58 59 60 61 62
/*
 * Static table for now.  Ugh.
 * Feel free to make dynamic as needed.
 */
#define INTERP_SLOT_HOST	(0)
63 64 65 66 67
#define INTERP_SLOT_CANON_HOST	(1)
#define INTERP_SLOT_IP		(2)
#define INTERP_SLOT_PORT	(3)
#define INTERP_SLOT_DIR		(4)
#define INTERP_SLOT_PERCENT	(5)
68 69 70

static struct interp interp_table[] = {
	{ "%H", 0},
71 72 73
	{ "%CH", 0},
	{ "%IP", 0},
	{ "%P", 0},
74
	{ "%D", 0},
75
	{ "%%", 0},
76 77 78
};


79
static void logreport(int priority, const char *err, va_list params)
P
Petr Baudis 已提交
80 81 82 83 84 85 86 87
{
	/* We should do a single write so that it is atomic and output
	 * of several processes do not get intermingled. */
	char buf[1024];
	int buflen;
	int maxlen, msglen;

	/* sizeof(buf) should be big enough for "[pid] \n" */
J
Junio C Hamano 已提交
88
	buflen = snprintf(buf, sizeof(buf), "[%ld] ", (long) getpid());
P
Petr Baudis 已提交
89 90 91 92

	maxlen = sizeof(buf) - buflen - 1; /* -1 for our own LF */
	msglen = vsnprintf(buf + buflen, maxlen, err, params);

93 94 95 96 97
	if (log_syslog) {
		syslog(priority, "%s", buf);
		return;
	}

P
Petr Baudis 已提交
98 99 100 101 102 103 104 105 106 107 108 109 110 111
	/* maxlen counted our own LF but also counts space given to
	 * vsnprintf for the terminating NUL.  We want to make sure that
	 * we have space for our own LF and NUL after the "meat" of the
	 * message, so truncate it at maxlen - 1.
	 */
	if (msglen > maxlen - 1)
		msglen = maxlen - 1;
	else if (msglen < 0)
		msglen = 0; /* Protect against weird return values. */
	buflen += msglen;

	buf[buflen++] = '\n';
	buf[buflen] = '\0';

112
	write_in_full(2, buf, buflen);
P
Petr Baudis 已提交
113 114
}

115
static void logerror(const char *err, ...)
P
Petr Baudis 已提交
116 117 118
{
	va_list params;
	va_start(params, err);
119
	logreport(LOG_ERR, err, params);
P
Petr Baudis 已提交
120 121 122
	va_end(params);
}

123
static void loginfo(const char *err, ...)
P
Petr Baudis 已提交
124 125 126 127 128
{
	va_list params;
	if (!verbose)
		return;
	va_start(params, err);
129
	logreport(LOG_INFO, err, params);
P
Petr Baudis 已提交
130 131
	va_end(params);
}
132

133 134 135 136 137 138
static void NORETURN daemon_die(const char *err, va_list params)
{
	logreport(LOG_ERR, err, params);
	exit(1);
}

139 140 141 142
static int avoid_alias(char *p)
{
	int sl, ndot;

J
Junio C Hamano 已提交
143
	/*
144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185
	 * This resurrects the belts and suspenders paranoia check by HPA
	 * done in <435560F7.4080006@zytor.com> thread, now enter_repo()
	 * does not do getcwd() based path canonicalizations.
	 *
	 * sl becomes true immediately after seeing '/' and continues to
	 * be true as long as dots continue after that without intervening
	 * non-dot character.
	 */
	if (!p || (*p != '/' && *p != '~'))
		return -1;
	sl = 1; ndot = 0;
	p++;

	while (1) {
		char ch = *p++;
		if (sl) {
			if (ch == '.')
				ndot++;
			else if (ch == '/') {
				if (ndot < 3)
					/* reject //, /./ and /../ */
					return -1;
				ndot = 0;
			}
			else if (ch == 0) {
				if (0 < ndot && ndot < 3)
					/* reject /.$ and /..$ */
					return -1;
				return 0;
			}
			else
				sl = ndot = 0;
		}
		else if (ch == 0)
			return 0;
		else if (ch == '/') {
			sl = 1;
			ndot = 0;
		}
	}
}

186
static char *path_ok(struct interp *itable)
187
{
188
	static char rpath[PATH_MAX];
189
	static char interp_path[PATH_MAX];
J
Jens Axboe 已提交
190
	int retried_path = 0;
191
	char *path;
192 193 194
	char *dir;

	dir = itable[INTERP_SLOT_DIR].value;
195 196 197 198 199 200

	if (avoid_alias(dir)) {
		logerror("'%s': aliased", dir);
		return NULL;
	}

201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222
	if (*dir == '~') {
		if (!user_path) {
			logerror("'%s': User-path not allowed", dir);
			return NULL;
		}
		if (*user_path) {
			/* Got either "~alice" or "~alice/foo";
			 * rewrite them to "~alice/%s" or
			 * "~alice/%s/foo".
			 */
			int namlen, restlen = strlen(dir);
			char *slash = strchr(dir, '/');
			if (!slash)
				slash = dir + restlen;
			namlen = slash - dir;
			restlen -= namlen;
			loginfo("userpath <%s>, request <%s>, namlen %d, restlen %d, slash <%s>", user_path, dir, namlen, restlen, slash);
			snprintf(rpath, PATH_MAX, "%.*s/%s%.*s",
				 namlen, dir, user_path, restlen, slash);
			dir = rpath;
		}
	}
223 224 225 226 227 228 229 230 231 232 233 234 235
	else if (interpolated_path && saw_extended_args) {
		if (*dir != '/') {
			/* Allow only absolute */
			logerror("'%s': Non-absolute path denied (interpolated-path active)", dir);
			return NULL;
		}

		interpolate(interp_path, PATH_MAX, interpolated_path,
			    interp_table, ARRAY_SIZE(interp_table));
		loginfo("Interpolated dir '%s'", interp_path);

		dir = interp_path;
	}
236 237 238
	else if (base_path) {
		if (*dir != '/') {
			/* Allow only absolute */
239
			logerror("'%s': Non-absolute path denied (base-path active)", dir);
P
Petr Baudis 已提交
240 241
			return NULL;
		}
242 243
		snprintf(rpath, PATH_MAX, "%s%s", base_path, dir);
		dir = rpath;
P
Petr Baudis 已提交
244 245
	}

J
Jens Axboe 已提交
246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261
	do {
		path = enter_repo(dir, strict_paths);
		if (path)
			break;

		/*
		 * if we fail and base_path_relaxed is enabled, try without
		 * prefixing the base path
		 */
		if (base_path && base_path_relaxed && !retried_path) {
			dir = itable[INTERP_SLOT_DIR].value;
			retried_path = 1;
			continue;
		}
		break;
	} while (1);
262

263 264 265
	if (!path) {
		logerror("'%s': unable to chdir or not a git archive", dir);
		return NULL;
266 267 268
	}

	if ( ok_paths && *ok_paths ) {
J
Junio C Hamano 已提交
269
		char **pp;
270
		int pathlen = strlen(path);
271

J
Junio C Hamano 已提交
272
		/* The validation is done on the paths after enter_repo
J
Junio C Hamano 已提交
273
		 * appends optional {.git,.git/.git} and friends, but
274 275 276 277
		 * it does not use getcwd().  So if your /pub is
		 * a symlink to /mnt/pub, you can whitelist /pub and
		 * do not have to say /mnt/pub.
		 * Do not say /pub/.
J
Junio C Hamano 已提交
278
		 */
279 280
		for ( pp = ok_paths ; *pp ; pp++ ) {
			int len = strlen(*pp);
J
Junio C Hamano 已提交
281 282 283 284 285
			if (len <= pathlen &&
			    !memcmp(*pp, path, len) &&
			    (path[len] == '\0' ||
			     (!strict_paths && path[len] == '/')))
				return path;
286
		}
287 288 289 290 291
	}
	else {
		/* be backwards compatible */
		if (!strict_paths)
			return path;
292 293
	}

294 295
	logerror("'%s': not in whitelist", path);
	return NULL;		/* Fallthrough. Deny by default */
296
}
297

298 299 300 301 302 303 304 305 306 307 308 309
typedef int (*daemon_service_fn)(void);
struct daemon_service {
	const char *name;
	const char *config_name;
	daemon_service_fn fn;
	int enabled;
	int overridable;
};

static struct daemon_service *service_looking_at;
static int service_enabled;

310
static int git_daemon_config(const char *var, const char *value, void *cb)
311
{
312
	if (!prefixcmp(var, "daemon.") &&
313 314 315 316 317 318 319 320 321
	    !strcmp(var + 7, service_looking_at->config_name)) {
		service_enabled = git_config_bool(var, value);
		return 0;
	}

	/* we are not interested in parsing any other configuration here */
	return 0;
}

322
static int run_service(struct interp *itable, struct daemon_service *service)
323
{
324
	const char *path;
325 326
	int enabled = service->enabled;

327 328 329
	loginfo("Request %s for '%s'",
		service->name,
		itable[INTERP_SLOT_DIR].value);
330

331 332 333 334 335
	if (!enabled && !service->overridable) {
		logerror("'%s': service not enabled.", service->name);
		errno = EACCES;
		return -1;
	}
336

337
	if (!(path = path_ok(itable)))
338
		return -1;
H
H. Peter Anvin 已提交
339

340 341 342
	/*
	 * Security on the cheap.
	 *
343
	 * We want a readable HEAD, usable "objects" directory, and
344 345
	 * a "git-daemon-export-ok" flag that says that the other side
	 * is ok with us doing this.
346 347 348
	 *
	 * path_ok() uses enter_repo() and does whitelist checking.
	 * We only need to make sure the repository is exported.
349
	 */
350

351
	if (!export_all_trees && access("git-daemon-export-ok", F_OK)) {
352
		logerror("'%s': repository not exported.", path);
353 354 355 356
		errno = EACCES;
		return -1;
	}

357 358 359
	if (service->overridable) {
		service_looking_at = service;
		service_enabled = -1;
360
		git_config(git_daemon_config, NULL);
361 362 363 364 365 366 367 368 369 370
		if (0 <= service_enabled)
			enabled = service_enabled;
	}
	if (!enabled) {
		logerror("'%s': service not enabled for '%s'",
			 service->name, path);
		errno = EACCES;
		return -1;
	}

371 372 373 374 375 376
	/*
	 * We'll ignore SIGTERM from now on, we have a
	 * good client.
	 */
	signal(SIGTERM, SIG_IGN);

377 378 379 380 381 382 383 384
	return service->fn();
}

static int upload_pack(void)
{
	/* Timeout as string */
	char timeout_buf[64];

385 386
	snprintf(timeout_buf, sizeof timeout_buf, "--timeout=%u", timeout);

387
	/* git-upload-pack only ever reads stuff, so this is safe */
388
	execl_git_cmd("upload-pack", "--strict", timeout_buf, ".", NULL);
389 390 391
	return -1;
}

F
Franck Bui-Huu 已提交
392 393 394 395 396 397
static int upload_archive(void)
{
	execl_git_cmd("upload-archive", ".", NULL);
	return -1;
}

L
Linus Torvalds 已提交
398 399 400 401 402 403
static int receive_pack(void)
{
	execl_git_cmd("receive-pack", ".", NULL);
	return -1;
}

404
static struct daemon_service daemon_service[] = {
F
Franck Bui-Huu 已提交
405
	{ "upload-archive", "uploadarch", upload_archive, 0, 1 },
406
	{ "upload-pack", "uploadpack", upload_pack, 1, 1 },
L
Linus Torvalds 已提交
407
	{ "receive-pack", "receivepack", receive_pack, 0, 1 },
408 409
};

410 411
static void enable_service(const char *name, int ena)
{
412 413 414 415 416 417 418 419 420 421
	int i;
	for (i = 0; i < ARRAY_SIZE(daemon_service); i++) {
		if (!strcmp(daemon_service[i].name, name)) {
			daemon_service[i].enabled = ena;
			return;
		}
	}
	die("No such service %s", name);
}

422 423
static void make_service_overridable(const char *name, int ena)
{
424 425 426 427 428 429 430 431 432 433
	int i;
	for (i = 0; i < ARRAY_SIZE(daemon_service); i++) {
		if (!strcmp(daemon_service[i].name, name)) {
			daemon_service[i].overridable = ena;
			return;
		}
	}
	die("No such service %s", name);
}

434 435
/*
 * Separate the "extra args" information as supplied by the client connection.
P
Pavel Roskin 已提交
436
 * Any resulting data is squirreled away in the given interpolation table.
437 438
 */
static void parse_extra_args(struct interp *table, char *extra_args, int buflen)
439 440 441 442 443 444 445 446 447 448 449
{
	char *val;
	int vallen;
	char *end = extra_args + buflen;

	while (extra_args < end && *extra_args) {
		saw_extended_args = 1;
		if (strncasecmp("host=", extra_args, 5) == 0) {
			val = extra_args + 5;
			vallen = strlen(val) + 1;
			if (*val) {
450 451 452
				/* Split <host>:<port> at colon. */
				char *host = val;
				char *port = strrchr(host, ':');
453 454 455
				if (port) {
					*port = 0;
					port++;
456
					interp_set_entry(table, INTERP_SLOT_PORT, port);
457
				}
458
				interp_set_entry(table, INTERP_SLOT_HOST, host);
459
			}
460

461 462 463 464 465 466
			/* On to the next one */
			extra_args = val + vallen;
		}
	}
}

P
Pierre Habouzit 已提交
467
static void fill_in_extra_table_entries(struct interp *itable)
468 469 470 471 472 473 474
{
	char *hp;

	/*
	 * Replace literal host with lowercase-ized hostname.
	 */
	hp = interp_table[INTERP_SLOT_HOST].value;
475 476
	if (!hp)
		return;
477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499
	for ( ; *hp; hp++)
		*hp = tolower(*hp);

	/*
	 * Locate canonical hostname and its IP address.
	 */
#ifndef NO_IPV6
	{
		struct addrinfo hints;
		struct addrinfo *ai, *ai0;
		int gai;
		static char addrbuf[HOST_NAME_MAX + 1];

		memset(&hints, 0, sizeof(hints));
		hints.ai_flags = AI_CANONNAME;

		gai = getaddrinfo(interp_table[INTERP_SLOT_HOST].value, 0, &hints, &ai0);
		if (!gai) {
			for (ai = ai0; ai; ai = ai->ai_next) {
				struct sockaddr_in *sin_addr = (void *)ai->ai_addr;

				inet_ntop(AF_INET, &sin_addr->sin_addr,
					  addrbuf, sizeof(addrbuf));
500 501 502 503
				interp_set_entry(interp_table,
						 INTERP_SLOT_CANON_HOST, ai->ai_canonname);
				interp_set_entry(interp_table,
						 INTERP_SLOT_IP, addrbuf);
504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525
				break;
			}
			freeaddrinfo(ai0);
		}
	}
#else
	{
		struct hostent *hent;
		struct sockaddr_in sa;
		char **ap;
		static char addrbuf[HOST_NAME_MAX + 1];

		hent = gethostbyname(interp_table[INTERP_SLOT_HOST].value);

		ap = hent->h_addr_list;
		memset(&sa, 0, sizeof sa);
		sa.sin_family = hent->h_addrtype;
		sa.sin_port = htons(0);
		memcpy(&sa.sin_addr, *ap, hent->h_length);

		inet_ntop(hent->h_addrtype, &sa.sin_addr,
			  addrbuf, sizeof(addrbuf));
526 527 528

		interp_set_entry(interp_table, INTERP_SLOT_CANON_HOST, hent->h_name);
		interp_set_entry(interp_table, INTERP_SLOT_IP, addrbuf);
529 530 531 532 533
	}
#endif
}


534
static int execute(struct sockaddr *addr)
535
{
536
	static char line[1000];
537
	int pktlen, len, i;
538

539 540 541 542 543 544 545
	if (addr) {
		char addrbuf[256] = "";
		int port = -1;

		if (addr->sa_family == AF_INET) {
			struct sockaddr_in *sin_addr = (void *) addr;
			inet_ntop(addr->sa_family, &sin_addr->sin_addr, addrbuf, sizeof(addrbuf));
546
			port = ntohs(sin_addr->sin_port);
547 548 549 550 551 552 553 554 555
#ifndef NO_IPV6
		} else if (addr && addr->sa_family == AF_INET6) {
			struct sockaddr_in6 *sin6_addr = (void *) addr;

			char *buf = addrbuf;
			*buf++ = '['; *buf = '\0'; /* stpcpy() is cool */
			inet_ntop(AF_INET6, &sin6_addr->sin6_addr, buf, sizeof(addrbuf) - 1);
			strcat(buf, "]");

556
			port = ntohs(sin6_addr->sin6_port);
557 558 559 560 561
#endif
		}
		loginfo("Connection from %s:%d", addrbuf, port);
	}

562
	alarm(init_timeout ? init_timeout : timeout);
563
	pktlen = packet_read_line(0, line, sizeof(line));
564
	alarm(0);
565

566 567 568 569 570
	len = strlen(line);
	if (pktlen != len)
		loginfo("Extended attributes (%d bytes) exist <%.*s>",
			(int) pktlen - len,
			(int) pktlen - len, line + len + 1);
571
	if (len && line[len-1] == '\n') {
572
		line[--len] = 0;
573 574
		pktlen--;
	}
575

576 577 578 579 580 581
	/*
	 * Initialize the path interpolation table for this connection.
	 */
	interp_clear_table(interp_table, ARRAY_SIZE(interp_table));
	interp_set_entry(interp_table, INTERP_SLOT_PERCENT, "%");

582
	if (len != pktlen) {
583
	    parse_extra_args(interp_table, line + len + 1, pktlen - len - 1);
584 585
	    fill_in_extra_table_entries(interp_table);
	}
586

587 588 589
	for (i = 0; i < ARRAY_SIZE(daemon_service); i++) {
		struct daemon_service *s = &(daemon_service[i]);
		int namelen = strlen(s->name);
590
		if (!prefixcmp(line, "git-") &&
591
		    !strncmp(s->name, line + 4, namelen) &&
592
		    line[namelen + 4] == ' ') {
593 594 595 596 597 598
			/*
			 * Note: The directory here is probably context sensitive,
			 * and might depend on the actual service being performed.
			 */
			interp_set_entry(interp_table,
					 INTERP_SLOT_DIR, line + namelen + 5);
599 600
			return run_service(interp_table, s);
		}
601
	}
602

P
Petr Baudis 已提交
603
	logerror("Protocol error: '%s'", line);
604 605 606
	return -1;
}

607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622

/*
 * We count spawned/reaped separately, just to avoid any
 * races when updating them from signals. The SIGCHLD handler
 * will only update children_reaped, and the fork logic will
 * only update children_spawned.
 *
 * MAX_CHILDREN should be a power-of-two to make the modulus
 * operation cheap. It should also be at least twice
 * the maximum number of connections we will ever allow.
 */
#define MAX_CHILDREN 128

static int max_connections = 25;

/* These are updated by the signal handler */
623
static volatile unsigned int children_reaped;
L
Linus Torvalds 已提交
624
static pid_t dead_child[MAX_CHILDREN];
625 626

/* These are updated by the main loop */
627 628
static unsigned int children_spawned;
static unsigned int children_deleted;
629

L
Linus Torvalds 已提交
630
static struct child {
631
	pid_t pid;
J
Junio C Hamano 已提交
632
	int addrlen;
633
	struct sockaddr_storage address;
634 635
} live_child[MAX_CHILDREN];

J
Junio C Hamano 已提交
636
static void add_child(int idx, pid_t pid, struct sockaddr *addr, int addrlen)
637 638 639
{
	live_child[idx].pid = pid;
	live_child[idx].addrlen = addrlen;
640
	memcpy(&live_child[idx].address, addr, addrlen);
641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686
}

/*
 * Walk from "deleted" to "spawned", and remove child "pid".
 *
 * We move everything up by one, since the new "deleted" will
 * be one higher.
 */
static void remove_child(pid_t pid, unsigned deleted, unsigned spawned)
{
	struct child n;

	deleted %= MAX_CHILDREN;
	spawned %= MAX_CHILDREN;
	if (live_child[deleted].pid == pid) {
		live_child[deleted].pid = -1;
		return;
	}
	n = live_child[deleted];
	for (;;) {
		struct child m;
		deleted = (deleted + 1) % MAX_CHILDREN;
		if (deleted == spawned)
			die("could not find dead child %d\n", pid);
		m = live_child[deleted];
		live_child[deleted] = n;
		if (m.pid == pid)
			return;
		n = m;
	}
}

/*
 * This gets called if the number of connections grows
 * past "max_connections".
 *
 * We _should_ start off by searching for connections
 * from the same IP, and if there is some address wth
 * multiple connections, we should kill that first.
 *
 * As it is, we just "randomly" kill 25% of the connections,
 * and our pseudo-random generator sucks too. I have no
 * shame.
 *
 * Really, this is just a place-holder for a _real_ algorithm.
 */
687
static void kill_some_children(int signo, unsigned start, unsigned stop)
688 689 690 691 692
{
	start %= MAX_CHILDREN;
	stop %= MAX_CHILDREN;
	while (start != stop) {
		if (!(start & 3))
693
			kill(live_child[start].pid, signo);
694 695 696 697
		start = (start + 1) % MAX_CHILDREN;
	}
}

698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727
static void check_dead_children(void)
{
	unsigned spawned, reaped, deleted;

	spawned = children_spawned;
	reaped = children_reaped;
	deleted = children_deleted;

	while (deleted < reaped) {
		pid_t pid = dead_child[deleted % MAX_CHILDREN];
		const char *dead = pid < 0 ? " (with error)" : "";

		if (pid < 0)
			pid = -pid;

		/* XXX: Custom logging, since we don't wanna getpid() */
		if (verbose) {
			if (log_syslog)
				syslog(LOG_INFO, "[%d] Disconnected%s",
						pid, dead);
			else
				fprintf(stderr, "[%d] Disconnected%s\n",
						pid, dead);
		}
		remove_child(pid, deleted, spawned);
		deleted++;
	}
	children_deleted = deleted;
}

728
static void check_max_connections(void)
729
{
730
	for (;;) {
731
		int active;
732 733 734
		unsigned spawned, deleted;

		check_dead_children();
735

736 737 738 739
		spawned = children_spawned;
		deleted = children_deleted;

		active = spawned - deleted;
740 741
		if (active <= max_connections)
			break;
742

743 744 745 746 747 748 749 750 751 752 753
		/* Kill some unstarted connections with SIGTERM */
		kill_some_children(SIGTERM, deleted, spawned);
		if (active <= max_connections << 1)
			break;

		/* If the SIGTERM thing isn't helping use SIGKILL */
		kill_some_children(SIGKILL, deleted, spawned);
		sleep(1);
	}
}

J
Junio C Hamano 已提交
754
static void handle(int incoming, struct sockaddr *addr, int addrlen)
755 756 757 758 759 760 761 762 763 764 765 766 767
{
	pid_t pid = fork();

	if (pid) {
		unsigned idx;

		close(incoming);
		if (pid < 0)
			return;

		idx = children_spawned % MAX_CHILDREN;
		children_spawned++;
		add_child(idx, pid, addr, addrlen);
768

769
		check_max_connections();
770 771 772 773 774 775
		return;
	}

	dup2(incoming, 0);
	dup2(incoming, 1);
	close(incoming);
P
Petr Baudis 已提交
776

777
	exit(execute(addr));
778 779
}

780 781 782
static void child_handler(int signo)
{
	for (;;) {
783 784
		int status;
		pid_t pid = waitpid(-1, &status, WNOHANG);
785 786 787

		if (pid > 0) {
			unsigned reaped = children_reaped;
788 789
			if (!WIFEXITED(status) || WEXITSTATUS(status) > 0)
				pid = -pid;
790 791
			dead_child[reaped % MAX_CHILDREN] = pid;
			children_reaped = reaped + 1;
792
			write(child_handler_pipe[1], &status, 1);
793 794 795 796 797 798
			continue;
		}
		break;
	}
}

799 800 801 802 803 804 805 806 807 808
static int set_reuse_addr(int sockfd)
{
	int on = 1;

	if (!reuseaddr)
		return 0;
	return setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR,
			  &on, sizeof(on));
}

P
Peter Anvin 已提交
809 810
#ifndef NO_IPV6

811
static int socksetup(char *listen_addr, int listen_port, int **socklist_p)
812
{
813 814 815
	int socknum = 0, *socklist = NULL;
	int maxfd = -1;
	char pbuf[NI_MAXSERV];
P
Peter Anvin 已提交
816 817
	struct addrinfo hints, *ai0, *ai;
	int gai;
818
	long flags;
819

820
	sprintf(pbuf, "%d", listen_port);
821 822 823 824 825 826
	memset(&hints, 0, sizeof(hints));
	hints.ai_family = AF_UNSPEC;
	hints.ai_socktype = SOCK_STREAM;
	hints.ai_protocol = IPPROTO_TCP;
	hints.ai_flags = AI_PASSIVE;

827
	gai = getaddrinfo(listen_addr, pbuf, &hints, &ai0);
828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851
	if (gai)
		die("getaddrinfo() failed: %s\n", gai_strerror(gai));

	for (ai = ai0; ai; ai = ai->ai_next) {
		int sockfd;

		sockfd = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
		if (sockfd < 0)
			continue;
		if (sockfd >= FD_SETSIZE) {
			error("too large socket descriptor.");
			close(sockfd);
			continue;
		}

#ifdef IPV6_V6ONLY
		if (ai->ai_family == AF_INET6) {
			int on = 1;
			setsockopt(sockfd, IPPROTO_IPV6, IPV6_V6ONLY,
				   &on, sizeof(on));
			/* Note: error is not fatal */
		}
#endif

852 853
		if (set_reuse_addr(sockfd)) {
			close(sockfd);
854
			continue;
855 856
		}

857 858 859 860 861 862 863 864 865
		if (bind(sockfd, ai->ai_addr, ai->ai_addrlen) < 0) {
			close(sockfd);
			continue;	/* not fatal */
		}
		if (listen(sockfd, 5) < 0) {
			close(sockfd);
			continue;	/* not fatal */
		}

866 867 868 869
		flags = fcntl(sockfd, F_GETFD, 0);
		if (flags >= 0)
			fcntl(sockfd, F_SETFD, flags | FD_CLOEXEC);

J
Jonas Fonseca 已提交
870
		socklist = xrealloc(socklist, sizeof(int) * (socknum + 1));
871 872 873 874 875 876 877 878
		socklist[socknum++] = sockfd;

		if (maxfd < sockfd)
			maxfd = sockfd;
	}

	freeaddrinfo(ai0);

P
Peter Anvin 已提交
879 880 881 882 883 884
	*socklist_p = socklist;
	return socknum;
}

#else /* NO_IPV6 */

885
static int socksetup(char *listen_addr, int listen_port, int **socklist_p)
P
Peter Anvin 已提交
886 887 888
{
	struct sockaddr_in sin;
	int sockfd;
889
	long flags;
P
Peter Anvin 已提交
890

891 892 893 894 895 896 897 898 899 900 901 902
	memset(&sin, 0, sizeof sin);
	sin.sin_family = AF_INET;
	sin.sin_port = htons(listen_port);

	if (listen_addr) {
		/* Well, host better be an IP address here. */
		if (inet_pton(AF_INET, listen_addr, &sin.sin_addr.s_addr) <= 0)
			return 0;
	} else {
		sin.sin_addr.s_addr = htonl(INADDR_ANY);
	}

P
Peter Anvin 已提交
903 904 905 906
	sockfd = socket(AF_INET, SOCK_STREAM, 0);
	if (sockfd < 0)
		return 0;

907 908 909 910 911
	if (set_reuse_addr(sockfd)) {
		close(sockfd);
		return 0;
	}

P
Peter Anvin 已提交
912 913 914 915
	if ( bind(sockfd, (struct sockaddr *)&sin, sizeof sin) < 0 ) {
		close(sockfd);
		return 0;
	}
916

917 918 919 920 921
	if (listen(sockfd, 5) < 0) {
		close(sockfd);
		return 0;
	}

922 923 924 925
	flags = fcntl(sockfd, F_GETFD, 0);
	if (flags >= 0)
		fcntl(sockfd, F_SETFD, flags | FD_CLOEXEC);

H
H. Peter Anvin 已提交
926
	*socklist_p = xmalloc(sizeof(int));
P
Peter Anvin 已提交
927
	**socklist_p = sockfd;
928
	return 1;
P
Peter Anvin 已提交
929 930 931 932 933 934 935 936 937
}

#endif

static int service_loop(int socknum, int *socklist)
{
	struct pollfd *pfd;
	int i;

938 939 940 941
	if (pipe(child_handler_pipe) < 0)
		die ("Could not set up pipe for child handler");

	pfd = xcalloc(socknum + 1, sizeof(struct pollfd));
P
Peter Anvin 已提交
942 943 944 945 946

	for (i = 0; i < socknum; i++) {
		pfd[i].fd = socklist[i];
		pfd[i].events = POLLIN;
	}
947 948
	pfd[socknum].fd = child_handler_pipe[0];
	pfd[socknum].events = POLLIN;
949 950

	signal(SIGCHLD, child_handler);
951 952

	for (;;) {
953
		int i;
P
Peter Anvin 已提交
954

955
		if (poll(pfd, socknum + 1, -1) < 0) {
956
			if (errno != EINTR) {
P
Peter Anvin 已提交
957
				error("poll failed, resuming: %s",
958 959 960
				      strerror(errno));
				sleep(1);
			}
961 962
			continue;
		}
963 964
		if (pfd[socknum].revents & POLLIN) {
			read(child_handler_pipe[0], &i, 1);
965 966
			check_dead_children();
		}
967 968

		for (i = 0; i < socknum; i++) {
P
Peter Anvin 已提交
969
			if (pfd[i].revents & POLLIN) {
970
				struct sockaddr_storage ss;
971
				unsigned int sslen = sizeof(ss);
P
Peter Anvin 已提交
972
				int incoming = accept(pfd[i].fd, (struct sockaddr *)&ss, &sslen);
973 974 975 976 977 978 979 980 981 982 983
				if (incoming < 0) {
					switch (errno) {
					case EAGAIN:
					case EINTR:
					case ECONNABORTED:
						continue;
					default:
						die("accept returned %s", strerror(errno));
					}
				}
				handle(incoming, (struct sockaddr *)&ss, sslen);
984 985 986 987 988
			}
		}
	}
}

989 990 991 992 993 994 995 996 997 998 999 1000
/* if any standard file descriptor is missing open it to /dev/null */
static void sanitize_stdfds(void)
{
	int fd = open("/dev/null", O_RDWR, 0);
	while (fd != -1 && fd < 2)
		fd = dup(fd);
	if (fd == -1)
		die("open /dev/null or dup failed: %s", strerror(errno));
	if (fd > 2)
		close(fd);
}

1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018
static void daemonize(void)
{
	switch (fork()) {
		case 0:
			break;
		case -1:
			die("fork failed: %s", strerror(errno));
		default:
			exit(0);
	}
	if (setsid() == -1)
		die("setsid failed: %s", strerror(errno));
	close(0);
	close(1);
	close(2);
	sanitize_stdfds();
}

1019 1020 1021 1022 1023
static void store_pid(const char *path)
{
	FILE *f = fopen(path, "w");
	if (!f)
		die("cannot open pid file %s: %s", path, strerror(errno));
1024 1025
	if (fprintf(f, "%d\n", getpid()) < 0 || fclose(f) != 0)
		die("failed to write pid file %s: %s", path, strerror(errno));
1026 1027
}

1028
static int serve(char *listen_addr, int listen_port, struct passwd *pass, gid_t gid)
P
Peter Anvin 已提交
1029 1030
{
	int socknum, *socklist;
J
Junio C Hamano 已提交
1031

1032
	socknum = socksetup(listen_addr, listen_port, &socklist);
P
Peter Anvin 已提交
1033
	if (socknum == 0)
1034 1035
		die("unable to allocate any listen sockets on host %s port %u",
		    listen_addr, listen_port);
J
Junio C Hamano 已提交
1036

1037 1038 1039 1040 1041
	if (pass && gid &&
	    (initgroups(pass->pw_name, gid) || setgid (gid) ||
	     setuid(pass->pw_uid)))
		die("cannot drop privileges");

P
Peter Anvin 已提交
1042
	return service_loop(socknum, socklist);
J
Junio C Hamano 已提交
1043
}
P
Peter Anvin 已提交
1044

1045 1046
int main(int argc, char **argv)
{
1047 1048
	int listen_port = 0;
	char *listen_addr = NULL;
1049
	int inetd_mode = 0;
1050
	const char *pid_file = NULL, *user_name = NULL, *group_name = NULL;
1051
	int detach = 0;
1052 1053 1054
	struct passwd *pass = NULL;
	struct group *group;
	gid_t gid = 0;
1055 1056
	int i;

1057 1058 1059 1060 1061
	/* Without this we cannot rely on waitpid() to tell
	 * what happened to our children.
	 */
	signal(SIGCHLD, SIG_DFL);

1062 1063 1064
	for (i = 1; i < argc; i++) {
		char *arg = argv[i];

1065
		if (!prefixcmp(arg, "--listen=")) {
1066 1067 1068 1069 1070 1071 1072
		    char *p = arg + 9;
		    char *ph = listen_addr = xmalloc(strlen(arg + 9) + 1);
		    while (*p)
			*ph++ = tolower(*p++);
		    *ph = 0;
		    continue;
		}
1073
		if (!prefixcmp(arg, "--port=")) {
1074 1075 1076 1077
			char *end;
			unsigned long n;
			n = strtoul(arg+7, &end, 0);
			if (arg[7] && !*end) {
1078
				listen_port = n;
1079 1080 1081
				continue;
			}
		}
1082 1083
		if (!strcmp(arg, "--inetd")) {
			inetd_mode = 1;
A
Andreas Ericsson 已提交
1084
			log_syslog = 1;
1085 1086
			continue;
		}
P
Petr Baudis 已提交
1087 1088 1089 1090
		if (!strcmp(arg, "--verbose")) {
			verbose = 1;
			continue;
		}
1091 1092 1093 1094
		if (!strcmp(arg, "--syslog")) {
			log_syslog = 1;
			continue;
		}
1095 1096 1097 1098
		if (!strcmp(arg, "--export-all")) {
			export_all_trees = 1;
			continue;
		}
1099
		if (!prefixcmp(arg, "--timeout=")) {
1100
			timeout = atoi(arg+10);
A
Andreas Ericsson 已提交
1101
			continue;
1102
		}
1103
		if (!prefixcmp(arg, "--init-timeout=")) {
1104
			init_timeout = atoi(arg+15);
A
Andreas Ericsson 已提交
1105
			continue;
1106
		}
1107 1108 1109 1110
		if (!strcmp(arg, "--strict-paths")) {
			strict_paths = 1;
			continue;
		}
1111
		if (!prefixcmp(arg, "--base-path=")) {
P
Petr Baudis 已提交
1112 1113 1114
			base_path = arg+12;
			continue;
		}
J
Jens Axboe 已提交
1115 1116 1117 1118
		if (!strcmp(arg, "--base-path-relaxed")) {
			base_path_relaxed = 1;
			continue;
		}
1119
		if (!prefixcmp(arg, "--interpolated-path=")) {
1120 1121 1122
			interpolated_path = arg+20;
			continue;
		}
1123 1124 1125 1126
		if (!strcmp(arg, "--reuseaddr")) {
			reuseaddr = 1;
			continue;
		}
1127 1128 1129 1130
		if (!strcmp(arg, "--user-path")) {
			user_path = "";
			continue;
		}
1131
		if (!prefixcmp(arg, "--user-path=")) {
1132 1133 1134
			user_path = arg + 12;
			continue;
		}
1135
		if (!prefixcmp(arg, "--pid-file=")) {
1136 1137 1138
			pid_file = arg + 11;
			continue;
		}
1139 1140 1141 1142 1143
		if (!strcmp(arg, "--detach")) {
			detach = 1;
			log_syslog = 1;
			continue;
		}
1144
		if (!prefixcmp(arg, "--user=")) {
1145 1146 1147
			user_name = arg + 7;
			continue;
		}
1148
		if (!prefixcmp(arg, "--group=")) {
1149 1150 1151
			group_name = arg + 8;
			continue;
		}
1152
		if (!prefixcmp(arg, "--enable=")) {
1153 1154 1155
			enable_service(arg + 9, 1);
			continue;
		}
1156
		if (!prefixcmp(arg, "--disable=")) {
1157 1158 1159
			enable_service(arg + 10, 0);
			continue;
		}
1160
		if (!prefixcmp(arg, "--allow-override=")) {
J
Junio C Hamano 已提交
1161
			make_service_overridable(arg + 17, 1);
1162 1163
			continue;
		}
1164
		if (!prefixcmp(arg, "--forbid-override=")) {
J
Junio C Hamano 已提交
1165
			make_service_overridable(arg + 18, 0);
1166 1167
			continue;
		}
1168 1169 1170 1171 1172 1173 1174
		if (!strcmp(arg, "--")) {
			ok_paths = &argv[i+1];
			break;
		} else if (arg[0] != '-') {
			ok_paths = &argv[i];
			break;
		}
1175

1176 1177 1178
		usage(daemon_usage);
	}

1179 1180 1181 1182 1183
	if (log_syslog) {
		openlog("git-daemon", 0, LOG_DAEMON);
		set_die_routine(daemon_die);
	}

1184 1185 1186
	if (inetd_mode && (group_name || user_name))
		die("--user and --group are incompatible with --inetd");

1187 1188 1189 1190 1191
	if (inetd_mode && (listen_port || listen_addr))
		die("--listen= and --port= are incompatible with --inetd");
	else if (listen_port == 0)
		listen_port = DEFAULT_GIT_PORT;

1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210
	if (group_name && !user_name)
		die("--group supplied without --user");

	if (user_name) {
		pass = getpwnam(user_name);
		if (!pass)
			die("user not found - %s", user_name);

		if (!group_name)
			gid = pass->pw_gid;
		else {
			group = getgrnam(group_name);
			if (!group)
				die("group not found - %s", group_name);

			gid = group->gr_gid;
		}
	}

1211 1212 1213
	if (strict_paths && (!ok_paths || !*ok_paths))
		die("option --strict-paths requires a whitelist");

1214 1215 1216 1217 1218 1219 1220 1221
	if (base_path) {
		struct stat st;

		if (stat(base_path, &st) || !S_ISDIR(st.st_mode))
			die("base-path '%s' does not exist or "
			    "is not a directory", base_path);
	}

1222
	if (inetd_mode) {
1223 1224 1225 1226
		struct sockaddr_storage ss;
		struct sockaddr *peer = (struct sockaddr *)&ss;
		socklen_t slen = sizeof(ss);

1227
		freopen("/dev/null", "w", stderr);
1228 1229 1230 1231 1232

		if (getpeername(0, peer, &slen))
			peer = NULL;

		return execute(peer);
1233
	}
1234

1235 1236 1237 1238
	if (detach)
		daemonize();
	else
		sanitize_stdfds();
1239

1240 1241 1242
	if (pid_file)
		store_pid(pid_file);

1243
	return serve(listen_addr, listen_port, pass, gid);
1244
}