#include "lwip/netif.h"
#include "lwip/ip.h"
#include "lwip/tcp.h"
#include "lwip/init.h"
#include "netif/etharp.h"
#include "lwip/udp.h"
#include "lwip/pbuf.h"
#include <stdio.h>
#include <string.h>
#include "main.h"

static void client_err(void *arg, err_t err) // 出现错误时调用这个函数，打印错误信息，并尝试重新连接
{
    // printf("连接错误!!\n");
    // printf("尝试重连!!\n");

    // // 连接失败的时候释放TCP控制块的内存
    // printf("关闭连接，释放TCP控制块内存\n");
    // // tcp_close(client_pcb);

    // // 重新连接
    // printf("重新初始化客户端\n");
    TCP_Client_Init();
}

static err_t client_send(void *arg, struct tcp_pcb *tpcb) // 发送函数，调用了tcp_write函数
{
    // 定义一个用于发送的缓冲区
    uint8_t send_buf[] = "be connected successfully\n";

    // 发送数据到服务器
    tcp_write(tpcb, send_buf, sizeof(send_buf), 1);

    // 返回操作成功标志
    return ERR_OK;
}

static err_t client_recv(void *arg, struct tcp_pcb *tpcb, struct pbuf *p, err_t err)
{
    if (p != NULL)
    {
        /* 接收数据*/
        tcp_recved(tpcb, p->tot_len);

        /* 返回接收到的数据*/
        tcp_write(tpcb, p->payload, p->tot_len, 1);

        memset(p->payload, 0, p->tot_len);
        pbuf_free(p);
    }
    else if (err == ERR_OK)
    {
        // 服务器断开连接
        tcp_close(tpcb);

        // 重新连接
        TCP_Client_Init();
    }
    return ERR_OK;
}

/**
 * @brief 客户端连接成功的回调函数
 *
 * 当客户端成功连接到服务器时，此函数将被调用。
 *
 * @param arg 用户提供的参数
 * @param pcb TCP连接控制块指针
 * @param err 错误码
 *
 * @return 错误码，ERR_OK表示成功
 */
static err_t client_connected(void *arg, struct tcp_pcb *pcb, err_t err)
{
    // 注册一个周期性回调函数
    tcp_poll(pcb, client_send, 2);

    // 注册一个接收函数
    tcp_recv(pcb, client_recv);

    return ERR_OK;
}

void TCP_Client_Init(void)
{
    struct tcp_pcb *client_pcb = NULL; // 这一句一定要放在里面，否则会没用
    ip4_addr_t server_ip;              // 因为客户端要主动去连接服务器，所以要知道服务器的IP地址
    /* 创建一个TCP控制块  */
    client_pcb = tcp_new();

    IP4_ADDR(&server_ip, DEST_IP_ADDR0, DEST_IP_ADDR1, DEST_IP_ADDR2, DEST_IP_ADDR3); // 合并IP地址
    // 开始连接
    tcp_connect(client_pcb, &server_ip, TCP_CLIENT_PORT, client_connected);
    ip_set_option(client_pcb, SOF_KEEPALIVE);
    // 注册异常处理
    tcp_err(client_pcb, client_err);
}