#include "router.h"
#include "db.h"
#include "net_task_thread.h"

int main(int argc, char const *argv[])
{
    // 初始化
    net_init(); // 初始化网络和数据库

    // 打开菜单
    menu_choice();

    close_mysql(); // 关闭数据库连接

    return 0;
}

void net_init()
{

    if (connect_mysql("localhost", 3306, "flykhan", "1202", "router") == 0)
        printf("数据库连接成功\n");
    else
        printf("数据库连接错误\n");
}

// MySQL 测试
int is_blocked_ip(unsigned char *ip)
{
    char sql[1024] = "";
    sprintf(sql, "select * from ip_fw where ip = '%s'", ip);
    int rows = result_rows(sql);

    printf("打印 %s\n", sql);
    printf("rows: %d\n", rows);

    // execlp("iptables", "iptables", "-A", "FORWARD", "-s", ip, "-j", "DROP", NULL); // 将源IP地址加入防火墙
    execlp("iptables", "iptables", "-D", "FORWARD", "-s", ip, "-j", "DROP", NULL); // 将源IP地址加入防火墙

    // 获取行数
    if (rows > 0)
    {
        printf("IP地址 %s 在黑名单中\n", ip);
        // execlp("iptables", "iptables", "-A", "FORWARD", "-s", ip, "-j", "DROP", NULL); // 将源IP地址加入防火墙
        // 从防火墙删除
        execlp("iptables", "iptables", "-D", "FORWARD", "-s", ip, "-j", "DROP", NULL); // 将源IP地址加入防火墙
        return 0;
    }
    else
    {
        printf("IP地址 %s 不在黑名单中\n", ip);
        return -1;
    }
}

void show()
{
    printf("**************************************************\n");
    printf("**                                              **\n");
    printf("**              模拟路由后端管理系统            **\n");
    printf("**                                              **\n");
    printf("**              1.添加黑名单IP                  **\n");
    printf("**              2.显示所有黑名单                **\n");
    printf("**              3.删除黑名单                    **\n");
    printf("**              4.开启路由器                    **\n");
    printf("**              5.刷新ARP表缓存                 **\n");
    printf("**              6.显示所有ARP表缓存             **\n");
    printf("**              9.重新打印菜单                  **\n");
    printf("**              0.退出此系统                    **\n");
    printf("**                                              **\n");
    printf("**************************************************\n");
    printf("**********      版权所有  FLY-NOOB      **********\n");
    printf("**************************************************\n");
}

// 判断IP地址格式是否正确
int is_ip_valid(const char *ip)
{
    regex_t regex; // 正则表达式
    int ret;

    // 编译正则表达式
    ret = regcomp(&regex, "^((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]?)$", REG_EXTENDED); // REG_EXTENDED 扩展正则表达式
    if (ret != 0)
    {
        printf("regcomp error\n");
        return -1;
    }

    // 执行正则表达式
    ret = regexec(&regex, ip, 0, NULL, 0);
    regfree(&regex); // 释放正则表达式

    if (ret == 0)
    {
        printf("IP地址格式正确\n");
        return 0;
    }
    else if (ret == REG_NOMATCH)
    {
        printf("IP地址格式非法\n");
        return -1;
    }
    else
    {
        printf("正则表达式匹配错误\n");
        return -1;
    }
}

// 终端控制菜单
void menu_choice()
{
    // 选择标志位
    int choice = 0;
    show(); // 打印菜单
    int pthread_flag = 0;
    while (1)
    {
        printf("请输入你的选择: \n");
        scanf("%d", &choice);
        getchar(); // 清空缓冲区

        // 后端添加黑名单 ip
        if (1 == choice)
        {
            char add_blacked_ip[INET_ADDRSTRLEN] = "";
            printf("请输入要添加的黑名单IP：格式如 192.168.6.5\n");
            fgets(add_blacked_ip, sizeof(add_blacked_ip), stdin);
            add_blacked_ip[strlen(add_blacked_ip) - 1] = '\0'; // 将最后的换行符替换为字符串结束符
            printf("add_blacked_ip: %s\n", add_blacked_ip);
            // IP 格式检查
            if (is_ip_valid(add_blacked_ip) != 0)
            {
                printf("IP地址格式错误\n");
                continue;
            }

            // 查询数据库黑名单IP表中是否已存在此IP
            int ret = search_ip_fw(add_blacked_ip);

            if (ret > 0)
            {
                printf("黑名单中已存在此IP\n");
                continue;
            }

            // 将IP地址加入黑名单
            insert_ip_fw(add_blacked_ip);
        }

        // 后端查询黑名单 ip
        else if (2 == choice)
        {
            // 打印黑名单
            print_ip_fw();
        }

        // 后端删除黑名单 ip
        else if (3 == choice)
        {
            char delete_blacked_ip[INET_ADDRSTRLEN] = "";
            printf("请输入要删除的黑名单IP：格式如 192.168.6.5\n");
            fgets(delete_blacked_ip, sizeof(delete_blacked_ip), stdin);
            delete_blacked_ip[strlen(delete_blacked_ip) - 1] = '\0'; // 将最后的换行符替换为字符串结束符

            // IP 格式检查
            if (is_ip_valid(delete_blacked_ip) != 0)
            {
                printf("IP地址格式错误\n");
                continue;
            }

            // 查询数据库黑名单IP表中是否已存在此IP
            int ret = search_ip_fw(delete_blacked_ip);

            if (ret > 0)
            {
                printf("delete_blacked_ip: %s\n", delete_blacked_ip);

                // 将IP地址从黑名单中删除
                delete_ip_fw(delete_blacked_ip);
            }
            else
            {
                printf("黑名单中不存在此IP\n");
            }
        }

        else if (4 == choice)
        {
            if (0 == pthread_flag)
            {
                printf("开启路由器\n");
                pthread_flag += 1; // 线程标志位修改
                pthread_t thread_net_task;
                pthread_create(&thread_net_task, NULL, net_task, NULL);
                pthread_detach(thread_net_task);
            }
            else
            {
                printf("路由器正在运行......\n");
            }
        }

        // 发送 ARP 广播
        else if (5 == choice)
        {
            send_arp_scan_broadcast();
        }

        // 显示 ARP 表
        else if (6 == choice)
        {
            printf("ARP 表如下：\n");
            const char *select_arp_result_sql = "select * from ip_mac";
            query(select_arp_result_sql, printResult);
        }

        // 重新打印菜单
        else if (9 == choice)
        {
            show();
        }

        // 退出此系统
        else if (0 == choice)
        {
            printf("退出此系统\n");
            return;
        }

        else
        {
            printf("输入错误，请重新输入\n");
        }
    }
}