reputation: add ipv6 cidr test

diff --git a/src/tests/reputation.c b/src/tests/reputation.c
index 65cedf3d82291304dded15c6c6aab62af2490d6e..8b72a8a6b6753f96fee37af66f3a9621e6880e6b 100644 (file)
--- a/src/tests/reputation.c
+++ b/src/tests/reputation.c
@@ -36,6 +36,13 @@
     Address a;                                                          \
     uint8_t cat = 0, value = 0;
 
+#define TEST_INIT_WITH_PACKET_IPV6(src, dst)                                                       \
+    uint8_t *buf = (uint8_t *)"Hi all!";                                                           \
+    uint16_t buflen = strlen((char *)buf);                                                         \
+    Packet *p = UTHBuildPacketIPV6SrcDst((uint8_t *)buf, buflen, IPPROTO_TCP, (src), (dst));       \
+    FAIL_IF(p == NULL);                                                                            \
+    TEST_INIT
+
 #define TEST_INIT_WITH_PACKET(ip)                                       \
     uint8_t *buf = (uint8_t *)"Hi all!";                                \
     uint16_t buflen = strlen((char *)buf);                              \
@@ -143,6 +150,26 @@ static int SRepTest07(void) {
     PASS;
 }
 
+static int SRepTest08(void)
+{
+    TEST_INIT_WITH_PACKET_IPV6("2000:0000:0000:0000:0000:0000:0000:0001", "FFFF::1");
+
+    char str1[] = "0.0.0.0/0,1,10\n";
+    char str2[] = "192.168.0.0/16,2,127\n";
+    char str3[] = "2000::/3,1,10\n";
+    char str4[] = "FFFF::/127,2,127\n";
+    FAIL_IF(SRepSplitLine(de_ctx->srepCIDR_ctx, str1, &a, &cat, &value) != 1);
+    FAIL_IF(SRepSplitLine(de_ctx->srepCIDR_ctx, str2, &a, &cat, &value) != 1);
+    FAIL_IF(SRepSplitLine(de_ctx->srepCIDR_ctx, str3, &a, &cat, &value) != 1);
+    FAIL_IF(SRepSplitLine(de_ctx->srepCIDR_ctx, str4, &a, &cat, &value) != 1);
+
+    cat = 1;
+    FAIL_IF(SRepCIDRGetIPRepSrc(de_ctx->srepCIDR_ctx, p, cat, 0) != 10);
+
+    TEST_CLEANUP_WITH_PACKET;
+    PASS;
+}
+
 /** Register the following unittests for the Reputation module */
 void SCReputationRegisterTests(void)
 {
@@ -153,4 +180,5 @@ void SCReputationRegisterTests(void)
     UtRegisterTest("SRepTest05", SRepTest05);
     UtRegisterTest("SRepTest06", SRepTest06);
     UtRegisterTest("SRepTest07", SRepTest07);
+    UtRegisterTest("SRepTest08", SRepTest08);
 }