mirror of
https://github.com/cesanta/mongoose
synced 2026-08-25 22:26:06 -04:00
283 lines
9.1 KiB
C
283 lines
9.1 KiB
C
#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);
|
|
buf[len++] = (uint8_t) crc; // store as LE32
|
|
buf[len++] = (uint8_t) (crc >> 8);
|
|
buf[len++] = (uint8_t) (crc >> 16);
|
|
buf[len++] = (uint8_t) (crc >> 24);
|
|
return len;
|
|
}
|
|
|
|
#define DASHBOARD(x) \
|
|
printf("HEALTH_DASHBOARD\t\"%s\": %s,\n", x, s_error ? "false" : "true")
|
|
|
|
static void test_eth(void) {
|
|
struct mg_tcpip_if ifp;
|
|
struct mg_str raw, pay;
|
|
enum mg_l2proto proto;
|
|
size_t len;
|
|
uint8_t frame[64];
|
|
|
|
{ // Plain Ethernet frame is accepted
|
|
uint8_t payload[] = {1, 2, 3, 4};
|
|
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);
|
|
proto = MG_TCPIP_L2PROTO_ARP;
|
|
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);
|
|
}
|
|
{ // Unknown Ethernet type is discarded
|
|
uint8_t payload[] = {1, 2, 3, 4};
|
|
char sentinel[] = "discard";
|
|
reset_if(&ifp);
|
|
mg_l2_init(&ifp);
|
|
len = mk_eth(frame, s_mac, 0x1234, payload, sizeof(payload));
|
|
raw = mg_str_n((char *) frame, len);
|
|
pay = mg_str_n(sentinel, sizeof(sentinel));
|
|
proto = MG_TCPIP_L2PROTO_ARP;
|
|
ASSERT(mg_l2_rx(&ifp, &proto, &pay, &raw) == false);
|
|
ASSERT(proto == MG_TCPIP_L2PROTO_ARP);
|
|
}
|
|
}
|
|
|
|
static void test_mac(void) {
|
|
struct mg_tcpip_if ifp;
|
|
struct mg_str raw, pay;
|
|
enum mg_l2proto proto;
|
|
size_t len;
|
|
uint8_t frame[64];
|
|
|
|
{ // MAC check accepts frames addressed to us
|
|
uint8_t payload[] = {1, 2, 3, 4};
|
|
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);
|
|
proto = MG_TCPIP_L2PROTO_ARP;
|
|
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);
|
|
}
|
|
{ // MAC check discards alien unicast frames
|
|
uint8_t payload[] = {1, 2, 3, 4}, dst[6] = {1, 1, 1, 1, 1, 1};
|
|
char sentinel[] = "discard";
|
|
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);
|
|
pay = mg_str_n(sentinel, sizeof(sentinel));
|
|
proto = MG_TCPIP_L2PROTO_ARP;
|
|
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(void) {
|
|
struct mg_tcpip_if ifp;
|
|
struct mg_str raw, pay;
|
|
enum mg_l2proto proto;
|
|
size_t len;
|
|
uint8_t frame[64];
|
|
|
|
{ // verify add_fcs() stores the CRC bytes in the proper order
|
|
// (internal consistency test)
|
|
uint8_t payload[] = {1, 2, 3, 4};
|
|
uint8_t fcs[] = {0xf4, 0xd4, 0x37, 0xf6};
|
|
len = add_fcs(frame, mk_eth(frame, s_mac, 0x0800, payload,
|
|
sizeof(payload)));
|
|
ASSERT(memcmp(frame + len - sizeof(fcs), fcs, sizeof(fcs)) == 0);
|
|
}
|
|
{ // FCS check strips a valid CRC
|
|
uint8_t payload[] = {1, 2, 3, 4};
|
|
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);
|
|
proto = MG_TCPIP_L2PROTO_ARP;
|
|
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);
|
|
}
|
|
{ // FCS check rejects a bad CRC
|
|
uint8_t payload[] = {1, 2, 3, 4};
|
|
char sentinel[] = "discard";
|
|
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);
|
|
pay = mg_str_n(sentinel, sizeof(sentinel));
|
|
proto = MG_TCPIP_L2PROTO_ARP;
|
|
ASSERT(mg_l2_rx(&ifp, &proto, &pay, &raw) == false);
|
|
ASSERT(proto == MG_TCPIP_L2PROTO_ARP);
|
|
ASSERT(pay.buf == sentinel);
|
|
ASSERT(pay.len == sizeof(sentinel));
|
|
}
|
|
{ // Disabled FCS check leaves the CRC in payload
|
|
uint8_t payload[] = {1, 2, 3, 4};
|
|
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);
|
|
proto = MG_TCPIP_L2PROTO_ARP;
|
|
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(void) {
|
|
struct mg_tcpip_if ifp;
|
|
struct mg_str raw, pay;
|
|
enum mg_l2proto proto;
|
|
size_t len;
|
|
uint8_t frame[64];
|
|
|
|
{ // Configured VLAN accepts matching tagged frames
|
|
uint8_t payload[] = {5, 6, 7, 8};
|
|
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);
|
|
proto = MG_TCPIP_L2PROTO_ARP;
|
|
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);
|
|
}
|
|
{ // Configured VLAN rejects non-matching tags
|
|
uint8_t payload[] = {5, 6, 7, 8};
|
|
char sentinel[] = "discard";
|
|
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);
|
|
pay = mg_str_n(sentinel, sizeof(sentinel));
|
|
proto = MG_TCPIP_L2PROTO_ARP;
|
|
ASSERT(mg_l2_rx(&ifp, &proto, &pay, &raw) == false);
|
|
ASSERT(proto == MG_TCPIP_L2PROTO_ARP);
|
|
ASSERT(pay.buf == sentinel);
|
|
ASSERT(pay.len == sizeof(sentinel));
|
|
}
|
|
{ // Tagged frames are rejected when no VLAN is configured
|
|
uint8_t payload[] = {5, 6, 7, 8};
|
|
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);
|
|
proto = MG_TCPIP_L2PROTO_ARP;
|
|
ASSERT(mg_l2_rx(&ifp, &proto, &pay, &raw) == false);
|
|
ASSERT(proto == MG_TCPIP_L2PROTO_ARP);
|
|
}
|
|
}
|
|
|
|
int main(void) {
|
|
s_error = false;
|
|
test_eth();
|
|
DASHBOARD("eth");
|
|
|
|
s_error = false;
|
|
test_mac();
|
|
DASHBOARD("mac");
|
|
|
|
s_error = false;
|
|
test_fcs();
|
|
DASHBOARD("fcs");
|
|
|
|
s_error = false;
|
|
test_vlan();
|
|
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;
|
|
}
|