mirror of
https://github.com/cesanta/mongoose.git
synced 2026-11-01 13:10:43 +00:00
add tests for L2
This commit is contained in:
1 parent
fdcefb087e
commit
e5e13ff156
8 files changed
+779
-27
No files matched your search
+275
@@ -0,0 +1,275 @@
|
||||
#define MG_ENABLE_TCPIP 1
|
||||
#define MG_ENABLE_TCPIP_DRIVER_INIT 0
|
||||
|
||||
#include "mongoose.c"
|
||||
|
||||
static int s_num_tests = 0;
|
||||
static bool s_error = false;
|
||||
|
||||
#ifdef NO_ABORT
|
||||
static int s_abort = 0;
|
||||
#define ABORT() ++s_abort, s_error = true
|
||||
#else
|
||||
#ifdef NO_SLEEP_ABORT
|
||||
#define ABORT() abort()
|
||||
#else
|
||||
#define ABORT() \
|
||||
sleep(2); /* 2s, GH print reason */ \
|
||||
abort();
|
||||
#endif
|
||||
#endif
|
||||
|
||||
#define ASSERT(expr) \
|
||||
do { \
|
||||
s_num_tests++; \
|
||||
if (!(expr)) { \
|
||||
printf("FAILURE %s:%d: %s\n", __FILE__, __LINE__, #expr); \
|
||||
fflush(stdout); \
|
||||
ABORT(); \
|
||||
} \
|
||||
} while (0)
|
||||
|
||||
static const uint8_t s_mac[6] = {2, 3, 4, 5, 6, 7};
|
||||
static const uint8_t s_src[6] = {8, 9, 10, 11, 12, 13};
|
||||
|
||||
static void reset_if(struct mg_tcpip_if *ifp) {
|
||||
memset(ifp, 0, sizeof(*ifp));
|
||||
memcpy(ifp->mac, s_mac, sizeof(s_mac));
|
||||
ifp->l2type = MG_TCPIP_L2_ETH;
|
||||
}
|
||||
|
||||
static size_t mk_eth(uint8_t *buf, const uint8_t *dst, uint16_t type,
|
||||
const uint8_t *payload, size_t len) {
|
||||
struct eth *eth = (struct eth *) buf;
|
||||
memcpy(eth->dst, dst, sizeof(eth->dst));
|
||||
memcpy(eth->src, s_src, sizeof(eth->src));
|
||||
eth->type = mg_htons(type);
|
||||
memcpy(eth + 1, payload, len);
|
||||
return sizeof(*eth) + len;
|
||||
}
|
||||
|
||||
static size_t mk_vlan(uint8_t *buf, const uint8_t *dst, uint16_t vlan,
|
||||
uint16_t type, const uint8_t *payload, size_t len) {
|
||||
struct eth *eth = (struct eth *) buf;
|
||||
struct qtag *qtag = (struct qtag *) ð->type;
|
||||
memcpy(eth->dst, dst, sizeof(eth->dst));
|
||||
memcpy(eth->src, s_src, sizeof(eth->src));
|
||||
qtag->tpid = mg_htons(0x8100);
|
||||
qtag->tci = mg_htons(vlan);
|
||||
MG_STORE_BE16(qtag + 1, type);
|
||||
memcpy((uint8_t *) (qtag + 1) + sizeof(uint16_t), payload, len);
|
||||
return sizeof(*eth) + sizeof(*qtag) + len;
|
||||
}
|
||||
|
||||
static size_t add_fcs(uint8_t *buf, size_t len) {
|
||||
uint32_t crc = mg_crc32(0, (const char *) buf, len);
|
||||
memcpy(buf + len, &crc, sizeof(crc));
|
||||
return len + sizeof(crc);
|
||||
}
|
||||
|
||||
static void test_plain_accept(void) {
|
||||
uint8_t frame[64], payload[] = {1, 2, 3, 4};
|
||||
struct mg_tcpip_if ifp;
|
||||
struct mg_str raw, pay;
|
||||
enum mg_l2proto proto = MG_TCPIP_L2PROTO_ARP;
|
||||
size_t len;
|
||||
reset_if(&ifp);
|
||||
mg_l2_init(&ifp);
|
||||
MG_DEBUG(("MTU: %u, frame size: %u", ifp.l2mtu, ifp.framesize));
|
||||
ASSERT(ifp.l2mtu != 0);
|
||||
ASSERT(ifp.framesize != 0);
|
||||
len = mk_eth(frame, s_mac, 0x0800, payload, sizeof(payload));
|
||||
raw = mg_str_n((char *) frame, len);
|
||||
ASSERT(mg_l2_rx(&ifp, &proto, &pay, &raw) == true);
|
||||
ASSERT(proto == MG_TCPIP_L2PROTO_IPV4);
|
||||
ASSERT(pay.len == sizeof(payload));
|
||||
ASSERT(memcmp(pay.buf, payload, sizeof(payload)) == 0);
|
||||
}
|
||||
|
||||
static void test_plain_discard(void) {
|
||||
uint8_t frame[64], payload[] = {1, 2, 3, 4};
|
||||
char sentinel[] = "discard";
|
||||
struct mg_tcpip_if ifp;
|
||||
struct mg_str raw, pay = mg_str_n(sentinel, sizeof(sentinel));
|
||||
enum mg_l2proto proto = MG_TCPIP_L2PROTO_ARP;
|
||||
size_t len;
|
||||
reset_if(&ifp);
|
||||
mg_l2_init(&ifp);
|
||||
len = mk_eth(frame, s_mac, 0x1234, payload, sizeof(payload));
|
||||
raw = mg_str_n((char *) frame, len);
|
||||
ASSERT(mg_l2_rx(&ifp, &proto, &pay, &raw) == false);
|
||||
ASSERT(proto == MG_TCPIP_L2PROTO_ARP);
|
||||
}
|
||||
|
||||
static void test_mac_check_accept(void) {
|
||||
uint8_t frame[64], payload[] = {1, 2, 3, 4};
|
||||
struct mg_tcpip_if ifp;
|
||||
struct mg_str raw, pay;
|
||||
enum mg_l2proto proto = MG_TCPIP_L2PROTO_ARP;
|
||||
size_t len;
|
||||
reset_if(&ifp);
|
||||
ifp.enable_mac_check = true;
|
||||
mg_l2_init(&ifp);
|
||||
len = mk_eth(frame, s_mac, 0x0800, payload, sizeof(payload));
|
||||
raw = mg_str_n((char *) frame, len);
|
||||
ASSERT(mg_l2_rx(&ifp, &proto, &pay, &raw) == true);
|
||||
ASSERT(proto == MG_TCPIP_L2PROTO_IPV4);
|
||||
ASSERT(pay.len == sizeof(payload));
|
||||
ASSERT(memcmp(pay.buf, payload, sizeof(payload)) == 0);
|
||||
}
|
||||
|
||||
static void test_mac_check_discard(void) {
|
||||
uint8_t frame[64], payload[] = {1, 2, 3, 4}, dst[6] = {1, 1, 1, 1, 1, 1};
|
||||
char sentinel[] = "discard";
|
||||
struct mg_tcpip_if ifp;
|
||||
struct mg_str raw, pay = mg_str_n(sentinel, sizeof(sentinel));
|
||||
enum mg_l2proto proto = MG_TCPIP_L2PROTO_ARP;
|
||||
size_t len;
|
||||
reset_if(&ifp);
|
||||
ifp.enable_mac_check = true;
|
||||
mg_l2_init(&ifp);
|
||||
len = mk_eth(frame, dst, 0x0800, payload, sizeof(payload));
|
||||
raw = mg_str_n((char *) frame, len);
|
||||
ASSERT(mg_l2_rx(&ifp, &proto, &pay, &raw) == false);
|
||||
ASSERT(proto == MG_TCPIP_L2PROTO_ARP);
|
||||
ASSERT(pay.buf == sentinel);
|
||||
ASSERT(pay.len == sizeof(sentinel));
|
||||
}
|
||||
|
||||
static void test_fcs_check_accept(void) {
|
||||
uint8_t frame[64], payload[] = {1, 2, 3, 4};
|
||||
struct mg_tcpip_if ifp;
|
||||
struct mg_str raw, pay;
|
||||
enum mg_l2proto proto = MG_TCPIP_L2PROTO_ARP;
|
||||
size_t len;
|
||||
reset_if(&ifp);
|
||||
ifp.enable_fcs_check = true;
|
||||
mg_l2_init(&ifp);
|
||||
len = add_fcs(frame, mk_eth(frame, s_mac, 0x0800, payload, sizeof(payload)));
|
||||
raw = mg_str_n((char *) frame, len);
|
||||
ASSERT(mg_l2_rx(&ifp, &proto, &pay, &raw) == true);
|
||||
ASSERT(proto == MG_TCPIP_L2PROTO_IPV4);
|
||||
ASSERT(pay.len == sizeof(payload));
|
||||
ASSERT(memcmp(pay.buf, payload, sizeof(payload)) == 0);
|
||||
}
|
||||
|
||||
static void test_fcs_check_discard(void) {
|
||||
uint8_t frame[64], payload[] = {1, 2, 3, 4};
|
||||
char sentinel[] = "discard";
|
||||
struct mg_tcpip_if ifp;
|
||||
struct mg_str raw, pay = mg_str_n(sentinel, sizeof(sentinel));
|
||||
enum mg_l2proto proto = MG_TCPIP_L2PROTO_ARP;
|
||||
size_t len;
|
||||
reset_if(&ifp);
|
||||
ifp.enable_fcs_check = true;
|
||||
mg_l2_init(&ifp);
|
||||
len = add_fcs(frame, mk_eth(frame, s_mac, 0x0800, payload, sizeof(payload)));
|
||||
frame[len - 1] ^= 1;
|
||||
raw = mg_str_n((char *) frame, len);
|
||||
ASSERT(mg_l2_rx(&ifp, &proto, &pay, &raw) == false);
|
||||
ASSERT(proto == MG_TCPIP_L2PROTO_ARP);
|
||||
ASSERT(pay.buf == sentinel);
|
||||
ASSERT(pay.len == sizeof(sentinel));
|
||||
}
|
||||
|
||||
static void test_fcs_check_disabled(void) {
|
||||
uint8_t frame[64], payload[] = {1, 2, 3, 4};
|
||||
struct mg_tcpip_if ifp;
|
||||
struct mg_str raw, pay;
|
||||
enum mg_l2proto proto = MG_TCPIP_L2PROTO_ARP;
|
||||
size_t len;
|
||||
reset_if(&ifp);
|
||||
mg_l2_init(&ifp);
|
||||
len = add_fcs(frame, mk_eth(frame, s_mac, 0x0800, payload, sizeof(payload)));
|
||||
raw = mg_str_n((char *) frame, len);
|
||||
ASSERT(mg_l2_rx(&ifp, &proto, &pay, &raw) == true);
|
||||
ASSERT(proto == MG_TCPIP_L2PROTO_IPV4);
|
||||
ASSERT(pay.len == sizeof(payload) + sizeof(uint32_t));
|
||||
ASSERT(memcmp(pay.buf, payload, sizeof(payload)) == 0);
|
||||
}
|
||||
|
||||
static void test_vlan_accept(void) {
|
||||
uint8_t frame[64], payload[] = {5, 6, 7, 8};
|
||||
struct mg_tcpip_if ifp;
|
||||
struct mg_str raw, pay;
|
||||
enum mg_l2proto proto = MG_TCPIP_L2PROTO_ARP;
|
||||
size_t len;
|
||||
reset_if(&ifp);
|
||||
ifp.l2data.eth.vlan_id = 123;
|
||||
mg_l2_init(&ifp);
|
||||
MG_DEBUG(("MTU: %u, frame size: %u", ifp.l2mtu, ifp.framesize));
|
||||
ASSERT(ifp.l2mtu != 0);
|
||||
ASSERT(ifp.framesize != 0);
|
||||
len = mk_vlan(frame, s_mac, 123, 0x86dd, payload, sizeof(payload));
|
||||
raw = mg_str_n((char *) frame, len);
|
||||
ASSERT(mg_l2_rx(&ifp, &proto, &pay, &raw) == true);
|
||||
ASSERT(proto == MG_TCPIP_L2PROTO_IPV6);
|
||||
ASSERT(pay.len == sizeof(payload));
|
||||
ASSERT(memcmp(pay.buf, payload, sizeof(payload)) == 0);
|
||||
}
|
||||
|
||||
static void test_vlan_discard(void) {
|
||||
uint8_t frame[64], payload[] = {5, 6, 7, 8};
|
||||
char sentinel[] = "discard";
|
||||
struct mg_tcpip_if ifp;
|
||||
struct mg_str raw, pay = mg_str_n(sentinel, sizeof(sentinel));
|
||||
enum mg_l2proto proto = MG_TCPIP_L2PROTO_ARP;
|
||||
size_t len;
|
||||
reset_if(&ifp);
|
||||
ifp.l2data.eth.vlan_id = 123;
|
||||
mg_l2_init(&ifp);
|
||||
len = mk_vlan(frame, s_mac, 124, 0x86dd, payload, sizeof(payload));
|
||||
raw = mg_str_n((char *) frame, len);
|
||||
ASSERT(mg_l2_rx(&ifp, &proto, &pay, &raw) == false);
|
||||
ASSERT(proto == MG_TCPIP_L2PROTO_ARP);
|
||||
ASSERT(pay.buf == sentinel);
|
||||
ASSERT(pay.len == sizeof(sentinel));
|
||||
}
|
||||
|
||||
static void test_tagged_without_vlan_discard(void) {
|
||||
uint8_t frame[64], payload[] = {5, 6, 7, 8};
|
||||
struct mg_tcpip_if ifp;
|
||||
struct mg_str raw, pay;
|
||||
enum mg_l2proto proto = MG_TCPIP_L2PROTO_ARP;
|
||||
size_t len;
|
||||
reset_if(&ifp);
|
||||
mg_l2_init(&ifp);
|
||||
len = mk_vlan(frame, s_mac, 123, 0x86dd, payload, sizeof(payload));
|
||||
raw = mg_str_n((char *) frame, len);
|
||||
ASSERT(mg_l2_rx(&ifp, &proto, &pay, &raw) == false);
|
||||
ASSERT(proto == MG_TCPIP_L2PROTO_ARP);
|
||||
}
|
||||
|
||||
#define DASHBOARD(x) \
|
||||
printf("HEALTH_DASHBOARD\t\"%s\": %s,\n", x, s_error ? "false" : "true")
|
||||
|
||||
int main(void) {
|
||||
s_error = false;
|
||||
test_plain_accept();
|
||||
test_plain_discard();
|
||||
DASHBOARD("eth");
|
||||
|
||||
s_error = false;
|
||||
test_mac_check_accept();
|
||||
test_mac_check_discard();
|
||||
DASHBOARD("mac");
|
||||
|
||||
s_error = false;
|
||||
test_fcs_check_accept();
|
||||
test_fcs_check_discard();
|
||||
test_fcs_check_disabled();
|
||||
DASHBOARD("fcs");
|
||||
|
||||
s_error = false;
|
||||
test_vlan_accept();
|
||||
test_vlan_discard();
|
||||
test_tagged_without_vlan_discard();
|
||||
printf("HEALTH_DASHBOARD\t\"vlan\": %s\n", s_error ? "false" : "true");
|
||||
|
||||
#ifdef NO_ABORT
|
||||
if (s_abort != 0) return EXIT_FAILURE;
|
||||
#endif
|
||||
|
||||
printf("SUCCESS. Total tests: %d\n", s_num_tests);
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
Reference in new issue
Block a user