diff options
Diffstat (limited to 'src/main.c')
| -rw-r--r-- | src/main.c | 82 |
1 files changed, 74 insertions, 8 deletions
@@ -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) |