116 lines
3.3 KiB
C
116 lines
3.3 KiB
C
//
|
||
// Created by xajhuang on 2023/3/23.
|
||
//
|
||
|
||
#include <time.h>
|
||
#include "lease.h"
|
||
#include "user_errno.h"
|
||
#include "zlog_module.h"
|
||
#include "database.h"
|
||
#include "user_mgr.h"
|
||
#include "rfc2131.h"
|
||
#include "dhcp_network.h"
|
||
#include "db_interface.h"
|
||
|
||
#define PREALLOC_IP_TIMEOUT (60)
|
||
|
||
typedef struct {
|
||
U32 ipAddr;
|
||
U8 macAddr[ETH_ALEN];
|
||
U32 bitset;
|
||
PIPPOOL_INFO pCtx;
|
||
U32 timeStamp;
|
||
UT_hash_handle hh;
|
||
} PRE_ALLOC_IP, *PPRE_ALLOC_IP;
|
||
|
||
//static PMAC_FILTER g_allowTbl = NULL;
|
||
//static PMAC_FILTER g_blackListTbl = NULL;
|
||
//static PPRE_ALLOC_IP g_pPreAllocIp = NULL;
|
||
|
||
U32 lease_is_pre_assign(PDHCP_REQ pReq) {
|
||
char macStr[20] = {0};
|
||
U32 ipAddr;
|
||
|
||
MAC_TO_STR(pReq->cliMac, macStr);
|
||
if (lease_get_pre_assign(pReq->uid, macStr, pReq->hostName, &ipAddr) == ERR_ITEM_EXISTS) {
|
||
return ipAddr;
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
//int pre_alloc_dhcp_res(U32 uid, const char *pMac, U32 *pOutIp, PIPPOOL_INFO *pOutPool) {
|
||
int pre_alloc_dhcp_res(PDHCP_REQ pReq, U32 *pOutIp, PIPPOOL_INFO *pOutPool) {
|
||
PIPPOOL_INFO pPool, pTemp;
|
||
PPRE_ALLOC_IP pNewIp, pTmp, pCache = NULL;
|
||
PIPPOOL_INFO pUserPool;
|
||
|
||
if (pReq == NULL || pOutIp == NULL || pOutPool == NULL) {
|
||
LOG_MOD(error, ZLOG_MOD_DHCPD, "Input params error: %p, %p\n", pOutIp, pOutPool);
|
||
return -ERR_INPUT_PARAMS;
|
||
}
|
||
|
||
pUserPool = user_get_pool(pReq->uid);
|
||
if (pUserPool == NULL) {
|
||
LOG_MOD(error, ZLOG_MOD_DHCPD, "Can't found avaliable address pool\n");
|
||
return -ERR_DHCP_NO_POOL;
|
||
}
|
||
|
||
// 遍历当前用户所有IP地址池
|
||
HASH_ITER(hh, pUserPool, pPool, pTemp) {
|
||
U32 addr;
|
||
|
||
// 查看是否预分配过该设备
|
||
if ((addr = lease_is_pre_assign(pReq)) != 0) {
|
||
*pOutIp = addr;
|
||
*pOutPool = pPool;
|
||
return ERR_SUCCESS;
|
||
}
|
||
|
||
while ((addr = bitset_minimum(pPool->assignPool)) > 0) {
|
||
U32 ipAddr = (pPool->minAddr & pPool->netMask) + addr;
|
||
|
||
// 查找租约配置文件中是否记录了曾经分配的地址, 如果已经分配则获取下一个
|
||
// TODO: add process
|
||
// if (FALSE) {
|
||
// bitset_set(pPool->assignPool, addr);
|
||
// continue;
|
||
// }
|
||
|
||
// 查找IP地址是否已经被预分配, 未分配直接分配IP地址
|
||
if (lease_ip_is_pre_assign(pReq->uid, ipAddr) == FALSE) {
|
||
lease_db_add_pre_assign(pReq, ipAddr, pPool);
|
||
|
||
*pOutIp = ipAddr;
|
||
*pOutPool = pPool;
|
||
|
||
bitset_cls_bit(pPool->assignPool, addr);
|
||
|
||
LOG_MOD(trace, ZLOG_MOD_DHCPD, "Select ipaddr %08X at %d of %p\n", ipAddr, addr, pPool->assignPool);
|
||
return ERR_SUCCESS;
|
||
}
|
||
}
|
||
}
|
||
|
||
// 清理所有超时的预分配IP
|
||
lease_clearup_timeout_pre_assign();
|
||
// 没有可预分配的IP，报错
|
||
LOG_MOD(error,
|
||
ZLOG_MOD_DHCPD,
|
||
"No free ipaddress in poll: uid = %u, pool = 0x%08X\n",
|
||
pReq->uid,
|
||
pUserPool->poolKey);
|
||
return -ERR_DHCP_NO_ADDR;
|
||
}
|
||
|
||
int dhcp_lease_init() {
|
||
int rc = 0;
|
||
|
||
rc = lease_init_database();
|
||
|
||
if (rc != ERR_SUCCESS) {
|
||
return rc;
|
||
}
|
||
|
||
return ERR_SUCCESS;
|
||
} |