/*
Copyright (C) 2018 Srivats P.

This file is part of "Ostinato"

This is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>
*/

#include "winhostdevice.h"

#include <QUuid>

#ifdef Q_OS_WIN32

WindowsHostDevice::WindowsHostDevice(QString portName,
        DeviceManager *deviceManager)
    : Device(deviceManager)
{
    GUID guid = static_cast<GUID>(QUuid(portName.right(38)));
    ulong status = ConvertInterfaceGuidToLuid(&guid, &luid_);
    if (status != NO_ERROR) {
        qWarning("ConvertInterfaceGuidToLuid failed for %s with error %lx",
                 qPrintable(portName), status);
        luid_.Value = 0;
    }
}

void WindowsHostDevice::receivePacket(PacketBuffer* /*pktBuf*/)
{
    // Do Nothing
}

void WindowsHostDevice::clearNeighbors(Device::NeighborSet set)
{
    // No need to do anything - see AbstractPort::resolveDeviceNeighbors()
    // on when this is used
    if (set == kUnresolvedNeighbors)
        return;

    NET_IFINDEX ifIndex;
    ulong status = ConvertInterfaceLuidToIndex(&luid_, &ifIndex);
    if (status != NO_ERROR) {
        qWarning("luid2ifIdx convert failed with error %lx", status);
        return;
    }

    status = FlushIpNetTable2(AF_UNSPEC, ifIndex);
    if(status != NO_ERROR)
        qWarning("Flush ARP/ND table failed with error %lx", status);
}

void WindowsHostDevice::getNeighbors(OstEmul::DeviceNeighborList *neighbors)
{
    PMIB_IPNET_TABLE2 nbrs = NULL;
    // TODO: optimization: use AF_UNSPEC only if hasIp4 and hasIp6
    ulong status = GetIpNetTable2(AF_UNSPEC, &nbrs) != NO_ERROR;
    if (status != NO_ERROR) {
        qWarning("Get ARP/ND table failed with error %lx", status);
        return;
    }

    for (uint i = 0; i < nbrs->NumEntries; i++) {
        if (nbrs->Table[i].InterfaceLuid.Value != luid_.Value)
            continue;

        if (nbrs->Table[i].Address.si_family == AF_INET) {
            OstEmul::ArpEntry *arp = neighbors->add_arp();
            arp->set_ip4(qToBigEndian(quint32(
                            nbrs->Table[i].Address.Ipv4.sin_addr.s_addr)));
            arp->set_mac(qFromBigEndian<quint64>(
                            nbrs->Table[i].PhysicalAddress) >> 16);
        }
        // TODO: IPv6
    }

    FreeMibTable(nbrs);
}

quint64 WindowsHostDevice::arpLookup(quint32 ip)
{
    if (!luid_.Value)
        return 0;

    MIB_IPNET_ROW2 arpEntry;
    arpEntry.InterfaceLuid = luid_;
    arpEntry.Address.si_family = AF_INET;
    arpEntry.Address.Ipv4.sin_addr.s_addr = qToBigEndian(ip);

    if ((GetIpNetEntry2(&arpEntry) == NO_ERROR)
            && (arpEntry.PhysicalAddressLength == 6)) {
        return qFromBigEndian<quint64>(arpEntry.PhysicalAddress) >> 16;
    }
    else
        return 0;
}

quint64 WindowsHostDevice::ndpLookup(UInt128 /*ip*/)
{
    if (!luid_.Value)
        return 0;

    return 0; // TODO
}

void WindowsHostDevice::sendArpRequest(quint32 tgtIp)
{
    SOCKADDR_INET src;
    src.Ipv4.sin_addr.s_addr = qToBigEndian(ip4_);

    MIB_IPNET_ROW2 arpEntry;
    arpEntry.InterfaceLuid = luid_;
    arpEntry.Address.si_family = AF_INET;
    arpEntry.Address.Ipv4.sin_addr.s_addr = qToBigEndian(tgtIp);

    ulong status = ResolveIpNetEntry2(&arpEntry, &src);
    if (ResolveIpNetEntry2(&arpEntry, &src) != NO_ERROR)
        qWarning("Resolve arp failed with error %lx", status);
}

void WindowsHostDevice::sendNeighborSolicit(UInt128 /*tgtIp*/)
{
    // TODO
}

#endif