int
tr_blocklistFileSetContent (tr_blocklistFile * b, const char * filename)
{
tr_sys_file_t in;
tr_sys_file_t out;
int inCount = 0;
char line[2048];
const char * err_fmt = _("Couldn't read \"%1$s\": %2$s");
struct tr_ipv4_range * ranges = NULL;
size_t ranges_alloc = 0;
size_t ranges_count = 0;
tr_error * error = NULL;
if (!filename)
{
blocklistDelete (b);
return 0;
}
in = tr_sys_file_open (filename, TR_SYS_FILE_READ, 0, &error);
if (in == TR_BAD_SYS_FILE)
{
tr_logAddError (err_fmt, filename, error->message);
tr_error_free (error);
return 0;
}
blocklistClose (b);
out = tr_sys_file_open (b->filename,
TR_SYS_FILE_WRITE | TR_SYS_FILE_CREATE | TR_SYS_FILE_TRUNCATE,
0666, &error);
if (out == TR_BAD_SYS_FILE)
{
tr_logAddError (err_fmt, b->filename, error->message);
tr_error_free (error);
tr_sys_file_close (in, NULL);
return 0;
}
/* load the rules into memory */
while (tr_sys_file_read_line (in, line, sizeof (line), NULL))
{
struct tr_ipv4_range range;
++inCount;
if (!parseLine (line, &range))
{
/* don't try to display the actual lines - it causes issues */
tr_logAddError (_("blocklist skipped invalid address at line %d"), inCount);
continue;
}
if (ranges_alloc == ranges_count)
{
ranges_alloc += 4096; /* arbitrary */
ranges = tr_renew (struct tr_ipv4_range, ranges, ranges_alloc);
}
ranges[ranges_count++] = range;
}
static void
dht_bootstrap (void *closure)
{
struct bootstrap_closure *cl = closure;
int i;
int num = cl->len / 6, num6 = cl->len6 / 18;
if (session != cl->session)
return;
if (cl->len > 0)
tr_logAddNamedInfo ("DHT", "Bootstrapping from %d IPv4 nodes", num);
if (cl->len6 > 0)
tr_logAddNamedInfo ("DHT", "Bootstrapping from %d IPv6 nodes", num6);
for (i = 0; i < MAX (num, num6); i++) {
if (i < num && !bootstrap_done (cl->session, AF_INET)) {
tr_port port;
struct tr_address addr;
memset (&addr, 0, sizeof (addr));
addr.type = TR_AF_INET;
memcpy (&addr.addr.addr4, &cl->nodes[i * 6], 4);
memcpy (&port, &cl->nodes[i * 6 + 4], 2);
port = ntohs (port);
tr_dhtAddNode (cl->session, &addr, port, 1);
}
if (i < num6 && !bootstrap_done (cl->session, AF_INET6)) {
tr_port port;
struct tr_address addr;
memset (&addr, 0, sizeof (addr));
addr.type = TR_AF_INET6;
memcpy (&addr.addr.addr6, &cl->nodes6[i * 18], 16);
memcpy (&port, &cl->nodes6[i * 18 + 16], 2);
port = ntohs (port);
tr_dhtAddNode (cl->session, &addr, port, 1);
}
/* Our DHT code is able to take up to 9 nodes in a row without
dropping any. After that, it takes some time to split buckets.
So ping the first 8 nodes quickly, then slow down. */
if (i < 8)
nap (2);
else
nap (15);
if (bootstrap_done (session, 0))
break;
}
if (!bootstrap_done (cl->session, 0)) {
char *bootstrap_file;
tr_sys_file_t f = TR_BAD_SYS_FILE;
bootstrap_file =
tr_buildPath (cl->session->configDir, "dht.bootstrap", NULL);
if (bootstrap_file)
f = tr_sys_file_open (bootstrap_file, TR_SYS_FILE_READ, 0, NULL);
if (f != TR_BAD_SYS_FILE) {
tr_logAddNamedInfo ("DHT", "Attempting manual bootstrap");
for (;;) {
char buf[201];
char *p;
int port = 0;
if (!tr_sys_file_read_line (f, buf, 200, NULL))
break;
p = memchr (buf, ' ', strlen (buf));
if (p != NULL)
port = atoi (p + 1);
if (p == NULL || port <= 0 || port >= 0x10000) {
tr_logAddNamedError ("DHT", "Couldn't parse %s", buf);
continue;
}
*p = '\0';
bootstrap_from_name (buf, port, bootstrap_af (session));
if (bootstrap_done (cl->session, 0))
break;
}
tr_sys_file_close (f, NULL);
}
tr_free (bootstrap_file);
}
if (!bootstrap_done (cl->session, 0)) {
for (i = 0; i < 6; i++) {
/* We don't want to abuse our bootstrap nodes, so be very
slow. The initial wait is to give other nodes a chance
to contact us before we attempt to contact a bootstrap
node, for example because we've just been restarted. */
nap (40);
if (bootstrap_done (cl->session, 0))
break;
if (i == 0)
tr_logAddNamedInfo ("DHT",
"Attempting bootstrap from dht.transmissionbt.com");
bootstrap_from_name ("dht.transmissionbt.com", 6881,
bootstrap_af (session));
}
}
if (cl->nodes)
tr_free (cl->nodes);
if (cl->nodes6)
tr_free (cl->nodes6);
tr_free (closure);
tr_logAddNamedDbg ("DHT", "Finished bootstrapping");
}
