stellar-stellar/infra/packet_manager/test/gtest_packet_manager.cpp

#include <gtest/gtest.h>

#include "stellar/mq.h"
#include "packet_parser.h"
#include "packet_private.h"
#include "packet_manager_private.h"

/******************************************************************************
 * [Protocols in frame: eth:ethertype:ip:ipv6:tcp]
 ******************************************************************************
 *
 * Frame 1: 106 bytes on wire (848 bits), 106 bytes captured (848 bits)
 * Ethernet II, Src: JuniperN_45:88:29 (2c:6b:f5:45:88:29), Dst: JuniperN_2a:a2:00 (5c:5e:ab:2a:a2:00)
 * 	Destination: JuniperN_2a:a2:00 (5c:5e:ab:2a:a2:00)
 * 	Source: JuniperN_45:88:29 (2c:6b:f5:45:88:29)
 * 	Type: IPv4 (0x0800)
 * Internet Protocol Version 4, Src: 210.77.88.163, Dst: 59.66.4.50
 * 	0100 .... = Version: 4
 * 	.... 0101 = Header Length: 20 bytes (5)
 * 	Differentiated Services Field: 0x00 (DSCP: CS0, ECN: Not-ECT)
 * 	Total Length: 92
 * 	Identification: 0x0b4d (2893)
 * 	000. .... = Flags: 0x0
 * 	...0 0000 0000 0000 = Fragment Offset: 0
 * 	Time to Live: 59
 * 	Protocol: IPv6 (41)
 * 	Header Checksum: 0x09c8 [validation disabled]
 * 	[Header checksum status: Unverified]
 * 	Source Address: 210.77.88.163
 * 	Destination Address: 59.66.4.50
 * Internet Protocol Version 6, Src: 2001:da8:200:900e:200:5efe:d24d:58a3, Dst: 2600:140e:6::1702:1058
 * 	0110 .... = Version: 6
 * 	.... 0000 0000 .... .... .... .... .... = Traffic Class: 0x00 (DSCP: CS0, ECN: Not-ECT)
 * 	.... 0000 0000 0000 0000 0000 = Flow Label: 0x00000
 * 	Payload Length: 32
 * 	Next Header: TCP (6)
 * 	Hop Limit: 64
 * 	Source Address: 2001:da8:200:900e:200:5efe:d24d:58a3
 * 	Destination Address: 2600:140e:6::1702:1058
 * 	[Source ISATAP IPv4: 210.77.88.163]
 * Transmission Control Protocol, Src Port: 52556, Dst Port: 80, Seq: 0, Len: 0
 * 	Source Port: 52556
 * 	Destination Port: 80
 * 	[Stream index: 0]
 * 	[Conversation completeness: Complete, WITH_DATA (31)]
 * 	[TCP Segment Len: 0]
 * 	Sequence Number: 0    (relative sequence number)
 * 	Sequence Number (raw): 2172673142
 * 	[Next Sequence Number: 1    (relative sequence number)]
 * 	Acknowledgment Number: 0
 * 	Acknowledgment number (raw): 0
 * 	1000 .... = Header Length: 32 bytes (8)
 * 	Flags: 0x002 (SYN)
 * 	Window: 8192
 * 	[Calculated window size: 8192]
 * 	Checksum: 0xf757 [unverified]
 * 	[Checksum Status: Unverified]
 * 	Urgent Pointer: 0
 * 	Options: (12 bytes), Maximum segment size, No-Operation (NOP), Window scale, No-Operation (NOP), No-Operation (NOP), SACK permitted
 * 	[Timestamps]
 */

unsigned char data[] = {
	0x5c, 0x5e, 0xab, 0x2a, 0xa2, 0x00, 0x2c, 0x6b, 0xf5, 0x45, 0x88, 0x29, 0x08, 0x00, 0x45, 0x00, 0x00, 0x5c, 0x0b, 0x4d, 0x00, 0x00, 0x3b, 0x29, 0x09, 0xc8,
	0xd2, 0x4d, 0x58, 0xa3, 0x3b, 0x42, 0x04, 0x32, 0x60, 0x00, 0x00, 0x00, 0x00, 0x20, 0x06, 0x40, 0x20, 0x01, 0x0d, 0xa8, 0x02, 0x00, 0x90, 0x0e, 0x02, 0x00,
	0x5e, 0xfe, 0xd2, 0x4d, 0x58, 0xa3, 0x26, 0x00, 0x14, 0x0e, 0x00, 0x06, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x17, 0x02, 0x10, 0x58, 0xcd, 0x4c, 0x00, 0x50,
	0x81, 0x80, 0x5c, 0x76, 0x00, 0x00, 0x00, 0x00, 0x80, 0x02, 0x20, 0x00, 0xf7, 0x57, 0x00, 0x00, 0x02, 0x04, 0x04, 0xc4, 0x01, 0x03, 0x03, 0x08, 0x01, 0x01,
	0x04, 0x02};

#if 1
TEST(PACKET_MANAGER, NEW_FREE)
{
	struct mq_schema *mq_schema = mq_schema_new();
	EXPECT_TRUE(mq_schema);

	struct packet_manager *pkt_mgr = packet_manager_new(mq_schema, "./conf/stellar.toml");
	EXPECT_TRUE(pkt_mgr);

	EXPECT_TRUE(packet_manager_get_schema(pkt_mgr));
	EXPECT_TRUE(packet_manager_get_runtime(pkt_mgr, 0));
	EXPECT_TRUE(packet_manager_get_runtime(pkt_mgr, 1) == NULL);

	packet_manager_free(pkt_mgr);

	mq_schema_free(mq_schema);
}
#endif

static void on_packet_stage(enum packet_stage stage, struct packet *pkt, void *args)
{
	printf("stage: %s\n", packet_stage_to_str(stage));

	static int count = 0;
	EXPECT_TRUE(count == stage);
	count++;

	EXPECT_TRUE(packet_is_ctrl(pkt));
	EXPECT_TRUE(args == NULL);
}

#if 1
TEST(PACKET_MANAGER, NORNMAL)
{
	// global init
	struct mq_schema *mq_schema = mq_schema_new();
	EXPECT_TRUE(mq_schema);
	struct mq_runtime *mq_rt = mq_runtime_new(mq_schema);
	EXPECT_TRUE(mq_rt);

	// module init
	struct packet_manager *pkt_mgr = packet_manager_new(mq_schema, "./conf/stellar.toml");
	EXPECT_TRUE(pkt_mgr);
	struct packet_manager_schema *schema = packet_manager_get_schema(pkt_mgr);
	EXPECT_TRUE(schema);
	EXPECT_TRUE(packet_manager_schema_add_subscriber(schema, PACKET_STAGE_PREROUTING, on_packet_stage, NULL) == 0);
	EXPECT_TRUE(packet_manager_schema_add_subscriber(schema, PACKET_STAGE_INPUT, on_packet_stage, NULL) == 0);
	EXPECT_TRUE(packet_manager_schema_add_subscriber(schema, PACKET_STAGE_FORWARD, on_packet_stage, NULL) == 0);
	EXPECT_TRUE(packet_manager_schema_add_subscriber(schema, PACKET_STAGE_OUTPUT, on_packet_stage, NULL) == 0);
	EXPECT_TRUE(packet_manager_schema_add_subscriber(schema, PACKET_STAGE_POSTROUTING, on_packet_stage, NULL) == 0);

	// per-thread init
	struct packet_manager_runtime *runtime = packet_manager_get_runtime(pkt_mgr, 0);
	EXPECT_TRUE(runtime);
	packet_manager_runtime_init(runtime, mq_rt);

	// per-thread run
	struct packet pkt;
	memset(&pkt, 0, sizeof(pkt));
	packet_parse(&pkt, (const char *)data, sizeof(data));
	packet_set_ctrl(&pkt, true);

	packet_manager_runtime_input(runtime, &pkt);
	packet_manager_runtime_dispatch(runtime);
	EXPECT_TRUE(packet_manager_runtime_output(runtime) == &pkt);

	// per-thread free

	// module free
	packet_manager_free(pkt_mgr);

	// global free
	mq_runtime_free(mq_rt);
	mq_schema_free(mq_schema);
}
#endif

#if 1
TEST(PACKET_MANAGER, TAKE)
{
	// TODO
	// packet_manager_runtime_take_packet(struct packet_manager_runtime *pkt_mgr_rt, struct packet *pkt);
}
#endif

#if 1
TEST(PACKET_MANAGER, SCHEDULE)
{
	// TODO
	// packet_manager_runtime_schedule_packet(struct packet_manager_runtime *pkt_mgr_rt, struct packet *pkt, enum packet_stage stage);
}
#endif

int main(int argc, char **argv)
{
	::testing::InitGoogleTest(&argc, argv);
	return RUN_ALL_TESTS();
}