2019-10-22 15:13:14 +08:00
|
|
|
/*************************************************************************
|
|
|
|
|
> File Name: verify-policy.cpp
|
2020-01-09 14:32:00 +08:00
|
|
|
> Author:
|
|
|
|
|
> Mail:
|
2019-10-22 15:13:14 +08:00
|
|
|
> Created Time: 2019年08月23日 星期五 14时41分17秒
|
|
|
|
|
************************************************************************/
|
|
|
|
|
|
|
|
|
|
#include<iostream>
|
|
|
|
|
#include <stdio.h>
|
|
|
|
|
#include <stdlib.h>
|
|
|
|
|
#include <string.h>
|
|
|
|
|
#include <unistd.h>
|
|
|
|
|
#include <fcntl.h>
|
|
|
|
|
|
|
|
|
|
#include <event2/listener.h>
|
|
|
|
|
#include <event2/http.h>
|
|
|
|
|
#include <cjson/cJSON.h>
|
|
|
|
|
#include <event2/buffer.h>
|
|
|
|
|
|
2020-01-17 10:59:34 +08:00
|
|
|
#include <sys/socket.h>//inet_addr
|
|
|
|
|
#include <netinet/in.h>//inet_addr
|
|
|
|
|
#include <arpa/inet.h>//inet_addr
|
|
|
|
|
#include <MESA/stream.h>
|
|
|
|
|
|
2019-10-22 15:13:14 +08:00
|
|
|
#include "verify_policy.h"
|
2020-06-22 15:15:52 +08:00
|
|
|
#include <MESA/MESA_prof_load.h>
|
|
|
|
|
#include <MESA/MESA_handle_logger.h>
|
2020-01-17 10:59:34 +08:00
|
|
|
#include "verify_policy_utils.h"
|
2019-10-22 15:13:14 +08:00
|
|
|
#include "verify_policy_logging.h"
|
|
|
|
|
|
2020-02-18 17:54:15 +08:00
|
|
|
struct verify_policy * g_verify_proxy = NULL;
|
2019-10-22 15:13:14 +08:00
|
|
|
|
2020-01-09 14:32:00 +08:00
|
|
|
/* VERSION STRING */
|
|
|
|
|
#ifdef TARGET_GIT_VERSION
|
|
|
|
|
static __attribute__((__used__)) const char * git_ver = TARGET_GIT_VERSION;
|
2019-10-22 15:13:14 +08:00
|
|
|
#else
|
2020-01-09 14:32:00 +08:00
|
|
|
static __attribute__((__used__)) const char * git_ver = "1.1";
|
2019-10-22 15:13:14 +08:00
|
|
|
#endif
|
|
|
|
|
const char * version()
|
|
|
|
|
{
|
2020-01-09 14:32:00 +08:00
|
|
|
return git_ver;
|
2019-10-22 15:13:14 +08:00
|
|
|
}
|
|
|
|
|
|
2020-02-18 17:54:15 +08:00
|
|
|
extern int pangu_policy_init(struct verify_policy * verify, const char* profile_path);
|
2019-10-22 15:13:14 +08:00
|
|
|
|
2020-02-18 17:54:15 +08:00
|
|
|
static int verify_policy_init(struct verify_policy * verify, const char *profile)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
|
|
|
|
int xret = -1;
|
2020-01-09 14:32:00 +08:00
|
|
|
|
2019-10-22 15:13:14 +08:00
|
|
|
xret = MESA_load_profile_uint_nodef(profile, "CONFIG", "thread-nu", &(verify->nr_work_threads));
|
|
|
|
|
if (xret < 0){
|
|
|
|
|
mesa_runtime_log(RLOG_LV_FATAL, MODULE_NAME, "Reading the number of running threads failed");
|
|
|
|
|
}
|
|
|
|
|
xret = MESA_load_profile_short_nodef(profile, "LISTEN", "port", (short *)&(verify->listen_port));
|
|
|
|
|
if (xret < 0){
|
|
|
|
|
mesa_runtime_log(RLOG_LV_FATAL, MODULE_NAME, "Listen Port invalid");
|
|
|
|
|
}
|
2020-01-09 14:32:00 +08:00
|
|
|
mesa_runtime_log(RLOG_LV_INFO, MODULE_NAME, "%s:%d", "The Threads", verify->nr_work_threads);
|
|
|
|
|
mesa_runtime_log(RLOG_LV_INFO, MODULE_NAME, "%s:%d", "Libevent Port", verify->listen_port);
|
2019-10-22 15:13:14 +08:00
|
|
|
return xret;
|
|
|
|
|
}
|
|
|
|
|
|
2020-02-18 17:54:15 +08:00
|
|
|
enum verify_policy_type tsg_policy_type_str2idx(const char *action_str)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
2020-01-17 10:59:34 +08:00
|
|
|
const char * policy_name[__SCAN_POLICY_MAX];
|
|
|
|
|
policy_name[PXY_TABLE_SECURITY] = "tsg_security";
|
2020-04-15 19:06:31 +08:00
|
|
|
policy_name[PXY_TABLE_MANIPULATION] = "pxy_manipulation";
|
2020-01-17 10:59:34 +08:00
|
|
|
policy_name[PXY_TABLE_DEFENCE] = "active_defence";
|
2020-01-09 14:32:00 +08:00
|
|
|
|
2019-10-22 15:13:14 +08:00
|
|
|
size_t i = 0;
|
|
|
|
|
|
2020-01-17 10:59:34 +08:00
|
|
|
for (i = 0; i < sizeof(policy_name) / sizeof(const char *); i++)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
2020-01-17 10:59:34 +08:00
|
|
|
if (0 == strcasecmp(action_str, policy_name[i]))
|
2019-10-22 15:13:14 +08:00
|
|
|
break;
|
|
|
|
|
}
|
2020-01-17 18:57:51 +08:00
|
|
|
mesa_runtime_log(RLOG_LV_DEBUG, MODULE_NAME, "[I] policyType= %s", action_str);
|
2020-02-18 17:54:15 +08:00
|
|
|
return (enum verify_policy_type)i;
|
2019-10-22 15:13:14 +08:00
|
|
|
}
|
|
|
|
|
|
2020-02-18 17:54:15 +08:00
|
|
|
int protoco_field_type_str2idx(enum verify_policy_type type, const char *action_str, char *buff, char **p)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
2020-01-17 10:59:34 +08:00
|
|
|
const char * table_name[__SECURITY_TABLE_MAX] ={0};
|
2019-10-22 15:13:14 +08:00
|
|
|
|
2020-01-17 10:59:34 +08:00
|
|
|
switch(type)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
2020-01-17 10:59:34 +08:00
|
|
|
case PXY_TABLE_MANIPULATION:
|
|
|
|
|
table_name[PXY_CTRL_IP] = "TSG_OBJ_IP_ADDR";
|
|
|
|
|
table_name[PXY_CTRL_HTTP_URL] = "TSG_FIELD_HTTP_URL";
|
|
|
|
|
table_name[PXY_CTRL_HTTP_FQDN] = "TSG_FIELD_HTTP_HOST";
|
|
|
|
|
table_name[PXY_CTRL_HTTP_REQ_HDR] = "TSG_FIELD_HTTP_REQ_HDR";
|
|
|
|
|
table_name[PXY_CTRL_HTTP_REQ_BODY] = "TSG_FIELD_HTTP_REQ_CONTENT";
|
|
|
|
|
table_name[PXY_CTRL_HTTP_RES_HDR] = "TSG_FIELD_HTTP_RES_HDR";
|
|
|
|
|
table_name[PXY_CTRL_HTTP_RES_BODY] = "TSG_FIELD_HTTP_RES_CONTENT";
|
|
|
|
|
table_name[PXY_CTRL_SUBSCRIBE_ID] = "TSG_OBJ_SUBSCRIBER_ID";
|
2020-04-01 14:29:24 +08:00
|
|
|
table_name[PXY_CTRL_APP_ID] = "TSG_OBJ_APP_ID";
|
2020-07-03 19:16:53 +08:00
|
|
|
table_name[PXY_CTRL_DOH_QNAME]="TSG_FIELD_DOH_QNAME";
|
|
|
|
|
table_name[PXY_CTRL_DOH_HOST]="TSG_FIELD_DOH_HOST";
|
2020-01-17 10:59:34 +08:00
|
|
|
break;
|
|
|
|
|
case PXY_TABLE_SECURITY:
|
|
|
|
|
table_name[PXY_SECURITY_IP] = "TSG_OBJ_IP_ADDR";
|
|
|
|
|
table_name[PXY_SECURITY_HTTP_URL] = "TSG_FIELD_HTTP_URL";
|
|
|
|
|
table_name[PXY_SECURITY_HTTP_FQDN] = "TSG_FIELD_HTTP_HOST";
|
|
|
|
|
table_name[PXY_SECURITY_HTTP_REQ_HDR] = "TSG_FIELD_HTTP_REQ_HDR";
|
|
|
|
|
table_name[PXY_SECURITY_HTTP_REQ_BODY] = "TSG_FIELD_HTTP_REQ_CONTENT";
|
|
|
|
|
table_name[PXY_SECURITY_HTTP_RES_HDR] = "TSG_FIELD_HTTP_RES_HDR";
|
|
|
|
|
table_name[PXY_SECURITY_HTTP_RES_BODY] = "TSG_FIELD_HTTP_RES_CONTENT";
|
|
|
|
|
table_name[PXY_SECURITY_SUBSCRIBE_ID] = "TSG_OBJ_SUBSCRIBER_ID";
|
|
|
|
|
table_name[PXY_SECURITY_HTTPS_SNI] = "TSG_FIELD_SSL_SNI";
|
|
|
|
|
table_name[PXY_SECURITY_HTTPS_CN] = "TSG_FIELD_SSL_CN";
|
|
|
|
|
table_name[PXY_SECURITY_HTTPS_SAN] = "TSG_FIELD_SSL_SAN";
|
|
|
|
|
table_name[PXY_SECURITY_DNS_QNAME] = "TSG_FIELD_DNS_QNAME";
|
|
|
|
|
table_name[PXY_SECURITY_MAIL_ACCOUNT] = "TSG_FIELD_MAIL_ACCOUNT";
|
|
|
|
|
table_name[PXY_SECURITY_MAIL_FROM] = "TSG_FIELD_MAIL_FROM";
|
|
|
|
|
table_name[PXY_SECURITY_MAIL_TO] = "TSG_FIELD_MAIL_TO";
|
|
|
|
|
table_name[PXY_SECURITY_MAIL_SUBJECT] = "TSG_FIELD_MAIL_SUBJECT";
|
|
|
|
|
table_name[PXY_SECURITY_MAIL_CONTENT] = "TSG_FIELD_MAIL_CONTENT";
|
|
|
|
|
table_name[PXY_SECURITY_MAIL_ATT_NAME] = "TSG_FIELD_MAIL_ATT_NAME";
|
|
|
|
|
table_name[PXY_SECURITY_MAIL_ATT_CONTENT] = "TSG_FIELD_MAIL_ATT_CONTENT";
|
|
|
|
|
table_name[PXY_SECURITY_FTP_URI] = "TSG_FIELD_FTP_URI";
|
|
|
|
|
table_name[PXY_SECURITY_FTP_CONTENT] = "TSG_FIELD_FTP_CONTENT";
|
2020-01-20 18:22:36 +08:00
|
|
|
table_name[PXY_SECURITY_FTP_ACCOUNT] = "TSG_FIELD_FTP_ACCOUNT";
|
2020-03-18 15:40:21 +08:00
|
|
|
table_name[PXY_SECURITY_APP_ID] = "TSG_OBJ_APP_ID";
|
2020-01-17 10:59:34 +08:00
|
|
|
break;
|
|
|
|
|
case PXY_TABLE_DEFENCE:
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
break;
|
2020-01-09 14:32:00 +08:00
|
|
|
}
|
2020-01-17 10:59:34 +08:00
|
|
|
size_t i = 0;
|
2020-01-09 14:32:00 +08:00
|
|
|
|
2020-01-17 10:59:34 +08:00
|
|
|
for (i = 0; i < sizeof(table_name) / sizeof(const char *); i++)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
2020-01-17 10:59:34 +08:00
|
|
|
if (0 == strcasecmp(action_str, table_name[i]))
|
|
|
|
|
break;
|
2019-10-22 15:13:14 +08:00
|
|
|
}
|
2020-02-18 17:54:15 +08:00
|
|
|
*p += snprintf(*p, sizeof(buff) - (*p - buff), ", protocolField=%s,%d",action_str, (int)i);
|
2020-01-17 10:59:34 +08:00
|
|
|
|
|
|
|
|
return i;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct ipaddr *ip_to_stream_addr(char *clientIp1, unsigned int clientPort1, char *serverIp1, unsigned int serverPort1, int addr_type)
|
|
|
|
|
{
|
|
|
|
|
struct ipaddr *ip_addr = ALLOC(struct ipaddr, 1);
|
|
|
|
|
if(addr_type == 4)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
2020-01-17 10:59:34 +08:00
|
|
|
struct stream_tuple4_v4 *v4_addr = ALLOC(struct stream_tuple4_v4, 1);
|
|
|
|
|
ip_addr->addrtype=ADDR_TYPE_IPV4;
|
|
|
|
|
inet_pton(AF_INET,clientIp1,&(v4_addr->saddr));
|
|
|
|
|
v4_addr->source=htons(clientPort1);
|
|
|
|
|
inet_pton(AF_INET,serverIp1,&(v4_addr->daddr));
|
|
|
|
|
v4_addr->dest=htons(serverPort1);
|
|
|
|
|
ip_addr->v4=v4_addr;
|
2019-10-22 15:13:14 +08:00
|
|
|
}
|
2020-01-17 10:59:34 +08:00
|
|
|
if(addr_type == 6)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
2020-01-17 10:59:34 +08:00
|
|
|
struct stream_tuple4_v6 *v6_addr = ALLOC(struct stream_tuple4_v6, 1);
|
|
|
|
|
ip_addr->addrtype=ADDR_TYPE_IPV6;
|
|
|
|
|
inet_pton(AF_INET6,clientIp1,&(v6_addr->saddr));
|
|
|
|
|
v6_addr->source=htons(clientPort1);
|
|
|
|
|
inet_pton(AF_INET6,serverIp1,&(v6_addr->daddr));
|
|
|
|
|
v6_addr->dest=htons(serverPort1);
|
|
|
|
|
ip_addr->v6=v6_addr;
|
2019-10-22 15:13:14 +08:00
|
|
|
}
|
2020-02-18 17:54:15 +08:00
|
|
|
mesa_runtime_log(RLOG_LV_DEBUG, MODULE_NAME, "[I] attributeName = ip, clientIp1=%s, clientPort1=%d, serverIp=%s, serverPort=%d, addr_type = %d",
|
2020-01-17 18:57:51 +08:00
|
|
|
clientIp1, clientPort1, serverIp1, serverPort1, addr_type);
|
|
|
|
|
|
2020-01-17 10:59:34 +08:00
|
|
|
return ip_addr;
|
|
|
|
|
}
|
|
|
|
|
|
2020-01-20 18:22:36 +08:00
|
|
|
void ipaddr_free(struct ipaddr *ip_addr)
|
|
|
|
|
{
|
|
|
|
|
if(ip_addr->addrtype==ADDR_TYPE_IPV4)
|
|
|
|
|
{
|
|
|
|
|
free(ip_addr->v4);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if(ip_addr->addrtype==ADDR_TYPE_IPV6)
|
|
|
|
|
{
|
|
|
|
|
free(ip_addr->v6);
|
|
|
|
|
}
|
|
|
|
|
free(ip_addr);
|
|
|
|
|
}
|
|
|
|
|
|
2020-04-15 19:06:31 +08:00
|
|
|
static struct ipaddr * get_ip_from_json(cJSON *attributeValue)
|
|
|
|
|
{
|
|
|
|
|
cJSON* item = NULL;
|
|
|
|
|
int addr_type=0, __attribute__((__unused__))protocol=0;
|
|
|
|
|
char *clientIp1=NULL,*serverIp1=NULL;
|
|
|
|
|
unsigned int clientPort1=0,serverPort1=0;
|
|
|
|
|
|
|
|
|
|
item = cJSON_GetObjectItem(attributeValue,"clientIp");
|
|
|
|
|
if(item && item->type==cJSON_String) clientIp1 = item->valuestring;
|
|
|
|
|
item = cJSON_GetObjectItem(attributeValue,"serverIp");
|
|
|
|
|
if(item && item->type==cJSON_String) serverIp1 = (item->valuestring);
|
|
|
|
|
item = cJSON_GetObjectItem(attributeValue,"clientPort");
|
|
|
|
|
if(item && item->type==cJSON_String) clientPort1 =atoi(item->valuestring);
|
|
|
|
|
item = cJSON_GetObjectItem(attributeValue,"serverPort");
|
|
|
|
|
if(item && item->type==cJSON_String) serverPort1 =atoi(item->valuestring);
|
|
|
|
|
item = cJSON_GetObjectItem(attributeValue,"protocol");
|
|
|
|
|
if(item && item->type==cJSON_Number) protocol = item->valueint;
|
|
|
|
|
item=cJSON_GetObjectItem(attributeValue,"addrType");
|
|
|
|
|
if(item && item->type==cJSON_Number) addr_type = item->valueint;
|
|
|
|
|
|
|
|
|
|
struct ipaddr *ip_addr = NULL;
|
|
|
|
|
ip_addr = ip_to_stream_addr(clientIp1, clientPort1, serverIp1, serverPort1, addr_type);
|
|
|
|
|
|
|
|
|
|
return ip_addr;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static int get_attribute_from_json(int curr_id, cJSON* subchild, struct verify_policy_query *policy_query)
|
|
|
|
|
{
|
|
|
|
|
int xret = -1;
|
|
|
|
|
char buff[VERIFY_STRING_MAX], *p = NULL;
|
|
|
|
|
cJSON* item = NULL, *attributeValue=NULL;
|
|
|
|
|
|
|
|
|
|
p = buff;
|
|
|
|
|
item = cJSON_GetObjectItem(subchild, "attributeName");
|
|
|
|
|
if(item && item->type==cJSON_String)
|
|
|
|
|
{
|
|
|
|
|
policy_query->verify_object[curr_id].attri_name = item->valuestring;
|
|
|
|
|
p += snprintf(p, sizeof(buff) - (p - buff), "attributeName = %s",policy_query->verify_object[curr_id].attri_name);
|
|
|
|
|
}
|
|
|
|
|
policy_query->verify_object[curr_id].attributes=cJSON_Duplicate(subchild, 1);
|
|
|
|
|
|
|
|
|
|
item = cJSON_GetObjectItem(subchild, "tableName");
|
|
|
|
|
if(item && item->type==cJSON_String)
|
|
|
|
|
{
|
|
|
|
|
policy_query->verify_object[curr_id].protocol_field = protoco_field_type_str2idx(policy_query->type, item->valuestring, buff, &p);
|
|
|
|
|
if ((policy_query->type == PXY_TABLE_MANIPULATION && policy_query->verify_object[curr_id].protocol_field == __SCAN_TABLE_MAX)
|
|
|
|
|
|| (policy_query->type == PXY_TABLE_SECURITY && policy_query->verify_object[curr_id].protocol_field == __SECURITY_TABLE_MAX))
|
|
|
|
|
{
|
|
|
|
|
mesa_runtime_log(RLOG_LV_FATAL, MODULE_NAME, "policy table name error, table name = %s", item->valuestring);
|
|
|
|
|
goto finish;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
attributeValue = cJSON_GetObjectItem(subchild, "attributeValue");
|
|
|
|
|
if(attributeValue == NULL || attributeValue->type!=cJSON_Object)
|
|
|
|
|
{
|
|
|
|
|
goto finish;
|
|
|
|
|
}
|
|
|
|
|
if(0 == strcasecmp(policy_query->verify_object[curr_id].attri_name, "ip"))
|
|
|
|
|
{
|
|
|
|
|
policy_query->verify_object[curr_id].ip_addr = get_ip_from_json(attributeValue);
|
|
|
|
|
goto end;
|
|
|
|
|
}
|
|
|
|
|
item = cJSON_GetObjectItem(attributeValue,"string");
|
|
|
|
|
{
|
|
|
|
|
policy_query->verify_object[curr_id].keyword = item->valuestring;
|
|
|
|
|
p += snprintf(p, sizeof(buff) - (p - buff), ", content = %s",policy_query->verify_object[curr_id].keyword);
|
|
|
|
|
}
|
|
|
|
|
mesa_runtime_log(RLOG_LV_DEBUG, MODULE_NAME, "[I] %s", buff);
|
|
|
|
|
memset(buff, 0, VERIFY_STRING_MAX);
|
|
|
|
|
end:
|
|
|
|
|
xret = 1;
|
|
|
|
|
finish:
|
|
|
|
|
return xret;
|
|
|
|
|
}
|
|
|
|
|
|
2020-01-17 10:59:34 +08:00
|
|
|
cJSON *get_query_from_request(const char *data, int thread_id)
|
|
|
|
|
{
|
2020-01-17 18:57:51 +08:00
|
|
|
int i = 0;
|
2020-03-06 16:02:42 +08:00
|
|
|
int hit_cnt = -1;
|
2020-04-15 19:06:31 +08:00
|
|
|
struct verify_policy_query *verify_policy = NULL;
|
2020-01-17 10:59:34 +08:00
|
|
|
|
|
|
|
|
cJSON* data_json = cJSON_Parse(data);
|
|
|
|
|
if(data_json == NULL)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
2020-01-17 10:59:34 +08:00
|
|
|
mesa_runtime_log(RLOG_LV_FATAL, MODULE_NAME, "invalid policy parameter");
|
|
|
|
|
return NULL;
|
2019-10-22 15:13:14 +08:00
|
|
|
}
|
2020-02-18 17:54:15 +08:00
|
|
|
cJSON *policy_obj=NULL, *data_obj=NULL;
|
2020-01-17 10:59:34 +08:00
|
|
|
policy_obj=cJSON_CreateObject();
|
|
|
|
|
|
|
|
|
|
cJSON_AddNumberToObject(policy_obj, "code", 200);
|
|
|
|
|
cJSON_AddStringToObject(policy_obj, "msg", "Success");
|
|
|
|
|
|
|
|
|
|
data_obj = cJSON_CreateObject();
|
|
|
|
|
cJSON_AddItemToObject(policy_obj, "data", data_obj);
|
|
|
|
|
|
2020-04-15 19:06:31 +08:00
|
|
|
cJSON* item = NULL, *subitem = NULL, *subchild = NULL, *attributes=NULL;
|
2020-01-17 10:59:34 +08:00
|
|
|
item = cJSON_GetObjectItem(data_json,"verifyList");
|
2019-10-22 15:13:14 +08:00
|
|
|
if(item && item->type==cJSON_Array)
|
|
|
|
|
{
|
|
|
|
|
for (subitem = item->child; subitem != NULL; subitem = subitem->next)
|
|
|
|
|
{
|
2020-04-15 19:06:31 +08:00
|
|
|
verify_policy = ALLOC(struct verify_policy_query, 1);
|
2020-01-17 10:59:34 +08:00
|
|
|
item = cJSON_GetObjectItem(subitem,"policyType");
|
2019-10-22 15:13:14 +08:00
|
|
|
if(item && item->type==cJSON_String)
|
|
|
|
|
{
|
2020-04-15 19:06:31 +08:00
|
|
|
verify_policy->type = tsg_policy_type_str2idx(item->valuestring);
|
|
|
|
|
if (verify_policy->type >= __SCAN_POLICY_MAX)
|
2020-03-06 16:02:42 +08:00
|
|
|
{
|
2020-04-15 19:06:31 +08:00
|
|
|
mesa_runtime_log(RLOG_LV_FATAL, MODULE_NAME, "policy type error, policy id = %d", verify_policy->type);
|
2020-03-06 16:02:42 +08:00
|
|
|
goto free;
|
|
|
|
|
}
|
2020-02-18 17:54:15 +08:00
|
|
|
}
|
|
|
|
|
item = cJSON_GetObjectItem(subitem,"verifySession");
|
|
|
|
|
if(item == NULL || item->type!=cJSON_Object)
|
|
|
|
|
{
|
2020-03-06 16:02:42 +08:00
|
|
|
goto free;
|
2020-01-09 14:32:00 +08:00
|
|
|
}
|
2020-02-18 17:54:15 +08:00
|
|
|
attributes = cJSON_GetObjectItem(item,"attributes");
|
|
|
|
|
if(attributes && attributes->type==cJSON_Array)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
2020-01-17 18:57:51 +08:00
|
|
|
void *ctx = pangu_http_ctx_new(thread_id);
|
|
|
|
|
|
2020-02-18 17:54:15 +08:00
|
|
|
for (subchild = attributes->child; subchild != NULL; subchild = subchild->next)
|
2020-01-17 10:59:34 +08:00
|
|
|
{
|
2020-04-15 19:06:31 +08:00
|
|
|
hit_cnt = get_attribute_from_json(i, subchild, verify_policy);
|
|
|
|
|
if (hit_cnt < 0)
|
2020-02-18 17:54:15 +08:00
|
|
|
{
|
2020-03-06 16:02:42 +08:00
|
|
|
goto free;
|
2020-01-17 10:59:34 +08:00
|
|
|
}
|
2020-06-22 15:15:52 +08:00
|
|
|
hit_cnt = http_policy_scan(verify_policy->type, &verify_policy->verify_object[i], data_obj, ctx);
|
2020-04-15 19:06:31 +08:00
|
|
|
if(0 == strcasecmp(verify_policy->verify_object[i].attri_name, "ip"))
|
2020-01-17 10:59:34 +08:00
|
|
|
{
|
2020-04-15 19:06:31 +08:00
|
|
|
ipaddr_free(verify_policy->verify_object[i].ip_addr);
|
2020-01-17 10:59:34 +08:00
|
|
|
}
|
|
|
|
|
i++;
|
|
|
|
|
}
|
2020-03-06 16:02:42 +08:00
|
|
|
int item = 0;
|
|
|
|
|
cJSON *verfifySession = cJSON_CreateObject();
|
|
|
|
|
cJSON_AddItemToObject(data_obj, "verifySession", verfifySession);
|
|
|
|
|
cJSON *attributes=cJSON_CreateArray();
|
|
|
|
|
cJSON_AddItemToObject(verfifySession, "attributes", attributes);
|
|
|
|
|
for (item = 0; item < i; item++)
|
2020-02-18 17:54:15 +08:00
|
|
|
{
|
2020-06-22 15:15:52 +08:00
|
|
|
http_get_scan_status(&verify_policy->verify_object[item], attributes,data_obj, ctx);
|
2020-02-18 17:54:15 +08:00
|
|
|
}
|
2020-01-17 18:57:51 +08:00
|
|
|
pangu_http_ctx_free(ctx);
|
2019-10-22 15:13:14 +08:00
|
|
|
}
|
|
|
|
|
|
2020-01-17 10:59:34 +08:00
|
|
|
i=0;
|
2020-04-15 19:06:31 +08:00
|
|
|
FREE(&verify_policy);
|
2020-02-18 17:54:15 +08:00
|
|
|
}
|
2020-03-06 16:02:42 +08:00
|
|
|
goto end;
|
|
|
|
|
free:
|
2020-04-15 19:06:31 +08:00
|
|
|
if (verify_policy)
|
|
|
|
|
{
|
|
|
|
|
FREE(&verify_policy);
|
|
|
|
|
}
|
2020-02-18 17:54:15 +08:00
|
|
|
end:
|
2020-03-06 16:02:42 +08:00
|
|
|
if (hit_cnt >= 0)
|
2020-02-18 17:54:15 +08:00
|
|
|
{
|
|
|
|
|
cJSON_AddBoolToObject(policy_obj, "success", true);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
cJSON_AddBoolToObject(policy_obj, "success", false);
|
2020-01-09 14:32:00 +08:00
|
|
|
}
|
2019-10-22 15:13:14 +08:00
|
|
|
}
|
2020-01-17 10:59:34 +08:00
|
|
|
cJSON_Delete(data_json);
|
|
|
|
|
return policy_obj;
|
2019-10-22 15:13:14 +08:00
|
|
|
}
|
|
|
|
|
|
2020-01-17 10:59:34 +08:00
|
|
|
static int evhttp_socket_send(struct evhttp_request *req, char *sendbuf)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
|
|
|
|
struct evbuffer *evb = NULL;
|
|
|
|
|
|
|
|
|
|
/* This holds the content we're sending. */
|
|
|
|
|
evb = evbuffer_new();
|
|
|
|
|
|
|
|
|
|
if (sendbuf[0] == '\0' && req == NULL){
|
|
|
|
|
goto err;
|
|
|
|
|
}
|
|
|
|
|
evhttp_add_header(evhttp_request_get_output_headers(req),
|
2020-02-27 14:40:54 +08:00
|
|
|
"Content-Type", "application/json");
|
2019-10-22 15:13:14 +08:00
|
|
|
evhttp_add_header(evhttp_request_get_output_headers(req), "Connection", "keep-alive");
|
|
|
|
|
evbuffer_add_printf(evb, "%s", sendbuf);
|
|
|
|
|
evhttp_send_reply(req, HTTP_OK, "OK", evb);
|
|
|
|
|
goto done;
|
|
|
|
|
|
|
|
|
|
err:
|
|
|
|
|
evhttp_send_error(req, HTTP_NOTFOUND, "Document was not found");
|
|
|
|
|
done:
|
|
|
|
|
evbuffer_free(evb);
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void evhttp_request_cb(struct evhttp_request *evh_req, void *arg)
|
2020-01-09 14:32:00 +08:00
|
|
|
{
|
2020-01-17 10:59:34 +08:00
|
|
|
char *policy_payload= NULL; cJSON *policy_obj;
|
2019-10-22 15:13:14 +08:00
|
|
|
struct evbuffer * evbuf_body = NULL;
|
2020-01-09 14:32:00 +08:00
|
|
|
char *input = NULL; ssize_t inputlen=0;
|
|
|
|
|
|
2020-02-18 17:54:15 +08:00
|
|
|
struct verify_policy_thread *thread_ctx = (struct verify_policy_thread *)arg;
|
2019-10-22 15:13:14 +08:00
|
|
|
|
|
|
|
|
if (evhttp_request_get_command(evh_req) != EVHTTP_REQ_POST)
|
|
|
|
|
{
|
|
|
|
|
mesa_runtime_log(RLOG_LV_DEBUG, MODULE_NAME, "FAILED (post type)");
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
2020-01-09 14:32:00 +08:00
|
|
|
evbuf_body = evhttp_request_get_input_buffer(evh_req);
|
2019-10-22 15:13:14 +08:00
|
|
|
if (!evbuf_body || 0==(inputlen = evbuffer_get_length(evbuf_body)) ||!(input = (char *)evbuffer_pullup(evbuf_body,inputlen)))
|
|
|
|
|
{
|
|
|
|
|
mesa_runtime_log(RLOG_LV_FATAL, MODULE_NAME, "Failed to get post data information.");
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
2020-01-17 10:59:34 +08:00
|
|
|
|
|
|
|
|
policy_obj = get_query_from_request(input, thread_ctx->id);
|
|
|
|
|
if(policy_obj == NULL)
|
|
|
|
|
{
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
policy_payload = cJSON_PrintUnformatted(policy_obj);
|
2020-01-17 18:57:51 +08:00
|
|
|
mesa_runtime_log(RLOG_LV_DEBUG, MODULE_NAME, "[O] %s", policy_payload);
|
2020-01-17 10:59:34 +08:00
|
|
|
evhttp_socket_send(evh_req, policy_payload);
|
|
|
|
|
|
|
|
|
|
cJSON_Delete(policy_obj);
|
|
|
|
|
free(policy_payload);
|
|
|
|
|
|
2019-10-22 15:13:14 +08:00
|
|
|
goto finish;
|
2020-01-09 14:32:00 +08:00
|
|
|
|
2019-10-22 15:13:14 +08:00
|
|
|
error:
|
|
|
|
|
evhttp_send_error(evh_req, HTTP_BADREQUEST, 0);
|
|
|
|
|
finish:
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
2020-02-18 17:54:15 +08:00
|
|
|
void * verify_policy_thread_func(void * arg)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
|
|
|
|
struct evhttp_bound_socket *bound = NULL;
|
2020-02-18 17:54:15 +08:00
|
|
|
struct verify_policy_thread *thread_ctx = (struct verify_policy_thread *)arg;
|
2020-01-09 14:32:00 +08:00
|
|
|
|
2019-10-22 15:13:14 +08:00
|
|
|
thread_ctx->base = event_base_new();
|
2020-01-09 14:32:00 +08:00
|
|
|
if (! thread_ctx->base)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
|
|
|
|
mesa_runtime_log(RLOG_LV_FATAL, MODULE_NAME, "Can'thread_ctx allocate event base");
|
|
|
|
|
goto finish;
|
|
|
|
|
}
|
|
|
|
|
thread_ctx->http = evhttp_new(thread_ctx->base);
|
2020-01-09 14:32:00 +08:00
|
|
|
if (!thread_ctx->http)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
|
|
|
|
mesa_runtime_log(RLOG_LV_FATAL, MODULE_NAME, "couldn'thread_ctx create evhttp. Exiting.");
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
2020-01-09 14:32:00 +08:00
|
|
|
|
2020-02-18 17:54:15 +08:00
|
|
|
evhttp_set_cb(thread_ctx->http, "/v1/policy/verify", evhttp_request_cb, thread_ctx);
|
2020-01-09 14:32:00 +08:00
|
|
|
|
2019-10-22 15:13:14 +08:00
|
|
|
bound = evhttp_accept_socket_with_handle(thread_ctx->http, thread_ctx->accept_fd);
|
2020-01-09 14:32:00 +08:00
|
|
|
if (bound != NULL)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
|
|
|
|
mesa_runtime_log(RLOG_LV_INFO, MODULE_NAME, "Bound(%p) to port %d - Awaiting connections ... ", bound,
|
|
|
|
|
g_verify_proxy->listen_port);
|
|
|
|
|
}
|
|
|
|
|
mesa_runtime_log(RLOG_LV_INFO, MODULE_NAME, "Work thread %u is run...", thread_ctx->id);
|
2020-01-09 14:32:00 +08:00
|
|
|
|
2019-10-22 15:13:14 +08:00
|
|
|
event_base_dispatch(thread_ctx->base);
|
|
|
|
|
error:
|
|
|
|
|
event_base_free(thread_ctx->base);
|
|
|
|
|
finish:
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static int
|
|
|
|
|
evutil_fast_socket_nonblocking(evutil_socket_t fd)
|
|
|
|
|
{
|
|
|
|
|
#ifdef _WIN32
|
|
|
|
|
return evutil_make_socket_nonblocking(fd);
|
|
|
|
|
#else
|
|
|
|
|
if (fcntl(fd, F_SETFL, O_NONBLOCK) == -1) {
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
return 0;
|
|
|
|
|
#endif
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static int
|
|
|
|
|
evutil_fast_socket_closeonexec(evutil_socket_t fd)
|
|
|
|
|
{
|
|
|
|
|
#if !defined(_WIN32) && defined(EVENT__HAVE_SETFD)
|
|
|
|
|
if (fcntl(fd, F_SETFD, FD_CLOEXEC) == -1) {
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
#endif
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
evutil_socket_t
|
|
|
|
|
evutil_socket_(int domain, int type, int protocol)
|
|
|
|
|
{
|
|
|
|
|
evutil_socket_t r;
|
|
|
|
|
#if defined(SOCK_NONBLOCK) && defined(SOCK_CLOEXEC)
|
|
|
|
|
r = socket(domain, type, protocol);
|
|
|
|
|
if (r >= 0)
|
|
|
|
|
return r;
|
|
|
|
|
else if ((type & (SOCK_NONBLOCK|SOCK_CLOEXEC)) == 0)
|
|
|
|
|
return -1;
|
|
|
|
|
#endif
|
|
|
|
|
#define SOCKET_TYPE_MASK (~(EVUTIL_SOCK_NONBLOCK|EVUTIL_SOCK_CLOEXEC))
|
|
|
|
|
r = socket(domain, type & SOCKET_TYPE_MASK, protocol);
|
|
|
|
|
if (r < 0)
|
|
|
|
|
return -1;
|
|
|
|
|
if (type & EVUTIL_SOCK_NONBLOCK) {
|
|
|
|
|
if (evutil_fast_socket_nonblocking(r) < 0) {
|
|
|
|
|
evutil_closesocket(r);
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if (type & EVUTIL_SOCK_CLOEXEC) {
|
|
|
|
|
if (evutil_fast_socket_closeonexec(r) < 0) {
|
|
|
|
|
evutil_closesocket(r);
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return r;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static evutil_socket_t
|
|
|
|
|
evhttp_listen_socket_byuser(const struct sockaddr *sa, int socklen,
|
|
|
|
|
unsigned flags, int backlog)
|
|
|
|
|
{
|
|
|
|
|
evutil_socket_t fd;
|
|
|
|
|
int on = 1;
|
|
|
|
|
int family = sa ? sa->sa_family : AF_UNSPEC;
|
|
|
|
|
int socktype = SOCK_STREAM | EVUTIL_SOCK_NONBLOCK;
|
|
|
|
|
|
|
|
|
|
if (flags & LEV_OPT_CLOSE_ON_EXEC)
|
|
|
|
|
socktype |= EVUTIL_SOCK_CLOEXEC;
|
|
|
|
|
|
|
|
|
|
fd = evutil_socket_(family, socktype, 0);
|
|
|
|
|
if (fd == -1)
|
|
|
|
|
return fd;
|
|
|
|
|
|
|
|
|
|
if (setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, (void*)&on, sizeof(on))<0)
|
|
|
|
|
goto err;
|
|
|
|
|
if (flags & LEV_OPT_REUSEABLE) {
|
|
|
|
|
if (evutil_make_listen_socket_reuseable(fd) < 0)
|
|
|
|
|
goto err;
|
|
|
|
|
}
|
|
|
|
|
if (flags & LEV_OPT_REUSEABLE_PORT) {
|
|
|
|
|
if (evutil_make_listen_socket_reuseable_port(fd) < 0){
|
|
|
|
|
goto err;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if (sa) {
|
|
|
|
|
if (bind(fd, sa, socklen)<0)
|
|
|
|
|
goto err;
|
|
|
|
|
}
|
|
|
|
|
if (listen(fd, backlog) == -1) {
|
|
|
|
|
goto err;
|
|
|
|
|
}
|
|
|
|
|
return fd;
|
|
|
|
|
err:
|
|
|
|
|
evutil_closesocket(fd);
|
|
|
|
|
return fd;
|
|
|
|
|
}
|
|
|
|
|
|
2020-02-18 17:54:15 +08:00
|
|
|
int pangu_policy_work_thread_run(struct verify_policy * verify)
|
2019-10-22 15:13:14 +08:00
|
|
|
{
|
|
|
|
|
int xret = 0;
|
|
|
|
|
unsigned int tid = 0;
|
2020-02-18 17:54:15 +08:00
|
|
|
struct verify_policy_thread *thread_ctx = NULL;
|
2019-10-22 15:13:14 +08:00
|
|
|
|
|
|
|
|
struct sockaddr_in sin;
|
|
|
|
|
memset(&sin, 0, sizeof(struct sockaddr_in));
|
|
|
|
|
sin.sin_family = AF_INET;
|
|
|
|
|
sin.sin_port = htons(verify->listen_port);
|
|
|
|
|
evutil_socket_t accept_fd = evhttp_listen_socket_byuser((struct sockaddr*)&sin, sizeof(struct sockaddr_in),LEV_OPT_REUSEABLE_PORT|LEV_OPT_CLOSE_ON_FREE, -1);
|
|
|
|
|
if (accept_fd < 0)
|
|
|
|
|
{
|
|
|
|
|
mesa_runtime_log(RLOG_LV_FATAL, MODULE_NAME, "Could not create a listen!");
|
|
|
|
|
goto finish;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
for (tid = 0; tid < verify->nr_work_threads; tid++)
|
|
|
|
|
{
|
2020-02-18 17:54:15 +08:00
|
|
|
verify->work_threads[tid] = ALLOC(struct verify_policy_thread, 1);
|
2019-10-22 15:13:14 +08:00
|
|
|
thread_ctx = verify->work_threads[tid];
|
|
|
|
|
thread_ctx->id = tid;
|
|
|
|
|
thread_ctx->accept_fd =accept_fd;
|
2020-02-18 17:54:15 +08:00
|
|
|
thread_ctx->routine = verify_policy_thread_func;
|
2019-10-22 15:13:14 +08:00
|
|
|
|
|
|
|
|
if (pthread_create(&thread_ctx->pid, thread_ctx->attr, thread_ctx->routine, thread_ctx))
|
|
|
|
|
{
|
|
|
|
|
mesa_runtime_log(RLOG_LV_FATAL, MODULE_NAME, "%s", strerror(errno));
|
|
|
|
|
goto finish;
|
|
|
|
|
}
|
|
|
|
|
if (pthread_detach(thread_ctx->pid))
|
|
|
|
|
{
|
|
|
|
|
mesa_runtime_log(RLOG_LV_FATAL, MODULE_NAME, "%s", strerror(errno));
|
|
|
|
|
goto finish;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
FOREVER{
|
|
|
|
|
sleep(1);
|
|
|
|
|
}
|
|
|
|
|
finish:
|
|
|
|
|
return xret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int main(int argc, char * argv[])
|
|
|
|
|
{
|
|
|
|
|
const char * main_profile = "./conf/verify_policy.conf";
|
2020-06-24 16:36:16 +08:00
|
|
|
struct timespec start_time, end_time;
|
2019-10-22 15:13:14 +08:00
|
|
|
|
|
|
|
|
int ret = 0, opt = 0;
|
|
|
|
|
while ((opt = getopt(argc, argv, "v")) != -1)
|
|
|
|
|
{
|
|
|
|
|
switch (opt)
|
|
|
|
|
{
|
|
|
|
|
case 'v':
|
2020-01-09 14:32:00 +08:00
|
|
|
fprintf(stderr, "Welcome to Verify Policy Engine, Version: %s\n", version());
|
2019-10-22 15:13:14 +08:00
|
|
|
return 0;
|
|
|
|
|
default:
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
}
|
2020-02-18 17:54:15 +08:00
|
|
|
g_verify_proxy = ALLOC(struct verify_policy, 1);
|
2019-10-22 15:13:14 +08:00
|
|
|
assert(g_verify_proxy);
|
|
|
|
|
strcpy(g_verify_proxy->name, "verify_policy");
|
|
|
|
|
|
2020-06-24 16:36:16 +08:00
|
|
|
clock_gettime(CLOCK_REALTIME, &(start_time));
|
|
|
|
|
|
2019-10-22 15:13:14 +08:00
|
|
|
g_verify_proxy->logger = verify_syslog_init(main_profile);
|
|
|
|
|
CHECK_OR_EXIT(g_verify_proxy->logger != NULL, "Failed at init log module. Exit.");
|
|
|
|
|
|
|
|
|
|
ret = verify_policy_init(g_verify_proxy, main_profile);
|
|
|
|
|
CHECK_OR_EXIT(ret == 0, "Failed at loading profile %s, Exit.", main_profile);
|
2020-01-09 14:32:00 +08:00
|
|
|
|
2019-10-22 15:13:14 +08:00
|
|
|
ret = pangu_policy_init(g_verify_proxy, main_profile);
|
|
|
|
|
CHECK_OR_EXIT(ret == 0, "Failed at init panggu module, Exit.");
|
|
|
|
|
|
2020-06-24 16:36:16 +08:00
|
|
|
clock_gettime(CLOCK_REALTIME, &(end_time));
|
|
|
|
|
mesa_runtime_log(RLOG_LV_DEBUG, MODULE_NAME, "Read table_info_proxy.conf, take time %lu(s)", end_time.tv_sec - start_time.tv_sec);
|
|
|
|
|
printf("Read table_info_proxy.conf, take time %lu(s)\n", end_time.tv_sec - start_time.tv_sec);
|
|
|
|
|
|
|
|
|
|
clock_gettime(CLOCK_REALTIME, &(start_time));
|
2020-01-17 10:59:34 +08:00
|
|
|
ret = security_policy_init(g_verify_proxy, main_profile);
|
|
|
|
|
CHECK_OR_EXIT(ret == 0, "Failed at init security module, Exit.");
|
2020-06-24 16:36:16 +08:00
|
|
|
clock_gettime(CLOCK_REALTIME, &(end_time));
|
|
|
|
|
mesa_runtime_log(RLOG_LV_DEBUG, MODULE_NAME, "Read table_info_security.conf, take time %lu(s)", end_time.tv_sec - start_time.tv_sec);
|
|
|
|
|
printf("Read table_info_security.conf, take time %lu(s)\n", end_time.tv_sec - start_time.tv_sec);
|
2020-01-17 10:59:34 +08:00
|
|
|
|
2019-10-22 15:13:14 +08:00
|
|
|
ret = pangu_policy_work_thread_run(g_verify_proxy);
|
2020-01-09 14:32:00 +08:00
|
|
|
|
2019-10-22 15:13:14 +08:00
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|