1110 lines
35 KiB
C++
1110 lines
35 KiB
C++
//
|
|
// Created by dongwenzhe on 2022/10/9.
|
|
//
|
|
#include <string>
|
|
#include <map>
|
|
using namespace std;
|
|
|
|
#include <sys/ioctl.h>
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
#include <net/if.h>
|
|
#include <arpa/inet.h>
|
|
#include <unistd.h>
|
|
#include <memory.h>
|
|
#include <cstdlib>
|
|
#include "opendhcpd.h"
|
|
#include "s2j/cJSON.h"
|
|
#include "haywire.h"
|
|
#include "misc.h"
|
|
#include <net/if_arp.h>
|
|
#include <libconfig.h>
|
|
#include <zlog.h>
|
|
#include "config.h"
|
|
#include "proto.h"
|
|
#include "user_errno.h"
|
|
#include "uthash/uthash.h"
|
|
|
|
extern data2 cfig;
|
|
extern bool kRunning;
|
|
extern dhcpMap dhcpCache;
|
|
extern time_t t;
|
|
|
|
static int dhcp_get_user_info(data19 *req, const char *pRequest) {
|
|
char logBuff[512];
|
|
const char *pStrContent;
|
|
dhcpMap::iterator p;
|
|
|
|
dzlog_debug("Input: %s\n", pRequest);
|
|
|
|
if (pRequest == nullptr || strlen(pRequest) == 0) {
|
|
sprintf(logBuff, "Requeset Json");
|
|
logDHCPMess(logBuff, 1);
|
|
return ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
pStrContent = proto_decode_context(pRequest, nullptr, nullptr);
|
|
|
|
if (pStrContent == nullptr) {
|
|
sprintf(logBuff, "Requeset Json error %s", pRequest);
|
|
logDHCPMess(logBuff, 1);
|
|
return ERR_PROTO_DECODE;
|
|
}
|
|
|
|
cJSON *pRoot = cJSON_Parse(pStrContent);
|
|
free((void *)pStrContent);
|
|
|
|
if (!pRoot) {
|
|
return ERR_JSON_PRASE_OBJ;
|
|
}
|
|
|
|
cJSON *pUserMac = cJSON_GetObjectItem(pRoot, "userMac");
|
|
|
|
if (!pUserMac) {
|
|
cJSON_Delete(pRoot);
|
|
return ERR_JSON_PRASE_OBJ;
|
|
}
|
|
|
|
req->memSize = (int)(2048 + (135 * dhcpCache.size()) + (cfig.dhcpSize * 26));
|
|
req->dp = (char *)calloc(1, req->memSize);
|
|
|
|
if (!req->dp) {
|
|
sprintf(logBuff, "Memory Error");
|
|
logDHCPMess(logBuff, 1);
|
|
return ERR_JSON_PRASE_OBJ;
|
|
}
|
|
|
|
// cJSON *pRspRoot = cJSON_CreateObject();
|
|
// cJSON_AddNumberToObject(pRspRoot, "version", 3);
|
|
// cJSON_AddNumberToObject(pRspRoot, "cryptoType", 0);
|
|
// cJSON_AddNumberToObject(pRspRoot, "timeStamp", (unsigned int)time(nullptr));
|
|
|
|
cJSON *pRspMsg = cJSON_CreateObject();
|
|
cJSON *pMsgArray = cJSON_CreateArray();
|
|
|
|
cJSON_AddItemToObject(pRspMsg, "userInfo", pMsgArray);
|
|
|
|
for (int i = 0; i < cJSON_GetArraySize(pUserMac); i++) {
|
|
char tempbuff[512];
|
|
cJSON *pItem = cJSON_GetArrayItem(pUserMac, i);
|
|
cJSON *pRspItem = cJSON_CreateObject();
|
|
|
|
p = dhcpCache.find(pItem->valuestring);
|
|
cJSON_AddStringToObject(pRspItem, "userMac", pItem->valuestring);
|
|
|
|
if (p != dhcpCache.end()) {
|
|
data7 *dhcpEntry = p->second;
|
|
|
|
cJSON_AddStringToObject(pRspItem, "ip", IP2String(tempbuff, dhcpEntry->ip));
|
|
cJSON_AddStringToObject(pRspItem, "hostname", dhcpEntry->hostname);
|
|
if (dhcpEntry->display && dhcpEntry->expiry >= t) {
|
|
tm *ttm = localtime(&dhcpEntry->expiry);
|
|
strftime(tempbuff, sizeof(tempbuff), "%d-%b-%y %X", ttm);
|
|
cJSON_AddStringToObject(pRspItem, "leaseExpiry", tempbuff);
|
|
} else {
|
|
cJSON_AddStringToObject(pRspItem, "leaseExpiry", "Expiry");
|
|
}
|
|
} else {
|
|
cJSON_AddStringToObject(pRspItem, "ip", "");
|
|
cJSON_AddStringToObject(pRspItem, "hostname", "");
|
|
}
|
|
|
|
cJSON_AddItemToArray(pMsgArray, pRspItem);
|
|
}
|
|
|
|
cJSON_AddNumberToObject(pRspMsg, "status", ERR_SUCCESS);
|
|
cJSON_AddStringToObject(pRspMsg, "message", getErrorEnumDesc(ERR_SUCCESS));
|
|
|
|
const char *pStrPro = proto_create_new(pRspMsg, 200);
|
|
|
|
//cJSON_AddItemToObject(pRspRoot, "msgContent", pRspMsg);
|
|
|
|
char *fp = req->dp;
|
|
//char *maxData = req->dp + (req->memSize - 512);
|
|
|
|
//fp += sprintf(fp, send200, strlen(rspBuf));
|
|
fp += sprintf(fp, "%s", pStrPro);
|
|
req->bytes = (int)(fp - req->dp);
|
|
|
|
cJSON_Delete(pRoot);
|
|
free((void *)pStrPro);
|
|
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
static int dhcp_get_all_user(data19 *req) {
|
|
char logBuff[512];
|
|
data7 *dhcpEntry;
|
|
dhcpMap::iterator p;
|
|
|
|
req->memSize = (int)(2048 + (135 * dhcpCache.size()) + (cfig.dhcpSize * 26));
|
|
req->dp = (char *)calloc(1, req->memSize);
|
|
|
|
if (!req->dp) {
|
|
sprintf(logBuff, "Memory Error");
|
|
logDHCPMess(logBuff, 1);
|
|
return ERR_MALLOC_MEMORY;
|
|
}
|
|
|
|
char *fp = req->dp;
|
|
char *maxData = req->dp + (req->memSize - 512);
|
|
|
|
cJSON *pRspMsg = cJSON_CreateObject();
|
|
cJSON *pMsgArray = cJSON_CreateArray();
|
|
|
|
cJSON_AddItemToObject(pRspMsg, "users", pMsgArray);
|
|
|
|
for (p = dhcpCache.begin(); kRunning && p != dhcpCache.end() && fp < maxData; p++) {
|
|
//cJSON *pRspItem = cJSON_CreateObject();
|
|
if ((dhcpEntry = p->second)) {
|
|
cJSON_AddStringToObject(pMsgArray, "", dhcpEntry->mapname);
|
|
}
|
|
|
|
//cJSON_AddItemToArray(pMsgArray, pRspItem);
|
|
}
|
|
|
|
//cJSON_AddItemToObject(pRspRoot, "msgContent", pRspMsg);
|
|
cJSON_AddNumberToObject(pRspMsg, "status", ERR_SUCCESS);
|
|
cJSON_AddStringToObject(pRspMsg, "message", getErrorEnumDesc(ERR_SUCCESS));
|
|
|
|
const char *pStrPro = proto_create_new(pRspMsg, 200);
|
|
|
|
fp += sprintf(fp, "%s", pStrPro);
|
|
req->bytes = (int)(fp - req->dp);
|
|
|
|
free((void *)pStrPro);
|
|
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
#define VALUE_TO_DHCP_TLV(buf, val, tag) \
|
|
do { \
|
|
int valSize; \
|
|
int numbytes = myTokenize((buf), (val), "/,.", true); \
|
|
if (numbytes <= 255) { \
|
|
char *ptr = (buf); \
|
|
valSize = 0; \
|
|
for (; *ptr; ptr = myGetToken(ptr, 1)) { \
|
|
if (isInt(ptr)) { \
|
|
hoption[valSize] = STR2INT(ptr); \
|
|
valSize++; \
|
|
} else { \
|
|
break; \
|
|
} \
|
|
} \
|
|
memcpy((val), hoption, valSize); \
|
|
if (buffSize > valSize + 2) { \
|
|
*dp = (tag); \
|
|
dp++; \
|
|
*dp = valSize; \
|
|
dp++; \
|
|
memcpy(dp, (val), valSize); \
|
|
dp += valSize; \
|
|
buffSize -= (valSize + 2); \
|
|
} \
|
|
} \
|
|
} while (0)
|
|
|
|
static int dhcp_add_rangeset_to_options(POBJ_DHCP_RNG pRange) {
|
|
int ret;
|
|
char buff[1024];
|
|
MYBYTE hoption[256];
|
|
MYBYTE *dp;
|
|
MYWORD buffSize = sizeof(dhcp_packet) - sizeof(dhcp_header);
|
|
char value[256] = {0};
|
|
MYBYTE *options = nullptr;
|
|
data20 optionData {};
|
|
|
|
if (!pRange) {
|
|
return -ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
memset(&optionData, 0, sizeof(data20));
|
|
|
|
optionData.rangeSetInd = cfig.rangeCount;
|
|
cfig.rangeSet[cfig.rangeCount].active = true;
|
|
|
|
dp = optionData.options;
|
|
*dp = 0;
|
|
dp++;
|
|
|
|
//dhcp_range
|
|
ret = addDHCPRange(pRange->rangAddr);
|
|
if (ret != ERR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
if (strlen(pRange->subnet) != 0) {
|
|
strcpy(value, pRange->subnet);
|
|
VALUE_TO_DHCP_TLV(buff, value, DHCP_OPTION_NETMASK);
|
|
optionData.mask = (*((MYDWORD *)value));
|
|
}
|
|
|
|
if (strlen(pRange->dnsSvr) != 0) {
|
|
strcpy(value, pRange->dnsSvr);
|
|
VALUE_TO_DHCP_TLV(buff, value, DHCP_OPTION_DNS);
|
|
}
|
|
|
|
if (strlen(pRange->gateway) != 0) {
|
|
strcpy(value, pRange->gateway);
|
|
VALUE_TO_DHCP_TLV(buff, value, DHCP_OPTION_ROUTER);
|
|
}
|
|
|
|
//lease
|
|
if (pRange->lease != 0) {
|
|
MYDWORD j;
|
|
j = pRange->lease;
|
|
if (buffSize > 6) {
|
|
*dp = DHCP_OPTION_IPADDRLEASE;
|
|
dp++;
|
|
*dp = 4;
|
|
dp++;
|
|
dp += pUInt(dp, j);
|
|
}
|
|
}
|
|
|
|
*dp = DHCP_OPTION_END;
|
|
dp++;
|
|
optionData.optionSize = (dp - optionData.options);
|
|
|
|
if (optionData.optionSize > 3) {
|
|
options = (MYBYTE *)calloc(1, optionData.optionSize);
|
|
memcpy(options, optionData.options, optionData.optionSize);
|
|
}
|
|
|
|
cfig.dhcpRanges[cfig.rangeCount].rangeSetInd = optionData.rangeSetInd;
|
|
cfig.dhcpRanges[cfig.rangeCount].options = options;
|
|
cfig.dhcpRanges[cfig.rangeCount].mask = optionData.mask;
|
|
if (!cfig.dhcpRanges[cfig.rangeCount].mask) {
|
|
cfig.dhcpRanges[cfig.rangeCount].mask = cfig.mask;
|
|
}
|
|
cfig.rangeCount = (char)(cfig.rangeCount + 1);
|
|
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
static int add_dhcpd_rangeset(data19 *req, const char *pRequest) {
|
|
char logBuff[512];
|
|
const char *pStrContent;
|
|
OBJ_DHCP_RNG range;
|
|
char *fp;
|
|
cJSON *pRspRoot;
|
|
cJSON *pExpandArray;
|
|
|
|
dzlog_debug("Input: %s\n", pRequest);
|
|
|
|
if (pRequest == nullptr || strlen(pRequest) == 0) {
|
|
sprintf(logBuff, "Requeset Json");
|
|
logDHCPMess(logBuff, 1);
|
|
return ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
pStrContent = proto_decode_context(pRequest, nullptr, nullptr);
|
|
|
|
if (pStrContent == nullptr) {
|
|
sprintf(logBuff, "Requeset Json error %s", pRequest);
|
|
logDHCPMess(logBuff, 1);
|
|
return ERR_PROTO_DECODE;
|
|
}
|
|
|
|
cJSON *pRoot = cJSON_Parse(pStrContent);
|
|
free((void *)pStrContent);
|
|
|
|
if (!pRoot) {
|
|
return ERR_JSON_PRASE_OBJ;
|
|
}
|
|
|
|
cJSON *prange_set = cJSON_GetObjectItem(pRoot, "rangeSet");
|
|
|
|
req->memSize = (int)(2048 + (135 * dhcpCache.size()) + (cfig.dhcpSize * 26));
|
|
req->dp = (char *)calloc(1, req->memSize);
|
|
|
|
if (!req->dp) {
|
|
sprintf(logBuff, "Memory Error");
|
|
logDHCPMess(logBuff, 1);
|
|
cJSON_Delete(pRoot);
|
|
return ERR_MALLOC_MEMORY;
|
|
}
|
|
|
|
fp = req->dp;
|
|
pRspRoot = cJSON_CreateObject();
|
|
pExpandArray = cJSON_CreateArray();
|
|
cJSON_AddItemToObject(pRspRoot, "rangeSet", pExpandArray);
|
|
|
|
for (int i = 0; i < cJSON_GetArraySize(prange_set); i++) {
|
|
int ret;
|
|
cJSON *pItem = cJSON_GetArrayItem(prange_set, i);
|
|
cJSON *pdhcp_range = cJSON_GetObjectItem(pItem, "dhcpRange");
|
|
cJSON *pEx_range = cJSON_CreateObject();
|
|
|
|
if (!pdhcp_range) {
|
|
continue;
|
|
}
|
|
|
|
cJSON_AddStringToObject(pEx_range, "dhcpRange", pdhcp_range->valuestring);
|
|
|
|
cJSON *psubnet_mask = cJSON_GetObjectItem(pItem, "netmask");
|
|
cJSON *pdomain_server = cJSON_GetObjectItem(pItem, "domainServer");
|
|
cJSON *pgateway = cJSON_GetObjectItem(pItem, "gateway");
|
|
cJSON *please_time = cJSON_GetObjectItem(pItem, "leaseTime");
|
|
|
|
memset(&range, 0, sizeof(OBJ_DHCP_RNG));
|
|
|
|
strcpy(range.rangAddr, pdhcp_range->valuestring);
|
|
if (psubnet_mask) {
|
|
strcpy(range.subnet, psubnet_mask->valuestring);
|
|
}
|
|
|
|
if (pdomain_server) {
|
|
strcpy(range.dnsSvr, pdomain_server->valuestring);
|
|
}
|
|
|
|
if (pgateway) {
|
|
strcpy(range.gateway, pgateway->valuestring);
|
|
}
|
|
|
|
if (please_time) {
|
|
range.lease = please_time->valueint;
|
|
}
|
|
|
|
//写入cfig
|
|
ret = dhcp_add_rangeset_to_options(&range);
|
|
if (ret != ERR_SUCCESS) {
|
|
cJSON_AddNumberToObject(pEx_range, "status", ret);
|
|
cJSON_AddStringToObject(pEx_range, "message", getErrorEnumDesc(ret));
|
|
cJSON_AddItemToArray(pExpandArray, pEx_range);
|
|
continue;
|
|
}
|
|
|
|
cJSON_AddNumberToObject(pEx_range, "status", ERR_SUCCESS);
|
|
cJSON_AddStringToObject(pEx_range, "message", getErrorEnumDesc(ERR_SUCCESS));
|
|
cJSON_AddItemToArray(pExpandArray, pEx_range);
|
|
}
|
|
|
|
const char *pStrPro = proto_create_new(pRspRoot, 200);
|
|
|
|
fp += sprintf(fp, "%s", pStrPro);
|
|
|
|
cJSON_Delete(pRoot);
|
|
|
|
req->bytes = (int)(fp - req->dp);
|
|
free((void *)pStrPro);
|
|
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
struct hash_map {
|
|
unsigned int key;
|
|
unsigned int value;
|
|
UT_hash_handle hh;
|
|
};
|
|
|
|
static int delete_dhcpd_rangeset(data19 *req, const char *pRequest) {
|
|
char logBuff[512];
|
|
const char *pStrContent;
|
|
char *fp;
|
|
cJSON *pRspRoot;
|
|
cJSON *pdelArray;
|
|
data13 dhcpRanges[MAX_DHCP_RANGES];
|
|
hash_map *delMap = nullptr;
|
|
int resCount = 0;
|
|
|
|
dzlog_debug("Input: %s\n", pRequest);
|
|
|
|
if (pRequest == nullptr || strlen(pRequest) == 0) {
|
|
sprintf(logBuff, "Requeset Json");
|
|
logDHCPMess(logBuff, 1);
|
|
return ERR_INPUT_PARAMS;
|
|
}
|
|
|
|
pStrContent = proto_decode_context(pRequest, nullptr, nullptr);
|
|
if (pStrContent == nullptr) {
|
|
sprintf(logBuff, "Requeset Json error %s", pRequest);
|
|
logDHCPMess(logBuff, 1);
|
|
return ERR_PROTO_DECODE;
|
|
}
|
|
|
|
cJSON *pRoot = cJSON_Parse(pStrContent);
|
|
free((void *)pStrContent);
|
|
if (!pRoot) {
|
|
return ERR_JSON_PRASE_OBJ;
|
|
}
|
|
|
|
cJSON *pdhcp_range = cJSON_GetObjectItem(pRoot, "dhcpRange");
|
|
|
|
req->memSize = (int)(2048 + (135 * dhcpCache.size()) + (cfig.dhcpSize * 26));
|
|
req->dp = (char *)calloc(1, req->memSize);
|
|
|
|
if (!req->dp) {
|
|
sprintf(logBuff, "Memory Error");
|
|
logDHCPMess(logBuff, 1);
|
|
cJSON_Delete(pRoot);
|
|
return ERR_MALLOC_MEMORY;
|
|
}
|
|
|
|
fp = req->dp;
|
|
pRspRoot = cJSON_CreateObject();
|
|
pdelArray = cJSON_CreateArray();
|
|
cJSON_AddItemToObject(pRspRoot, "rangeSet", pdelArray);
|
|
|
|
for (int i = 0; i < cJSON_GetArraySize(pdhcp_range); i++) {
|
|
cJSON *pdel_Item = cJSON_CreateObject();
|
|
cJSON *pdelRange = cJSON_GetArrayItem(pdhcp_range, i);
|
|
char del_range[256];
|
|
auto *delItem = (struct hash_map *)malloc(sizeof(hash_map));
|
|
|
|
if (!pdelRange) {
|
|
continue;
|
|
}
|
|
|
|
strcpy(del_range, pdelRange->valuestring);
|
|
|
|
MYDWORD st_addr;
|
|
MYDWORD en_addr;
|
|
char start[128];
|
|
char end[128];
|
|
mySplit(start, end, del_range, '-');
|
|
st_addr = htonl(inet_addr(start));
|
|
en_addr = htonl(inet_addr(end));
|
|
if(en_addr == 0) {
|
|
cJSON_AddStringToObject(pdel_Item, "dhcpRange", del_range);
|
|
|
|
cJSON_AddNumberToObject(pdel_Item, "status", ERR_INPUT_PARAMS);
|
|
cJSON_AddStringToObject(pdel_Item, "message", getErrorEnumDesc(ERR_INPUT_PARAMS));
|
|
cJSON_AddItemToArray(pdelArray, pdel_Item);
|
|
}
|
|
|
|
HASH_FIND_INT(delMap, &st_addr, delItem);
|
|
if (delItem == nullptr) {
|
|
auto *s = (struct hash_map *)malloc(sizeof(struct hash_map));
|
|
s->key = st_addr;
|
|
s->value = en_addr;
|
|
HASH_ADD_INT(delMap, key, s);
|
|
}
|
|
}
|
|
|
|
for (int i = 0; i < cfig.rangeCount; i++) {
|
|
cJSON *pdel_Item = cJSON_CreateObject();
|
|
hash_map *delRange;
|
|
char del_range[256];
|
|
char saddr[128];
|
|
char eaddr[128];
|
|
memset(del_range, 0, 256);
|
|
|
|
HASH_FIND_INT(delMap, &cfig.dhcpRanges[i].rangeStart, delRange);
|
|
if (delRange != nullptr) {
|
|
IP2String(saddr, ntohl(delRange->key));
|
|
IP2String(eaddr, ntohl(delRange->value));
|
|
|
|
sprintf(del_range, "%s-%s", saddr, eaddr);
|
|
cJSON_AddStringToObject(pdel_Item, "dhcpRange", del_range);
|
|
//judge whether the input has error
|
|
if(delRange->value == cfig.dhcpRanges[i].rangeEnd) {
|
|
cJSON_AddNumberToObject(pdel_Item, "status", ERR_SUCCESS);
|
|
cJSON_AddStringToObject(pdel_Item, "message", getErrorEnumDesc(ERR_SUCCESS));
|
|
cJSON_AddItemToArray(pdelArray, pdel_Item);
|
|
|
|
} else {
|
|
memcpy(&dhcpRanges[resCount], &cfig.dhcpRanges[i], sizeof(struct data13));
|
|
resCount++;
|
|
cJSON_AddNumberToObject(pdel_Item, "status", ERR_INPUT_PARAMS);
|
|
cJSON_AddStringToObject(pdel_Item, "message", getErrorEnumDesc(ERR_INPUT_PARAMS));
|
|
cJSON_AddItemToArray(pdelArray, pdel_Item);
|
|
}
|
|
|
|
HASH_DEL(delMap, delRange);
|
|
} else {
|
|
//write to dhcpRanges
|
|
memcpy(&dhcpRanges[resCount], &cfig.dhcpRanges[i], sizeof(struct data13));
|
|
resCount++;
|
|
}
|
|
}
|
|
|
|
//输入参数不存在的情况
|
|
if(resCount > cfig.rangeCount - cJSON_GetArraySize(pdhcp_range)){
|
|
cJSON *pdel_Item = cJSON_CreateObject();
|
|
hash_map *s = (struct hash_map*)malloc(sizeof(struct hash_map));
|
|
hash_map *tmp;
|
|
char saddr[128];
|
|
char eaddr[128];
|
|
char del_range[256];
|
|
memset(del_range, 0, 256);
|
|
|
|
HASH_ITER(hh, delMap, s, tmp){
|
|
IP2String(saddr, ntohl(s->key));
|
|
IP2String(eaddr, ntohl(s->value));
|
|
|
|
sprintf(del_range, "%s-%s", saddr, eaddr);
|
|
cJSON_AddStringToObject(pdel_Item, "dhcpRange", del_range);
|
|
|
|
cJSON_AddNumberToObject(pdel_Item, "status", ERR_ITEM_UNEXISTS);
|
|
cJSON_AddStringToObject(pdel_Item, "message", getErrorEnumDesc(ERR_ITEM_UNEXISTS));
|
|
cJSON_AddItemToArray(pdelArray, pdel_Item);
|
|
}
|
|
}
|
|
|
|
//rewite cfig.dhcpRanges
|
|
for (int i = 0; i < cfig.rangeCount; i++) {
|
|
if (i < resCount) {
|
|
memcpy(&cfig.dhcpRanges[i], &dhcpRanges[i], sizeof(struct data13));
|
|
} else {
|
|
memset(&cfig.dhcpRanges[i], 0, sizeof(struct data13));
|
|
}
|
|
}
|
|
|
|
cfig.rangeCount = (char)resCount;
|
|
|
|
const char *pStrPro = proto_create_new(pRspRoot, 200);
|
|
fp += sprintf(fp, "%s", pStrPro);
|
|
|
|
cJSON_Delete(pRoot);
|
|
req->bytes = (int)(fp - req->dp);
|
|
free((void *)pStrPro);
|
|
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
static void revert(unsigned int *num) {
|
|
unsigned int v = *num;
|
|
v = ((v & 0x000000FF) << 24) | ((v & 0x0000FF00) << 8) | ((v & 0x00FF0000) >> 8) | ((v & 0xFF000000) >> 24);
|
|
*num = v;
|
|
}
|
|
|
|
static int query_dhcpd_rangeset(data19 *req) {
|
|
char logBuff[512];
|
|
req->memSize = (int)(2048 + (135 * dhcpCache.size()) + (cfig.dhcpSize * 26));
|
|
req->dp = (char *)calloc(1, req->memSize);
|
|
|
|
if (!req->dp) {
|
|
sprintf(logBuff, "Memory Error");
|
|
logDHCPMess(logBuff, 1);
|
|
return ERR_MALLOC_MEMORY;
|
|
}
|
|
|
|
char *fp = req->dp;
|
|
char *maxData = req->dp + (req->memSize - 512);
|
|
|
|
cJSON *pRspMsg = cJSON_CreateObject();
|
|
cJSON *pMsgArray = cJSON_CreateArray();
|
|
cJSON_AddItemToObject(pRspMsg, "rangeSet", pMsgArray);
|
|
|
|
for (char rangeInd = 0; rangeInd < cfig.rangeCount; rangeInd++) {
|
|
char addrStart[64];
|
|
char addrEnd[64];
|
|
char rangeSet[128];
|
|
char rangeMask[128];
|
|
char domainServer[128];
|
|
char gateway[128];
|
|
cJSON *pRangeItem = cJSON_CreateObject();
|
|
unsigned int lease;
|
|
|
|
memset(domainServer, 0, 128);
|
|
IP2String(addrStart, ntohl(cfig.dhcpRanges[rangeInd].rangeStart));
|
|
IP2String(addrEnd, ntohl(cfig.dhcpRanges[rangeInd].rangeEnd));
|
|
IP2String(rangeMask, cfig.dhcpRanges[rangeInd].mask);
|
|
|
|
sprintf(rangeSet, "%s-%s", addrStart, addrEnd);
|
|
cJSON_AddStringToObject(pRangeItem, "dhcpRange", rangeSet);
|
|
cJSON_AddStringToObject(pRangeItem, "netmask", rangeMask);
|
|
|
|
MYBYTE *opPointer = cfig.dhcpRanges[rangeInd].options;
|
|
data3 op {};
|
|
if (opPointer) {
|
|
opPointer++;
|
|
while (*opPointer && *opPointer != DHCP_OPTION_END) {
|
|
unsigned int tmpVal = 0;
|
|
unsigned char dnsSize;
|
|
unsigned char offset = 0;
|
|
char dns_op[64];
|
|
|
|
op.opt_code = *opPointer;
|
|
opPointer++;
|
|
op.size = *opPointer;
|
|
opPointer++;
|
|
|
|
memcpy(op.value, opPointer, op.size);
|
|
if (op.opt_code == DHCP_OPTION_DNS) {
|
|
dnsSize = op.size;
|
|
do {
|
|
tmpVal = fIP(op.value + offset);
|
|
revert(&tmpVal);
|
|
IP2String(dns_op, ntohl(tmpVal));
|
|
sprintf(domainServer, "%s%s", domainServer, dns_op);
|
|
if (dnsSize != 4) {
|
|
sprintf(domainServer, "%s,", domainServer);
|
|
}
|
|
dnsSize -= 4;
|
|
offset = op.size - dnsSize;
|
|
} while (dnsSize != 0);
|
|
|
|
cJSON_AddStringToObject(pRangeItem, "domainServer", domainServer);
|
|
} else if (op.opt_code == DHCP_OPTION_ROUTER) {
|
|
tmpVal = fIP(op.value);
|
|
revert(&tmpVal);
|
|
IP2String(gateway, ntohl(tmpVal));
|
|
|
|
cJSON_AddStringToObject(pRangeItem, "gateway", gateway);
|
|
} else if (op.opt_code == DHCP_OPTION_IPADDRLEASE) {
|
|
lease = fUInt(op.value);
|
|
|
|
cJSON_AddNumberToObject(pRangeItem, "lease", lease);
|
|
}
|
|
opPointer += op.size;
|
|
}
|
|
}
|
|
|
|
cJSON_AddItemToArray(pMsgArray, pRangeItem);
|
|
}
|
|
cJSON_AddNumberToObject(pRspMsg, "status", ERR_SUCCESS);
|
|
cJSON_AddStringToObject(pRspMsg, "message", getErrorEnumDesc(ERR_SUCCESS));
|
|
|
|
const char *pStrPro = proto_create_new(pRspMsg, 200);
|
|
|
|
fp += sprintf(fp, "%s", pStrPro);
|
|
req->bytes = (int)(fp - req->dp);
|
|
|
|
free((void *)pStrPro);
|
|
return ERR_SUCCESS;
|
|
}
|
|
|
|
#pragma clang diagnostic push
|
|
#pragma ide diagnostic ignored "cert-err34-c"
|
|
|
|
int getHwAddr(char *buff, char *mac) {
|
|
if (buff == nullptr || mac == nullptr) {
|
|
return -1;
|
|
}
|
|
|
|
int i;
|
|
unsigned int p[6];
|
|
|
|
if (sscanf(mac, "%x:%x:%x:%x:%x:%x", &p[0], &p[1], &p[2], &p[3], &p[4], &p[5]) < 6) {
|
|
return -1;
|
|
}
|
|
|
|
for (i = 0; i < 6; i++) {
|
|
buff[i] = (char)p[i];
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
#pragma clang diagnostic pop
|
|
|
|
int arpSet(const char *ifname, char *ipStr, char *mac) {
|
|
if (ifname == nullptr || ipStr == nullptr || mac == nullptr) {
|
|
dzlog_error("para is null.\n");
|
|
return -1;
|
|
}
|
|
|
|
struct arpreq req {};
|
|
struct sockaddr_in *sin;
|
|
int ret;
|
|
int sock_fd;
|
|
|
|
memset(&req, 0, sizeof(struct arpreq));
|
|
sin = (struct sockaddr_in *)&req.arp_pa;
|
|
sin->sin_family = AF_INET;
|
|
sin->sin_addr.s_addr = inet_addr(ipStr);
|
|
//arp_dev长度为[16]，注意越界
|
|
strncpy(req.arp_dev, ifname, 15);
|
|
req.arp_flags = ATF_PERM | ATF_COM;
|
|
|
|
if (getHwAddr((char *)req.arp_ha.sa_data, mac) < 0) {
|
|
dzlog_error("get mac error.\n");
|
|
return -1;
|
|
}
|
|
|
|
sock_fd = socket(AF_INET, SOCK_DGRAM, 0);
|
|
if (sock_fd < 0) {
|
|
dzlog_error("get socket error.\n");
|
|
return -1;
|
|
}
|
|
|
|
ret = ioctl(sock_fd, SIOCSARP, &req);
|
|
if (ret < 0) {
|
|
dzlog_error("ioctl error.\n");
|
|
close(sock_fd);
|
|
return -1;
|
|
}
|
|
|
|
close(sock_fd);
|
|
return 0;
|
|
}
|
|
|
|
sockaddr_in get_cliAddr(char *nicif, char *tempbuff, data9 *req) {
|
|
arpSet(nicif, IP2String(tempbuff, req->dhcpp.header.bp_yiaddr), req->chaddr);
|
|
|
|
sockaddr_in cliAddr {};
|
|
memcpy(&cliAddr, &req->remote, sizeof(sockaddr_in));
|
|
cliAddr.sin_addr.s_addr = inet_addr(IP2String(tempbuff, req->dhcpp.header.bp_yiaddr));
|
|
return cliAddr;
|
|
}
|
|
|
|
static void response_complete(void *user_data) {
|
|
auto *req = (data19 *)user_data;
|
|
if (req) {
|
|
|
|
if (req->dp) {
|
|
free(req->dp);
|
|
}
|
|
|
|
free(req);
|
|
}
|
|
}
|
|
|
|
static void proto_response_error(hw_http_response *response, int httpCode, const char *httpCodeStr, int errCode) {
|
|
cJSON *pRspMsg = cJSON_CreateObject();
|
|
|
|
cJSON_AddNumberToObject(pRspMsg, "status", errCode);
|
|
cJSON_AddStringToObject(pRspMsg, "message", getErrorEnumDesc(errCode));
|
|
|
|
const char *pStrPro = proto_create_new(pRspMsg, httpCode);
|
|
|
|
hw_http_response_send_error(response, httpCodeStr, pStrPro);
|
|
free((void *)pStrPro);
|
|
}
|
|
|
|
static void opendhcp_http_info(http_request *request, hw_http_response *response, void *UNUSED(user_data)) {
|
|
hw_string status_code;
|
|
hw_string content_type_name;
|
|
hw_string content_type_value;
|
|
hw_string body;
|
|
hw_string keep_alive_name;
|
|
hw_string keep_alive_value;
|
|
|
|
auto *req = (data19 *)malloc(sizeof(struct data19));
|
|
|
|
if (req == nullptr) {
|
|
proto_response_error(response, 500, HTTP_STATUS_500, ERR_MALLOC_MEMORY);
|
|
return;
|
|
}
|
|
|
|
if (request->method != HW_HTTP_GET) {
|
|
proto_response_error(response, 405, HTTP_STATUS_405, ERR_HTTP_UNSUP_METHOD);
|
|
return;
|
|
}
|
|
|
|
memset(req, 0, sizeof(struct data19));
|
|
|
|
SETSTRING(content_type_name, "Content-Type");
|
|
SETSTRING(content_type_value, "text/html");
|
|
hw_set_response_header(response, &content_type_name, &content_type_value);
|
|
|
|
SETSTRING(status_code, HTTP_STATUS_200);
|
|
prepareUserHtmlRespStatus(req);
|
|
SETSTRING(body, req->dp);
|
|
hw_set_body(response, &body);
|
|
hw_set_response_status_code(response, &status_code);
|
|
|
|
if (request->keep_alive) {
|
|
SETSTRING(keep_alive_name, "Connection");
|
|
SETSTRING(keep_alive_value, "close");
|
|
hw_set_response_header(response, &keep_alive_name, &keep_alive_value);
|
|
} else {
|
|
hw_set_http_version(response, 1, 0);
|
|
}
|
|
|
|
hw_http_response_send(response, req, response_complete);
|
|
}
|
|
|
|
static void opendhcp_http_get_userinfo(http_request *request, hw_http_response *response, void *UNUSED(user_data)) {
|
|
hw_string status_code;
|
|
hw_string content_type_name;
|
|
hw_string content_type_value;
|
|
hw_string body;
|
|
hw_string keep_alive_name;
|
|
hw_string keep_alive_value;
|
|
int ret;
|
|
|
|
auto *req = (data19 *)malloc(sizeof(struct data19));
|
|
|
|
if (req == nullptr) {
|
|
proto_response_error(response, 500, HTTP_STATUS_500, ERR_MALLOC_MEMORY);
|
|
return;
|
|
}
|
|
|
|
if (request->method != HW_HTTP_POST) {
|
|
proto_response_error(response, 405, HTTP_STATUS_405, ERR_HTTP_UNSUP_METHOD);
|
|
return;
|
|
}
|
|
|
|
memset(req, 0, sizeof(struct data19));
|
|
|
|
SETSTRING(content_type_name, "Content-Type");
|
|
SETSTRING(content_type_value, "application/json");
|
|
hw_set_response_header(response, &content_type_name, &content_type_value);
|
|
|
|
SETSTRING(status_code, HTTP_STATUS_200);
|
|
ret = dhcp_get_user_info(req, request->body->value);
|
|
if (ret != ERR_SUCCESS) {
|
|
proto_response_error(response, 500, HTTP_STATUS_500, ret);
|
|
return;
|
|
}
|
|
SETSTRING(body, req->dp);
|
|
hw_set_body(response, &body);
|
|
hw_set_response_status_code(response, &status_code);
|
|
|
|
if (request->keep_alive) {
|
|
SETSTRING(keep_alive_name, "Connection");
|
|
SETSTRING(keep_alive_value, "close");
|
|
hw_set_response_header(response, &keep_alive_name, &keep_alive_value);
|
|
} else {
|
|
hw_set_http_version(response, 1, 0);
|
|
}
|
|
|
|
hw_http_response_send(response, req, response_complete);
|
|
}
|
|
|
|
static void opendhcp_http_get_alluser(http_request *request, hw_http_response *response, void *UNUSED(user_data)) {
|
|
hw_string status_code;
|
|
hw_string content_type_name;
|
|
hw_string content_type_value;
|
|
hw_string body;
|
|
hw_string keep_alive_name;
|
|
hw_string keep_alive_value;
|
|
int ret;
|
|
auto *req = (data19 *)malloc(sizeof(struct data19));
|
|
|
|
if (req == nullptr) {
|
|
proto_response_error(response, 500, HTTP_STATUS_500, ERR_MALLOC_MEMORY);
|
|
return;
|
|
}
|
|
|
|
if (request->method != HW_HTTP_GET) {
|
|
proto_response_error(response, 405, HTTP_STATUS_405, ERR_HTTP_UNSUP_METHOD);
|
|
return;
|
|
}
|
|
|
|
memset(req, 0, sizeof(struct data19));
|
|
|
|
SETSTRING(content_type_name, "Content-Type");
|
|
SETSTRING(content_type_value, "application/json");
|
|
hw_set_response_header(response, &content_type_name, &content_type_value);
|
|
|
|
SETSTRING(status_code, HTTP_STATUS_200);
|
|
ret = dhcp_get_all_user(req);
|
|
if (ret != ERR_SUCCESS) {
|
|
proto_response_error(response, 500, HTTP_STATUS_500, ret);
|
|
return;
|
|
}
|
|
SETSTRING(body, req->dp);
|
|
hw_set_body(response, &body);
|
|
hw_set_response_status_code(response, &status_code);
|
|
|
|
if (request->keep_alive) {
|
|
SETSTRING(keep_alive_name, "Connection");
|
|
SETSTRING(keep_alive_value, "close");
|
|
hw_set_response_header(response, &keep_alive_name, &keep_alive_value);
|
|
} else {
|
|
hw_set_http_version(response, 1, 0);
|
|
}
|
|
|
|
hw_http_response_send(response, req, response_complete);
|
|
}
|
|
|
|
static void opendhcp_http_add_rangeset(http_request *request, hw_http_response *response, void *UNUSED(user_data)) {
|
|
hw_string status_code;
|
|
hw_string content_type_name;
|
|
hw_string content_type_value;
|
|
hw_string body;
|
|
hw_string keep_alive_name;
|
|
hw_string keep_alive_value;
|
|
int ret;
|
|
auto *req = (data19 *)malloc(sizeof(struct data19));
|
|
|
|
if (req == nullptr) {
|
|
proto_response_error(response, 500, HTTP_STATUS_500, ERR_MALLOC_MEMORY);
|
|
return;
|
|
}
|
|
|
|
if (request->method != HW_HTTP_POST) {
|
|
proto_response_error(response, 405, HTTP_STATUS_405, ERR_HTTP_UNSUP_METHOD);
|
|
return;
|
|
}
|
|
|
|
memset(req, 0, sizeof(struct data19));
|
|
SETSTRING(content_type_name, "Content-Type");
|
|
SETSTRING(content_type_value, "application/json");
|
|
hw_set_response_header(response, &content_type_name, &content_type_value);
|
|
|
|
SETSTRING(status_code, HTTP_STATUS_200);
|
|
ret = add_dhcpd_rangeset(req, request->body->value);
|
|
if (ret != ERR_SUCCESS) {
|
|
proto_response_error(response, 500, HTTP_STATUS_500, ret);
|
|
return;
|
|
}
|
|
|
|
SETSTRING(body, req->dp);
|
|
hw_set_body(response, &body);
|
|
hw_set_response_status_code(response, &status_code);
|
|
|
|
if (request->keep_alive) {
|
|
SETSTRING(keep_alive_name, "Connection");
|
|
SETSTRING(keep_alive_value, "close");
|
|
hw_set_response_header(response, &keep_alive_name, &keep_alive_value);
|
|
} else {
|
|
hw_set_http_version(response, 1, 0);
|
|
}
|
|
|
|
hw_http_response_send(response, req, response_complete);
|
|
}
|
|
|
|
static void opendhcp_http_delete_rangeset(http_request *request, hw_http_response *response, void *UNUSED(user_data)) {
|
|
hw_string status_code;
|
|
hw_string content_type_name;
|
|
hw_string content_type_value;
|
|
hw_string body;
|
|
hw_string keep_alive_name;
|
|
hw_string keep_alive_value;
|
|
int ret;
|
|
auto *req = (data19 *)malloc(sizeof(struct data19));
|
|
|
|
if (req == nullptr) {
|
|
proto_response_error(response, 500, HTTP_STATUS_500, ERR_MALLOC_MEMORY);
|
|
return;
|
|
}
|
|
|
|
if (request->method != HW_HTTP_POST) {
|
|
proto_response_error(response, 405, HTTP_STATUS_405, ERR_HTTP_UNSUP_METHOD);
|
|
return;
|
|
}
|
|
|
|
memset(req, 0, sizeof(struct data19));
|
|
SETSTRING(content_type_name, "Content-Type");
|
|
SETSTRING(content_type_value, "application/json");
|
|
hw_set_response_header(response, &content_type_name, &content_type_value);
|
|
|
|
SETSTRING(status_code, HTTP_STATUS_200);
|
|
ret = delete_dhcpd_rangeset(req, request->body->value);
|
|
if (ret != ERR_SUCCESS) {
|
|
proto_response_error(response, 500, HTTP_STATUS_500, ret);
|
|
return;
|
|
}
|
|
|
|
SETSTRING(body, req->dp);
|
|
hw_set_body(response, &body);
|
|
hw_set_response_status_code(response, &status_code);
|
|
|
|
if (request->keep_alive) {
|
|
SETSTRING(keep_alive_name, "Connection");
|
|
SETSTRING(keep_alive_value, "close");
|
|
hw_set_response_header(response, &keep_alive_name, &keep_alive_value);
|
|
} else {
|
|
hw_set_http_version(response, 1, 0);
|
|
}
|
|
|
|
hw_http_response_send(response, req, response_complete);
|
|
}
|
|
|
|
static void opendhcp_http_query_rangeset(http_request *request, hw_http_response *response, void *UNUSED(user_data)) {
|
|
hw_string status_code;
|
|
hw_string content_type_name;
|
|
hw_string content_type_value;
|
|
hw_string body;
|
|
hw_string keep_alive_name;
|
|
hw_string keep_alive_value;
|
|
int ret;
|
|
auto *req = (data19 *)malloc(sizeof(struct data19));
|
|
|
|
if (req == nullptr) {
|
|
proto_response_error(response, 500, HTTP_STATUS_500, ERR_MALLOC_MEMORY);
|
|
return;
|
|
}
|
|
|
|
if (request->method != HW_HTTP_GET) {
|
|
proto_response_error(response, 405, HTTP_STATUS_405, ERR_HTTP_UNSUP_METHOD);
|
|
return;
|
|
}
|
|
|
|
memset(req, 0, sizeof(struct data19));
|
|
SETSTRING(content_type_name, "Content-Type");
|
|
SETSTRING(content_type_value, "application/json");
|
|
hw_set_response_header(response, &content_type_name, &content_type_value);
|
|
|
|
SETSTRING(status_code, HTTP_STATUS_200);
|
|
ret = query_dhcpd_rangeset(req);
|
|
if (ret != ERR_SUCCESS) {
|
|
proto_response_error(response, 500, HTTP_STATUS_500, ret);
|
|
return;
|
|
}
|
|
|
|
SETSTRING(body, req->dp);
|
|
hw_set_body(response, &body);
|
|
hw_set_response_status_code(response, &status_code);
|
|
|
|
if (request->keep_alive) {
|
|
SETSTRING(keep_alive_name, "Connection");
|
|
SETSTRING(keep_alive_value, "close");
|
|
hw_set_response_header(response, &keep_alive_name, &keep_alive_value);
|
|
} else {
|
|
hw_set_http_version(response, 1, 0);
|
|
}
|
|
|
|
hw_http_response_send(response, req, response_complete);
|
|
}
|
|
|
|
/**
|
|
* 添加配置文件监听接口
|
|
* @return
|
|
*/
|
|
int opendhcp_add_listener() {
|
|
int i;
|
|
vector listen_ip = config_get_dhcp_listen_on();
|
|
|
|
for (i = 0; listen_ip && i < vect_size(listen_ip); i++) {
|
|
const char *pIp = (const char *)vect_get_at(listen_ip, i);
|
|
if (pIp && strlen(pIp) > 0) {
|
|
MYDWORD addr = inet_addr(pIp);
|
|
addServer(cfig.specifiedServers, MAX_SERVERS, addr);
|
|
}
|
|
}
|
|
|
|
return i;
|
|
}
|
|
|
|
/**
|
|
* 增加 DHCP Server HTTP服务接口
|
|
*/
|
|
void opendhcp_init_http_server() {
|
|
static int added = FALSE;
|
|
|
|
if (!added) {
|
|
hw_http_add_route("/", opendhcp_http_info, nullptr);
|
|
hw_http_add_route("dhcp/info/getuser", opendhcp_http_get_userinfo, nullptr);
|
|
hw_http_add_route("dhcp/info/allusers", opendhcp_http_get_alluser, nullptr);
|
|
hw_http_add_route("dhcp/config/rangeset", opendhcp_http_add_rangeset, nullptr);
|
|
hw_http_add_route("dhcp/delete/rangeset", opendhcp_http_delete_rangeset, nullptr);
|
|
hw_http_add_route("dhcp/query/rangeset", opendhcp_http_query_rangeset, nullptr);
|
|
added = TRUE;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 增加 DHCP 主、从服务器配置
|
|
*/
|
|
void opendhcp_set_replication_svr() {
|
|
vector replication = config_get_dhcp_replication_svr();
|
|
|
|
if (replication && vect_size(replication) == 2) {
|
|
cfig.zoneServers[0] = inet_addr((const char *)vect_get_at(replication, 0));
|
|
cfig.zoneServers[1] = inet_addr((const char *)vect_get_at(replication, 1));
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 添加配置文件中的 DHCP 地址池池配置到服务中
|
|
*/
|
|
void opendhcp_add_ip_pool_set() {
|
|
vector pool = config_get_dhcp_server_range_set();
|
|
|
|
for (int i = 0; (pool && i < vect_size(pool)); i++) {
|
|
auto pRange = (POBJ_DHCP_RNG)vect_get_at(pool, i);
|
|
|
|
if (pRange) {
|
|
if (dhcp_add_rangeset_to_options(pRange) != ERR_SUCCESS) {
|
|
dzlog_error("Add rangeset(\"%s\") failed!", pRange->rangAddr);
|
|
}
|
|
}
|
|
}
|
|
} |