1103 lines
60 KiB
C++
1103 lines
60 KiB
C++
#include <stdio.h>
|
||
#include <string.h>
|
||
#include <unistd.h>
|
||
#include <arpa/inet.h>
|
||
|
||
#include "tsg_stat.h"
|
||
#include "tsg_rule.h"
|
||
#include "tsg_label.h"
|
||
#include "tsg_entry.h"
|
||
#include "tsg_variable.h"
|
||
#include "tsg_rule_internal.h"
|
||
#include "tsg_protocol_common.h"
|
||
|
||
#include <gtest/gtest.h>
|
||
|
||
extern struct maat_runtime_para g_tsg_maat_rt_para;
|
||
|
||
const struct session_runtime_attribute *session_runtime_attribute_new(const struct streaminfo * a_stream)
|
||
{
|
||
return NULL;
|
||
}
|
||
|
||
const struct session_runtime_attribute *session_runtime_attribute_get(const struct streaminfo * a_stream)
|
||
{
|
||
return 0;
|
||
}
|
||
|
||
int session_runtine_attribute_get_umts_user_info(const struct streaminfo * a_stream, struct umts_user_info * * user_info)
|
||
{
|
||
return 0;
|
||
}
|
||
|
||
int session_mirror_packets_sync(const struct streaminfo * a_stream, struct maat_rule * result, struct mirrored_vlan * vlan)
|
||
{
|
||
return 0;
|
||
}
|
||
|
||
int session_capture_packets_sync(const struct streaminfo * a_stream, struct maat_rule * result, int depth)
|
||
{
|
||
return 0;
|
||
}
|
||
|
||
TEST(TM, ExDataGTPC)
|
||
{
|
||
/******************************************************************************************************
|
||
table name: TSG_DYN_MOBILE_IDENTITY_APN_TEID
|
||
id teid imsi phone_number apn imei is_valid
|
||
"1\t111039813\t460045157065560\t861440152009856\tcmiott.gxqli.mcto60g.com\t8626070583075127\t1",
|
||
"2\t111052899\t460045157053102\t861440152041083\tcmiott.wkctf.mcto60g.com\t8626070583008402\t1"
|
||
******************************************************************************************************/
|
||
|
||
int teid=111039813;
|
||
struct umts_user_info *user_info=(struct umts_user_info *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_GTP_IP2SIGNALING].id, (const char *)&teid, sizeof(int));
|
||
EXPECT_NE(nullptr, user_info);
|
||
EXPECT_STREQ("460045157065560", user_info->imsi);
|
||
EXPECT_STREQ("861440152009856", user_info->msisdn);
|
||
EXPECT_STREQ("111039813.cmiott.gxqli.mcto60g.com", user_info->apn);
|
||
EXPECT_STREQ("8626070583075127", user_info->imei);
|
||
|
||
teid=111052899;
|
||
user_info=(struct umts_user_info *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_GTP_IP2SIGNALING].id, (const char *)&teid, sizeof(int));
|
||
EXPECT_NE(nullptr, user_info);
|
||
EXPECT_STREQ("460045157053102", user_info->imsi);
|
||
EXPECT_STREQ("861440152041083", user_info->msisdn);
|
||
EXPECT_STREQ("111052899.cmiott.wkctf.mcto60g.com", user_info->apn);
|
||
EXPECT_STREQ("8626070583008402", user_info->imei);
|
||
}
|
||
|
||
TEST(TM, ExDataAPPIDDictUnknown)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: APP_ID_DICT
|
||
app_id app_name parent_app_id parent_app_name category subcategory technology risk characteristics depends_on_app_ids implicitly_uses_app_ids deny_action continue_scanning tcp_timeout udp_timeout tcp_half_close tcp_time_wait is_valid
|
||
"4\tunknown\t0\tnull\tcategory\tsubcategory\ttechnology\trisk\tcharacteristics\tnull\tnull\tnull\t1\t3600\t3600\t1800\t1800\t1"
|
||
**************************************************************************************************************************/
|
||
long long ll_app_id=4;
|
||
struct app_id_dict *dict=(struct app_id_dict *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_APP_ID_DICT].id, (const char *)&ll_app_id, sizeof(long long));
|
||
dict=(struct app_id_dict *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_APP_ID_DICT].id, (const char *)&ll_app_id, sizeof(long long));
|
||
EXPECT_NE(nullptr, dict);
|
||
EXPECT_EQ(4, dict->app_id);
|
||
EXPECT_STREQ("unknown", dict->app_name);
|
||
EXPECT_EQ(0, dict->parent_app_id);
|
||
EXPECT_STREQ(nullptr, dict->parent_app_name);
|
||
|
||
EXPECT_STREQ("category", dict->category);
|
||
EXPECT_STREQ("subcategory", dict->subcategory);
|
||
EXPECT_STREQ("technology", dict->technology);
|
||
EXPECT_STREQ("risk", dict->risk);
|
||
EXPECT_STREQ("characteristics", dict->characteristics);
|
||
//depends_on_app_ids
|
||
//implicitly_uses_app_ids
|
||
// deny_action
|
||
EXPECT_EQ(1, dict->continue_scanning);
|
||
EXPECT_EQ(3600, dict->tcp_timeout);
|
||
EXPECT_EQ(3600, dict->udp_timeout);
|
||
EXPECT_EQ(1800, dict->tcp_half_close);
|
||
EXPECT_EQ(1800, dict->tcp_time_wait);
|
||
|
||
dict=(struct app_id_dict *)matched_rule_cites_app_id_dict(g_tsg_maat_feather, ll_app_id);
|
||
}
|
||
|
||
TEST(TM, ExDataAPPIDDictDenyActionDrop)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: APP_ID_DICT
|
||
app_id app_name parent_app_id parent_app_name category subcategory technology risk characteristics depends_on_app_ids implicitly_uses_app_ids deny_action continue_scanning tcp_timeout udp_timeout tcp_half_close tcp_time_wait is_valid
|
||
"67\thttp\t0\tnull\tnetworking\tinfrastructure\tnetwork-protocol\t3\tused-by-malware,vulnerability,widely-used\tnull\tnull\t{\"method\":\"drop\",\"after_n_packets\":0,\"send_icmp_unreachable\":1,\"send_tcp_reset\":1}\t0\t60\t120\t30\t30\t1",
|
||
**************************************************************************************************************************/
|
||
long long ll_app_id=67;
|
||
struct app_id_dict *dict=(struct app_id_dict *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_APP_ID_DICT].id, (const char *)&ll_app_id, sizeof(long long));
|
||
EXPECT_NE(nullptr, dict);
|
||
EXPECT_EQ(67, dict->app_id);
|
||
EXPECT_STREQ("http", dict->app_name);
|
||
EXPECT_EQ(0, dict->parent_app_id);
|
||
EXPECT_STREQ(nullptr, dict->parent_app_name);
|
||
|
||
EXPECT_STREQ("networking", dict->category);
|
||
EXPECT_STREQ("infrastructure", dict->subcategory);
|
||
EXPECT_STREQ("network-protocol", dict->technology);
|
||
EXPECT_STREQ("3", dict->risk);
|
||
EXPECT_STREQ("used-by-malware,vulnerability,widely-used", dict->characteristics);
|
||
//depends_on_app_ids
|
||
//implicitly_uses_app_ids
|
||
|
||
//deny_action
|
||
EXPECT_EQ(TSG_DENY_TYPE_APP_DROP, dict->deny_app_para.type);
|
||
EXPECT_EQ(0, dict->deny_app_para.after_n_packets);
|
||
EXPECT_EQ(1, dict->deny_app_para.drop_para.send_icmp_enable);
|
||
EXPECT_EQ(1, dict->deny_app_para.drop_para.send_reset_enable);
|
||
|
||
EXPECT_EQ(0, dict->continue_scanning);
|
||
EXPECT_EQ(60, dict->tcp_timeout);
|
||
EXPECT_EQ(120, dict->udp_timeout);
|
||
EXPECT_EQ(30, dict->tcp_half_close);
|
||
EXPECT_EQ(30, dict->tcp_time_wait);
|
||
|
||
dict=(struct app_id_dict *)matched_rule_cites_app_id_dict(g_tsg_maat_feather, ll_app_id);
|
||
}
|
||
|
||
TEST(TM, ExDataAPPIDDictDenyActionRatelimit)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: APP_ID_DICT
|
||
app_id app_name parent_app_id parent_app_name category subcategory technology risk characteristics depends_on_app_ids implicitly_uses_app_ids deny_action continue_scanning tcp_timeout udp_timeout tcp_half_close tcp_time_wait is_valid
|
||
"68\thttps\t0\tnull\tnetworking\tinfrastructure\tnetwork-protocol\t3\tused-by-malware,vulnerability,widely-used\tnull\tnull\t{\"method\":\"rate_limit\","bps":1000}\t0\t0\t0\t0\t0\t1",
|
||
**************************************************************************************************************************/
|
||
long long ll_app_id=68;
|
||
struct app_id_dict *dict=(struct app_id_dict *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_APP_ID_DICT].id, (const char *)&ll_app_id, sizeof(long long));
|
||
EXPECT_NE(nullptr, dict);
|
||
EXPECT_EQ(68, dict->app_id);
|
||
EXPECT_STREQ("https", dict->app_name);
|
||
EXPECT_EQ(0, dict->parent_app_id);
|
||
EXPECT_STREQ(nullptr, dict->parent_app_name);
|
||
|
||
EXPECT_STREQ("networking", dict->category);
|
||
EXPECT_STREQ("infrastructure", dict->subcategory);
|
||
EXPECT_STREQ("network-protocol", dict->technology);
|
||
EXPECT_STREQ("3", dict->risk);
|
||
EXPECT_STREQ("used-by-malware,vulnerability,widely-used", dict->characteristics);
|
||
//depends_on_app_ids
|
||
//implicitly_uses_app_ids
|
||
|
||
//deny_action
|
||
EXPECT_EQ(TSG_DENY_TYPE_APP_RATELIMIT, dict->deny_app_para.type);
|
||
EXPECT_EQ(1000, dict->deny_app_para.bps);
|
||
|
||
EXPECT_EQ(0, dict->continue_scanning);
|
||
EXPECT_EQ(0, dict->tcp_timeout);
|
||
EXPECT_EQ(0, dict->udp_timeout);
|
||
EXPECT_EQ(0, dict->tcp_half_close);
|
||
EXPECT_EQ(0, dict->tcp_time_wait);
|
||
|
||
dict=(struct app_id_dict *)matched_rule_cites_app_id_dict(g_tsg_maat_feather, ll_app_id);
|
||
}
|
||
|
||
TEST(TM, ExDataASNBuiltIn)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_IP_ASN_BUILT_IN
|
||
**************************************************************************************************************************/
|
||
EXPECT_EQ(1, 1);
|
||
}
|
||
|
||
TEST(TM, ExDataASNUserDefine)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_IP_ASN_USER_DEFINED
|
||
**************************************************************************************************************************/
|
||
EXPECT_EQ(1, 1);
|
||
}
|
||
|
||
TEST(TM, ExDataLocationBuiltIn)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_IP_LOCATION_BUILT_IN
|
||
location_id geoname_id addr_type start_ip end_ip latitude longitude coords language continent_abbr continent_full country_abbr country_full province_abbr province_full city_full subdivision_addr time_zone is_valid
|
||
"6777621\t1819730\t4\t124.156.128.0\t124.156.191.255\t22.25\t114.1667\t50.0\ten\tAS\tAsia\tHK\tHong\\bKong\tOther\tOther\tOther\tRoad1\tAsia/Hong_Kong\t1",
|
||
"3716523\t1814992\t4\t192.168.50.1\t192.168.50.255\t34.7725\t113.7266\t50.0\ten\tAS\tAsia\tCN\tChina\tBeijing\tBeijing\tXin\\bXi\\bGang\tRoad1\tAsia/Shanghai\t1"
|
||
**************************************************************************************************************************/
|
||
struct ip_addr temp_addr = {0};
|
||
temp_addr.ip_type = 4;
|
||
temp_addr.ipv4 = inet_addr("124.156.128.1");
|
||
struct location_info *location = NULL;
|
||
maat_ip_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_LOCATION_BUILT_IN].id, &temp_addr, (void **)&location, 1);
|
||
|
||
EXPECT_STREQ("Other", location->city_full);
|
||
EXPECT_STREQ("Hong Kong", location->country_full);
|
||
EXPECT_STREQ("Other", location->province_full);
|
||
EXPECT_STREQ(NULL, location->subdivision_addr);
|
||
EXPECT_EQ(2, location->ref_cnt);
|
||
|
||
temp_addr.ipv4 = inet_addr("192.168.50.2");
|
||
maat_ip_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_LOCATION_BUILT_IN].id, &temp_addr, (void **)&location, 1);
|
||
|
||
EXPECT_STREQ("Xin Xi Gang", location->city_full);
|
||
EXPECT_STREQ("China", location->country_full);
|
||
EXPECT_STREQ("Beijing", location->province_full);
|
||
EXPECT_STREQ(NULL, location->subdivision_addr); // location_field_num == 18; not 19
|
||
EXPECT_EQ(2, location->ref_cnt);
|
||
|
||
location = NULL;
|
||
temp_addr.ipv4 = inet_addr("124.156.127.1");
|
||
maat_ip_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_LOCATION_BUILT_IN].id, &temp_addr, (void **)&location, 1);
|
||
EXPECT_EQ(NULL, location);
|
||
}
|
||
|
||
TEST(TM, ExDataLocationUserDefine)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_IP_LOCATION_USER_DEFINED
|
||
location_id geoname_id addr_type start_ip end_ip latitude longitude coords language continent_abbr continent_full country_abbr country_full province_abbr province_full city_full subdivision_addr time_zone is_valid
|
||
"371652\t181499\t4\t192.168.50.10\t192.168.50.255\t34.7725\t113.7266\t50.0\ten\tAS\tAsia\tCN\tChina\tBeijing\tBeijing\tHua\\bYan\\bBei\\bLi\tRoad1\tAsia/Shanghai\t1"
|
||
**************************************************************************************************************************/
|
||
struct ip_addr temp_addr = {0};
|
||
temp_addr.ip_type = 4;
|
||
temp_addr.ipv4 = inet_addr("192.168.50.11");
|
||
struct location_info *location = NULL;
|
||
maat_ip_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_LOCATION_USER_DEFINED].id, &temp_addr, (void **)&location, 1);
|
||
|
||
EXPECT_STREQ("Hua Yan Bei Li", location->city_full);
|
||
EXPECT_STREQ("China", location->country_full);
|
||
EXPECT_STREQ("Beijing", location->province_full);
|
||
EXPECT_STREQ(NULL, location->subdivision_addr); // location_field_num == 18; not 19
|
||
EXPECT_EQ(2, location->ref_cnt);
|
||
|
||
location = NULL;
|
||
temp_addr.ipv4 = inet_addr("124.156.127.1");
|
||
maat_ip_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_LOCATION_USER_DEFINED].id, &temp_addr, (void **)&location, 1);
|
||
EXPECT_EQ(NULL, location);
|
||
}
|
||
|
||
TEST(TM, ExDataFQDNCategoryBuiltIn)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_FQDN_CATEGORY_BUILT_IN
|
||
id category_id fqdn match_method(0=complete,1=suf) is_valid
|
||
"106285681\t4\t106285681.201198.com\t1\t1",
|
||
"106285682\t5\t106285682.201198.com\t1\t1"
|
||
**************************************************************************************************************************/
|
||
struct fqdn_category *category_ids[2] = {0};
|
||
int ret = maat_fqdn_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_FQDN_CAT_BUILT_IN].id, "106285681.201198.com", (void **)category_ids, 2);
|
||
EXPECT_EQ(1, ret);
|
||
EXPECT_EQ(4, category_ids[0]->category_id);
|
||
EXPECT_EQ(NULL, category_ids[1]);
|
||
|
||
category_ids[0] = NULL;
|
||
ret = 0;
|
||
ret = maat_fqdn_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_FQDN_CAT_BUILT_IN].id, "106285682.201198.com", (void **)category_ids, 2);
|
||
EXPECT_EQ(1, ret);
|
||
EXPECT_EQ(5, category_ids[0]->category_id);
|
||
EXPECT_EQ(NULL, category_ids[1]);
|
||
|
||
category_ids[0] = NULL;
|
||
}
|
||
|
||
TEST(TM, ExDataFQDNCategoryUserDefine)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_FQDN_CATEGORY_USER_DEFINED
|
||
id category_id fqdn match_method(0=complete,1=suf) is_valid
|
||
"1106285681\t4\t1106285681.201198.com\t1\t1",
|
||
"1106285682\t5\t1106285682.201198.com\t1\t1"
|
||
**************************************************************************************************************************/
|
||
struct fqdn_category *category_ids[2] = {0};
|
||
int ret = maat_fqdn_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_FQDN_CAT_USER_DEFINED].id, "1106285681.201198.com", (void **)category_ids, 2);
|
||
EXPECT_EQ(1, ret);
|
||
EXPECT_EQ(4, category_ids[0]->category_id);
|
||
EXPECT_EQ(NULL, category_ids[1]);
|
||
|
||
category_ids[0] = NULL;
|
||
ret = 0;
|
||
ret = maat_fqdn_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_FQDN_CAT_USER_DEFINED].id, "1106285682.201198.com", (void **)category_ids, 2);
|
||
EXPECT_EQ(1, ret);
|
||
EXPECT_EQ(5, category_ids[0]->category_id);
|
||
EXPECT_EQ(NULL, category_ids[1]);
|
||
|
||
category_ids[0] = NULL;
|
||
}
|
||
|
||
TEST(TM, ExDataTunnelCatalog)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_TUNNEL_CATALOG
|
||
tunnel_id tunnel_name tunnel_type composition is_valid
|
||
"977\t1.1.1.1-1.1.1.1\tGTP\t1367\t1",
|
||
"978\t1.1.1.1-1.1.1.1\tGTP\t1367&1605\t1"
|
||
**************************************************************************************************************************/
|
||
struct tunnel_catalog *t_catalog[3] = {0};
|
||
unsigned long long bool_id_array[2] = {1367, 1605};
|
||
size_t n_bool_id_array = 2;
|
||
int ret = maat_bool_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_TUNNEL_CATALOG].id, (unsigned long long *)bool_id_array, n_bool_id_array, (void **)(&t_catalog), 3);
|
||
|
||
EXPECT_EQ(2, ret);
|
||
EXPECT_EQ(978, t_catalog[0]->id);
|
||
EXPECT_STREQ("1.1.1.1-1.1.1.1", t_catalog[0]->name);
|
||
EXPECT_STREQ("GTP", t_catalog[0]->type);
|
||
EXPECT_STREQ("1367&1605", t_catalog[0]->composition);
|
||
|
||
EXPECT_EQ(977, t_catalog[1]->id);
|
||
EXPECT_STREQ("1.1.1.1-1.1.1.1", t_catalog[1]->name);
|
||
EXPECT_STREQ("GTP", t_catalog[1]->type);
|
||
EXPECT_STREQ("1367", t_catalog[1]->composition);
|
||
|
||
EXPECT_EQ(NULL, t_catalog[2]);
|
||
t_catalog[0] = NULL;
|
||
t_catalog[1] = NULL;
|
||
}
|
||
|
||
TEST(TM, ExDataTunnelEndpoint)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_TUNNEL_ENDPOINT
|
||
endpoint_id endpoint_group_id addr_type addr_format ip1 ip2 description is_valid
|
||
"989\t4\t192.50.0.0\t192.50.255.255\ttest\t1",
|
||
"990\t4\t192.50.0.0\t192.50.255.255\ttest\t1"
|
||
**************************************************************************************************************************/
|
||
struct tunnel_endpoint *all_endpoint[3] = {0};
|
||
struct ip_addr temp_addr = {0};
|
||
temp_addr.ip_type = 4;
|
||
temp_addr.ipv4 = inet_addr("192.50.0.0");
|
||
int ret = maat_ip_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_TUNNEL_ENDPOINT].id, (const struct ip_addr *)&temp_addr, (void **)all_endpoint, 3);
|
||
|
||
EXPECT_EQ(2, ret);
|
||
EXPECT_EQ(989, all_endpoint[0]->id);
|
||
EXPECT_STREQ("test", all_endpoint[0]->description);
|
||
EXPECT_EQ(990, all_endpoint[1]->id);
|
||
EXPECT_STREQ("test", all_endpoint[1]->description);
|
||
|
||
ret = 0;
|
||
all_endpoint[0] = NULL;
|
||
all_endpoint[1] = NULL;
|
||
temp_addr.ipv4 = inet_addr("192.50.255.255");
|
||
ret = maat_ip_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_TUNNEL_ENDPOINT].id, (const struct ip_addr *)&temp_addr, (void **)all_endpoint, 3);
|
||
|
||
EXPECT_EQ(2, ret);
|
||
EXPECT_EQ(989, all_endpoint[0]->id);
|
||
EXPECT_STREQ("test", all_endpoint[0]->description);
|
||
EXPECT_EQ(990, all_endpoint[1]->id);
|
||
EXPECT_STREQ("test", all_endpoint[1]->description);
|
||
|
||
all_endpoint[0] = NULL;
|
||
all_endpoint[1] = NULL;
|
||
}
|
||
|
||
TEST(TM, ExDataTunnelLabel)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_TUNNEL_LABEL
|
||
label_id label_group_id label is_valid
|
||
"15560\t15560\tVLAN_ID\t1",
|
||
"15561\t15561\tVLAN_ID\t1"
|
||
**************************************************************************************************************************/
|
||
void *label_id = maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_TUNNEL_LABEL].id, "VLAN_ID", strlen("VLAN_ID"));
|
||
EXPECT_EQ(0, (long long)label_id);
|
||
}
|
||
|
||
TEST(TM, ExDataSubscriberID)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_DYN_SUBSCRIBER_IP
|
||
id addr_type ip subscriber_id is_valid
|
||
"1299\t4\t192.168.56.28\ttest5628\t1",
|
||
"1300\t4\t192.168.56.27\ttest5627\t1",
|
||
"1301\t6\t1030::C9B4:FF12:48AA:1A2B\ttest5630\t1",
|
||
"1302\t6\t1030::C9B4:FF12:3799:1A2B\ttest5629\t1"
|
||
**************************************************************************************************************************/
|
||
struct streaminfo a_stream = {0};
|
||
a_stream.addr.addrtype = ADDR_TYPE_IPV4;
|
||
struct stream_tuple4_v4 tuple4_v4 = {0};
|
||
a_stream.addr.tuple4_v4 = &tuple4_v4;
|
||
inet_pton(AF_INET, "192.168.56.28", &tuple4_v4.daddr);
|
||
inet_pton(AF_INET, "192.168.56.27", &tuple4_v4.saddr);
|
||
struct subscribe_id_info *dest_subscribe_id = NULL;
|
||
struct subscribe_id_info *source_subscribe_id = NULL;
|
||
srt_attribute_set_subscriber_id(&a_stream, g_tsg_maat_feather, &source_subscribe_id, &dest_subscribe_id);
|
||
EXPECT_STREQ("test5628", dest_subscribe_id->subscribe_id);
|
||
EXPECT_STREQ("test5627", source_subscribe_id->subscribe_id);
|
||
|
||
a_stream.addr.addrtype = ADDR_TYPE_IPV6;
|
||
dest_subscribe_id = NULL;
|
||
source_subscribe_id = NULL;
|
||
struct stream_tuple4_v6 tuple4_v6 = {0};
|
||
a_stream.addr.tuple4_v6 = &tuple4_v6;
|
||
inet_pton(AF_INET6, "1030::C9B4:FF12:48AA:1A2B", &tuple4_v6.daddr);
|
||
inet_pton(AF_INET6, "1030::C9B4:FF12:3799:1A2B", &tuple4_v6.saddr);
|
||
srt_attribute_set_subscriber_id(&a_stream, g_tsg_maat_feather, &source_subscribe_id, &dest_subscribe_id);
|
||
EXPECT_STREQ("test5630", dest_subscribe_id->subscribe_id);
|
||
EXPECT_STREQ("test5629", source_subscribe_id->subscribe_id);
|
||
}
|
||
|
||
TEST(TM, ExDataDNSRecordsProfileA)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_PROFILE_DNS_RECORDS
|
||
record_id record_name record_type record_values is_valid
|
||
"8119\teditTypeA\tA\t[{\"value\":\"1.1.1.1\",\"priority\":null},{\"value\":\"2.2.2.2\",\"priority\":null},{\"value\":\"3.3.3.3\",\"priority\":null}]\t1",
|
||
"7961\tFile\tAAAA\t[{\"value\":\"1030::C9B4:FF12:48AA:1A2B\",\"priority\":null},{\"value\":\"1030::C9B4:FF12:48AA:1A2C\",\"priority\":null}]\t1",
|
||
"7701\tTypeCNAME\tCNAME\t[{\"value\":\"www.facebook.com\",\"priority\":null},{\"value\":\"www.twitter.com\",\"priority\":null}]\t1"
|
||
**************************************************************************************************************************/
|
||
long long profile_id = 8119;
|
||
struct dns_profile_records *profile_records = (struct dns_profile_records *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_DNS_PROFILE_RECORD].id, (const char *)&profile_id, sizeof(long long));
|
||
EXPECT_EQ(8119, profile_records->record_id);
|
||
EXPECT_EQ(1, profile_records->answer_type);
|
||
EXPECT_EQ(3, profile_records->record_num);
|
||
|
||
EXPECT_EQ(1, profile_records->record_val[0].answer_type);
|
||
EXPECT_EQ(4, profile_records->record_val[0].len);
|
||
EXPECT_EQ(0, profile_records->record_val[0].selected_flag);
|
||
EXPECT_EQ(inet_addr("1.1.1.1"), profile_records->record_val[0].v4_addr.s_addr);
|
||
|
||
EXPECT_EQ(1, profile_records->record_val[1].answer_type);
|
||
EXPECT_EQ(4, profile_records->record_val[1].len);
|
||
EXPECT_EQ(0, profile_records->record_val[1].selected_flag);
|
||
EXPECT_EQ(inet_addr("2.2.2.2"), profile_records->record_val[1].v4_addr.s_addr);
|
||
|
||
EXPECT_EQ(1, profile_records->record_val[2].answer_type);
|
||
EXPECT_EQ(4, profile_records->record_val[2].len);
|
||
EXPECT_EQ(0, profile_records->record_val[2].selected_flag);
|
||
EXPECT_EQ(inet_addr("3.3.3.3"), profile_records->record_val[2].v4_addr.s_addr);
|
||
|
||
profile_id = 8110;
|
||
profile_records = (struct dns_profile_records *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_DNS_PROFILE_RECORD].id, (const char *)&profile_id, sizeof(long long));
|
||
EXPECT_EQ(NULL, profile_records);
|
||
}
|
||
|
||
TEST(TM, ExDataDNSRecordsProfileAAAA)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_PROFILE_DNS_RECORDS
|
||
record_id record_name record_type record_values is_valid
|
||
"8119\teditTypeA\tA\t[{\"value\":\"1.1.1.1\",\"priority\":null},{\"value\":\"2.2.2.2\",\"priority\":null},{\"value\":\"3.3.3.3\",\"priority\":null}]\t1",
|
||
"7961\tFile\tAAAA\t[{\"value\":\"1030::C9B4:FF12:48AA:1A2B\",\"priority\":null},{\"value\":\"1030::C9B4:FF12:48AA:1A2C\",\"priority\":null}]\t1",
|
||
"7701\tTypeCNAME\tCNAME\t[{\"value\":\"www.facebook.com\",\"priority\":null},{\"value\":\"www.twitter.com\",\"priority\":null}]\t1"
|
||
**************************************************************************************************************************/
|
||
long long profile_id = 7961;
|
||
uint8_t u6_addr8[16] = {0};
|
||
struct dns_profile_records *profile_records = (struct dns_profile_records *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_DNS_PROFILE_RECORD].id, (const char *)&profile_id, sizeof(long long));
|
||
EXPECT_EQ(7961, profile_records->record_id);
|
||
EXPECT_EQ(28, profile_records->answer_type);
|
||
EXPECT_EQ(2, profile_records->record_num);
|
||
|
||
EXPECT_EQ(28, profile_records->record_val[0].answer_type);
|
||
EXPECT_EQ(16, profile_records->record_val[0].len);
|
||
EXPECT_EQ(0, profile_records->record_val[0].selected_flag);
|
||
inet_pton(AF_INET6, "1030::C9B4:FF12:48AA:1A2B", u6_addr8);
|
||
EXPECT_EQ(0, memcmp(u6_addr8, profile_records->record_val[0].v6_addr.__in6_u.__u6_addr8, 16));
|
||
|
||
EXPECT_EQ(28, profile_records->record_val[1].answer_type);
|
||
EXPECT_EQ(16, profile_records->record_val[1].len);
|
||
EXPECT_EQ(0, profile_records->record_val[1].selected_flag);
|
||
inet_pton(AF_INET6, "1030::C9B4:FF12:48AA:1A2C", u6_addr8);
|
||
EXPECT_EQ(0, memcmp(u6_addr8, profile_records->record_val[1].v6_addr.__in6_u.__u6_addr8, 16));
|
||
|
||
profile_id = 8110;
|
||
profile_records = (struct dns_profile_records *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_DNS_PROFILE_RECORD].id, (const char *)&profile_id, sizeof(long long));
|
||
EXPECT_EQ(NULL, profile_records);
|
||
}
|
||
|
||
TEST(TM, ExDataDNSRecordsProfileCNAME)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_PROFILE_DNS_RECORDS
|
||
record_id record_name record_type record_values is_valid
|
||
"8119\teditTypeA\tA\t[{\"value\":\"1.1.1.1\",\"priority\":null},{\"value\":\"2.2.2.2\",\"priority\":null},{\"value\":\"3.3.3.3\",\"priority\":null}]\t1",
|
||
"7961\tFile\tAAAA\t[{\"value\":\"1030::C9B4:FF12:48AA:1A2B\",\"priority\":null},{\"value\":\"1030::C9B4:FF12:48AA:1A2C\",\"priority\":null}]\t1",
|
||
"7701\tTypeCNAME\tCNAME\t[{\"value\":\"www.facebook.com\",\"priority\":null},{\"value\":\"www.twitter.com\",\"priority\":null}]\t1"
|
||
**************************************************************************************************************************/
|
||
long long profile_id = 7701;
|
||
struct dns_profile_records *profile_records = (struct dns_profile_records *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_DNS_PROFILE_RECORD].id, (const char *)&profile_id, sizeof(long long));
|
||
EXPECT_EQ(7701, profile_records->record_id);
|
||
EXPECT_EQ(5, profile_records->answer_type);
|
||
EXPECT_EQ(2, profile_records->record_num);
|
||
|
||
EXPECT_EQ(5, profile_records->record_val[0].answer_type);
|
||
EXPECT_EQ(strlen("www.facebook.com"), profile_records->record_val[0].len);
|
||
EXPECT_EQ(0, profile_records->record_val[0].selected_flag);
|
||
EXPECT_STREQ("www.facebook.com", profile_records->record_val[0].cname);
|
||
|
||
EXPECT_EQ(5, profile_records->record_val[1].answer_type);
|
||
EXPECT_EQ(strlen("www.twitter.com"), profile_records->record_val[1].len);
|
||
EXPECT_EQ(0, profile_records->record_val[1].selected_flag);
|
||
EXPECT_STREQ("www.twitter.com", profile_records->record_val[1].cname);
|
||
|
||
profile_id = 8110;
|
||
profile_records = (struct dns_profile_records *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_DNS_PROFILE_RECORD].id, (const char *)&profile_id, sizeof(long long));
|
||
EXPECT_EQ(NULL, profile_records);
|
||
}
|
||
|
||
TEST(TM, ExDataResponsePagesProfile)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_PROFILE_RESPONSE_PAGES
|
||
profile_id profile_name format path is_valid
|
||
"957\ttest-html-1\thtml\tforeign_files/TSG_PROFILE_RESPONSE_PAGES.1\t1",
|
||
"958\ttest-html-2\thtml\tforeign_files/TSG_PROFILE_RESPONSE_PAGES.2\t1"
|
||
**************************************************************************************************************************/
|
||
long long profile_id = 957;
|
||
struct http_response_pages *response_pages = (struct http_response_pages *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_RESPONSE_PAGES].id, (const char *)&profile_id, sizeof(long long));
|
||
EXPECT_EQ(957, response_pages->profile_id);
|
||
EXPECT_EQ(HTTP_RESPONSE_FORMAT_HTML, response_pages->format);
|
||
EXPECT_EQ(strlen("test1"), response_pages->content_len);
|
||
EXPECT_STREQ("test1", response_pages->content);
|
||
|
||
profile_id = 958;
|
||
response_pages = (struct http_response_pages *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_RESPONSE_PAGES].id, (const char *)&profile_id, sizeof(long long));
|
||
EXPECT_EQ(958, response_pages->profile_id);
|
||
EXPECT_EQ(HTTP_RESPONSE_FORMAT_HTML, response_pages->format);
|
||
EXPECT_EQ(strlen("test2"), response_pages->content_len);
|
||
EXPECT_STREQ("test2", response_pages->content);
|
||
|
||
profile_id = 959;
|
||
response_pages = (struct http_response_pages *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_RESPONSE_PAGES].id, (const char *)&profile_id, sizeof(long long));
|
||
EXPECT_EQ(NULL, response_pages);
|
||
}
|
||
|
||
TEST(TM, ExDataTrafficMirrorProfile)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_PROFILE_TRAFFIC_MIRROR
|
||
profile_id profile_name vlan_ids is_valid
|
||
"845\t168.50.28yinyong\t[3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34]\t1",
|
||
"123\ttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest\t[66]\t1"
|
||
**************************************************************************************************************************/
|
||
int profile_id = 845;
|
||
struct traffic_mirror_profile *mirror_profile = (struct traffic_mirror_profile *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_PROFILE_MIRROR].id, (const char *)&(profile_id), sizeof(int));
|
||
EXPECT_EQ(845, mirror_profile->profile_id);
|
||
EXPECT_EQ(32, mirror_profile->vlan.num);
|
||
for (int i = 0; i < 32; i++)
|
||
{
|
||
EXPECT_EQ(i + 3, mirror_profile->vlan.id[i]);
|
||
}
|
||
|
||
profile_id = 123;
|
||
mirror_profile = (struct traffic_mirror_profile *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_PROFILE_MIRROR].id, (const char *)&(profile_id), sizeof(int));
|
||
EXPECT_EQ(123, mirror_profile->profile_id);
|
||
EXPECT_EQ(1, mirror_profile->vlan.num);
|
||
EXPECT_EQ(66, mirror_profile->vlan.id[0]);
|
||
|
||
profile_id = 124;
|
||
mirror_profile = (struct traffic_mirror_profile *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_PROFILE_MIRROR].id, (const char *)&(profile_id), sizeof(long long));
|
||
EXPECT_EQ(NULL, mirror_profile);
|
||
}
|
||
|
||
TEST(TM, ExDataSessionRecordEnable)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: T_VSYS_INFO
|
||
id is_enable_session_record is_valid
|
||
"6\t1\t1"
|
||
**************************************************************************************************************************/
|
||
EXPECT_EQ(g_tsg_maat_rt_para.session_record_switch, 1);
|
||
}
|
||
|
||
TEST(TMAPI, GetUmtsUserInfoFromRedis)
|
||
{
|
||
/******************************************************************************************************
|
||
table name: TSG_DYN_MOBILE_IDENTITY_APN_TEID
|
||
id teid imsi phone_number apn imei is_valid
|
||
"1\t111039813\t460045157065560\t861440152009856\tcmiott.gxqli.mcto60g.com\t8626070583075127\t1",
|
||
"2\t111052899\t460045157053102\t861440152041083\tcmiott.wkctf.mcto60g.com\t8626070583008402\t1"
|
||
******************************************************************************************************/
|
||
|
||
int teid = 111039813;
|
||
struct umts_user_info *user_info = (struct umts_user_info *)tsg_get_umts_user_info_form_redis(g_tsg_maat_feather, teid);
|
||
struct umts_user_info *user_info_right = (struct umts_user_info *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_GTP_IP2SIGNALING].id, (const char *)&teid, sizeof(int));
|
||
EXPECT_EQ(user_info, user_info_right);
|
||
|
||
teid = 111052899;
|
||
user_info = (struct umts_user_info *)tsg_get_umts_user_info_form_redis(g_tsg_maat_feather, teid);
|
||
user_info_right = (struct umts_user_info *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_GTP_IP2SIGNALING].id, (const char *)&teid, sizeof(int));
|
||
EXPECT_EQ(user_info, user_info_right);
|
||
|
||
user_info = NULL;
|
||
user_info_right = NULL;
|
||
}
|
||
|
||
TEST(TMAPI, GetAppIdDictUnknown)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: APP_ID_DICT
|
||
app_id app_name parent_app_id parent_app_name category subcategory technology risk characteristics depends_on_app_ids implicitly_uses_app_ids deny_action continue_scanning tcp_timeout udp_timeout tcp_half_close tcp_time_wait is_valid
|
||
"4\tunknown\t0\tnull\tcategory\tsubcategory\ttechnology\trisk\tcharacteristics\tnull\tnull\tnull\t1\t3600\t3600\t1800\t1800\t1"
|
||
**************************************************************************************************************************/
|
||
long long ll_app_id = 4;
|
||
struct app_id_dict *dict_right = (struct app_id_dict *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_APP_ID_DICT].id, (const char *)&ll_app_id, sizeof(long long));
|
||
struct app_id_dict *dict = (struct app_id_dict *)matched_rule_cites_app_id_dict(g_tsg_maat_feather, ll_app_id);
|
||
EXPECT_EQ(dict_right, dict);
|
||
|
||
dict_right = NULL;
|
||
dict = NULL;
|
||
}
|
||
|
||
TEST(TMAPI, GetAppIdDictDenyActionDrop)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: APP_ID_DICT
|
||
app_id app_name parent_app_id parent_app_name category subcategory technology risk characteristics depends_on_app_ids implicitly_uses_app_ids deny_action continue_scanning tcp_timeout udp_timeout tcp_half_close tcp_time_wait is_valid
|
||
"67\thttp\t0\tnull\tnetworking\tinfrastructure\tnetwork-protocol\t3\tused-by-malware,vulnerability,widely-used\tnull\tnull\t{\"method\":\"drop\",\"after_n_packets\":0,\"send_icmp_unreachable\":1,\"send_tcp_reset\":1}\t0\t60\t120\t30\t30\t1",
|
||
**************************************************************************************************************************/
|
||
long long ll_app_id = 67;
|
||
struct app_id_dict *dict_right = (struct app_id_dict *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_APP_ID_DICT].id, (const char *)&ll_app_id, sizeof(long long));
|
||
struct app_id_dict *dict = (struct app_id_dict *)matched_rule_cites_app_id_dict(g_tsg_maat_feather, ll_app_id);
|
||
EXPECT_EQ(dict_right, dict);
|
||
|
||
dict_right = NULL;
|
||
dict = NULL;
|
||
}
|
||
|
||
TEST(TMAPI, GetAppIdDictDenyActionRatelimit)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: APP_ID_DICT
|
||
app_id app_name parent_app_id parent_app_name category subcategory technology risk characteristics depends_on_app_ids implicitly_uses_app_ids deny_action continue_scanning tcp_timeout udp_timeout tcp_half_close tcp_time_wait is_valid
|
||
"68\thttps\t0\tnull\tnetworking\tinfrastructure\tnetwork-protocol\t3\tused-by-malware,vulnerability,widely-used\tnull\tnull\t{\"method\":\"rate_limit\","bps":1000}\t0\t0\t0\t0\t0\t1",
|
||
**************************************************************************************************************************/
|
||
long long ll_app_id = 68;
|
||
struct app_id_dict *dict_right = (struct app_id_dict *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_APP_ID_DICT].id, (const char *)&ll_app_id, sizeof(long long));
|
||
struct app_id_dict *dict = (struct app_id_dict *)matched_rule_cites_app_id_dict(g_tsg_maat_feather, ll_app_id);
|
||
EXPECT_EQ(dict_right, dict);
|
||
|
||
dict_right = NULL;
|
||
dict = NULL;
|
||
}
|
||
|
||
TEST(TMAPI, GetAppNameByIdIllegalPara)
|
||
{
|
||
// int tsg_get_app_name_by_id(int app_id, char *app_name, int app_name_len, int is_joint_parent)
|
||
char app_name[128] = {0};
|
||
EXPECT_EQ(0, tsg_get_app_name_by_id(g_tsg_maat_feather, 0, app_name, 128, 1));
|
||
EXPECT_EQ(0, tsg_get_app_name_by_id(g_tsg_maat_feather, 68, NULL, 128, 1));
|
||
EXPECT_EQ(0, tsg_get_app_name_by_id(g_tsg_maat_feather, 68, app_name, 0, 1));
|
||
}
|
||
|
||
TEST(TMAPI, GetAppNameByIdParentAppId0)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: APP_ID_DICT
|
||
app_id app_name parent_app_id parent_app_name category subcategory technology risk characteristics depends_on_app_ids implicitly_uses_app_ids deny_action continue_scanning tcp_timeout udp_timeout tcp_half_close tcp_time_wait is_valid
|
||
"68\thttps\t0\tnull\tnetworking\tinfrastructure\tnetwork-protocol\t3\tused-by-malware,vulnerability,widely-used\tnull\tnull\t{\"method\":\"rate_limit\","bps":1000}\t0\t0\t0\t0\t0\t1",
|
||
**************************************************************************************************************************/
|
||
int app_id = 68;
|
||
char app_name[128] = {0};
|
||
EXPECT_EQ(0, tsg_get_app_name_by_id(g_tsg_maat_feather, app_id, app_name, 2, 1));
|
||
EXPECT_EQ(strlen("https"), tsg_get_app_name_by_id(g_tsg_maat_feather, app_id, app_name, 128, 1));
|
||
}
|
||
|
||
TEST(TMAPI, GetAppNameByIdParentAppId1)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: APP_ID_DICT
|
||
app_id app_name parent_app_id parent_app_name category subcategory technology risk characteristics depends_on_app_ids implicitly_uses_app_ids deny_action continue_scanning tcp_timeout udp_timeout tcp_half_close tcp_time_wait is_valid
|
||
"70\thttps\t1\tssl\tnetworking\tinfrastructure\tnetwork-protocol\t3\tused-by-malware,vulnerability,widely-used\tnull\tnull\t{\"method\":\"rate_limit\","bps":1000}\t0\t0\t0\t0\t0\t1",
|
||
**************************************************************************************************************************/
|
||
int app_id = 70;
|
||
char app_name[128] = {0};
|
||
EXPECT_EQ(0, tsg_get_app_name_by_id(g_tsg_maat_feather, app_id, app_name, 2, 1));
|
||
EXPECT_EQ(strlen("ssl.https"), tsg_get_app_name_by_id(g_tsg_maat_feather, app_id, app_name, 128, 1));
|
||
}
|
||
|
||
TEST(TMAPI, GetLocationInfoLocationBuiltIn)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_IP_LOCATION_BUILT_IN
|
||
location_id geoname_id addr_type start_ip end_ip latitude longitude coords language continent_abbr continent_full country_abbr country_full province_abbr province_full city_full subdivision_addr time_zone is_valid
|
||
"6777621\t1819730\t4\t124.156.128.0\t124.156.191.255\t22.25\t114.1667\t50.0\ten\tAS\tAsia\tHK\tHong\\bKong\tOther\tOther\tOther\tRoad1\tAsia/Hong_Kong\t1",
|
||
"3716523\t1814992\t4\t192.168.50.1\t192.168.50.255\t34.7725\t113.7266\t50.0\ten\tAS\tAsia\tCN\tChina\tBeijing\tBeijing\tXin\\bXi\\bGang\tRoad1\tAsia/Shanghai\t1"
|
||
**************************************************************************************************************************/
|
||
struct streaminfo a_stream = {0};
|
||
struct stream_tuple4_v4 tuple4_v4 = {0};
|
||
struct location_info *client_location = NULL, *server_location = NULL;
|
||
tuple4_v4.daddr = inet_addr("124.156.128.2");
|
||
tuple4_v4.saddr = inet_addr("192.168.50.3");
|
||
a_stream.addr.addrtype = ADDR_TYPE_IPV4;
|
||
a_stream.addr.tuple4_v4 = &tuple4_v4;
|
||
srt_attribute_set_ip_location(&a_stream, g_tsg_maat_feather, &client_location, &server_location);
|
||
|
||
struct ip_addr temp_addr = {0};
|
||
temp_addr.ip_type = 4;
|
||
temp_addr.ipv4 = inet_addr("124.156.128.1");
|
||
struct location_info *location = NULL;
|
||
maat_ip_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_LOCATION_BUILT_IN].id, &temp_addr, (void **)&location, 1);
|
||
EXPECT_EQ(server_location, location);
|
||
|
||
temp_addr.ipv4 = inet_addr("192.168.50.2");
|
||
maat_ip_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_LOCATION_BUILT_IN].id, &temp_addr, (void **)&location, 1);
|
||
EXPECT_EQ(client_location, location);
|
||
|
||
location = NULL;
|
||
client_location = NULL;
|
||
server_location = NULL;
|
||
tuple4_v4.daddr = inet_addr("124.156.127.1");
|
||
tuple4_v4.saddr = inet_addr("192.168.49.1");
|
||
a_stream.addr.tuple4_v4 = &tuple4_v4;
|
||
srt_attribute_set_ip_location(&a_stream, g_tsg_maat_feather, &client_location, &server_location);
|
||
EXPECT_EQ(NULL, server_location);
|
||
EXPECT_EQ(NULL, client_location);
|
||
}
|
||
|
||
TEST(TMAPI, GetLocationInfoLocationUserDefine)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_IP_LOCATION_USER_DEFINED
|
||
location_id geoname_id addr_type start_ip end_ip latitude longitude coords language continent_abbr continent_full country_abbr country_full province_abbr province_full city_full subdivision_addr time_zone is_valid
|
||
"371652\t181499\t4\t192.168.50.10\t192.168.50.255\t34.7725\t113.7266\t50.0\ten\tAS\tAsia\tCN\tChina\tBeijing\tBeijing\tHua\\bYan\\bBei\\bLi\tRoad1\tAsia/Shanghai\t1"
|
||
**************************************************************************************************************************/
|
||
struct streaminfo a_stream = {0};
|
||
struct stream_tuple4_v4 tuple4_v4 = {0};
|
||
struct location_info *client_location = NULL, *server_location = NULL;
|
||
tuple4_v4.saddr = inet_addr("192.168.50.11");
|
||
tuple4_v4.daddr = inet_addr("192.168.50.12");
|
||
a_stream.addr.addrtype = ADDR_TYPE_IPV4;
|
||
a_stream.addr.tuple4_v4 = &tuple4_v4;
|
||
srt_attribute_set_ip_location(&a_stream, g_tsg_maat_feather, &client_location, &server_location);
|
||
|
||
struct ip_addr temp_addr = {0};
|
||
temp_addr.ip_type = 4;
|
||
temp_addr.ipv4 = inet_addr("192.168.50.11");
|
||
struct location_info *location = NULL;
|
||
maat_ip_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_LOCATION_USER_DEFINED].id, &temp_addr, (void **)&location, 1);
|
||
EXPECT_EQ(client_location, location);
|
||
EXPECT_EQ(server_location, location);
|
||
}
|
||
|
||
TEST(TMAPI, GetLocationInfoBoth)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_IP_LOCATION_USER_DEFINED
|
||
location_id geoname_id addr_type start_ip end_ip latitude longitude coords language continent_abbr continent_full country_abbr country_full province_abbr province_full city_full subdivision_addr time_zone is_valid
|
||
"371652\t181499\t4\t192.168.50.10\t192.168.50.255\t34.7725\t113.7266\t50.0\ten\tAS\tAsia\tCN\tChina\tBeijing\tBeijing\tHua\\bYan\\bBei\\bLi\tRoad1\tAsia/Shanghai\t1"
|
||
**************************************************************************************************************************/
|
||
/*************************************************************************************************************************
|
||
table name: TSG_IP_LOCATION_BUILT_IN
|
||
location_id geoname_id addr_type start_ip end_ip latitude longitude coords language continent_abbr continent_full country_abbr country_full province_abbr province_full city_full subdivision_addr time_zone is_valid
|
||
"6777621\t1819730\t4\t124.156.128.0\t124.156.191.255\t22.25\t114.1667\t50.0\ten\tAS\tAsia\tHK\tHong\\bKong\tOther\tOther\tOther\tRoad1\tAsia/Hong_Kong\t1",
|
||
"3716523\t1814992\t4\t192.168.50.1\t192.168.50.255\t34.7725\t113.7266\t50.0\ten\tAS\tAsia\tCN\tChina\tBeijing\tBeijing\tXin\\bXi\\bGang\tRoad1\tAsia/Shanghai\t1"
|
||
**************************************************************************************************************************/
|
||
struct streaminfo a_stream = {0};
|
||
struct stream_tuple4_v4 tuple4_v4 = {0};
|
||
struct location_info *client_location = NULL, *server_location = NULL;
|
||
tuple4_v4.saddr = inet_addr("124.156.128.11");
|
||
tuple4_v4.daddr = inet_addr("192.168.50.12");
|
||
a_stream.addr.addrtype = ADDR_TYPE_IPV4;
|
||
a_stream.addr.tuple4_v4 = &tuple4_v4;
|
||
srt_attribute_set_ip_location(&a_stream, g_tsg_maat_feather, &client_location, &server_location); // find in MAAT_PLUGIN_LOCATION_USER_DEFINED
|
||
|
||
struct ip_addr temp_addr = {0};
|
||
temp_addr.ip_type = 4;
|
||
temp_addr.ipv4 = inet_addr("192.168.50.11");
|
||
struct location_info *location = NULL;
|
||
maat_ip_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_LOCATION_BUILT_IN].id, &temp_addr, (void **)&location, 1);
|
||
EXPECT_NE(server_location, client_location);
|
||
EXPECT_NE(client_location, location);
|
||
EXPECT_NE(server_location, location);
|
||
}
|
||
|
||
TEST(TMAPI, GetFQDNCategoryIdUserDefine)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_FQDN_CATEGORY_USER_DEFINED
|
||
id category_id fqdn match_method(0=complete,1=suf) is_valid
|
||
"1106285683\t6\t1106285683.201198.com\t1\t1",
|
||
"1106285684\t7\t1106285684.201198.com\t1\t1"
|
||
"1106285685\t7\t1106285684.201198.com\t1\t1"
|
||
**************************************************************************************************************************/
|
||
unsigned int category_ids[3] = {0};
|
||
int ret = tsg_get_fqdn_category_ids(g_tsg_maat_feather, (char *)"1106285683.201198.com", category_ids, MAX_CATEGORY_ID_NUM);
|
||
EXPECT_EQ(1, ret);
|
||
EXPECT_EQ(6, category_ids[0]);
|
||
|
||
category_ids[0] = 0;
|
||
ret = 0;
|
||
ret = tsg_get_fqdn_category_ids(g_tsg_maat_feather, (char *)"1106285684.201198.com", category_ids, MAX_CATEGORY_ID_NUM); // get_fqdn_category_id 去重
|
||
EXPECT_EQ(1, ret);
|
||
EXPECT_EQ(7, category_ids[0]);
|
||
EXPECT_EQ(0, category_ids[1]);
|
||
}
|
||
|
||
TEST(TMAPI, GetFQDNCategoryIdBuiltIn)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_FQDN_CATEGORY_BUILT_IN
|
||
id category_id fqdn match_method(0=complete,1=suf) is_valid
|
||
"106285688\t8\t106285688.201198.com\t1\t1",
|
||
"106285689\t9\t106285689.201198.com\t1\t1"
|
||
"106285690\t9\t106285689.201198.com\t1\t1"
|
||
"106285691\t10\t1106285683.201198.com\t1\t1"
|
||
**************************************************************************************************************************/
|
||
unsigned int category_ids[3] = {0};
|
||
int ret = tsg_get_fqdn_category_ids(g_tsg_maat_feather, (char *)"106285688.201198.com", category_ids, MAX_CATEGORY_ID_NUM);
|
||
EXPECT_EQ(1, ret);
|
||
EXPECT_EQ(8, category_ids[0]);
|
||
|
||
category_ids[0] = 0;
|
||
ret = 0;
|
||
ret = tsg_get_fqdn_category_ids(g_tsg_maat_feather, (char *)"106285689.201198.com", category_ids, MAX_CATEGORY_ID_NUM); // get_fqdn_category_id 去重
|
||
EXPECT_EQ(1, ret);
|
||
EXPECT_EQ(9, category_ids[0]);
|
||
EXPECT_EQ(0, category_ids[1]);
|
||
}
|
||
|
||
TEST(TMAPI, GetFQDNCategoryIdBoth)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_FQDN_CATEGORY_BUILT_IN
|
||
id category_id fqdn match_method(0=complete,1=suf) is_valid
|
||
"106285691\t10\t1106285683.201198.com\t1\t1"
|
||
|
||
table name: TSG_FQDN_CATEGORY_USER_DEFINED
|
||
id category_id fqdn match_method(0=complete,1=suf) is_valid
|
||
"1106285683\t6\t1106285683.201198.com\t1\t1",
|
||
**************************************************************************************************************************/
|
||
unsigned int category_ids[3] = {0};
|
||
int ret = tsg_get_fqdn_category_ids(g_tsg_maat_feather, (char *)"1106285683.201198.com", category_ids, MAX_CATEGORY_ID_NUM); // 优先get TSG_FQDN_CATEGORY_USER_DEFINED
|
||
EXPECT_EQ(1, ret);
|
||
EXPECT_EQ(6, category_ids[0]);
|
||
EXPECT_EQ(0, category_ids[1]);
|
||
}
|
||
|
||
TEST(TMAPI, ScanTunnelId)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_TUNNEL_CATALOG
|
||
tunnel_id tunnel_name tunnel_type composition is_valid
|
||
"977\t1.1.1.1-1.1.1.1\tGTP\t1367\t1",
|
||
"978\t1.1.1.1-1.1.1.1\tGTP\t1367&1605\t1"
|
||
**************************************************************************************************************************/
|
||
|
||
// tsg_scan_tunnel_id(const struct streaminfo *a_stream, struct maat_rule *results, size_t n_results, struct maat_state *s_mid, long long *bool_id_array, size_t n_bool_id_array)
|
||
long long bool_id_array[2] = {1367, 1605};
|
||
size_t n_bool_id_array = 2, n_results = 2;
|
||
struct streaminfo a_stream = {0};
|
||
struct maat_rule results[2] = {0};
|
||
struct maat_state *s_mid = maat_state_new(g_tsg_maat_feather, 0);
|
||
int ret = tsg_scan_tunnel_id(&a_stream, g_tsg_maat_feather, results, n_results, s_mid, bool_id_array, n_bool_id_array);
|
||
|
||
EXPECT_EQ(2, ret);
|
||
EXPECT_EQ(4, results[0].rule_id);
|
||
EXPECT_EQ(16, results[0].action);
|
||
EXPECT_EQ(0, results[0].service_id);
|
||
EXPECT_EQ(1, results[0].do_log);
|
||
|
||
EXPECT_EQ(3, results[1].rule_id);
|
||
EXPECT_EQ(1, results[1].action);
|
||
EXPECT_EQ(0, results[1].service_id);
|
||
EXPECT_EQ(1, results[1].do_log);
|
||
maat_state_free(s_mid);
|
||
s_mid=NULL;
|
||
}
|
||
|
||
extern int tsg_get_endpoint_id(const struct streaminfo *a_stream, struct maat *feather, struct tunnel_endpoint **client_endpoint, struct tunnel_endpoint **server_endpoint, long long *endpoint_id_array, int endpoint_id_array_num);
|
||
TEST(TMAPI, GetEndPointId)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_TUNNEL_ENDPOINT
|
||
endpoint_id endpoint_group_id addr_type addr_format ip1 ip2 description is_valid
|
||
"991\t4\t192.40.128.0\t192.40.255.255\ttest\t1",
|
||
"992\t4\t192.40.0.0\t192.40.127.255\ttest\t1"
|
||
**************************************************************************************************************************/
|
||
|
||
// int tsg_get_endpoint_id(const struct streaminfo *a_stream, struct maat *feather, struct tunnel_endpoint **client_endpoint, struct tunnel_endpoint **server_endpoint, long long *endpoint_id_array, int endpoint_id_array_num)
|
||
long long endpoint_id_array[128] = {0};
|
||
int endpoint_id_array_num = 128;
|
||
struct streaminfo a_stream = {0};
|
||
struct stream_tuple4_v4 tuple4_v4 = {0};
|
||
struct tunnel_endpoint *client_endpoint = NULL, *server_endpoint = NULL;
|
||
tuple4_v4.daddr = inet_addr("192.40.128.1");
|
||
tuple4_v4.saddr = inet_addr("192.40.0.1");
|
||
a_stream.addr.addrtype = ADDR_TYPE_IPV4;
|
||
a_stream.addr.tuple4_v4 = &tuple4_v4;
|
||
|
||
int ret = tsg_get_endpoint_id(&a_stream, g_tsg_maat_feather, &client_endpoint, &server_endpoint, endpoint_id_array, endpoint_id_array_num);
|
||
EXPECT_EQ(2, ret);
|
||
EXPECT_EQ(992, endpoint_id_array[0]);
|
||
EXPECT_EQ(991, endpoint_id_array[1]);
|
||
|
||
ret = 0;
|
||
struct tunnel_endpoint *all_endpoint[2] = {0};
|
||
struct ip_addr temp_addr = {0};
|
||
temp_addr.ip_type = 4;
|
||
temp_addr.ipv4 = inet_addr("192.40.128.1");
|
||
ret = maat_ip_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_TUNNEL_ENDPOINT].id, (const struct ip_addr *)&temp_addr, (void **)all_endpoint, 2);
|
||
|
||
EXPECT_EQ(1, ret);
|
||
EXPECT_EQ(server_endpoint, all_endpoint[0]);
|
||
|
||
ret = 0;
|
||
all_endpoint[0] = NULL;
|
||
all_endpoint[1] = NULL;
|
||
temp_addr.ipv4 = inet_addr("192.40.0.1");
|
||
ret = maat_ip_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_TUNNEL_ENDPOINT].id, (const struct ip_addr *)&temp_addr, (void **)all_endpoint, 2);
|
||
|
||
EXPECT_EQ(1, ret);
|
||
EXPECT_EQ(client_endpoint, all_endpoint[0]);
|
||
|
||
all_endpoint[0] = NULL;
|
||
all_endpoint[1] = NULL;
|
||
}
|
||
|
||
TEST(TMAPI, GetEndPointIdMultipleTunnelEndpoint)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_TUNNEL_ENDPOINT
|
||
endpoint_id endpoint_group_id addr_type addr_format ip1 ip2 description is_valid
|
||
"989\t4\t192.50.0.0\t192.50.255.255\ttest\t1",
|
||
"990\t4\t192.50.0.0\t192.50.255.255\ttest\t1",
|
||
"992\t4\t192.40.0.0\t192.40.127.255\ttest\t1"
|
||
**************************************************************************************************************************/
|
||
|
||
// int tsg_get_endpoint_id(const struct streaminfo *a_stream, struct maat *feather, struct tunnel_endpoint **client_endpoint, struct tunnel_endpoint **server_endpoint, long long *endpoint_id_array, int endpoint_id_array_num)
|
||
long long endpoint_id_array[128] = {0};
|
||
int endpoint_id_array_num = 128;
|
||
struct streaminfo a_stream = {0};
|
||
struct stream_tuple4_v4 tuple4_v4 = {0};
|
||
struct tunnel_endpoint *client_endpoint = NULL, *server_endpoint = NULL;
|
||
tuple4_v4.daddr = inet_addr("192.50.128.1");
|
||
tuple4_v4.saddr = inet_addr("192.40.0.1");
|
||
a_stream.addr.addrtype = ADDR_TYPE_IPV4;
|
||
a_stream.addr.tuple4_v4 = &tuple4_v4;
|
||
|
||
int ret = tsg_get_endpoint_id(&a_stream, g_tsg_maat_feather, &client_endpoint, &server_endpoint, endpoint_id_array, endpoint_id_array_num);
|
||
EXPECT_EQ(3, ret);
|
||
EXPECT_EQ(992, endpoint_id_array[0]);
|
||
EXPECT_EQ(989, endpoint_id_array[1]);
|
||
EXPECT_EQ(990, endpoint_id_array[2]);
|
||
|
||
EXPECT_EQ(992, client_endpoint->id);
|
||
EXPECT_EQ(990, server_endpoint->id);
|
||
}
|
||
|
||
TEST(TMAPI, GetEndPointIdSourceIpEqualToDestIp) // 也可能在一个网段，是否可以两个不同的endpoint_id拥有相同的网段；是否有可能源ip和目的ip同时可以处于一个endpoint的网段；endpoint_id_array是否可能重复
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_TUNNEL_ENDPOINT
|
||
endpoint_id endpoint_group_id addr_type addr_format ip1 ip2 description is_valid
|
||
"989\t4\t192.50.0.0\t192.50.255.255\ttest\t1",
|
||
"990\t4\t192.50.0.0\t192.50.255.255\ttest\t1",
|
||
**************************************************************************************************************************/
|
||
|
||
// int tsg_get_endpoint_id(const struct streaminfo *a_stream, struct maat *feather, struct tunnel_endpoint **client_endpoint, struct tunnel_endpoint **server_endpoint, long long *endpoint_id_array, int endpoint_id_array_num)
|
||
long long endpoint_id_array[128] = {0};
|
||
int endpoint_id_array_num = 128;
|
||
struct streaminfo a_stream = {0};
|
||
struct stream_tuple4_v4 tuple4_v4 = {0};
|
||
struct tunnel_endpoint *client_endpoint = NULL, *server_endpoint = NULL;
|
||
tuple4_v4.daddr = inet_addr("192.50.128.1");
|
||
tuple4_v4.saddr = inet_addr("192.50.0.1");
|
||
a_stream.addr.addrtype = ADDR_TYPE_IPV4;
|
||
a_stream.addr.tuple4_v4 = &tuple4_v4;
|
||
|
||
int ret = tsg_get_endpoint_id(&a_stream, g_tsg_maat_feather, &client_endpoint, &server_endpoint, endpoint_id_array, endpoint_id_array_num);
|
||
EXPECT_EQ(4, ret);
|
||
EXPECT_EQ(989, endpoint_id_array[0]);
|
||
EXPECT_EQ(990, endpoint_id_array[1]);
|
||
EXPECT_EQ(989, endpoint_id_array[2]);
|
||
EXPECT_EQ(990, endpoint_id_array[3]);
|
||
|
||
EXPECT_EQ(990, client_endpoint->id);
|
||
EXPECT_EQ(990, server_endpoint->id);
|
||
}
|
||
|
||
extern int tsg_get_vlan_label_id(struct maat *feather, struct single_layer_vlan_addr *vlan_array, int vlan_array_num, long long *label_id_array, int label_id_array_num);
|
||
TEST(TMAPI, GetVlanLabelIdIllegalPara)
|
||
{
|
||
// int tsg_get_vlan_label_id(struct maat *feather, struct single_layer_vlan_addr *vlan_array, int vlan_array_num, long long *label_id_array, int label_id_array_num)
|
||
struct single_layer_vlan_addr vlan_array[2] = {0};
|
||
long long label_id_array[2] = {0};
|
||
EXPECT_EQ(0, tsg_get_vlan_label_id(g_tsg_maat_feather, vlan_array, 2, label_id_array, 0));
|
||
EXPECT_EQ(0, tsg_get_vlan_label_id(g_tsg_maat_feather, vlan_array, 2, NULL, 2));
|
||
EXPECT_EQ(0, tsg_get_vlan_label_id(g_tsg_maat_feather, vlan_array, 0, label_id_array, 2));
|
||
EXPECT_EQ(0, tsg_get_vlan_label_id(g_tsg_maat_feather, NULL, 2, label_id_array, 2));
|
||
}
|
||
|
||
TEST(TMAPI, GetVlanLabelIdFunction)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_TUNNEL_LABEL
|
||
label_id label_group_id label is_valid
|
||
"15560\t15560\tVLAN_ID\t1",
|
||
"15561\t15561\tVLAN_ID\t1"
|
||
**************************************************************************************************************************/
|
||
// int tsg_get_vlan_label_id(struct maat *feather, struct single_layer_vlan_addr *vlan_array, int vlan_array_num, long long *label_id_array, int label_id_array_num)
|
||
EXPECT_EQ(1, 1);
|
||
}
|
||
|
||
TEST(TMAPI, GetDNSProfileRecord)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_PROFILE_DNS_RECORDS
|
||
record_id record_name record_type record_values is_valid
|
||
"8119\teditTypeA\tA\t[{\"value\":\"1.1.1.1\",\"priority\":null},{\"value\":\"2.2.2.2\",\"priority\":null},{\"value\":\"3.3.3.3\",\"priority\":null}]\t1",
|
||
"7961\tFile\tAAAA\t[{\"value\":\"1030::C9B4:FF12:48AA:1A2B\",\"priority\":null},{\"value\":\"1030::C9B4:FF12:48AA:1A2C\",\"priority\":null}]\t1",
|
||
"7701\tTypeCNAME\tCNAME\t[{\"value\":\"www.facebook.com\",\"priority\":null},{\"value\":\"www.twitter.com\",\"priority\":null}]\t1"
|
||
**************************************************************************************************************************/
|
||
|
||
// void *matched_rule_cites_dns_profile_record(struct maat * feather, long long profile_id);
|
||
|
||
long long profile_id = 8119;
|
||
struct dns_profile_records *profile_records_right = (struct dns_profile_records *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_DNS_PROFILE_RECORD].id, (const char *)&profile_id, sizeof(long long));
|
||
struct dns_profile_records *profile_records = (struct dns_profile_records *)matched_rule_cites_dns_profile_record(g_tsg_maat_feather, profile_id);
|
||
EXPECT_EQ(profile_records_right, profile_records);
|
||
|
||
profile_records = NULL;
|
||
profile_records_right = NULL;
|
||
profile_id = 7961;
|
||
profile_records_right = (struct dns_profile_records *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_DNS_PROFILE_RECORD].id, (const char *)&profile_id, sizeof(long long));
|
||
profile_records = (struct dns_profile_records *)matched_rule_cites_dns_profile_record(g_tsg_maat_feather, profile_id);
|
||
EXPECT_TRUE(profile_records_right);
|
||
EXPECT_EQ(profile_records_right, profile_records);
|
||
|
||
profile_records = NULL;
|
||
profile_records_right = NULL;
|
||
profile_id = 7701;
|
||
profile_records_right = (struct dns_profile_records *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_DNS_PROFILE_RECORD].id, (const char *)&profile_id, sizeof(long long));
|
||
profile_records = (struct dns_profile_records *)matched_rule_cites_dns_profile_record(g_tsg_maat_feather, profile_id);
|
||
EXPECT_TRUE(profile_records_right);
|
||
EXPECT_EQ(profile_records_right, profile_records);
|
||
}
|
||
|
||
TEST(TMAPI, GetHttpResponsePagesProfile)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_PROFILE_RESPONSE_PAGES
|
||
profile_id profile_name format path is_valid
|
||
"957\ttest-html-1\thtml\tforeign_files/TSG_PROFILE_RESPONSE_PAGES.1\t1",
|
||
"958\ttest-html-2\thtml\tforeign_files/TSG_PROFILE_RESPONSE_PAGES.2\t1"
|
||
**************************************************************************************************************************/
|
||
|
||
// void *matched_rule_cites_http_response_pages(struct maat *feather, long long profile_id)
|
||
long long profile_id = 957;
|
||
struct http_response_pages *response_pages_right = (struct http_response_pages *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_RESPONSE_PAGES].id, (const char *)&profile_id, sizeof(long long));
|
||
struct http_response_pages *response_pages = (struct http_response_pages *)matched_rule_cites_http_response_pages(g_tsg_maat_feather, profile_id);
|
||
EXPECT_EQ(957, response_pages->profile_id);
|
||
EXPECT_EQ(response_pages_right, response_pages);
|
||
|
||
profile_id = 958;
|
||
response_pages_right = (struct http_response_pages *)maat_plugin_table_get_ex_data(g_tsg_maat_feather, g_tsg_maat_rt_para.plugin_tb[MAAT_PLUGIN_RESPONSE_PAGES].id, (const char *)&profile_id, sizeof(long long));
|
||
response_pages = (struct http_response_pages *)matched_rule_cites_http_response_pages(g_tsg_maat_feather, profile_id);
|
||
EXPECT_EQ(958, response_pages->profile_id);
|
||
EXPECT_EQ(response_pages_right, response_pages);
|
||
}
|
||
|
||
TEST(TMAPI, GetSessionRecordSwitch)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: T_VSYS_INFO
|
||
id is_enable_session_record is_valid
|
||
"6\t1\t1"
|
||
**************************************************************************************************************************/
|
||
EXPECT_EQ(1, tsg_session_record_switch_get());
|
||
}
|
||
|
||
TEST(TMAPI, NotifyPacketCaptureByPolicy)
|
||
{
|
||
/*************************************************************************************************************************
|
||
table name: TSG_PROFILE_TRAFFIC_MIRROR
|
||
profile_id profile_name vlan_ids is_valid
|
||
"845\t168.50.28yinyong\t[3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34]\t1",
|
||
"123\ttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest\t[66]\t1"
|
||
**************************************************************************************************************************/
|
||
// int session_packet_capture_by_rules_notify(const struct streaminfo *a_stream, struct maat_rule *rules, int n_rules, int thread_seq)
|
||
}
|
||
|
||
TEST(TMAPI, ScanNestingAddr)
|
||
{
|
||
// int tsg_scan_nesting_addr(const struct streaminfo *a_stream, enum TSG_PROTOCOL proto, struct maat_state **s_mid, struct maat_rule *results, size_t n_results)
|
||
}
|
||
|
||
TEST(TMAPI, ScanAppPropertiesPolicy)
|
||
{
|
||
// int tsg_scan_app_properties_policy(const struct streaminfo *a_stream, char *property, char *district, struct maat_state *s_mid, struct maat_rule *results, int n_results)
|
||
}
|
||
|
||
int main(int argc, char *argv[])
|
||
{
|
||
tsg_stat_create("./tsgconf/main.conf");
|
||
tsg_stat_init();
|
||
tsg_maat_rule_init("tsgconf/main.conf");
|
||
tsg_stat_start();
|
||
|
||
testing::InitGoogleTest(&argc, argv);
|
||
return RUN_ALL_TESTS();
|
||
}
|