/* * Copyright (c) 2006, 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. * * This file is part of the Contiki operating system. * * Author: Oliver Schmidt * * $Id: wpcap-service.c,v 1.5 2006/12/29 00:06:32 oliverschmidt Exp $ */ #define WIN32_LEAN_AND_MEAN #include #include #include #include #include #pragma comment(lib, "wsock32") #pragma comment(lib, "iphlpapi") #define htons /* Avoid 'redefinition' error. */ #include "contiki.h" #include "contiki-net.h" void send_packet(void); SERVICE(wpcap_service, packet_service, { (u8_t (*)(void))send_packet }); PROCESS(wpcap_service_process, "WinPcap driver"); struct pcap; struct pcap_if { struct pcap_if *next; char *name; char *description; struct pcap_addr { struct pcap_addr *next; struct sockaddr *addr; struct sockaddr *netmask; struct sockaddr *broadaddr; struct sockaddr *dstaddr; } *addresses; unsigned __int32 flags; }; struct pcap_pkthdr { struct timeval ts; unsigned __int32 caplen; unsigned __int32 len; }; static struct pcap *pcap; static int (* pcap_findalldevs)(struct pcap_if **, char *); static struct pcap *(* pcap_open_live)(char *, int, int, int, char *); static int (* pcap_next_ex)(struct pcap *, struct pcap_pkthdr **, unsigned char **); static int (* pcap_sendpacket)(struct pcap *, unsigned char *, int); #define BUF ((struct uip_eth_hdr *)&uip_buf[0]) /*---------------------------------------------------------------------------*/ static void pollhandler(void) { struct pcap_pkthdr *packet_header; unsigned char *packet; process_poll(&wpcap_service_process); switch(pcap_next_ex(pcap, &packet_header, &packet)) { case -1: error_exit("Error on poll\n"); case 0: return; } uip_len = packet_header->caplen; CopyMemory(uip_buf, packet, uip_len); if(BUF->type == HTONS(UIP_ETHTYPE_IP)) { debug_printf("I"); uip_len -= sizeof(struct uip_eth_hdr); tcpip_input(); } else if(BUF->type == HTONS(UIP_ETHTYPE_ARP)) { debug_printf("A"); uip_arp_arpin(); if(uip_len > 0) { if(pcap_sendpacket(pcap, uip_buf, uip_len) == -1) { error_exit("Error on ARP response\n"); } } } } /*---------------------------------------------------------------------------*/ static void send_packet(void) { uip_arp_out(); if(pcap_sendpacket(pcap, uip_buf, uip_len) == -1) { error_exit("Error on send\n"); } } /*---------------------------------------------------------------------------*/ static void init_pcap(struct in_addr addr) { struct pcap_if *interfaces; char error[256]; if(pcap_findalldevs(&interfaces, error) == -1) { error_exit(error); } while(interfaces != NULL) { debug_printf("Found interface: %s\n", interfaces->description); if(interfaces->addresses != NULL && interfaces->addresses->addr != NULL && interfaces->addresses->addr->sa_family == AF_INET) { struct in_addr interface_addr; interface_addr = ((struct sockaddr_in *)interfaces->addresses->addr)->sin_addr; debug_printf(" With address: %s\n", inet_ntoa(interface_addr)); if(interface_addr.s_addr == addr.s_addr) { break; } } interfaces = interfaces->next; } if(interfaces == NULL) { error_exit("No interface found with IP addr specified on cmdline\n"); } pcap = pcap_open_live(interfaces->name, UIP_BUFSIZE, 0, -1, error); if(pcap == NULL) { error_exit(error); } } /*---------------------------------------------------------------------------*/ static void set_ethaddr(struct in_addr addr) { PIP_ADAPTER_ADDRESSES adapters; ULONG size = 0; if(GetAdaptersAddresses(AF_INET, GAA_FLAG_SKIP_ANYCAST | GAA_FLAG_SKIP_MULTICAST | GAA_FLAG_SKIP_DNS_SERVER, NULL, NULL, &size) != ERROR_BUFFER_OVERFLOW) { error_exit("Error on access to adapter list size\n"); } adapters = alloca(size); if(GetAdaptersAddresses(AF_INET, GAA_FLAG_SKIP_ANYCAST | GAA_FLAG_SKIP_MULTICAST | GAA_FLAG_SKIP_DNS_SERVER, NULL, adapters, &size) != ERROR_SUCCESS) { error_exit("Error on access to adapter list\n"); } while(adapters != NULL) { char description[256]; WideCharToMultiByte(CP_ACP, 0, adapters->Description, -1, description, sizeof(description), NULL, NULL); debug_printf("Found adapter: %s\n", description); if(adapters->FirstUnicastAddress != NULL && adapters->FirstUnicastAddress->Address.lpSockaddr != NULL && adapters->FirstUnicastAddress->Address.lpSockaddr->sa_family == AF_INET) { struct in_addr adapter_addr; adapter_addr = ((struct sockaddr_in *)adapters->FirstUnicastAddress->Address.lpSockaddr)->sin_addr; debug_printf(" With address: %s\n", inet_ntoa(adapter_addr)); if(adapter_addr.s_addr == addr.s_addr) { if(adapters->PhysicalAddressLength != 6) { error_exit("IP addr specified on cmdline does not belong to an Ethernet card\n"); } debug_printf(" Ethernetaddr: %02X-%02X-%02X-%02X-%02X-%02X\n", adapters->PhysicalAddress[0], adapters->PhysicalAddress[1], adapters->PhysicalAddress[2], adapters->PhysicalAddress[3], adapters->PhysicalAddress[4], adapters->PhysicalAddress[5]); uip_setethaddr((*(struct uip_eth_addr *)adapters->PhysicalAddress)); break; } } adapters = adapters->Next; } if(adapters == NULL) { error_exit("No adapter found with IP addr specified on cmdline\n"); } } /*---------------------------------------------------------------------------*/ PROCESS_THREAD(wpcap_service_process, ev, data) { static struct in_addr addr; static HMODULE wpcap; PROCESS_POLLHANDLER(pollhandler()); PROCESS_BEGIN(); addr.s_addr = inet_addr(__argv[1]); if(addr.s_addr == INADDR_NONE) { error_exit("Usage: contiki \n"); } debug_printf("Cmdline address: %s\n", inet_ntoa(addr)); wpcap = LoadLibrary("wpcap.dll"); (FARPROC)pcap_findalldevs = GetProcAddress(wpcap, "pcap_findalldevs"); (FARPROC)pcap_open_live = GetProcAddress(wpcap, "pcap_open_live"); (FARPROC)pcap_next_ex = GetProcAddress(wpcap, "pcap_next_ex"); (FARPROC)pcap_sendpacket = GetProcAddress(wpcap, "pcap_sendpacket"); if(pcap_findalldevs == NULL || pcap_open_live == NULL || pcap_next_ex == NULL || pcap_sendpacket == NULL) { error_exit("Error on access to WinPcap library\n"); } init_pcap(addr); set_ethaddr(addr); uip_arp_init(); SERVICE_REGISTER(wpcap_service); process_poll(&wpcap_service_process); PROCESS_WAIT_EVENT_UNTIL(ev == PROCESS_EVENT_EXIT || ev == PROCESS_EVENT_SERVICE_REMOVED); SERVICE_REMOVE(wpcap_service); FreeLibrary(wpcap); PROCESS_END(); } /*---------------------------------------------------------------------------*/