torrent-stats/torrent-stats.c

/***************************************************************************
 *   Copyright (C) 07/2007 by Olaf Rempel                                  *
 *   razzor@kopf-tisch.de                                                  *
 *                                                                         *
 *   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; version 2 of the License                *
 *                                                                         *
 *   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 <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#include <getopt.h>
#include <pwd.h>

#include "configfile.h"
#include "connection.h"
#include "event.h"
#include "httpd.h"
#include "list.h"
#include "logging.h"
#include "sockaddr.h"
#include "tcpsocket.h"

#define DEFAULT_CONFIG "torrent-stats.conf"
#define DEFAULT_LOGFILE "torrent-stats.log"

static struct option opts[] = {
	{"config",	1, 0, 'c'},
	{"user",	1, 0, 'u'},
	{"debug",	0, 0, 'd'},
	{"help",	0, 0, 'h'},
	{0, 0, 0, 0}
};

static int listen_cb(const char *parameter, void *privdata)
{
	struct sockaddr_in addr;

	if (parse_sockaddr(parameter, &addr) < 0) {
		log_print(LOG_WARN, "listen_cb(): invalid address");
		return -1;
	}

	int sockfd = tcp_listen(&addr);
	if (sockfd < 0) {
		log_print(LOG_WARN, "listen_cb(): tcp_listen()");
		return -1;
	}

	log_print(LOG_INFO, "listen on %s", get_sockaddr_buf(&addr));
	event_add_readfd(NULL, sockfd, privdata, NULL);
	return 0;
}

int main(int argc, char *argv[])
{
	char *config = DEFAULT_CONFIG;
	char *user = NULL;
	int debug = 0;
	int code, arg = 0;

	do {
		code = getopt_long(argc, argv, "c:u:dh", opts, &arg);

		switch (code) {
		case 'c':	/* config */
				config = optarg;
				break;

		case 'u':	/* user */
				user = optarg;
				break;

		case 'd':	/* debug */
				debug = 1;
				break;

		case 'h':	/* help */
				printf("Usage: torrent-stats [options]\n"
					"Options: \n"
					"  --config       -c  configfile  use this configfile\n"
					"  --debug        -d              do not fork and log to stderr\n"
					"  --user	  -u  username    change uid to username\n"
					"  --help         -h              this help\n"
					"\n");
				exit(0);
				break;

		case '?':	/* error */
				exit(1);
				break;

		default:	/* unknown / all options parsed */
				break;
		}
	} while (code != -1);

	/* userwechsel */
	if (user) {
		struct  passwd *pwl;
		if (!(pwl = getpwnam(user))) {
			log_print(LOG_ERROR, "unknown user: %s", user);
			exit(-1);
		}

		if (setgid(pwl->pw_gid) || setuid(pwl->pw_uid)) {
			log_print(LOG_ERROR, "setgid/setuid");
			exit(-1);
		}
	}

	/* parse config file */
	if (config_parse(config) < 0)
		exit(1);

	/* check logfile */
	const char *logfile = config_get_string("global", "logfile", DEFAULT_LOGFILE);
	if (!debug) {
		/* start logging */
		if (log_init(logfile) < 0)
			exit(-1);

		/* zum daemon mutieren */
		daemon(-1, 0);
	}

	log_print(LOG_INFO, "torrent-stats started (user: %s, pid: %d)", getpwuid(getuid())->pw_name, getpid());

	config_get_strings("global", "listen", listen_cb, ctcs_accept_handler);
	config_get_strings("global", "listen-http", listen_cb, httpd_accept_handler);

	httpd_add_cb("/quit", 0, ctcs_httpd_quit, NULL);
	httpd_add_cb("/", 1, ctcs_httpd_show, NULL);

	int interval = config_get_int("global", "status-interval", 10);
	int timeout = config_get_int("global", "seed-timeout", 300);

	struct timeval tv = { .tv_sec = interval, .tv_usec = 0 };
	event_add_timeout(&tv, ctcs_trigger_status, (void *)timeout);

	event_loop();
	return 0;
}