diff options
Diffstat (limited to 'tests/contrib/test_toeplitz.c')
| -rw-r--r-- | tests/contrib/test_toeplitz.c | 93 |
1 files changed, 93 insertions, 0 deletions
diff --git a/tests/contrib/test_toeplitz.c b/tests/contrib/test_toeplitz.c new file mode 100644 index 0000000..244137c --- /dev/null +++ b/tests/contrib/test_toeplitz.c @@ -0,0 +1,93 @@ +/* Copyright (C) 2022 CZ.NIC, z.s.p.o. <knot-dns@labs.nic.cz> + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation, either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program. If not, see <https://www.gnu.org/licenses/>. + */ + +#include <arpa/inet.h> +#include <netinet/in.h> +#include <sys/socket.h> +#include <tap/basic.h> + +#include "contrib/toeplitz.h" +#include "contrib/wire_ctx.h" + +// Test vectors come from Intel Ethernet Controller X710/XXV710/XL710 Series Datasheet +const uint8_t key[] = { + 0x6d, 0x5a, 0x56, 0xda, 0x25, 0x5b, 0x0e, 0xc2, + 0x41, 0x67, 0x25, 0x3d, 0x43, 0xa3, 0x8f, 0xb0, + 0xd0, 0xca, 0x2b, 0xcb, 0xae, 0x7b, 0x30, 0xb4, + 0x77, 0xcb, 0x2d, 0xa3, 0x80, 0x30, 0xf2, 0x0c, + 0x6a, 0x42, 0xb7, 0x3b, 0xbe, 0xac, 0x01, 0xfa, + 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00 +}; + +void toeplitz_check(int family, const char *src_ip, const char *dst_ip, + uint16_t src_port, uint16_t dst_port, uint32_t expected) +{ + uint8_t data[2 * sizeof(struct in6_addr) + 2 * sizeof(uint16_t)]; + + wire_ctx_t ctx = wire_ctx_init(data, sizeof(data)); + + struct in_addr src_addr4, dst_addr4; + struct in6_addr src_addr6, dst_addr6; + + if (family == AF_INET && + inet_pton(AF_INET, src_ip, &src_addr4) == 1 && + inet_pton(AF_INET, dst_ip, &dst_addr4) == 1) { + wire_ctx_write(&ctx, (uint8_t *)&(src_addr4.s_addr), sizeof(struct in_addr)); + wire_ctx_write(&ctx, (uint8_t *)&(dst_addr4.s_addr), sizeof(struct in_addr)); + } else if (family == AF_INET6 && + inet_pton(AF_INET6, src_ip, &src_addr6) == 1 && + inet_pton(AF_INET6, dst_ip, &dst_addr6) == 1) { + wire_ctx_write(&ctx, (uint8_t *)&(src_addr6.s6_addr), sizeof(struct in6_addr)); + wire_ctx_write(&ctx, (uint8_t *)&(dst_addr6.s6_addr), sizeof(struct in6_addr)); + } else { + assert(0); + } + + wire_ctx_write_u16(&ctx, src_port); + wire_ctx_write_u16(&ctx, dst_port); + + if (ctx.error != KNOT_EOK) { + assert(0); + } + + uint32_t value = toeplitz_hash(key, sizeof(key), data, wire_ctx_offset(&ctx)); + is_int(expected, value, "toeplitz_hash: %u", expected); + + toeplitz_ctx_t toepl; + for (int i = 0; i <= wire_ctx_offset(&ctx); i++) { + toeplitz_init(&toepl, i, key, sizeof(key), data, wire_ctx_offset(&ctx)); + value = toeplitz_finish(&toepl); + is_int(expected, value, "toeplitz_init to %i: %u", i, expected); + } +} + +int main(void) +{ + plan_lazy(); + + toeplitz_check(AF_INET, "66.9.149.187", "161.142.100.80", 2794, 1766, 0x51ccc178); + toeplitz_check(AF_INET, "199.92.111.2", "65.69.140.83", 14230, 4739, 0xc626b0ea); + toeplitz_check(AF_INET, "24.19.198.95", "12.22.207.184", 12898, 38024, 0x5c2b394a); + toeplitz_check(AF_INET, "38.27.205.30", "209.142.163.6", 48228, 2217, 0xafc7327f); + toeplitz_check(AF_INET, "153.39.163.191", "202.188.127.2", 44251, 1303, 0x10e828a2); + + toeplitz_check(AF_INET6, "3ffe:2501:200:1fff::7", "3ffe:2501:200:3::1", 2794, 1766, 0x40207d3d); + toeplitz_check(AF_INET6, "3ffe:501:8::260:97ff:fe40:efab", "ff02::1", 14230, 4739, 0xdde51bbf); + toeplitz_check(AF_INET6, "3ffe:1900:4545:3:200:f8ff:fe21:67cf", "fe80::200:f8ff:fe21:67cf", 44251, 38024, 0x02d1feef); + + return 0; +} |