637 lines
16 KiB
C
637 lines
16 KiB
C
#include "../include/parsefile.h"
|
||
#include "../include/configm.h"
|
||
#include "../../../netlink_uapi/libnetlinku.h"
|
||
#include "../../../../common/rpc/rpc.h"
|
||
#include "authfree.h"
|
||
#include <cjson/cJSON.h>
|
||
#include "../../../../../Common/s2j/s2j.h"
|
||
#include "../../../../../Common/commuapinl.h"
|
||
|
||
/*定义结构体数组 存在免认证规则 */
|
||
freeauth_configure_t freeauth_array[RULE_MAX_NUM] = {0};
|
||
|
||
|
||
#ifdef FREEAUTH_ACK_COOKIES
|
||
#define CFG_FREEAUTH_ACK_COOKIES
|
||
#endif
|
||
|
||
/*全局变量初始化 失败为1 成功为0*/
|
||
int authfreeInit(freeauth_configure_t **localuser)
|
||
{
|
||
*localuser = (freeauth_configure_t *)malloc(sizeof(freeauth_configure_t));
|
||
if (NULL == *localuser)
|
||
{
|
||
return 1;
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
/* 判断IPv4格式是否正确*/
|
||
int isIpV4Addr(const char *ipAddr)
|
||
{
|
||
int ip_part_1 = 0;
|
||
int ip_part_2 = 0;
|
||
int ip_part_3 = 0;
|
||
int ip_part_4 = 0;
|
||
char end_char = 0;
|
||
if((NULL == ipAddr) || (0 == strlen(ipAddr)))
|
||
{
|
||
return -1;
|
||
}
|
||
if(4 == sscanf(ipAddr,"%d.%d.%d.%d%c",&ip_part_1,&ip_part_2,&ip_part_3,&ip_part_4,&end_char))
|
||
{
|
||
if((ip_part_1 >= 0) && (ip_part_1 <= 255) &&
|
||
(ip_part_2 >= 0) && (ip_part_2 <= 255) &&
|
||
(ip_part_3 >= 0) && (ip_part_3 <= 255) &&
|
||
(ip_part_4 >= 0) && (ip_part_4 <= 255)
|
||
)
|
||
{
|
||
return 0;
|
||
}
|
||
}
|
||
return -1;
|
||
}
|
||
|
||
/*下发配置到内核态 */
|
||
int set_freeauthcfg_waitack(freeauth_configure_t *struct_freeauth)
|
||
{
|
||
int freeauth_len = 0;
|
||
struct nlmsghdr *ack = NULL;
|
||
struct nlmsghdr **answer = &ack;
|
||
|
||
struct{
|
||
struct nlmsghdr n;
|
||
char buf[1024];
|
||
} req ={
|
||
.n.nlmsg_len = NLMSG_LENGTH(0),
|
||
#ifdef CFG_FREEAUTH_ACK_COOKIES
|
||
.n.nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK,/*set NLM_F_ACK：use kernel auto ack*/
|
||
#else
|
||
.n.nlmsg_flags = NLM_F_REQUEST, /*not use kernel auto ack */
|
||
#endif
|
||
.n.nlmsg_type = FREEAUTH_CFG,
|
||
.n.nlmsg_pid = getpid(),
|
||
};
|
||
|
||
/*判断要发送的数据是否为NULL,不为NULL,打印出来 */
|
||
if (struct_freeauth == NULL)
|
||
{
|
||
printf("set_freeauthcfg_waitack is error: input struct_freeauth is NULL.\r\n");
|
||
return -1;
|
||
}else
|
||
{
|
||
char str[32];
|
||
memset(str, 0, 32);
|
||
inet_ntop(AF_INET, (void *)&struct_freeauth->sip, str, 32);
|
||
char *sip_addr = str;
|
||
char dtr[32];
|
||
memset(dtr, 0, 32);
|
||
inet_ntop(AF_INET, (void *)&struct_freeauth->dip, dtr, 32);
|
||
char *dip_addr = dtr;
|
||
printf("set_freeauthcfg_waitack :name %s sip %s dip %s dport %d\n",
|
||
struct_freeauth->name, sip_addr, dip_addr,
|
||
struct_freeauth->dport);
|
||
}
|
||
|
||
/*计算需要发送的数据的长度 */
|
||
freeauth_len = sizeof(freeauth_configure_t);
|
||
|
||
/*可选属性 */
|
||
commnl_addattr_l(&req.n, sizeof(req), 1, struct_freeauth, freeauth_len);
|
||
|
||
/*发送组装好的netlink消息 */
|
||
if(pdeliv_talk(1, &req.n, answer) < 0)
|
||
{
|
||
printf("set_user_freeauth_waitack rcv ack msg faild.\r\n");
|
||
return -2;
|
||
}
|
||
else
|
||
{
|
||
printf("set_user_freeauth_waitack rcv ack msg success.\r\n");
|
||
}
|
||
|
||
if(*answer != NULL)
|
||
{
|
||
printf("set_user_freeauth_waitack rcv answer.\r\n");
|
||
}
|
||
else{
|
||
printf("set_user_freeauth_waitack rcv answer error.\r\n");
|
||
return -3;
|
||
}
|
||
|
||
#ifdef CFG_FREEAUTH_ACK_COOKIES
|
||
/*recv answer*/
|
||
if((*answer)->nlmsg_type == NLMSG_ERROR){
|
||
nl_debugfs_extack(*answer);
|
||
}
|
||
#else
|
||
/*recv answer*/
|
||
if((*answer)->nlmsg_type == FREEAUTH_CFG)
|
||
{
|
||
nl_debugfs(*answer);
|
||
}
|
||
#endif
|
||
|
||
return 0;
|
||
}
|
||
|
||
/* 判断免认证规则格式
|
||
* iuput格式：{"type": 0, "data": {"name": "armink","sip": 1027824,"dip": 103427824,"dport": 24}}*/
|
||
ret_code freeauth_config_json_parse(pointer input, uint *conf_type, freeauth_configure_t *freeauth_buff)
|
||
{
|
||
ret_code ret = RET_OK;
|
||
cJSON *cjson, *type, *data;
|
||
|
||
/*JSON字符串到JSON格式 */
|
||
cjson = cJSON_Parse(input);
|
||
if(!cjson)
|
||
{
|
||
ret = RET_INPUTERR;
|
||
ASSERT_RET(ret);
|
||
return ret;
|
||
}
|
||
|
||
/*获取操作类型 add、mod、del */
|
||
type = cJSON_GetObjectItem(cjson, "type");
|
||
if(!type)
|
||
{
|
||
ret = RET_INPUTERR;
|
||
cJSON_Delete(cjson);
|
||
return ret;
|
||
}
|
||
|
||
*conf_type = type->valueint;
|
||
|
||
/*获取免认证规则的data部分 */
|
||
data = cJSON_GetObjectItem(cjson, "data");
|
||
if(!data)
|
||
{
|
||
ret = RET_INPUTERR;
|
||
cJSON_Delete(cjson);
|
||
cJSON_Delete(type);
|
||
return ret;
|
||
}
|
||
|
||
/*创建freeauth_configure_t结构体对象 */
|
||
s2j_create_struct_obj(struct_freeauth, freeauth_configure_t);
|
||
if(struct_freeauth == NULL)
|
||
{
|
||
cJSON_Delete(cjson);
|
||
cJSON_Delete(type);
|
||
cJSON_Delete(data);
|
||
return RET_NOMEM;
|
||
}
|
||
|
||
/*反序列化数据到freeauth_configure_t结构体对象 */
|
||
s2j_struct_get_basic_element(struct_freeauth, data, string, name);
|
||
s2j_struct_get_basic_element(struct_freeauth, data, int, sip);
|
||
s2j_struct_get_basic_element(struct_freeauth, data, int, dip);
|
||
s2j_struct_get_basic_element(struct_freeauth, data, int, dport);
|
||
|
||
memcpy(freeauth_buff->name, struct_freeauth->name, sizeof(char)*32);
|
||
freeauth_buff->sip = struct_freeauth->sip;
|
||
freeauth_buff->dip = struct_freeauth->dip;
|
||
freeauth_buff->dport = struct_freeauth->dport;
|
||
|
||
s2j_delete_struct_obj(struct_freeauth);
|
||
cJSON_Delete(cjson);
|
||
cJSON_Delete(type);
|
||
cJSON_Delete(data);
|
||
|
||
return RET_OK;
|
||
}
|
||
|
||
/*检查增加的参数格式是否正确 */
|
||
ret_code freeauth_config_add_chk(uint source,uint config_type,
|
||
pointer input, int input_len,
|
||
pointer output, int *output_len)
|
||
{
|
||
ret_code ret = RET_OK;
|
||
freeauth_configure_t *freeauth_configure;
|
||
freeauth_configure = (freeauth_configure_t *)input;
|
||
char str[32] = {0};
|
||
char dtr[32] = {0};
|
||
|
||
if((input_len < sizeof(freeauth_configure_t)) || (input_len > sizeof(freeauth_configure_t)))
|
||
{
|
||
ret = RET_INPUTERR;
|
||
return ret;
|
||
}
|
||
|
||
memset(str, 0, 32);
|
||
inet_ntop(AF_INET, (void *)&freeauth_configure->sip, str, 32);
|
||
char *sip_addr = str;
|
||
if( isIpV4Addr(sip_addr) < 0 )
|
||
{
|
||
ret = RET_IPINVALID;
|
||
return ret;
|
||
}
|
||
|
||
memset(dtr, 0, 32);
|
||
inet_ntop(AF_INET, (void *)&freeauth_configure->dip, dtr, 32);
|
||
char *dip_addr = dtr;
|
||
if( isIpV4Addr(dip_addr) < 0 )
|
||
{
|
||
ret = RET_IPINVALID;
|
||
return ret;
|
||
}
|
||
|
||
if ( (freeauth_configure->dport < DPORT_MIN_NUM) && (freeauth_configure->dport > DPORT_MAX_NUM ))
|
||
{
|
||
ret = RET_IPINVALID; /*先用IPVAILD表示，后面加PORTVAILD */
|
||
return ret;
|
||
}
|
||
|
||
return RET_OK;
|
||
}
|
||
|
||
|
||
/*删除的时候以免认证规则名作为参数，检查免认证规则名是否存在 */
|
||
ret_code freeauth_config_del_chk(uint source,uint config_type,
|
||
pointer input, int input_len,
|
||
pointer output, int *output_len)
|
||
{
|
||
ret_code ret = RET_OK;
|
||
freeauth_configure_t *freeauth_configure;
|
||
freeauth_configure = (freeauth_configure_t *)input;
|
||
int i;
|
||
|
||
if((input_len < sizeof(freeauth_configure_t)) || (input_len > sizeof(freeauth_configure_t)))
|
||
{
|
||
ret = RET_INPUTERR;
|
||
return ret;
|
||
}
|
||
|
||
for(i = 0; i < RULE_MAX_NUM; i++)
|
||
{
|
||
if (0 != strcmp(freeauth_array[i].name, freeauth_configure->name))
|
||
return RET_NOTFOUND;
|
||
}
|
||
|
||
return RET_OK;
|
||
}
|
||
|
||
/*修改 查询要修改的内容是否存在 */
|
||
ret_code freeauth_config_mod_chk(uint source,uint config_type,
|
||
pointer input, int input_len,
|
||
pointer output, int *output_len)
|
||
{
|
||
ret_code ret = RET_OK;
|
||
freeauth_configure_t *freeauth_configure;
|
||
freeauth_configure = (freeauth_configure_t *)input;
|
||
int i;
|
||
|
||
if((input_len < sizeof(freeauth_configure_t)) || (input_len > sizeof(freeauth_configure_t)))
|
||
{
|
||
ret = RET_INPUTERR;
|
||
return ret;
|
||
}
|
||
|
||
/*检查修改的内容是否存在 */
|
||
for(i = 0; i < RULE_MAX_NUM; i++)
|
||
{
|
||
if (0 != strcmp(freeauth_array[i].name, freeauth_configure->name))
|
||
return RET_NOTFOUND;
|
||
}
|
||
return RET_OK;
|
||
|
||
}
|
||
|
||
/*chk data格式 */
|
||
ret_code freeauth_config_chk(uint source,uint *config_type,
|
||
pointer input, int *input_len,
|
||
pointer output, int *output_len)
|
||
{
|
||
|
||
ret_code ret = RET_OK;
|
||
freeauth_configure_t freeauth_configure = {0};
|
||
int config_len = sizeof(freeauth_configure_t);
|
||
uint conf_type = FREEAUTH_CONFIG_GET;
|
||
int code = 0;
|
||
|
||
freeauth_config_json_parse(input, &conf_type, &freeauth_configure);
|
||
|
||
switch (conf_type)
|
||
{
|
||
case FREEAUTH_CONFIG_ADD:
|
||
ret = freeauth_config_add_chk(source, conf_type,
|
||
&freeauth_configure, config_len,
|
||
output, output_len);
|
||
break;
|
||
case FREEAUTH_CONFIG_MOD:
|
||
ret = freeauth_config_mod_chk(source, conf_type,
|
||
&freeauth_configure, config_len,
|
||
output, output_len);
|
||
break;
|
||
case FREEAUTH_CONFIG_DEL:
|
||
ret = freeauth_config_del_chk(source, conf_type,
|
||
&freeauth_configure, config_len,
|
||
output, output_len);
|
||
break;
|
||
default:
|
||
ret = RET_NOTSUPPORT;
|
||
}
|
||
|
||
if(config_len <= CM_BUFF_SIZE)
|
||
{
|
||
memset(input, 0, *input_len);
|
||
memcpy(input, &freeauth_configure, config_len);
|
||
*config_type = conf_type;
|
||
*input_len = config_len;
|
||
}
|
||
else
|
||
{
|
||
ret = RET_NOMEM;
|
||
}
|
||
|
||
RET_ERR_FORMART(ret, code, output, *output_len);
|
||
|
||
return ret;
|
||
}
|
||
|
||
|
||
ret_code freeauth_config_add_proc(uint source, uint config_type,
|
||
pointer input, int input_len,
|
||
pointer output, int *output_len)
|
||
{
|
||
ret_code ret = RET_OK;
|
||
cJSON *res;
|
||
char * ret_char = NULL;
|
||
unsigned int ret_int = 0;
|
||
freeauth_configure_t *freeauth_configure;
|
||
freeauth_configure = (freeauth_configure_t *)input;
|
||
|
||
/*增加数据库*/
|
||
|
||
/*存入全局变量*/
|
||
|
||
/*用户态下发到内核态auth_hook */
|
||
int r = -1;
|
||
printf("cfgchannel main begin:\r\n");
|
||
|
||
/*创建通道 */
|
||
r = commcfgnl_open();
|
||
if(r < 0)
|
||
{
|
||
printf(" pdlivnl_open fail, exit.\r\n");
|
||
return RET_ERR;
|
||
}
|
||
|
||
/*下发配置到内核态 */
|
||
r = set_freeauthcfg_waitack(freeauth_configure);
|
||
if(r < 0)
|
||
{
|
||
printf("set_cfg_debug_waitack failed.\r\n");
|
||
return RET_ERR;
|
||
}
|
||
|
||
/*关闭netlink通道 */
|
||
commcfgnl_close();
|
||
printf("cfgchannel main exit!\r\n");
|
||
|
||
/*创建json对象 */
|
||
res = cJSON_CreateObject();
|
||
if(!res)
|
||
{
|
||
free(freeauth_configure);
|
||
ret = RET_ERR;
|
||
return ret;
|
||
}
|
||
|
||
/*将json对象转换成json字符串 返回处理结果*/
|
||
cJSON_AddNumberToObject(res, "result", r);
|
||
ret_char = cJSON_PrintUnformatted(res);
|
||
ret_int = strlen(ret_char);
|
||
if(output_len)
|
||
{
|
||
*output_len = ret_int;
|
||
}
|
||
|
||
/*超出2k的内存，报错 */
|
||
if(ret_int >= 1024 * 2)
|
||
{
|
||
free(freeauth_configure);
|
||
free(ret_char);
|
||
cJSON_Delete(res);
|
||
return RET_NOMEM;
|
||
}
|
||
|
||
memcpy(output, 0, ret_int + 1);
|
||
strcpy(output, ret_char);
|
||
|
||
free(ret_char);
|
||
cJSON_Delete(res);
|
||
free(freeauth_configure);
|
||
return RET_OK;
|
||
}
|
||
|
||
|
||
ret_code freeauth_config_mod_proc(uint source, uint config_type,
|
||
pointer input, int input_len,
|
||
pointer output, int *output_len)
|
||
{
|
||
ret_code ret = RET_OK;
|
||
cJSON *res;
|
||
char * ret_char = NULL;
|
||
unsigned int ret_int = 0;
|
||
freeauth_configure_t *freeauth_configure;
|
||
freeauth_configure = (freeauth_configure_t *)input;
|
||
|
||
/*数据库修改*/
|
||
|
||
/*存入全局变量*/
|
||
|
||
/*用户态下发到内核态auth_hook */
|
||
int r = -1;
|
||
printf("cfgchannel main begin:\r\n");
|
||
|
||
/*创建通道 */
|
||
r = commcfgnl_open();
|
||
if(r < 0)
|
||
{
|
||
printf(" pdlivnl_open fail, exit.\r\n");
|
||
return RET_ERR;
|
||
}
|
||
|
||
/*下发配置到内核态 */
|
||
r = set_freeauthcfg_waitack(freeauth_configure);
|
||
if(r < 0)
|
||
{
|
||
printf("set_cfg_debug_waitack failed.\r\n");
|
||
return RET_ERR;
|
||
}
|
||
|
||
/*关闭netlink通道 */
|
||
commcfgnl_close();
|
||
printf("cfgchannel main exit!\r\n");
|
||
|
||
/*创建json对象 */
|
||
res = cJSON_CreateObject();
|
||
if(!res)
|
||
{
|
||
free(freeauth_configure);
|
||
ret = RET_ERR;
|
||
return ret;
|
||
}
|
||
|
||
cJSON_AddNumberToObject(res, "result", r);
|
||
|
||
/*将json对象转换成json字符串 */
|
||
ret_char = cJSON_PrintUnformatted(res);
|
||
ret_int = strlen(ret_char);
|
||
if(output_len)
|
||
{
|
||
*output_len = ret_int;
|
||
}
|
||
|
||
/*超出2k的内存，报错 */
|
||
if(ret_int >= 1024 * 2)
|
||
{
|
||
free(freeauth_configure);
|
||
free(ret_char);
|
||
cJSON_Delete(res);
|
||
return RET_NOMEM;
|
||
}
|
||
|
||
memcpy(output, 0, ret_int + 1);
|
||
strcpy(output, ret_char);
|
||
|
||
free(ret_char);
|
||
cJSON_Delete(res);
|
||
free(freeauth_configure);
|
||
return RET_OK;
|
||
|
||
}
|
||
|
||
ret_code freeauth_config_del_proc(uint source, uint config_type,
|
||
pointer input, int input_len,
|
||
pointer output, int *output_len)
|
||
{
|
||
ret_code ret = RET_OK;
|
||
cJSON *res;
|
||
char * ret_char = NULL;
|
||
unsigned int ret_int = 0;
|
||
freeauth_configure_t *freeauth_configure;
|
||
freeauth_configure = (freeauth_configure_t *)input;
|
||
|
||
/*数据库删除*/
|
||
|
||
/*存入全局变量*/
|
||
|
||
/*用户态下发到内核态auth_hook */
|
||
int r = -1;
|
||
printf("cfgchannel main begin:\r\n");
|
||
|
||
/*创建通道 */
|
||
r = commcfgnl_open();
|
||
if(r < 0)
|
||
{
|
||
printf(" pdlivnl_open fail, exit.\r\n");
|
||
return RET_ERR;
|
||
}
|
||
|
||
/*下发配置到内核态 */
|
||
r = set_freeauthcfg_waitack(freeauth_configure);
|
||
if(r < 0)
|
||
{
|
||
printf("set_cfg_debug_waitack failed.\r\n");
|
||
return RET_ERR;
|
||
}
|
||
|
||
/*关闭netlink通道 */
|
||
commcfgnl_close();
|
||
printf("cfgchannel main exit!\r\n");
|
||
|
||
/*创建json对象 */
|
||
res = cJSON_CreateObject();
|
||
if(!res)
|
||
{
|
||
free(freeauth_configure);
|
||
ret = RET_ERR;
|
||
return ret;
|
||
}
|
||
|
||
cJSON_AddNumberToObject(res, "result", r);
|
||
|
||
/*将json对象转换成json字符串 */
|
||
ret_char = cJSON_PrintUnformatted(res);
|
||
ret_int = strlen(ret_char);
|
||
if(output_len)
|
||
{
|
||
*output_len = ret_int;
|
||
}
|
||
|
||
/*超出2k的内存，报错 */
|
||
if(ret_int >= 1024 * 2)
|
||
{
|
||
free(freeauth_configure);
|
||
free(ret_char);
|
||
cJSON_Delete(res);
|
||
return RET_NOMEM;
|
||
}
|
||
|
||
memcpy(output, 0, ret_int + 1);
|
||
strcpy(output, ret_char);
|
||
|
||
free(ret_char);
|
||
cJSON_Delete(res);
|
||
free(freeauth_configure);
|
||
return RET_OK;
|
||
|
||
}
|
||
|
||
|
||
ret_code freeauth_config_proc(uint source, uint config_type,
|
||
pointer input, int input_len,
|
||
pointer output, int *output_len)
|
||
{
|
||
|
||
uint conf_type = config_type;
|
||
ret_code ret = RET_OK;
|
||
int r = -1;
|
||
cJSON *res;
|
||
char * ret_char = NULL;
|
||
unsigned int ret_int = 0;
|
||
freeauth_configure_t conf_buff = {0};
|
||
freeauth_configure_t *freeauth_configure = &conf_buff;
|
||
|
||
freeauth_configure = (freeauth_configure_t *)input;
|
||
|
||
rpc_log_info("config type is %d, name %s sip %d dip %d dport %d\n",
|
||
conf_type, freeauth_configure->name,
|
||
freeauth_configure->sip, freeauth_configure->dip,
|
||
freeauth_configure->dport);
|
||
|
||
switch (conf_type)
|
||
{
|
||
case FREEAUTH_CONFIG_ADD:
|
||
ret = freeauth_config_add_proc(source, conf_type,
|
||
&freeauth_configure, input_len,
|
||
output, output_len);
|
||
break;
|
||
case FREEAUTH_CONFIG_MOD:
|
||
ret = freeauth_config_mod_proc(source, conf_type,
|
||
&freeauth_configure, input_len,
|
||
output, output_len);
|
||
break;
|
||
case FREEAUTH_CONFIG_DEL:
|
||
ret = freeauth_config_del_proc(source, conf_type,
|
||
&freeauth_configure, input_len,
|
||
output, output_len);
|
||
break;
|
||
default:
|
||
ret = RET_NOTSUPPORT;
|
||
}
|
||
|
||
return RET_OK;
|
||
}
|
||
|
||
|
||
int main()
|
||
{
|
||
|
||
}
|
||
|
||
|