/*
 * Copyright (c) 2001,2002 Florian Schulze.
 * 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.
 * 3. Neither the name of the authors nor the names of the contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHORS 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 AUTHORS 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.
 *
 * pktif.c - This file is part of lwIPtest
 *
 ****************************************************************************
 *
 * This file is derived from an example in lwIP with the following license:
 *
 * Copyright (c) 2001, Swedish Institute of Computer Science.
 * 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.
 * 3. Neither the name of the Institute nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE 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 INSTITUTE 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.
 *
 */

#define WIN32_LEAN_AND_MEAN
/* get the windows definitions of the following 4 functions out of the way */
#include <stdlib.h>
#include <stdio.h>
#include <windows.h>
#include <packet32.h>
#include <ntddndis.h>

/** @todo use the lwip header file */
#define ETHARP_HWADDR_LEN 6

extern void process_input(void);

/* global variables for only one adapter */
LPADAPTER  lpAdapter;
LPPACKET   lpPacket;

char buffer[256000];  /* buffer to hold the data coming from the driver */
unsigned char *cur_packet;
int cur_length;
unsigned char ethaddr[ETHARP_HWADDR_LEN];

/*-----------------------------------------------------------------------------------*/
int
init_adapter(int adapter_num, char* mac_addr)
{
  #define Max_Num_Adapter 10

  void *AdapterList[Max_Num_Adapter];

  int i;
  DWORD dwVersion;
  DWORD dwWindowsMajorVersion;

  /* unicode strings (winnt) */
  char   AdapterName[8192]; /* string that contains a list of the network adapters */
  char   *temp,*temp1;

  /* ascii strings (win95) */
  char    AdapterNamea[8192]; /* string that contains a list of the network adapters */
  char    *tempa,*temp1a;

  int      AdapterNum=0;
  ULONG    AdapterLength;

  PPACKET_OID_DATA ppacket_oid_data;

  /* obtain the name of the adapters installed on this machine */
  AdapterLength=4096;

  memset(AdapterList,0,sizeof(AdapterList));
  memset(AdapterName,0,sizeof(AdapterName));
  memset(AdapterNamea,0,sizeof(AdapterNamea));

  i=0;

  /* the data returned by PacketGetAdapterNames is different in Win95 and in WinNT.
   * We have to check the os on which we are running */
  dwVersion=GetVersion();
  dwWindowsMajorVersion =  (DWORD)(LOBYTE(LOWORD(dwVersion)));
  if (!(dwVersion >= 0x80000000 && dwWindowsMajorVersion >= 4)) {
    /* Windows NT */
    if (PacketGetAdapterNames((char *)AdapterName,&AdapterLength)==FALSE){
      printf("Unable to retrieve the list of the adapters!\n");
      return -1;
    }
    temp=AdapterName;
    temp1=AdapterName;
    while ((*temp!='\0')||(*(temp-1)!='\0')) {
      if (*temp=='\0') {
        AdapterList[i] = temp1;
        temp1=temp+1;
        i++;
      }

      temp++;
    }

    AdapterNum=i;
    for (i=0; i<AdapterNum; i++) {
      printf("%2i: %s\n", i, AdapterList[i]);
    }
  } else {
    /* Windows 95 */
    if (PacketGetAdapterNames(AdapterNamea,&AdapterLength)==FALSE) {
      printf("Unable to retrieve the list of the adapters!\n");
      return -1;
    }
    tempa = AdapterNamea;
    temp1a = AdapterNamea;

    while ((*tempa!='\0')||(*(tempa-1)!='\0')) {
      if (*tempa=='\0') {
        AdapterList[i] = temp1a;
        temp1a=tempa+1;
        i++;
      }
      tempa++;
    }

    AdapterNum=i;
    AdapterNum=i;
    for (i=0; i<AdapterNum; i++) {
      printf("%2i: %s", i, AdapterList[i]);
    }
  }

  if (AdapterNum<=0) {
    return -1;
  }
  if (adapter_num < 0) {
    return -1;
  }
  if (adapter_num >= AdapterNum) {
    return -1;
  }
  ppacket_oid_data=malloc(sizeof(PACKET_OID_DATA)+ETHARP_HWADDR_LEN);
  lpAdapter=PacketOpenAdapter(AdapterList[adapter_num]);
  if (!lpAdapter || (lpAdapter->hFile == INVALID_HANDLE_VALUE)) {
    lpAdapter = NULL;
    return -1;
  }
  ppacket_oid_data->Oid=OID_802_3_PERMANENT_ADDRESS;
  ppacket_oid_data->Length=ETHARP_HWADDR_LEN;
  if (!PacketRequest(lpAdapter,FALSE,ppacket_oid_data)) {
    lpAdapter = NULL;
    return -1;
  }
  memcpy(&ethaddr,ppacket_oid_data->Data,ETHARP_HWADDR_LEN);
  free(ppacket_oid_data);
  memcpy(mac_addr, ethaddr, ETHARP_HWADDR_LEN);
  printf("MAC: %02X:%02X:%02X:%02X:%02X:%02X\n", ethaddr[0], ethaddr[1], ethaddr[2], ethaddr[3], ethaddr[4], ethaddr[5]);
  PacketSetBuff(lpAdapter,512000);
  PacketSetReadTimeout(lpAdapter,1);
  PacketSetHwFilter(lpAdapter,NDIS_PACKET_TYPE_ALL_LOCAL | NDIS_PACKET_TYPE_PROMISCUOUS);
  if ((lpPacket = PacketAllocatePacket())==NULL) {
    lpAdapter = NULL;
    return -1;
  }

  PacketInitPacket(lpPacket,(char*)buffer, sizeof(buffer));

  return 0;
}

void
shutdown_adapter(void)
{
  if (lpAdapter != NULL) {
    PacketFreePacket(lpPacket);
    PacketCloseAdapter(lpAdapter);
  }
}

int
packet_send(void *buffer, int len)
{
  LPPACKET lpPacket;

  if (lpAdapter == NULL) {
    return -1;
  }
  if ((lpPacket = PacketAllocatePacket()) == NULL) {
    return -1;
  }
  PacketInitPacket(lpPacket,buffer,len);
  if (!PacketSendPacket(lpAdapter,lpPacket,TRUE)) {
    return -1;
  }
  PacketFreePacket(lpPacket);

  return 0;
}

static void
ProcessPackets(LPPACKET lpPacket)
{
  ULONG  ulLines, ulBytesReceived;
  char  *base;
  char  *buf;
  u_int off=0;
  u_int tlen,tlen1;
  struct bpf_hdr *hdr;

  ulBytesReceived = lpPacket->ulBytesReceived;

  buf = lpPacket->Buffer;

  off=0;

  while (off<ulBytesReceived) {
    hdr=(struct bpf_hdr *)(buf+off);
    tlen1=hdr->bh_datalen;
    cur_length=tlen1;
    tlen=hdr->bh_caplen;
    off+=hdr->bh_hdrlen;

    ulLines = (tlen + 15) / 16;
    if (ulLines > 5) {
      ulLines=5;
    }

    base =(char*)(buf + off);
    cur_packet = base;
    off = Packet_WORDALIGN(off + tlen);

    process_input();
  }
}

void
update_adapter(void)
{
  if ((lpAdapter != NULL) && (PacketReceivePacket(lpAdapter, lpPacket, TRUE))) {
    ProcessPackets(lpPacket);
  }
  cur_length=0;
  cur_packet=NULL;
}