8 #if defined(NMRPFLASH_WINDOWS)
9 #define NMRPFLASH_NETALIAS_PREFIX "net"
15 #if defined(NMRPFLASH_LINUX)
16 #define NMRPFLASH_AF_PACKET AF_PACKET
17 #include <linux/if_packet.h>
19 #define NMRPFLASH_AF_PACKET AF_LINK
20 #include <net/if_types.h>
28 #ifndef NMRPFLASH_WINDOWS
38 const char *mac_to_str(uint8_t *mac)
41 snprintf(buf, sizeof(buf), "%02x:%02x:%02x:%02x:%02x:%02x",
42 mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
46 static int x_pcap_findalldevs(pcap_if_t **devs)
48 char errbuf[PCAP_ERRBUF_SIZE];
49 if (pcap_findalldevs(devs, errbuf) != 0) {
50 fprintf(stderr, "%s.\n", errbuf);
57 #ifndef NMRPFLASH_WINDOWS
58 static inline bool sockaddr_get_hwaddr(struct sockaddr *sa, uint8_t *hwaddr)
62 if (sa->sa_family != NMRPFLASH_AF_PACKET) {
66 #ifndef NMRPFLASH_LINUX
67 if (((struct sockaddr_dl*)sa)->sdl_type != IFT_ETHER) {
70 src = LLADDR((struct sockaddr_dl*)sa);
72 src = ((struct sockaddr_ll*)sa)->sll_addr;
75 memcpy(hwaddr, src, 6);
79 static bool get_intf_info(const char *intf, uint8_t *hwaddr, void *dummy)
81 struct ifaddrs *ifas, *ifa;
84 if (getifaddrs(&ifas) != 0) {
91 for (ifa = ifas; ifa; ifa = ifa->ifa_next) {
92 if (!strcmp(ifa->ifa_name, intf)) {
93 if (sockaddr_get_hwaddr(ifa->ifa_addr, hwaddr)) {
106 void win_perror2(const char *msg, DWORD err)
109 FormatMessageA(FORMAT_MESSAGE_ALLOCATE_BUFFER |
110 FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
111 NULL, err, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
112 (LPTSTR)&buf, 0, NULL);
115 /* FormatMessageA terminates buf with CRLF! */
116 fprintf(stderr, "%s: %s", msg, buf);
119 fprintf(stderr, "%s: error %d\n", msg, (int)err);
123 static bool get_intf_info(const char *intf, uint8_t *hwaddr, DWORD *index)
125 PIP_ADAPTER_INFO adapters, adapter;
130 if ((ret = GetAdaptersInfo(NULL, &bufLen)) != ERROR_BUFFER_OVERFLOW) {
131 win_perror2("GetAdaptersInfo", ret);
135 adapters = malloc(bufLen);
141 if ((ret = GetAdaptersInfo(adapters, &bufLen) == NO_ERROR)) {
142 for (adapter = adapters; adapter; adapter = adapter->Next) {
143 if (adapter->Type != MIB_IF_TYPE_ETHERNET) {
147 /* Interface names from WinPcap are "\Device\NPF_{GUID}", while
148 * AdapterName from GetAdaptersInfo is just "{GUID}".*/
149 if (strstr(intf, adapter->AdapterName)) {
150 if (adapter->AddressLength == 6) {
151 memcpy(hwaddr, adapter->Address, 6);
153 *index = adapter->Index;
161 win_perror2("GetAdaptersInfo", ret);
168 static const char *intf_alias_to_wpcap(const char *intf)
170 static char buf[128];
171 pcap_if_t *devs, *dev;
172 unsigned i = 0, dev_num = 0;
174 if (intf[0] == '\\') {
176 } else if (sscanf(intf, NMRPFLASH_NETALIAS_PREFIX "%u", &dev_num) != 1) {
177 fprintf(stderr, "Invalid interface alias.\n");
181 if (x_pcap_findalldevs(&devs) != 0) {
185 for (dev = devs; dev; dev = dev->next, ++i) {
188 printf("%s%u: %s\n", NMRPFLASH_NETALIAS_PREFIX, i, dev->name);
190 strncpy(buf, dev->name, sizeof(buf) - 1);
191 buf[sizeof(buf) - 1] = '\0';
196 pcap_freealldevs(devs);
199 fprintf(stderr, "Interface alias not found.\n");
206 static const char *intf_get_pretty_name(const char *intf)
208 static char buf[512];
214 guid = strstr(intf, "NPF_{");
221 snprintf(buf, sizeof(buf),
222 "System\\CurrentControlSet\\Control\\Network\\"
223 "{4D36E972-E325-11CE-BFC1-08002BE10318}\\"
224 "%s\\Connection", guid);
225 err = RegOpenKeyExA(HKEY_LOCAL_MACHINE, buf, 0, KEY_READ, &hkey);
226 if (err != ERROR_SUCCESS) {
228 win_perror2("RegOpenKeyExA", err);
234 err = RegQueryValueExA(hkey, "Name", NULL, NULL, (LPBYTE)buf, &len);
235 if (err == ERROR_SUCCESS) {
239 win_perror2("RegQueryValueExA", err);
249 inline uint8_t *ethsock_get_hwaddr(struct ethsock *sock)
254 struct ethsock *ethsock_create(const char *intf, uint16_t protocol)
256 char buf[PCAP_ERRBUF_SIZE];
257 struct bpf_program fp;
258 struct ethsock *sock;
261 sock = malloc(sizeof(struct ethsock));
267 #ifdef NMRPFLASH_WINDOWS
268 intf = intf_alias_to_wpcap(intf);
277 sock->pcap = pcap_open_live(sock->intf, BUFSIZ, 1, 1, buf);
279 fprintf(stderr, "%s.\n", buf);
284 fprintf(stderr, "Warning: %s.\n", buf);
287 if (pcap_datalink(sock->pcap) != DLT_EN10MB) {
288 fprintf(stderr, "%s is not an ethernet interface.\n",
293 #ifndef NMRPFLASH_WINDOWS
294 err = !get_intf_info(intf, sock->hwaddr, NULL);
296 err = !get_intf_info(intf, sock->hwaddr, &sock->index);
299 fprintf(stderr, "Failed to get interface info.\n");
303 #ifndef NMRPFLASH_WINDOWS
304 sock->fd = pcap_get_selectable_fd(sock->pcap);
305 if (sock->fd == -1) {
306 pcap_perror(sock->pcap, "pcap_get_selectable_fd");
310 sock->handle = pcap_getevent(sock->pcap);
312 pcap_perror(sock->pcap, "pcap_getevent");
316 err = pcap_setmintocopy(sock->pcap, 1);
318 pcap_perror(sock->pcap, "pcap_setmintocopy");
323 snprintf(buf, sizeof(buf), "ether proto 0x%04x and not ether src %s",
324 protocol, mac_to_str(sock->hwaddr));
326 err = pcap_compile(sock->pcap, &fp, buf, 0, 0);
328 pcap_perror(sock->pcap, "pcap_compile");
332 err = pcap_setfilter(sock->pcap, &fp);
336 pcap_perror(sock->pcap, "pcap_setfilter");
343 pcap_close(sock->pcap);
349 int select_fd(int fd, unsigned timeout)
358 tv.tv_sec = timeout / 1000;
359 tv.tv_usec = 1000 * (timeout % 1000);
361 status = select(fd + 1, &fds, NULL, NULL, &tv);
363 sock_perror("select");
369 ssize_t ethsock_recv(struct ethsock *sock, void *buf, size_t len)
371 struct pcap_pkthdr* hdr;
372 const u_char *capbuf;
374 #ifdef NMRPFLASH_WINDOWS
378 ret = WaitForSingleObject(sock->handle, sock->timeout);
379 if (ret == WAIT_TIMEOUT) {
381 } else if (ret != WAIT_OBJECT_0) {
382 win_perror2("WaitForSingleObject", ret);
388 status = select_fd(sock->fd, sock->timeout);
391 } else if (status == 0) {
397 status = pcap_next_ex(sock->pcap, &hdr, &capbuf);
400 memcpy(buf, capbuf, MIN(len, hdr->caplen));
405 pcap_perror(sock->pcap, "pcap_next_ex");
408 fprintf(stderr, "pcap_next_ex: returned %d.\n", status);
413 int ethsock_send(struct ethsock *sock, void *buf, size_t len)
415 #ifdef NMRPFLASH_WINDOWS
416 if (pcap_sendpacket(sock->pcap, buf, len) == 0) {
419 pcap_perror(sock->pcap, "pcap_sendpacket");
423 if (pcap_inject(sock->pcap, buf, len) == len) {
426 pcap_perror(sock->pcap, "pcap_inject");
432 int ethsock_close(struct ethsock *sock)
434 pcap_close(sock->pcap);
439 inline int ethsock_set_timeout(struct ethsock *sock, unsigned msec)
441 sock->timeout = msec;
445 #ifndef NMRPFLASH_WINDOWS
446 int ethsock_arp_add(struct ethsock *sock, uint8_t *hwaddr, struct in_addr *ipaddr)
451 int ethsock_arp_del(struct ethsock *sock, uint8_t *hwaddr, struct in_addr *ipaddr)
456 static int ethsock_arp(struct ethsock *sock, uint8_t *hwaddr, struct in_addr *ipaddr, int add)
460 .dwIndex = sock->index,
462 .dwAddr = ipaddr->s_addr,
463 .dwType = MIB_IPNET_TYPE_STATIC
466 memcpy(arp.bPhysAddr, hwaddr, 6);
469 ret = CreateIpNetEntry(&arp);
470 if (ret != NO_ERROR) {
471 win_perror2("CreateIpNetEntry", ret);
475 DeleteIpNetEntry(&arp);
481 int ethsock_arp_add(struct ethsock *sock, uint8_t *hwaddr, struct in_addr *ipaddr)
483 ethsock_arp_del(sock, hwaddr, ipaddr);
484 return ethsock_arp(sock, hwaddr, ipaddr, 1);
487 int ethsock_arp_del(struct ethsock *sock, uint8_t *hwaddr, struct in_addr *ipaddr)
489 return ethsock_arp(sock, hwaddr, ipaddr, 0);
493 static bool get_hwaddr_from_pcap(const pcap_if_t *dev, uint8_t *hwaddr)
495 #ifndef NMRPFLASH_WINDOWS
499 for (addr = dev->addresses; addr; addr = addr->next) {
501 printf("%s: sa_family=%d, sa_data={ ", dev->name,
502 addr->addr->sa_family);
503 for (i = 0; i != sizeof(addr->addr->sa_data); ++i) {
504 printf("%02x ", addr->addr->sa_data[i] & 0xff);
509 if (sockaddr_get_hwaddr(addr->addr, hwaddr)) {
515 return get_intf_info(dev->name, hwaddr, NULL);
518 int ethsock_list_all(void)
520 pcap_if_t *devs, *dev;
523 unsigned dev_num = 0, dev_ok = 0;
524 #ifdef NMRPFLASH_WINDOWS
528 if (x_pcap_findalldevs(&devs) != 0) {
532 memset(hwaddr, 0, 6);
534 for (dev = devs; dev; dev = dev->next, ++dev_num) {
535 if (dev->flags & PCAP_IF_LOOPBACK) {
537 printf("%-15s (loopback device)\n", dev->name);
542 if (!get_hwaddr_from_pcap(dev, hwaddr)) {
544 printf("%-15s (not an ethernet device)\n",
550 #ifndef NMRPFLASH_WINDOWS
551 printf("%-15s", dev->name);
553 /* Call this here so *_perror() calls don't happen within a line */
554 pretty = intf_get_pretty_name(dev->name);
557 printf("%s%u", NMRPFLASH_NETALIAS_PREFIX, dev_num);
559 printf("%s", dev->name);
563 for (addr = dev->addresses; addr; addr = addr->next) {
564 if (addr->addr->sa_family == AF_INET) {
566 inet_ntoa(((struct sockaddr_in*)addr->addr)->sin_addr));
572 printf(" %-15s", "0.0.0.0");
575 printf(" %s", mac_to_str(hwaddr));
577 #ifdef NMRPFLASH_WINDOWS
579 printf(" (%s)", pretty);
580 } else if (dev->description) {
581 printf(" (%s)", dev->description);
590 printf("No suitable network interfaces found.\n");
596 int ethsock_for_each_ip(struct ethsock *sock, ethsock_ip_callback_t callback,
599 struct ethsock_ip_callback_args args;
600 pcap_if_t *devs, *dev;
604 if (x_pcap_findalldevs(&devs) != 0) {
610 for (dev = devs; dev; dev = dev->next) {
611 if (strcmp(sock->intf, dev->name)) {
615 for (addr = dev->addresses; addr; addr = addr->next) {
616 if (addr->addr->sa_family == AF_INET) {
617 args.ipaddr = &((struct sockaddr_in*)addr->addr)->sin_addr;
618 args.ipmask = &((struct sockaddr_in*)addr->netmask)->sin_addr;
620 status = callback(&args);
630 pcap_freealldevs(devs);
632 return status <= 0 ? status : 0;