510 lines
20 KiB
C
510 lines
20 KiB
C
//
|
||
// Created by HuangXin on 2023/4/24.
|
||
//
|
||
|
||
#include <sqlite3.h>
|
||
#include "user_errno.h"
|
||
#include "database.h"
|
||
#include "dhcp_network.h"
|
||
#include "rfc2131.h"
|
||
#include "ip_pool.h"
|
||
#include "db_interface.h"
|
||
#include "misc.h"
|
||
#include "zlog_module.h"
|
||
#include "lease.h"
|
||
|
||
// 10小时以上清理无效的预分配IP
|
||
#define DCHP_STEP_CLEANUP_TIMEOUT (36000)
|
||
// Discover 预分配IP 和 Request 请求间的超时时间
|
||
#define DCHP_STEP_REQUEST_TIMEOUT (600000)
|
||
|
||
#define CREATE_LEASE_TABLE() \
|
||
"CREATE TABLE IF NOT EXISTS lease " \
|
||
" ( id INTEGER PRIMARY KEY AUTOINCREMENT," \
|
||
" uid INTEGER NOT NULL," \
|
||
" mac CHAR(20) NOT NULL," \
|
||
" ip INTEGER NOT NULL," \
|
||
" lease INTEGER NOT NULL," \
|
||
" netmask INTEGER," \
|
||
" gateway INTEGER," \
|
||
" dns1 INTEGER," \
|
||
" dns2 INTEGER," \
|
||
" server INTEGER NOT NULL," \
|
||
" hostname CHAR(64) DEFAULT '' NOT NULL," \
|
||
" createTm TIMESTAMP DEFAULT (datetime('now', 'localtime')) NOT NULL" \
|
||
");" \
|
||
"CREATE INDEX IF NOT EXISTS " \
|
||
"lease_index ON lease (uid, mac);"
|
||
|
||
#define CREATE_PRE_ASSIGN_TABLE() \
|
||
"CREATE TABLE IF NOT EXISTS pre_assign" \
|
||
" ( id INTEGER PRIMARY KEY AUTOINCREMENT," \
|
||
" uid INTEGER NOT NULL," \
|
||
" xid INTEGER NOT NULL," \
|
||
" hostname CHAR(64) DEFAULT '' NOT NULL," \
|
||
" mac CHAR(20) NOT NULL," \
|
||
" ip CHAR(24) NOT NULL," \
|
||
" lease INTEGER NOT NULL," \
|
||
" netmask CHAR(24)," \
|
||
" gateway CHAR(24)," \
|
||
" dns1 CHAR(24)," \
|
||
" dns2 CHAR(24)," \
|
||
" server CHAR(24) NOT NULL," \
|
||
" createTm TIMESTAMP DEFAULT (datetime('now', 'localtime')) NOT NULL" \
|
||
"); CREATE INDEX IF NOT EXISTS pre_assign_index ON pre_assign(ip, uid);"
|
||
|
||
#define INSERT_PRE_ASSIGN_ROW_FMT \
|
||
"INSERT INTO pre_assign (uid, xid, hostname, mac, ip, lease, netmask, gateway, dns1, dns2, server) " \
|
||
"VALUES (%d, %d, '%s', '%s', '%s', %d, '%s', '%s', '%s', '%s', '%s');"
|
||
|
||
#define GET_PRE_ASSIGN_EXISTS_ROW_FMT \
|
||
"SELECT ip, id FROM pre_assign WHERE mac = '%s' AND hostname = '%s' AND uid = %d ORDER BY createTm DESC LIMIT 1;"
|
||
|
||
#define GET_ASSIGN_IP_INFO_FMT \
|
||
"SELECT id, ip, lease, netmask, gateway, dns1, dns2 FROM pre_assign WHERE mac = '%s' AND hostname = '%s' AND " \
|
||
"uid = %d AND server = '%s' AND (strftime('%%s', 'now', 'localtime') - strftime('%%s', createTm)) < " \
|
||
"%d ORDER BY createTm DESC LIMIT 1;"
|
||
|
||
#define IP_IS_PRE_ASSIGN_NOT_TIMEOUT_FMT \
|
||
"SELECT (strftime('%%s', 'now', 'localtime') - strftime('%%s', createTm)) as tm FROM pre_assign WHERE ip = " \
|
||
"'%s' AND tm < %d AND uid = %d ORDER BY tm DESC LIMIT 1;"
|
||
|
||
#define IP_IS_PRE_ASSIGN "SELECT ip FROM pre_assign WHERE ip = '%s' AND uid = %d"
|
||
|
||
#define IP_IS_PRE_ASSIGN_TIMEOUT_FMT \
|
||
"SELECT (strftime('%%s', 'now', 'localtime') - strftime('%%s', createTm)) as tm, id FROM pre_assign WHERE ip = " \
|
||
"'%s' AND tm >= %d AND uid = %d ORDER BY tm DESC LIMIT 1;"
|
||
|
||
#define GET_PRE_ASSIGN_ROW_FMT \
|
||
"SELECT ip, uid FROM pre_assign WHERE (strftime('%%s', 'now', 'localtime') - strftime('%%s', createTm)) < %d;"
|
||
|
||
#define CLS_TIMEOUT_PRE_ASSIGN_ROW_FMT \
|
||
"DELETE FROM pre_assign WHERE (strftime('%%s', 'now', 'localtime') - strftime('%%s', createTm)) > %d;"
|
||
|
||
#define UPDATE_CREATE_TIME_BY_ID_FMT "UPDATE pre_assign SET createTm = datetime('now', 'localtime') WHERE id = %s"
|
||
|
||
#define INSERT_LEASE_FMT \
|
||
"INSERT INTO lease(uid, hostname, mac, ip, lease, netmask, gateway, dns1, dns2, server) " \
|
||
"VALUES (%d, '%s', '%s', '%s', %d, '%s', '%s', '%s', '%s', '%s');"
|
||
|
||
#define LEASE_FIND_ITEM_FMT \
|
||
"SELECT id FROM lease WHERE mac = '%s' AND hostname = '%s' AND uid = %d AND ip = '%s' ORDER BY createTm DESC " \
|
||
"LIMIT 1;"
|
||
|
||
#define LEASE_ITEM_UPDATE_FMT \
|
||
"UPDATE lease SET lease = %d, netmask = '%s', gateway = '%s', dns1 = '%s', dns2 = '%s', server = '%s', " \
|
||
"createTm = datetime('now', 'localtime') WHERE id = %s;"
|
||
|
||
#define CLS_TIMEOUT_LEASE_ROW_FMT \
|
||
"DELETE FROM lease WHERE (strftime('%%s', 'now', 'localtime') - - strftime('%%s', createTm)) > lease.lease"
|
||
|
||
#define GET_LEASE_ROW_FMT \
|
||
"SELECT ip, uid FROM lease WHERE (strftime('%%s', 'now', 'localtime') - strftime('%%s', createTm)) < lease.lease;"
|
||
|
||
#define GET_LEASE_EXISTS_ROW_FMT \
|
||
"SELECT ip, id FROM lease WHERE mac = '%s' AND hostname = '%s' AND uid = %d AND (strftime('%%s', 'now', " \
|
||
"'localtime') - strftime('%%s', createTm)) < lease.lease ORDER BY createTm DESC LIMIT 1;"
|
||
|
||
#define GET_LEASE_INFO_FMT \
|
||
"SELECT id, ip, lease, netmask, gateway, dns1, dns2 FROM lease WHERE mac = '%s' AND hostname = '%s' AND " \
|
||
"uid = %d AND server = '%s' AND (strftime('%%s', 'now','localtime') - strftime('%%s', createTm)) < lease.lease " \
|
||
"ORDER BY createTm DESC LIMIT 1;"
|
||
|
||
#define UPDATE_LEASE_TIME_BY_ID_FMT "UPDATE lease SET createTm = datetime('now', 'localtime') WHERE id = %llu"
|
||
|
||
#define LEASE_RELEASE_FMT "DELETE FROM lease WHERE uid = %d AND mac = '%s' AND ip = '%s'"
|
||
|
||
static int lease_add(PDHCP_REQ pReq,
|
||
const char *ip,
|
||
const char *netmask,
|
||
const char *gw,
|
||
const char *dns1,
|
||
const char *dns2,
|
||
U32 leaseTime) {
|
||
int rc;
|
||
char buf[2048] = {0};
|
||
char macStr[20] = {0};
|
||
char **dbResult;
|
||
int nRow = 0, nColumn = 0;
|
||
|
||
MAC_TO_STR(pReq->cliMac, macStr);
|
||
snprintf(buf, 2048, LEASE_FIND_ITEM_FMT, macStr, pReq->hostName, pReq->uid, ip);
|
||
|
||
rc = db_sqlite3_get_rows(buf, &dbResult, &nRow, &nColumn, NULL);
|
||
if (rc == ERR_SUCCESS) {
|
||
// 已经存在，则更新
|
||
if (nRow > 0 && nColumn > 0) {
|
||
memset(buf, 0, 2048);
|
||
snprintf(buf, 2048, LEASE_ITEM_UPDATE_FMT, leaseTime, netmask, gw, dns1, dns2,
|
||
u32_to_str_ip(pReq->serverAddr), dbResult[nColumn]);
|
||
rc = db_sqlite3_sql_exec(buf, NULL, NULL, NULL);
|
||
if (rc != ERR_SUCCESS) {
|
||
LOG_MOD(error, ZM_DHCP_DB, "Add upgrade to lease db error: id = %s\n", dbResult[nColumn]);
|
||
}
|
||
} else {
|
||
//添加一个新的
|
||
// 记录设备分配的DHCP信息到数据库
|
||
memset(buf, 0, 2048);
|
||
snprintf(buf, 2048, INSERT_LEASE_FMT, pReq->uid, pReq->hostName, macStr, ip, leaseTime, netmask, gw, dns1,
|
||
dns2, u32_to_str_ip(pReq->serverAddr));
|
||
|
||
rc = db_sqlite3_sql_exec(buf, NULL, NULL, NULL);
|
||
|
||
if (rc != ERR_SUCCESS) {
|
||
LOG_MOD(error, ZM_DHCP_DB, "Add date to lease db error\n");
|
||
}
|
||
}
|
||
}
|
||
|
||
sqlite3_free_table(dbResult);
|
||
return ERR_SUCCESS;
|
||
}
|
||
|
||
int db_add_lease(PDHCP_REQ pReq, PPOOL_CTX pCtx) {
|
||
const char *pNetmask = u32_to_str_ip_safe(htonl(pCtx->netMask));
|
||
const char *pGateway = u32_to_str_ip_safe(htonl(pCtx->gwAddr));
|
||
const char *pDns1 = u32_to_str_ip_safe(htonl(pCtx->primeDNS));
|
||
const char *pDns2 = u32_to_str_ip_safe(htonl(pCtx->salveDNS));
|
||
const char *pIp = u32_to_str_ip_safe(htonl(pReq->cliAddr));
|
||
|
||
// 添加租约信息到lease数据库
|
||
lease_add(pReq, pIp, pNetmask, pGateway, pDns1, pDns2, pCtx->leaseTime);
|
||
|
||
free((void *)pNetmask);
|
||
free((void *)pGateway);
|
||
free((void *)pDns1);
|
||
free((void *)pDns2);
|
||
free((void *)pIp);
|
||
return ERR_SUCCESS;
|
||
}
|
||
|
||
int db_release_lease(PDHCP_REQ pReq) {
|
||
int rc;
|
||
char buf[1024] = {0};
|
||
char macStr[20] = {0};
|
||
MAC_TO_STR(pReq->cliMac, macStr);
|
||
snprintf(buf, 1024, LEASE_RELEASE_FMT, pReq->uid, macStr, u32_to_str_ip(htonl(pReq->reqIpAddr)));
|
||
|
||
rc = db_sqlite3_sql_exec(buf, NULL, NULL, NULL);
|
||
|
||
if (rc != ERR_SUCCESS) {
|
||
LOG_MOD(error, ZM_DHCP_DB, "DB remove lease item error: %s\n", buf);
|
||
}
|
||
return ERR_SUCCESS;
|
||
}
|
||
|
||
int db_lock_pre_assign_ip() {
|
||
int rc;
|
||
char buf[1024] = {0};
|
||
char **dbResult;
|
||
int nRow = 0, nColumn = 0;
|
||
|
||
snprintf(buf, 1024, GET_PRE_ASSIGN_ROW_FMT, DCHP_STEP_CLEANUP_TIMEOUT);
|
||
|
||
rc = db_sqlite3_get_rows(buf, &dbResult, &nRow, &nColumn, NULL);
|
||
if (rc == ERR_SUCCESS && nRow > 0 && nColumn > 0) {
|
||
int i;
|
||
for (i = 1; i <= nRow; i++) {
|
||
U32 uid = strtoul(dbResult[i * nColumn + 1], NULL, 10);
|
||
U32 ip = ntohl(inet_addr(dbResult[i * nColumn]));
|
||
PDHCP_USER pUser = dhcp_user_create(uid);
|
||
|
||
if (pUser) {
|
||
usr_lease_lock_ip(pUser, ip);
|
||
LOG_MOD(debug, ZM_DHCP_DB, "Lock prepare assign ip %s for user %u\n", dbResult[i * nColumn], uid);
|
||
}
|
||
// printf("-- Row %d value %s, %s, %s\n", i, dbResult[i * nColumn], dbResult[i * nColumn + 1], dbResult[i * nColumn + 2]);
|
||
}
|
||
}
|
||
sqlite3_free_table(dbResult);
|
||
|
||
memset(buf, 0, 1024);
|
||
snprintf(buf, 1024, GET_LEASE_ROW_FMT);
|
||
rc = db_sqlite3_get_rows(buf, &dbResult, &nRow, &nColumn, NULL);
|
||
if (rc == ERR_SUCCESS && nRow > 0 && nColumn > 0) {
|
||
int i;
|
||
for (i = 1; i <= nRow; i++) {
|
||
U32 uid = strtoul(dbResult[i * nColumn + 1], NULL, 10);
|
||
U32 ip = ntohl(inet_addr(dbResult[i * nColumn]));
|
||
PDHCP_USER pUser = dhcp_user_create(uid);
|
||
if (pUser) {
|
||
usr_lease_lock_ip(pUser, ip);
|
||
LOG_MOD(debug, ZM_DHCP_DB, "Lock prepare assign ip %s for user %u\n", dbResult[i * nColumn], uid);
|
||
}
|
||
}
|
||
}
|
||
sqlite3_free_table(dbResult);
|
||
|
||
return ERR_SUCCESS;
|
||
}
|
||
|
||
int db_clearup_timeout_lease() {
|
||
int rc;
|
||
char buf[1024] = {0};
|
||
|
||
snprintf(buf, 1024, CLS_TIMEOUT_PRE_ASSIGN_ROW_FMT, DCHP_STEP_CLEANUP_TIMEOUT);
|
||
|
||
rc = db_sqlite3_sql_exec(buf, NULL, NULL, NULL);
|
||
|
||
LOG_MOD(trace, ZM_DHCP_DB, "Cleanup pre_assign database timeout(%u seconds) data\n", DCHP_STEP_CLEANUP_TIMEOUT);
|
||
|
||
if (rc != ERR_SUCCESS) {
|
||
LOG_MOD(error, ZM_DHCP_DB, "Cleanup pre_assign database error: %s\n", buf);
|
||
}
|
||
|
||
rc = db_sqlite3_sql_exec(CLS_TIMEOUT_LEASE_ROW_FMT, NULL, NULL, NULL);
|
||
|
||
LOG_MOD(trace, ZM_DHCP_DB, "Cleanup assign database lease expire data\n");
|
||
if (rc != ERR_SUCCESS) {
|
||
LOG_MOD(error, ZM_DHCP_DB, "Cleanup lease database error: %s\n", buf);
|
||
}
|
||
|
||
return ERR_SUCCESS;
|
||
}
|
||
|
||
int db_ip_is_pre_assign(U32 uid, U32 ip) {
|
||
int rc;
|
||
char buf[1024] = {0};
|
||
char **dbResult;
|
||
int nRow = 0, nColumn = 0;
|
||
|
||
// 判断IP是否存曾经被预分配
|
||
snprintf(buf, 1024, IP_IS_PRE_ASSIGN, u32_to_str_ip(htonl(ip)), uid);
|
||
rc = db_sqlite3_get_rows(buf, &dbResult, &nRow, &nColumn, NULL);
|
||
sqlite3_free_table(dbResult);
|
||
if (rc == ERR_SUCCESS && nRow == 0) {
|
||
// 数据库没有相关记录，直接返回
|
||
LOG_MOD(trace, ZM_DHCP_DB, "New prepare assign ipaddr %s form user %u\n", u32_to_str_ip(htonl(ip)), uid);
|
||
return FALSE;
|
||
}
|
||
|
||
// 判断数据库中该IP是否未超时
|
||
memset(buf, 0, 1024);
|
||
snprintf(buf, 1024, IP_IS_PRE_ASSIGN_NOT_TIMEOUT_FMT, u32_to_str_ip(htonl(ip)), DCHP_STEP_CLEANUP_TIMEOUT, uid);
|
||
rc = db_sqlite3_get_rows(buf, &dbResult, &nRow, &nColumn, NULL);
|
||
sqlite3_free_table(dbResult);
|
||
|
||
if (rc == ERR_SUCCESS && nRow > 0 && nColumn > 0) {
|
||
// 如果数据库存在记录说明该IP暂时不可以用
|
||
LOG_MOD(trace, ZM_DHCP_DB, "No free ip address form user %u\n", uid);
|
||
return TRUE;
|
||
}
|
||
|
||
// 判断是否有超时预分配的IP， 存在的话可以预分配给其它请求
|
||
memset(buf, 0, 1024);
|
||
snprintf(buf, 1024, IP_IS_PRE_ASSIGN_TIMEOUT_FMT, u32_to_str_ip(htonl(ip)), DCHP_STEP_CLEANUP_TIMEOUT, uid);
|
||
rc = db_sqlite3_get_rows(buf, &dbResult, &nRow, &nColumn, NULL);
|
||
if (rc == ERR_SUCCESS && nRow > 0 && nColumn > 0) {
|
||
// 清理超时资源并预分配
|
||
// 更新时间戳
|
||
memset(buf, 0, 1024);
|
||
snprintf(buf, 1024, UPDATE_CREATE_TIME_BY_ID_FMT, dbResult[3]);
|
||
db_sqlite3_sql_exec(buf, NULL, NULL, NULL);
|
||
sqlite3_free_table(dbResult);
|
||
LOG_MOD(trace, ZM_DHCP_DB, "New prepare assign ipaddr %s form user %u by clearup resource\n",
|
||
u32_to_str_ip(htonl(ip)), uid);
|
||
return FALSE;
|
||
}
|
||
sqlite3_free_table(dbResult);
|
||
|
||
// 处理失败，默认返回
|
||
return TRUE;
|
||
}
|
||
|
||
static int db_get_table_items(PPOOL_CTX pAssign, const char *pRunSql, U64 *pDbId) {
|
||
int rc;
|
||
char **dbResult;
|
||
int nRow = 0, nColumn = 0;
|
||
|
||
rc = db_sqlite3_get_rows(pRunSql, &dbResult, &nRow, &nColumn, NULL);
|
||
if (rc == ERR_SUCCESS && nRow > 0 && nColumn > 0) {
|
||
char *pNetmask = dbResult[nColumn + 3];
|
||
char *pGateway = dbResult[nColumn + 4];
|
||
char *pDns1 = dbResult[nColumn + 5];
|
||
char *pDns2 = dbResult[nColumn + 6];
|
||
|
||
// minAddr 记录可分配的IP
|
||
pAssign->minAddr = ntohl(inet_addr(dbResult[nColumn + 1]));
|
||
pAssign->leaseTime = strtoul(dbResult[nColumn + 2], NULL, 10);
|
||
|
||
if (strlen(pNetmask) > 0) {
|
||
pAssign->netMask = ntohl(inet_addr(pNetmask));
|
||
}
|
||
|
||
if (strlen(pGateway) > 0) {
|
||
pAssign->gwAddr = ntohl(inet_addr(pGateway));
|
||
}
|
||
|
||
if (strlen(pDns1) > 0) {
|
||
pAssign->primeDNS = ntohl(inet_addr(pDns1));
|
||
}
|
||
|
||
if (strlen(pDns2) > 0) {
|
||
pAssign->salveDNS = ntohl(inet_addr(pDns2));
|
||
}
|
||
|
||
if (pDbId) {
|
||
*pDbId = strtoull(dbResult[nColumn], NULL, 10);
|
||
}
|
||
|
||
sqlite3_free_table(dbResult);
|
||
return ERR_SUCCESS;
|
||
} else {
|
||
sqlite3_free_table(dbResult);
|
||
return -ERR_ITEM_UNEXISTS;
|
||
}
|
||
}
|
||
|
||
int db_get_lease(PDHCP_REQ pReq, PPOOL_CTX pAssign) {
|
||
int rc;
|
||
U64 id;
|
||
char buf[2048] = {0};
|
||
char macStr[20] = {0};
|
||
|
||
MAC_TO_STR(pReq->cliMac, macStr);
|
||
snprintf(buf, 2048, GET_LEASE_INFO_FMT, macStr, pReq->hostName, pReq->uid, u32_to_str_ip(pReq->serverAddr));
|
||
|
||
rc = db_get_table_items(pAssign, buf, &id);
|
||
if (rc == ERR_SUCCESS) {
|
||
// 更新租约时间信息
|
||
memset(buf, 0, 1024);
|
||
snprintf(buf, 1024, UPDATE_LEASE_TIME_BY_ID_FMT, id);
|
||
|
||
if (db_sqlite3_sql_exec(buf, NULL, NULL, NULL) != ERR_SUCCESS) {
|
||
LOG_MOD(warn, ZM_DHCP_DB, "Update %llu lease time error: %s\n", id, buf);
|
||
}
|
||
return ERR_SUCCESS;
|
||
}
|
||
|
||
return -ERR_ITEM_UNEXISTS;
|
||
}
|
||
|
||
int db_get_pre_lease(PDHCP_REQ pReq, PPOOL_CTX pAssign) {
|
||
int rc;
|
||
char buf[2048] = {0};
|
||
char macStr[20] = {0};
|
||
U64 id;
|
||
|
||
MAC_TO_STR(pReq->cliMac, macStr);
|
||
snprintf(buf, 2048, GET_ASSIGN_IP_INFO_FMT, macStr, pReq->hostName, pReq->uid, u32_to_str_ip(pReq->serverAddr),
|
||
DCHP_STEP_REQUEST_TIMEOUT);
|
||
|
||
rc = db_get_table_items(pAssign, buf, &id);
|
||
|
||
if (rc == ERR_SUCCESS) {
|
||
const char *pNetmask = u32_to_str_ip_safe(htonl(pAssign->netMask));
|
||
const char *pGateway = u32_to_str_ip_safe(htonl(pAssign->gwAddr));
|
||
const char *pDns1 = u32_to_str_ip_safe(htonl(pAssign->primeDNS));
|
||
const char *pDns2 = u32_to_str_ip_safe(htonl(pAssign->salveDNS));
|
||
const char *pIp = u32_to_str_ip_safe(htonl(pAssign->minAddr));
|
||
|
||
// 删除预分配IP信息
|
||
#if 0
|
||
memset(buf, 0, 2048);
|
||
sprintf(buf, "DELETE FROM pre_assign WHERE id = %llu;", id);
|
||
db_sqlite3_sql_exec(buf, NULL, NULL, NULL);
|
||
#endif
|
||
// 添加租约信息到lease数据库
|
||
lease_add(pReq, pIp, pNetmask, pGateway, pDns1, pDns2, pAssign->leaseTime);
|
||
|
||
free((void *)pNetmask);
|
||
free((void *)pGateway);
|
||
free((void *)pDns1);
|
||
free((void *)pDns2);
|
||
free((void *)pIp);
|
||
return ERR_SUCCESS;
|
||
}
|
||
return -ERR_ITEM_UNEXISTS;
|
||
}
|
||
|
||
int db_get_pre_assign(U32 uid, const char *mac, const char *hostname, U32 *preAssign) {
|
||
int rc;
|
||
char buf[1024] = {0};
|
||
char **dbResult;
|
||
int nRow = 0, nColumn = 0;
|
||
|
||
snprintf(buf, 1024, GET_PRE_ASSIGN_EXISTS_ROW_FMT, mac, hostname, uid);
|
||
|
||
rc = db_sqlite3_get_rows(buf, &dbResult, &nRow, &nColumn, NULL);
|
||
if (rc == ERR_SUCCESS && nRow > 0 && nColumn > 0) {
|
||
if (preAssign) {
|
||
*preAssign = ntohl(inet_addr(dbResult[2]));
|
||
// 更新时间戳
|
||
memset(buf, 0, 1024);
|
||
snprintf(buf, 1024, UPDATE_CREATE_TIME_BY_ID_FMT, dbResult[3]);
|
||
db_sqlite3_sql_exec(buf, NULL, NULL, NULL);
|
||
}
|
||
|
||
sqlite3_free_table(dbResult);
|
||
return ERR_ITEM_EXISTS;
|
||
} else {
|
||
memset(buf, 0, 1024);
|
||
snprintf(buf, 1024, GET_LEASE_EXISTS_ROW_FMT, mac, hostname, uid);
|
||
|
||
rc = db_sqlite3_get_rows(buf, &dbResult, &nRow, &nColumn, NULL);
|
||
if (rc == ERR_SUCCESS && nRow > 0 && nColumn > 0) {
|
||
if (preAssign) {
|
||
*preAssign = ntohl(inet_addr(dbResult[2]));
|
||
// 更新时间戳
|
||
memset(buf, 0, 1024);
|
||
snprintf(buf, 1024, UPDATE_CREATE_TIME_BY_ID_FMT, dbResult[3]);
|
||
db_sqlite3_sql_exec(buf, NULL, NULL, NULL);
|
||
}
|
||
|
||
sqlite3_free_table(dbResult);
|
||
return ERR_ITEM_EXISTS;
|
||
}
|
||
}
|
||
|
||
sqlite3_free_table(dbResult);
|
||
return ERR_ITEM_UNEXISTS;
|
||
}
|
||
|
||
int db_add_pre_assign(PDHCP_REQ pReq, U32 ip, PPOOL_CTX pPool) {
|
||
int rc;
|
||
char buf[1024] = {0};
|
||
char macStr[20] = {0};
|
||
const char *pIp = u32_to_str_ip_safe(htonl(ip));
|
||
const char *pMask = u32_to_str_ip_safe(htonl(pPool->netMask));
|
||
const char *pGw = u32_to_str_ip_safe(htonl(pPool->gwAddr));
|
||
const char *pDns1 = u32_to_str_ip_safe(htonl(pPool->primeDNS));
|
||
const char *pDns2 = u32_to_str_ip_safe(htonl(pPool->salveDNS));
|
||
const char *pServer = u32_to_str_ip_safe(pReq->serverAddr);
|
||
|
||
MAC_TO_STR(pReq->cliMac, macStr);
|
||
snprintf(buf, 1024, INSERT_PRE_ASSIGN_ROW_FMT, pReq->uid, pReq->xid, pReq->hostName, macStr, pIp, pPool->leaseTime,
|
||
pMask, pGw, pDns1, pDns2, pServer);
|
||
|
||
free((void *)pIp);
|
||
free((void *)pMask);
|
||
free((void *)pGw);
|
||
free((void *)pDns1);
|
||
free((void *)pDns2);
|
||
free((void *)pServer);
|
||
|
||
rc = db_sqlite3_sql_exec(buf, NULL, NULL, NULL);
|
||
|
||
if (rc != ERR_SUCCESS) {
|
||
return rc;
|
||
}
|
||
|
||
return ERR_SUCCESS;
|
||
}
|
||
|
||
int db_init_lease_database() {
|
||
int rc;
|
||
|
||
rc = db_sqlite3_sql_exec(CREATE_LEASE_TABLE(), NULL, NULL, NULL);
|
||
|
||
if (rc != ERR_SUCCESS) {
|
||
return rc;
|
||
}
|
||
|
||
rc = db_sqlite3_sql_exec(CREATE_PRE_ASSIGN_TABLE(), NULL, NULL, NULL);
|
||
|
||
if (rc != ERR_SUCCESS) {
|
||
return rc;
|
||
}
|
||
|
||
return ERR_SUCCESS;
|
||
} |