vcpe/srcs/open_dhcp/query.cpp

//
// 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"

#define STR2INT(val) ((int)strtol((val), nullptr, 10))

extern data2   cfig;
extern bool    kRunning;
extern dhcpMap dhcpCache;
extern time_t  t;

static void sendUserList(data19 *req, const char *pRequest) {
    char              logBuff[512];
    dhcpMap::iterator p;

    dzlog_debug("Input: %s\n", pRequest);

    if (pRequest == nullptr || strlen(pRequest) == 0) {
        sprintf(logBuff, "Requeset Json");
        logDHCPMess(logBuff, 1);
        return;
    }

    cJSON *pRoot = cJSON_Parse(pRequest);

    if (!pRoot) {
        return;
    }

    cJSON *pMsgContent = cJSON_GetObjectItem(pRoot, "msgContent");

    if (!pMsgContent) {
        cJSON_Delete(pRoot);
        return;
    }

    cJSON *pUserMac = cJSON_GetObjectItem(pMsgContent, "userMac");

    if (!pUserMac) {
        cJSON_Delete(pRoot);
        return;
    }

    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;
    }

    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_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", cJSON_Print(pRspRoot));

    cJSON_Delete(pRoot);
    cJSON_Delete(pRspRoot);

    req->bytes = (int)(fp - req->dp);
}

static void sendAllLists(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;
    }

    char *fp      = req->dp;
    char *maxData = req->dp + (req->memSize - 512);

    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, "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);

    fp += sprintf(fp, "%s", cJSON_Print(pRspRoot));

    cJSON_Delete(pRspRoot);

    req->bytes = (int)(fp - req->dp);
}

static void add_options(OBJ_DHCP_RNG pRange, data20* optionData){
    MYBYTE *dp         = optionData->options;
    MYBYTE  opTag      = 0;
    MYBYTE  valSize    = 0;
    MYWORD  buffSize   = sizeof(dhcp_packet) - sizeof(dhcp_header);
    char    value[512] = {0};
    bool    map_Flag[3]= {false};
    *dp                = 0;
    dp++;

    //dhcp_range
    addDHCPRange(pRange.rangAddr);

    for(int i = 0; i < 3; i++){
        char    buff[1024];
        MYBYTE  hoption[256];

        if(strlen(pRange.subnet) != 0 && !map_Flag[0]){
            map_Flag[0] = true;
            opTag = DHCP_OPTION_NETMASK;
            strcpy(value, pRange.subnet);
        }else if(strlen(pRange.dnsSvr) != 0 && !map_Flag[1]){
            map_Flag[1] = true;
            opTag = DHCP_OPTION_DNS;
            strcpy(value, pRange.dnsSvr);
        }else if(strlen(pRange.gateway) != 0 && !map_Flag[2]){
            map_Flag[2] = true;
            opTag = DHCP_OPTION_ROUTER;
            strcpy(value, pRange.gateway);
        }else
            break;

        int numbytes = myTokenize(buff, value, "/,.", true);
        if(numbytes > 255)
            break;
        else{
            char    *ptr = buff;
            valSize   = 0;
            for (; *ptr; ptr = myGetToken(ptr, 1)) {
                if (isInt(ptr)) {
                    hoption[valSize] = STR2INT(ptr);
                    valSize++;
                } else {
                    break;
                }
            }

            memcpy(value, hoption, valSize);
        }

        if(opTag == DHCP_OPTION_NETMASK && map_Flag[0]){
            optionData->mask = (*((MYDWORD *)value));
        }

        if(buffSize > valSize + 2){
            *dp = opTag;
            dp++;
            *dp = valSize;
            dp++;
            memcpy(dp, value, valSize);
            dp += valSize;
            buffSize -= (valSize + 2);
        }
    }

    //lease
    if (pRange.lease != 0) {
        MYDWORD j;
        j = pRange.lease;
        opTag = 51;
        if (buffSize > 6) {
            *dp = DHCP_OPTION_IPADDRLEASE;
            dp++;
            *dp = 4;
            dp++;
            dp += pUInt(dp, j);
            buffSize -= 6;
        }
    }

    *dp = DHCP_OPTION_END;
    dp++;
    optionData->optionSize = (dp - optionData->options);
}

static void expand_range_set(data19 *req, const char *pRequest) {
    char logBuff[512];
    OBJ_DHCP_RNG pRange;
    memset(&pRange, 0, sizeof(OBJ_DHCP_RNG));

    dzlog_debug("Input: %s\n", pRequest);

    if (pRequest == nullptr || strlen(pRequest) == 0) {
        sprintf(logBuff, "Requeset Json");
        logDHCPMess(logBuff, 1);
        return;
    }

    cJSON *pRoot = cJSON_Parse(pRequest);
    if (!pRoot) {
        return;
    }

    cJSON *pdhcp_range = cJSON_GetObjectItem(pRoot, "dhcp_range");

    if (!pdhcp_range) {
        cJSON_Delete(pRoot);
        return;
    }

    cJSON *psubnet_mask   = cJSON_GetObjectItem(pRoot, "subnet_mask");
    cJSON *pdomain_server = cJSON_GetObjectItem(pRoot, "domain_server");
    cJSON *pgateway       = cJSON_GetObjectItem(pRoot, "gateway");
    cJSON *please_time    = cJSON_GetObjectItem(pRoot, "lease_time");

    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;
    }
    char *fp      = req->dp;
    char *maxData = req->dp + (req->memSize - 512);

    cJSON *pRspRoot = cJSON_CreateObject();

    strcpy(pRange.rangAddr, pdhcp_range->valuestring);

    if (psubnet_mask) {
        strcpy(pRange.subnet, psubnet_mask->valuestring);
    }

    if (pdomain_server){
        strcpy(pRange.dnsSvr, pdomain_server->valuestring);
    }

    if (pgateway){
        strcpy(pRange.gateway, pgateway->valuestring);
    }

    if (please_time){
        pRange.lease = STR2INT(please_time->valuestring);
    }

    //写入cfig
    MYBYTE m = cfig.rangeCount;
    data20 optionData {};
    optionData.rangeSetInd = m;

    MYBYTE *options                              = nullptr;
    cfig.rangeSet[optionData.rangeSetInd].active = true;

    add_options(pRange, &optionData);

    if (optionData.optionSize > 3) {
        options = (MYBYTE *)calloc(1, optionData.optionSize);
        memcpy(options, optionData.options, optionData.optionSize);
    }

    cfig.dhcpRanges[m].rangeSetInd = optionData.rangeSetInd;
    cfig.dhcpRanges[m].options     = options;
    cfig.dhcpRanges[m].mask        = optionData.mask;
    if (!cfig.dhcpRanges[m].mask) {
        cfig.dhcpRanges[m].mask = cfig.mask;
    }
    cfig.rangeCount = (char)(m + 1);

    cJSON_AddNumberToObject(pRspRoot, "expand_start", cfig.dhcpRanges[m].rangeStart);
    cJSON_AddNumberToObject(pRspRoot, "expand_end", cfig.dhcpRanges[m].rangeEnd);

    fp += sprintf(fp, "%s", cJSON_Print(pRspRoot));

    cJSON_Delete(pRspRoot);

    req->bytes = (int)(fp - req->dp);
}

#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 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) {
        hw_http_response_send_error(response, HTTP_STATUS_500, "memory error");
        return;
    }

    if (request->method != HW_HTTP_GET) {
        hw_http_response_send_error(response, HTTP_STATUS_405, HTTP_STATUS_405);
        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;

    auto *req = (data19 *)malloc(sizeof(struct data19));

    if (req == nullptr) {
        hw_http_response_send_error(response, HTTP_STATUS_500, "memory error");
        return;
    }

    if (request->method != HW_HTTP_POST) {
        hw_http_response_send_error(response, HTTP_STATUS_405, HTTP_STATUS_405);
        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);
    sendUserList(req, request->body->value);
    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;

    auto *req = (data19 *)malloc(sizeof(struct data19));

    if (req == nullptr) {
        hw_http_response_send_error(response, HTTP_STATUS_500, "memory error");
        return;
    }

    if (request->method != HW_HTTP_GET) {
        hw_http_response_send_error(response, HTTP_STATUS_405, HTTP_STATUS_405);
        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);
    sendAllLists(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_expand_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;

    auto *req = (data19 *)malloc(sizeof(struct data19));

    if (req == nullptr) {
        hw_http_response_send_error(response, HTTP_STATUS_500, "memory error");
        return;
    }

    if (request->method != HW_HTTP_POST) {
        hw_http_response_send_error(response, HTTP_STATUS_405, HTTP_STATUS_405);
        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);
    expand_range_set(req, request->body->value);
    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);
}

void opendhcp_init_http_server() {
    static int added = FALSE;

    if (!added) {
        hw_http_add_route("/", opendhcp_http_info, nullptr);
        hw_http_add_route("getuser", opendhcp_http_get_userinfo, nullptr);
        hw_http_add_route("allusers", opendhcp_http_get_alluser, nullptr);
        hw_http_add_route("expansion", opendhcp_http_expand_rangeset, nullptr);
        added = TRUE;
    }
}