#include "../include/arp.h"
#include "../include/etherframe.h"
#include "../include/memory.h"
#include "../include/pcnet.h"
uint8_t ARP_flags = 1;
uint64_t ARP_mac_address[MAX_ARP_TABLE];
uint32_t ARP_ip_address[MAX_ARP_TABLE];
uint32_t ARP_write_pointer = 0;

uint8_t *ARP_Packet(uint64_t dest_mac, uint32_t dest_ip, uint64_t src_mac,
                    uint32_t src_ip, uint16_t command) {
    struct ARPMessage *res =
            (struct ARPMessage *)kmalloc(sizeof(struct ARPMessage));
    res->hardwareType = 0x0100;
    res->protocol = 0x0008;
    res->hardwareAddressSize = 6;
    res->protocolAddressSize = 4;
    res->command = ((command & 0xff00) >> 8) | ((command & 0x00ff) << 8);
    res->dest_mac[0] = (uint8_t)dest_mac;
    res->dest_mac[1] = (uint8_t)(dest_mac >> 8);
    res->dest_mac[2] = (uint8_t)(dest_mac >> 16);
    res->dest_mac[3] = (uint8_t)(dest_mac >> 24);
    res->dest_mac[4] = (uint8_t)(dest_mac >> 32);
    res->dest_mac[5] = (uint8_t)(dest_mac >> 40);
    res->dest_ip = ((dest_ip << 24) & 0xff000000) |
                   ((dest_ip << 8) & 0x00ff0000) | ((dest_ip >> 8) & 0xff00) |
                   ((dest_ip >> 24) & 0xff);
    res->src_mac[0] = (uint8_t)src_mac;
    res->src_mac[1] = (uint8_t)(src_mac >> 8);
    res->src_mac[2] = (uint8_t)(src_mac >> 16);
    res->src_mac[3] = (uint8_t)(src_mac >> 24);
    res->src_mac[4] = (uint8_t)(src_mac >> 32);
    res->src_mac[5] = (uint8_t)(src_mac >> 40);
    res->src_ip = ((src_ip << 24) & 0xff000000) | ((src_ip << 8) & 0x00ff0000) |
                  ((src_ip >> 8) & 0xff00) | ((src_ip >> 24) & 0xff);
    return (uint8_t *)res;
}

uint64_t IPParseMAC(uint32_t dstIP) {
    extern uint8_t ARP_flags;
    extern uint32_t ARP_write_pointer;
    extern uint64_t ARP_mac_address[MAX_ARP_TABLE];
    extern uint32_t ARP_ip_address[MAX_ARP_TABLE];
    extern uint32_t ip;
    extern uint8_t mac0;
    extern uint32_t gateway;
    //extern struct TIMERCTL timerctl;
    if ((dstIP & 0xffffff00) != (ip & 0xffffff00)) {
        dstIP = gateway;
    }
    for (int i = 0; i != ARP_write_pointer; i++) {
        if (dstIP == ARP_ip_address[i]) {
            return ARP_mac_address[i];
        }
    }
    ARP_flags = 1;
    //printk("send\n");
    ether_frame_provider_send(
            0xffffffffffff, 0x0806,
            ARP_Packet(0xffffffffffff, dstIP, *(uint64_t *)&mac0, ip, 1),
            sizeof(struct ARPMessage));
    //printk("ok\n");
    /*
    uint32_t time = timerctl.count;
    while (ARP_flags) {
        if (timerctl.count - time > ARP_WAITTIME) {
            return -1;
        }
    }
     */
    return ARP_mac_address[ARP_write_pointer - 1];
}
void arp_handler(void *base) {
    extern uint32_t ip;
    extern uint8_t mac0;
    struct EthernetFrame_head *header = (struct EthernetFrame_head *)(base);
    if ((*(uint64_t *)&header->dest_mac[0] & 0xffffffffffff) ==
        (*(uint64_t *)&mac0 & 0xffffffffffff)) { // ARP广播回应
        struct ARPMessage *arp =
                (struct ARPMessage *)(base + sizeof(struct EthernetFrame_head));
        if (arp->command == 0x0200) {
            // printk("ARP MAC Address Reply\n");
            if (ARP_write_pointer < MAX_ARP_TABLE) {
                ARP_mac_address[ARP_write_pointer] =
                        *(uint64_t *)&header->src_mac[0] & 0xffffffffffff;
                ARP_ip_address[ARP_write_pointer] = swap32(arp->src_ip);
                ARP_write_pointer++;
                ARP_flags = 0;
            }
        }
        // 如果发送方不知道我们的MAC地址
        // 要发ARP数据包返回给发送方 告诉发送方我的MAC地址（确立联系）
    } else if ((*(uint64_t *)&header->dest_mac[0] & 0xffffffffffff) ==
               0xffffffffffff) { // dest_mac = 0xffffffffffff && ARP广播请求
        struct ARPMessage *arp =
                (struct ARPMessage *)(base + sizeof(struct EthernetFrame_head));
        if (arp->command == 0x0100 && arp->dest_ip == swap32(ip)) {
            // printk("ARP MAC Address request\n");
            uint32_t src_ip = ((arp->src_ip << 24) & 0xff000000) |
                              ((arp->src_ip << 8) & 0x00ff0000) |
                              ((arp->src_ip >> 8) & 0xff00) |
                              ((arp->src_ip >> 24) & 0xff);
            ether_frame_provider_send(*(uint64_t *)&header->src_mac[0], 0x0806,
                                      ARP_Packet(*(uint64_t *)&header->src_mac[0],
                                                 src_ip, *(uint64_t *)&mac0, ip, 2),
                                      sizeof(struct ARPMessage));
        }
    }
}