[or-cvs] Add getinfo logic for accounting; add options for helper no...

[nickm@xxxxxxxx (Nick Mathewson)]

Update of /home/or/cvsroot/tor/src/or
In directory moria:/tmp/cvs-serv14115/src/or

Modified Files:
	config.c control.c hibernate.c or.h 
Log Message:
Add getinfo logic for accounting; add options for helper nodes

Index: config.c
===================================================================
RCS file: /home/or/cvsroot/tor/src/or/config.c,v
retrieving revision 1.369
retrieving revision 1.370
diff -u -d -r1.369 -r1.370
--- config.c	14 Jul 2005 22:37:52 -0000	1.369
+++ config.c	22 Jul 2005 14:55:09 -0000	1.370
@@ -179,6 +179,8 @@
   VAR("SysLog",              LINELIST_S, OldLogOptions,      NULL),
   OBSOLETE("TrafficShaping"),
   VAR("User",                STRING,   User,                 NULL),
+  VAR("UseHelperNodes",      BOOL,     UseHelperNodes,       "0"),
+  VAR("NumHelperNodes",      UINT,     NumHelperNodes,       "3"),
   VAR("__LeaveStreamsUnattached", BOOL,LeaveStreamsUnattached, "0"),
   { NULL, CONFIG_TYPE_OBSOLETE, 0, NULL }
 };
@@ -1564,6 +1566,11 @@
     result = -1;
   }
 
+  if (options->UseHelperNodes && ! options->NumHelperNodes) {
+    log_fn(LOG_WARN, "Cannot enable UseHelperNodes with NumHelperNodes set to 0");
+    result = -1;
+  }
+
   if (check_nickname_list(options->ExitNodes, "ExitNodes"))
     result = -1;
   if (check_nickname_list(options->EntryNodes, "EntryNodes"))

Index: control.c
===================================================================
RCS file: /home/or/cvsroot/tor/src/or/control.c,v
retrieving revision 1.108
retrieving revision 1.109
diff -u -d -r1.108 -r1.109
--- control.c	17 Jul 2005 22:28:01 -0000	1.108
+++ control.c	22 Jul 2005 14:55:09 -0000	1.109
@@ -1121,6 +1121,8 @@
   *answer = NULL; /* unrecognized key by default */
   if (!strcmp(question, "version")) {
     *answer = tor_strdup(VERSION);
+  } else if (!strcmpstart(question, "accounting/")) {
+    return accounting_getinfo_helper(question, answer);
   } else if (!strcmpstart(question, "desc/id/")) {
     routerinfo_t *ri = router_get_by_hexdigest(question+strlen("desc/id/"));
     if (ri && ri->signed_descriptor)

Index: hibernate.c
===================================================================
RCS file: /home/or/cvsroot/tor/src/or/hibernate.c,v
retrieving revision 1.54
retrieving revision 1.55
diff -u -d -r1.54 -r1.55
--- hibernate.c	11 Jun 2005 18:52:11 -0000	1.54
+++ hibernate.c	22 Jul 2005 14:55:09 -0000	1.55
@@ -100,6 +100,8 @@
  * Functions for bandwidth accounting.
  * ************/
 
+
+
 /** Configure accounting start/end time settings based on
  * options->AccountingStart.  Return 0 on success, -1 on failure. If
  * <b>validate_only</b> is true, do not change the current settings. */
@@ -846,3 +848,41 @@
   }
 }
 
+/** DOCDOC */
+int
+accounting_getinfo_helper(const char *question, char **answer)
+{
+  if (!strcmp(question, "accounting/enabled")) {
+    *answer = tor_strdup(get_options()->AccountingMax ? "1" : "0");
+  } else if (!strcmp(question, "accounting/hibernating")) {
+    if (hibernate_state == HIBERNATE_STATE_DORMANT)
+      *answer = tor_strdup("hard");
+    else if (hibernate_state == HIBERNATE_STATE_LOWBANDWIDTH)
+      *answer = tor_strdup("soft");
+    else
+      *answer = tor_strdup("awake");
+  } else if (!strcmp(question, "accounting/bytes")) {
+    *answer = tor_malloc(32);
+    tor_snprintf(*answer, 32, U64_FORMAT" "U64_FORMAT,
+                 U64_PRINTF_ARG(n_bytes_read_in_interval),
+                 U64_PRINTF_ARG(n_bytes_written_in_interval));
+  } else if (!strcmp(question, "accounting/bytes-left")) {
+    *answer = tor_malloc(32);
+    uint64_t limit = get_options()->AccountingMax;
+    tor_snprintf(*answer, 32, U64_FORMAT" "U64_FORMAT,
+                 U64_PRINTF_ARG(limit - n_bytes_read_in_interval),
+                 U64_PRINTF_ARG(limit - n_bytes_written_in_interval));
+  } else if (!strcmp(question, "accounting/interval-start")) {
+    *answer = tor_malloc(ISO_TIME_LEN+1);
+    format_iso_time(*answer, interval_start_time);
+  } else if (!strcmp(question, "accounting/interval-wake")) {
+    *answer = tor_malloc(ISO_TIME_LEN+1);
+    format_iso_time(*answer, interval_wakeup_time);
+  } else if (!strcmp(question, "accounting/interval-end")) {
+    *answer = tor_malloc(ISO_TIME_LEN+1);
+    format_iso_time(*answer, interval_end_time);
+  } else {
+    *answer = NULL;
+  }
+  return 0;
+}

Index: or.h
===================================================================
RCS file: /home/or/cvsroot/tor/src/or/or.h,v
retrieving revision 1.626
retrieving revision 1.627
diff -u -d -r1.626 -r1.627
--- or.h	14 Jul 2005 08:43:19 -0000	1.626
+++ or.h	22 Jul 2005 14:55:09 -0000	1.627
@@ -1164,6 +1164,9 @@
                     * such as addresses (0), or do we scrub them first (1)? */
   int HardwareAccel; /**< Boolean: Should we enable OpenSSL hardware
                       * acceleration where available? */
+  int UseHelperNodes; /**< Boolean: Do we try to enter from a smallish number
+                       * of fixed nodes? */
+  int NumHelperNodes; /**< How many helper nodes do we try to establish? */
 } or_options_t;
 
 #define MAX_SOCKS_REPLY_LEN 1024
@@ -1619,6 +1622,8 @@
 void hibernate_begin_shutdown(void);
 int we_are_hibernating(void);
 void consider_hibernation(time_t now);
+int accounting_getinfo_helper(const char *question, char **answer);
+
 
 /********************************* main.c ***************************/