#include <gtest/gtest.h>

#include "mpack.h"
#include "control_packet.h"

static u_char control_packet_active0[] = {
	0x85, 0xA5, 0x74, 0x73, 0x79, 0x6E, 0x63, 0xA3, 0x32, 0x2E, 0x30, 0xAA, 0x73, 0x65, 0x73, 0x73,
	0x69, 0x6F, 0x6E, 0x5F, 0x69, 0x64, 0xCF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xA5,
	0x73, 0x74, 0x61, 0x74, 0x65, 0xA6, 0x61, 0x63, 0x74, 0x69, 0x76, 0x65, 0xA6, 0x6D, 0x65, 0x74,
	0x68, 0x6F, 0x64, 0xAD, 0x70, 0x6F, 0x6C, 0x69, 0x63, 0x79, 0x5F, 0x75, 0x70, 0x64, 0x61, 0x74,
	0x65, 0xA6, 0x70, 0x61, 0x72, 0x61, 0x6D, 0x73, 0x83, 0xA3, 0x73, 0x63, 0x65, 0x81, 0xA8, 0x72,
	0x75, 0x6C, 0x65, 0x5F, 0x69, 0x64, 0x73, 0x92, 0xCD, 0x04, 0x57, 0xCD, 0x08, 0xAE, 0xA6, 0x73,
	0x68, 0x61, 0x70, 0x65, 0x72, 0x81, 0xA8, 0x72, 0x75, 0x6C, 0x65, 0x5F, 0x69, 0x64, 0x73, 0x92,
	0xCD, 0x0D, 0x05, 0xCD, 0x11, 0x5C, 0xA5, 0x70, 0x72, 0x6F, 0x78, 0x79, 0x82, 0xA8, 0x72, 0x75,
	0x6C, 0x65, 0x5F, 0x69, 0x64, 0x73, 0x92, 0xCD, 0x15, 0xB3, 0xCD, 0x1A, 0x0A, 0xAD, 0x74, 0x63,
	0x70, 0x5F, 0x68, 0x61, 0x6E, 0x64, 0x73, 0x68, 0x61, 0x6B, 0x65, 0xDC, 0x00, 0x24, 0x01, 0x01,
	0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0xA3, 0x31,
	0x32, 0x33, 0xA3, 0x31, 0x32, 0x33, 0xA3, 0x31, 0x32, 0x33, 0xA3, 0x31, 0x32, 0x33, 0xA3, 0x31,
	0x32, 0x33, 0xA3, 0x31, 0x32, 0x33, 0xA3, 0x31, 0x32, 0x33, 0xA3, 0x31, 0x32, 0x33, 0xA3, 0x31,
	0x32, 0x33, 0xA3, 0x31, 0x32, 0x33, 0xA3, 0x31, 0x32, 0x33, 0xA3, 0x31, 0x32, 0x33, 0xA3, 0x31,
	0x32, 0x33, 0xA3, 0x31, 0x32, 0x33, 0xA3, 0x31, 0x32, 0x33, 0x92, 0xCD, 0x15, 0xB3, 0xCD, 0x1A,
	0x0A, 0x92, 0xCD, 0x15, 0xB3, 0xCD, 0x1A, 0x0A, 0x92, 0xCD, 0x15, 0xB3, 0xCD, 0x1A, 0x0A, 0x92,
	0x01, 0x01, 0x92, 0x01, 0x01};

static u_char control_packet_active1[] = {
	0x85, 0xA5, 0x74, 0x73, 0x79, 0x6E, 0x63, 0xA3, 0x32, 0x2E, 0x30, 0xAA, 0x73, 0x65, 0x73, 0x73,
	0x69, 0x6F, 0x6E, 0x5F, 0x69, 0x64, 0xCF, 0x04, 0x08, 0x02, 0x1B, 0x68, 0x4C, 0x03, 0xE9, 0xA5,
	0x73, 0x74, 0x61, 0x74, 0x65, 0xA6, 0x61, 0x63, 0x74, 0x69, 0x76, 0x65, 0xA6, 0x6D, 0x65, 0x74,
	0x68, 0x6F, 0x64, 0xAD, 0x70, 0x6F, 0x6C, 0x69, 0x63, 0x79, 0x5F, 0x75, 0x70, 0x64, 0x61, 0x74,
	0x65, 0xA6, 0x70, 0x61, 0x72, 0x61, 0x6D, 0x73, 0x81, 0xA3, 0x73, 0x63, 0x65, 0x81, 0xA8, 0x72,
	0x75, 0x6C, 0x65, 0x5F, 0x69, 0x64, 0x73, 0x91, 0xCE, 0x00, 0x0F, 0x2F, 0x7F};

static u_char control_packet_opening[] = {
	0x83, 0xA5, 0x74, 0x73, 0x79, 0x6E, 0x63, 0xA3, 0x32, 0x2E, 0x30, 0xAA, 0x73, 0x65, 0x73, 0x73,
	0x69, 0x6F, 0x6E, 0x5F, 0x69, 0x64, 0xCF, 0x04, 0x08, 0x02, 0x1B, 0x68, 0x4C, 0x03, 0xE9, 0xA5,
	0x73, 0x74, 0x61, 0x74, 0x65, 0xA7, 0x6F, 0x70, 0x65, 0x6E, 0x69, 0x6E, 0x67};

TEST(CONTROL_PACKET, PACKAGE0)
{
	char *data;
	size_t size;
	mpack_writer_t writer;
	mpack_writer_init_growable(&writer, &data, &size);

	// write the example on the msgpack homepage
	mpack_build_map(&writer); // root begin

	// tsync
	mpack_write_cstr(&writer, "tsync");
	mpack_write_cstr(&writer, "2.0");

	// session_id
	mpack_write_cstr(&writer, "session_id");
	mpack_write_u64(&writer, 18446744073709551615);

	// state
	mpack_write_cstr(&writer, "state");
	mpack_write_cstr(&writer, "active");

	// method
	mpack_write_cstr(&writer, "method");
	mpack_write_cstr(&writer, "policy_update");

	// params
	{
		mpack_write_cstr(&writer, "params");
		mpack_build_map(&writer); // params value begin

		// sce
		{
			mpack_write_cstr(&writer, "sce");
			mpack_build_map(&writer);

			mpack_write_cstr(&writer, "rule_ids");
			mpack_build_array(&writer);
			mpack_write_u64(&writer, 1111);
			mpack_write_u64(&writer, 2222);
			mpack_complete_array(&writer);

			mpack_complete_map(&writer);
		}

		// shaper
		{
			mpack_write_cstr(&writer, "shaper");
			mpack_build_map(&writer);

			mpack_write_cstr(&writer, "rule_ids");
			mpack_build_array(&writer);
			mpack_write_u64(&writer, 3333);
			mpack_write_u64(&writer, 4444);
			mpack_complete_array(&writer);

			mpack_complete_map(&writer);
		}

		// proxy
		{
			mpack_write_cstr(&writer, "proxy");
			mpack_build_map(&writer); // proxy value begin

			mpack_write_cstr(&writer, "rule_ids");
			mpack_build_array(&writer);
			mpack_write_u64(&writer, 5555);
			mpack_write_u64(&writer, 6666);
			mpack_complete_array(&writer);

			mpack_write_cstr(&writer, "tcp_handshake");

			mpack_build_array(&writer);			// tcp_handshake value begin
			mpack_write_u32(&writer, 1);		// tcp_seq
			mpack_write_u32(&writer, 1);		// tcp_ack
			mpack_write_u16(&writer, 1);		// tcp_mss_client
			mpack_write_u16(&writer, 1);		// tcp_mss_server
			mpack_write_u8(&writer, 1);			// tcp_wsacle_client
			mpack_write_u8(&writer, 1);			// tcp_wsacle_server
			mpack_write_u8(&writer, 1);			// tcp_sack_client
			mpack_write_u8(&writer, 1);			// tcp_sack_server
			mpack_write_u8(&writer, 1);			// tcp_ts_client
			mpack_write_u8(&writer, 1);			// tcp_ts_server
			mpack_write_u8(&writer, 1);			// tcp_protocol
			mpack_write_u16(&writer, 1);		// tcp_window_client
			mpack_write_u16(&writer, 1);		// tcp_window_server
			mpack_write_u32(&writer, 1);		// tcp_ts_client_val
			mpack_write_u32(&writer, 1);		// tcp_ts_server_val
			mpack_write_u8(&writer, 1);			// tcp_info_packet_cur_dir
			mpack_write_str(&writer, "123", 3); // src_sub_id
			mpack_write_str(&writer, "123", 3); // dst_sub_id
			mpack_write_str(&writer, "123", 3); // src_asn
			mpack_write_str(&writer, "123", 3); // dst_asn
			mpack_write_str(&writer, "123", 3); // src_organization
			mpack_write_str(&writer, "123", 3); // dst_organization
			mpack_write_str(&writer, "123", 3); // src_ip_location_country
			mpack_write_str(&writer, "123", 3); // dst_ip_location_country
			mpack_write_str(&writer, "123", 3); // src_ip_location_provine
			mpack_write_str(&writer, "123", 3); // dst_ip_location_provine
			mpack_write_str(&writer, "123", 3); // src_ip_location_city
			mpack_write_str(&writer, "123", 3); // dst_ip_location_city
			mpack_write_str(&writer, "123", 3); // src_ip_location_subdivision
			mpack_write_str(&writer, "123", 3); // dst_ip_location_subdivision
			mpack_write_str(&writer, "123", 3); // ssl_client_ja3_fingerprint
			mpack_build_array(&writer);			// fqdn_cat_id_val
			mpack_write_u32(&writer, 5555);
			mpack_write_u32(&writer, 6666);
			mpack_complete_array(&writer);
			mpack_build_array(&writer); // tcp_seq_sids
			mpack_write_u16(&writer, 5555);
			mpack_write_u16(&writer, 6666);
			mpack_complete_array(&writer);
			mpack_build_array(&writer); // tcp_ack_sids
			mpack_write_u16(&writer, 5555);
			mpack_write_u16(&writer, 6666);
			mpack_complete_array(&writer);
			mpack_build_array(&writer); // tcp_seq_route_ctx
			mpack_write_u8(&writer, 1);
			mpack_write_u8(&writer, 1);
			mpack_complete_array(&writer);
			mpack_build_array(&writer); // tcp_ack_route_ctx
			mpack_write_u8(&writer, 1);
			mpack_write_u8(&writer, 1);
			mpack_complete_array(&writer);
			mpack_complete_array(&writer); // tcp_handshake value end

			mpack_complete_map(&writer); // proxy value end
		}

		mpack_complete_map(&writer); // params value end
	}

	mpack_complete_map(&writer); // root end

	// finish writing
	EXPECT_TRUE(mpack_writer_destroy(&writer) == mpack_ok);
	EXPECT_TRUE(size == sizeof(control_packet_active0));
	EXPECT_TRUE(memcmp(data, control_packet_active0, size) == 0);

	// data to hex
	for (size_t i = 0; i < size; i++)
	{
		printf("%02X, ", data[i]);
	}
	printf("\n");

	free(data);
}

TEST(CONTROL_PACKET, PACKAGE1)
{
	char *data;
	size_t size;
	mpack_writer_t writer;
	mpack_writer_init_growable(&writer, &data, &size);

	// write the example on the msgpack homepage
	mpack_build_map(&writer); // root begin

	// tsync
	mpack_write_cstr(&writer, "tsync");
	mpack_write_cstr(&writer, "2.0");

	// session_id
	mpack_write_cstr(&writer, "session_id");
	mpack_write_u64(&writer, 290484492702581737);

	// state
	mpack_write_cstr(&writer, "state");
	mpack_write_cstr(&writer, "active");

	// method
	mpack_write_cstr(&writer, "method");
	mpack_write_cstr(&writer, "policy_update");

	// params
	{
		mpack_write_cstr(&writer, "params");
		mpack_build_map(&writer); // params value begin

		// sce
		{
			mpack_write_cstr(&writer, "sce");
			mpack_build_map(&writer);

			mpack_write_cstr(&writer, "rule_ids");
			mpack_build_array(&writer);
			mpack_write_u64(&writer, 995199);
			mpack_complete_array(&writer);

			mpack_complete_map(&writer);
		}

		mpack_complete_map(&writer); // params value end
	}

	mpack_complete_map(&writer); // root end

	// finish writing
	EXPECT_TRUE(mpack_writer_destroy(&writer) == mpack_ok);
	EXPECT_TRUE(size == sizeof(control_packet_active1));
	EXPECT_TRUE(memcmp(data, control_packet_active1, size) == 0);

	// data to hex
	for (size_t i = 0; i < size; i++)
	{
		printf("%02X, ", data[i]);
	}
	printf("\n");

	free(data);
}

TEST(CONTROL_PACKET, PACKAGE2)
{
	char *data;
	size_t size;
	mpack_writer_t writer;
	mpack_writer_init_growable(&writer, &data, &size);

	// write the example on the msgpack homepage
	mpack_build_map(&writer); // root begin

	// tsync
	mpack_write_cstr(&writer, "tsync");
	mpack_write_cstr(&writer, "2.0");

	// session_id
	mpack_write_cstr(&writer, "session_id");
	mpack_write_u64(&writer, 290484492702581737);

	// state
	mpack_write_cstr(&writer, "state");
	mpack_write_cstr(&writer, "opening");

	mpack_complete_map(&writer); // root end

	// finish writing
	EXPECT_TRUE(mpack_writer_destroy(&writer) == mpack_ok);
	EXPECT_TRUE(size == sizeof(control_packet_opening));
	EXPECT_TRUE(memcmp(data, control_packet_opening, size) == 0);

	// data to hex
	for (size_t i = 0; i < size; i++)
	{
		printf("%02X, ", data[i]);
	}
	printf("\n");

	free(data);
}

TEST(CONTROL_PACKET, PARSE0)
{
	struct control_packet handler;
	EXPECT_TRUE(control_packet_parse(&handler, (const char *)control_packet_active0, sizeof(control_packet_active0)) == CTRL_PKT_SUCCESS);
	control_packet_dump(&handler);

	EXPECT_STREQ(handler.tsync, "2.0");
	EXPECT_TRUE(handler.session_id == 18446744073709551615);
	EXPECT_TRUE(handler.state == SESSION_STATE_ACTIVE);
	EXPECT_STREQ(handler.method, "policy_update");
	EXPECT_TRUE(handler.rule_id_num == 2);
	EXPECT_TRUE(handler.rule_ids[0] == 1111);
	EXPECT_TRUE(handler.rule_ids[1] == 2222);
}

TEST(CONTROL_PACKET, PARSE1)
{
	struct control_packet handler;
	EXPECT_TRUE(control_packet_parse(&handler, (const char *)control_packet_active1, sizeof(control_packet_active1)) == CTRL_PKT_SUCCESS);
	control_packet_dump(&handler);

	EXPECT_STREQ(handler.tsync, "2.0");
	EXPECT_TRUE(handler.session_id == 290484492702581737);
	EXPECT_TRUE(handler.state == SESSION_STATE_ACTIVE);
	EXPECT_STREQ(handler.method, "policy_update");
	EXPECT_TRUE(handler.rule_id_num == 1);
	EXPECT_TRUE(handler.rule_ids[0] == 995199);
}

TEST(CONTROL_PACKET, PARSE2)
{
	struct control_packet handler;
	EXPECT_TRUE(control_packet_parse(&handler, (const char *)control_packet_opening, sizeof(control_packet_opening)) == CTRL_PKT_SUCCESS);
	control_packet_dump(&handler);

	EXPECT_STREQ(handler.tsync, "2.0");
	EXPECT_TRUE(handler.session_id == 290484492702581737);
	EXPECT_TRUE(handler.state == SESSION_STATE_OPENING);
	EXPECT_STREQ(handler.method, "");
	EXPECT_TRUE(handler.rule_id_num == 0);
}

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