修复Verify Policy中fqdn entry的命中路径错误,增加Verify Policy测试用例代码框架
This commit is contained in:
19
test/CMakeLists.txt
Normal file
19
test/CMakeLists.txt
Normal file
@@ -0,0 +1,19 @@
|
||||
cmake_minimum_required(VERSION 3.5)
|
||||
|
||||
include_directories(${CMAKE_SOURCE_DIR}/common/include)
|
||||
add_executable(verify_policy_test verify_policy_test.cpp ${CMAKE_SOURCE_DIR}/platform/src/verify_matcher.cpp)
|
||||
target_link_libraries(verify_policy_test common MESA_prof_load maatframe cjson gtest)
|
||||
|
||||
set(CMAKE_BUILD_DIR ${CMAKE_SOURCE_DIR}/build)
|
||||
set(TEST_RUN_DIR ${CMAKE_BUILD_DIR}/test)
|
||||
file(COPY ${CMAKE_SOURCE_DIR}/resource DESTINATION ${TEST_RUN_DIR})
|
||||
file(COPY ${CMAKE_SOURCE_DIR}/conf DESTINATION ${TEST_RUN_DIR})
|
||||
file(COPY ${CMAKE_SOURCE_DIR}/test/resource/VerifyPolicyHit.json DESTINATION ${TEST_RUN_DIR}/resource)
|
||||
file(COPY ${CMAKE_SOURCE_DIR}/test/resource/HitPolicyResult.json DESTINATION ${TEST_RUN_DIR}/resource)
|
||||
file(COPY ${CMAKE_SOURCE_DIR}/test/resource/HitPolicyRequest.json DESTINATION ${TEST_RUN_DIR}/resource)
|
||||
|
||||
add_test(NAME UPDATE_MAAT_MODE COMMAND bash -c "sed -i 's/maat_input_mode=1/maat_input_mode=0/' ${TEST_RUN_DIR}/conf/verify_policy.conf")
|
||||
add_test(NAME UPDATE_LOG_LEVEL COMMAND bash -c "sed -i 's/log_level=5/log_level=1/' ${TEST_RUN_DIR}/conf/verify_policy.conf")
|
||||
add_test(NAME UPDATE_VSYS_NUM COMMAND bash -c "sed -i '32i load_vsys_num=2' ${TEST_RUN_DIR}/conf/verify_policy.conf")
|
||||
add_test(NAME UPDATE_START_VSYS COMMAND bash -c "sed -i '33i load_start_vsys=1' ${TEST_RUN_DIR}/conf/verify_policy.conf")
|
||||
add_test(NAME UPDATE_JSON_FILE COMMAND bash -c "sed -i 's/json_cfg_file=\\.\\/resource\\/verify-policy\\.json/json_cfg_file=\\.\\/resource\\/VerifyPolicyHit\\.json/' ${TEST_RUN_DIR}/conf/verify_policy.conf")
|
||||
70
test/resource/HitPolicyRequest.json
Normal file
70
test/resource/HitPolicyRequest.json
Normal file
@@ -0,0 +1,70 @@
|
||||
{
|
||||
"Verify_Policy_Request": [
|
||||
{
|
||||
"vsys_id": 1,
|
||||
"verify_list": [
|
||||
{
|
||||
"type": "pxy_manipulation",
|
||||
"vsys_id": 1,
|
||||
"verify_session": {
|
||||
"attributes": [
|
||||
{
|
||||
"attribute_type": "ip",
|
||||
"table_name": "ATTR_SOURCE_IP",
|
||||
"attribute_name": "source",
|
||||
"attribute_value": {
|
||||
"ip": "192.168.0.1",
|
||||
"port":"8080",
|
||||
"addr_type": 4
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
],
|
||||
"verify_type": "policy"
|
||||
},
|
||||
{
|
||||
"vsys_id": 1,
|
||||
"verify_list": [
|
||||
{
|
||||
"type": "pxy_manipulation",
|
||||
"vsys_id": 1,
|
||||
"verify_session": {
|
||||
"attributes": [
|
||||
{
|
||||
"attribute_type": "ip",
|
||||
"table_name": "ATTR_SOURCE_IP",
|
||||
"attribute_name": "source",
|
||||
"attribute_value": {
|
||||
"ip": "192.168.1.1",
|
||||
"port":"8080",
|
||||
"addr_type": 4
|
||||
}
|
||||
},
|
||||
{
|
||||
"attribute_type": "ip",
|
||||
"table_name": "ATTR_INTERNAL_IP",
|
||||
"attribute_name": "internal",
|
||||
"attribute_value": {
|
||||
"ip": "192.168.1.2",
|
||||
"port": "80",
|
||||
"addr_type": 4
|
||||
}
|
||||
},
|
||||
{
|
||||
"attribute_type": "string",
|
||||
"table_name": "ATTR_SERVER_FQDN",
|
||||
"attribute_name": "server_fqdn",
|
||||
"attribute_value": {
|
||||
"string": "www.126.com"
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
],
|
||||
"verify_type": "policy"
|
||||
}
|
||||
]
|
||||
}
|
||||
134
test/resource/HitPolicyResult.json
Normal file
134
test/resource/HitPolicyResult.json
Normal file
@@ -0,0 +1,134 @@
|
||||
{
|
||||
"Verify_Policy_Result": [
|
||||
{
|
||||
"code": 200,
|
||||
"msg": "Success",
|
||||
"data": {
|
||||
"hitPolicyList": [
|
||||
{
|
||||
"id": 1021,
|
||||
"policyName": "",
|
||||
"is_execute_policy": true,
|
||||
"top_object_list": [
|
||||
{
|
||||
"object_id": 101,
|
||||
"table_name": "ATTR_SOURCE_IP",
|
||||
"not_flag": 0,
|
||||
"nth_clause": 0
|
||||
}
|
||||
]
|
||||
}
|
||||
],
|
||||
"verify_session": {
|
||||
"attributes": [
|
||||
{
|
||||
"attribute_type": "ip",
|
||||
"table_name": "ATTR_SOURCE_IP",
|
||||
"attribute_name": "source",
|
||||
"attribute_value": {
|
||||
"ip": "192.168.0.1",
|
||||
"port": "8080",
|
||||
"addr_type": 4
|
||||
},
|
||||
"hit_paths": [
|
||||
{
|
||||
"item_id": 1,
|
||||
"superior_object_id": 101
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
},
|
||||
"success": true
|
||||
},
|
||||
{
|
||||
"code": 200,
|
||||
"msg": "Success",
|
||||
"data": {
|
||||
"hitPolicyList": [
|
||||
{
|
||||
"id": 1022,
|
||||
"policyName": "",
|
||||
"is_execute_policy": true,
|
||||
"top_object_list": [
|
||||
{
|
||||
"tag_id": 11,
|
||||
"table_name": "ATTR_SOURCE_IP",
|
||||
"not_flag": 0,
|
||||
"nth_clause": 0
|
||||
},
|
||||
{
|
||||
"tag_id": 12,
|
||||
"table_name": "ATTR_INTERNAL_IP",
|
||||
"not_flag": 0,
|
||||
"nth_clause": 1
|
||||
},
|
||||
{
|
||||
"tag_id": 1,
|
||||
"table_name": "ATTR_SERVER_FQDN",
|
||||
"not_flag": 0,
|
||||
"nth_clause": 2
|
||||
}
|
||||
]
|
||||
}
|
||||
],
|
||||
"verify_session": {
|
||||
"attributes": [
|
||||
{
|
||||
"attribute_type": "ip",
|
||||
"table_name": "ATTR_SOURCE_IP",
|
||||
"attribute_name": "source",
|
||||
"attribute_value": {
|
||||
"ip": "192.168.1.1",
|
||||
"port": "8080",
|
||||
"addr_type": 4
|
||||
},
|
||||
"hit_paths": [
|
||||
{
|
||||
"entry_id": 1,
|
||||
"tag_id": 11
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"attribute_type": "ip",
|
||||
"table_name": "ATTR_INTERNAL_IP",
|
||||
"attribute_name": "internal",
|
||||
"attribute_value": {
|
||||
"ip": "192.168.1.2",
|
||||
"port": "80",
|
||||
"addr_type": 4
|
||||
},
|
||||
"hit_paths": [
|
||||
{
|
||||
"entry_id": 2,
|
||||
"tag_id": 12
|
||||
},
|
||||
{
|
||||
"entry_id": 2,
|
||||
"tag_id": 13
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"attribute_type": "string",
|
||||
"table_name": "ATTR_SERVER_FQDN",
|
||||
"attribute_name": "server_fqdn",
|
||||
"attribute_value": {
|
||||
"string": "www.126.com"
|
||||
},
|
||||
"hit_paths": [
|
||||
{
|
||||
"entry_id": 1,
|
||||
"tag_id": 1
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
},
|
||||
"success": true
|
||||
}
|
||||
]
|
||||
}
|
||||
101
test/resource/VerifyPolicyHit.json
Normal file
101
test/resource/VerifyPolicyHit.json
Normal file
@@ -0,0 +1,101 @@
|
||||
{
|
||||
"compile_table": "PXY_CTRL_COMPILE",
|
||||
"group2compile_table": "GROUP_PXY_CTRL_COMPILE_RELATION",
|
||||
"group2group_table": "GROUP_GROUP_RELATION",
|
||||
"rules": [
|
||||
{
|
||||
"compile_id": 1021,
|
||||
"service": 1,
|
||||
"action": 48,
|
||||
"do_blacklist": 1,
|
||||
"do_log": 1,
|
||||
"effective_range": 0,
|
||||
"tags":"anything",
|
||||
"user_region": "anything",
|
||||
"is_valid": "yes",
|
||||
"groups": [
|
||||
{
|
||||
"not_flag": 0,
|
||||
"group_id": 101,
|
||||
"group_name":"IPv4TCPSoureVeiryPolicy01",
|
||||
"virtual_table": "ATTR_SOURCE_IP",
|
||||
"regions": [
|
||||
{
|
||||
"table_type": "ip",
|
||||
"table_name": "TSG_OBJ_IP_ADDR",
|
||||
"table_content": {
|
||||
"addr_type": "ipv4",
|
||||
"addr_format": "range",
|
||||
"ip1": "192.168.0.1",
|
||||
"ip2": "192.168.0.1"
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"compile_id": 1022,
|
||||
"service": 1,
|
||||
"action": 48,
|
||||
"do_blacklist": 1,
|
||||
"do_log": 1,
|
||||
"effective_range": 0,
|
||||
"tags":"anything",
|
||||
"user_region": "anything",
|
||||
"is_valid": "yes",
|
||||
"groups": [
|
||||
{
|
||||
"group_id": 11,
|
||||
"group_name": "IPv4TCPSoureEntry.11",
|
||||
"virtual_table": "ATTR_SOURCE_IP"
|
||||
},
|
||||
{
|
||||
"group_id": 12,
|
||||
"group_name": "IPv4TCPSoureEntry.12",
|
||||
"virtual_table": "ATTR_INTERNAL_IP"
|
||||
},
|
||||
{
|
||||
"group_id": 1,
|
||||
"group_name": "FQDNEntry.1",
|
||||
"virtual_table": "ATTR_SERVER_FQDN"
|
||||
}
|
||||
]
|
||||
}
|
||||
],
|
||||
"plugin_table": [
|
||||
{
|
||||
"table_name": "FQDN_ENTRY",
|
||||
"table_content": [
|
||||
"1\t1\twww.126.com\t1\t1",
|
||||
"2\t2,3\twww.baidu.com\t1\t1",
|
||||
"4\t4,5,6\twww.qq.com\t1\t1"
|
||||
]
|
||||
},
|
||||
{
|
||||
"table_name": "IP_ADDR_ENTRY",
|
||||
"table_content": [
|
||||
"1\t11\t4\tsingle\t192.168.1.1\t192.168.1.1\t1",
|
||||
"2\t12,13\t4\tsingle\t192.168.1.2\t192.168.1.2\t1",
|
||||
"4\t14,15,16\t4\trange\t192.168.1.3\t192.168.1.3\t1"
|
||||
]
|
||||
},
|
||||
{
|
||||
"table_name": "LIBRARY_TAG",
|
||||
"table_content": [
|
||||
"1\tnone\twebsite_category\tfqdn1\tsearch\\bengines\t1",
|
||||
"2\tnone\twebsite_category\tfqdn2\tRecreation\band\bHobbies\t1",
|
||||
"3\tnone\twebsite_category\tfqdn3\tbusiness\t1",
|
||||
"4\tnone\twebsite_category\tfqdn4\tsearch bengines\t1",
|
||||
"5\tnone\twebsite_category\tfqdn5\tsearch\\bengines\t1",
|
||||
"6\tnone\twebsite_category\tfqdn6\tsearch\\bengines\t1",
|
||||
"11\tnone\tgeoip\tadministrative_area\tColombia.Departamento\bdel\bVaupes\t1",
|
||||
"12\tnone\tgeoip\tadministrative_area\tColombia.Departamento\bdel\bVaupes.Mitú\t1",
|
||||
"13\tnone\tgeoip\tadministrative_area\tColombia.Antioquia.Marinilla\t1",
|
||||
"14\tnone\tgeoip\tsuper_administrative_area\tColombia.Departamento\bdel\bVaupes\t1",
|
||||
"15\tnone\tgeoip\tadministrative_area\tGermany.Bavaria.Mauern\t1",
|
||||
"16\tnone\tgeoip\tadministrative_area\tGermany.Bavaria.Mellrichstadt\t1"
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
308
test/verify_policy_test.cpp
Normal file
308
test/verify_policy_test.cpp
Normal file
@@ -0,0 +1,308 @@
|
||||
/*************************************************************************
|
||||
> File Name:
|
||||
> Author:
|
||||
> Mail:
|
||||
> Created Time: 2020<32><30>05<30><35>28<32><38> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> 19ʱ21<32><31>37<33><37>
|
||||
************************************************************************/
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <unistd.h>
|
||||
#include <fcntl.h>
|
||||
#include <cjson/cJSON.h>
|
||||
#include <sys/stat.h>
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
#include <MESA/MESA_prof_load.h>
|
||||
|
||||
#include "verify_policy.h"
|
||||
#include "utils.h"
|
||||
|
||||
cJSON *verify_policy_result;
|
||||
cJSON *verify_policy_request;
|
||||
|
||||
struct verify_policy * g_verify_proxy = NULL;
|
||||
extern cJSON *get_library_search_query(const char *data, ssize_t data_len);
|
||||
extern cJSON *get_verify_policy_query(const char *data, ssize_t data_len, int thread_id);
|
||||
|
||||
int load_json_file_system_cmd(const char *load_json_file, const char *run_json_file)
|
||||
{
|
||||
char command[1024] = {0};
|
||||
snprintf(command, sizeof(command), "cp ./resource/%s ./resource/%s", load_json_file, run_json_file);
|
||||
system(command);
|
||||
sleep(2);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static char *select_hit_policy_result_item(int gtest_id)
|
||||
{
|
||||
if(verify_policy_result == NULL || verify_policy_result->type!=cJSON_Array)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
int foreach=0;
|
||||
char *hit_policy_result = NULL;
|
||||
cJSON *subitem = NULL;
|
||||
|
||||
for (subitem = verify_policy_result->child; subitem != NULL; subitem = subitem->next)
|
||||
{
|
||||
if(foreach == gtest_id)
|
||||
{
|
||||
hit_policy_result = cJSON_PrintUnformatted(subitem);
|
||||
break;
|
||||
}
|
||||
foreach++;
|
||||
}
|
||||
return hit_policy_result;
|
||||
}
|
||||
|
||||
static char *select_hit_policy_request_item(int gtest_id)
|
||||
{
|
||||
if(verify_policy_request == NULL || verify_policy_request->type!=cJSON_Array)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
int foreach=0;
|
||||
char *hit_policy_request = NULL;
|
||||
cJSON *subitem = NULL;
|
||||
|
||||
for (subitem = verify_policy_request->child; subitem != NULL; subitem = subitem->next)
|
||||
{
|
||||
if(foreach == gtest_id)
|
||||
{
|
||||
hit_policy_request = cJSON_PrintUnformatted(subitem);
|
||||
break;
|
||||
}
|
||||
foreach++;
|
||||
}
|
||||
return hit_policy_request;
|
||||
}
|
||||
|
||||
TEST(LibrarySearch, HitFqdnEntry)
|
||||
{
|
||||
const char *cm_http_request = "{\"ip\":null,\"fqdn\":\"www.126.com\",\"vsys_id\":1}";
|
||||
const char *expect_result="{\"code\":200,\"msg\":\"Success\",\"data\":{\"hit_library\":[{\"entry_id\":1,\"tag_ids\":\"1\"}]},\"success\":true}";
|
||||
|
||||
cJSON *result_json = get_library_search_query(cm_http_request, strlen(cm_http_request));
|
||||
ASSERT_TRUE(result_json != NULL);
|
||||
|
||||
char *hit_policy_list = cJSON_PrintUnformatted(result_json);
|
||||
ASSERT_TRUE(hit_policy_list != NULL);
|
||||
|
||||
int equal = strncasecmp(hit_policy_list, expect_result, strlen(expect_result));
|
||||
EXPECT_EQ(equal, 0);
|
||||
|
||||
cJSON_Delete(result_json);
|
||||
FREE(&hit_policy_list);
|
||||
|
||||
cm_http_request = "{\"ip\":null,\"fqdn\":\"www.baidu.com\",\"vsys_id\":1}";
|
||||
expect_result="{\"code\":200,\"msg\":\"Success\",\"data\":{\"hit_library\":[{\"entry_id\":2,\"tag_ids\":\"2,3\"}]},\"success\":true}";
|
||||
|
||||
result_json = get_library_search_query(cm_http_request, strlen(cm_http_request));
|
||||
ASSERT_TRUE(result_json != NULL);
|
||||
|
||||
hit_policy_list = cJSON_PrintUnformatted(result_json);
|
||||
ASSERT_TRUE(hit_policy_list != NULL);
|
||||
|
||||
equal = strncasecmp(hit_policy_list, expect_result, strlen(expect_result));
|
||||
EXPECT_EQ(equal, 0);
|
||||
|
||||
cJSON_Delete(result_json);
|
||||
FREE(&hit_policy_list);
|
||||
|
||||
cm_http_request = "{\"ip\":null,\"fqdn\":\"www.qq.com\",\"vsys_id\":1}";
|
||||
expect_result="{\"code\":200,\"msg\":\"Success\",\"data\":{\"hit_library\":[{\"entry_id\":4,\"tag_ids\":\"4,5,6\"}]},\"success\":true}";
|
||||
|
||||
result_json = get_library_search_query(cm_http_request, strlen(cm_http_request));
|
||||
ASSERT_TRUE(result_json != NULL);
|
||||
|
||||
hit_policy_list = cJSON_PrintUnformatted(result_json);
|
||||
ASSERT_TRUE(hit_policy_list != NULL);
|
||||
|
||||
equal = strncasecmp(hit_policy_list, expect_result, strlen(expect_result));
|
||||
EXPECT_EQ(equal, 0);
|
||||
|
||||
cJSON_Delete(result_json);
|
||||
FREE(&hit_policy_list);
|
||||
}
|
||||
|
||||
TEST(LibrarySearch, HitIpEntry)
|
||||
{
|
||||
const char *cm_http_request = "{\"ip\":\"192.168.1.1\",\"fqdn\":null,\"vsys_id\":1}";
|
||||
const char *expect_result="{\"code\":200,\"msg\":\"Success\",\"data\":{\"hit_library\":[{\"entry_id\":1,\"tag_ids\":\"11\"}]},\"success\":true}";
|
||||
|
||||
cJSON *result_json = get_library_search_query(cm_http_request, strlen(cm_http_request));
|
||||
ASSERT_TRUE(result_json != NULL);
|
||||
|
||||
char *hit_policy_list = cJSON_PrintUnformatted(result_json);
|
||||
ASSERT_TRUE(hit_policy_list != NULL);
|
||||
|
||||
int equal = strncasecmp(hit_policy_list, expect_result, strlen(expect_result));
|
||||
EXPECT_EQ(equal, 0);
|
||||
|
||||
cJSON_Delete(result_json);
|
||||
FREE(&hit_policy_list);
|
||||
|
||||
cm_http_request ="{\"ip\":\"192.168.1.2\",\"fqdn\":null,\"vsys_id\":1}";
|
||||
expect_result="{\"code\":200,\"msg\":\"Success\",\"data\":{\"hit_library\":[{\"entry_id\":2,\"tag_ids\":\"12,13\"}]},\"success\":true}";
|
||||
|
||||
result_json = get_library_search_query(cm_http_request, strlen(cm_http_request));
|
||||
ASSERT_TRUE(result_json != NULL);
|
||||
|
||||
hit_policy_list = cJSON_PrintUnformatted(result_json);
|
||||
ASSERT_TRUE(hit_policy_list != NULL);
|
||||
|
||||
equal = strncasecmp(hit_policy_list, expect_result, strlen(expect_result));
|
||||
EXPECT_EQ(equal, 0);
|
||||
|
||||
cJSON_Delete(result_json);
|
||||
FREE(&hit_policy_list);
|
||||
|
||||
cm_http_request = "{\"ip\":\"192.168.1.3\",\"fqdn\":null,\"vsys_id\":1}";
|
||||
expect_result="{\"code\":200,\"msg\":\"Success\",\"data\":{\"hit_library\":[{\"entry_id\":4,\"tag_ids\":\"14,15,16\"}]},\"success\":true}";
|
||||
|
||||
result_json = get_library_search_query(cm_http_request, strlen(cm_http_request));
|
||||
ASSERT_TRUE(result_json != NULL);
|
||||
|
||||
hit_policy_list = cJSON_PrintUnformatted(result_json);
|
||||
ASSERT_TRUE(hit_policy_list != NULL);
|
||||
|
||||
equal = strncasecmp(hit_policy_list, expect_result, strlen(expect_result));
|
||||
EXPECT_EQ(equal, 0);
|
||||
|
||||
cJSON_Delete(result_json);
|
||||
FREE(&hit_policy_list);
|
||||
}
|
||||
|
||||
TEST(VerifyPolicy, HitIpPolicy)
|
||||
{
|
||||
char *hit_policy_request = select_hit_policy_request_item(0);
|
||||
ASSERT_TRUE(hit_policy_request != NULL);
|
||||
char *hit_policy_result = select_hit_policy_result_item(0);
|
||||
ASSERT_TRUE(hit_policy_result != NULL);
|
||||
|
||||
cJSON *result_json = get_verify_policy_query(hit_policy_request, strlen(hit_policy_request), 1);
|
||||
ASSERT_TRUE(result_json != NULL);
|
||||
|
||||
char *hit_policy_query = cJSON_PrintUnformatted(result_json);
|
||||
ASSERT_TRUE(hit_policy_query != NULL);
|
||||
|
||||
int equal = strncasecmp(hit_policy_query, hit_policy_result, strlen(hit_policy_result));
|
||||
EXPECT_EQ(equal, 0);
|
||||
|
||||
cJSON_Delete(result_json);
|
||||
FREE(&hit_policy_query);
|
||||
FREE(&hit_policy_request);
|
||||
FREE(&hit_policy_result);
|
||||
}
|
||||
|
||||
TEST(VerifyPolicy, HitLibraryPolicy)
|
||||
{
|
||||
char *hit_policy_request = select_hit_policy_request_item(1);
|
||||
ASSERT_TRUE(hit_policy_request != NULL);
|
||||
char *hit_policy_result = select_hit_policy_result_item(1);
|
||||
ASSERT_TRUE(hit_policy_result != NULL);
|
||||
|
||||
cJSON *result_json = get_verify_policy_query(hit_policy_request, strlen(hit_policy_request), 1);
|
||||
ASSERT_TRUE(result_json != NULL);
|
||||
|
||||
char *hit_policy_query = cJSON_PrintUnformatted(result_json);
|
||||
ASSERT_TRUE(hit_policy_query != NULL);
|
||||
//printf("hit_policy_query =%s\n", hit_policy_query);
|
||||
|
||||
int equal = strncasecmp(hit_policy_query, hit_policy_result, strlen(hit_policy_result));
|
||||
EXPECT_EQ(equal, 0);
|
||||
|
||||
cJSON_Delete(result_json);
|
||||
FREE(&hit_policy_query);
|
||||
FREE(&hit_policy_request);
|
||||
FREE(&hit_policy_result);
|
||||
}
|
||||
|
||||
static char *read_json_file(const char *filename, size_t *input_sz)
|
||||
{
|
||||
FILE* fp=NULL;
|
||||
struct stat file_info;
|
||||
stat(filename, &file_info);
|
||||
*input_sz=file_info.st_size;
|
||||
|
||||
fp=fopen(filename,"r");
|
||||
if(fp==NULL)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
char* input=(char*)malloc(*input_sz);
|
||||
fread(input,1,*input_sz,fp);
|
||||
fclose(fp);
|
||||
return input;
|
||||
}
|
||||
|
||||
cJSON *load_verify_policy_result_by_file(const char *filename, int hit_policy_result)
|
||||
{
|
||||
size_t input_sz = 0;
|
||||
char *input = read_json_file(filename, &input_sz);
|
||||
if(input == NULL)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
cJSON* data = cJSON_Parse(input);
|
||||
if(data == NULL)
|
||||
{
|
||||
FREE(&input);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if(hit_policy_result)
|
||||
{
|
||||
verify_policy_result = cJSON_GetObjectItem(data,"Verify_Policy_Result");
|
||||
}
|
||||
else
|
||||
{
|
||||
verify_policy_request = cJSON_GetObjectItem(data,"Verify_Policy_Request");
|
||||
}
|
||||
|
||||
FREE(&input);
|
||||
return data;
|
||||
}
|
||||
|
||||
int main(int argc, char ** argv)
|
||||
{
|
||||
int ret = 0;
|
||||
int log_level=0;
|
||||
const char * main_profile = "./conf/verify_policy.conf";
|
||||
|
||||
g_verify_proxy = ALLOC(struct verify_policy, 1);
|
||||
assert(g_verify_proxy);
|
||||
strcpy(g_verify_proxy->name, "verify_policy");
|
||||
|
||||
const char *log_path="./logs/verify_policy.log";
|
||||
MESA_load_profile_int_def(main_profile, "SYSTEM", "log_level", &log_level, LOG_FATAL);
|
||||
g_verify_proxy->logger = log_handle_create(log_path, log_level);
|
||||
CHECK_OR_EXIT(g_verify_proxy->logger != NULL, "Failed at init log module. Exit.");
|
||||
|
||||
ret = maat_table_init(g_verify_proxy, main_profile);
|
||||
CHECK_OR_EXIT(ret == 0, "Failed at init maat module, Exit.");
|
||||
|
||||
const char *filename1 = "./resource/HitPolicyResult.json";
|
||||
cJSON *data1=load_verify_policy_result_by_file(filename1, 1);
|
||||
const char *filename2 = "./resource/HitPolicyRequest.json";
|
||||
cJSON *data2=load_verify_policy_result_by_file(filename2, 0);
|
||||
|
||||
testing::InitGoogleTest(&argc, argv);
|
||||
ret=RUN_ALL_TESTS();
|
||||
|
||||
if(data1 != NULL)
|
||||
{
|
||||
cJSON_Delete(data1);
|
||||
}
|
||||
if(data2 != NULL)
|
||||
{
|
||||
cJSON_Delete(data2);
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
185
test/verify_policy_tool.cpp
Normal file
185
test/verify_policy_tool.cpp
Normal file
@@ -0,0 +1,185 @@
|
||||
/*************************************************************************
|
||||
> File Name:
|
||||
> Author:
|
||||
> Mail:
|
||||
> Created Time: 2020<32><30>05<30><35>28<32><38> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> 19ʱ21<32><31>37<33><37>
|
||||
************************************************************************/
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <unistd.h>
|
||||
#include <fcntl.h>
|
||||
#include <cjson/cJSON.h>
|
||||
#include <sys/stat.h>
|
||||
#include "utils.h"
|
||||
|
||||
enum curl_post_type
|
||||
{
|
||||
CURL_POST_HTTP,
|
||||
CURL_POST_SSL,
|
||||
__CURL_POST_MAX
|
||||
};
|
||||
|
||||
struct curl_post_data
|
||||
{
|
||||
char *input_result_data;
|
||||
char *curl_post_file;
|
||||
enum curl_post_type protocol_type;
|
||||
};
|
||||
|
||||
char *curl_read_file(char *result_json_file)
|
||||
{
|
||||
FILE* fp=NULL;
|
||||
struct stat file_info;
|
||||
stat(result_json_file, &file_info);
|
||||
size_t input_sz=file_info.st_size;
|
||||
|
||||
fp=fopen(result_json_file,"r");
|
||||
if(fp==NULL)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
char* input=(char*)malloc(input_sz);
|
||||
fread(input,1,input_sz,fp);
|
||||
fclose(fp);
|
||||
|
||||
return input;
|
||||
}
|
||||
|
||||
int curl_post_system_cmd(char *result, size_t result_len, char *curl_post_json)
|
||||
{
|
||||
int line_num=0;
|
||||
char command[1024] = {0};
|
||||
char *cmd_str=NULL;
|
||||
|
||||
snprintf(command, sizeof(command), "curl -X POST -H 'Content-Type: application/json' http://127.0.0.1:9994/v1/policy/verify -d@%s", curl_post_json);
|
||||
|
||||
char *p=result;
|
||||
char line[2048] = {0};
|
||||
FILE *fp = NULL;
|
||||
|
||||
if((fp = popen(command, "r")) == NULL)
|
||||
{
|
||||
printf("popen error!\n");
|
||||
return 0;
|
||||
}
|
||||
memset(result, 0, result_len);
|
||||
while (fgets(line, sizeof(line), fp))
|
||||
{
|
||||
if((p - result) < (int)result_len)
|
||||
{
|
||||
if(line_num)
|
||||
{
|
||||
p += snprintf(p, result_len - (p - result), ",");
|
||||
}
|
||||
|
||||
p += snprintf(p, result_len - (p - result), "%s", line);
|
||||
}
|
||||
line_num++;
|
||||
}
|
||||
pclose(fp);
|
||||
|
||||
free(cmd_str);
|
||||
return 1;
|
||||
}
|
||||
|
||||
int curl_exec_expect_result(char *result, char *input_json_file)
|
||||
{
|
||||
int ret=-1;
|
||||
cJSON *result_json=NULL, *input_json=NULL;
|
||||
cJSON_bool successful = false;
|
||||
|
||||
result_json = cJSON_Parse(result);
|
||||
if(!result_json)
|
||||
{
|
||||
goto finish;
|
||||
}
|
||||
input_json = cJSON_Parse(input_json_file);
|
||||
if(!input_json)
|
||||
{
|
||||
goto finish;
|
||||
}
|
||||
successful = cJSON_Compare(result_json, result_json, true);
|
||||
if(successful)
|
||||
{
|
||||
ret=0;
|
||||
}
|
||||
finish:
|
||||
cJSON_Delete(result_json);
|
||||
cJSON_Delete(input_json);
|
||||
return ret;
|
||||
}
|
||||
|
||||
int call_curl_post(struct curl_post_data *post_data)
|
||||
{
|
||||
int ret=-1;
|
||||
char result[81920]={0};
|
||||
|
||||
curl_post_system_cmd(result, sizeof(result), post_data->curl_post_file);
|
||||
|
||||
switch(post_data->protocol_type)
|
||||
{
|
||||
case CURL_POST_HTTP:
|
||||
ret = curl_exec_expect_result(result, post_data->input_result_data);
|
||||
break;
|
||||
case CURL_POST_SSL:
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
static void help()
|
||||
{
|
||||
fprintf(stderr,
|
||||
"verify_policy_test <-j| -t | -c > arg\n"
|
||||
"Usage:\n"
|
||||
" -j <post json data>\n"
|
||||
" -t <protocol type>\n"
|
||||
" -c <input json>\n");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
int main(int argc, char ** argv)
|
||||
{
|
||||
int i=0, ret = 0;
|
||||
int curl_post_type=0;
|
||||
char result_json_file[256]={0};
|
||||
char curl_post_file[256]={0};
|
||||
|
||||
if(argc < 3) {help();}
|
||||
for (i = 1; i < argc; i++)
|
||||
{
|
||||
int lastarg = i==argc-1;
|
||||
if (!strcmp(argv[i], "-j") && !lastarg)
|
||||
{
|
||||
strncpy(curl_post_file, argv[++i], sizeof(curl_post_file));
|
||||
}
|
||||
else if (!strcmp(argv[i], "-t") && !lastarg)
|
||||
{
|
||||
sscanf(argv[++i], "%u", &curl_post_type);
|
||||
}
|
||||
else if (!strcmp(argv[i], "-c") && !lastarg)
|
||||
{
|
||||
strncpy(result_json_file, argv[++i], sizeof(result_json_file));
|
||||
}
|
||||
else
|
||||
{
|
||||
help();
|
||||
}
|
||||
}
|
||||
struct curl_post_data *post_data=NULL;
|
||||
post_data = ALLOC(struct curl_post_data, 1);
|
||||
post_data->protocol_type=(enum curl_post_type)curl_post_type;
|
||||
post_data->input_result_data=curl_read_file(result_json_file);
|
||||
post_data->curl_post_file=curl_post_file;
|
||||
|
||||
ret = call_curl_post(post_data);
|
||||
|
||||
return ret;
|
||||
}
|
||||
Reference in New Issue
Block a user