[or-cvs] Remember list of digests for trusted dirservers; use this l...

[nickm@seul.org (Nick Mathewson)]

Update of /home/or/cvsroot/src/or
In directory moria.mit.edu:/tmp/cvs-serv4330/src/or

Modified Files:
	or.h router.c routerlist.c 
Log Message:
Remember list of digests for trusted dirservers; use this list to tell whether you are a trusted dirserver

Index: or.h
===================================================================
RCS file: /home/or/cvsroot/src/or/or.h,v
retrieving revision 1.385
retrieving revision 1.386
diff -u -d -r1.385 -r1.386
--- or.h	20 Jul 2004 19:45:29 -0000	1.385
+++ or.h	20 Jul 2004 20:09:59 -0000	1.386
@@ -1369,6 +1369,7 @@
 routerinfo_t *router_get_by_nickname(const char *nickname);
 routerinfo_t *router_get_by_hexdigest(const char *hexdigest);
 routerinfo_t *router_get_by_digest(const char *digest);
+int router_digest_is_trusted_dir(const char *digest);
 void router_get_routerlist(routerlist_t **prouterlist);
 void routerlist_free(routerlist_t *routerlist);
 void routerlist_clear_trusted_directories(void);

Index: router.c
===================================================================
RCS file: /home/or/cvsroot/src/or/router.c,v
retrieving revision 1.66
retrieving revision 1.67
diff -u -d -r1.66 -r1.67
--- router.c	20 Jul 2004 19:45:29 -0000	1.66
+++ router.c	20 Jul 2004 20:09:59 -0000	1.67
@@ -516,12 +516,8 @@
     log_fn(LOG_WARN, "Couldn't dump router to string.");
     return -1;
   }
-  /* XXX008 NM: no, we shouldn't just blindly assume we're an
-   * authdirserver just because our dir_port is set. We should
-   * take these next two lines out, and then set our is_trusted_dir
-   * variable if we find ourselves in the dirservers file. Yes/no? */
-  if (ri->dir_port)
-    ri->is_trusted_dir = 1;
+  ri->is_trusted_dir = ri->dir_port &&
+    router_digest_is_trusted_dir(ri->identity_digest);
   return 0;
 }
 

Index: routerlist.c
===================================================================
RCS file: /home/or/cvsroot/src/or/routerlist.c,v
retrieving revision 1.100
retrieving revision 1.101
diff -u -d -r1.100 -r1.101
--- routerlist.c	20 Jul 2004 06:44:16 -0000	1.100
+++ routerlist.c	20 Jul 2004 20:09:59 -0000	1.101
@@ -26,6 +26,9 @@
 
 /****************************************************************************/
 
+/** List of digests of keys for servers that are trusted directories. */
+static smartlist_t *trusted_dir_digests = NULL;
+
 /****
  * Functions to manage and access our list of known routers. (Note:
  * dirservers maintain a separate, independent list of known router
@@ -306,6 +309,16 @@
   return NULL;
 }
 
+/** Return true iff <b>digest</b> is the digest of the identity key of
+ * a trusted directory. */
+int router_digest_is_trusted_dir(const char *digest) {
+  if (!trusted_dir_digests)
+    return 0;
+  SMARTLIST_FOREACH(trusted_dir_digests, char *, cp,
+                    if (!memcmp(digest, cp, DIGEST_LEN)) return 1);
+  return 0;
+}
+
 /** Return the router in our routerlist whose hexadecimal key digest
  * is <b>hexdigest</b>.  Return NULL if no such router is known. */
 routerinfo_t *router_get_by_hexdigest(const char *hexdigest) {
@@ -527,9 +540,14 @@
 /** Mark all directories in the routerlist as nontrusted. */
 void routerlist_clear_trusted_directories(void)
 {
-  if (!routerlist) return;
-  SMARTLIST_FOREACH(routerlist->routers, routerinfo_t *, r,
-                    r->is_trusted_dir = 0);
+  if (routerlist) {
+    SMARTLIST_FOREACH(routerlist->routers, routerinfo_t *, r,
+                      r->is_trusted_dir = 0);
+  }
+  if (trusted_dir_digests) {
+    SMARTLIST_FOREACH(trusted_dir_digests, char *, cp, tor_free(cp));
+    smartlist_clear(trusted_dir_digests);
+  }
 }
 
 /** Helper function: read routerinfo elements from s, and throw out the
@@ -546,8 +564,19 @@
     return -1;
   }
   if (trusted) {
-    SMARTLIST_FOREACH(new_list->routers, routerinfo_t *, r,
-                      if (r->dir_port) r->is_trusted_dir = 1);
+    int i;
+    if (!trusted_dir_digests)
+      trusted_dir_digests = smartlist_create();
+    for (i=0;i<smartlist_len(new_list->routers);++i) {
+      routerinfo_t *r = smartlist_get(new_list->routers, i);
+      if (r->dir_port) {
+        char *b;
+        r->is_trusted_dir = 1;
+        b = tor_malloc(DIGEST_LEN);
+        memcpy(b, r->identity_digest, DIGEST_LEN);
+        smartlist_add(trusted_dir_digests, b);
+      }
+    }
   }
   if (routerlist) {
     SMARTLIST_FOREACH(new_list->routers, routerinfo_t *, r,