Allow test program to compare input to a reference file.

[utcp] / test.c

diff --git a/test.c b/test.c
index d138fd60bc168adf713ed4c068b18d0f92abf29a..4a1bf16e3bff39ca60ac67472b3c4e0c8f9f0cd5 100644 (file)
--- a/test.c
+++ b/test.c
@@ -30,11 +30,13 @@ double dropin;
 double dropout;
 long total_out;
 long total_in;
+FILE *reference;
 
 char *reorder_data;
 size_t reorder_len;
 int reorder_countdown;
 
+#if UTCP_DEBUG
 void debug(const char *format, ...) {
        struct timeval now;
        gettimeofday(&now, NULL);
@@ -44,6 +46,9 @@ void debug(const char *format, ...) {
        vfprintf(stderr, format, ap);
        va_end(ap);
 }
+#else
+#define debug(...)
+#endif
 
 ssize_t do_recv(struct utcp_connection *c, const void *data, size_t len) {
        if(!data || !len) {
@@ -56,6 +61,17 @@ ssize_t do_recv(struct utcp_connection *c, const void *data, size_t len) {
                }
                return -1;
        }
+       if(reference) {
+               char buf[len];
+               if(fread(buf, len, 1, reference) != 1) {
+                       debug("Error reading reference\n");
+                       abort();
+               }
+               if(memcmp(buf, data, len)) {
+                       debug("Received data differs from reference\n");
+                       abort();
+               }
+       }
        return write(1, data, len);
 }
 
@@ -69,8 +85,10 @@ ssize_t do_send(struct utcp *utcp, const void *data, size_t len) {
        int s = *(int *)utcp->priv;
        outpktno++;
        if(outpktno >= dropfrom && outpktno < dropto) {
-               if(drand48() < dropout)
+               if(drand48() < dropout) {
+                       debug("Dropped outgoing packet\n");
                        return len;
+               }
                if(!reorder_data && drand48() < reorder) {
                        reorder_data = malloc(len);
                        if(!reorder_data) {
@@ -117,6 +135,10 @@ int main(int argc, char *argv[]) {
        if(getenv("REORDER")) reorder = atof(getenv("REORDER"));
        if(getenv("REORDER_DIST")) reorder_dist = atoi(getenv("REORDER_DIST"));
 
+       char *reference_filename = getenv("REFERENCE");
+       if(reference_filename)
+               reference = fopen(reference_filename, "r");
+
        if(dropto < dropfrom)
                dropto = 1 << 30;
 
@@ -213,7 +235,10 @@ int main(int argc, char *argv[]) {
                        inpktno++;
                        if(inpktno >= dropto || inpktno < dropfrom || drand48() >= dropin) {
                                total_in += len;
-                               utcp_recv(u, buf, len);
+                               if(utcp_recv(u, buf, len) == -1)
+                                       debug("Error receiving UTCP packet: %s\n", strerror(errno));
+                       } else {
+                               debug("Dropped incoming packet\n");
                        }
                }