54 lines
2.3 KiB
C++
54 lines
2.3 KiB
C++
|
#include <gtest/gtest.h>
|
||
|
|
|
||
|
|
#include "policy.h"
|
||
|
|
#include "raw_packet.h"
|
||
|
|
|
||
|
|
unsigned char data1[] = {
|
||
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0xa4, 0xc6, 0x4f, 0x3b, 0xb3, 0x9a, 0x81, 0x00, 0x66, 0x58, 0x81, 0x00, 0x61, 0xf9, 0x08, 0x00, 0x45, 0xb8, 0x00, 0x94,
|
||
|
|
0xe8, 0x58, 0x00, 0x00, 0xff, 0x04, 0x11, 0x48, 0x45, 0x43, 0x23, 0x92, 0x29, 0xca, 0x2e, 0x6e, 0x45, 0xb8, 0x00, 0x80, 0x00, 0x01, 0x00, 0x00, 0xfe, 0x11,
|
||
|
|
0xde, 0x84, 0x0a, 0x0a, 0x64, 0x19, 0x0a, 0x0a, 0x65, 0x02, 0xf3, 0x9f, 0x42, 0x68, 0x00, 0x6c, 0x4b, 0x9a, 0x00, 0x02, 0x00, 0x00, 0x04, 0x73, 0x6c, 0x10,
|
||
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd,
|
||
|
|
0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd,
|
||
|
|
0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd,
|
||
|
|
0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd, 0xab, 0xcd};
|
||
|
|
|
||
|
|
TEST(POLICY, SELECTED_CHAINING_LIFE_CYCLE)
|
||
|
|
{
|
||
|
|
struct selected_chaining *chaining = NULL;
|
||
|
|
|
||
|
|
chaining = selected_chaining_create(128);
|
||
|
|
EXPECT_TRUE(chaining != nullptr);
|
||
|
|
|
||
|
|
selected_chaining_destory(chaining);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(POLICY, POLICY_ENFORCER_LIFE_CYCLE)
|
||
|
|
{
|
||
|
|
struct raw_pkt_parser *parser = raw_packet_parser_create(LAYER_TYPE_ALL, 8);
|
||
|
|
EXPECT_TRUE(parser != nullptr);
|
||
|
|
const void *payload = raw_packet_parser_parse(parser, (const void *)data1, sizeof(data1));
|
||
|
|
EXPECT_TRUE(payload != nullptr);
|
||
|
|
EXPECT_TRUE((char *)payload - (char *)&data1 == 70);
|
||
|
|
|
||
|
|
const char *profile = "./test_resource/sce.conf";
|
||
|
|
struct policy_enforcer *enforcer = policy_enforcer_create("SCE", profile, 8, NULL);
|
||
|
|
EXPECT_TRUE(enforcer != nullptr);
|
||
|
|
EXPECT_TRUE(policy_enforcer_register(enforcer) == 0);
|
||
|
|
|
||
|
|
int policy_id = 2;
|
||
|
|
int dir_is_internal = 1;
|
||
|
|
struct selected_chaining *chaining = policy_enforce_select_chaining(enforcer, parser, policy_id, dir_is_internal);
|
||
|
|
EXPECT_TRUE(chaining != nullptr);
|
||
|
|
selected_chaining_dump(chaining);
|
||
|
|
selected_chaining_bref(chaining);
|
||
|
|
|
||
|
|
selected_chaining_destory(chaining);
|
||
|
|
policy_enforcer_destory(enforcer);
|
||
|
|
raw_packet_parser_destory(parser);
|
||
|
|
}
|
||
|
|
|
||
|
|
int main(int argc, char **argv)
|
||
|
|
{
|
||
|
|
::testing::InitGoogleTest(&argc, argv);
|
||
|
|
return RUN_ALL_TESTS();
|
||
|
|
}
|