From 0d47952611198ef6b1163f366dc03922d20b1475 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Wed, 17 Apr 2024 09:42:04 +0200
Subject: Adding upstream version 7.94+git20230807.3be01efb1+dfsg.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 ncat/test/addrset.c | 104 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 104 insertions(+)
 create mode 100644 ncat/test/addrset.c

(limited to 'ncat/test/addrset.c')

diff --git a/ncat/test/addrset.c b/ncat/test/addrset.c
new file mode 100644
index 0000000..61a0f0d
--- /dev/null
+++ b/ncat/test/addrset.c
@@ -0,0 +1,104 @@
+/*
+    Usage: ./addrset [<specification> ...]
+
+    This program tests the addrset functions in nbase/nbase_addrset.c,
+    the ones that maintain the lists of addresses for --allow and
+    --deny. It takes as arguments specifications that are added to an
+    addrset. It then reads whitespace-separated host names or IP
+    addresses from standard input and echoes only those that are in the
+    addrset.
+
+    David Fifield
+
+    Example:
+    $ echo "1.2.3.4 1.0.0.5 1.2.3.8" | ./addrset "1.2.3.10/24"
+    1.2.3.4
+    1.2.3.8
+*/
+
+#include <errno.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+#include "ncat_core.h"
+
+#ifdef WIN32
+#include "../nsock/src/error.h"
+#endif
+
+
+#ifdef WIN32
+static void win_init(void)
+{
+  WSADATA data;
+  int rc;
+
+  rc = WSAStartup(MAKEWORD(2,2), &data);
+  if (rc)
+    fatal("failed to start winsock: %s\n", socket_strerror(rc));
+}
+#endif
+
+static int resolve_name(const char *name, struct addrinfo **result)
+{
+    struct addrinfo hints = { 0 };
+
+    hints.ai_protocol = IPPROTO_TCP;
+    *result = NULL;
+
+    return getaddrinfo(name, NULL, &hints, result);
+}
+
+int main(int argc, char *argv[])
+{
+    struct addrset *set;
+    char line[1024];
+    int i;
+
+#ifdef WIN32
+    win_init();
+#endif
+
+    set = addrset_new();
+
+    options_init();
+
+    for (i = 1; i < argc; i++) {
+        if (!addrset_add_spec(set, argv[i], o.af, !o.nodns)) {
+            fprintf(stderr, "Error adding spec \"%s\".\n", argv[i]);
+            exit(1);
+        }
+    }
+
+    while (fgets(line, sizeof(line), stdin) != NULL) {
+        char *s, *hostname;
+        struct addrinfo *addrs;
+
+        s = line;
+        while ((hostname = strtok(s, " \t\n")) != NULL) {
+            int rc;
+
+            s = NULL;
+
+            rc = resolve_name(hostname, &addrs);
+            if (rc != 0) {
+                fprintf(stderr, "Error resolving \"%s\": %s.\n", hostname, gai_strerror(rc));
+                continue;
+            }
+            if (addrs == NULL) {
+                fprintf(stderr, "No addresses found for \"%s\".\n", hostname);
+                continue;
+            }
+
+            /* Check just the first address returned. */
+            if (addrset_contains(set, addrs->ai_addr))
+                    printf("%s\n", hostname);
+
+            freeaddrinfo(addrs);
+        }
+    }
+
+    addrset_free(set);
+
+    return 0;
+}
-- 
cgit v1.2.3