93 lines
3.6 KiB
C
93 lines
3.6 KiB
C
/* 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;
|
|
}
|