chiark / gitweb /
proxy: Add a `drop' filter to randomly discard packets.
authorMark Wooding <mdw@distorted.org.uk>
Wed, 1 Feb 2012 21:29:56 +0000 (21:29 +0000)
committerMark Wooding <mdw@distorted.org.uk>
Wed, 21 Mar 2012 16:11:49 +0000 (16:11 +0000)
I'm surprised there wasn't one of these already.

proxy/tripe-mitm.8.in
proxy/tripe-mitm.c

index 38062a0..f9cea37 100644 (file)
@@ -201,6 +201,11 @@ the queue) then it has a 1 in
 .I p-replay
 probability (default 1 in 20) of being left in the queue.
 .TP
+.BI drop\fR[\fP: p-drop\fR]
+Randomly drop messages.  Each message has a 1 in
+.I p-drop
+probability (default 1 in 5) of being discarded.
+.TP
 .BI corrupt\fR[\fP: p-corrupt\fR]
 Randomly corrupt messages.  Each message has a 1 in
 .I p-corrupt
index 4b0e447..eb1c7fd 100644 (file)
@@ -257,7 +257,7 @@ static void addcorrupt(filter *f, unsigned ac, char **av)
 {
   corrupt *c;
   if (ac > 1)
-    die(1, "syntax: filt:corrupt[:PCORRUPT]");
+    die(1, "syntax: filt:corrupt[:P-CORRUPT]");
   c = CREATE(corrupt);
   if (ac > 0)
     c->p_corrupt = atoi(av[0]);
@@ -267,6 +267,36 @@ static void addcorrupt(filter *f, unsigned ac, char **av)
   f->func = docorrupt;
 }
 
+/*----- Drop filter -------------------------------------------------------*/
+
+typedef struct drop {
+  unsigned p_drop;
+} drop;
+
+static void dodrop(filter *f, const octet *buf, size_t sz)
+{
+  drop *d = f->state;
+
+  if (!RND(d->p_drop))
+    puts("drop packet");
+  else
+    PASS(f->next, buf, sz);
+}
+
+static void adddrop(filter *f, unsigned ac, char **av)
+{
+  drop *d;
+  if (ac > 1)
+    die(1, "syntax: filt:drop[:P-DROP]");
+  d = CREATE(drop);
+  if (ac > 0)
+    d->p_drop = atoi(av[0]);
+  else
+    d->p_drop = 5;
+  f->state = d;
+  f->func = dodrop;
+}
+
 /*----- Delay filter ------------------------------------------------------*/
 
 typedef struct delaynode {
@@ -379,7 +409,7 @@ static void adddelay(filter *f, unsigned ac, char **av)
   unsigned i;
 
   if (ac < 1 || ac > 3)
-    die(1, "syntax: filt:delay:QLEN[:MILLIS:PREPLAY]");
+    die(1, "syntax: filt:delay:QLEN[:MILLIS:P-REPLAY]");
   d = CREATE(delay);
   d->max = atoi(av[0]);
   if (ac > 1)
@@ -426,6 +456,7 @@ const struct filtab {
   { "send",    addsend },
   { "fork",    addfork },
   { "delay",   adddelay },
+  { "drop",    adddrop },
   { "corrupt", addcorrupt },
   { 0,         0 }
 };
@@ -633,6 +664,7 @@ Filters:\n\
   send\n\
   fork:TAG\n\
   delay:QLEN[:MILLIS:P-REPLAY]\n\
+  drop[:P-DROP]\n\
   corrupt[:P-CORRUPT]\n",
        fp);
 }