/*
 *  libnet
 *  libnet_link_win32.c - low-level win32 libwpcap routines
 *
 *  Copyright (c) 2001 - 2002 Don Bowman <don@sandvine.com>
 *  Copyright (c) 1998 - 2004 Mike D. Schiffman <mike@infonexus.com>
 *  Copyright (c) 2002 Roberto Larcher <roberto.larcher@libero.it>
 *  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 */

/* Libnet's unnamespaced ICMP6_ macros stomp on the enumerated versions of
   these names in the MS headers, so pre-include this header. */

#include <winsock2.h>
#include <iphlpapi.h> /* From the Microsoft Platform SDK */
#include <iprtrmib.h>
#include <assert.h>

#include "common.h"

/*
 * These are the types that are the same on all platforms, and that
 * have been defined by <net/bpf.h> for ages.
 */

#ifndef DLT_NULL
#define DLT_NULL	0	/* BSD loopback encapsulation */
#endif

#ifndef DLT_EN10MB
#define DLT_EN10MB	1	/* Ethernet (10Mb) */
#endif

#ifndef DLT_EN3MB
#define DLT_EN3MB	2	/* Experimental Ethernet (3Mb) */
#endif

#ifndef DLT_AX25
#define DLT_AX25	3	/* Amateur Radio AX.25 */
#endif

#ifndef DLT_PRONET
#define DLT_PRONET	4	/* Proteon ProNET Token Ring */
#endif

#ifndef DLT_CHAOS
#define DLT_CHAOS	5	/* Chaos */
#endif

#ifndef DLT_IEEE802
#define DLT_IEEE802	6	/* 802.5 Token Ring */
#endif

#ifndef DLT_ARCNET
#define DLT_ARCNET	7	/* ARCNET, with BSD-style header */
#endif

#ifndef DLT_SLIP
#define DLT_SLIP	8	/* Serial Line IP */
#endif

#ifndef DLT_PPP
#define DLT_PPP		9	/* Point-to-point Protocol */
#endif

#ifndef DLT_FDDI
#define DLT_FDDI	10	/* FDDI */
#endif

/*
 * These are types that are different on some platforms, and that
 * have been defined by <net/bpf.h> for ages.  We use #ifdefs to
 * detect the BSDs that define them differently from the traditional
 * libpcap <net/bpf.h>
 *
 * XXX - DLT_ATM_RFC1483 is 13 in BSD/OS, and DLT_RAW is 14 in BSD/OS,
 * but I don't know what the right #define is for BSD/OS.
 */

#ifndef DLT_ATM_RFC1483
#define DLT_ATM_RFC1483	11	/* LLC-encapsulated ATM */
#endif

int
libnet_open_link(libnet_t *l)
{
    DWORD dwErrorCode;
    NetType IFType;

    if (l == NULL)
    {
        return (-1);
    }

    if (l->device == NULL)
    {
        snprintf(l->err_buf, LIBNET_ERRBUF_SIZE,
                 "%s(): NULL device", __func__);
        return (-1);
    }

    l->lpAdapter = 0;

    /* open adapter */
    l->lpAdapter = PacketOpenAdapter(l->device);
    if (!l->lpAdapter || (l->lpAdapter->hFile == INVALID_HANDLE_VALUE))
    {
        dwErrorCode=GetLastError();
        snprintf(l->err_buf, LIBNET_ERRBUF_SIZE,
                 "%s(): unable to open the driver, error Code : %lx",
                 __func__, dwErrorCode);
        return (-1);
    }

    /* increase the send buffer */
    PacketSetBuff(l->lpAdapter, 512000);

    /*
     *  Assign link type and offset.
     */
    if (PacketGetNetType(l->lpAdapter, &IFType))
    {
        switch(IFType.LinkType)
        {
        case NdisMedium802_3:
            l->link_type = DLT_EN10MB;
            l->link_offset = LIBNET_ETH_H;
            break;
        case NdisMedium802_5:
            l->link_type = DLT_IEEE802;
            l->link_offset = LIBNET_TOKEN_RING_H;
            break;
        case NdisMediumFddi:
            l->link_type = DLT_FDDI;
            l->link_offset = 0x15;
            break;
        case NdisMediumWan:
            snprintf(l->err_buf, LIBNET_ERRBUF_SIZE,
                     "%s():, WinPcap has disabled support for Network type (%d)",
                     __func__, IFType.LinkType);
            return (-1);
            break;
        case NdisMediumAtm:
            l->link_type = DLT_ATM_RFC1483;
            break;
        case NdisMediumArcnet878_2:
            l->link_type = DLT_ARCNET;
            break;
        default:
            snprintf(l->err_buf, LIBNET_ERRBUF_SIZE,
                     "%s(): network type (%d) is not supported",
                     __func__, IFType.LinkType);
            return (-1);
            break;
        }
    }
    else
    {
        dwErrorCode=GetLastError();
        snprintf(l->err_buf, LIBNET_ERRBUF_SIZE,
                 "%s(): unable to determine the network type, error Code : %lx",
                 __func__, dwErrorCode);
        return (-1);
    }
    return (1);
}

int
libnet_close_link_interface(libnet_t *l)
{
    if (l->lpAdapter)
    {
        PacketSetHwFilter(l->lpAdapter, NDIS_PACKET_TYPE_ALL_LOCAL);
        PacketCloseAdapter(l->lpAdapter);
    }
    return (1);
}

int
libnet_write_link(libnet_t *l, const uint8_t *data, uint32_t size)
{
    PACKET pkt;
    DWORD  BytesTransfered = -1;

    /* Packet* arguments aren't const, but aren't actually modified.
     */
    PacketInitPacket(&pkt, (PVOID)data, size);

    if (PacketSendPacket(l->lpAdapter, &pkt, TRUE))
       BytesTransfered = size;

    return (BytesTransfered);
 }

struct libnet_ether_addr *
libnet_get_hwaddr(libnet_t *l)
{
    struct libnet_ether_addr *mac = &l->link_addr;
    ULONG IoCtlBufferLength = (sizeof(PACKET_OID_DATA) + sizeof(ULONG) - 1);
    PPACKET_OID_DATA OidData;

    int i = 0;

    if (l == NULL)
    {
        return (NULL);
    }

    if (l->device == NULL)
    {
        if (libnet_select_device(l) == -1)
        {
            snprintf(l->err_buf, LIBNET_ERRBUF_SIZE,
                     "%s(): can't figure out a device to use", __func__);
            return (NULL);
        }
    }

    OidData = (struct _PACKET_OID_DATA *)malloc(IoCtlBufferLength);
    if (OidData == NULL)
    {
        snprintf(l->err_buf, LIBNET_ERRBUF_SIZE,
                 "%s(): OidData is NULL", __func__);
        return(NULL);
    }

    if (l->link_type == DLT_IEEE802)
    {
        OidData->Oid = OID_802_5_CURRENT_ADDRESS;
    }
    else
    {
        OidData->Oid = OID_802_3_CURRENT_ADDRESS;
    }

    OidData->Length = 6;
    if ((PacketRequest(l->lpAdapter, FALSE, OidData)) == FALSE)
    {
        memset(mac, 0, 6);
    }
    else
    {
        for (i = 0; i < 6; i++)
        {
            mac->ether_addr_octet[i] = OidData->Data[i];
        }
    }
    free(OidData);
    return(mac);
}


BYTE *
libnet_win32_get_remote_mac(libnet_t *l, DWORD DestIP)
{
	HRESULT hr;
    ULONG   pulMac[6];
    ULONG   ulLen = 6;
	static PBYTE pbHexMac;
	PIP_ADAPTER_INFO pinfo = NULL;
	DWORD dwSize = 0;
	struct sockaddr_in sin;
	static BYTE bcastmac[]= {0xFF,0xFF,0xFF,0xFF,0xFF,0xFF};

	BYTE *MAC = libnet_win32_read_arp_table(DestIP);

	if (MAC==NULL)
	{
		memset(pulMac, 0xff, sizeof (pulMac));
		memset(&sin, 0, sizeof(sin));

		if((hr = SendARP (DestIP, 0, pulMac, &ulLen)) != NO_ERROR)
		{
			*(int32_t *)&sin.sin_addr = DestIP;
			GetAdaptersInfo(NULL, &dwSize);
			pinfo = (PIP_ADAPTER_INFO)GlobalAlloc(GPTR, dwSize);
			GetAdaptersInfo(pinfo, &dwSize);
			if(pinfo != NULL)
			{
				DestIP = inet_addr(pinfo->GatewayList.IpAddress.String);
				memset (pulMac, 0xff, sizeof (pulMac));
				ulLen = 6;
				if((hr = SendARP (DestIP, 0, pulMac, &ulLen)) != NO_ERROR)
				{
					GlobalFree(pinfo);
					return(bcastmac);
				}
			}
			else
			{
				GlobalFree(pinfo);
				return(bcastmac); /* ff:ff:ff:ff:ff:ff */
			}
		}

		pbHexMac = (PBYTE) pulMac;

		return (pbHexMac);
	}
	else
	{
		return (MAC);
	}
}

BYTE *libnet_win32_read_arp_table(DWORD DestIP)
{
    static BYTE buffMAC[6];
    BOOL fOrder = TRUE;
    DWORD status;

    MIB_IPNETTABLE *pIpNetTable = NULL;
    DWORD Size = 0;

    memset(buffMAC, 0, sizeof(buffMAC));
    status = GetIpNetTable(NULL, &Size, fOrder);
    if (status == ERROR_INSUFFICIENT_BUFFER)
    {
        pIpNetTable = alloca(Size);
        status = GetIpNetTable(pIpNetTable, &Size, fOrder);
    }

    if (status == NO_ERROR)
    {
        DWORD i, ci = pIpNetTable->table[0].dwIndex;  /* set current interface */

        for (i = 0; i < pIpNetTable->dwNumEntries; ++i)
        {
            if (pIpNetTable->table[i].dwIndex != ci)
                ci = pIpNetTable->table[i].dwIndex;

            if(pIpNetTable->table[i].dwAddr == DestIP) /* found IP in arp cache */
            {
                memcpy(buffMAC, pIpNetTable->table[i].bPhysAddr, sizeof(buffMAC));
                return buffMAC;
            }
        }
    }

    return(NULL);
}

/**
 * Local Variables:
 *  indent-tabs-mode: nil
 *  c-file-style: "stroustrup"
 * End:
 */