##include <stdio.h>
##include <rte_eal.h>
##include <rte_mbuf.h>
##include <rte_ethdev.h>
##include <arpa/inet.h>

##define MBUF_COUNT 4096
##define BURST_SIZE 32

int gDpdkPortId = 0; // 逻辑值与eth0无关，先绑定0就是0

// 只配置接受rx的设置
static const struct rte_eth_conf port_conf_default = {
    .rxmode = { .max_rx_pkt_len = RTE_ETHER_MAX_LEN }
};

int main(int argc, char *argv[])
{
    if(rte_eal_init(argc, argv) < 0){
        rte_exit(EXIT_FAILURE, "Failed to init EAL\n");
    }

    /* 1.开辟用于接受数据的mBuf pool内存池 */
    // 名称随便写，MBUF_COUNT巨页大小
    // rte_socket_id()内存插槽id
    struct rte_mempool *mbuf_pool = rte_pktmbuf_pool_create("mbufpool", MBUF_COUNT, 0, 0, RTE_MBUF_DEFAULT_BUF_SIZE, rte_socket_id());
    
    /* 2.配置网卡中用于收发数据的rx、tx队列 */
    struct rte_eth_conf port_conf = port_conf_default;
    int num_rx_queues = 1; // 需要小于多队列网卡最多的rx队列数，1代表只用1个，表示只活动一个rx用于收数据
    int num_tx_queues = 0; // 需要小于多队列网卡最多的tx队列数，0代表都不用，表示不写数据
    rte_eth_dev_configure(gDpdkPortId, num_rx_queues, num_tx_queues, &port_conf );

    /* 3.初始化rx队列 */
    // 绑定的网卡id，队列的id（因为只有一个，从0开始编号），接受用的环形队列的大小（随便写但是不能大于RTE_MBUF_DEFAULT_BUF_SIZE）
    // 。。。，rx配置，mBuf pool
    rte_eth_rx_queue_setup(gDpdkPortId, 0, 128, rte_eth_dev_socket_id(gDpdkPortId), NULL, mbuf_pool);
    
    /* 4.启动 */
    rte_eth_dev_start(gDpdkPortId);

    /* 5.开始解析接收到的数据 */
    while(1)
    {
        struct rte_mbuf *mbufs[BURST_SIZE];
        // 网口id，rx队列id，接收放在哪儿，接收大小
        // 这个函数传入的mbufs是个指针数组，是没有进行拷贝的，且不需要释放mbufs
        unsigned num_recvd = rte_eth_rx_burst(gDpdkPortId, 0, mbufs, BURST_SIZE);
        if(num_recvd > BURST_SIZE){
            rte_exit(EXIT_FAILURE, "Failed to rte_eth_rx_burst\n");
        }
        unsigned i = 0;
        for(; i < num_recvd ; i++){
            /* 解以太网头，14字节=6字节源mac地址+6字节目的mac地址+2字节类型 */
            // 使用宏定义，直接解出以太网头
            struct rte_ether_hdr *ehdr = rte_pktmbuf_mtod(mbufs[i], struct rte_ether_hdr *); // mbufs就是一个完整的数据包
            // 判断是不是ipv4，即最后两个字节的值，注意网络字节序转换
            if(ehdr->ether_type == htons(RTE_ETHER_TYPE_IPV4)) 
            // or ntohs(ehdr->ether_type) == RTE_ETHER_TYPE_IPV4
            {
                /* 
                    解ip头，将近20字节
                    4位版本，4位首部长度，8位服务类型(TOS)
                    16位总长度，16位标识，3位标志，13位ip片偏移
                    8位生存时间（TTL，跳数，缺省64），8位协议（TCP、UDP等），16位首部校验和
                    32位源ip，32位目的ip
                */
                // 使用宏定义，需要加上以太网头偏移
                struct rte_ipv4_hdr *iphdr = rte_pktmbuf_mtod_offset(mbufs[i], struct rte_ipv4_hdr *, sizeof(struct rte_ether_hdr));
                // 判断协议类型，即8位协议的值
                if(iphdr->next_proto_id == IPPROTO_UDP) 
                {
                    /*
                        解udp头，8个字节=2字节源端口+2字节目的端口+2个字节udp长度(这个是包含头的)+2字节校验和
                    */
                    // 注意这里为什么没有越界？因为使用了0长数组char a[0];
                    /*
                        0长数组示例：
                        struct udppkt{
                            struct ethhdr eh;
                            struct iphdr ip;
                            struct udphdr udp;
                            char data[0]; 
                            这里使用柔性数组，因为预留多少都不合适，我们可以根据前面udphdr中的长度算出这里的数组结尾在哪儿
                            直接对内存进行寻址
                        }
                    */
                    struct rte_udp_hdr *udphdr = (struct rte_udp_hdr *)(iphdr + 1);
                    uint16_t length = ntohs(udphdr->dgram_len);
                    *((char*)udphdr + length) = '\0';

                    printf("data : %s\n", (char*)(udphdr + 1));
                }
            }
        }
    }
}