1 files changed, 74 insertions, 8 deletions

diff --git a/src/main.c b/src/main.c
index c054e31..c14af38 100644
--- a/src/main.c
+++ b/src/main.c
@@ -6,51 +6,117 @@
 #include <error.h>
 #include <fcntl.h>
 #include <signal.h>
+#include <poll.h>
+#include <sys/time.h>
 #define S0(x) (x ? x : "")
 #include <dht.c>
+int samomor = 0;
+int periodično = 0;
+void handler (int s) {
+	switch (s) {
+		case SIGINT:
+		case SIGTERM:
+			samomor++;
+			break;
+		case SIGALRM:
+			periodično++;
+			break;
+	}
+}
 int main (int argc, char ** argv) {
 	int r = 0;
+	struct dht * dht = NULL;
+	struct sigaction sigact = {
+		.sa_handler = handler,
+		.sa_flags = SA_RESTART
+	};
+	if (sigaction(SIGALRM, &sigact, NULL) == -1)
+		error_at_line(1, errno, __FILE__, __LINE__, "sigaction(SIGALRM)");
+	if (sigaction(SIGINT, &sigact, NULL) == -1)
+		error_at_line(2, errno, __FILE__, __LINE__, "sigaction(SIGINT)");
+	if (sigaction(SIGTERM, &sigact, NULL) == -1)
+		error_at_line(3, errno, __FILE__, __LINE__, "sigaction(SIGTERM)");
 	if (argc != 1+1)
-		error_at_line(1, 0, __FILE__, __LINE__, "%s configfile.ben", S0(argv[0]));
+		error_at_line(4, 0, __FILE__, __LINE__, "%s configfile.ben", S0(argv[0]));
 	int cf = open(argv[1], O_RDWR | O_CLOEXEC | O_CREAT, 00664);
 	if (cf == -1)
-		error_at_line(2, errno, __FILE__, __LINE__, "open(%s)", argv[1]);
+		error_at_line(5, errno, __FILE__, __LINE__, "open(%s)", argv[1]);
 	struct stat statbuf;
 	if (fstat(cf, &statbuf) == -1) {
 		error_at_line(0, errno, __FILE__, __LINE__, "fstat(cf, &statbuf)");
-		r = 3;
+		r = 6;
 		goto r;
 	}
 	char * cfr = NULL;
 	if (statbuf.st_size && !(cfr = mmap(NULL, statbuf.st_size, PROT_READ, MAP_SHARED, cf, 0))) {
 		error_at_line(0, errno, __FILE__, __LINE__, "mmap(NULL, %ld, PROT_READ, MAP_SHARED, cf, 0)", statbuf.st_size);
-		r = 4;
+		r = 7;
 		goto r;
 	}
 	struct bencoding * config = bdecode(cfr, statbuf.st_size, replace);
-	struct dht * dht = dht_init(config);
+	dht = dht_init(config);
 	free_bencoding(config);
+	struct torrent * torrent = calloc(1, sizeof *torrent);
+	memcpy(torrent->hash, "\xdd\x82\x55\xec\xdc\x7c\xa5\x5f\xb0\xbb\xf8\x13\x23\xd8\x70\x62\xdb\x1f\x6d\x1c", 20);
+	torrent->type = announce | peers;
+	add_torrent(dht, torrent);
+	work(dht);
+	struct pollfd pollfd = {
+		.fd = dht->socket,
+		.events = POLLIN
+	};
+	struct itimerval itimerval = {
+		.it_interval = {
+			.tv_sec = 13*60
+		}
+	};
+	setitimer(ITIMER_REAL, &itimerval, NULL);
+w:
+	while (poll(&pollfd, 1, -1) == 1) {
+		work(dht);
+	}
+	if (errno == EINTR) {
+		if (periodično) {
+			periodično = 0;
+			work(dht);
+			goto w;
+		}
+		if (!samomor) {
+			error_at_line(0, errno, __FILE__, __LINE__, "poll");
+			r = 108;
+		}
+	} else {
+		error_at_line(0, errno, __FILE__,  __LINE__, "poll");
+		r = 109;
+		goto r;
+	}
 	config = persistent(dht);
 	dht_free(dht);
+	dht = NULL;
 	if (cfr && munmap(cfr, statbuf.st_size) == -1) {
 		error_at_line(0, errno, __FILE__, __LINE__, "munmap(cf, %ld)", statbuf.st_size);
-		r = 105;
+		r = 110;
 		goto r;
 	}
 	cfr = NULL;
 	if (ftruncate(cf, (statbuf.st_size = bencode_length(config))) == -1) {
 		error_at_line(0, errno, __FILE__, __LINE__, "ftruncate(cf, %ld)", statbuf.st_size);
-		r = 106;
+		r = 111;
 		goto r;
 	}
 	if (!(cfr = mmap(NULL, statbuf.st_size, PROT_WRITE, MAP_SHARED, cf, 0))) {
 		error_at_line(0, errno, __FILE__, __LINE__, "mmap(NULL, %ld, PROT_READ, MAP_SHARED, cf, 0)", statbuf.st_size);
-		r = 107;
+		r = 112;
 		goto r;
 	}
 	bencode(cfr, config);
 	free_bencoding(config);
+	config = NULL;
 	r:
+	if (dht) {
+		dht_free(dht);
+		dht = NULL;
+	}
 	if (cfr && munmap(cfr, statbuf.st_size) == -1)
 		error_at_line(0, errno, __FILE__, __LINE__, "munmap(cf, %ld)", statbuf.st_size);
 	if (close(cf) == -1)