Added option to run without the assumption of PlanetLab Vservers. We still assume...

[distributedratelimiting.git] / drl / ulogd_DRL.c

diff --git a/drl/ulogd_DRL.c b/drl/ulogd_DRL.c
index 70f096b..cb32f3b 100644 (file)
--- a/drl/ulogd_DRL.c
+++ b/drl/ulogd_DRL.c
@@ -124,8 +124,16 @@
  * Add the config options for DRL. 
  */
 
-static config_entry_t bind_addr = {
+static config_entry_t leaves = {
     .next = NULL,
+    .key = "leaves",
+    .type = CONFIG_TYPE_STRING,
+    .options = CONFIG_OPT_NONE,
+    .u = { .string = "PLANETLAB" },
+};
+
+static config_entry_t bind_addr = {
+    .next = &leaves,
     .key = "bind_addr",
     .type = CONFIG_TYPE_STRING,
     .options = CONFIG_OPT_NONE,
@@ -231,7 +239,7 @@ static config_entry_t policy = {
     .key = "policy",
     .type = CONFIG_TYPE_STRING,
     .options = CONFIG_OPT_MANDATORY,
-    .u = { .string = "GRD" },
+    .u = { .string = "FPS" },
 };
 
 /** The estimate interval, in milliseconds. */
@@ -240,7 +248,7 @@ static config_entry_t estintms = {
     .key = "estintms",
     .type = CONFIG_TYPE_INT,
     .options = CONFIG_OPT_MANDATORY,
-    .u = { .value = 100 },
+    .u = { .value = 500 },
 };
 
 #define config_entries (&estintms)
@@ -725,15 +733,29 @@ static int validate_htb_exists(int node, int parent) {
     FILE *pipe = popen("/sbin/tc class show dev eth0", "r");
     char line[200];
 
-    while (fgets(line, 200, pipe) != NULL) {
-        int n, p;
-        char ignore[200];
+    if (parent != 0) {
+        while (fgets(line, 200, pipe) != NULL) {
+            int n, p;
+            char ignore[200];
 
-        sscanf(line, "class htb 1:%x parent 1:%x prio %s", &n, &p, ignore);
-        if (n == node && p == parent) {
-            pclose(pipe);
-            return 0;
+            sscanf(line, "class htb 1:%x parent 1:%x prio %s", &n, &p, ignore);
+            if (n == node && p == parent) {
+                pclose(pipe);
+                return 0;
+            }
+        }
+    } else {
+        while (fgets(line, 200, pipe) != NULL) {
+            int n, p;
+            char ignore[200];
+
+            sscanf(line, "class htb 1:%x root prio %d %s", &n, &p, ignore);
+            if (n == node && strstr(line, "root") != NULL) {
+                pclose(pipe);
+                return 0;
+            }
         }
+
     }
 
     pclose(pipe);
@@ -1594,9 +1616,18 @@ static int init_drl(void) {
         return false;
     }
 
-    if (get_eligible_leaves(&limiter.stable_instance)) {
-        printlog(LOG_CRITICAL, "Failed to read eligigle leaves.\n");
-        return false;
+    /* If no leaves are specified, assume we're on planetlab and read them out
+     * of /proc/virtual.  Otherwise, read the specified line. */
+    if (!strncmp(leaves.u.string, "PLANETLAB", 9)) {
+        if (get_eligible_leaves(&limiter.stable_instance)) {
+            printlog(LOG_CRITICAL, "Failed to read eligigle leaves.\n");
+            return false;
+        }
+    } else {
+        if (parse_leaves(&limiter.stable_instance, leaves.u.string)) {
+            printlog(LOG_CRITICAL, "Failed to parse leaf string.\n");
+            return false;
+        }
     }
 
     if (parse_drl_config(drl_configfile.u.string, &configs)) {
@@ -1683,10 +1714,18 @@ static void reconfig() {
         return;
     }
 
-    if (get_eligible_leaves(&limiter.new_instance)) {
-        free_failed_config(configs, &limiter.new_instance);
-        printlog(LOG_CRITICAL, "Failed to read leaves during reconfig().\n");
-        return;
+    if (!strncmp(leaves.u.string, "PLANETLAB", 9)) {
+        if (get_eligible_leaves(&limiter.new_instance)) {
+            free_failed_config(configs, &limiter.new_instance);
+            printlog(LOG_CRITICAL, "Failed to read eligigle leaves.\n");
+            return false;
+        }
+    } else {
+        if (parse_leaves(&limiter.new_instance, leaves.u.string)) {
+            free_failed_config(configs, &limiter.new_instance);
+            printlog(LOG_CRITICAL, "Failed to parse leaf string.\n");
+            return false;
+        }
     }
 
     if (parse_drl_config(drl_configfile.u.string, &configs)) {