From c5e5b8415e9c91d132678dcfccde8df848ee70c8 Mon Sep 17 00:00:00 2001
From: Phil Sutter <psutter@redhat.com>
Date: Fri, 25 Nov 2016 11:38:02 +0100
Subject: [PATCH] test: Implement self-test functionality

This allows to run 'test' unattended by setting environment variable
SELFTEST=on. Instead of printing the settings libkeepalive should have
changed, it uses the input values still present in environment to assert
them being set correctly.

Signed-off-by: Phil Sutter <psutter@redhat.com>
---
 test/test.c | 42 ++++++++++++++++++++++++++++++++++++++----
 1 file changed, 38 insertions(+), 4 deletions(-)

diff --git a/test/test.c b/test/test.c
index 224c1b4944fc5..7eaaaed2e9840 100644
--- a/test/test.c
+++ b/test/test.c
@@ -32,14 +32,23 @@
 */
 
 #define _GNU_SOURCE
+#include <stdbool.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <strings.h>
 #include <unistd.h>
 #include <sys/types.h>
 #include <sys/socket.h>
 #include <netinet/in.h>
 #include <netinet/tcp.h>
 
+#define assert(x) do { \
+  if (!(x)) { \
+    printf("%s:%d: assertion '" #x "' failed!\n", __FILE__, __LINE__); \
+    exit(EXIT_FAILURE); \
+  } \
+} while (0)
+
 int main(void);
 
 int main()
@@ -47,6 +56,11 @@ int main()
   int s;
   int optval;
   socklen_t optlen = sizeof(optval);
+  const char *env;
+  bool selftest = false;
+
+  env = getenv("SELFTEST");
+	selftest = env && !strcasecmp(env, "on");
 
   if((s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0) {
     perror("socket()");
@@ -58,7 +72,12 @@ int main()
     close(s);
     exit(EXIT_FAILURE);
   }
-  printf("SO_KEEPALIVE is %s\n", (optval ? "ON" : "OFF"));
+  if (selftest) {
+    env = getenv("KEEPALIVE");
+    assert((env && !strcasecmp(env, "off")) ^ optval);
+  } else {
+    printf("SO_KEEPALIVE is %s\n", (optval ? "ON" : "OFF"));
+  }
 
   if(optval) {
     #ifdef TCP_KEEPCNT
@@ -67,7 +86,12 @@ int main()
       close(s);
       exit(EXIT_FAILURE);
     }
-    printf("TCP_KEEPCNT   = %d\n", optval);
+    if (selftest) {
+      env = getenv("KEEPCNT");
+      assert(!env || atoi(env) == optval);
+    } else {
+      printf("TCP_KEEPCNT   = %d\n", optval);
+    }
     #endif
 
     #ifdef TCP_KEEPIDLE
@@ -76,7 +100,12 @@ int main()
       close(s);
       exit(EXIT_FAILURE);
     }
-    printf("TCP_KEEPIDLE  = %d\n", optval);
+    if (selftest) {
+      env = getenv("KEEPIDLE");
+      assert(!env || atoi(env) == optval);
+    } else {
+      printf("TCP_KEEPIDLE  = %d\n", optval);
+    }
     #endif
 
     #ifdef TCP_KEEPINTVL
@@ -85,7 +114,12 @@ int main()
       close(s);
       exit(EXIT_FAILURE);
     }
-    printf("TCP_KEEPINTVL = %d\n", optval);
+    if (selftest) {
+      env = getenv("KEEPINTVL");
+      assert(!env || atoi(env) == optval);
+    } else {
+      printf("TCP_KEEPINTVL = %d\n", optval);
+    }
     #endif
   }
 
-- 
2.10.0