oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 1 | #define WIN32_LEAN_AND_MEAN |
| 2 | #include <winsock2.h> |
| 3 | #include <ws2tcpip.h> |
| 4 | #include <stdlib.h> |
oliverschmidt | 7f0b22b | 2005-01-29 23:10:43 +0000 | [diff] [blame] | 5 | #include <malloc.h> |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 6 | |
| 7 | /* from mstcpip.h */ |
oliverschmidt | 7f0b22b | 2005-01-29 23:10:43 +0000 | [diff] [blame] | 8 | |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 9 | #define SIO_RCVALL _WSAIOW(IOC_VENDOR,1) |
| 10 | |
oliverschmidt | 7f0b22b | 2005-01-29 23:10:43 +0000 | [diff] [blame] | 11 | /* from iphlpapi.h */ |
| 12 | |
| 13 | #pragma pack(push, 4) |
| 14 | |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 15 | typedef struct mib_ipaddrrow_t { |
| 16 | DWORD dwAddr; |
| 17 | DWORD dwIndex; |
| 18 | DWORD dwMask; |
| 19 | DWORD dwBCastAddr; |
| 20 | DWORD dwReasmSize; |
| 21 | DWORD unused; |
oliverschmidt | 7f0b22b | 2005-01-29 23:10:43 +0000 | [diff] [blame] | 22 | } MIB_IPADDRROW; |
| 23 | |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 24 | typedef struct mib_ipaddrtable_t { |
oliverschmidt | ed66a1d | 2005-02-15 21:18:33 +0000 | [diff] [blame] | 25 | DWORD dwNumEntries; |
oliverschmidt | 7f0b22b | 2005-01-29 23:10:43 +0000 | [diff] [blame] | 26 | MIB_IPADDRROW table[0]; |
| 27 | } MIB_IPADDRTABLE; |
| 28 | |
oliverschmidt | 7f0b22b | 2005-01-29 23:10:43 +0000 | [diff] [blame] | 29 | #pragma pack(pop) |
| 30 | |
ryohji | 55cd87a | 2007-05-02 04:06:56 +0000 | [diff] [blame] | 31 | #define htons contiki_htons /* htons is defined in winsock2 */ |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 32 | #include "uip.h" |
ryohji | 55cd87a | 2007-05-02 04:06:56 +0000 | [diff] [blame] | 33 | #undef htons |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 34 | #include "tcpip.h" |
| 35 | #include "ctk.h" |
ryohji | 55cd87a | 2007-05-02 04:06:56 +0000 | [diff] [blame] | 36 | |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 37 | #define BUF ((uip_tcpip_hdr *)uip_buf) |
| 38 | |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 39 | typedef DWORD (WINAPI *ARPFUNC)(DWORD, DWORD, DWORD); |
| 40 | |
| 41 | extern void debug_printf(char *format, ...); |
| 42 | |
| 43 | static void error_exit(const char *message, int value); |
| 44 | static void create_proxyarp(void); |
| 45 | static void delete_proxyarp(void); |
oliverschmidt | 7f0b22b | 2005-01-29 23:10:43 +0000 | [diff] [blame] | 46 | |
oliverschmidt | e7851ab | 2005-01-30 20:25:28 +0000 | [diff] [blame] | 47 | static DWORD interfaceindex; |
oliverschmidt | 7f0b22b | 2005-01-29 23:10:43 +0000 | [diff] [blame] | 48 | static DWORD proxyaddr; |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 49 | static SOCKET rawsock = INVALID_SOCKET; |
| 50 | |
ryohji | 55cd87a | 2007-05-02 04:06:56 +0000 | [diff] [blame] | 51 | |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 52 | #define END_OF(ARRAY) (ARRAY + sizeof(ARRAY)/sizeof(ARRAY[0])) |
| 53 | #define HOSTADDR() (*(DWORD*)uip_hostaddr) |
| 54 | #define NETMASK() (*(DWORD*)uip_netmask) |
| 55 | |
| 56 | #define DELAY_LOAD(NAME,MODULE) \ |
| 57 | ( GetProcAddress(GetModuleHandle(MODULE), NAME) \ |
| 58 | ? GetProcAddress(GetModuleHandle(MODULE), NAME) \ |
| 59 | : (error_exit("GetProcAddress("#NAME") error: %d\n", GetLastError()), 0)) |
| 60 | |
| 61 | #define SOCKCTL_(ON,CMD) \ |
| 62 | { unsigned long on = ON; if (SOCKET_ERROR == \ |
| 63 | ioctlsocket(rawsock, CMD, &on)) \ |
| 64 | error_exit("ioctlsocket("#CMD") error: %d\n", WSAGetLastError()); } |
| 65 | |
| 66 | #define SOCKOPT_ON_(OPT,LAYER) \ |
| 67 | { unsigned long on = 1; if (SOCKET_ERROR == \ |
| 68 | setsockopt(rawsock, LAYER, OPT, (char*)&on, sizeof(on))) \ |
| 69 | error_exit("setsockopt("#OPT") error: %d\n", WSAGetLastError()); } |
| 70 | |
| 71 | #define TRAP(ERR,HEAD,TAIL) \ |
| 72 | ( match(ERR,HEAD,TAIL) ? ERR : (error_exit("send/recv() error: %d\n",ERR),0) ) |
| 73 | |
ryohji | 55cd87a | 2007-05-02 04:06:56 +0000 | [diff] [blame] | 74 | /*-----------------------------------------------------------------------------------*/ |
| 75 | static int match(const int x, const int *head, const int *const tail) |
| 76 | { |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 77 | for( ;tail != head; ++head) if(*head == x) return 1; |
ryohji | 55cd87a | 2007-05-02 04:06:56 +0000 | [diff] [blame] | 78 | return 0; |
| 79 | } |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 80 | /*-----------------------------------------------------------------------------------*/ |
| 81 | static void |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 82 | error_exit(const char *message, int value) |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 83 | { |
| 84 | console_exit(); |
| 85 | cprintf(message, value); |
| 86 | exit(EXIT_FAILURE); |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 87 | } |
| 88 | /*-----------------------------------------------------------------------------------*/ |
| 89 | static DWORD |
| 90 | get_interface_index_for_(DWORD ip) |
| 91 | { |
| 92 | DWORD (WINAPI *Table)(MIB_IPADDRTABLE *, DWORD *, BOOL); |
| 93 | DWORD retval; |
| 94 | DWORD size = sizeof(MIB_IPADDRROW); |
| 95 | MIB_IPADDRTABLE *d = 0; |
| 96 | MIB_IPADDRROW *r; |
| 97 | |
| 98 | (FARPROC)Table = DELAY_LOAD("GetIpAddrTable","iphlpapi"); |
| 99 | do { d = realloc(d, size *= 2); } |
| 100 | while(ERROR_INSUFFICIENT_BUFFER == (retval = Table(d, &size, TRUE))); |
| 101 | if(NO_ERROR != retval) error_exit("GetIpAddrTable error: %d\n", retval); |
| 102 | |
| 103 | for(r = d->table; r != d->table + d->dwNumEntries; ++r) |
| 104 | if(r->dwAddr == ip) return retval = r->dwIndex, realloc(d,0), retval; |
| 105 | return error_exit("Parameter error: Unknown host IP address\n", 0), 0; |
| 106 | } |
| 107 | /*-----------------------------------------------------------------------------------*/ |
| 108 | static void* |
| 109 | sock_in_addr(unsigned long ipaddr, u16_t port) |
| 110 | { |
| 111 | static struct sockaddr_in addr; |
| 112 | addr.sin_family = AF_INET; |
| 113 | addr.sin_port = port; |
| 114 | addr.sin_addr.s_addr = ipaddr; |
| 115 | return &addr; |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 116 | } |
| 117 | /*-----------------------------------------------------------------------------------*/ |
| 118 | void |
| 119 | rawsock_init(void) |
| 120 | { |
oliverschmidt | ed66a1d | 2005-02-15 21:18:33 +0000 | [diff] [blame] | 121 | DWORD hostaddr; |
oliverschmidt | ed66a1d | 2005-02-15 21:18:33 +0000 | [diff] [blame] | 122 | WSADATA wsadata; |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 123 | |
| 124 | if(WSAStartup(2, &wsadata) != 0) { |
| 125 | error_exit("WSAStartup() error\n", 0); |
| 126 | } |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 127 | |
oliverschmidt | 7f0b22b | 2005-01-29 23:10:43 +0000 | [diff] [blame] | 128 | hostaddr = inet_addr(__argv[1]); |
| 129 | if(hostaddr == INADDR_NONE) { |
| 130 | error_exit("Parameter error: Invalid host IP address\n", 0); |
| 131 | } |
| 132 | |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 133 | rawsock = socket(PF_INET, SOCK_RAW, IPPROTO_IP); |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 134 | if(rawsock == INVALID_SOCKET) { |
| 135 | error_exit("socket() error: %d\n", WSAGetLastError()); |
| 136 | } |
| 137 | |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 138 | if(bind(rawsock, sock_in_addr(hostaddr,0), sizeof(struct sockaddr)) == SOCKET_ERROR) { |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 139 | error_exit("bind() error: %d\n", WSAGetLastError()); |
| 140 | } |
| 141 | |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 142 | SOCKOPT_ON_(SO_BROADCAST, SOL_SOCKET); |
| 143 | SOCKOPT_ON_(IP_HDRINCL, IPPROTO_IP); |
| 144 | SOCKCTL_(1, SIO_RCVALL); |
| 145 | SOCKCTL_(1, FIONBIO); |
oliverschmidt | c160fca | 2005-01-30 13:26:05 +0000 | [diff] [blame] | 146 | |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 147 | LoadLibrary("iphlpapi"); /* map iphlpapi.dll to process address space */ |
| 148 | if(atexit(delete_proxyarp)) error_exit("atexit() error\n", 0); |
| 149 | interfaceindex = get_interface_index_for_(hostaddr); |
| 150 | } |
| 151 | /*-----------------------------------------------------------------------------------*/ |
| 152 | void |
| 153 | rawsock_fini(void) |
| 154 | { |
| 155 | delete_proxyarp(); |
| 156 | closesocket(rawsock); |
| 157 | rawsock = INVALID_SOCKET; |
| 158 | FreeLibrary(GetModuleHandle("iphlpapi")); |
| 159 | WSACleanup(); |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 160 | } |
| 161 | /*-----------------------------------------------------------------------------------*/ |
| 162 | void |
ryohji | 55cd87a | 2007-05-02 04:06:56 +0000 | [diff] [blame] | 163 | rawsock_send(u8_t *hdr, u16_t hdrlen, u8_t *data, u16_t datalen) |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 164 | { |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 165 | static const int ignore[] = {WSAEADDRNOTAVAIL,WSAEHOSTUNREACH}; |
ryohji | 55cd87a | 2007-05-02 04:06:56 +0000 | [diff] [blame] | 166 | WSABUF sendbuf[2] = {{hdrlen, hdr},{datalen, data}}; |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 167 | DWORD value; |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 168 | |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 169 | SOCKCTL_(0, FIONBIO); /* switch Non-blocking I/O off */ |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 170 | |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 171 | value = WSASendTo(rawsock, sendbuf, 2, &value, 0 |
| 172 | , sock_in_addr(*(unsigned*)BUF->destipaddr,0), sizeof(struct sockaddr_in) |
| 173 | , NULL, NULL); |
| 174 | if(SOCKET_ERROR == value) TRAP(WSAGetLastError(), ignore, END_OF(ignore)); |
oliverschmidt | 2605c1c | 2004-08-02 11:36:34 +0000 | [diff] [blame] | 175 | |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 176 | SOCKCTL_(1, FIONBIO); /* switch Non-blocking I/O on */ |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 177 | } |
| 178 | /*-----------------------------------------------------------------------------------*/ |
| 179 | u16_t |
| 180 | rawsock_poll(void) |
| 181 | { |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 182 | static const int ignore[] = {WSAEWOULDBLOCK,WSAEMSGSIZE,WSANOTINITIALISED}; |
| 183 | const int nr = (create_proxyarp(),recv(rawsock, uip_buf, UIP_BUFSIZE, 0)); |
| 184 | return SOCKET_ERROR != nr ? nr : TRAP(WSAGetLastError(), ignore, END_OF(ignore)); |
ryohji | 55cd87a | 2007-05-02 04:06:56 +0000 | [diff] [blame] | 185 | } |
| 186 | /*-----------------------------------------------------------------------------------*/ |
| 187 | void |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 188 | create_proxyarp(void) |
ryohji | 55cd87a | 2007-05-02 04:06:56 +0000 | [diff] [blame] | 189 | { |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 190 | if(proxyaddr != HOSTADDR()) { |
| 191 | FARPROC f = DELAY_LOAD("CreateProxyArpEntry","iphlpapi"); |
| 192 | DWORD retval = (delete_proxyarp(),((ARPFUNC)f)(HOSTADDR(), NETMASK(), interfaceindex)); |
| 193 | proxyaddr = HOSTADDR(); |
| 194 | debug_printf("CreateProxyArpEntry(%s): %d\n", inet_ntoa(*(struct in_addr*)&proxyaddr), retval); |
oliverschmidt | 7f0b22b | 2005-01-29 23:10:43 +0000 | [diff] [blame] | 195 | } |
ryohji | 55cd87a | 2007-05-02 04:06:56 +0000 | [diff] [blame] | 196 | } |
| 197 | /*-----------------------------------------------------------------------------------*/ |
| 198 | void |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 199 | delete_proxyarp(void) |
ryohji | 55cd87a | 2007-05-02 04:06:56 +0000 | [diff] [blame] | 200 | { |
ryohji | 09e3a0d | 2007-05-06 13:01:52 +0000 | [diff] [blame] | 201 | if(proxyaddr) { |
| 202 | FARPROC f = DELAY_LOAD("DeleteProxyArpEntry","iphlpapi"); |
| 203 | DWORD retval = ((ARPFUNC)f)(HOSTADDR(), NETMASK(), interfaceindex); |
| 204 | debug_printf("DeleteProxyArpEntry(%s): %d\n", inet_ntoa(*(struct in_addr*)&proxyaddr), retval); |
| 205 | proxyaddr = 0; |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 206 | } |
oliverschmidt | de98fae | 2004-07-31 14:47:56 +0000 | [diff] [blame] | 207 | } |
| 208 | /*-----------------------------------------------------------------------------------*/ |