Possible fix for broken country settings in ExcludeExitNodes.
[tor/rransom.git] / src / or / routerlist.c
blob080603b7e9f400020151379fe7b127b57311352e
1 /* Copyright (c) 2001 Matej Pfajfar.
2 * Copyright (c) 2001-2004, Roger Dingledine.
3 * Copyright (c) 2004-2006, Roger Dingledine, Nick Mathewson.
4 * Copyright (c) 2007-2008, The Tor Project, Inc. */
5 /* See LICENSE for licensing information */
7 /**
8 * \file routerlist.c
9 * \brief Code to
10 * maintain and access the global list of routerinfos for known
11 * servers.
12 **/
14 #include "or.h"
16 // #define DEBUG_ROUTERLIST
18 /****************************************************************************/
20 /* static function prototypes */
21 static routerstatus_t *router_pick_directory_server_impl(
22 authority_type_t auth, int flags);
23 static routerstatus_t *router_pick_trusteddirserver_impl(
24 authority_type_t auth, int flags, int *n_busy_out);
25 static void mark_all_trusteddirservers_up(void);
26 static int router_nickname_matches(routerinfo_t *router, const char *nickname);
27 static void trusted_dir_server_free(trusted_dir_server_t *ds);
28 static void launch_router_descriptor_downloads(smartlist_t *downloadable,
29 time_t now);
30 static void update_consensus_router_descriptor_downloads(time_t now);
31 static int signed_desc_digest_is_recognized(signed_descriptor_t *desc);
32 static void update_router_have_minimum_dir_info(void);
33 static const char *signed_descriptor_get_body_impl(signed_descriptor_t *desc,
34 int with_annotations);
35 static void list_pending_downloads(digestmap_t *result,
36 int purpose, const char *prefix);
38 DECLARE_TYPED_DIGESTMAP_FNS(sdmap_, digest_sd_map_t, signed_descriptor_t)
39 DECLARE_TYPED_DIGESTMAP_FNS(rimap_, digest_ri_map_t, routerinfo_t)
40 DECLARE_TYPED_DIGESTMAP_FNS(eimap_, digest_ei_map_t, extrainfo_t)
41 #define SDMAP_FOREACH(map, keyvar, valvar) \
42 DIGESTMAP_FOREACH(sdmap_to_digestmap(map), keyvar, signed_descriptor_t *, \
43 valvar)
44 #define RIMAP_FOREACH(map, keyvar, valvar) \
45 DIGESTMAP_FOREACH(rimap_to_digestmap(map), keyvar, routerinfo_t *, valvar)
46 #define EIMAP_FOREACH(map, keyvar, valvar) \
47 DIGESTMAP_FOREACH(eimap_to_digestmap(map), keyvar, extrainfo_t *, valvar)
49 /****************************************************************************/
51 /** Global list of a trusted_dir_server_t object for each trusted directory
52 * server. */
53 static smartlist_t *trusted_dir_servers = NULL;
55 /** List of for a given authority, and download status for latest certificate.
57 typedef struct cert_list_t {
58 download_status_t dl_status;
59 smartlist_t *certs;
60 } cert_list_t;
61 /** Map from v3 identity key digest to cert_list_t. */
62 static digestmap_t *trusted_dir_certs = NULL;
63 /** True iff any key certificate in at least one member of
64 * <b>trusted_dir_certs</b> has changed since we last flushed the
65 * certificates to disk. */
66 static int trusted_dir_servers_certs_changed = 0;
68 /** Global list of all of the routers that we know about. */
69 static routerlist_t *routerlist = NULL;
71 /** List of strings for nicknames we've already warned about and that are
72 * still unknown / unavailable. */
73 static smartlist_t *warned_nicknames = NULL;
75 /** The last time we tried to download any routerdesc, or 0 for "never". We
76 * use this to rate-limit download attempts when the number of routerdescs to
77 * download is low. */
78 static time_t last_routerdesc_download_attempted = 0;
80 /** When we last computed the weights to use for bandwidths on directory
81 * requests, what were the total weighted bandwidth, and our share of that
82 * bandwidth? Used to determine what fraction of directory requests we should
83 * expect to see. */
84 static uint64_t sl_last_total_weighted_bw = 0,
85 sl_last_weighted_bw_of_me = 0;
87 /** Return the number of directory authorities whose type matches some bit set
88 * in <b>type</b> */
89 int
90 get_n_authorities(authority_type_t type)
92 int n = 0;
93 if (!trusted_dir_servers)
94 return 0;
95 SMARTLIST_FOREACH(trusted_dir_servers, trusted_dir_server_t *, ds,
96 if (ds->type & type)
97 ++n);
98 return n;
101 #define get_n_v2_authorities() get_n_authorities(V2_AUTHORITY)
103 /** Helper: Return the cert_list_t for an authority whose authority ID is
104 * <b>id_digest</b>, allocating a new list if necessary. */
105 static cert_list_t *
106 get_cert_list(const char *id_digest)
108 cert_list_t *cl;
109 if (!trusted_dir_certs)
110 trusted_dir_certs = digestmap_new();
111 cl = digestmap_get(trusted_dir_certs, id_digest);
112 if (!cl) {
113 cl = tor_malloc_zero(sizeof(cert_list_t));
114 cl->dl_status.schedule = DL_SCHED_CONSENSUS;
115 cl->certs = smartlist_create();
116 digestmap_set(trusted_dir_certs, id_digest, cl);
118 return cl;
121 /** Reload the cached v3 key certificates from the cached-certs file in
122 * the data directory. Return 0 on success, -1 on failure. */
124 trusted_dirs_reload_certs(void)
126 char *filename;
127 char *contents;
128 int r;
130 filename = get_datadir_fname("cached-certs");
131 contents = read_file_to_str(filename, RFTS_IGNORE_MISSING, NULL);
132 tor_free(filename);
133 if (!contents)
134 return 0;
135 r = trusted_dirs_load_certs_from_string(contents, 1, 1);
136 tor_free(contents);
137 return r;
140 /** Helper: return true iff we already have loaded the exact cert
141 * <b>cert</b>. */
142 static INLINE int
143 already_have_cert(authority_cert_t *cert)
145 cert_list_t *cl = get_cert_list(cert->cache_info.identity_digest);
147 SMARTLIST_FOREACH(cl->certs, authority_cert_t *, c,
149 if (!memcmp(c->cache_info.signed_descriptor_digest,
150 cert->cache_info.signed_descriptor_digest,
151 DIGEST_LEN))
152 return 1;
154 return 0;
157 /** Load a bunch of new key certificates from the string <b>contents</b>. If
158 * <b>from_store</b> is true, the certificates are from the cache, and we
159 * don't need to flush them to disk. If <b>from_store</b> is false, we need
160 * to flush any changed certificates to disk. Return 0 on success, -1 on
161 * failure. */
163 trusted_dirs_load_certs_from_string(const char *contents, int from_store,
164 int flush)
166 trusted_dir_server_t *ds;
167 const char *s, *eos;
169 for (s = contents; *s; s = eos) {
170 authority_cert_t *cert = authority_cert_parse_from_string(s, &eos);
171 cert_list_t *cl;
172 if (!cert)
173 break;
174 ds = trusteddirserver_get_by_v3_auth_digest(
175 cert->cache_info.identity_digest);
176 log_debug(LD_DIR, "Parsed certificate for %s",
177 ds ? ds->nickname : "unknown authority");
179 if (already_have_cert(cert)) {
180 /* we already have this one. continue. */
181 log_info(LD_DIR, "Skipping %s certificate for %s that we "
182 "already have.",
183 from_store ? "cached" : "downloaded",
184 ds ? ds->nickname : "??");
186 /* a duplicate on a download should be treated as a failure, since it
187 * probably means we wanted a different secret key or we are trying to
188 * replace an expired cert that has not in fact been updated. */
189 if (!from_store) {
190 log_warn(LD_DIR, "Got a certificate for %s that we already have. "
191 "Maybe they haven't updated it. Waiting for a while.",
192 ds ? ds->nickname : "??");
193 authority_cert_dl_failed(cert->cache_info.identity_digest, 404);
196 authority_cert_free(cert);
197 continue;
200 if (ds) {
201 log_info(LD_DIR, "Adding %s certificate for directory authority %s with "
202 "signing key %s", from_store ? "cached" : "downloaded",
203 ds->nickname, hex_str(cert->signing_key_digest,DIGEST_LEN));
204 } else {
205 int adding = directory_caches_dir_info(get_options());
206 log_info(LD_DIR, "%s %s certificate for unrecognized directory "
207 "authority with signing key %s",
208 adding ? "Adding" : "Not adding",
209 from_store ? "cached" : "downloaded",
210 hex_str(cert->signing_key_digest,DIGEST_LEN));
211 if (!adding) {
212 authority_cert_free(cert);
213 continue;
217 cl = get_cert_list(cert->cache_info.identity_digest);
218 smartlist_add(cl->certs, cert);
219 if (ds && cert->cache_info.published_on > ds->addr_current_at) {
220 /* Check to see whether we should update our view of the authority's
221 * address. */
222 if (cert->addr && cert->dir_port &&
223 (ds->addr != cert->addr ||
224 ds->dir_port != cert->dir_port)) {
225 char *a = tor_dup_ip(cert->addr);
226 log_notice(LD_DIR, "Updating address for directory authority %s "
227 "from %s:%d to %s:%d based on in certificate.",
228 ds->nickname, ds->address, (int)ds->dir_port,
229 a, cert->dir_port);
230 tor_free(a);
231 ds->addr = cert->addr;
232 ds->dir_port = cert->dir_port;
234 ds->addr_current_at = cert->cache_info.published_on;
237 if (!from_store)
238 trusted_dir_servers_certs_changed = 1;
241 if (flush)
242 trusted_dirs_flush_certs_to_disk();
244 networkstatus_note_certs_arrived();
245 return 0;
248 /** Save all v3 key certificates to the cached-certs file. */
249 void
250 trusted_dirs_flush_certs_to_disk(void)
252 char *filename;
253 smartlist_t *chunks;
255 if (!trusted_dir_servers_certs_changed || !trusted_dir_certs)
256 return;
258 chunks = smartlist_create();
259 DIGESTMAP_FOREACH(trusted_dir_certs, key, cert_list_t *, cl) {
260 SMARTLIST_FOREACH(cl->certs, authority_cert_t *, cert,
262 sized_chunk_t *c = tor_malloc(sizeof(sized_chunk_t));
263 c->bytes = cert->cache_info.signed_descriptor_body;
264 c->len = cert->cache_info.signed_descriptor_len;
265 smartlist_add(chunks, c);
267 } DIGESTMAP_FOREACH_END;
269 filename = get_datadir_fname("cached-certs");
270 if (write_chunks_to_file(filename, chunks, 0)) {
271 log_warn(LD_FS, "Error writing certificates to disk.");
273 tor_free(filename);
274 SMARTLIST_FOREACH(chunks, sized_chunk_t *, c, tor_free(c));
275 smartlist_free(chunks);
277 trusted_dir_servers_certs_changed = 0;
280 /** Remove all v3 authority certificates that have been superseded for more
281 * than 48 hours. (If the most recent cert was published more than 48 hours
282 * ago, then we aren't going to get any consensuses signed with older
283 * keys.) */
284 static void
285 trusted_dirs_remove_old_certs(void)
287 time_t now = time(NULL);
288 #define DEAD_CERT_LIFETIME (2*24*60*60)
289 #define OLD_CERT_LIFETIME (7*24*60*60)
290 if (!trusted_dir_certs)
291 return;
293 DIGESTMAP_FOREACH(trusted_dir_certs, key, cert_list_t *, cl) {
294 authority_cert_t *newest = NULL;
295 SMARTLIST_FOREACH(cl->certs, authority_cert_t *, cert,
296 if (!newest || (cert->cache_info.published_on >
297 newest->cache_info.published_on))
298 newest = cert);
299 if (newest) {
300 const time_t newest_published = newest->cache_info.published_on;
301 SMARTLIST_FOREACH_BEGIN(cl->certs, authority_cert_t *, cert) {
302 int expired;
303 time_t cert_published;
304 if (newest == cert)
305 continue;
306 expired = ftime_definitely_after(now, cert->expires);
307 cert_published = cert->cache_info.published_on;
308 /* Store expired certs for 48 hours after a newer arrives;
310 if (expired ?
311 (newest_published + DEAD_CERT_LIFETIME < now) :
312 (cert_published + OLD_CERT_LIFETIME < newest_published)) {
313 SMARTLIST_DEL_CURRENT(cl->certs, cert);
314 authority_cert_free(cert);
315 trusted_dir_servers_certs_changed = 1;
317 } SMARTLIST_FOREACH_END(cert);
319 } DIGESTMAP_FOREACH_END;
320 #undef OLD_CERT_LIFETIME
322 trusted_dirs_flush_certs_to_disk();
325 /** Return the newest v3 authority certificate whose v3 authority identity key
326 * has digest <b>id_digest</b>. Return NULL if no such authority is known,
327 * or it has no certificate. */
328 authority_cert_t *
329 authority_cert_get_newest_by_id(const char *id_digest)
331 cert_list_t *cl;
332 authority_cert_t *best = NULL;
333 if (!trusted_dir_certs ||
334 !(cl = digestmap_get(trusted_dir_certs, id_digest)))
335 return NULL;
337 SMARTLIST_FOREACH(cl->certs, authority_cert_t *, cert,
339 if (!best || cert->cache_info.published_on > best->cache_info.published_on)
340 best = cert;
342 return best;
345 /** Return the newest v3 authority certificate whose directory signing key has
346 * digest <b>sk_digest</b>. Return NULL if no such certificate is known.
348 authority_cert_t *
349 authority_cert_get_by_sk_digest(const char *sk_digest)
351 authority_cert_t *c;
352 if (!trusted_dir_certs)
353 return NULL;
355 if ((c = get_my_v3_authority_cert()) &&
356 !memcmp(c->signing_key_digest, sk_digest, DIGEST_LEN))
357 return c;
358 if ((c = get_my_v3_legacy_cert()) &&
359 !memcmp(c->signing_key_digest, sk_digest, DIGEST_LEN))
360 return c;
362 DIGESTMAP_FOREACH(trusted_dir_certs, key, cert_list_t *, cl) {
363 SMARTLIST_FOREACH(cl->certs, authority_cert_t *, cert,
365 if (!memcmp(cert->signing_key_digest, sk_digest, DIGEST_LEN))
366 return cert;
368 } DIGESTMAP_FOREACH_END;
369 return NULL;
372 /** Return the v3 authority certificate with signing key matching
373 * <b>sk_digest</b>, for the authority with identity digest <b>id_digest</b>.
374 * Return NULL if no such authority is known. */
375 authority_cert_t *
376 authority_cert_get_by_digests(const char *id_digest,
377 const char *sk_digest)
379 cert_list_t *cl;
380 if (!trusted_dir_certs ||
381 !(cl = digestmap_get(trusted_dir_certs, id_digest)))
382 return NULL;
383 SMARTLIST_FOREACH(cl->certs, authority_cert_t *, cert,
384 if (!memcmp(cert->signing_key_digest, sk_digest, DIGEST_LEN))
385 return cert; );
387 return NULL;
390 /** Add every known authority_cert_t to <b>certs_out</b>. */
391 void
392 authority_cert_get_all(smartlist_t *certs_out)
394 tor_assert(certs_out);
395 if (!trusted_dir_certs)
396 return;
398 DIGESTMAP_FOREACH(trusted_dir_certs, key, cert_list_t *, cl) {
399 SMARTLIST_FOREACH(cl->certs, authority_cert_t *, c,
400 smartlist_add(certs_out, c));
401 } DIGESTMAP_FOREACH_END;
404 /** Called when an attempt to download a certificate with the authority with
405 * ID <b>id_digest</b> fails with HTTP response code <b>status</b>: remember
406 * the failure, so we don't try again immediately. */
407 void
408 authority_cert_dl_failed(const char *id_digest, int status)
410 cert_list_t *cl;
411 if (!trusted_dir_certs ||
412 !(cl = digestmap_get(trusted_dir_certs, id_digest)))
413 return;
415 download_status_failed(&cl->dl_status, status);
418 /** How many times will we try to fetch a certificate before giving up? */
419 #define MAX_CERT_DL_FAILURES 8
421 /** Try to download any v3 authority certificates that we may be missing. If
422 * <b>status</b> is provided, try to get all the ones that were used to sign
423 * <b>status</b>. Additionally, try to have a non-expired certificate for
424 * every V3 authority in trusted_dir_servers. Don't fetch certificates we
425 * already have.
427 void
428 authority_certs_fetch_missing(networkstatus_t *status, time_t now)
430 digestmap_t *pending;
431 authority_cert_t *cert;
432 smartlist_t *missing_digests;
433 char *resource = NULL;
434 cert_list_t *cl;
435 const int cache = directory_caches_dir_info(get_options());
437 if (should_delay_dir_fetches(get_options()))
438 return;
440 pending = digestmap_new();
441 missing_digests = smartlist_create();
443 list_pending_downloads(pending, DIR_PURPOSE_FETCH_CERTIFICATE, "fp/");
444 if (status) {
445 SMARTLIST_FOREACH(status->voters, networkstatus_voter_info_t *, voter,
447 if (tor_digest_is_zero(voter->signing_key_digest))
448 continue; /* This authority never signed this consensus, so don't
449 * go looking for a cert with key digest 0000000000. */
450 if (!cache &&
451 !trusteddirserver_get_by_v3_auth_digest(voter->identity_digest))
452 continue; /* We are not a cache, and we don't know this authority.*/
453 cl = get_cert_list(voter->identity_digest);
454 cert = authority_cert_get_by_digests(voter->identity_digest,
455 voter->signing_key_digest);
456 if (cert) {
457 if (now < cert->expires)
458 download_status_reset(&cl->dl_status);
459 continue;
461 if (download_status_is_ready(&cl->dl_status, now,
462 MAX_CERT_DL_FAILURES) &&
463 !digestmap_get(pending, voter->identity_digest)) {
464 log_notice(LD_DIR, "We're missing a certificate from authority "
465 "with signing key %s: launching request.",
466 hex_str(voter->signing_key_digest, DIGEST_LEN));
467 smartlist_add(missing_digests, voter->identity_digest);
471 SMARTLIST_FOREACH(trusted_dir_servers, trusted_dir_server_t *, ds,
473 int found = 0;
474 if (!(ds->type & V3_AUTHORITY))
475 continue;
476 if (smartlist_digest_isin(missing_digests, ds->v3_identity_digest))
477 continue;
478 cl = get_cert_list(ds->v3_identity_digest);
479 SMARTLIST_FOREACH(cl->certs, authority_cert_t *, cert,
481 if (!ftime_definitely_after(now, cert->expires)) {
482 /* It's not expired, and we weren't looking for something to
483 * verify a consensus with. Call it done. */
484 download_status_reset(&cl->dl_status);
485 found = 1;
486 break;
489 if (!found &&
490 download_status_is_ready(&cl->dl_status, now,MAX_CERT_DL_FAILURES) &&
491 !digestmap_get(pending, ds->v3_identity_digest)) {
492 log_notice(LD_DIR, "No current certificate known for authority %s; "
493 "launching request.", ds->nickname);
494 smartlist_add(missing_digests, ds->v3_identity_digest);
498 if (!smartlist_len(missing_digests)) {
499 goto done;
500 } else {
501 smartlist_t *fps = smartlist_create();
502 smartlist_add(fps, tor_strdup("fp/"));
503 SMARTLIST_FOREACH(missing_digests, const char *, d, {
504 char *fp;
505 if (digestmap_get(pending, d))
506 continue;
507 fp = tor_malloc(HEX_DIGEST_LEN+2);
508 base16_encode(fp, HEX_DIGEST_LEN+1, d, DIGEST_LEN);
509 fp[HEX_DIGEST_LEN] = '+';
510 fp[HEX_DIGEST_LEN+1] = '\0';
511 smartlist_add(fps, fp);
513 if (smartlist_len(fps) == 1) {
514 /* we didn't add any: they were all pending */
515 SMARTLIST_FOREACH(fps, char *, cp, tor_free(cp));
516 smartlist_free(fps);
517 goto done;
519 resource = smartlist_join_strings(fps, "", 0, NULL);
520 resource[strlen(resource)-1] = '\0';
521 SMARTLIST_FOREACH(fps, char *, cp, tor_free(cp));
522 smartlist_free(fps);
524 directory_get_from_dirserver(DIR_PURPOSE_FETCH_CERTIFICATE, 0,
525 resource, PDS_RETRY_IF_NO_SERVERS);
527 done:
528 tor_free(resource);
529 smartlist_free(missing_digests);
530 digestmap_free(pending, NULL);
533 /* Router descriptor storage.
535 * Routerdescs are stored in a big file, named "cached-descriptors". As new
536 * routerdescs arrive, we append them to a journal file named
537 * "cached-descriptors.new".
539 * From time to time, we replace "cached-descriptors" with a new file
540 * containing only the live, non-superseded descriptors, and clear
541 * cached-routers.new.
543 * On startup, we read both files.
546 /** Helper: return 1 iff the router log is so big we want to rebuild the
547 * store. */
548 static int
549 router_should_rebuild_store(desc_store_t *store)
551 if (store->store_len > (1<<16))
552 return (store->journal_len > store->store_len / 2 ||
553 store->bytes_dropped > store->store_len / 2);
554 else
555 return store->journal_len > (1<<15);
558 /** Return the desc_store_t in <b>rl</b> that should be used to store
559 * <b>sd</b>. */
560 static INLINE desc_store_t *
561 desc_get_store(routerlist_t *rl, signed_descriptor_t *sd)
563 if (sd->is_extrainfo)
564 return &rl->extrainfo_store;
565 else
566 return &rl->desc_store;
569 /** Add the signed_descriptor_t in <b>desc</b> to the router
570 * journal; change its saved_location to SAVED_IN_JOURNAL and set its
571 * offset appropriately. */
572 static int
573 signed_desc_append_to_journal(signed_descriptor_t *desc,
574 desc_store_t *store)
576 char *fname = get_datadir_fname_suffix(store->fname_base, ".new");
577 const char *body = signed_descriptor_get_body_impl(desc,1);
578 size_t len = desc->signed_descriptor_len + desc->annotations_len;
580 tor_assert(len == strlen(body));
582 if (append_bytes_to_file(fname, body, len, 1)) {
583 log_warn(LD_FS, "Unable to store router descriptor");
584 tor_free(fname);
585 return -1;
587 desc->saved_location = SAVED_IN_JOURNAL;
588 tor_free(fname);
590 desc->saved_offset = store->journal_len;
591 store->journal_len += len;
593 return 0;
596 /** Sorting helper: return &lt;0, 0, or &gt;0 depending on whether the
597 * signed_descriptor_t* in *<b>a</b> is older, the same age as, or newer than
598 * the signed_descriptor_t* in *<b>b</b>. */
599 static int
600 _compare_signed_descriptors_by_age(const void **_a, const void **_b)
602 const signed_descriptor_t *r1 = *_a, *r2 = *_b;
603 return (int)(r1->published_on - r2->published_on);
606 #define RRS_FORCE 1
607 #define RRS_DONT_REMOVE_OLD 2
609 /** If the journal of <b>store</b> is too long, or if RRS_FORCE is set in
610 * <b>flags</b>, then atomically replace the saved router store with the
611 * routers currently in our routerlist, and clear the journal. Unless
612 * RRS_DONT_REMOVE_OLD is set in <b>flags</b>, delete expired routers before
613 * rebuilding the store. Return 0 on success, -1 on failure.
615 static int
616 router_rebuild_store(int flags, desc_store_t *store)
618 smartlist_t *chunk_list = NULL;
619 char *fname = NULL, *fname_tmp = NULL;
620 int r = -1;
621 off_t offset = 0;
622 smartlist_t *signed_descriptors = NULL;
623 int nocache=0;
624 size_t total_expected_len = 0;
625 int had_any;
626 int force = flags & RRS_FORCE;
628 if (!force && !router_should_rebuild_store(store)) {
629 r = 0;
630 goto done;
632 if (!routerlist) {
633 r = 0;
634 goto done;
637 if (store->type == EXTRAINFO_STORE)
638 had_any = !eimap_isempty(routerlist->extra_info_map);
639 else
640 had_any = (smartlist_len(routerlist->routers)+
641 smartlist_len(routerlist->old_routers))>0;
643 /* Don't save deadweight. */
644 if (!(flags & RRS_DONT_REMOVE_OLD))
645 routerlist_remove_old_routers();
647 log_info(LD_DIR, "Rebuilding %s cache", store->description);
649 fname = get_datadir_fname(store->fname_base);
650 fname_tmp = get_datadir_fname_suffix(store->fname_base, ".tmp");
652 chunk_list = smartlist_create();
654 /* We sort the routers by age to enhance locality on disk. */
655 signed_descriptors = smartlist_create();
656 if (store->type == EXTRAINFO_STORE) {
657 eimap_iter_t *iter;
658 for (iter = eimap_iter_init(routerlist->extra_info_map);
659 !eimap_iter_done(iter);
660 iter = eimap_iter_next(routerlist->extra_info_map, iter)) {
661 const char *key;
662 extrainfo_t *ei;
663 eimap_iter_get(iter, &key, &ei);
664 smartlist_add(signed_descriptors, &ei->cache_info);
666 } else {
667 SMARTLIST_FOREACH(routerlist->old_routers, signed_descriptor_t *, sd,
668 smartlist_add(signed_descriptors, sd));
669 SMARTLIST_FOREACH(routerlist->routers, routerinfo_t *, ri,
670 smartlist_add(signed_descriptors, &ri->cache_info));
673 smartlist_sort(signed_descriptors, _compare_signed_descriptors_by_age);
675 /* Now, add the appropriate members to chunk_list */
676 SMARTLIST_FOREACH(signed_descriptors, signed_descriptor_t *, sd,
678 sized_chunk_t *c;
679 const char *body = signed_descriptor_get_body_impl(sd, 1);
680 if (!body) {
681 log_warn(LD_BUG, "No descriptor available for router.");
682 goto done;
684 if (sd->do_not_cache) {
685 ++nocache;
686 continue;
688 c = tor_malloc(sizeof(sized_chunk_t));
689 c->bytes = body;
690 c->len = sd->signed_descriptor_len + sd->annotations_len;
691 total_expected_len += c->len;
692 smartlist_add(chunk_list, c);
695 if (write_chunks_to_file(fname_tmp, chunk_list, 1)<0) {
696 log_warn(LD_FS, "Error writing router store to disk.");
697 goto done;
700 /* Our mmap is now invalid. */
701 if (store->mmap) {
702 tor_munmap_file(store->mmap);
703 store->mmap = NULL;
706 if (replace_file(fname_tmp, fname)<0) {
707 log_warn(LD_FS, "Error replacing old router store: %s", strerror(errno));
708 goto done;
711 errno = 0;
712 store->mmap = tor_mmap_file(fname);
713 if (! store->mmap) {
714 if (errno == ERANGE) {
715 /* empty store.*/
716 if (total_expected_len) {
717 log_warn(LD_FS, "We wrote some bytes to a new descriptor file at '%s',"
718 " but when we went to mmap it, it was empty!", fname);
719 } else if (had_any) {
720 log_info(LD_FS, "We just removed every descriptor in '%s'. This is "
721 "okay if we're just starting up after a long time. "
722 "Otherwise, it's a bug.", fname);
724 } else {
725 log_warn(LD_FS, "Unable to mmap new descriptor file at '%s'.",fname);
729 log_info(LD_DIR, "Reconstructing pointers into cache");
731 offset = 0;
732 SMARTLIST_FOREACH(signed_descriptors, signed_descriptor_t *, sd,
734 if (sd->do_not_cache)
735 continue;
736 sd->saved_location = SAVED_IN_CACHE;
737 if (store->mmap) {
738 tor_free(sd->signed_descriptor_body); // sets it to null
739 sd->saved_offset = offset;
741 offset += sd->signed_descriptor_len + sd->annotations_len;
742 signed_descriptor_get_body(sd); /* reconstruct and assert */
745 tor_free(fname);
746 fname = get_datadir_fname_suffix(store->fname_base, ".new");
747 write_str_to_file(fname, "", 1);
749 r = 0;
750 store->store_len = (size_t) offset;
751 store->journal_len = 0;
752 store->bytes_dropped = 0;
753 done:
754 if (signed_descriptors)
755 smartlist_free(signed_descriptors);
756 tor_free(fname);
757 tor_free(fname_tmp);
758 if (chunk_list) {
759 SMARTLIST_FOREACH(chunk_list, sized_chunk_t *, c, tor_free(c));
760 smartlist_free(chunk_list);
763 return r;
766 /** Helper: Reload a cache file and its associated journal, setting metadata
767 * appropriately. If <b>extrainfo</b> is true, reload the extrainfo store;
768 * else reload the router descriptor store. */
769 static int
770 router_reload_router_list_impl(desc_store_t *store)
772 char *fname = NULL, *altname = NULL, *contents = NULL;
773 struct stat st;
774 int read_from_old_location = 0;
775 int extrainfo = (store->type == EXTRAINFO_STORE);
776 time_t now = time(NULL);
777 store->journal_len = store->store_len = 0;
779 fname = get_datadir_fname(store->fname_base);
780 if (store->fname_alt_base)
781 altname = get_datadir_fname(store->fname_alt_base);
783 if (store->mmap) /* get rid of it first */
784 tor_munmap_file(store->mmap);
785 store->mmap = NULL;
787 store->mmap = tor_mmap_file(fname);
788 if (!store->mmap && altname && file_status(altname) == FN_FILE) {
789 read_from_old_location = 1;
790 log_notice(LD_DIR, "Couldn't read %s; trying to load routers from old "
791 "location %s.", fname, altname);
792 if ((store->mmap = tor_mmap_file(altname)))
793 read_from_old_location = 1;
795 if (altname && !read_from_old_location) {
796 remove_file_if_very_old(altname, now);
798 if (store->mmap) {
799 store->store_len = store->mmap->size;
800 if (extrainfo)
801 router_load_extrainfo_from_string(store->mmap->data,
802 store->mmap->data+store->mmap->size,
803 SAVED_IN_CACHE, NULL, 0);
804 else
805 router_load_routers_from_string(store->mmap->data,
806 store->mmap->data+store->mmap->size,
807 SAVED_IN_CACHE, NULL, 0, NULL);
810 tor_free(fname);
811 fname = get_datadir_fname_suffix(store->fname_base, ".new");
812 if (file_status(fname) == FN_FILE)
813 contents = read_file_to_str(fname, RFTS_BIN|RFTS_IGNORE_MISSING, &st);
814 if (read_from_old_location) {
815 tor_free(altname);
816 altname = get_datadir_fname_suffix(store->fname_alt_base, ".new");
817 if (!contents)
818 contents = read_file_to_str(altname, RFTS_BIN|RFTS_IGNORE_MISSING, &st);
819 else
820 remove_file_if_very_old(altname, now);
822 if (contents) {
823 if (extrainfo)
824 router_load_extrainfo_from_string(contents, NULL,SAVED_IN_JOURNAL,
825 NULL, 0);
826 else
827 router_load_routers_from_string(contents, NULL, SAVED_IN_JOURNAL,
828 NULL, 0, NULL);
829 store->journal_len = (size_t) st.st_size;
830 tor_free(contents);
833 tor_free(fname);
834 tor_free(altname);
836 if (store->journal_len || read_from_old_location) {
837 /* Always clear the journal on startup.*/
838 router_rebuild_store(RRS_FORCE, store);
839 } else if (!extrainfo) {
840 /* Don't cache expired routers. (This is in an else because
841 * router_rebuild_store() also calls remove_old_routers().) */
842 routerlist_remove_old_routers();
845 return 0;
848 /** Load all cached router descriptors and extra-info documents from the
849 * store. Return 0 on success and -1 on failure.
852 router_reload_router_list(void)
854 routerlist_t *rl = router_get_routerlist();
855 if (router_reload_router_list_impl(&rl->desc_store))
856 return -1;
857 if (router_reload_router_list_impl(&rl->extrainfo_store))
858 return -1;
859 return 0;
862 /** Return a smartlist containing a list of trusted_dir_server_t * for all
863 * known trusted dirservers. Callers must not modify the list or its
864 * contents.
866 smartlist_t *
867 router_get_trusted_dir_servers(void)
869 if (!trusted_dir_servers)
870 trusted_dir_servers = smartlist_create();
872 return trusted_dir_servers;
875 /** Try to find a running dirserver that supports operations of <b>type</b>.
877 * If there are no running dirservers in our routerlist and the
878 * <b>PDS_RETRY_IF_NO_SERVERS</b> flag is set, set all the authoritative ones
879 * as running again, and pick one.
881 * If the <b>PDS_IGNORE_FASCISTFIREWALL</b> flag is set, then include
882 * dirservers that we can't reach.
884 * If the <b>PDS_ALLOW_SELF</b> flag is not set, then don't include ourself
885 * (if we're a dirserver).
887 * Don't pick an authority if any non-authority is viable; try to avoid using
888 * servers that have returned 503 recently.
890 routerstatus_t *
891 router_pick_directory_server(authority_type_t type, int flags)
893 routerstatus_t *choice;
894 if (get_options()->PreferTunneledDirConns)
895 flags |= _PDS_PREFER_TUNNELED_DIR_CONNS;
897 if (!routerlist)
898 return NULL;
900 choice = router_pick_directory_server_impl(type, flags);
901 if (choice || !(flags & PDS_RETRY_IF_NO_SERVERS))
902 return choice;
904 log_info(LD_DIR,
905 "No reachable router entries for dirservers. "
906 "Trying them all again.");
907 /* mark all authdirservers as up again */
908 mark_all_trusteddirservers_up();
909 /* try again */
910 choice = router_pick_directory_server_impl(type, flags);
911 return choice;
914 /** Try to determine which fraction of v2 and v3 directory requsts aimed at
915 * caches will be sent to us. Set *<b>v2_share_out</b> and
916 * *<b>v3_share_out</b> to the fractions of v2 and v3 protocol shares we
917 * expect to see, respectively. Return 0 on success, negative on failue. */
919 router_get_my_share_of_directory_requests(double *v2_share_out,
920 double *v3_share_out)
922 routerinfo_t *me = router_get_my_routerinfo();
923 routerstatus_t *rs;
924 const int pds_flags = PDS_ALLOW_SELF|PDS_IGNORE_FASCISTFIREWALL;
925 *v2_share_out = *v3_share_out = 0.0;
926 if (!me)
927 return -1;
928 rs = router_get_consensus_status_by_id(me->cache_info.identity_digest);
929 if (!rs)
930 return -1;
932 /* Calling for side effect */
933 /* XXXX This is a bit of a kludge */
934 if (rs->is_v2_dir) {
935 sl_last_total_weighted_bw = 0;
936 router_pick_directory_server(V2_AUTHORITY, pds_flags);
937 if (sl_last_total_weighted_bw != 0) {
938 *v2_share_out = U64_TO_DBL(sl_last_weighted_bw_of_me) /
939 U64_TO_DBL(sl_last_total_weighted_bw);
943 if (rs->version_supports_v3_dir) {
944 sl_last_total_weighted_bw = 0;
945 router_pick_directory_server(V3_AUTHORITY, pds_flags);
946 if (sl_last_total_weighted_bw != 0) {
947 *v3_share_out = U64_TO_DBL(sl_last_weighted_bw_of_me) /
948 U64_TO_DBL(sl_last_total_weighted_bw);
952 return 0;
955 /** Return the trusted_dir_server_t for the directory authority whose identity
956 * key hashes to <b>digest</b>, or NULL if no such authority is known.
958 trusted_dir_server_t *
959 router_get_trusteddirserver_by_digest(const char *digest)
961 if (!trusted_dir_servers)
962 return NULL;
964 SMARTLIST_FOREACH(trusted_dir_servers, trusted_dir_server_t *, ds,
966 if (!memcmp(ds->digest, digest, DIGEST_LEN))
967 return ds;
970 return NULL;
973 /** Return the trusted_dir_server_t for the directory authority whose identity
974 * key hashes to <b>digest</b>, or NULL if no such authority is known.
976 trusted_dir_server_t *
977 trusteddirserver_get_by_v3_auth_digest(const char *digest)
979 if (!trusted_dir_servers)
980 return NULL;
982 SMARTLIST_FOREACH(trusted_dir_servers, trusted_dir_server_t *, ds,
984 if (!memcmp(ds->v3_identity_digest, digest, DIGEST_LEN) &&
985 (ds->type & V3_AUTHORITY))
986 return ds;
989 return NULL;
992 /** Try to find a running trusted dirserver. Flags are as for
993 * router_pick_directory_server.
995 routerstatus_t *
996 router_pick_trusteddirserver(authority_type_t type, int flags)
998 routerstatus_t *choice;
999 int busy = 0;
1000 if (get_options()->PreferTunneledDirConns)
1001 flags |= _PDS_PREFER_TUNNELED_DIR_CONNS;
1003 choice = router_pick_trusteddirserver_impl(type, flags, &busy);
1004 if (choice || !(flags & PDS_RETRY_IF_NO_SERVERS))
1005 return choice;
1006 if (busy) {
1007 /* If the reason that we got no server is that servers are "busy",
1008 * we must be excluding good servers because we already have serverdesc
1009 * fetches with them. Do not mark down servers up because of this. */
1010 tor_assert((flags & PDS_NO_EXISTING_SERVERDESC_FETCH));
1011 return NULL;
1014 log_info(LD_DIR,
1015 "No trusted dirservers are reachable. Trying them all again.");
1016 mark_all_trusteddirservers_up();
1017 return router_pick_trusteddirserver_impl(type, flags, NULL);
1020 /** How long do we avoid using a directory server after it's given us a 503? */
1021 #define DIR_503_TIMEOUT (60*60)
1023 /** Pick a random running valid directory server/mirror from our
1024 * routerlist. Arguments are as for router_pick_directory_server(), except
1025 * that RETRY_IF_NO_SERVERS is ignored, and:
1027 * If the _PDS_PREFER_TUNNELED_DIR_CONNS flag is set, prefer directory servers
1028 * that we can use with BEGINDIR.
1030 static routerstatus_t *
1031 router_pick_directory_server_impl(authority_type_t type, int flags)
1033 routerstatus_t *result;
1034 smartlist_t *direct, *tunnel;
1035 smartlist_t *trusted_direct, *trusted_tunnel;
1036 smartlist_t *overloaded_direct, *overloaded_tunnel;
1037 time_t now = time(NULL);
1038 const networkstatus_t *consensus = networkstatus_get_latest_consensus();
1039 int requireother = ! (flags & PDS_ALLOW_SELF);
1040 int fascistfirewall = ! (flags & PDS_IGNORE_FASCISTFIREWALL);
1041 int prefer_tunnel = (flags & _PDS_PREFER_TUNNELED_DIR_CONNS);
1043 if (!consensus)
1044 return NULL;
1046 direct = smartlist_create();
1047 tunnel = smartlist_create();
1048 trusted_direct = smartlist_create();
1049 trusted_tunnel = smartlist_create();
1050 overloaded_direct = smartlist_create();
1051 overloaded_tunnel = smartlist_create();
1053 /* Find all the running dirservers we know about. */
1054 SMARTLIST_FOREACH_BEGIN(consensus->routerstatus_list, routerstatus_t *,
1055 status) {
1056 int is_trusted;
1057 int is_overloaded = status->last_dir_503_at + DIR_503_TIMEOUT > now;
1058 tor_addr_t addr;
1059 if (!status->is_running || !status->dir_port || !status->is_valid)
1060 continue;
1061 if (status->is_bad_directory)
1062 continue;
1063 if (requireother && router_digest_is_me(status->identity_digest))
1064 continue;
1065 if (type & V3_AUTHORITY) {
1066 if (!(status->version_supports_v3_dir ||
1067 router_digest_is_trusted_dir_type(status->identity_digest,
1068 V3_AUTHORITY)))
1069 continue;
1071 is_trusted = router_digest_is_trusted_dir(status->identity_digest);
1072 if ((type & V2_AUTHORITY) && !(status->is_v2_dir || is_trusted))
1073 continue;
1074 if ((type & EXTRAINFO_CACHE) &&
1075 !router_supports_extrainfo(status->identity_digest, 0))
1076 continue;
1078 /* XXXX IP6 proposal 118 */
1079 tor_addr_from_ipv4h(&addr, status->addr);
1081 if (prefer_tunnel &&
1082 status->version_supports_begindir &&
1083 (!fascistfirewall ||
1084 fascist_firewall_allows_address_or(&addr, status->or_port)))
1085 smartlist_add(is_trusted ? trusted_tunnel :
1086 is_overloaded ? overloaded_tunnel : tunnel, status);
1087 else if (!fascistfirewall ||
1088 fascist_firewall_allows_address_dir(&addr, status->dir_port))
1089 smartlist_add(is_trusted ? trusted_direct :
1090 is_overloaded ? overloaded_direct : direct, status);
1091 } SMARTLIST_FOREACH_END(status);
1093 if (smartlist_len(tunnel)) {
1094 result = routerstatus_sl_choose_by_bandwidth(tunnel);
1095 } else if (smartlist_len(overloaded_tunnel)) {
1096 result = routerstatus_sl_choose_by_bandwidth(overloaded_tunnel);
1097 } else if (smartlist_len(trusted_tunnel)) {
1098 /* FFFF We don't distinguish between trusteds and overloaded trusteds
1099 * yet. Maybe one day we should. */
1100 /* FFFF We also don't load balance over authorities yet. I think this
1101 * is a feature, but it could easily be a bug. -RD */
1102 result = smartlist_choose(trusted_tunnel);
1103 } else if (smartlist_len(direct)) {
1104 result = routerstatus_sl_choose_by_bandwidth(direct);
1105 } else if (smartlist_len(overloaded_direct)) {
1106 result = routerstatus_sl_choose_by_bandwidth(overloaded_direct);
1107 } else {
1108 result = smartlist_choose(trusted_direct);
1110 smartlist_free(direct);
1111 smartlist_free(tunnel);
1112 smartlist_free(trusted_direct);
1113 smartlist_free(trusted_tunnel);
1114 smartlist_free(overloaded_direct);
1115 smartlist_free(overloaded_tunnel);
1116 return result;
1119 /** Choose randomly from among the trusted dirservers that are up. Flags
1120 * are as for router_pick_directory_server_impl().
1122 static routerstatus_t *
1123 router_pick_trusteddirserver_impl(authority_type_t type, int flags,
1124 int *n_busy_out)
1126 smartlist_t *direct, *tunnel;
1127 smartlist_t *overloaded_direct, *overloaded_tunnel;
1128 routerinfo_t *me = router_get_my_routerinfo();
1129 routerstatus_t *result;
1130 time_t now = time(NULL);
1131 const int requireother = ! (flags & PDS_ALLOW_SELF);
1132 const int fascistfirewall = ! (flags & PDS_IGNORE_FASCISTFIREWALL);
1133 const int prefer_tunnel = (flags & _PDS_PREFER_TUNNELED_DIR_CONNS);
1134 const int no_serverdesc_fetching =(flags & PDS_NO_EXISTING_SERVERDESC_FETCH);
1135 int n_busy = 0;
1137 if (!trusted_dir_servers)
1138 return NULL;
1140 direct = smartlist_create();
1141 tunnel = smartlist_create();
1142 overloaded_direct = smartlist_create();
1143 overloaded_tunnel = smartlist_create();
1145 SMARTLIST_FOREACH_BEGIN(trusted_dir_servers, trusted_dir_server_t *, d)
1147 int is_overloaded =
1148 d->fake_status.last_dir_503_at + DIR_503_TIMEOUT > now;
1149 tor_addr_t addr;
1150 if (!d->is_running) continue;
1151 if ((type & d->type) == 0)
1152 continue;
1153 if ((type & EXTRAINFO_CACHE) &&
1154 !router_supports_extrainfo(d->digest, 1))
1155 continue;
1156 if (requireother && me && router_digest_is_me(d->digest))
1157 continue;
1159 /* XXXX IP6 proposal 118 */
1160 tor_addr_from_ipv4h(&addr, d->addr);
1162 if (no_serverdesc_fetching) {
1163 if (connection_get_by_type_addr_port_purpose(
1164 CONN_TYPE_DIR, &addr, d->dir_port, DIR_PURPOSE_FETCH_SERVERDESC)
1165 || connection_get_by_type_addr_port_purpose(
1166 CONN_TYPE_DIR, &addr, d->dir_port, DIR_PURPOSE_FETCH_EXTRAINFO)) {
1167 //log_debug(LD_DIR, "We have an existing connection to fetch "
1168 // "descriptor from %s; delaying",d->description);
1169 ++n_busy;
1170 continue;
1174 if (prefer_tunnel &&
1175 d->or_port &&
1176 (!fascistfirewall ||
1177 fascist_firewall_allows_address_or(&addr, d->or_port)))
1178 smartlist_add(is_overloaded ? overloaded_tunnel : tunnel,
1179 &d->fake_status);
1180 else if (!fascistfirewall ||
1181 fascist_firewall_allows_address_dir(&addr, d->dir_port))
1182 smartlist_add(is_overloaded ? overloaded_direct : direct,
1183 &d->fake_status);
1185 SMARTLIST_FOREACH_END(d);
1187 if (smartlist_len(tunnel)) {
1188 result = smartlist_choose(tunnel);
1189 } else if (smartlist_len(overloaded_tunnel)) {
1190 result = smartlist_choose(overloaded_tunnel);
1191 } else if (smartlist_len(direct)) {
1192 result = smartlist_choose(direct);
1193 } else {
1194 result = smartlist_choose(overloaded_direct);
1197 if (n_busy_out)
1198 *n_busy_out = n_busy;
1200 smartlist_free(direct);
1201 smartlist_free(tunnel);
1202 smartlist_free(overloaded_direct);
1203 smartlist_free(overloaded_tunnel);
1204 return result;
1207 /** Go through and mark the authoritative dirservers as up. */
1208 static void
1209 mark_all_trusteddirservers_up(void)
1211 if (routerlist) {
1212 SMARTLIST_FOREACH(routerlist->routers, routerinfo_t *, router,
1213 if (router_digest_is_trusted_dir(router->cache_info.identity_digest) &&
1214 router->dir_port > 0) {
1215 router->is_running = 1;
1218 if (trusted_dir_servers) {
1219 SMARTLIST_FOREACH(trusted_dir_servers, trusted_dir_server_t *, dir,
1221 routerstatus_t *rs;
1222 dir->is_running = 1;
1223 download_status_reset(&dir->v2_ns_dl_status);
1224 rs = router_get_consensus_status_by_id(dir->digest);
1225 if (rs && !rs->is_running) {
1226 rs->is_running = 1;
1227 rs->last_dir_503_at = 0;
1228 control_event_networkstatus_changed_single(rs);
1232 router_dir_info_changed();
1235 /** Reset all internal variables used to count failed downloads of network
1236 * status objects. */
1237 void
1238 router_reset_status_download_failures(void)
1240 mark_all_trusteddirservers_up();
1243 /** Return true iff router1 and router2 have the same /16 network. */
1244 static INLINE int
1245 routers_in_same_network_family(routerinfo_t *r1, routerinfo_t *r2)
1247 return (r1->addr & 0xffff0000) == (r2->addr & 0xffff0000);
1250 /** Look through the routerlist and identify routers that
1251 * advertise the same /16 network address as <b>router</b>.
1252 * Add each of them to <b>sl</b>.
1254 static void
1255 routerlist_add_network_family(smartlist_t *sl, routerinfo_t *router)
1257 SMARTLIST_FOREACH(routerlist->routers, routerinfo_t *, r,
1259 if (router != r && routers_in_same_network_family(router, r))
1260 smartlist_add(sl, r);
1264 /** Add all the family of <b>router</b> to the smartlist <b>sl</b>.
1265 * This is used to make sure we don't pick siblings in a single path,
1266 * or pick more than one relay from a family for our entry guard list.
1268 void
1269 routerlist_add_family(smartlist_t *sl, routerinfo_t *router)
1271 routerinfo_t *r;
1272 config_line_t *cl;
1273 or_options_t *options = get_options();
1275 /* First, add any routers with similar network addresses. */
1276 if (options->EnforceDistinctSubnets)
1277 routerlist_add_network_family(sl, router);
1279 if (router->declared_family) {
1280 /* Add every r such that router declares familyness with r, and r
1281 * declares familyhood with router. */
1282 SMARTLIST_FOREACH(router->declared_family, const char *, n,
1284 if (!(r = router_get_by_nickname(n, 0)))
1285 continue;
1286 if (!r->declared_family)
1287 continue;
1288 SMARTLIST_FOREACH(r->declared_family, const char *, n2,
1290 if (router_nickname_matches(router, n2))
1291 smartlist_add(sl, r);
1296 /* If the user declared any families locally, honor those too. */
1297 for (cl = options->NodeFamilies; cl; cl = cl->next) {
1298 if (router_nickname_is_in_list(router, cl->value)) {
1299 add_nickname_list_to_smartlist(sl, cl->value, 0);
1304 /** Return true iff r is named by some nickname in <b>lst</b>. */
1305 static INLINE int
1306 router_in_nickname_smartlist(smartlist_t *lst, routerinfo_t *r)
1308 if (!lst) return 0;
1309 SMARTLIST_FOREACH(lst, const char *, name,
1310 if (router_nickname_matches(r, name))
1311 return 1;);
1312 return 0;
1315 /** Return true iff r1 and r2 are in the same family, but not the same
1316 * router. */
1318 routers_in_same_family(routerinfo_t *r1, routerinfo_t *r2)
1320 or_options_t *options = get_options();
1321 config_line_t *cl;
1323 if (options->EnforceDistinctSubnets && routers_in_same_network_family(r1,r2))
1324 return 1;
1326 if (router_in_nickname_smartlist(r1->declared_family, r2) &&
1327 router_in_nickname_smartlist(r2->declared_family, r1))
1328 return 1;
1330 for (cl = options->NodeFamilies; cl; cl = cl->next) {
1331 if (router_nickname_is_in_list(r1, cl->value) &&
1332 router_nickname_is_in_list(r2, cl->value))
1333 return 1;
1335 return 0;
1338 /** Given a (possibly NULL) comma-and-whitespace separated list of nicknames,
1339 * see which nicknames in <b>list</b> name routers in our routerlist, and add
1340 * the routerinfos for those routers to <b>sl</b>. If <b>must_be_running</b>,
1341 * only include routers that we think are running.
1342 * Warn if any non-Named routers are specified by nickname.
1344 void
1345 add_nickname_list_to_smartlist(smartlist_t *sl, const char *list,
1346 int must_be_running)
1348 routerinfo_t *router;
1349 smartlist_t *nickname_list;
1350 int have_dir_info = router_have_minimum_dir_info();
1352 if (!list)
1353 return; /* nothing to do */
1354 tor_assert(sl);
1356 nickname_list = smartlist_create();
1357 if (!warned_nicknames)
1358 warned_nicknames = smartlist_create();
1360 smartlist_split_string(nickname_list, list, ",",
1361 SPLIT_SKIP_SPACE|SPLIT_IGNORE_BLANK, 0);
1363 SMARTLIST_FOREACH(nickname_list, const char *, nick, {
1364 int warned;
1365 if (!is_legal_nickname_or_hexdigest(nick)) {
1366 log_warn(LD_CONFIG, "Nickname '%s' is misformed; skipping", nick);
1367 continue;
1369 router = router_get_by_nickname(nick, 1);
1370 warned = smartlist_string_isin(warned_nicknames, nick);
1371 if (router) {
1372 if (!must_be_running || router->is_running) {
1373 smartlist_add(sl,router);
1375 } else if (!router_get_consensus_status_by_nickname(nick,1)) {
1376 if (!warned) {
1377 log_fn(have_dir_info ? LOG_WARN : LOG_INFO, LD_CONFIG,
1378 "Nickname list includes '%s' which isn't a known router.",nick);
1379 smartlist_add(warned_nicknames, tor_strdup(nick));
1383 SMARTLIST_FOREACH(nickname_list, char *, nick, tor_free(nick));
1384 smartlist_free(nickname_list);
1387 /** Return 1 iff any member of the (possibly NULL) comma-separated list
1388 * <b>list</b> is an acceptable nickname or hexdigest for <b>router</b>. Else
1389 * return 0.
1392 router_nickname_is_in_list(routerinfo_t *router, const char *list)
1394 smartlist_t *nickname_list;
1395 int v = 0;
1397 if (!list)
1398 return 0; /* definitely not */
1399 tor_assert(router);
1401 nickname_list = smartlist_create();
1402 smartlist_split_string(nickname_list, list, ",",
1403 SPLIT_SKIP_SPACE|SPLIT_STRIP_SPACE|SPLIT_IGNORE_BLANK, 0);
1404 SMARTLIST_FOREACH(nickname_list, const char *, cp,
1405 if (router_nickname_matches(router, cp)) {v=1;break;});
1406 SMARTLIST_FOREACH(nickname_list, char *, cp, tor_free(cp));
1407 smartlist_free(nickname_list);
1408 return v;
1411 /** Add every suitable router from our routerlist to <b>sl</b>, so that
1412 * we can pick a node for a circuit.
1414 static void
1415 router_add_running_routers_to_smartlist(smartlist_t *sl, int allow_invalid,
1416 int need_uptime, int need_capacity,
1417 int need_guard)
1419 if (!routerlist)
1420 return;
1422 SMARTLIST_FOREACH(routerlist->routers, routerinfo_t *, router,
1424 if (router->is_running &&
1425 router->purpose == ROUTER_PURPOSE_GENERAL &&
1426 (router->is_valid || allow_invalid) &&
1427 !router_is_unreliable(router, need_uptime,
1428 need_capacity, need_guard)) {
1429 /* If it's running, and it's suitable according to the
1430 * other flags we had in mind */
1431 smartlist_add(sl, router);
1436 /** Look through the routerlist until we find a router that has my key.
1437 Return it. */
1438 routerinfo_t *
1439 routerlist_find_my_routerinfo(void)
1441 if (!routerlist)
1442 return NULL;
1444 SMARTLIST_FOREACH(routerlist->routers, routerinfo_t *, router,
1446 if (router_is_me(router))
1447 return router;
1449 return NULL;
1452 /** Find a router that's up, that has this IP address, and
1453 * that allows exit to this address:port, or return NULL if there
1454 * isn't a good one.
1456 routerinfo_t *
1457 router_find_exact_exit_enclave(const char *address, uint16_t port)
1459 uint32_t addr;
1460 struct in_addr in;
1461 tor_addr_t a;
1463 if (!tor_inet_aton(address, &in))
1464 return NULL; /* it's not an IP already */
1465 addr = ntohl(in.s_addr);
1467 tor_addr_from_ipv4h(&a, addr);
1469 SMARTLIST_FOREACH(routerlist->routers, routerinfo_t *, router,
1471 if (router->addr == addr &&
1472 router->is_running &&
1473 compare_tor_addr_to_addr_policy(&a, port, router->exit_policy) ==
1474 ADDR_POLICY_ACCEPTED)
1475 return router;
1477 return NULL;
1480 /** Return 1 if <b>router</b> is not suitable for these parameters, else 0.
1481 * If <b>need_uptime</b> is non-zero, we require a minimum uptime.
1482 * If <b>need_capacity</b> is non-zero, we require a minimum advertised
1483 * bandwidth.
1484 * If <b>need_guard</b>, we require that the router is a possible entry guard.
1487 router_is_unreliable(routerinfo_t *router, int need_uptime,
1488 int need_capacity, int need_guard)
1490 if (need_uptime && !router->is_stable)
1491 return 1;
1492 if (need_capacity && !router->is_fast)
1493 return 1;
1494 if (need_guard && !router->is_possible_guard)
1495 return 1;
1496 return 0;
1499 /** Return the smaller of the router's configured BandwidthRate
1500 * and its advertised capacity. */
1501 uint32_t
1502 router_get_advertised_bandwidth(routerinfo_t *router)
1504 if (router->bandwidthcapacity < router->bandwidthrate)
1505 return router->bandwidthcapacity;
1506 return router->bandwidthrate;
1509 /** Do not weight any declared bandwidth more than this much when picking
1510 * routers by bandwidth. */
1511 #define DEFAULT_MAX_BELIEVABLE_BANDWIDTH 10000000 /* 10 MB/sec */
1513 /** Return the smaller of the router's configured BandwidthRate
1514 * and its advertised capacity, capped by max-believe-bw. */
1515 uint32_t
1516 router_get_advertised_bandwidth_capped(routerinfo_t *router)
1518 uint32_t result = router->bandwidthcapacity;
1519 if (result > router->bandwidthrate)
1520 result = router->bandwidthrate;
1521 if (result > DEFAULT_MAX_BELIEVABLE_BANDWIDTH)
1522 result = DEFAULT_MAX_BELIEVABLE_BANDWIDTH;
1523 return result;
1526 /** Eventually, the number we return will come from the directory
1527 * consensus, so clients can dynamically update to better numbers.
1529 * But for now, or in case there is no consensus available, just return
1530 * a sufficient default. */
1531 static uint32_t
1532 get_max_believable_bandwidth(void)
1534 return DEFAULT_MAX_BELIEVABLE_BANDWIDTH;
1537 /** Helper function:
1538 * choose a random element of smartlist <b>sl</b>, weighted by
1539 * the advertised bandwidth of each element.
1541 * If <b>statuses</b> is zero, then <b>sl</b> is a list of
1542 * routerinfo_t's. Otherwise it's a list of routerstatus_t's.
1544 * If <b>rule</b>==WEIGHT_FOR_EXIT. we're picking an exit node: consider all
1545 * nodes' bandwidth equally regardless of their Exit status, since there may
1546 * be some in the list because they exit to obscure ports. If
1547 * <b>rule</b>==NO_WEIGHTING, we're picking a non-exit node: weight
1548 * exit-node's bandwidth less depending on the smallness of the fraction of
1549 * Exit-to-total bandwidth. If <b>rule</b>==WEIGHT_FOR_GUARD, we're picking a
1550 * guard node: consider all guard's bandwidth equally. Otherwise, weight
1551 * guards proportionally less.
1553 static void *
1554 smartlist_choose_by_bandwidth(smartlist_t *sl, bandwidth_weight_rule_t rule,
1555 int statuses)
1557 unsigned int i;
1558 routerinfo_t *router;
1559 routerstatus_t *status=NULL;
1560 int32_t *bandwidths;
1561 int is_exit;
1562 int is_guard;
1563 uint64_t total_nonexit_bw = 0, total_exit_bw = 0, total_bw = 0;
1564 uint64_t total_nonguard_bw = 0, total_guard_bw = 0;
1565 uint64_t rand_bw, tmp;
1566 double exit_weight;
1567 double guard_weight;
1568 int n_unknown = 0;
1569 bitarray_t *exit_bits;
1570 bitarray_t *guard_bits;
1571 uint32_t max_believable_bw = get_max_believable_bandwidth();
1572 int me_idx = -1;
1574 /* Can't choose exit and guard at same time */
1575 tor_assert(rule == NO_WEIGHTING ||
1576 rule == WEIGHT_FOR_EXIT ||
1577 rule == WEIGHT_FOR_GUARD);
1579 /* First count the total bandwidth weight, and make a list
1580 * of each value. <0 means "unknown; no routerinfo." We use the
1581 * bits of negative values to remember whether the router was fast (-x)&1
1582 * and whether it was an exit (-x)&2 or guard (-x)&4. Yes, it's a hack. */
1583 bandwidths = tor_malloc(sizeof(int32_t)*smartlist_len(sl));
1584 exit_bits = bitarray_init_zero(smartlist_len(sl));
1585 guard_bits = bitarray_init_zero(smartlist_len(sl));
1587 /* Iterate over all the routerinfo_t or routerstatus_t, and */
1588 for (i = 0; i < (unsigned)smartlist_len(sl); ++i) {
1589 /* first, learn what bandwidth we think i has */
1590 int is_known = 1;
1591 int32_t flags = 0;
1592 uint32_t this_bw = 0;
1593 if (statuses) {
1594 /* need to extract router info */
1595 status = smartlist_get(sl, i);
1596 if (router_digest_is_me(status->identity_digest))
1597 me_idx = i;
1598 router = router_get_by_digest(status->identity_digest);
1599 is_exit = status->is_exit;
1600 is_guard = status->is_possible_guard;
1601 if (router) {
1602 this_bw = router_get_advertised_bandwidth(router);
1603 } else { /* guess */
1604 is_known = 0;
1605 flags = status->is_fast ? 1 : 0;
1606 flags |= is_exit ? 2 : 0;
1607 flags |= is_guard ? 4 : 0;
1609 } else {
1610 router = smartlist_get(sl, i);
1611 if (router_digest_is_me(router->cache_info.identity_digest))
1612 me_idx = i;
1613 is_exit = router->is_exit;
1614 is_guard = router->is_possible_guard;
1615 this_bw = router_get_advertised_bandwidth(router);
1617 if (is_exit)
1618 bitarray_set(exit_bits, i);
1619 if (is_guard)
1620 bitarray_set(guard_bits, i);
1621 /* if they claim something huge, don't believe it */
1622 if (this_bw > max_believable_bw) {
1623 char fp[HEX_DIGEST_LEN+1];
1624 base16_encode(fp, sizeof(fp), statuses ?
1625 status->identity_digest :
1626 router->cache_info.identity_digest,
1627 DIGEST_LEN);
1628 log_fn(LOG_PROTOCOL_WARN, LD_DIR,
1629 "Bandwidth %d for router %s (%s) exceeds allowed max %d, capping",
1630 this_bw, router ? router->nickname : "(null)",
1631 fp, max_believable_bw);
1632 this_bw = max_believable_bw;
1634 if (is_known) {
1635 bandwidths[i] = (int32_t) this_bw; // safe since MAX_BELIEVABLE<INT32_MAX
1636 if (is_guard)
1637 total_guard_bw += this_bw;
1638 else
1639 total_nonguard_bw += this_bw;
1640 if (is_exit)
1641 total_exit_bw += this_bw;
1642 else
1643 total_nonexit_bw += this_bw;
1644 } else {
1645 ++n_unknown;
1646 bandwidths[i] = -flags;
1650 /* Now, fill in the unknown values. */
1651 if (n_unknown) {
1652 int32_t avg_fast, avg_slow;
1653 if (total_exit_bw+total_nonexit_bw) {
1654 /* if there's some bandwidth, there's at least one known router,
1655 * so no worries about div by 0 here */
1656 int n_known = smartlist_len(sl)-n_unknown;
1657 avg_fast = avg_slow = (int32_t)
1658 ((total_exit_bw+total_nonexit_bw)/((uint64_t) n_known));
1659 } else {
1660 avg_fast = 40000;
1661 avg_slow = 20000;
1663 for (i=0; i<(unsigned)smartlist_len(sl); ++i) {
1664 int32_t bw = bandwidths[i];
1665 if (bw>=0)
1666 continue;
1667 is_exit = ((-bw)&2);
1668 is_guard = ((-bw)&4);
1669 bandwidths[i] = ((-bw)&1) ? avg_fast : avg_slow;
1670 if (is_exit)
1671 total_exit_bw += bandwidths[i];
1672 else
1673 total_nonexit_bw += bandwidths[i];
1674 if (is_guard)
1675 total_guard_bw += bandwidths[i];
1676 else
1677 total_nonguard_bw += bandwidths[i];
1681 /* If there's no bandwidth at all, pick at random. */
1682 if (!(total_exit_bw+total_nonexit_bw)) {
1683 tor_free(bandwidths);
1684 tor_free(exit_bits);
1685 tor_free(guard_bits);
1686 return smartlist_choose(sl);
1689 /* Figure out how to weight exits and guards */
1691 double all_bw = U64_TO_DBL(total_exit_bw+total_nonexit_bw);
1692 double exit_bw = U64_TO_DBL(total_exit_bw);
1693 double guard_bw = U64_TO_DBL(total_guard_bw);
1695 * For detailed derivation of this formula, see
1696 * http://archives.seul.org/or/dev/Jul-2007/msg00056.html
1698 if (rule == WEIGHT_FOR_EXIT)
1699 exit_weight = 1.0;
1700 else
1701 exit_weight = 1.0 - all_bw/(3.0*exit_bw);
1703 if (rule == WEIGHT_FOR_GUARD)
1704 guard_weight = 1.0;
1705 else
1706 guard_weight = 1.0 - all_bw/(3.0*guard_bw);
1708 if (exit_weight <= 0.0)
1709 exit_weight = 0.0;
1711 if (guard_weight <= 0.0)
1712 guard_weight = 0.0;
1714 total_bw = 0;
1715 sl_last_weighted_bw_of_me = 0;
1716 for (i=0; i < (unsigned)smartlist_len(sl); i++) {
1717 uint64_t bw;
1718 is_exit = bitarray_is_set(exit_bits, i);
1719 is_guard = bitarray_is_set(guard_bits, i);
1720 if (is_exit && is_guard)
1721 bw = ((uint64_t)(bandwidths[i] * exit_weight * guard_weight));
1722 else if (is_guard)
1723 bw = ((uint64_t)(bandwidths[i] * guard_weight));
1724 else if (is_exit)
1725 bw = ((uint64_t)(bandwidths[i] * exit_weight));
1726 else
1727 bw = bandwidths[i];
1728 total_bw += bw;
1729 if (i == (unsigned) me_idx)
1730 sl_last_weighted_bw_of_me = bw;
1734 /* XXXX022 this is a kludge to expose these values. */
1735 sl_last_total_weighted_bw = total_bw;
1737 log_debug(LD_CIRC, "Total weighted bw = "U64_FORMAT
1738 ", exit bw = "U64_FORMAT
1739 ", nonexit bw = "U64_FORMAT", exit weight = %lf "
1740 "(for exit == %d)"
1741 ", guard bw = "U64_FORMAT
1742 ", nonguard bw = "U64_FORMAT", guard weight = %lf "
1743 "(for guard == %d)",
1744 U64_PRINTF_ARG(total_bw),
1745 U64_PRINTF_ARG(total_exit_bw), U64_PRINTF_ARG(total_nonexit_bw),
1746 exit_weight, (int)(rule == WEIGHT_FOR_EXIT),
1747 U64_PRINTF_ARG(total_guard_bw), U64_PRINTF_ARG(total_nonguard_bw),
1748 guard_weight, (int)(rule == WEIGHT_FOR_GUARD));
1750 /* Almost done: choose a random value from the bandwidth weights. */
1751 rand_bw = crypto_rand_uint64(total_bw);
1753 /* Last, count through sl until we get to the element we picked */
1754 tmp = 0;
1755 for (i=0; i < (unsigned)smartlist_len(sl); i++) {
1756 is_exit = bitarray_is_set(exit_bits, i);
1757 is_guard = bitarray_is_set(guard_bits, i);
1759 /* Weights can be 0 if not counting guards/exits */
1760 if (is_exit && is_guard)
1761 tmp += ((uint64_t)(bandwidths[i] * exit_weight * guard_weight));
1762 else if (is_guard)
1763 tmp += ((uint64_t)(bandwidths[i] * guard_weight));
1764 else if (is_exit)
1765 tmp += ((uint64_t)(bandwidths[i] * exit_weight));
1766 else
1767 tmp += bandwidths[i];
1769 if (tmp >= rand_bw)
1770 break;
1772 if (i == (unsigned)smartlist_len(sl)) {
1773 /* This was once possible due to round-off error, but shouldn't be able
1774 * to occur any longer. */
1775 tor_fragile_assert();
1776 --i;
1777 log_warn(LD_BUG, "Round-off error in computing bandwidth had an effect on "
1778 " which router we chose. Please tell the developers. "
1779 U64_FORMAT " " U64_FORMAT " " U64_FORMAT, U64_PRINTF_ARG(tmp),
1780 U64_PRINTF_ARG(rand_bw), U64_PRINTF_ARG(total_bw));
1782 tor_free(bandwidths);
1783 tor_free(exit_bits);
1784 tor_free(guard_bits);
1785 return smartlist_get(sl, i);
1788 /** Choose a random element of router list <b>sl</b>, weighted by
1789 * the advertised bandwidth of each router.
1791 routerinfo_t *
1792 routerlist_sl_choose_by_bandwidth(smartlist_t *sl,
1793 bandwidth_weight_rule_t rule)
1795 return smartlist_choose_by_bandwidth(sl, rule, 0);
1798 /** Choose a random element of status list <b>sl</b>, weighted by
1799 * the advertised bandwidth of each status.
1801 routerstatus_t *
1802 routerstatus_sl_choose_by_bandwidth(smartlist_t *sl)
1804 /* We are choosing neither exit nor guard here. Weight accordingly. */
1805 return smartlist_choose_by_bandwidth(sl, NO_WEIGHTING, 1);
1808 /** Return a random running router from the routerlist. If any node
1809 * named in <b>preferred</b> is available, pick one of those. Never
1810 * pick a node whose routerinfo is in
1811 * <b>excludedsmartlist</b>, or whose routerinfo matches <b>excludedset</b>,
1812 * even if they are the only nodes
1813 * available. If <b>CRN_STRICT_PREFERRED</b> is set in flags, never pick
1814 * any node besides those in <b>preferred</b>.
1815 * If <b>CRN_NEED_UPTIME</b> is set in flags and any router has more than
1816 * a minimum uptime, return one of those.
1817 * If <b>CRN_NEED_CAPACITY</b> is set in flags, weight your choice by the
1818 * advertised capacity of each router.
1819 * If <b>CRN_ALLOW_INVALID</b> is not set in flags, consider only Valid
1820 * routers.
1821 * If <b>CRN_NEED_GUARD</b> is set in flags, consider only Guard routers.
1822 * If <b>CRN_WEIGHT_AS_EXIT</b> is set in flags, we weight bandwidths as if
1823 * picking an exit node, otherwise we weight bandwidths for picking a relay
1824 * node (that is, possibly discounting exit nodes).
1826 routerinfo_t *
1827 router_choose_random_node(const char *preferred,
1828 smartlist_t *excludedsmartlist,
1829 routerset_t *excludedset,
1830 router_crn_flags_t flags)
1832 const int need_uptime = (flags & CRN_NEED_UPTIME) != 0;
1833 const int need_capacity = (flags & CRN_NEED_CAPACITY) != 0;
1834 const int need_guard = (flags & CRN_NEED_GUARD) != 0;
1835 const int allow_invalid = (flags & CRN_ALLOW_INVALID) != 0;
1836 const int strict = (flags & CRN_STRICT_PREFERRED) != 0;
1837 const int weight_for_exit = (flags & CRN_WEIGHT_AS_EXIT) != 0;
1839 smartlist_t *sl, *excludednodes;
1840 routerinfo_t *choice = NULL, *r;
1841 bandwidth_weight_rule_t rule;
1843 tor_assert(!(weight_for_exit && need_guard));
1844 rule = weight_for_exit ? WEIGHT_FOR_EXIT :
1845 (need_guard ? WEIGHT_FOR_GUARD : NO_WEIGHTING);
1847 excludednodes = smartlist_create();
1849 /* Exclude relays that allow single hop exit circuits, if the user
1850 * wants to (such relays might be risky) */
1851 if (get_options()->ExcludeSingleHopRelays) {
1852 routerlist_t *rl = router_get_routerlist();
1853 SMARTLIST_FOREACH(rl->routers, routerinfo_t *, r,
1854 if (r->allow_single_hop_exits) {
1855 smartlist_add(excludednodes, r);
1859 if ((r = routerlist_find_my_routerinfo())) {
1860 smartlist_add(excludednodes, r);
1861 routerlist_add_family(excludednodes, r);
1864 /* Try the preferred nodes first. Ignore need_uptime and need_capacity
1865 * and need_guard, since the user explicitly asked for these nodes. */
1866 if (preferred) {
1867 sl = smartlist_create();
1868 add_nickname_list_to_smartlist(sl,preferred,1);
1869 smartlist_subtract(sl,excludednodes);
1870 if (excludedsmartlist)
1871 smartlist_subtract(sl,excludedsmartlist);
1872 if (excludedset)
1873 routerset_subtract_routers(sl,excludedset);
1874 choice = smartlist_choose(sl);
1875 smartlist_free(sl);
1877 if (!choice && !strict) {
1878 /* Then give up on our preferred choices: any node
1879 * will do that has the required attributes. */
1880 sl = smartlist_create();
1881 router_add_running_routers_to_smartlist(sl, allow_invalid,
1882 need_uptime, need_capacity,
1883 need_guard);
1884 smartlist_subtract(sl,excludednodes);
1885 if (excludedsmartlist)
1886 smartlist_subtract(sl,excludedsmartlist);
1887 if (excludedset)
1888 routerset_subtract_routers(sl,excludedset);
1890 if (need_capacity || need_guard)
1891 choice = routerlist_sl_choose_by_bandwidth(sl, rule);
1892 else
1893 choice = smartlist_choose(sl);
1895 smartlist_free(sl);
1896 if (!choice && (need_uptime || need_capacity || need_guard)) {
1897 /* try once more -- recurse but with fewer restrictions. */
1898 log_info(LD_CIRC,
1899 "We couldn't find any live%s%s%s routers; falling back "
1900 "to list of all routers.",
1901 need_capacity?", fast":"",
1902 need_uptime?", stable":"",
1903 need_guard?", guard":"");
1904 flags &= ~ (CRN_NEED_UPTIME|CRN_NEED_CAPACITY|CRN_NEED_GUARD);
1905 choice = router_choose_random_node(
1906 NULL, excludedsmartlist, excludedset, flags);
1909 smartlist_free(excludednodes);
1910 if (!choice) {
1911 if (strict) {
1912 log_warn(LD_CIRC, "All preferred nodes were down when trying to choose "
1913 "node, and the Strict[...]Nodes option is set. Failing.");
1914 } else {
1915 log_warn(LD_CIRC,
1916 "No available nodes when trying to choose node. Failing.");
1919 return choice;
1922 /** Helper: Return true iff the <b>identity_digest</b> and <b>nickname</b>
1923 * combination of a router, encoded in hexadecimal, matches <b>hexdigest</b>
1924 * (which is optionally prefixed with a single dollar sign). Return false if
1925 * <b>hexdigest</b> is malformed, or it doesn't match. */
1926 static INLINE int
1927 hex_digest_matches(const char *hexdigest, const char *identity_digest,
1928 const char *nickname, int is_named)
1930 char digest[DIGEST_LEN];
1931 size_t len;
1932 tor_assert(hexdigest);
1933 if (hexdigest[0] == '$')
1934 ++hexdigest;
1936 len = strlen(hexdigest);
1937 if (len < HEX_DIGEST_LEN)
1938 return 0;
1939 else if (len > HEX_DIGEST_LEN &&
1940 (hexdigest[HEX_DIGEST_LEN] == '=' ||
1941 hexdigest[HEX_DIGEST_LEN] == '~')) {
1942 if (strcasecmp(hexdigest+HEX_DIGEST_LEN+1, nickname))
1943 return 0;
1944 if (hexdigest[HEX_DIGEST_LEN] == '=' && !is_named)
1945 return 0;
1948 if (base16_decode(digest, DIGEST_LEN, hexdigest, HEX_DIGEST_LEN)<0)
1949 return 0;
1950 return (!memcmp(digest, identity_digest, DIGEST_LEN));
1953 /** Return true iff the digest of <b>router</b>'s identity key,
1954 * encoded in hexadecimal, matches <b>hexdigest</b> (which is
1955 * optionally prefixed with a single dollar sign). Return false if
1956 * <b>hexdigest</b> is malformed, or it doesn't match. */
1957 static INLINE int
1958 router_hex_digest_matches(routerinfo_t *router, const char *hexdigest)
1960 return hex_digest_matches(hexdigest, router->cache_info.identity_digest,
1961 router->nickname, router->is_named);
1964 /** Return true if <b>router</b>'s nickname matches <b>nickname</b>
1965 * (case-insensitive), or if <b>router's</b> identity key digest
1966 * matches a hexadecimal value stored in <b>nickname</b>. Return
1967 * false otherwise. */
1968 static int
1969 router_nickname_matches(routerinfo_t *router, const char *nickname)
1971 if (nickname[0]!='$' && !strcasecmp(router->nickname, nickname))
1972 return 1;
1973 return router_hex_digest_matches(router, nickname);
1976 /** Return the router in our routerlist whose (case-insensitive)
1977 * nickname or (case-sensitive) hexadecimal key digest is
1978 * <b>nickname</b>. Return NULL if no such router is known.
1980 routerinfo_t *
1981 router_get_by_nickname(const char *nickname, int warn_if_unnamed)
1983 int maybedigest;
1984 char digest[DIGEST_LEN];
1985 routerinfo_t *best_match=NULL;
1986 int n_matches = 0;
1987 const char *named_digest = NULL;
1989 tor_assert(nickname);
1990 if (!routerlist)
1991 return NULL;
1992 if (nickname[0] == '$')
1993 return router_get_by_hexdigest(nickname);
1994 if (!strcasecmp(nickname, UNNAMED_ROUTER_NICKNAME))
1995 return NULL;
1996 if (server_mode(get_options()) &&
1997 !strcasecmp(nickname, get_options()->Nickname))
1998 return router_get_my_routerinfo();
2000 maybedigest = (strlen(nickname) >= HEX_DIGEST_LEN) &&
2001 (base16_decode(digest,DIGEST_LEN,nickname,HEX_DIGEST_LEN) == 0);
2003 if ((named_digest = networkstatus_get_router_digest_by_nickname(nickname))) {
2004 return rimap_get(routerlist->identity_map, named_digest);
2006 if (networkstatus_nickname_is_unnamed(nickname))
2007 return NULL;
2009 /* If we reach this point, there's no canonical value for the nickname. */
2011 SMARTLIST_FOREACH(routerlist->routers, routerinfo_t *, router,
2013 if (!strcasecmp(router->nickname, nickname)) {
2014 ++n_matches;
2015 if (n_matches <= 1 || router->is_running)
2016 best_match = router;
2017 } else if (maybedigest &&
2018 !memcmp(digest, router->cache_info.identity_digest, DIGEST_LEN)
2020 if (router_hex_digest_matches(router, nickname))
2021 return router;
2022 /* If we reach this point, we have a ID=name syntax that matches the
2023 * identity but not the name. That isn't an acceptable match. */
2027 if (best_match) {
2028 if (warn_if_unnamed && n_matches > 1) {
2029 smartlist_t *fps = smartlist_create();
2030 int any_unwarned = 0;
2031 SMARTLIST_FOREACH(routerlist->routers, routerinfo_t *, router,
2033 routerstatus_t *rs;
2034 char *desc;
2035 size_t dlen;
2036 char fp[HEX_DIGEST_LEN+1];
2037 if (strcasecmp(router->nickname, nickname))
2038 continue;
2039 rs = router_get_consensus_status_by_id(
2040 router->cache_info.identity_digest);
2041 if (rs && !rs->name_lookup_warned) {
2042 rs->name_lookup_warned = 1;
2043 any_unwarned = 1;
2045 base16_encode(fp, sizeof(fp),
2046 router->cache_info.identity_digest, DIGEST_LEN);
2047 dlen = 32 + HEX_DIGEST_LEN + strlen(router->address);
2048 desc = tor_malloc(dlen);
2049 tor_snprintf(desc, dlen, "\"$%s\" for the one at %s:%d",
2050 fp, router->address, router->or_port);
2051 smartlist_add(fps, desc);
2053 if (any_unwarned) {
2054 char *alternatives = smartlist_join_strings(fps, "; ",0,NULL);
2055 log_warn(LD_CONFIG,
2056 "There are multiple matches for the nickname \"%s\","
2057 " but none is listed as named by the directory authorities. "
2058 "Choosing one arbitrarily. If you meant one in particular, "
2059 "you should say %s.", nickname, alternatives);
2060 tor_free(alternatives);
2062 SMARTLIST_FOREACH(fps, char *, cp, tor_free(cp));
2063 smartlist_free(fps);
2064 } else if (warn_if_unnamed) {
2065 routerstatus_t *rs = router_get_consensus_status_by_id(
2066 best_match->cache_info.identity_digest);
2067 if (rs && !rs->name_lookup_warned) {
2068 char fp[HEX_DIGEST_LEN+1];
2069 base16_encode(fp, sizeof(fp),
2070 best_match->cache_info.identity_digest, DIGEST_LEN);
2071 log_warn(LD_CONFIG, "You specified a server \"%s\" by name, but this "
2072 "name is not registered, so it could be used by any server, "
2073 "not just the one you meant. "
2074 "To make sure you get the same server in the future, refer to "
2075 "it by key, as \"$%s\".", nickname, fp);
2076 rs->name_lookup_warned = 1;
2079 return best_match;
2082 return NULL;
2085 /** Try to find a routerinfo for <b>digest</b>. If we don't have one,
2086 * return 1. If we do, ask tor_version_as_new_as() for the answer.
2089 router_digest_version_as_new_as(const char *digest, const char *cutoff)
2091 routerinfo_t *router = router_get_by_digest(digest);
2092 if (!router)
2093 return 1;
2094 return tor_version_as_new_as(router->platform, cutoff);
2097 /** Return true iff <b>digest</b> is the digest of the identity key of a
2098 * trusted directory matching at least one bit of <b>type</b>. If <b>type</b>
2099 * is zero, any authority is okay. */
2101 router_digest_is_trusted_dir_type(const char *digest, authority_type_t type)
2103 if (!trusted_dir_servers)
2104 return 0;
2105 if (authdir_mode(get_options()) && router_digest_is_me(digest))
2106 return 1;
2107 SMARTLIST_FOREACH(trusted_dir_servers, trusted_dir_server_t *, ent,
2108 if (!memcmp(digest, ent->digest, DIGEST_LEN)) {
2109 return (!type) || ((type & ent->type) != 0);
2111 return 0;
2114 /** Return true iff <b>addr</b> is the address of one of our trusted
2115 * directory authorities. */
2117 router_addr_is_trusted_dir(uint32_t addr)
2119 if (!trusted_dir_servers)
2120 return 0;
2121 SMARTLIST_FOREACH(trusted_dir_servers, trusted_dir_server_t *, ent,
2122 if (ent->addr == addr)
2123 return 1;
2125 return 0;
2128 /** If hexdigest is correctly formed, base16_decode it into
2129 * digest, which must have DIGEST_LEN space in it.
2130 * Return 0 on success, -1 on failure.
2133 hexdigest_to_digest(const char *hexdigest, char *digest)
2135 if (hexdigest[0]=='$')
2136 ++hexdigest;
2137 if (strlen(hexdigest) < HEX_DIGEST_LEN ||
2138 base16_decode(digest,DIGEST_LEN,hexdigest,HEX_DIGEST_LEN) < 0)
2139 return -1;
2140 return 0;
2143 /** Return the router in our routerlist whose hexadecimal key digest
2144 * is <b>hexdigest</b>. Return NULL if no such router is known. */
2145 routerinfo_t *
2146 router_get_by_hexdigest(const char *hexdigest)
2148 char digest[DIGEST_LEN];
2149 size_t len;
2150 routerinfo_t *ri;
2152 tor_assert(hexdigest);
2153 if (!routerlist)
2154 return NULL;
2155 if (hexdigest[0]=='$')
2156 ++hexdigest;
2157 len = strlen(hexdigest);
2158 if (hexdigest_to_digest(hexdigest, digest) < 0)
2159 return NULL;
2161 ri = router_get_by_digest(digest);
2163 if (ri && len > HEX_DIGEST_LEN) {
2164 if (hexdigest[HEX_DIGEST_LEN] == '=') {
2165 if (strcasecmp(ri->nickname, hexdigest+HEX_DIGEST_LEN+1) ||
2166 !ri->is_named)
2167 return NULL;
2168 } else if (hexdigest[HEX_DIGEST_LEN] == '~') {
2169 if (strcasecmp(ri->nickname, hexdigest+HEX_DIGEST_LEN+1))
2170 return NULL;
2171 } else {
2172 return NULL;
2176 return ri;
2179 /** Return the router in our routerlist whose 20-byte key digest
2180 * is <b>digest</b>. Return NULL if no such router is known. */
2181 routerinfo_t *
2182 router_get_by_digest(const char *digest)
2184 tor_assert(digest);
2186 if (!routerlist) return NULL;
2188 // routerlist_assert_ok(routerlist);
2190 return rimap_get(routerlist->identity_map, digest);
2193 /** Return the router in our routerlist whose 20-byte descriptor
2194 * is <b>digest</b>. Return NULL if no such router is known. */
2195 signed_descriptor_t *
2196 router_get_by_descriptor_digest(const char *digest)
2198 tor_assert(digest);
2200 if (!routerlist) return NULL;
2202 return sdmap_get(routerlist->desc_digest_map, digest);
2205 /** Return the signed descriptor for the router in our routerlist whose
2206 * 20-byte extra-info digest is <b>digest</b>. Return NULL if no such router
2207 * is known. */
2208 signed_descriptor_t *
2209 router_get_by_extrainfo_digest(const char *digest)
2211 tor_assert(digest);
2213 if (!routerlist) return NULL;
2215 return sdmap_get(routerlist->desc_by_eid_map, digest);
2218 /** Return the signed descriptor for the extrainfo_t in our routerlist whose
2219 * extra-info-digest is <b>digest</b>. Return NULL if no such extra-info
2220 * document is known. */
2221 signed_descriptor_t *
2222 extrainfo_get_by_descriptor_digest(const char *digest)
2224 extrainfo_t *ei;
2225 tor_assert(digest);
2226 if (!routerlist) return NULL;
2227 ei = eimap_get(routerlist->extra_info_map, digest);
2228 return ei ? &ei->cache_info : NULL;
2231 /** Return a pointer to the signed textual representation of a descriptor.
2232 * The returned string is not guaranteed to be NUL-terminated: the string's
2233 * length will be in desc-\>signed_descriptor_len.
2235 * If <b>with_annotations</b> is set, the returned string will include
2236 * the annotations
2237 * (if any) preceding the descriptor. This will increase the length of the
2238 * string by desc-\>annotations_len.
2240 * The caller must not free the string returned.
2242 static const char *
2243 signed_descriptor_get_body_impl(signed_descriptor_t *desc,
2244 int with_annotations)
2246 const char *r = NULL;
2247 size_t len = desc->signed_descriptor_len;
2248 off_t offset = desc->saved_offset;
2249 if (with_annotations)
2250 len += desc->annotations_len;
2251 else
2252 offset += desc->annotations_len;
2254 tor_assert(len > 32);
2255 if (desc->saved_location == SAVED_IN_CACHE && routerlist) {
2256 desc_store_t *store = desc_get_store(router_get_routerlist(), desc);
2257 if (store && store->mmap) {
2258 tor_assert(desc->saved_offset + len <= store->mmap->size);
2259 r = store->mmap->data + offset;
2260 } else if (store) {
2261 log_err(LD_DIR, "We couldn't read a descriptor that is supposedly "
2262 "mmaped in our cache. Is another process running in our data "
2263 "directory? Exiting.");
2264 exit(1);
2267 if (!r) /* no mmap, or not in cache. */
2268 r = desc->signed_descriptor_body +
2269 (with_annotations ? 0 : desc->annotations_len);
2271 tor_assert(r);
2272 if (!with_annotations) {
2273 if (memcmp("router ", r, 7) && memcmp("extra-info ", r, 11)) {
2274 char *cp = tor_strndup(r, 64);
2275 log_err(LD_DIR, "descriptor at %p begins with unexpected string %s. "
2276 "Is another process running in our data directory? Exiting.",
2277 desc, escaped(cp));
2278 exit(1);
2282 return r;
2285 /** Return a pointer to the signed textual representation of a descriptor.
2286 * The returned string is not guaranteed to be NUL-terminated: the string's
2287 * length will be in desc-\>signed_descriptor_len.
2289 * The caller must not free the string returned.
2291 const char *
2292 signed_descriptor_get_body(signed_descriptor_t *desc)
2294 return signed_descriptor_get_body_impl(desc, 0);
2297 /** As signed_descriptor_get_body(), but points to the beginning of the
2298 * annotations section rather than the beginning of the descriptor. */
2299 const char *
2300 signed_descriptor_get_annotations(signed_descriptor_t *desc)
2302 return signed_descriptor_get_body_impl(desc, 1);
2305 /** Return the current list of all known routers. */
2306 routerlist_t *
2307 router_get_routerlist(void)
2309 if (PREDICT_UNLIKELY(!routerlist)) {
2310 routerlist = tor_malloc_zero(sizeof(routerlist_t));
2311 routerlist->routers = smartlist_create();
2312 routerlist->old_routers = smartlist_create();
2313 routerlist->identity_map = rimap_new();
2314 routerlist->desc_digest_map = sdmap_new();
2315 routerlist->desc_by_eid_map = sdmap_new();
2316 routerlist->extra_info_map = eimap_new();
2318 routerlist->desc_store.fname_base = "cached-descriptors";
2319 routerlist->desc_store.fname_alt_base = "cached-routers";
2320 routerlist->extrainfo_store.fname_base = "cached-extrainfo";
2322 routerlist->desc_store.type = ROUTER_STORE;
2323 routerlist->extrainfo_store.type = EXTRAINFO_STORE;
2325 routerlist->desc_store.description = "router descriptors";
2326 routerlist->extrainfo_store.description = "extra-info documents";
2328 return routerlist;
2331 /** Free all storage held by <b>router</b>. */
2332 void
2333 routerinfo_free(routerinfo_t *router)
2335 if (!router)
2336 return;
2338 tor_free(router->cache_info.signed_descriptor_body);
2339 tor_free(router->address);
2340 tor_free(router->nickname);
2341 tor_free(router->platform);
2342 tor_free(router->contact_info);
2343 if (router->onion_pkey)
2344 crypto_free_pk_env(router->onion_pkey);
2345 if (router->identity_pkey)
2346 crypto_free_pk_env(router->identity_pkey);
2347 if (router->declared_family) {
2348 SMARTLIST_FOREACH(router->declared_family, char *, s, tor_free(s));
2349 smartlist_free(router->declared_family);
2351 addr_policy_list_free(router->exit_policy);
2353 /* XXXX Remove if this turns out to affect performance. */
2354 memset(router, 77, sizeof(routerinfo_t));
2356 tor_free(router);
2359 /** Release all storage held by <b>extrainfo</b> */
2360 void
2361 extrainfo_free(extrainfo_t *extrainfo)
2363 if (!extrainfo)
2364 return;
2365 tor_free(extrainfo->cache_info.signed_descriptor_body);
2366 tor_free(extrainfo->pending_sig);
2368 /* XXXX remove this if it turns out to slow us down. */
2369 memset(extrainfo, 88, sizeof(extrainfo_t)); /* debug bad memory usage */
2370 tor_free(extrainfo);
2373 /** Release storage held by <b>sd</b>. */
2374 static void
2375 signed_descriptor_free(signed_descriptor_t *sd)
2377 tor_free(sd->signed_descriptor_body);
2379 /* XXXX remove this once more bugs go away. */
2380 memset(sd, 99, sizeof(signed_descriptor_t)); /* Debug bad mem usage */
2381 tor_free(sd);
2384 /** Extract a signed_descriptor_t from a routerinfo, and free the routerinfo.
2386 static signed_descriptor_t *
2387 signed_descriptor_from_routerinfo(routerinfo_t *ri)
2389 signed_descriptor_t *sd = tor_malloc_zero(sizeof(signed_descriptor_t));
2390 memcpy(sd, &(ri->cache_info), sizeof(signed_descriptor_t));
2391 sd->routerlist_index = -1;
2392 ri->cache_info.signed_descriptor_body = NULL;
2393 routerinfo_free(ri);
2394 return sd;
2397 /** Helper: free the storage held by the extrainfo_t in <b>e</b>. */
2398 static void
2399 _extrainfo_free(void *e)
2401 extrainfo_free(e);
2404 /** Free all storage held by a routerlist <b>rl</b>. */
2405 void
2406 routerlist_free(routerlist_t *rl)
2408 tor_assert(rl);
2409 rimap_free(rl->identity_map, NULL);
2410 sdmap_free(rl->desc_digest_map, NULL);
2411 sdmap_free(rl->desc_by_eid_map, NULL);
2412 eimap_free(rl->extra_info_map, _extrainfo_free);
2413 SMARTLIST_FOREACH(rl->routers, routerinfo_t *, r,
2414 routerinfo_free(r));
2415 SMARTLIST_FOREACH(rl->old_routers, signed_descriptor_t *, sd,
2416 signed_descriptor_free(sd));
2417 smartlist_free(rl->routers);
2418 smartlist_free(rl->old_routers);
2419 if (routerlist->desc_store.mmap)
2420 tor_munmap_file(routerlist->desc_store.mmap);
2421 if (routerlist->extrainfo_store.mmap)
2422 tor_munmap_file(routerlist->extrainfo_store.mmap);
2423 tor_free(rl);
2425 router_dir_info_changed();
2428 /** Log information about how much memory is being used for routerlist,
2429 * at log level <b>severity</b>. */
2430 void
2431 dump_routerlist_mem_usage(int severity)
2433 uint64_t livedescs = 0;
2434 uint64_t olddescs = 0;
2435 if (!routerlist)
2436 return;
2437 SMARTLIST_FOREACH(routerlist->routers, routerinfo_t *, r,
2438 livedescs += r->cache_info.signed_descriptor_len);
2439 SMARTLIST_FOREACH(routerlist->old_routers, signed_descriptor_t *, sd,
2440 olddescs += sd->signed_descriptor_len);
2442 log(severity, LD_DIR,
2443 "In %d live descriptors: "U64_FORMAT" bytes. "
2444 "In %d old descriptors: "U64_FORMAT" bytes.",
2445 smartlist_len(routerlist->routers), U64_PRINTF_ARG(livedescs),
2446 smartlist_len(routerlist->old_routers), U64_PRINTF_ARG(olddescs));
2448 #if 0
2450 const smartlist_t *networkstatus_v2_list = networkstatus_get_v2_list();
2451 networkstatus_t *consensus = networkstatus_get_latest_consensus();
2452 log(severity, LD_DIR, "Now let's look through old_descriptors!");
2453 SMARTLIST_FOREACH(routerlist->old_routers, signed_descriptor_t *, sd, {
2454 int in_v2 = 0;
2455 int in_v3 = 0;
2456 char published[ISO_TIME_LEN+1];
2457 char last_valid_until[ISO_TIME_LEN+1];
2458 char last_served_at[ISO_TIME_LEN+1];
2459 char id[HEX_DIGEST_LEN+1];
2460 routerstatus_t *rs;
2461 format_iso_time(published, sd->published_on);
2462 format_iso_time(last_valid_until, sd->last_listed_as_valid_until);
2463 format_iso_time(last_served_at, sd->last_served_at);
2464 base16_encode(id, sizeof(id), sd->identity_digest, DIGEST_LEN);
2465 SMARTLIST_FOREACH(networkstatus_v2_list, networkstatus_v2_t *, ns,
2467 rs = networkstatus_v2_find_entry(ns, sd->identity_digest);
2468 if (rs && !memcmp(rs->descriptor_digest,
2469 sd->signed_descriptor_digest, DIGEST_LEN)) {
2470 in_v2 = 1; break;
2473 if (consensus) {
2474 rs = networkstatus_vote_find_entry(consensus, sd->identity_digest);
2475 if (rs && !memcmp(rs->descriptor_digest,
2476 sd->signed_descriptor_digest, DIGEST_LEN))
2477 in_v3 = 1;
2479 log(severity, LD_DIR,
2480 "Old descriptor for %s (published %s) %sin v2 ns, %sin v3 "
2481 "consensus. Last valid until %s; last served at %s.",
2482 id, published, in_v2 ? "" : "not ", in_v3 ? "" : "not ",
2483 last_valid_until, last_served_at);
2486 #endif
2489 /** Debugging helper: If <b>idx</b> is nonnegative, assert that <b>ri</b> is
2490 * in <b>sl</b> at position <b>idx</b>. Otherwise, search <b>sl</b> for
2491 * <b>ri</b>. Return the index of <b>ri</b> in <b>sl</b>, or -1 if <b>ri</b>
2492 * is not in <b>sl</b>. */
2493 static INLINE int
2494 _routerlist_find_elt(smartlist_t *sl, void *ri, int idx)
2496 if (idx < 0) {
2497 idx = -1;
2498 SMARTLIST_FOREACH(sl, routerinfo_t *, r,
2499 if (r == ri) {
2500 idx = r_sl_idx;
2501 break;
2503 } else {
2504 tor_assert(idx < smartlist_len(sl));
2505 tor_assert(smartlist_get(sl, idx) == ri);
2507 return idx;
2510 /** Insert an item <b>ri</b> into the routerlist <b>rl</b>, updating indices
2511 * as needed. There must be no previous member of <b>rl</b> with the same
2512 * identity digest as <b>ri</b>: If there is, call routerlist_replace
2513 * instead.
2515 static void
2516 routerlist_insert(routerlist_t *rl, routerinfo_t *ri)
2518 routerinfo_t *ri_old;
2520 /* XXXX Remove if this slows us down. */
2521 routerinfo_t *ri_generated = router_get_my_routerinfo();
2522 tor_assert(ri_generated != ri);
2524 tor_assert(ri->cache_info.routerlist_index == -1);
2526 ri_old = rimap_set(rl->identity_map, ri->cache_info.identity_digest, ri);
2527 tor_assert(!ri_old);
2528 sdmap_set(rl->desc_digest_map, ri->cache_info.signed_descriptor_digest,
2529 &(ri->cache_info));
2530 if (!tor_digest_is_zero(ri->cache_info.extra_info_digest))
2531 sdmap_set(rl->desc_by_eid_map, ri->cache_info.extra_info_digest,
2532 &ri->cache_info);
2533 smartlist_add(rl->routers, ri);
2534 ri->cache_info.routerlist_index = smartlist_len(rl->routers) - 1;
2535 router_dir_info_changed();
2536 #ifdef DEBUG_ROUTERLIST
2537 routerlist_assert_ok(rl);
2538 #endif
2541 /** Adds the extrainfo_t <b>ei</b> to the routerlist <b>rl</b>, if there is a
2542 * corresponding router in rl-\>routers or rl-\>old_routers. Return true iff
2543 * we actually inserted <b>ei</b>. Free <b>ei</b> if it isn't inserted. */
2544 static int
2545 extrainfo_insert(routerlist_t *rl, extrainfo_t *ei)
2547 int r = 0;
2548 routerinfo_t *ri = rimap_get(rl->identity_map,
2549 ei->cache_info.identity_digest);
2550 signed_descriptor_t *sd =
2551 sdmap_get(rl->desc_by_eid_map, ei->cache_info.signed_descriptor_digest);
2552 extrainfo_t *ei_tmp;
2555 /* XXXX remove this code if it slows us down. */
2556 extrainfo_t *ei_generated = router_get_my_extrainfo();
2557 tor_assert(ei_generated != ei);
2560 if (!ri) {
2561 /* This router is unknown; we can't even verify the signature. Give up.*/
2562 goto done;
2564 if (routerinfo_incompatible_with_extrainfo(ri, ei, sd, NULL)) {
2565 goto done;
2568 /* Okay, if we make it here, we definitely have a router corresponding to
2569 * this extrainfo. */
2571 ei_tmp = eimap_set(rl->extra_info_map,
2572 ei->cache_info.signed_descriptor_digest,
2573 ei);
2574 r = 1;
2575 if (ei_tmp) {
2576 rl->extrainfo_store.bytes_dropped +=
2577 ei_tmp->cache_info.signed_descriptor_len;
2578 extrainfo_free(ei_tmp);
2581 done:
2582 if (r == 0)
2583 extrainfo_free(ei);
2585 #ifdef DEBUG_ROUTERLIST
2586 routerlist_assert_ok(rl);
2587 #endif
2588 return r;
2591 #define should_cache_old_descriptors() \
2592 directory_caches_dir_info(get_options())
2594 /** If we're a directory cache and routerlist <b>rl</b> doesn't have
2595 * a copy of router <b>ri</b> yet, add it to the list of old (not
2596 * recommended but still served) descriptors. Else free it. */
2597 static void
2598 routerlist_insert_old(routerlist_t *rl, routerinfo_t *ri)
2601 /* XXXX remove this code if it slows us down. */
2602 routerinfo_t *ri_generated = router_get_my_routerinfo();
2603 tor_assert(ri_generated != ri);
2605 tor_assert(ri->cache_info.routerlist_index == -1);
2607 if (should_cache_old_descriptors() &&
2608 ri->purpose == ROUTER_PURPOSE_GENERAL &&
2609 !sdmap_get(rl->desc_digest_map,
2610 ri->cache_info.signed_descriptor_digest)) {
2611 signed_descriptor_t *sd = signed_descriptor_from_routerinfo(ri);
2612 sdmap_set(rl->desc_digest_map, sd->signed_descriptor_digest, sd);
2613 smartlist_add(rl->old_routers, sd);
2614 sd->routerlist_index = smartlist_len(rl->old_routers)-1;
2615 if (!tor_digest_is_zero(sd->extra_info_digest))
2616 sdmap_set(rl->desc_by_eid_map, sd->extra_info_digest, sd);
2617 } else {
2618 routerinfo_free(ri);
2620 #ifdef DEBUG_ROUTERLIST
2621 routerlist_assert_ok(rl);
2622 #endif
2625 /** Remove an item <b>ri</b> from the routerlist <b>rl</b>, updating indices
2626 * as needed. If <b>idx</b> is nonnegative and smartlist_get(rl-&gt;routers,
2627 * idx) == ri, we don't need to do a linear search over the list to decide
2628 * which to remove. We fill the gap in rl-&gt;routers with a later element in
2629 * the list, if any exists. <b>ri</b> is freed.
2631 * If <b>make_old</b> is true, instead of deleting the router, we try adding
2632 * it to rl-&gt;old_routers. */
2633 void
2634 routerlist_remove(routerlist_t *rl, routerinfo_t *ri, int make_old)
2636 routerinfo_t *ri_tmp;
2637 extrainfo_t *ei_tmp;
2638 int idx = ri->cache_info.routerlist_index;
2639 tor_assert(0 <= idx && idx < smartlist_len(rl->routers));
2640 tor_assert(smartlist_get(rl->routers, idx) == ri);
2642 ri->cache_info.routerlist_index = -1;
2643 smartlist_del(rl->routers, idx);
2644 if (idx < smartlist_len(rl->routers)) {
2645 routerinfo_t *r = smartlist_get(rl->routers, idx);
2646 r->cache_info.routerlist_index = idx;
2649 ri_tmp = rimap_remove(rl->identity_map, ri->cache_info.identity_digest);
2650 router_dir_info_changed();
2651 tor_assert(ri_tmp == ri);
2653 if (make_old && should_cache_old_descriptors() &&
2654 ri->purpose == ROUTER_PURPOSE_GENERAL) {
2655 signed_descriptor_t *sd;
2656 sd = signed_descriptor_from_routerinfo(ri);
2657 smartlist_add(rl->old_routers, sd);
2658 sd->routerlist_index = smartlist_len(rl->old_routers)-1;
2659 sdmap_set(rl->desc_digest_map, sd->signed_descriptor_digest, sd);
2660 if (!tor_digest_is_zero(sd->extra_info_digest))
2661 sdmap_set(rl->desc_by_eid_map, sd->extra_info_digest, sd);
2662 } else {
2663 signed_descriptor_t *sd_tmp;
2664 sd_tmp = sdmap_remove(rl->desc_digest_map,
2665 ri->cache_info.signed_descriptor_digest);
2666 tor_assert(sd_tmp == &(ri->cache_info));
2667 rl->desc_store.bytes_dropped += ri->cache_info.signed_descriptor_len;
2668 ei_tmp = eimap_remove(rl->extra_info_map,
2669 ri->cache_info.extra_info_digest);
2670 if (ei_tmp) {
2671 rl->extrainfo_store.bytes_dropped +=
2672 ei_tmp->cache_info.signed_descriptor_len;
2673 extrainfo_free(ei_tmp);
2675 if (!tor_digest_is_zero(ri->cache_info.extra_info_digest))
2676 sdmap_remove(rl->desc_by_eid_map, ri->cache_info.extra_info_digest);
2677 routerinfo_free(ri);
2679 #ifdef DEBUG_ROUTERLIST
2680 routerlist_assert_ok(rl);
2681 #endif
2684 /** Remove a signed_descriptor_t <b>sd</b> from <b>rl</b>-\>old_routers, and
2685 * adjust <b>rl</b> as appropriate. <b>idx</b> is -1, or the index of
2686 * <b>sd</b>. */
2687 static void
2688 routerlist_remove_old(routerlist_t *rl, signed_descriptor_t *sd, int idx)
2690 signed_descriptor_t *sd_tmp;
2691 extrainfo_t *ei_tmp;
2692 desc_store_t *store;
2693 if (idx == -1) {
2694 idx = sd->routerlist_index;
2696 tor_assert(0 <= idx && idx < smartlist_len(rl->old_routers));
2697 /* XXXX edmanm's bridge relay triggered the following assert while
2698 * running 0.2.0.12-alpha. If anybody triggers this again, see if we
2699 * can get a backtrace. */
2700 tor_assert(smartlist_get(rl->old_routers, idx) == sd);
2701 tor_assert(idx == sd->routerlist_index);
2703 sd->routerlist_index = -1;
2704 smartlist_del(rl->old_routers, idx);
2705 if (idx < smartlist_len(rl->old_routers)) {
2706 signed_descriptor_t *d = smartlist_get(rl->old_routers, idx);
2707 d->routerlist_index = idx;
2709 sd_tmp = sdmap_remove(rl->desc_digest_map,
2710 sd->signed_descriptor_digest);
2711 tor_assert(sd_tmp == sd);
2712 store = desc_get_store(rl, sd);
2713 if (store)
2714 store->bytes_dropped += sd->signed_descriptor_len;
2716 ei_tmp = eimap_remove(rl->extra_info_map,
2717 sd->extra_info_digest);
2718 if (ei_tmp) {
2719 rl->extrainfo_store.bytes_dropped +=
2720 ei_tmp->cache_info.signed_descriptor_len;
2721 extrainfo_free(ei_tmp);
2723 if (!tor_digest_is_zero(sd->extra_info_digest))
2724 sdmap_remove(rl->desc_by_eid_map, sd->extra_info_digest);
2726 signed_descriptor_free(sd);
2727 #ifdef DEBUG_ROUTERLIST
2728 routerlist_assert_ok(rl);
2729 #endif
2732 /** Remove <b>ri_old</b> from the routerlist <b>rl</b>, and replace it with
2733 * <b>ri_new</b>, updating all index info. If <b>idx</b> is nonnegative and
2734 * smartlist_get(rl-&gt;routers, idx) == ri, we don't need to do a linear
2735 * search over the list to decide which to remove. We put ri_new in the same
2736 * index as ri_old, if possible. ri is freed as appropriate.
2738 * If should_cache_descriptors() is true, instead of deleting the router,
2739 * we add it to rl-&gt;old_routers. */
2740 static void
2741 routerlist_replace(routerlist_t *rl, routerinfo_t *ri_old,
2742 routerinfo_t *ri_new)
2744 int idx;
2746 routerinfo_t *ri_tmp;
2747 extrainfo_t *ei_tmp;
2749 /* XXXX Remove this if it turns out to slow us down. */
2750 routerinfo_t *ri_generated = router_get_my_routerinfo();
2751 tor_assert(ri_generated != ri_new);
2753 tor_assert(ri_old != ri_new);
2754 tor_assert(ri_new->cache_info.routerlist_index == -1);
2756 idx = ri_old->cache_info.routerlist_index;
2757 tor_assert(0 <= idx && idx < smartlist_len(rl->routers));
2758 tor_assert(smartlist_get(rl->routers, idx) == ri_old);
2760 router_dir_info_changed();
2761 if (idx >= 0) {
2762 smartlist_set(rl->routers, idx, ri_new);
2763 ri_old->cache_info.routerlist_index = -1;
2764 ri_new->cache_info.routerlist_index = idx;
2765 /* Check that ri_old is not in rl->routers anymore: */
2766 tor_assert( _routerlist_find_elt(rl->routers, ri_old, -1) == -1 );
2767 } else {
2768 log_warn(LD_BUG, "Appending entry from routerlist_replace.");
2769 routerlist_insert(rl, ri_new);
2770 return;
2772 if (memcmp(ri_old->cache_info.identity_digest,
2773 ri_new->cache_info.identity_digest, DIGEST_LEN)) {
2774 /* digests don't match; digestmap_set won't replace */
2775 rimap_remove(rl->identity_map, ri_old->cache_info.identity_digest);
2777 ri_tmp = rimap_set(rl->identity_map,
2778 ri_new->cache_info.identity_digest, ri_new);
2779 tor_assert(!ri_tmp || ri_tmp == ri_old);
2780 sdmap_set(rl->desc_digest_map,
2781 ri_new->cache_info.signed_descriptor_digest,
2782 &(ri_new->cache_info));
2784 if (!tor_digest_is_zero(ri_new->cache_info.extra_info_digest)) {
2785 sdmap_set(rl->desc_by_eid_map, ri_new->cache_info.extra_info_digest,
2786 &ri_new->cache_info);
2789 if (should_cache_old_descriptors() &&
2790 ri_old->purpose == ROUTER_PURPOSE_GENERAL) {
2791 signed_descriptor_t *sd = signed_descriptor_from_routerinfo(ri_old);
2792 smartlist_add(rl->old_routers, sd);
2793 sd->routerlist_index = smartlist_len(rl->old_routers)-1;
2794 sdmap_set(rl->desc_digest_map, sd->signed_descriptor_digest, sd);
2795 if (!tor_digest_is_zero(sd->extra_info_digest))
2796 sdmap_set(rl->desc_by_eid_map, sd->extra_info_digest, sd);
2797 } else {
2798 if (memcmp(ri_old->cache_info.signed_descriptor_digest,
2799 ri_new->cache_info.signed_descriptor_digest,
2800 DIGEST_LEN)) {
2801 /* digests don't match; digestmap_set didn't replace */
2802 sdmap_remove(rl->desc_digest_map,
2803 ri_old->cache_info.signed_descriptor_digest);
2806 ei_tmp = eimap_remove(rl->extra_info_map,
2807 ri_old->cache_info.extra_info_digest);
2808 if (ei_tmp) {
2809 rl->extrainfo_store.bytes_dropped +=
2810 ei_tmp->cache_info.signed_descriptor_len;
2811 extrainfo_free(ei_tmp);
2813 if (!tor_digest_is_zero(ri_old->cache_info.extra_info_digest)) {
2814 sdmap_remove(rl->desc_by_eid_map,
2815 ri_old->cache_info.extra_info_digest);
2817 rl->desc_store.bytes_dropped += ri_old->cache_info.signed_descriptor_len;
2818 routerinfo_free(ri_old);
2820 #ifdef DEBUG_ROUTERLIST
2821 routerlist_assert_ok(rl);
2822 #endif
2825 /** Extract the descriptor <b>sd</b> from old_routerlist, and re-parse
2826 * it as a fresh routerinfo_t. */
2827 static routerinfo_t *
2828 routerlist_reparse_old(routerlist_t *rl, signed_descriptor_t *sd)
2830 routerinfo_t *ri;
2831 const char *body;
2833 body = signed_descriptor_get_annotations(sd);
2835 ri = router_parse_entry_from_string(body,
2836 body+sd->signed_descriptor_len+sd->annotations_len,
2837 0, 1, NULL);
2838 if (!ri)
2839 return NULL;
2840 memcpy(&ri->cache_info, sd, sizeof(signed_descriptor_t));
2841 sd->signed_descriptor_body = NULL; /* Steal reference. */
2842 ri->cache_info.routerlist_index = -1;
2844 routerlist_remove_old(rl, sd, -1);
2846 return ri;
2849 /** Free all memory held by the routerlist module. */
2850 void
2851 routerlist_free_all(void)
2853 if (routerlist)
2854 routerlist_free(routerlist);
2855 routerlist = NULL;
2856 if (warned_nicknames) {
2857 SMARTLIST_FOREACH(warned_nicknames, char *, cp, tor_free(cp));
2858 smartlist_free(warned_nicknames);
2859 warned_nicknames = NULL;
2861 if (trusted_dir_servers) {
2862 SMARTLIST_FOREACH(trusted_dir_servers, trusted_dir_server_t *, ds,
2863 trusted_dir_server_free(ds));
2864 smartlist_free(trusted_dir_servers);
2865 trusted_dir_servers = NULL;
2867 if (trusted_dir_certs) {
2868 DIGESTMAP_FOREACH(trusted_dir_certs, key, cert_list_t *, cl) {
2869 SMARTLIST_FOREACH(cl->certs, authority_cert_t *, cert,
2870 authority_cert_free(cert));
2871 smartlist_free(cl->certs);
2872 tor_free(cl);
2873 } DIGESTMAP_FOREACH_END;
2874 digestmap_free(trusted_dir_certs, NULL);
2875 trusted_dir_certs = NULL;
2879 /** Forget that we have issued any router-related warnings, so that we'll
2880 * warn again if we see the same errors. */
2881 void
2882 routerlist_reset_warnings(void)
2884 if (!warned_nicknames)
2885 warned_nicknames = smartlist_create();
2886 SMARTLIST_FOREACH(warned_nicknames, char *, cp, tor_free(cp));
2887 smartlist_clear(warned_nicknames); /* now the list is empty. */
2889 networkstatus_reset_warnings();
2892 /** Mark the router with ID <b>digest</b> as running or non-running
2893 * in our routerlist. */
2894 void
2895 router_set_status(const char *digest, int up)
2897 routerinfo_t *router;
2898 routerstatus_t *status;
2899 tor_assert(digest);
2901 SMARTLIST_FOREACH(trusted_dir_servers, trusted_dir_server_t *, d,
2902 if (!memcmp(d->digest, digest, DIGEST_LEN))
2903 d->is_running = up);
2905 router = router_get_by_digest(digest);
2906 if (router) {
2907 log_debug(LD_DIR,"Marking router '%s/%s' as %s.",
2908 router->nickname, router->address, up ? "up" : "down");
2909 if (!up && router_is_me(router) && !we_are_hibernating())
2910 log_warn(LD_NET, "We just marked ourself as down. Are your external "
2911 "addresses reachable?");
2912 router->is_running = up;
2914 status = router_get_consensus_status_by_id(digest);
2915 if (status && status->is_running != up) {
2916 status->is_running = up;
2917 control_event_networkstatus_changed_single(status);
2919 router_dir_info_changed();
2922 /** Add <b>router</b> to the routerlist, if we don't already have it. Replace
2923 * older entries (if any) with the same key. Note: Callers should not hold
2924 * their pointers to <b>router</b> if this function fails; <b>router</b>
2925 * will either be inserted into the routerlist or freed. Similarly, even
2926 * if this call succeeds, they should not hold their pointers to
2927 * <b>router</b> after subsequent calls with other routerinfo's -- they
2928 * might cause the original routerinfo to get freed.
2930 * Returns the status for the operation. Might set *<b>msg</b> if it wants
2931 * the poster of the router to know something.
2933 * If <b>from_cache</b>, this descriptor came from our disk cache. If
2934 * <b>from_fetch</b>, we received it in response to a request we made.
2935 * (If both are false, that means it was uploaded to us as an auth dir
2936 * server or via the controller.)
2938 * This function should be called *after*
2939 * routers_update_status_from_consensus_networkstatus; subsequently, you
2940 * should call router_rebuild_store and routerlist_descriptors_added.
2942 was_router_added_t
2943 router_add_to_routerlist(routerinfo_t *router, const char **msg,
2944 int from_cache, int from_fetch)
2946 const char *id_digest;
2947 int authdir = authdir_mode_handles_descs(get_options(), router->purpose);
2948 int authdir_believes_valid = 0;
2949 routerinfo_t *old_router;
2950 networkstatus_t *consensus = networkstatus_get_latest_consensus();
2951 const smartlist_t *networkstatus_v2_list = networkstatus_get_v2_list();
2952 int in_consensus = 0;
2954 tor_assert(msg);
2956 if (!routerlist)
2957 router_get_routerlist();
2959 id_digest = router->cache_info.identity_digest;
2961 /* Make sure that we haven't already got this exact descriptor. */
2962 if (sdmap_get(routerlist->desc_digest_map,
2963 router->cache_info.signed_descriptor_digest)) {
2964 log_info(LD_DIR,
2965 "Dropping descriptor that we already have for router '%s'",
2966 router->nickname);
2967 *msg = "Router descriptor was not new.";
2968 routerinfo_free(router);
2969 return ROUTER_WAS_NOT_NEW;
2972 if (authdir) {
2973 if (authdir_wants_to_reject_router(router, msg,
2974 !from_cache && !from_fetch)) {
2975 tor_assert(*msg);
2976 routerinfo_free(router);
2977 return ROUTER_AUTHDIR_REJECTS;
2979 authdir_believes_valid = router->is_valid;
2980 } else if (from_fetch) {
2981 /* Only check the descriptor digest against the network statuses when
2982 * we are receiving in response to a fetch. */
2984 if (!signed_desc_digest_is_recognized(&router->cache_info) &&
2985 !routerinfo_is_a_configured_bridge(router)) {
2986 /* We asked for it, so some networkstatus must have listed it when we
2987 * did. Save it if we're a cache in case somebody else asks for it. */
2988 log_info(LD_DIR,
2989 "Received a no-longer-recognized descriptor for router '%s'",
2990 router->nickname);
2991 *msg = "Router descriptor is not referenced by any network-status.";
2993 /* Only journal this desc if we'll be serving it. */
2994 if (!from_cache && should_cache_old_descriptors())
2995 signed_desc_append_to_journal(&router->cache_info,
2996 &routerlist->desc_store);
2997 routerlist_insert_old(routerlist, router);
2998 return ROUTER_NOT_IN_CONSENSUS_OR_NETWORKSTATUS;
3002 /* We no longer need a router with this descriptor digest. */
3003 SMARTLIST_FOREACH(networkstatus_v2_list, networkstatus_v2_t *, ns,
3005 routerstatus_t *rs =
3006 networkstatus_v2_find_entry(ns, router->cache_info.identity_digest);
3007 if (rs && !memcmp(rs->descriptor_digest,
3008 router->cache_info.signed_descriptor_digest,
3009 DIGEST_LEN))
3010 rs->need_to_mirror = 0;
3012 if (consensus) {
3013 routerstatus_t *rs = networkstatus_vote_find_entry(consensus,
3014 router->cache_info.identity_digest);
3015 if (rs && !memcmp(rs->descriptor_digest,
3016 router->cache_info.signed_descriptor_digest,
3017 DIGEST_LEN)) {
3018 in_consensus = 1;
3019 rs->need_to_mirror = 0;
3023 if (router->purpose == ROUTER_PURPOSE_GENERAL &&
3024 consensus && !in_consensus && !authdir) {
3025 /* If it's a general router not listed in the consensus, then don't
3026 * consider replacing the latest router with it. */
3027 if (!from_cache && should_cache_old_descriptors())
3028 signed_desc_append_to_journal(&router->cache_info,
3029 &routerlist->desc_store);
3030 routerlist_insert_old(routerlist, router);
3031 *msg = "Skipping router descriptor: not in consensus.";
3032 return ROUTER_NOT_IN_CONSENSUS;
3035 /* If we have a router with the same identity key, choose the newer one. */
3036 old_router = rimap_get(routerlist->identity_map,
3037 router->cache_info.identity_digest);
3038 if (old_router) {
3039 if (!in_consensus && (router->cache_info.published_on <=
3040 old_router->cache_info.published_on)) {
3041 /* Same key, but old. This one is not listed in the consensus. */
3042 log_debug(LD_DIR, "Skipping not-new descriptor for router '%s'",
3043 router->nickname);
3044 /* Only journal this desc if we'll be serving it. */
3045 if (!from_cache && should_cache_old_descriptors())
3046 signed_desc_append_to_journal(&router->cache_info,
3047 &routerlist->desc_store);
3048 routerlist_insert_old(routerlist, router);
3049 *msg = "Router descriptor was not new.";
3050 return ROUTER_WAS_NOT_NEW;
3051 } else {
3052 /* Same key, and either new, or listed in the consensus. */
3053 log_debug(LD_DIR, "Replacing entry for router '%s/%s' [%s]",
3054 router->nickname, old_router->nickname,
3055 hex_str(id_digest,DIGEST_LEN));
3056 if (router->addr == old_router->addr &&
3057 router->or_port == old_router->or_port) {
3058 /* these carry over when the address and orport are unchanged. */
3059 router->last_reachable = old_router->last_reachable;
3060 router->testing_since = old_router->testing_since;
3062 routerlist_replace(routerlist, old_router, router);
3063 if (!from_cache) {
3064 signed_desc_append_to_journal(&router->cache_info,
3065 &routerlist->desc_store);
3067 directory_set_dirty();
3068 *msg = authdir_believes_valid ? "Valid server updated" :
3069 ("Invalid server updated. (This dirserver is marking your "
3070 "server as unapproved.)");
3071 return ROUTER_ADDED_SUCCESSFULLY;
3075 if (!in_consensus && from_cache &&
3076 router->cache_info.published_on < time(NULL) - OLD_ROUTER_DESC_MAX_AGE) {
3077 *msg = "Router descriptor was really old.";
3078 return ROUTER_WAS_NOT_NEW;
3081 /* We haven't seen a router with this identity before. Add it to the end of
3082 * the list. */
3083 routerlist_insert(routerlist, router);
3084 if (!from_cache)
3085 signed_desc_append_to_journal(&router->cache_info,
3086 &routerlist->desc_store);
3087 directory_set_dirty();
3088 return ROUTER_ADDED_SUCCESSFULLY;
3091 /** Insert <b>ei</b> into the routerlist, or free it. Other arguments are
3092 * as for router_add_to_routerlist(). Return ROUTER_ADDED_SUCCESSFULLY iff
3093 * we actually inserted it, ROUTER_BAD_EI otherwise.
3095 was_router_added_t
3096 router_add_extrainfo_to_routerlist(extrainfo_t *ei, const char **msg,
3097 int from_cache, int from_fetch)
3099 int inserted;
3100 (void)from_fetch;
3101 if (msg) *msg = NULL;
3102 /*XXXX022 Do something with msg */
3104 inserted = extrainfo_insert(router_get_routerlist(), ei);
3106 if (inserted && !from_cache)
3107 signed_desc_append_to_journal(&ei->cache_info,
3108 &routerlist->extrainfo_store);
3110 if (inserted)
3111 return ROUTER_ADDED_SUCCESSFULLY;
3112 else
3113 return ROUTER_BAD_EI;
3116 /** Sorting helper: return &lt;0, 0, or &gt;0 depending on whether the
3117 * signed_descriptor_t* in *<b>a</b> has an identity digest preceding, equal
3118 * to, or later than that of *<b>b</b>. */
3119 static int
3120 _compare_old_routers_by_identity(const void **_a, const void **_b)
3122 int i;
3123 const signed_descriptor_t *r1 = *_a, *r2 = *_b;
3124 if ((i = memcmp(r1->identity_digest, r2->identity_digest, DIGEST_LEN)))
3125 return i;
3126 return (int)(r1->published_on - r2->published_on);
3129 /** Internal type used to represent how long an old descriptor was valid,
3130 * where it appeared in the list of old descriptors, and whether it's extra
3131 * old. Used only by routerlist_remove_old_cached_routers_with_id(). */
3132 struct duration_idx_t {
3133 int duration;
3134 int idx;
3135 int old;
3138 /** Sorting helper: compare two duration_idx_t by their duration. */
3139 static int
3140 _compare_duration_idx(const void *_d1, const void *_d2)
3142 const struct duration_idx_t *d1 = _d1;
3143 const struct duration_idx_t *d2 = _d2;
3144 return d1->duration - d2->duration;
3147 /** The range <b>lo</b> through <b>hi</b> inclusive of routerlist->old_routers
3148 * must contain routerinfo_t with the same identity and with publication time
3149 * in ascending order. Remove members from this range until there are no more
3150 * than max_descriptors_per_router() remaining. Start by removing the oldest
3151 * members from before <b>cutoff</b>, then remove members which were current
3152 * for the lowest amount of time. The order of members of old_routers at
3153 * indices <b>lo</b> or higher may be changed.
3155 static void
3156 routerlist_remove_old_cached_routers_with_id(time_t now,
3157 time_t cutoff, int lo, int hi,
3158 digestset_t *retain)
3160 int i, n = hi-lo+1;
3161 unsigned n_extra, n_rmv = 0;
3162 struct duration_idx_t *lifespans;
3163 uint8_t *rmv, *must_keep;
3164 smartlist_t *lst = routerlist->old_routers;
3165 #if 1
3166 const char *ident;
3167 tor_assert(hi < smartlist_len(lst));
3168 tor_assert(lo <= hi);
3169 ident = ((signed_descriptor_t*)smartlist_get(lst, lo))->identity_digest;
3170 for (i = lo+1; i <= hi; ++i) {
3171 signed_descriptor_t *r = smartlist_get(lst, i);
3172 tor_assert(!memcmp(ident, r->identity_digest, DIGEST_LEN));
3174 #endif
3175 /* Check whether we need to do anything at all. */
3177 int mdpr = directory_caches_dir_info(get_options()) ? 2 : 1;
3178 if (n <= mdpr)
3179 return;
3180 n_extra = n - mdpr;
3183 lifespans = tor_malloc_zero(sizeof(struct duration_idx_t)*n);
3184 rmv = tor_malloc_zero(sizeof(uint8_t)*n);
3185 must_keep = tor_malloc_zero(sizeof(uint8_t)*n);
3186 /* Set lifespans to contain the lifespan and index of each server. */
3187 /* Set rmv[i-lo]=1 if we're going to remove a server for being too old. */
3188 for (i = lo; i <= hi; ++i) {
3189 signed_descriptor_t *r = smartlist_get(lst, i);
3190 signed_descriptor_t *r_next;
3191 lifespans[i-lo].idx = i;
3192 if (r->last_listed_as_valid_until >= now ||
3193 (retain && digestset_isin(retain, r->signed_descriptor_digest))) {
3194 must_keep[i-lo] = 1;
3196 if (i < hi) {
3197 r_next = smartlist_get(lst, i+1);
3198 tor_assert(r->published_on <= r_next->published_on);
3199 lifespans[i-lo].duration = (int)(r_next->published_on - r->published_on);
3200 } else {
3201 r_next = NULL;
3202 lifespans[i-lo].duration = INT_MAX;
3204 if (!must_keep[i-lo] && r->published_on < cutoff && n_rmv < n_extra) {
3205 ++n_rmv;
3206 lifespans[i-lo].old = 1;
3207 rmv[i-lo] = 1;
3211 if (n_rmv < n_extra) {
3213 * We aren't removing enough servers for being old. Sort lifespans by
3214 * the duration of liveness, and remove the ones we're not already going to
3215 * remove based on how long they were alive.
3217 qsort(lifespans, n, sizeof(struct duration_idx_t), _compare_duration_idx);
3218 for (i = 0; i < n && n_rmv < n_extra; ++i) {
3219 if (!must_keep[lifespans[i].idx-lo] && !lifespans[i].old) {
3220 rmv[lifespans[i].idx-lo] = 1;
3221 ++n_rmv;
3226 i = hi;
3227 do {
3228 if (rmv[i-lo])
3229 routerlist_remove_old(routerlist, smartlist_get(lst, i), i);
3230 } while (--i >= lo);
3231 tor_free(must_keep);
3232 tor_free(rmv);
3233 tor_free(lifespans);
3236 /** Deactivate any routers from the routerlist that are more than
3237 * ROUTER_MAX_AGE seconds old and not recommended by any networkstatuses;
3238 * remove old routers from the list of cached routers if we have too many.
3240 void
3241 routerlist_remove_old_routers(void)
3243 int i, hi=-1;
3244 const char *cur_id = NULL;
3245 time_t now = time(NULL);
3246 time_t cutoff;
3247 routerinfo_t *router;
3248 signed_descriptor_t *sd;
3249 digestset_t *retain;
3250 int caches = directory_caches_dir_info(get_options());
3251 const networkstatus_t *consensus = networkstatus_get_latest_consensus();
3252 const smartlist_t *networkstatus_v2_list = networkstatus_get_v2_list();
3253 int have_enough_v2;
3255 trusted_dirs_remove_old_certs();
3257 if (!routerlist || !consensus)
3258 return;
3260 // routerlist_assert_ok(routerlist);
3262 /* We need to guess how many router descriptors we will wind up wanting to
3263 retain, so that we can be sure to allocate a large enough Bloom filter
3264 to hold the digest set. Overestimating is fine; underestimating is bad.
3267 /* We'll probably retain everything in the consensus. */
3268 int n_max_retain = smartlist_len(consensus->routerstatus_list);
3269 if (caches && networkstatus_v2_list) {
3270 /* If we care about v2 statuses, we'll retain at most as many as are
3271 listed any of the v2 statues. This will be at least the length of
3272 the largest v2 networstatus, and in the worst case, this set will be
3273 equal to the sum of the lengths of all v2 consensuses. Take the
3274 worst case.
3276 SMARTLIST_FOREACH(networkstatus_v2_list, networkstatus_v2_t *, ns,
3277 n_max_retain += smartlist_len(ns->entries));
3279 retain = digestset_new(n_max_retain);
3282 cutoff = now - OLD_ROUTER_DESC_MAX_AGE;
3283 /* Build a list of all the descriptors that _anybody_ lists. */
3284 if (caches && networkstatus_v2_list) {
3285 SMARTLIST_FOREACH(networkstatus_v2_list, networkstatus_v2_t *, ns,
3287 /* XXXX The inner loop here gets pretty expensive, and actually shows up
3288 * on some profiles. It may be the reason digestmap_set shows up in
3289 * profiles too. If instead we kept a per-descriptor digest count of
3290 * how many networkstatuses recommended each descriptor, and changed
3291 * that only when the networkstatuses changed, that would be a speed
3292 * improvement, possibly 1-4% if it also removes digestmap_set from the
3293 * profile. Not worth it for 0.1.2.x, though. The new directory
3294 * system will obsolete this whole thing in 0.2.0.x. */
3295 SMARTLIST_FOREACH(ns->entries, routerstatus_t *, rs,
3296 if (rs->published_on >= cutoff)
3297 digestset_add(retain, rs->descriptor_digest));
3301 /* Retain anything listed in the consensus. */
3302 if (consensus) {
3303 SMARTLIST_FOREACH(consensus->routerstatus_list, routerstatus_t *, rs,
3304 if (rs->published_on >= cutoff)
3305 digestset_add(retain, rs->descriptor_digest));
3308 /* If we have a consensus, and nearly as many v2 networkstatuses as we want,
3309 * we should consider pruning current routers that are too old and that
3310 * nobody recommends. (If we don't have a consensus or enough v2
3311 * networkstatuses, then we should get more before we decide to kill
3312 * routers.) */
3313 /* we set this to true iff we don't care about v2 info, or we have enough. */
3314 have_enough_v2 = !caches ||
3315 (networkstatus_v2_list &&
3316 smartlist_len(networkstatus_v2_list) > get_n_v2_authorities() / 2);
3318 if (have_enough_v2 && consensus) {
3319 cutoff = now - ROUTER_MAX_AGE;
3320 /* Remove too-old unrecommended members of routerlist->routers. */
3321 for (i = 0; i < smartlist_len(routerlist->routers); ++i) {
3322 router = smartlist_get(routerlist->routers, i);
3323 if (router->cache_info.published_on <= cutoff &&
3324 router->cache_info.last_listed_as_valid_until < now &&
3325 !digestset_isin(retain,
3326 router->cache_info.signed_descriptor_digest)) {
3327 /* Too old: remove it. (If we're a cache, just move it into
3328 * old_routers.) */
3329 log_info(LD_DIR,
3330 "Forgetting obsolete (too old) routerinfo for router '%s'",
3331 router->nickname);
3332 routerlist_remove(routerlist, router, 1);
3333 i--;
3338 //routerlist_assert_ok(routerlist);
3340 /* Remove far-too-old members of routerlist->old_routers. */
3341 cutoff = now - OLD_ROUTER_DESC_MAX_AGE;
3342 for (i = 0; i < smartlist_len(routerlist->old_routers); ++i) {
3343 sd = smartlist_get(routerlist->old_routers, i);
3344 if (sd->published_on <= cutoff &&
3345 sd->last_listed_as_valid_until < now &&
3346 !digestset_isin(retain, sd->signed_descriptor_digest)) {
3347 /* Too old. Remove it. */
3348 routerlist_remove_old(routerlist, sd, i--);
3352 //routerlist_assert_ok(routerlist);
3354 log_info(LD_DIR, "We have %d live routers and %d old router descriptors.",
3355 smartlist_len(routerlist->routers),
3356 smartlist_len(routerlist->old_routers));
3358 /* Now we might have to look at routerlist->old_routers for extraneous
3359 * members. (We'd keep all the members if we could, but we need to save
3360 * space.) First, check whether we have too many router descriptors, total.
3361 * We're okay with having too many for some given router, so long as the
3362 * total number doesn't approach max_descriptors_per_router()*len(router).
3364 if (smartlist_len(routerlist->old_routers) <
3365 smartlist_len(routerlist->routers))
3366 goto done;
3368 /* Sort by identity, then fix indices. */
3369 smartlist_sort(routerlist->old_routers, _compare_old_routers_by_identity);
3370 /* Fix indices. */
3371 for (i = 0; i < smartlist_len(routerlist->old_routers); ++i) {
3372 signed_descriptor_t *r = smartlist_get(routerlist->old_routers, i);
3373 r->routerlist_index = i;
3376 /* Iterate through the list from back to front, so when we remove descriptors
3377 * we don't mess up groups we haven't gotten to. */
3378 for (i = smartlist_len(routerlist->old_routers)-1; i >= 0; --i) {
3379 signed_descriptor_t *r = smartlist_get(routerlist->old_routers, i);
3380 if (!cur_id) {
3381 cur_id = r->identity_digest;
3382 hi = i;
3384 if (memcmp(cur_id, r->identity_digest, DIGEST_LEN)) {
3385 routerlist_remove_old_cached_routers_with_id(now,
3386 cutoff, i+1, hi, retain);
3387 cur_id = r->identity_digest;
3388 hi = i;
3391 if (hi>=0)
3392 routerlist_remove_old_cached_routers_with_id(now, cutoff, 0, hi, retain);
3393 //routerlist_assert_ok(routerlist);
3395 done:
3396 digestset_free(retain);
3397 router_rebuild_store(RRS_DONT_REMOVE_OLD, &routerlist->desc_store);
3398 router_rebuild_store(RRS_DONT_REMOVE_OLD,&routerlist->extrainfo_store);
3401 /** We just added a new set of descriptors. Take whatever extra steps
3402 * we need. */
3403 static void
3404 routerlist_descriptors_added(smartlist_t *sl, int from_cache)
3406 tor_assert(sl);
3407 control_event_descriptors_changed(sl);
3408 SMARTLIST_FOREACH(sl, routerinfo_t *, ri,
3409 if (ri->purpose == ROUTER_PURPOSE_BRIDGE)
3410 learned_bridge_descriptor(ri, from_cache);
3415 * Code to parse a single router descriptor and insert it into the
3416 * routerlist. Return -1 if the descriptor was ill-formed; 0 if the
3417 * descriptor was well-formed but could not be added; and 1 if the
3418 * descriptor was added.
3420 * If we don't add it and <b>msg</b> is not NULL, then assign to
3421 * *<b>msg</b> a static string describing the reason for refusing the
3422 * descriptor.
3424 * This is used only by the controller.
3427 router_load_single_router(const char *s, uint8_t purpose, int cache,
3428 const char **msg)
3430 routerinfo_t *ri;
3431 was_router_added_t r;
3432 smartlist_t *lst;
3433 char annotation_buf[ROUTER_ANNOTATION_BUF_LEN];
3434 tor_assert(msg);
3435 *msg = NULL;
3437 tor_snprintf(annotation_buf, sizeof(annotation_buf),
3438 "@source controller\n"
3439 "@purpose %s\n", router_purpose_to_string(purpose));
3441 if (!(ri = router_parse_entry_from_string(s, NULL, 1, 0, annotation_buf))) {
3442 log_warn(LD_DIR, "Error parsing router descriptor; dropping.");
3443 *msg = "Couldn't parse router descriptor.";
3444 return -1;
3446 tor_assert(ri->purpose == purpose);
3447 if (router_is_me(ri)) {
3448 log_warn(LD_DIR, "Router's identity key matches mine; dropping.");
3449 *msg = "Router's identity key matches mine.";
3450 routerinfo_free(ri);
3451 return 0;
3454 if (!cache) /* obey the preference of the controller */
3455 ri->cache_info.do_not_cache = 1;
3457 lst = smartlist_create();
3458 smartlist_add(lst, ri);
3459 routers_update_status_from_consensus_networkstatus(lst, 0);
3461 r = router_add_to_routerlist(ri, msg, 0, 0);
3462 if (!WRA_WAS_ADDED(r)) {
3463 /* we've already assigned to *msg now, and ri is already freed */
3464 tor_assert(*msg);
3465 if (r == ROUTER_AUTHDIR_REJECTS)
3466 log_warn(LD_DIR, "Couldn't add router to list: %s Dropping.", *msg);
3467 smartlist_free(lst);
3468 return 0;
3469 } else {
3470 routerlist_descriptors_added(lst, 0);
3471 smartlist_free(lst);
3472 log_debug(LD_DIR, "Added router to list");
3473 return 1;
3477 /** Given a string <b>s</b> containing some routerdescs, parse it and put the
3478 * routers into our directory. If saved_location is SAVED_NOWHERE, the routers
3479 * are in response to a query to the network: cache them by adding them to
3480 * the journal.
3482 * Return the number of routers actually added.
3484 * If <b>requested_fingerprints</b> is provided, it must contain a list of
3485 * uppercased fingerprints. Do not update any router whose
3486 * fingerprint is not on the list; after updating a router, remove its
3487 * fingerprint from the list.
3489 * If <b>descriptor_digests</b> is non-zero, then the requested_fingerprints
3490 * are descriptor digests. Otherwise they are identity digests.
3493 router_load_routers_from_string(const char *s, const char *eos,
3494 saved_location_t saved_location,
3495 smartlist_t *requested_fingerprints,
3496 int descriptor_digests,
3497 const char *prepend_annotations)
3499 smartlist_t *routers = smartlist_create(), *changed = smartlist_create();
3500 char fp[HEX_DIGEST_LEN+1];
3501 const char *msg;
3502 int from_cache = (saved_location != SAVED_NOWHERE);
3503 int allow_annotations = (saved_location != SAVED_NOWHERE);
3504 int any_changed = 0;
3506 router_parse_list_from_string(&s, eos, routers, saved_location, 0,
3507 allow_annotations, prepend_annotations);
3509 routers_update_status_from_consensus_networkstatus(routers, !from_cache);
3511 log_info(LD_DIR, "%d elements to add", smartlist_len(routers));
3513 SMARTLIST_FOREACH_BEGIN(routers, routerinfo_t *, ri) {
3514 was_router_added_t r;
3515 if (requested_fingerprints) {
3516 base16_encode(fp, sizeof(fp), descriptor_digests ?
3517 ri->cache_info.signed_descriptor_digest :
3518 ri->cache_info.identity_digest,
3519 DIGEST_LEN);
3520 if (smartlist_string_isin(requested_fingerprints, fp)) {
3521 smartlist_string_remove(requested_fingerprints, fp);
3522 } else {
3523 char *requested =
3524 smartlist_join_strings(requested_fingerprints," ",0,NULL);
3525 log_warn(LD_DIR,
3526 "We received a router descriptor with a fingerprint (%s) "
3527 "that we never requested. (We asked for: %s.) Dropping.",
3528 fp, requested);
3529 tor_free(requested);
3530 routerinfo_free(ri);
3531 continue;
3535 r = router_add_to_routerlist(ri, &msg, from_cache, !from_cache);
3536 if (WRA_WAS_ADDED(r)) {
3537 any_changed++;
3538 smartlist_add(changed, ri);
3539 routerlist_descriptors_added(changed, from_cache);
3540 smartlist_clear(changed);
3541 } else if (WRA_WAS_REJECTED(r)) {
3542 download_status_t *dl_status;
3543 dl_status = router_get_dl_status_by_descriptor_digest(
3544 ri->cache_info.signed_descriptor_digest);
3545 if (dl_status) {
3546 log_info(LD_GENERAL, "Marking router %s as never downloadable",
3547 ri->cache_info.signed_descriptor_digest);
3548 download_status_mark_impossible(dl_status);
3551 } SMARTLIST_FOREACH_END(ri);
3553 routerlist_assert_ok(routerlist);
3555 if (any_changed)
3556 router_rebuild_store(0, &routerlist->desc_store);
3558 smartlist_free(routers);
3559 smartlist_free(changed);
3561 return any_changed;
3564 /** Parse one or more extrainfos from <b>s</b> (ending immediately before
3565 * <b>eos</b> if <b>eos</b> is present). Other arguments are as for
3566 * router_load_routers_from_string(). */
3567 void
3568 router_load_extrainfo_from_string(const char *s, const char *eos,
3569 saved_location_t saved_location,
3570 smartlist_t *requested_fingerprints,
3571 int descriptor_digests)
3573 smartlist_t *extrainfo_list = smartlist_create();
3574 const char *msg;
3575 int from_cache = (saved_location != SAVED_NOWHERE);
3577 router_parse_list_from_string(&s, eos, extrainfo_list, saved_location, 1, 0,
3578 NULL);
3580 log_info(LD_DIR, "%d elements to add", smartlist_len(extrainfo_list));
3582 SMARTLIST_FOREACH(extrainfo_list, extrainfo_t *, ei, {
3583 was_router_added_t added =
3584 router_add_extrainfo_to_routerlist(ei, &msg, from_cache, !from_cache);
3585 if (WRA_WAS_ADDED(added) && requested_fingerprints) {
3586 char fp[HEX_DIGEST_LEN+1];
3587 base16_encode(fp, sizeof(fp), descriptor_digests ?
3588 ei->cache_info.signed_descriptor_digest :
3589 ei->cache_info.identity_digest,
3590 DIGEST_LEN);
3591 smartlist_string_remove(requested_fingerprints, fp);
3592 /* We silently let people stuff us with extrainfos we didn't ask for,
3593 * so long as we would have wanted them anyway. Since we always fetch
3594 * all the extrainfos we want, and we never actually act on them
3595 * inside Tor, this should be harmless. */
3599 routerlist_assert_ok(routerlist);
3600 router_rebuild_store(0, &router_get_routerlist()->extrainfo_store);
3602 smartlist_free(extrainfo_list);
3605 /** Return true iff any networkstatus includes a descriptor whose digest
3606 * is that of <b>desc</b>. */
3607 static int
3608 signed_desc_digest_is_recognized(signed_descriptor_t *desc)
3610 routerstatus_t *rs;
3611 networkstatus_t *consensus = networkstatus_get_latest_consensus();
3612 int caches = directory_caches_dir_info(get_options());
3613 const smartlist_t *networkstatus_v2_list = networkstatus_get_v2_list();
3615 if (consensus) {
3616 rs = networkstatus_vote_find_entry(consensus, desc->identity_digest);
3617 if (rs && !memcmp(rs->descriptor_digest,
3618 desc->signed_descriptor_digest, DIGEST_LEN))
3619 return 1;
3621 if (caches && networkstatus_v2_list) {
3622 SMARTLIST_FOREACH(networkstatus_v2_list, networkstatus_v2_t *, ns,
3624 if (!(rs = networkstatus_v2_find_entry(ns, desc->identity_digest)))
3625 continue;
3626 if (!memcmp(rs->descriptor_digest,
3627 desc->signed_descriptor_digest, DIGEST_LEN))
3628 return 1;
3631 return 0;
3634 /** Clear all our timeouts for fetching v2 and v3 directory stuff, and then
3635 * give it all a try again. */
3636 void
3637 routerlist_retry_directory_downloads(time_t now)
3639 router_reset_status_download_failures();
3640 router_reset_descriptor_download_failures();
3641 update_networkstatus_downloads(now);
3642 update_router_descriptor_downloads(now);
3645 /** Return 1 if all running sufficiently-stable routers will reject
3646 * addr:port, return 0 if any might accept it. */
3648 router_exit_policy_all_routers_reject(uint32_t addr, uint16_t port,
3649 int need_uptime)
3651 addr_policy_result_t r;
3652 if (!routerlist) return 1;
3654 SMARTLIST_FOREACH(routerlist->routers, routerinfo_t *, router,
3656 if (router->is_running &&
3657 !router_is_unreliable(router, need_uptime, 0, 0)) {
3658 r = compare_addr_to_addr_policy(addr, port, router->exit_policy);
3659 if (r != ADDR_POLICY_REJECTED && r != ADDR_POLICY_PROBABLY_REJECTED)
3660 return 0; /* this one could be ok. good enough. */
3663 return 1; /* all will reject. */
3666 /** Return true iff <b>router</b> does not permit exit streams.
3669 router_exit_policy_rejects_all(routerinfo_t *router)
3671 return router->policy_is_reject_star;
3674 /** Add to the list of authorized directory servers one at
3675 * <b>address</b>:<b>port</b>, with identity key <b>digest</b>. If
3676 * <b>address</b> is NULL, add ourself. Return 0 if success, -1 if
3677 * we couldn't add it. */
3678 trusted_dir_server_t *
3679 add_trusted_dir_server(const char *nickname, const char *address,
3680 uint16_t dir_port, uint16_t or_port,
3681 const char *digest, const char *v3_auth_digest,
3682 authority_type_t type)
3684 trusted_dir_server_t *ent;
3685 uint32_t a;
3686 char *hostname = NULL;
3687 size_t dlen;
3688 if (!trusted_dir_servers)
3689 trusted_dir_servers = smartlist_create();
3691 if (!address) { /* The address is us; we should guess. */
3692 if (resolve_my_address(LOG_WARN, get_options(), &a, &hostname) < 0) {
3693 log_warn(LD_CONFIG,
3694 "Couldn't find a suitable address when adding ourself as a "
3695 "trusted directory server.");
3696 return NULL;
3698 } else {
3699 if (tor_lookup_hostname(address, &a)) {
3700 log_warn(LD_CONFIG,
3701 "Unable to lookup address for directory server at '%s'",
3702 address);
3703 return NULL;
3705 hostname = tor_strdup(address);
3708 ent = tor_malloc_zero(sizeof(trusted_dir_server_t));
3709 ent->nickname = nickname ? tor_strdup(nickname) : NULL;
3710 ent->address = hostname;
3711 ent->addr = a;
3712 ent->dir_port = dir_port;
3713 ent->or_port = or_port;
3714 ent->is_running = 1;
3715 ent->type = type;
3716 memcpy(ent->digest, digest, DIGEST_LEN);
3717 if (v3_auth_digest && (type & V3_AUTHORITY))
3718 memcpy(ent->v3_identity_digest, v3_auth_digest, DIGEST_LEN);
3720 dlen = 64 + strlen(hostname) + (nickname?strlen(nickname):0);
3721 ent->description = tor_malloc(dlen);
3722 if (nickname)
3723 tor_snprintf(ent->description, dlen, "directory server \"%s\" at %s:%d",
3724 nickname, hostname, (int)dir_port);
3725 else
3726 tor_snprintf(ent->description, dlen, "directory server at %s:%d",
3727 hostname, (int)dir_port);
3729 ent->fake_status.addr = ent->addr;
3730 memcpy(ent->fake_status.identity_digest, digest, DIGEST_LEN);
3731 if (nickname)
3732 strlcpy(ent->fake_status.nickname, nickname,
3733 sizeof(ent->fake_status.nickname));
3734 else
3735 ent->fake_status.nickname[0] = '\0';
3736 ent->fake_status.dir_port = ent->dir_port;
3737 ent->fake_status.or_port = ent->or_port;
3739 if (ent->or_port)
3740 ent->fake_status.version_supports_begindir = 1;
3741 /* XX021 - wait until authorities are upgraded */
3742 #if 0
3743 ent->fake_status.version_supports_conditional_consensus = 1;
3744 #else
3745 ent->fake_status.version_supports_conditional_consensus = 0;
3746 #endif
3748 smartlist_add(trusted_dir_servers, ent);
3749 router_dir_info_changed();
3750 return ent;
3753 /** Free storage held in <b>cert</b>. */
3754 void
3755 authority_cert_free(authority_cert_t *cert)
3757 if (!cert)
3758 return;
3760 tor_free(cert->cache_info.signed_descriptor_body);
3761 if (cert->signing_key)
3762 crypto_free_pk_env(cert->signing_key);
3763 if (cert->identity_key)
3764 crypto_free_pk_env(cert->identity_key);
3766 tor_free(cert);
3769 /** Free storage held in <b>ds</b>. */
3770 static void
3771 trusted_dir_server_free(trusted_dir_server_t *ds)
3773 tor_free(ds->nickname);
3774 tor_free(ds->description);
3775 tor_free(ds->address);
3776 tor_free(ds);
3779 /** Remove all members from the list of trusted dir servers. */
3780 void
3781 clear_trusted_dir_servers(void)
3783 if (trusted_dir_servers) {
3784 SMARTLIST_FOREACH(trusted_dir_servers, trusted_dir_server_t *, ent,
3785 trusted_dir_server_free(ent));
3786 smartlist_clear(trusted_dir_servers);
3787 } else {
3788 trusted_dir_servers = smartlist_create();
3790 router_dir_info_changed();
3793 /** Return 1 if any trusted dir server supports v1 directories,
3794 * else return 0. */
3796 any_trusted_dir_is_v1_authority(void)
3798 if (trusted_dir_servers)
3799 return get_n_authorities(V1_AUTHORITY) > 0;
3801 return 0;
3804 /** For every current directory connection whose purpose is <b>purpose</b>,
3805 * and where the resource being downloaded begins with <b>prefix</b>, split
3806 * rest of the resource into base16 fingerprints, decode them, and set the
3807 * corresponding elements of <b>result</b> to a nonzero value. */
3808 static void
3809 list_pending_downloads(digestmap_t *result,
3810 int purpose, const char *prefix)
3812 const size_t p_len = strlen(prefix);
3813 smartlist_t *tmp = smartlist_create();
3814 smartlist_t *conns = get_connection_array();
3816 tor_assert(result);
3818 SMARTLIST_FOREACH(conns, connection_t *, conn,
3820 if (conn->type == CONN_TYPE_DIR &&
3821 conn->purpose == purpose &&
3822 !conn->marked_for_close) {
3823 const char *resource = TO_DIR_CONN(conn)->requested_resource;
3824 if (!strcmpstart(resource, prefix))
3825 dir_split_resource_into_fingerprints(resource + p_len,
3826 tmp, NULL, 1, 0);
3829 SMARTLIST_FOREACH(tmp, char *, d,
3831 digestmap_set(result, d, (void*)1);
3832 tor_free(d);
3834 smartlist_free(tmp);
3837 /** For every router descriptor (or extra-info document if <b>extrainfo</b> is
3838 * true) we are currently downloading by descriptor digest, set result[d] to
3839 * (void*)1. */
3840 static void
3841 list_pending_descriptor_downloads(digestmap_t *result, int extrainfo)
3843 int purpose =
3844 extrainfo ? DIR_PURPOSE_FETCH_EXTRAINFO : DIR_PURPOSE_FETCH_SERVERDESC;
3845 list_pending_downloads(result, purpose, "d/");
3848 /** Launch downloads for all the descriptors whose digests are listed
3849 * as digests[i] for lo <= i < hi. (Lo and hi may be out of range.)
3850 * If <b>source</b> is given, download from <b>source</b>; otherwise,
3851 * download from an appropriate random directory server.
3853 static void
3854 initiate_descriptor_downloads(routerstatus_t *source,
3855 int purpose,
3856 smartlist_t *digests,
3857 int lo, int hi, int pds_flags)
3859 int i, n = hi-lo;
3860 char *resource, *cp;
3861 size_t r_len;
3862 if (n <= 0)
3863 return;
3864 if (lo < 0)
3865 lo = 0;
3866 if (hi > smartlist_len(digests))
3867 hi = smartlist_len(digests);
3869 r_len = 8 + (HEX_DIGEST_LEN+1)*n;
3870 cp = resource = tor_malloc(r_len);
3871 memcpy(cp, "d/", 2);
3872 cp += 2;
3873 for (i = lo; i < hi; ++i) {
3874 base16_encode(cp, r_len-(cp-resource),
3875 smartlist_get(digests,i), DIGEST_LEN);
3876 cp += HEX_DIGEST_LEN;
3877 *cp++ = '+';
3879 memcpy(cp-1, ".z", 3);
3881 if (source) {
3882 /* We know which authority we want. */
3883 directory_initiate_command_routerstatus(source, purpose,
3884 ROUTER_PURPOSE_GENERAL,
3885 0, /* not private */
3886 resource, NULL, 0, 0);
3887 } else {
3888 directory_get_from_dirserver(purpose, ROUTER_PURPOSE_GENERAL, resource,
3889 pds_flags);
3891 tor_free(resource);
3894 /** Return 0 if this routerstatus is obsolete, too new, isn't
3895 * running, or otherwise not a descriptor that we would make any
3896 * use of even if we had it. Else return 1. */
3897 static INLINE int
3898 client_would_use_router(routerstatus_t *rs, time_t now, or_options_t *options)
3900 if (!rs->is_running && !options->FetchUselessDescriptors) {
3901 /* If we had this router descriptor, we wouldn't even bother using it.
3902 * But, if we want to have a complete list, fetch it anyway. */
3903 return 0;
3905 if (rs->published_on + options->TestingEstimatedDescriptorPropagationTime
3906 > now) {
3907 /* Most caches probably don't have this descriptor yet. */
3908 return 0;
3910 if (rs->published_on + OLD_ROUTER_DESC_MAX_AGE < now) {
3911 /* We'd drop it immediately for being too old. */
3912 return 0;
3914 return 1;
3917 /** Max amount of hashes to download per request.
3918 * Since squid does not like URLs >= 4096 bytes we limit it to 96.
3919 * 4096 - strlen(http://255.255.255.255/tor/server/d/.z) == 4058
3920 * 4058/41 (40 for the hash and 1 for the + that separates them) => 98
3921 * So use 96 because it's a nice number.
3923 #define MAX_DL_PER_REQUEST 96
3924 /** Don't split our requests so finely that we are requesting fewer than
3925 * this number per server. */
3926 #define MIN_DL_PER_REQUEST 4
3927 /** To prevent a single screwy cache from confusing us by selective reply,
3928 * try to split our requests into at least this this many requests. */
3929 #define MIN_REQUESTS 3
3930 /** If we want fewer than this many descriptors, wait until we
3931 * want more, or until MAX_CLIENT_INTERVAL_WITHOUT_REQUEST has
3932 * passed. */
3933 #define MAX_DL_TO_DELAY 16
3934 /** When directory clients have only a few servers to request, they batch
3935 * them until they have more, or until this amount of time has passed. */
3936 #define MAX_CLIENT_INTERVAL_WITHOUT_REQUEST (10*60)
3938 /** Given a list of router descriptor digests in <b>downloadable</b>, decide
3939 * whether to delay fetching until we have more. If we don't want to delay,
3940 * launch one or more requests to the appropriate directory authorities. */
3941 static void
3942 launch_router_descriptor_downloads(smartlist_t *downloadable, time_t now)
3944 int should_delay = 0, n_downloadable;
3945 or_options_t *options = get_options();
3947 n_downloadable = smartlist_len(downloadable);
3948 if (!directory_fetches_dir_info_early(options)) {
3949 if (n_downloadable >= MAX_DL_TO_DELAY) {
3950 log_debug(LD_DIR,
3951 "There are enough downloadable routerdescs to launch requests.");
3952 should_delay = 0;
3953 } else {
3954 should_delay = (last_routerdesc_download_attempted +
3955 MAX_CLIENT_INTERVAL_WITHOUT_REQUEST) > now;
3956 if (!should_delay && n_downloadable) {
3957 if (last_routerdesc_download_attempted) {
3958 log_info(LD_DIR,
3959 "There are not many downloadable routerdescs, but we've "
3960 "been waiting long enough (%d seconds). Downloading.",
3961 (int)(now-last_routerdesc_download_attempted));
3962 } else {
3963 log_info(LD_DIR,
3964 "There are not many downloadable routerdescs, but we haven't "
3965 "tried downloading descriptors recently. Downloading.");
3970 /* XXX should we consider having even the dir mirrors delay
3971 * a little bit, so we don't load the authorities as much? -RD
3972 * I don't think so. If we do, clients that want those descriptors may
3973 * not actually find them if the caches haven't got them yet. -NM
3976 if (! should_delay && n_downloadable) {
3977 int i, n_per_request;
3978 const char *req_plural = "", *rtr_plural = "";
3979 int pds_flags = PDS_RETRY_IF_NO_SERVERS;
3980 if (! authdir_mode_any_nonhidserv(options)) {
3981 /* If we wind up going to the authorities, we want to only open one
3982 * connection to each authority at a time, so that we don't overload
3983 * them. We do this by setting PDS_NO_EXISTING_SERVERDESC_FETCH
3984 * regardless of whether we're a cache or not; it gets ignored if we're
3985 * not calling router_pick_trusteddirserver.
3987 * Setting this flag can make initiate_descriptor_downloads() ignore
3988 * requests. We need to make sure that we do in fact call
3989 * update_router_descriptor_downloads() later on, once the connections
3990 * have succeeded or failed.
3992 pds_flags |= PDS_NO_EXISTING_SERVERDESC_FETCH;
3995 n_per_request = (n_downloadable+MIN_REQUESTS-1) / MIN_REQUESTS;
3996 if (n_per_request > MAX_DL_PER_REQUEST)
3997 n_per_request = MAX_DL_PER_REQUEST;
3998 if (n_per_request < MIN_DL_PER_REQUEST)
3999 n_per_request = MIN_DL_PER_REQUEST;
4001 if (n_downloadable > n_per_request)
4002 req_plural = rtr_plural = "s";
4003 else if (n_downloadable > 1)
4004 rtr_plural = "s";
4006 log_info(LD_DIR,
4007 "Launching %d request%s for %d router%s, %d at a time",
4008 (n_downloadable+n_per_request-1)/n_per_request,
4009 req_plural, n_downloadable, rtr_plural, n_per_request);
4010 smartlist_sort_digests(downloadable);
4011 for (i=0; i < n_downloadable; i += n_per_request) {
4012 initiate_descriptor_downloads(NULL, DIR_PURPOSE_FETCH_SERVERDESC,
4013 downloadable, i, i+n_per_request,
4014 pds_flags);
4016 last_routerdesc_download_attempted = now;
4020 /** Launch downloads for router status as needed, using the strategy used by
4021 * authorities and caches: based on the v2 networkstatuses we have, download
4022 * every descriptor we don't have but would serve, from a random authority
4023 * that lists it. */
4024 static void
4025 update_router_descriptor_cache_downloads_v2(time_t now)
4027 smartlist_t **downloadable; /* For each authority, what can we dl from it? */
4028 smartlist_t **download_from; /* ... and, what will we dl from it? */
4029 digestmap_t *map; /* Which descs are in progress, or assigned? */
4030 int i, j, n;
4031 int n_download;
4032 or_options_t *options = get_options();
4033 const smartlist_t *networkstatus_v2_list = networkstatus_get_v2_list();
4035 if (! directory_fetches_dir_info_early(options)) {
4036 log_warn(LD_BUG, "Called update_router_descriptor_cache_downloads_v2() "
4037 "on a non-dir-mirror?");
4040 if (!networkstatus_v2_list || !smartlist_len(networkstatus_v2_list))
4041 return;
4043 map = digestmap_new();
4044 n = smartlist_len(networkstatus_v2_list);
4046 downloadable = tor_malloc_zero(sizeof(smartlist_t*) * n);
4047 download_from = tor_malloc_zero(sizeof(smartlist_t*) * n);
4049 /* Set map[d]=1 for the digest of every descriptor that we are currently
4050 * downloading. */
4051 list_pending_descriptor_downloads(map, 0);
4053 /* For the digest of every descriptor that we don't have, and that we aren't
4054 * downloading, add d to downloadable[i] if the i'th networkstatus knows
4055 * about that descriptor, and we haven't already failed to get that
4056 * descriptor from the corresponding authority.
4058 n_download = 0;
4059 SMARTLIST_FOREACH(networkstatus_v2_list, networkstatus_v2_t *, ns,
4061 trusted_dir_server_t *ds;
4062 smartlist_t *dl;
4063 dl = downloadable[ns_sl_idx] = smartlist_create();
4064 download_from[ns_sl_idx] = smartlist_create();
4065 if (ns->published_on + MAX_NETWORKSTATUS_AGE+10*60 < now) {
4066 /* Don't download if the networkstatus is almost ancient. */
4067 /* Actually, I suspect what's happening here is that we ask
4068 * for the descriptor when we have a given networkstatus,
4069 * and then we get a newer networkstatus, and then we receive
4070 * the descriptor. Having a networkstatus actually expire is
4071 * probably a rare event, and we'll probably be happiest if
4072 * we take this clause out. -RD */
4073 continue;
4076 /* Don't try dirservers that we think are down -- we might have
4077 * just tried them and just marked them as down. */
4078 ds = router_get_trusteddirserver_by_digest(ns->identity_digest);
4079 if (ds && !ds->is_running)
4080 continue;
4082 SMARTLIST_FOREACH(ns->entries, routerstatus_t * , rs,
4084 if (!rs->need_to_mirror)
4085 continue;
4086 if (router_get_by_descriptor_digest(rs->descriptor_digest)) {
4087 log_warn(LD_BUG,
4088 "We have a router descriptor, but need_to_mirror=1.");
4089 rs->need_to_mirror = 0;
4090 continue;
4092 if (authdir_mode(options) && dirserv_would_reject_router(rs)) {
4093 rs->need_to_mirror = 0;
4094 continue;
4096 if (digestmap_get(map, rs->descriptor_digest)) {
4097 /* We're downloading it already. */
4098 continue;
4099 } else {
4100 /* We could download it from this guy. */
4101 smartlist_add(dl, rs->descriptor_digest);
4102 ++n_download;
4107 /* At random, assign descriptors to authorities such that:
4108 * - if d is a member of some downloadable[x], d is a member of some
4109 * download_from[y]. (Everything we want to download, we try to download
4110 * from somebody.)
4111 * - If d is a member of download_from[y], d is a member of downloadable[y].
4112 * (We only try to download descriptors from authorities who claim to have
4113 * them.)
4114 * - No d is a member of download_from[x] and download_from[y] s.t. x != y.
4115 * (We don't try to download anything from two authorities concurrently.)
4117 while (n_download) {
4118 int which_ns = crypto_rand_int(n);
4119 smartlist_t *dl = downloadable[which_ns];
4120 int idx;
4121 char *d;
4122 if (!smartlist_len(dl))
4123 continue;
4124 idx = crypto_rand_int(smartlist_len(dl));
4125 d = smartlist_get(dl, idx);
4126 if (! digestmap_get(map, d)) {
4127 smartlist_add(download_from[which_ns], d);
4128 digestmap_set(map, d, (void*) 1);
4130 smartlist_del(dl, idx);
4131 --n_download;
4134 /* Now, we can actually launch our requests. */
4135 for (i=0; i<n; ++i) {
4136 networkstatus_v2_t *ns = smartlist_get(networkstatus_v2_list, i);
4137 trusted_dir_server_t *ds =
4138 router_get_trusteddirserver_by_digest(ns->identity_digest);
4139 smartlist_t *dl = download_from[i];
4140 int pds_flags = PDS_RETRY_IF_NO_SERVERS;
4141 if (! authdir_mode_any_nonhidserv(options))
4142 pds_flags |= PDS_NO_EXISTING_SERVERDESC_FETCH; /* XXXX ignored*/
4144 if (!ds) {
4145 log_warn(LD_BUG, "Networkstatus with no corresponding authority!");
4146 continue;
4148 if (! smartlist_len(dl))
4149 continue;
4150 log_info(LD_DIR, "Requesting %d descriptors from authority \"%s\"",
4151 smartlist_len(dl), ds->nickname);
4152 for (j=0; j < smartlist_len(dl); j += MAX_DL_PER_REQUEST) {
4153 initiate_descriptor_downloads(&(ds->fake_status),
4154 DIR_PURPOSE_FETCH_SERVERDESC, dl, j,
4155 j+MAX_DL_PER_REQUEST, pds_flags);
4159 for (i=0; i<n; ++i) {
4160 smartlist_free(download_from[i]);
4161 smartlist_free(downloadable[i]);
4163 tor_free(download_from);
4164 tor_free(downloadable);
4165 digestmap_free(map,NULL);
4168 /** For any descriptor that we want that's currently listed in the live
4169 * consensus, download it as appropriate. */
4170 static void
4171 update_consensus_router_descriptor_downloads(time_t now)
4173 or_options_t *options = get_options();
4174 digestmap_t *map = NULL;
4175 smartlist_t *no_longer_old = smartlist_create();
4176 smartlist_t *downloadable = smartlist_create();
4177 int authdir = authdir_mode(options);
4178 networkstatus_t *consensus =
4179 networkstatus_get_reasonably_live_consensus(now);
4180 int n_delayed=0, n_have=0, n_would_reject=0, n_wouldnt_use=0,
4181 n_inprogress=0, n_in_oldrouters=0;
4183 if (directory_too_idle_to_fetch_descriptors(options, now))
4184 goto done;
4185 if (!consensus)
4186 goto done;
4188 map = digestmap_new();
4189 list_pending_descriptor_downloads(map, 0);
4190 SMARTLIST_FOREACH(consensus->routerstatus_list, routerstatus_t *, rs,
4192 signed_descriptor_t *sd;
4193 if ((sd = router_get_by_descriptor_digest(rs->descriptor_digest))) {
4194 routerinfo_t *ri;
4195 ++n_have;
4196 if (!(ri = router_get_by_digest(rs->identity_digest)) ||
4197 memcmp(ri->cache_info.signed_descriptor_digest,
4198 sd->signed_descriptor_digest, DIGEST_LEN)) {
4199 /* We have a descriptor with this digest, but either there is no
4200 * entry in routerlist with the same ID (!ri), or there is one,
4201 * but the identity digest differs (memcmp).
4203 smartlist_add(no_longer_old, sd);
4204 ++n_in_oldrouters; /* We have it in old_routers. */
4206 continue; /* We have it already. */
4208 if (digestmap_get(map, rs->descriptor_digest)) {
4209 ++n_inprogress;
4210 continue; /* We have an in-progress download. */
4212 if (!download_status_is_ready(&rs->dl_status, now,
4213 MAX_ROUTERDESC_DOWNLOAD_FAILURES)) {
4214 ++n_delayed; /* Not ready for retry. */
4215 continue;
4217 if (authdir && dirserv_would_reject_router(rs)) {
4218 ++n_would_reject;
4219 continue; /* We would throw it out immediately. */
4221 if (!directory_caches_dir_info(options) &&
4222 !client_would_use_router(rs, now, options)) {
4223 ++n_wouldnt_use;
4224 continue; /* We would never use it ourself. */
4226 smartlist_add(downloadable, rs->descriptor_digest);
4229 if (!authdir_mode_handles_descs(options, ROUTER_PURPOSE_GENERAL)
4230 && smartlist_len(no_longer_old)) {
4231 routerlist_t *rl = router_get_routerlist();
4232 log_info(LD_DIR, "%d router descriptors listed in consensus are "
4233 "currently in old_routers; making them current.",
4234 smartlist_len(no_longer_old));
4235 SMARTLIST_FOREACH(no_longer_old, signed_descriptor_t *, sd, {
4236 const char *msg;
4237 was_router_added_t r;
4238 routerinfo_t *ri = routerlist_reparse_old(rl, sd);
4239 if (!ri) {
4240 log_warn(LD_BUG, "Failed to re-parse a router.");
4241 continue;
4243 r = router_add_to_routerlist(ri, &msg, 1, 0);
4244 if (WRA_WAS_OUTDATED(r)) {
4245 log_warn(LD_DIR, "Couldn't add re-parsed router: %s",
4246 msg?msg:"???");
4249 routerlist_assert_ok(rl);
4252 log_info(LD_DIR,
4253 "%d router descriptors downloadable. %d delayed; %d present "
4254 "(%d of those were in old_routers); %d would_reject; "
4255 "%d wouldnt_use; %d in progress.",
4256 smartlist_len(downloadable), n_delayed, n_have, n_in_oldrouters,
4257 n_would_reject, n_wouldnt_use, n_inprogress);
4259 launch_router_descriptor_downloads(downloadable, now);
4261 digestmap_free(map, NULL);
4262 done:
4263 smartlist_free(downloadable);
4264 smartlist_free(no_longer_old);
4267 /** How often should we launch a server/authority request to be sure of getting
4268 * a guess for our IP? */
4269 /*XXXX021 this info should come from netinfo cells or something, or we should
4270 * do this only when we aren't seeing incoming data. see bug 652. */
4271 #define DUMMY_DOWNLOAD_INTERVAL (20*60)
4273 /** Launch downloads for router status as needed. */
4274 void
4275 update_router_descriptor_downloads(time_t now)
4277 or_options_t *options = get_options();
4278 static time_t last_dummy_download = 0;
4279 if (should_delay_dir_fetches(options))
4280 return;
4281 if (directory_fetches_dir_info_early(options)) {
4282 update_router_descriptor_cache_downloads_v2(now);
4284 update_consensus_router_descriptor_downloads(now);
4286 /* XXXX021 we could be smarter here; see notes on bug 652. */
4287 /* If we're a server that doesn't have a configured address, we rely on
4288 * directory fetches to learn when our address changes. So if we haven't
4289 * tried to get any routerdescs in a long time, try a dummy fetch now. */
4290 if (!options->Address &&
4291 server_mode(options) &&
4292 last_routerdesc_download_attempted + DUMMY_DOWNLOAD_INTERVAL < now &&
4293 last_dummy_download + DUMMY_DOWNLOAD_INTERVAL < now) {
4294 last_dummy_download = now;
4295 directory_get_from_dirserver(DIR_PURPOSE_FETCH_SERVERDESC,
4296 ROUTER_PURPOSE_GENERAL, "authority.z",
4297 PDS_RETRY_IF_NO_SERVERS);
4301 /** Launch extrainfo downloads as needed. */
4302 void
4303 update_extrainfo_downloads(time_t now)
4305 or_options_t *options = get_options();
4306 routerlist_t *rl;
4307 smartlist_t *wanted;
4308 digestmap_t *pending;
4309 int old_routers, i;
4310 int n_no_ei = 0, n_pending = 0, n_have = 0, n_delay = 0;
4311 if (! options->DownloadExtraInfo)
4312 return;
4313 if (should_delay_dir_fetches(options))
4314 return;
4315 if (!router_have_minimum_dir_info())
4316 return;
4318 pending = digestmap_new();
4319 list_pending_descriptor_downloads(pending, 1);
4320 rl = router_get_routerlist();
4321 wanted = smartlist_create();
4322 for (old_routers = 0; old_routers < 2; ++old_routers) {
4323 smartlist_t *lst = old_routers ? rl->old_routers : rl->routers;
4324 for (i = 0; i < smartlist_len(lst); ++i) {
4325 signed_descriptor_t *sd;
4326 char *d;
4327 if (old_routers)
4328 sd = smartlist_get(lst, i);
4329 else
4330 sd = &((routerinfo_t*)smartlist_get(lst, i))->cache_info;
4331 if (sd->is_extrainfo)
4332 continue; /* This should never happen. */
4333 if (old_routers && !router_get_by_digest(sd->identity_digest))
4334 continue; /* Couldn't check the signature if we got it. */
4335 if (sd->extrainfo_is_bogus)
4336 continue;
4337 d = sd->extra_info_digest;
4338 if (tor_digest_is_zero(d)) {
4339 ++n_no_ei;
4340 continue;
4342 if (eimap_get(rl->extra_info_map, d)) {
4343 ++n_have;
4344 continue;
4346 if (!download_status_is_ready(&sd->ei_dl_status, now,
4347 MAX_ROUTERDESC_DOWNLOAD_FAILURES)) {
4348 ++n_delay;
4349 continue;
4351 if (digestmap_get(pending, d)) {
4352 ++n_pending;
4353 continue;
4355 smartlist_add(wanted, d);
4358 digestmap_free(pending, NULL);
4360 log_info(LD_DIR, "Extrainfo download status: %d router with no ei, %d "
4361 "with present ei, %d delaying, %d pending, %d downloadable.",
4362 n_no_ei, n_have, n_delay, n_pending, smartlist_len(wanted));
4364 smartlist_shuffle(wanted);
4365 for (i = 0; i < smartlist_len(wanted); i += MAX_DL_PER_REQUEST) {
4366 initiate_descriptor_downloads(NULL, DIR_PURPOSE_FETCH_EXTRAINFO,
4367 wanted, i, i + MAX_DL_PER_REQUEST,
4368 PDS_RETRY_IF_NO_SERVERS|PDS_NO_EXISTING_SERVERDESC_FETCH);
4371 smartlist_free(wanted);
4374 /** True iff, the last time we checked whether we had enough directory info
4375 * to build circuits, the answer was "yes". */
4376 static int have_min_dir_info = 0;
4377 /** True iff enough has changed since the last time we checked whether we had
4378 * enough directory info to build circuits that our old answer can no longer
4379 * be trusted. */
4380 static int need_to_update_have_min_dir_info = 1;
4381 /** String describing what we're missing before we have enough directory
4382 * info. */
4383 static char dir_info_status[128] = "";
4385 /** Return true iff we have enough networkstatus and router information to
4386 * start building circuits. Right now, this means "more than half the
4387 * networkstatus documents, and at least 1/4 of expected routers." */
4388 //XXX should consider whether we have enough exiting nodes here.
4390 router_have_minimum_dir_info(void)
4392 if (PREDICT_UNLIKELY(need_to_update_have_min_dir_info)) {
4393 update_router_have_minimum_dir_info();
4394 need_to_update_have_min_dir_info = 0;
4396 return have_min_dir_info;
4399 /** Called when our internal view of the directory has changed. This can be
4400 * when the authorities change, networkstatuses change, the list of routerdescs
4401 * changes, or number of running routers changes.
4403 void
4404 router_dir_info_changed(void)
4406 need_to_update_have_min_dir_info = 1;
4407 rend_hsdir_routers_changed();
4410 /** Return a string describing what we're missing before we have enough
4411 * directory info. */
4412 const char *
4413 get_dir_info_status_string(void)
4415 return dir_info_status;
4418 /** Iterate over the servers listed in <b>consensus</b>, and count how many of
4419 * them seem like ones we'd use, and how many of <em>those</em> we have
4420 * descriptors for. Store the former in *<b>num_usable</b> and the latter in
4421 * *<b>num_present</b>. */
4422 static void
4423 count_usable_descriptors(int *num_present, int *num_usable,
4424 const networkstatus_t *consensus,
4425 or_options_t *options, time_t now)
4427 *num_present = 0, *num_usable=0;
4429 SMARTLIST_FOREACH(consensus->routerstatus_list, routerstatus_t *, rs,
4431 if (client_would_use_router(rs, now, options)) {
4432 ++*num_usable; /* the consensus says we want it. */
4433 if (router_get_by_descriptor_digest(rs->descriptor_digest)) {
4434 /* we have the descriptor listed in the consensus. */
4435 ++*num_present;
4440 log_debug(LD_DIR, "%d usable, %d present.", *num_usable, *num_present);
4443 /** We just fetched a new set of descriptors. Compute how far through
4444 * the "loading descriptors" bootstrapping phase we are, so we can inform
4445 * the controller of our progress. */
4447 count_loading_descriptors_progress(void)
4449 int num_present = 0, num_usable=0;
4450 time_t now = time(NULL);
4451 const networkstatus_t *consensus =
4452 networkstatus_get_reasonably_live_consensus(now);
4453 double fraction;
4455 if (!consensus)
4456 return 0; /* can't count descriptors if we have no list of them */
4458 count_usable_descriptors(&num_present, &num_usable,
4459 consensus, get_options(), now);
4461 if (num_usable == 0)
4462 return 0; /* don't div by 0 */
4463 fraction = num_present / (num_usable/4.);
4464 if (fraction > 1.0)
4465 return 0; /* it's not the number of descriptors holding us back */
4466 return BOOTSTRAP_STATUS_LOADING_DESCRIPTORS + (int)
4467 (fraction*(BOOTSTRAP_STATUS_CONN_OR-1 -
4468 BOOTSTRAP_STATUS_LOADING_DESCRIPTORS));
4471 /** Change the value of have_min_dir_info, setting it true iff we have enough
4472 * network and router information to build circuits. Clear the value of
4473 * need_to_update_have_min_dir_info. */
4474 static void
4475 update_router_have_minimum_dir_info(void)
4477 int num_present = 0, num_usable=0;
4478 time_t now = time(NULL);
4479 int res;
4480 or_options_t *options = get_options();
4481 const networkstatus_t *consensus =
4482 networkstatus_get_reasonably_live_consensus(now);
4484 if (!consensus) {
4485 if (!networkstatus_get_latest_consensus())
4486 strlcpy(dir_info_status, "We have no network-status consensus.",
4487 sizeof(dir_info_status));
4488 else
4489 strlcpy(dir_info_status, "We have no recent network-status consensus.",
4490 sizeof(dir_info_status));
4491 res = 0;
4492 goto done;
4495 if (should_delay_dir_fetches(get_options())) {
4496 log_notice(LD_DIR, "no known bridge descriptors running yet; stalling");
4497 strlcpy(dir_info_status, "No live bridge descriptors.",
4498 sizeof(dir_info_status));
4499 res = 0;
4500 goto done;
4503 count_usable_descriptors(&num_present, &num_usable, consensus, options, now);
4505 if (num_present < num_usable/4) {
4506 tor_snprintf(dir_info_status, sizeof(dir_info_status),
4507 "We have only %d/%d usable descriptors.", num_present, num_usable);
4508 res = 0;
4509 control_event_bootstrap(BOOTSTRAP_STATUS_REQUESTING_DESCRIPTORS, 0);
4510 } else if (num_present < 2) {
4511 tor_snprintf(dir_info_status, sizeof(dir_info_status),
4512 "Only %d descriptor%s here and believed reachable!",
4513 num_present, num_present ? "" : "s");
4514 res = 0;
4515 } else {
4516 res = 1;
4519 done:
4520 if (res && !have_min_dir_info) {
4521 log(LOG_NOTICE, LD_DIR,
4522 "We now have enough directory information to build circuits.");
4523 control_event_client_status(LOG_NOTICE, "ENOUGH_DIR_INFO");
4524 control_event_bootstrap(BOOTSTRAP_STATUS_CONN_OR, 0);
4526 if (!res && have_min_dir_info) {
4527 int quiet = directory_too_idle_to_fetch_descriptors(options, now);
4528 log(quiet ? LOG_INFO : LOG_NOTICE, LD_DIR,
4529 "Our directory information is no longer up-to-date "
4530 "enough to build circuits: %s", dir_info_status);
4531 control_event_client_status(LOG_NOTICE, "NOT_ENOUGH_DIR_INFO");
4533 have_min_dir_info = res;
4534 need_to_update_have_min_dir_info = 0;
4537 /** Reset the descriptor download failure count on all routers, so that we
4538 * can retry any long-failed routers immediately.
4540 void
4541 router_reset_descriptor_download_failures(void)
4543 networkstatus_reset_download_failures();
4544 last_routerdesc_download_attempted = 0;
4545 if (!routerlist)
4546 return;
4547 SMARTLIST_FOREACH(routerlist->routers, routerinfo_t *, ri,
4549 download_status_reset(&ri->cache_info.ei_dl_status);
4551 SMARTLIST_FOREACH(routerlist->old_routers, signed_descriptor_t *, sd,
4553 download_status_reset(&sd->ei_dl_status);
4557 /** Any changes in a router descriptor's publication time larger than this are
4558 * automatically non-cosmetic. */
4559 #define ROUTER_MAX_COSMETIC_TIME_DIFFERENCE (12*60*60)
4561 /** We allow uptime to vary from how much it ought to be by this much. */
4562 #define ROUTER_ALLOW_UPTIME_DRIFT (6*60*60)
4564 /** Return true iff the only differences between r1 and r2 are such that
4565 * would not cause a recent (post 0.1.1.6) dirserver to republish.
4568 router_differences_are_cosmetic(routerinfo_t *r1, routerinfo_t *r2)
4570 time_t r1pub, r2pub;
4571 long time_difference;
4572 tor_assert(r1 && r2);
4574 /* r1 should be the one that was published first. */
4575 if (r1->cache_info.published_on > r2->cache_info.published_on) {
4576 routerinfo_t *ri_tmp = r2;
4577 r2 = r1;
4578 r1 = ri_tmp;
4581 /* If any key fields differ, they're different. */
4582 if (strcasecmp(r1->address, r2->address) ||
4583 strcasecmp(r1->nickname, r2->nickname) ||
4584 r1->or_port != r2->or_port ||
4585 r1->dir_port != r2->dir_port ||
4586 r1->purpose != r2->purpose ||
4587 crypto_pk_cmp_keys(r1->onion_pkey, r2->onion_pkey) ||
4588 crypto_pk_cmp_keys(r1->identity_pkey, r2->identity_pkey) ||
4589 strcasecmp(r1->platform, r2->platform) ||
4590 (r1->contact_info && !r2->contact_info) || /* contact_info is optional */
4591 (!r1->contact_info && r2->contact_info) ||
4592 (r1->contact_info && r2->contact_info &&
4593 strcasecmp(r1->contact_info, r2->contact_info)) ||
4594 r1->is_hibernating != r2->is_hibernating ||
4595 r1->has_old_dnsworkers != r2->has_old_dnsworkers ||
4596 cmp_addr_policies(r1->exit_policy, r2->exit_policy))
4597 return 0;
4598 if ((r1->declared_family == NULL) != (r2->declared_family == NULL))
4599 return 0;
4600 if (r1->declared_family && r2->declared_family) {
4601 int i, n;
4602 if (smartlist_len(r1->declared_family)!=smartlist_len(r2->declared_family))
4603 return 0;
4604 n = smartlist_len(r1->declared_family);
4605 for (i=0; i < n; ++i) {
4606 if (strcasecmp(smartlist_get(r1->declared_family, i),
4607 smartlist_get(r2->declared_family, i)))
4608 return 0;
4612 /* Did bandwidth change a lot? */
4613 if ((r1->bandwidthcapacity < r2->bandwidthcapacity/2) ||
4614 (r2->bandwidthcapacity < r1->bandwidthcapacity/2))
4615 return 0;
4617 /* Did more than 12 hours pass? */
4618 if (r1->cache_info.published_on + ROUTER_MAX_COSMETIC_TIME_DIFFERENCE
4619 < r2->cache_info.published_on)
4620 return 0;
4622 /* Did uptime fail to increase by approximately the amount we would think,
4623 * give or take some slop? */
4624 r1pub = r1->cache_info.published_on;
4625 r2pub = r2->cache_info.published_on;
4626 time_difference = labs(r2->uptime - (r1->uptime + (r2pub - r1pub)));
4627 if (time_difference > ROUTER_ALLOW_UPTIME_DRIFT &&
4628 time_difference > r1->uptime * .05 &&
4629 time_difference > r2->uptime * .05)
4630 return 0;
4632 /* Otherwise, the difference is cosmetic. */
4633 return 1;
4636 /** Check whether <b>ri</b> (a.k.a. sd) is a router compatible with the
4637 * extrainfo document
4638 * <b>ei</b>. If no router is compatible with <b>ei</b>, <b>ei</b> should be
4639 * dropped. Return 0 for "compatible", return 1 for "reject, and inform
4640 * whoever uploaded <b>ei</b>, and return -1 for "reject silently.". If
4641 * <b>msg</b> is present, set *<b>msg</b> to a description of the
4642 * incompatibility (if any).
4645 routerinfo_incompatible_with_extrainfo(routerinfo_t *ri, extrainfo_t *ei,
4646 signed_descriptor_t *sd,
4647 const char **msg)
4649 int digest_matches, r=1;
4650 tor_assert(ri);
4651 tor_assert(ei);
4652 if (!sd)
4653 sd = &ri->cache_info;
4655 if (ei->bad_sig) {
4656 if (msg) *msg = "Extrainfo signature was bad, or signed with wrong key.";
4657 return 1;
4660 digest_matches = !memcmp(ei->cache_info.signed_descriptor_digest,
4661 sd->extra_info_digest, DIGEST_LEN);
4663 /* The identity must match exactly to have been generated at the same time
4664 * by the same router. */
4665 if (memcmp(ri->cache_info.identity_digest, ei->cache_info.identity_digest,
4666 DIGEST_LEN)) {
4667 if (msg) *msg = "Extrainfo nickname or identity did not match routerinfo";
4668 goto err; /* different servers */
4671 if (ei->pending_sig) {
4672 char signed_digest[128];
4673 if (crypto_pk_public_checksig(ri->identity_pkey, signed_digest,
4674 ei->pending_sig, ei->pending_sig_len) != DIGEST_LEN ||
4675 memcmp(signed_digest, ei->cache_info.signed_descriptor_digest,
4676 DIGEST_LEN)) {
4677 ei->bad_sig = 1;
4678 tor_free(ei->pending_sig);
4679 if (msg) *msg = "Extrainfo signature bad, or signed with wrong key";
4680 goto err; /* Bad signature, or no match. */
4683 ei->cache_info.send_unencrypted = ri->cache_info.send_unencrypted;
4684 tor_free(ei->pending_sig);
4687 if (ei->cache_info.published_on < sd->published_on) {
4688 if (msg) *msg = "Extrainfo published time did not match routerdesc";
4689 goto err;
4690 } else if (ei->cache_info.published_on > sd->published_on) {
4691 if (msg) *msg = "Extrainfo published time did not match routerdesc";
4692 r = -1;
4693 goto err;
4696 if (!digest_matches) {
4697 if (msg) *msg = "Extrainfo digest did not match value from routerdesc";
4698 goto err; /* Digest doesn't match declared value. */
4701 return 0;
4702 err:
4703 if (digest_matches) {
4704 /* This signature was okay, and the digest was right: This is indeed the
4705 * corresponding extrainfo. But insanely, it doesn't match the routerinfo
4706 * that lists it. Don't try to fetch this one again. */
4707 sd->extrainfo_is_bogus = 1;
4710 return r;
4713 /** Assert that the internal representation of <b>rl</b> is
4714 * self-consistent. */
4715 void
4716 routerlist_assert_ok(routerlist_t *rl)
4718 routerinfo_t *r2;
4719 signed_descriptor_t *sd2;
4720 if (!rl)
4721 return;
4722 SMARTLIST_FOREACH(rl->routers, routerinfo_t *, r,
4724 r2 = rimap_get(rl->identity_map, r->cache_info.identity_digest);
4725 tor_assert(r == r2);
4726 sd2 = sdmap_get(rl->desc_digest_map,
4727 r->cache_info.signed_descriptor_digest);
4728 tor_assert(&(r->cache_info) == sd2);
4729 tor_assert(r->cache_info.routerlist_index == r_sl_idx);
4730 /* XXXX
4732 * Hoo boy. We need to fix this one, and the fix is a bit tricky, so
4733 * commenting this out is just a band-aid.
4735 * The problem is that, although well-behaved router descriptors
4736 * should never have the same value for their extra_info_digest, it's
4737 * possible for ill-behaved routers to claim whatever they like there.
4739 * The real answer is to trash desc_by_eid_map and instead have
4740 * something that indicates for a given extra-info digest we want,
4741 * what its download status is. We'll do that as a part of routerlist
4742 * refactoring once consensus directories are in. For now,
4743 * this rep violation is probably harmless: an adversary can make us
4744 * reset our retry count for an extrainfo, but that's not the end
4745 * of the world. Changing the representation in 0.2.0.x would just
4746 * destabilize the codebase.
4747 if (!tor_digest_is_zero(r->cache_info.extra_info_digest)) {
4748 signed_descriptor_t *sd3 =
4749 sdmap_get(rl->desc_by_eid_map, r->cache_info.extra_info_digest);
4750 tor_assert(sd3 == &(r->cache_info));
4754 SMARTLIST_FOREACH(rl->old_routers, signed_descriptor_t *, sd,
4756 r2 = rimap_get(rl->identity_map, sd->identity_digest);
4757 tor_assert(sd != &(r2->cache_info));
4758 sd2 = sdmap_get(rl->desc_digest_map, sd->signed_descriptor_digest);
4759 tor_assert(sd == sd2);
4760 tor_assert(sd->routerlist_index == sd_sl_idx);
4761 /* XXXX see above.
4762 if (!tor_digest_is_zero(sd->extra_info_digest)) {
4763 signed_descriptor_t *sd3 =
4764 sdmap_get(rl->desc_by_eid_map, sd->extra_info_digest);
4765 tor_assert(sd3 == sd);
4770 RIMAP_FOREACH(rl->identity_map, d, r) {
4771 tor_assert(!memcmp(r->cache_info.identity_digest, d, DIGEST_LEN));
4772 } DIGESTMAP_FOREACH_END;
4773 SDMAP_FOREACH(rl->desc_digest_map, d, sd) {
4774 tor_assert(!memcmp(sd->signed_descriptor_digest, d, DIGEST_LEN));
4775 } DIGESTMAP_FOREACH_END;
4776 SDMAP_FOREACH(rl->desc_by_eid_map, d, sd) {
4777 tor_assert(!tor_digest_is_zero(d));
4778 tor_assert(sd);
4779 tor_assert(!memcmp(sd->extra_info_digest, d, DIGEST_LEN));
4780 } DIGESTMAP_FOREACH_END;
4781 EIMAP_FOREACH(rl->extra_info_map, d, ei) {
4782 signed_descriptor_t *sd;
4783 tor_assert(!memcmp(ei->cache_info.signed_descriptor_digest,
4784 d, DIGEST_LEN));
4785 sd = sdmap_get(rl->desc_by_eid_map,
4786 ei->cache_info.signed_descriptor_digest);
4787 // tor_assert(sd); // XXXX see above
4788 if (sd) {
4789 tor_assert(!memcmp(ei->cache_info.signed_descriptor_digest,
4790 sd->extra_info_digest, DIGEST_LEN));
4792 } DIGESTMAP_FOREACH_END;
4795 /** Allocate and return a new string representing the contact info
4796 * and platform string for <b>router</b>,
4797 * surrounded by quotes and using standard C escapes.
4799 * THIS FUNCTION IS NOT REENTRANT. Don't call it from outside the main
4800 * thread. Also, each call invalidates the last-returned value, so don't
4801 * try log_warn(LD_GENERAL, "%s %s", esc_router_info(a), esc_router_info(b));
4803 * If <b>router</b> is NULL, it just frees its internal memory and returns.
4805 const char *
4806 esc_router_info(routerinfo_t *router)
4808 static char *info=NULL;
4809 char *esc_contact, *esc_platform;
4810 size_t len;
4811 if (info)
4812 tor_free(info);
4813 if (!router)
4814 return NULL; /* we're exiting; just free the memory we use */
4816 esc_contact = esc_for_log(router->contact_info);
4817 esc_platform = esc_for_log(router->platform);
4819 len = strlen(esc_contact)+strlen(esc_platform)+32;
4820 info = tor_malloc(len);
4821 tor_snprintf(info, len, "Contact %s, Platform %s", esc_contact,
4822 esc_platform);
4823 tor_free(esc_contact);
4824 tor_free(esc_platform);
4826 return info;
4829 /** Helper for sorting: compare two routerinfos by their identity
4830 * digest. */
4831 static int
4832 _compare_routerinfo_by_id_digest(const void **a, const void **b)
4834 routerinfo_t *first = *(routerinfo_t **)a, *second = *(routerinfo_t **)b;
4835 return memcmp(first->cache_info.identity_digest,
4836 second->cache_info.identity_digest,
4837 DIGEST_LEN);
4840 /** Sort a list of routerinfo_t in ascending order of identity digest. */
4841 void
4842 routers_sort_by_identity(smartlist_t *routers)
4844 smartlist_sort(routers, _compare_routerinfo_by_id_digest);
4847 /** A routerset specifies constraints on a set of possible routerinfos, based
4848 * on their names, identities, or addresses. It is optimized for determining
4849 * whether a router is a member or not, in O(1+P) time, where P is the number
4850 * of address policy constraints. */
4851 struct routerset_t {
4852 /** A list of strings for the elements of the policy. Each string is either
4853 * a nickname, a hexadecimal identity fingerprint, or an address policy. A
4854 * router belongs to the set if its nickname OR its identity OR its address
4855 * matches an entry here. */
4856 smartlist_t *list;
4857 /** A map from lowercase nicknames of routers in the set to (void*)1 */
4858 strmap_t *names;
4859 /** A map from identity digests routers in the set to (void*)1 */
4860 digestmap_t *digests;
4861 /** An address policy for routers in the set. For implementation reasons,
4862 * a router belongs to the set if it is _rejected_ by this policy. */
4863 smartlist_t *policies;
4865 /** A human-readable description of what this routerset is for. Used in
4866 * log messages. */
4867 char *description;
4869 /** A list of the country codes in this set. */
4870 smartlist_t *country_names;
4871 /** Total number of countries we knew about when we built <b>countries</b>.*/
4872 int n_countries;
4873 /** Bit array mapping the return value of geoip_get_country() to 1 iff the
4874 * country is a member of this routerset. Note that we MUST call
4875 * routerset_refresh_countries() whenever the geoip country list is
4876 * reloaded. */
4877 bitarray_t *countries;
4880 /** Return a new empty routerset. */
4881 routerset_t *
4882 routerset_new(void)
4884 routerset_t *result = tor_malloc_zero(sizeof(routerset_t));
4885 result->list = smartlist_create();
4886 result->names = strmap_new();
4887 result->digests = digestmap_new();
4888 result->policies = smartlist_create();
4889 result->country_names = smartlist_create();
4890 return result;
4893 /** If <b>c</b> is a country code in the form {cc}, return a newly allocated
4894 * string holding the "cc" part. Else, return NULL. */
4895 static char *
4896 routerset_get_countryname(const char *c)
4898 char *country;
4900 if (strlen(c) < 4 || c[0] !='{' || c[3] !='}')
4901 return NULL;
4903 country = tor_strndup(c+1, 2);
4904 tor_strlower(country);
4905 return country;
4908 #if 0
4909 /** Add the GeoIP database's integer index (+1) of a valid two-character
4910 * country code to the routerset's <b>countries</b> bitarray. Return the
4911 * integer index if the country code is valid, -1 otherwise.*/
4912 static int
4913 routerset_add_country(const char *c)
4915 char country[3];
4916 country_t cc;
4918 /* XXXX: Country codes must be of the form \{[a-z\?]{2}\} but this accepts
4919 \{[.]{2}\}. Do we need to be strict? -RH */
4920 /* Nope; if the country code is bad, we'll get 0 when we look it up. */
4922 if (!geoip_is_loaded()) {
4923 log(LOG_WARN, LD_CONFIG, "GeoIP database not loaded: Cannot add country"
4924 "entry %s, ignoring.", c);
4925 return -1;
4928 memcpy(country, c+1, 2);
4929 country[2] = '\0';
4930 tor_strlower(country);
4932 if ((cc=geoip_get_country(country))==-1) {
4933 log(LOG_WARN, LD_CONFIG, "Country code '%s' is not valid, ignoring.",
4934 country);
4936 return cc;
4938 #endif
4940 /** Update the routerset's <b>countries</b> bitarray_t. Called whenever
4941 * the GeoIP database is reloaded.
4943 void
4944 routerset_refresh_countries(routerset_t *target)
4946 int cc;
4947 if (target->countries) {
4948 bitarray_free(target->countries);
4950 if (!geoip_is_loaded()) {
4951 target->countries = NULL;
4952 target->n_countries = 0;
4953 return;
4955 target->n_countries = geoip_get_n_countries();
4956 target->countries = bitarray_init_zero(target->n_countries);
4957 SMARTLIST_FOREACH_BEGIN(target->country_names, const char *, country) {
4958 cc = geoip_get_country(country);
4959 if (cc >= 0) {
4960 tor_assert(cc < target->n_countries);
4961 bitarray_set(target->countries, cc);
4962 } else {
4963 log(LOG_WARN, LD_CONFIG, "Country code '%s' is not recognized.",
4964 country);
4966 } SMARTLIST_FOREACH_END(country);
4969 /** Parse the string <b>s</b> to create a set of routerset entries, and add
4970 * them to <b>target</b>. In log messages, refer to the string as
4971 * <b>description</b>. Return 0 on success, -1 on failure.
4973 * Three kinds of elements are allowed in routersets: nicknames, IP address
4974 * patterns, and fingerprints. They may be surrounded by optional space, and
4975 * mst be separated by commas.
4978 routerset_parse(routerset_t *target, const char *s, const char *description)
4980 int r = 0;
4981 int added_countries = 0;
4982 char *countryname;
4983 smartlist_t *list = smartlist_create();
4984 smartlist_split_string(list, s, ",",
4985 SPLIT_SKIP_SPACE | SPLIT_IGNORE_BLANK, 0);
4986 SMARTLIST_FOREACH_BEGIN(list, char *, nick) {
4987 addr_policy_t *p;
4988 if (is_legal_hexdigest(nick)) {
4989 char d[DIGEST_LEN];
4990 if (*nick == '$')
4991 ++nick;
4992 log_debug(LD_CONFIG, "Adding identity %s to %s", nick, description);
4993 base16_decode(d, sizeof(d), nick, HEX_DIGEST_LEN);
4994 digestmap_set(target->digests, d, (void*)1);
4995 } else if (is_legal_nickname(nick)) {
4996 log_debug(LD_CONFIG, "Adding nickname %s to %s", nick, description);
4997 strmap_set_lc(target->names, nick, (void*)1);
4998 } else if ((countryname = routerset_get_countryname(nick)) != NULL) {
4999 log_debug(LD_CONFIG, "Adding country %s to %s", nick,
5000 description);
5001 smartlist_add(target->country_names, countryname);
5002 added_countries = 1;
5003 } else if ((strchr(nick,'.') || strchr(nick, '*')) &&
5004 (p = router_parse_addr_policy_item_from_string(
5005 nick, ADDR_POLICY_REJECT))) {
5006 log_debug(LD_CONFIG, "Adding address %s to %s", nick, description);
5007 smartlist_add(target->policies, p);
5008 } else {
5009 log_warn(LD_CONFIG, "Entry '%s' in %s is misformed.", nick,
5010 description);
5011 r = -1;
5012 tor_free(nick);
5013 SMARTLIST_DEL_CURRENT(list, nick);
5015 } SMARTLIST_FOREACH_END(nick);
5016 smartlist_add_all(target->list, list);
5017 smartlist_free(list);
5018 if (added_countries)
5019 routerset_refresh_countries(target);
5020 return r;
5023 /** DOCDOC */
5024 void
5025 refresh_all_country_info(void)
5027 or_options_t *options = get_options();
5029 if (options->EntryNodes)
5030 routerset_refresh_countries(options->EntryNodes);
5031 if (options->ExitNodes)
5032 routerset_refresh_countries(options->ExitNodes);
5033 if (options->ExcludeNodes)
5034 routerset_refresh_countries(options->ExcludeNodes);
5035 if (options->ExcludeExitNodes)
5036 routerset_refresh_countries(options->ExcludeExitNodes);
5037 if (options->_ExcludeExitNodesUnion)
5038 routerset_refresh_countries(options->_ExcludeExitNodesUnion);
5040 routerlist_refresh_countries();
5043 /** Add all members of the set <b>source</b> to <b>target</b>. */
5044 void
5045 routerset_union(routerset_t *target, const routerset_t *source)
5047 char *s;
5048 tor_assert(target);
5049 if (!source || !source->list)
5050 return;
5051 s = routerset_to_string(source);
5052 routerset_parse(target, s, "other routerset");
5053 tor_free(s);
5056 /** Return true iff <b>set</b> lists only nicknames and digests, and includes
5057 * no IP ranges or countries. */
5059 routerset_is_list(const routerset_t *set)
5061 return smartlist_len(set->country_names) == 0 &&
5062 smartlist_len(set->policies) == 0;
5065 /** Return true iff we need a GeoIP IP-to-country database to make sense of
5066 * <b>set</b>. */
5068 routerset_needs_geoip(const routerset_t *set)
5070 return set && smartlist_len(set->country_names);
5073 /** Return true iff there are no entries in <b>set</b>. */
5074 static int
5075 routerset_is_empty(const routerset_t *set)
5077 return !set || smartlist_len(set->list) == 0;
5080 /** Helper. Return true iff <b>set</b> contains a router based on the other
5081 * provided fields. Return higher values for more specific subentries: a
5082 * single router is more specific than an address range of routers, which is
5083 * more specific in turn than a country code.
5085 * (If country is -1, then we take the country
5086 * from addr.) */
5087 static int
5088 routerset_contains(const routerset_t *set, const tor_addr_t *addr,
5089 uint16_t orport,
5090 const char *nickname, const char *id_digest, int is_named,
5091 country_t country)
5093 if (!set || !set->list) return 0;
5094 (void) is_named; /* not supported */
5095 if (nickname && strmap_get_lc(set->names, nickname))
5096 return 4;
5097 if (id_digest && digestmap_get(set->digests, id_digest))
5098 return 4;
5099 if (addr && compare_tor_addr_to_addr_policy(addr, orport, set->policies)
5100 == ADDR_POLICY_REJECTED)
5101 return 3;
5102 if (set->countries) {
5103 if (country < 0 && addr)
5104 country = geoip_get_country_by_ip(tor_addr_to_ipv4h(addr));
5106 if (country >= 0 && country < set->n_countries &&
5107 bitarray_is_set(set->countries, country))
5108 return 2;
5110 return 0;
5113 /** Return true iff we can tell that <b>ei</b> is a member of <b>set</b>. */
5115 routerset_contains_extendinfo(const routerset_t *set, extend_info_t *ei)
5117 return routerset_contains(set,
5118 &ei->addr,
5119 ei->port,
5120 ei->nickname,
5121 ei->identity_digest,
5122 -1, /*is_named*/
5123 -1 /*country*/);
5126 /** Return true iff <b>ri</b> is in <b>set</b>. */
5128 routerset_contains_router(const routerset_t *set, routerinfo_t *ri)
5130 tor_addr_t addr;
5131 tor_addr_from_ipv4h(&addr, ri->addr);
5132 return routerset_contains(set,
5133 &addr,
5134 ri->or_port,
5135 ri->nickname,
5136 ri->cache_info.identity_digest,
5137 ri->is_named,
5138 ri->country);
5141 /** Return true iff <b>rs</b> is in <b>set</b>. */
5143 routerset_contains_routerstatus(const routerset_t *set, routerstatus_t *rs)
5145 tor_addr_t addr;
5146 tor_addr_from_ipv4h(&addr, rs->addr);
5147 return routerset_contains(set,
5148 &addr,
5149 rs->or_port,
5150 rs->nickname,
5151 rs->identity_digest,
5152 rs->is_named,
5153 -1);
5156 /** Add every known routerinfo_t that is a member of <b>routerset</b> to
5157 * <b>out</b>. If <b>running_only</b>, only add the running ones. */
5158 void
5159 routerset_get_all_routers(smartlist_t *out, const routerset_t *routerset,
5160 int running_only)
5162 tor_assert(out);
5163 if (!routerset || !routerset->list)
5164 return;
5165 if (!warned_nicknames)
5166 warned_nicknames = smartlist_create();
5167 if (routerset_is_list(routerset)) {
5169 /* No routers are specified by type; all are given by name or digest.
5170 * we can do a lookup in O(len(list)). */
5171 SMARTLIST_FOREACH(routerset->list, const char *, name, {
5172 routerinfo_t *router = router_get_by_nickname(name, 1);
5173 if (router) {
5174 if (!running_only || router->is_running)
5175 smartlist_add(out, router);
5178 } else {
5179 /* We need to iterate over the routerlist to get all the ones of the
5180 * right kind. */
5181 routerlist_t *rl = router_get_routerlist();
5182 SMARTLIST_FOREACH(rl->routers, routerinfo_t *, router, {
5183 if (running_only && !router->is_running)
5184 continue;
5185 if (routerset_contains_router(routerset, router))
5186 smartlist_add(out, router);
5191 /** Add to <b>target</b> every routerinfo_t from <b>source</b> that is in
5192 * <b>include</b>, but not excluded in a more specific fashion by
5193 * <b>exclude</b>. If <b>running_only</b>, only include running routers.
5195 void
5196 routersets_get_disjunction(smartlist_t *target,
5197 const smartlist_t *source,
5198 const routerset_t *include,
5199 const routerset_t *exclude, int running_only)
5201 SMARTLIST_FOREACH(source, routerinfo_t *, router, {
5202 int include_result;
5203 if (running_only && !router->is_running)
5204 continue;
5205 if (!routerset_is_empty(include))
5206 include_result = routerset_contains_router(include, router);
5207 else
5208 include_result = 1;
5210 if (include_result) {
5211 int exclude_result = routerset_contains_router(exclude, router);
5212 if (include_result >= exclude_result)
5213 smartlist_add(target, router);
5218 /** Remove every routerinfo_t from <b>lst</b> that is in <b>routerset</b>. */
5219 void
5220 routerset_subtract_routers(smartlist_t *lst, const routerset_t *routerset)
5222 tor_assert(lst);
5223 if (!routerset)
5224 return;
5225 SMARTLIST_FOREACH(lst, routerinfo_t *, r, {
5226 if (routerset_contains_router(routerset, r)) {
5227 //log_debug(LD_DIR, "Subtracting %s",r->nickname);
5228 SMARTLIST_DEL_CURRENT(lst, r);
5233 /** Return a new string that when parsed by routerset_parse_string() will
5234 * yield <b>set</b>. */
5235 char *
5236 routerset_to_string(const routerset_t *set)
5238 if (!set || !set->list)
5239 return tor_strdup("");
5240 return smartlist_join_strings(set->list, ",", 0, NULL);
5243 /** Helper: return true iff old and new are both NULL, or both non-NULL
5244 * equal routersets. */
5246 routerset_equal(const routerset_t *old, const routerset_t *new)
5248 if (old == NULL && new == NULL)
5249 return 1;
5250 else if (old == NULL || new == NULL)
5251 return 0;
5253 if (smartlist_len(old->list) != smartlist_len(new->list))
5254 return 0;
5256 SMARTLIST_FOREACH(old->list, const char *, cp1, {
5257 const char *cp2 = smartlist_get(new->list, cp1_sl_idx);
5258 if (strcmp(cp1, cp2))
5259 return 0;
5262 return 1;
5264 #if 0
5265 /* XXXX: This won't work if the names/digests are identical but in a
5266 different order. Checking for exact equality would be heavy going,
5267 is it worth it? -RH*/
5268 /* This code is totally bogus; sizeof doesn't work even remotely like this
5269 * code seems to think. Let's revert to a string-based comparison for
5270 * now. -NM*/
5271 if (sizeof(old->names) != sizeof(new->names))
5272 return 0;
5274 if (memcmp(old->names,new->names,sizeof(new->names)))
5275 return 0;
5276 if (sizeof(old->digests) != sizeof(new->digests))
5277 return 0;
5278 if (memcmp(old->digests,new->digests,sizeof(new->digests)))
5279 return 0;
5280 if (sizeof(old->countries) != sizeof(new->countries))
5281 return 0;
5282 if (memcmp(old->countries,new->countries,sizeof(new->countries)))
5283 return 0;
5284 return 1;
5285 #endif
5288 /** Free all storage held in <b>routerset</b>. */
5289 void
5290 routerset_free(routerset_t *routerset)
5292 SMARTLIST_FOREACH(routerset->list, char *, cp, tor_free(cp));
5293 smartlist_free(routerset->list);
5294 SMARTLIST_FOREACH(routerset->policies, addr_policy_t *, p,
5295 addr_policy_free(p));
5296 smartlist_free(routerset->policies);
5297 SMARTLIST_FOREACH(routerset->country_names, char *, cp, tor_free(cp));
5298 smartlist_free(routerset->country_names);
5300 strmap_free(routerset->names, NULL);
5301 digestmap_free(routerset->digests, NULL);
5302 if (routerset->countries)
5303 bitarray_free(routerset->countries);
5304 tor_free(routerset);
5307 /** Refresh the country code of <b>ri</b>. This function MUST be called on
5308 * each router when the GeoIP database is reloaded, and on all new routers. */
5309 void
5310 routerinfo_set_country(routerinfo_t *ri)
5312 ri->country = geoip_get_country_by_ip(ri->addr);
5315 /** Set the country code of all routers in the routerlist. */
5316 void
5317 routerlist_refresh_countries(void)
5319 routerlist_t *rl = router_get_routerlist();
5320 SMARTLIST_FOREACH(rl->routers, routerinfo_t *, ri,
5321 routerinfo_set_country(ri));
5324 /** Determine the routers that are responsible for <b>id</b> (binary) and
5325 * add pointers to those routers' routerstatus_t to <b>responsible_dirs</b>.
5326 * Return -1 if we're returning an empty smartlist, else return 0.
5329 hid_serv_get_responsible_directories(smartlist_t *responsible_dirs,
5330 const char *id)
5332 int start, found, n_added = 0, i;
5333 networkstatus_t *c = networkstatus_get_latest_consensus();
5334 int use_begindir = get_options()->TunnelDirConns;
5335 if (!c || !smartlist_len(c->routerstatus_list)) {
5336 log_warn(LD_REND, "We don't have a consensus, so we can't perform v2 "
5337 "rendezvous operations.");
5338 return -1;
5340 tor_assert(id);
5341 start = networkstatus_vote_find_entry_idx(c, id, &found);
5342 if (start == smartlist_len(c->routerstatus_list)) start = 0;
5343 i = start;
5344 do {
5345 routerstatus_t *r = smartlist_get(c->routerstatus_list, i);
5346 if (r->is_hs_dir) {
5347 if (r->dir_port || use_begindir)
5348 smartlist_add(responsible_dirs, r);
5349 else
5350 log_info(LD_REND, "Not adding router '%s' to list of responsible "
5351 "hidden service directories, because we have no way of "
5352 "reaching it.", r->nickname);
5353 if (++n_added == REND_NUMBER_OF_CONSECUTIVE_REPLICAS)
5354 break;
5356 if (++i == smartlist_len(c->routerstatus_list))
5357 i = 0;
5358 } while (i != start);
5360 /* Even though we don't have the desired number of hidden service
5361 * directories, be happy if we got any. */
5362 return smartlist_len(responsible_dirs) ? 0 : -1;
5365 /** Return true if this node is currently acting as hidden service
5366 * directory, false otherwise. */
5368 hid_serv_acting_as_directory(void)
5370 routerinfo_t *me = router_get_my_routerinfo();
5371 networkstatus_t *c;
5372 routerstatus_t *rs;
5373 if (!me)
5374 return 0;
5375 if (!get_options()->HidServDirectoryV2) {
5376 log_info(LD_REND, "We are not acting as hidden service directory, "
5377 "because we have not been configured as such.");
5378 return 0;
5380 if (!(c = networkstatus_get_latest_consensus())) {
5381 log_info(LD_REND, "There's no consensus, so I can't tell if I'm a hidden "
5382 "service directory");
5383 return 0;
5385 rs = networkstatus_vote_find_entry(c, me->cache_info.identity_digest);
5386 if (!rs) {
5387 log_info(LD_REND, "We're not listed in the consensus, so we're not "
5388 "being a hidden service directory.");
5389 return 0;
5391 if (!rs->is_hs_dir) {
5392 log_info(LD_REND, "We're not listed as a hidden service directory in "
5393 "the consensus, so we won't be one.");
5394 return 0;
5396 return 1;
5399 /** Return true if this node is responsible for storing the descriptor ID
5400 * in <b>query</b> and false otherwise. */
5402 hid_serv_responsible_for_desc_id(const char *query)
5404 routerinfo_t *me;
5405 routerstatus_t *last_rs;
5406 const char *my_id, *last_id;
5407 int result;
5408 smartlist_t *responsible;
5409 if (!hid_serv_acting_as_directory())
5410 return 0;
5411 if (!(me = router_get_my_routerinfo()))
5412 return 0; /* This is redundant, but let's be paranoid. */
5413 my_id = me->cache_info.identity_digest;
5414 responsible = smartlist_create();
5415 if (hid_serv_get_responsible_directories(responsible, query) < 0) {
5416 smartlist_free(responsible);
5417 return 0;
5419 last_rs = smartlist_get(responsible, smartlist_len(responsible)-1);
5420 last_id = last_rs->identity_digest;
5421 result = rend_id_is_in_interval(my_id, query, last_id);
5422 smartlist_free(responsible);
5423 return result;