cmhi
/
secgateway
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
secgateway/Product/user/object_manager/main.c

491 lines
15 KiB
C
Raw Blame History

#include <stdio.h>
#include <unistd.h>
#include <getopt.h>
#include <fcntl.h>
#include <errno.h>
#include <string.h>
#include <regex.h>
#include <pthread.h>
#include <uthash/uthash.h>
#include <arpa/inet.h>
#include "compile.h"
#include "log.h"
#include "object_manager.h"
#include "json_interface.h"
#define MAX_OBJ (300)
#define ADD_JS_FILE ("/home/hx/secogateway/Product/user/object_manager/json_test/add.json")
#define REGEX_SVR_PORT ("^[0-9]{1,5}$")
#define REGEX_IP_ADDR "^((25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\\.){3}(25[0-5]|2[0-4][0-9]|[01]\
?[0-9][0-9]?)(\\/([0-9]|[1-2][0-9]|3[0-2]))?$|^([0-9a-fA-F]{1,4}:){6}((25[0-5]|2[0-4][0-9]|[01]?[0-9]\
[0-9]?)\\.){3}(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)(\\/([1-9]?[0-9]|(1([0-1][0-9]|2[0-8]))))?$|^::(\
[0-9a-fA-F]{1,4}:){0,4}((25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\\.){3}(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\
(\\/([1-9]?[0-9]|(1([0-1][0-9]|2[0-8]))))?$|^([0-9a-fA-F]{1,4}:):([0-9a-fA-F]{1,4}:){0,3}((25[0-5]|2[0-4]\
[0-9]|[01]?[0-9][0-9]?)\\.){3}(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)(\\/([1-9]?[0-9]|(1([0-1][0-9]|2[0-8]))\
))?$|^([0-9a-fA-F]{1,4}:){2}:([0-9a-fA-F]{1,4}:){0,2}((25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\\.){3}(25[0-5]\
|2[0-4][0-9]|[01]?[0-9][0-9]?)(\\/([1-9]?[0-9]|(1([0-1][0-9]|2[0-8]))))?$|^([0-9a-fA-F]{1,4}:){3}:([0-9a-fA-F]\
{1,4}:){0,1}((25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\\.){3}(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)(\\/([1-9]?[0-9]|\
(1([0-1][0-9]|2[0-8]))))?$|^([0-9a-fA-F]{1,4}:){4}:((25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\\.){3}(25[0-5]|2[0-4]\
[0-9]|[01]?[0-9][0-9]?)(\\/([1-9]?[0-9]|(1([0-1][0-9]|2[0-8]))))?$|^([0-9a-fA-F]{1,4}:){7}[0-9a-fA-F]{1,4}\
(\\/([1-9]?[0-9]|(1([0-1][0-9]|2[0-8]))))?$|^:((:[0-9a-fA-F]{1,4}){1,6}|:)(\\/([1-9]?[0-9]|(1([0-1][0-9]|2[0-8]\
))))?$|^[0-9a-fA-F]{1,4}:((:[0-9a-fA-F]{1,4}){1,5}|:)(\\/([1-9]?[0-9]|(1([0-1][0-9]|2[0-8]))))?$|^([0-9a-fA-F]\
{1,4}:){2}((:[0-9a-fA-F]{1,4}){1,4}|:)(\\/([1-9]?[0-9]|(1([0-1][0-9]|2[0-8]))))?$|^([0-9a-fA-F]{1,4}:){3}\
((:[0-9a-fA-F]{1,4}){1,3}|:)(\\/([1-9]?[0-9]|(1([0-1][0-9]|2[0-8]))))?$|^([0-9a-fA-F]{1,4}:){4}((:[0-9a-fA-F]\
{1,4}){1,2}|:)(\\/([1-9]?[0-9]|(1([0-1][0-9]|2[0-8]))))?$|^([0-9a-fA-F]{1,4}:){5}:([0-9a-fA-F]{1,4})?\
(\\/([1-9]?[0-9]|(1([0-1][0-9]|2[0-8]))))?$|^([0-9a-fA-F]{1,4}:){6}:(\\/([1-9]?[0-9]|(1([0-1][0-9]|2[0-8]))))?$"
static PCMHI_OBJECT g_pObject = NULL;
static OBJECT_K g_objItem[MAX_OBJ];
int regex_match(const char *regex, const char *pStr)
{
int result = FALSE;
regex_t reg;
int regexInit;
if(!regex || !pStr) {
LOG_EX(LOG_Error, "Input params error: %p, %p\n", regex, pStr);
return FALSE;
}
regexInit = regcomp(&reg, regex, REG_EXTENDED);
if(regexInit) {
LOG_EX(LOG_Error, "regcomp error\n");
return FALSE;
} else {
int reti = regexec(&reg, pStr, 0, NULL, 0);
if(REG_NOERROR != reti) {
char buf[1024];
memset(buf, 0, 1024);
//Error print: match failed!
regerror(reti, regexInit, buf, 1024);
LOG_EX(LOG_Error, "%s match %s failed(%d): %s\n", regex, pStr, reti, buf);
} else {
//LOG_EX(LOG_Debug, "%s match %s OK\n", regex, pStr);
result = TRUE;
}
}
regfree(&reg);
return result;
}
int split_params(char *pInput, char **pFirst, char **pSecond, const char *split)
{
char *pStr = (char *)pInput;
char *p;
if(!pInput || !pFirst || !pSecond) {
LOG_EX(LOG_Error, "Input params error: %p, %p, %p\n", pInput, pFirst, pSecond);
return -RET_INPUTERR;
}
p = strsep(&pStr, split);
if(strlen(p) == 0) {
*pFirst = pInput;
*pSecond = NULL;
} else {
*pFirst = p;
*pSecond = pStr;
}
LOG_EX(LOG_Debug, "[%s] Split to First = %s, Second = %s\n", pInput, *pFirst, *pSecond);
return RET_OK;
}
static int get_free_pos(void)
{
int i;
for(i = 0; i < MAX_OBJ; i++) {
if(strlen(g_objItem[i].memTag) == 0) {
return i;
}
}
return -1;
}
static void prase_port_object(PSVR_OBJECT pSvr)
{
int i = 0;
if(!pSvr) {
return;
}
for(i = 0; i < strlen(pSvr->str_val); i++) {
//if(pSvr->str_val[i] >)
}
if(strchr(pSvr->str_val, ',') != NULL) {
} else if(strchr(pSvr->str_val, '-') != NULL) {
}
}
int object_add(PCMHI_OBJECT pObj, POBJECT_K pObjK)
{
int i, pos;
PCMHI_OBJECT p;
PDT_OBJECT pDt;
ADDR_OBJECT pAddr;
if(!pObj || !pObjK) {
LOG_EX(LOG_Error, "Input params error %p, %p\n", pObj, pObjK);
return -RET_INPUTERR;
}
pos = get_free_pos();
if(pos < 0) {
LOG_EX(LOG_Error, "Share memory full\n");
return -RET_NOMEM;
}
strcpy(pObjK->memTag, OBJ_MEM_TAG);
for(i = 0; i < pObjK->ctx_num && i < MAX_OBJ_CONTENT; i++) {
char *pStart = NULL, *pEnd = NULL;
char *pSubStart = NULL, *pSubEnd = NULL;
PSVR_OBJECT pSvr = &pObjK->objs.svr_obj[i];
PADDR_OBJECT pAddr = &pObjK->objs.addr_obj[i];
split_params(pSvr->str_val, &pStart, &pEnd, "-");
switch(pObjK->type / 10) {
case OBJ_TYPE_SERVER:
if(regex_match(REGEX_SVR_PORT, pStart)) {
LOG_EX(LOG_Error, "Input %s format error\n", pStart);
return -RET_INPUTERR;
}
if(pEnd && strlen(pEnd) > 0) {
unsigned short first, last;
if(regex_match(REGEX_SVR_PORT, pEnd)) {
LOG_EX(LOG_Error, "Input %s format error\n", pEnd);
return -RET_INPUTERR;
}
first = strtoul(pStart, NULL, 10) & 0xFFFF;
last = strtoul(pEnd, NULL, 10) & 0xFFFF;
if(pSvr->pro_type <= PORT_UDP) {
pSvr->min_port = htons(first);
pSvr->max_port = htons(last);
} else {
pSvr->min_port = first;
pSvr->max_port = last;
}
} else {
unsigned short first;
first = strtoul(pStart, NULL, 10) & 0xFFFF;
pSvr->min_port = htons(first);
pSvr->max_port = 0;
}
break;
case OBJ_TYPE_ADDR:
if(regex_match(REGEX_IP_ADDR, pStart)) {
LOG_EX(LOG_Error, "Input %s format error\n", pStart);
return -RET_INPUTERR;
}
split_params(pStart, &pSubStart, &pSubEnd, "/");
if(pSubEnd && strlen(pSubEnd) > 0) {
pAddr->net_mask = strtoul(pSubEnd, NULL, 10);
}
// IPv4 格式
if(strchr(pStart, ':') == NULL) {
if(inet_pton(AF_INET, pSubStart, &pAddr->min_addr.addr_v4.ip4_addr) != 1) {
LOG_EX(LOG_Error, "%s convert to ip address error\n", pStart);
return -RET_INPUTERR;
}
pAddr->ip_version = IP_V4;
} else {
pAddr->ip_version = IP_V6;
if(inet_pton(AF_INET6, pSubStart, &pAddr->min_addr.addr_v6.addr.addr8) != 1) {
LOG_EX(LOG_Error, "%s convert to ip address error\n", pStart);
return -RET_INPUTERR;
}
}
if(pEnd && strlen(pEnd) > 0) {
if(regex_match(REGEX_IP_ADDR, pEnd)) {
LOG_EX(LOG_Error, "Input %s format error\n", pEnd);
return -RET_INPUTERR;
}
if(pAddr->ip_version == IP_V4) {
if(inet_pton(AF_INET, pEnd, &pAddr->max_addr.addr_v4.ip4_addr) != 1) {
LOG_EX(LOG_Error, "%s convert to ip address error\n", pEnd);
return -RET_INPUTERR;
}
} else {
if(inet_pton(AF_INET6, pEnd, &pAddr->max_addr.addr_v6.addr.addr8) != 1) {
LOG_EX(LOG_Error, "%s convert to ip address error\n", pEnd);
return -RET_INPUTERR;
}
}
}
//LOG_EX(LOG_Info, " value: %s\n", pObjK->objs.addr_obj[j].str_val);
break;
case OBJ_TYPE_DATETIME:
//LOG_EX(LOG_Info, " repeat: %d\n", pObjK->objs.dt_obj[j].rep_mode);
//LOG_EX(LOG_Info, " value: %s\n", pObjK->objs.dt_obj[j].str_val);
break;
}
}
pObj->obj_index = pos;
pthread_mutex_init(&pObj->lock, NULL);
HASH_FIND_STR(g_pObject, pObj->name, p);
if(p != NULL) {
LOG_EX(LOG_Error, "Item %s exists\n", pObj->name);
return -RET_EXIST;
}
HASH_ADD_STR(g_pObject, name, pObj);
}
static const char *read_json_file(const char *pPath)
{
FILE *pFile;
unsigned char *pBuf;
int f_size = 0;
GET_FILE_SIZE(pPath, f_size);
if(f_size <= 0) {
LOG_EX(LOG_Error, "Get file %p size error: %d\n", pPath, f_size);
return NULL;
}
pBuf = (unsigned char *)malloc(f_size + 1);
if(!pBuf) {
LOG_EX(LOG_Error, "Malloc %d bytes memory error: %d\n", f_size + 1);
return NULL;
}
pFile = fopen(pPath, "r");
if(!pFile) {
LOG_EX(LOG_Error, "Open file %p size error\n", pPath);
free(pBuf);
return NULL;
}
fread(pBuf, 1, f_size, pFile);
fclose(pFile);
pBuf[f_size] = 0;
return (const char *)pBuf;
}
static void test_add_object(void)
{
PJSON_INTERFACE p = NULL;
PIFACE_ADD_OBJ pAdd = NULL;
int i, j, ret;
const char *pJson = read_json_file(ADD_JS_FILE);
ret = Json2Struct(pJson, &p, JE_INTERFACE, FALSE);
if(ret != RET_OK || p == NULL) {
LOG_EX(LOG_Error, "Decode json error: %d\n", ret);
free(pJson);
return;
}
LOG_EX(LOG_Info, "cmdId: %d\n", p->cmdId);
LOG_EX(LOG_Info, "ver: %d\n", p->ver);
LOG_EX(LOG_Info, "cryptoType: %d\n", p->cryptoType);
LOG_EX(LOG_Info, "timeStamp: %d\n", p->timeStamp);
LOG_EX(LOG_Info, "msgContent: %s\n", p->msgContent);
ret = Json2Struct(p->msgContent, &pAdd, JE_OBJ_ADD, FALSE);
if(ret != RET_OK || pAdd == NULL) {
LOG_EX(LOG_Error, "Decode json error: %d\n", ret);
if(p->msgContent) {
free(p->msgContent);
}
free(p);
free(pJson);
return;
}
for(i = 0; i < pAdd->n_obj; i++) {
LOG_EX(LOG_Info, "name: %s\n", pAdd->pCtx[i].obj.name);
LOG_EX(LOG_Info, "desc: %s\n", pAdd->pCtx[i].obj.desc);
LOG_EX(LOG_Info, "prio: %d\n", pAdd->pCtx[i].obj.prio);
LOG_EX(LOG_Info, "type: %d\n", pAdd->pCtx[i].objk.type);
LOG_EX(LOG_Info, "items: %d\n", pAdd->pCtx[i].objk.ctx_num);
for(j = 0; j < pAdd->pCtx[i].objk.ctx_num; j++) {
POBJECT_K pObjK = &pAdd->pCtx[i].objk;
switch(pObjK->type / 10) {
case OBJ_TYPE_SERVER:
LOG_EX(LOG_Info, " porType: %d\n", pObjK->objs.svr_obj[j].pro_type);
LOG_EX(LOG_Info, " value: %s\n", pObjK->objs.svr_obj[j].str_val);
break;
case OBJ_TYPE_ADDR:
LOG_EX(LOG_Info, " value: %s\n", pObjK->objs.addr_obj[j].str_val);
break;
case OBJ_TYPE_DATETIME:
LOG_EX(LOG_Info, " repeat: %d\n", pObjK->objs.dt_obj[j].rep_mode);
LOG_EX(LOG_Info, " value: %s\n", pObjK->objs.dt_obj[j].str_val);
break;
}
}
}
if(pAdd) {
free(pAdd);
}
if(p->msgContent) {
free(p->msgContent);
}
free(p);
free(pJson);
}
static void test_strsep(char *pVal)
{
char *pStart = NULL, *pEnd = NULL;
split_params(pVal, &pStart, &pEnd, "-");
if(pEnd) {
LOG_EX(LOG_Debug, "[%s] Split to First = %s, Second = %s\n", pVal, pStart, pEnd);
} else {
LOG_EX(LOG_Debug, "[%s] Split to First = %s\n", pVal, pStart);
}
}
static void test_regex(char *pVal)
{
LOG_EX(LOG_Debug, "Ret : [%s] ==> %d\n", pVal, regex_match(REGEX_IP_ADDR, pVal));
//LOG_EX(LOG_Debug, "Ret : [%s] ==> %d\n", pVal, regex_match(REGEX_SVR_PORT, pVal));
}
static void test_ipaddr_format(char *pVal)
{
int ret = 0;
IP4_ADDR ip_v4;
IP6_ADDR ip_v6;
memset(&ip_v4, 0, sizeof(IP4_ADDR));
memset(&ip_v6, 0, sizeof(IP6_ADDR));
ret = regex_match(REGEX_IP_ADDR, pVal);
if(ret != TRUE) {
LOG_EX(LOG_Error, "Verify ipadd format error\n");
}
if(strchr(pVal, ':') == NULL) {
ret = inet_pton(AF_INET, pVal, &ip_v4.ip4_addr);
print_hex_dump_bytes("IP", 0, &ip_v4, sizeof(IP4_ADDR));
} else {
ret = inet_pton(AF_INET6, pVal, ip_v6.addr.addr8);
print_hex_dump_bytes("IP", 0, &ip_v6, sizeof(IP6_ADDR));
}
LOG_EX(LOG_Debug, "Convert %s to ip address ret: %d\n", pVal, ret);
}
/**
* @brief 应用程序主函数
* @param argc 输入参数个数
* @param argv 输入参数详细内容
* @return 0
*/
int main(int argc, char **argv)
{
int c, optidx = 0;
static const struct option long_opts[] = {
{ "help", no_argument, NULL, 'h' },
{ "version", no_argument, NULL, 'v' },
// TODO 添加其它需要处理的参数配置
{ "interface", no_argument, NULL, 'i' },
{ "regex", 1, NULL, 'r' },
{ "split_str", 1, NULL, 's' },
{ "ip_addr", 1, NULL, 'p' },
{NULL, 0, NULL, 0}
};
IHW_InitLOG("obj", NULL, TRUE);
LOG_EX(LOG_Debug, "OBJECT_K = %u bytes\n", sizeof(OBJECT_K));
LOG_EX(LOG_Debug, "DT_OBJECT = %u bytes\n", sizeof(DT_OBJECT));
LOG_EX(LOG_Debug, "SVR_OBJECT = %u bytes\n", sizeof(SVR_OBJECT));
LOG_EX(LOG_Debug, "ADDR_OBJECT = %u bytes\n", sizeof(ADDR_OBJECT));
while((c = getopt_long(argc, argv, "hvir:s:p:", long_opts, &optidx)) != -1) {
switch(c) {
case 'v':
LOG_EX(LOG_Info, "User demo version: %s(%s)\n", sGATE_GIT_TAGS, sGATE_GIT_VERS);
break;
case '?':
case 'h':
LOG_EX2(LOG_Info, "Usage: %s [-h] [-v] ...\n", argv[0]);
LOG_EX2(LOG_Info, "options:\n");
LOG_EX2(LOG_Info, "\t-v, --version show program version\n");
LOG_EX2(LOG_Info, "\t-h, --help print this message\n");
LOG_EX2(LOG_Info, "\t-i, --interface test object add interface\n");
break;
//TODO 添加其它必要处理参数过程
case 'i':
test_add_object();
break;
case 's':
test_strsep(optarg);
break;
case 'r':
test_regex(optarg);
break;
case 'p':
test_ipaddr_format(optarg);
break;
}
}
while(TRUE) {
sleep(1);
}
return 0;
}
Reference in New Issue View Git Blame Copy Permalink