8 #if defined(NMRPFLASH_WINDOWS)
9 #define NMRPFLASH_NETALIAS_PREFIX "net"
15 #if defined(NMRPFLASH_LINUX)
16 #include <linux/if_packet.h>
17 #elif defined (NMRPFLASH_OSX)
18 #include <net/if_dl.h>
23 #define MIN(a, b) ((a) < (b) ? (a) : (b))
30 #ifndef NMRPFLASH_WINDOWS
39 const char *mac_to_str(uint8_t *mac)
42 snprintf(buf, sizeof(buf), "%02x:%02x:%02x:%02x:%02x:%02x",
43 mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
47 static int x_pcap_findalldevs(pcap_if_t **devs)
49 char errbuf[PCAP_ERRBUF_SIZE];
50 if (pcap_findalldevs(devs, errbuf) != 0) {
51 fprintf(stderr, "%s.\n", errbuf);
58 #ifndef NMRPFLASH_WINDOWS
59 static bool get_hwaddr(uint8_t *hwaddr, const char *intf)
61 struct ifaddrs *ifas, *ifa;
65 if (getifaddrs(&ifas) != 0) {
72 for (ifa = ifas; ifa; ifa = ifa->ifa_next) {
73 if (!strcmp(ifa->ifa_name, intf)) {
74 #ifdef NMRPFLASH_LINUX
75 if (ifa->ifa_addr->sa_family != AF_PACKET) {
78 src = ((struct sockaddr_ll*)ifa->ifa_addr)->sll_addr;
80 if (ifa->ifa_addr->sa_family != AF_LINK) {
83 src = LLADDR((struct sockaddr_dl*)ifa->ifa_addr);
85 memcpy(hwaddr, src, 6);
96 void win_perror2(const char *msg, DWORD err)
99 FormatMessageA(FORMAT_MESSAGE_ALLOCATE_BUFFER |
100 FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
101 NULL, err, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
102 (LPTSTR)&buf, 0, NULL);
105 /* FormatMessageA terminates buf with CRLF! */
106 fprintf(stderr, "%s: %s", msg, buf);
109 fprintf(stderr, "%s: error %d\n", msg, (int)err);
113 static bool get_hwaddr(uint8_t *hwaddr, const char *intf)
115 PIP_ADAPTER_INFO adapters, adapter;
120 if ((ret = GetAdaptersInfo(NULL, &bufLen)) != ERROR_BUFFER_OVERFLOW) {
121 win_perror2("GetAdaptersInfo", ret);
125 adapters = malloc(bufLen);
131 if ((ret = GetAdaptersInfo(adapters, &bufLen) == NO_ERROR)) {
132 for (adapter = adapters; adapter; adapter = adapter->Next) {
133 if (adapter->Type != MIB_IF_TYPE_ETHERNET) {
137 #ifndef NMRPFLASH_WINDOWS
138 if (!strcmp(intf, adapter->AdapterName))
140 /* Interface names from WinPcap are "\Device\NPF_{GUID}", while
141 * AdapterName from GetAdaptersInfo is just "{GUID}".*/
142 if (strstr(intf, adapter->AdapterName))
145 if (adapter->AddressLength == 6) {
146 for (i = 0; i != 6; ++i) {
147 hwaddr[i] = adapter->Address[i];
156 win_perror2("GetAdaptersInfo", ret);
163 static const char *intf_alias_to_wpcap(const char *intf)
165 static char buf[128];
166 pcap_if_t *devs, *dev;
167 unsigned i = 0, dev_num = 0;
169 if (intf[0] == '\\') {
171 } else if (sscanf(intf, NMRPFLASH_NETALIAS_PREFIX "%u", &dev_num) != 1) {
172 fprintf(stderr, "Invalid interface alias.\n");
176 if (x_pcap_findalldevs(&devs) != 0) {
180 for (dev = devs; dev; dev = dev->next, ++i) {
183 printf("%s%u: %s\n", NMRPFLASH_NETALIAS_PREFIX, i, dev->name);
185 strncpy(buf, dev->name, sizeof(buf) - 1);
186 buf[sizeof(buf) - 1] = '\0';
191 pcap_freealldevs(devs);
194 fprintf(stderr, "Interface alias not found.\n");
201 static const char *intf_get_pretty_name(const char *intf)
203 static char buf[512];
209 guid = strstr(intf, "NPF_{");
216 snprintf(buf, sizeof(buf),
217 "System\\CurrentControlSet\\Control\\Network\\"
218 "{4D36E972-E325-11CE-BFC1-08002BE10318}\\"
219 "%s\\Connection", guid);
220 err = RegOpenKeyExA(HKEY_LOCAL_MACHINE, buf, 0, KEY_READ, &hkey);
221 if (err != ERROR_SUCCESS) {
223 win_perror2("RegOpenKeyExA", err);
229 err = RegQueryValueExA(hkey, "Name", NULL, NULL, (LPBYTE)buf, &len);
230 if (err == ERROR_SUCCESS) {
234 win_perror2("RegQueryValueExA", err);
245 inline uint8_t *ethsock_get_hwaddr(struct ethsock *sock)
250 struct ethsock *ethsock_create(const char *intf, uint16_t protocol)
252 char buf[PCAP_ERRBUF_SIZE];
253 struct bpf_program fp;
254 struct ethsock *sock;
257 sock = malloc(sizeof(struct ethsock));
263 #ifdef NMRPFLASH_WINDOWS
264 intf = intf_alias_to_wpcap(intf);
272 sock->pcap = pcap_open_live(intf, BUFSIZ, 1, 1, buf);
274 fprintf(stderr, "%s.\n", buf);
279 fprintf(stderr, "Warning: %s.\n", buf);
282 if (pcap_datalink(sock->pcap) != DLT_EN10MB) {
283 fprintf(stderr, "%s is not an ethernet interface.\n",
288 if (!get_hwaddr(sock->hwaddr, intf)) {
289 fprintf(stderr, "Failed to get MAC address of interface.\n");
293 #ifndef NMRPFLASH_WINDOWS
294 sock->fd = pcap_get_selectable_fd(sock->pcap);
295 if (sock->fd == -1) {
296 fprintf(stderr, "No selectable file descriptor available.\n");
300 sock->handle = pcap_getevent(sock->pcap);
302 fprintf(stderr, "No event handle available.\n");
306 err = pcap_setmintocopy(sock->pcap, 1);
308 pcap_perror(sock->pcap, "pcap_setmintocopy");
313 snprintf(buf, sizeof(buf), "ether proto 0x%04x and not ether src %s",
314 protocol, mac_to_str(sock->hwaddr));
316 err = pcap_compile(sock->pcap, &fp, buf, 0, 0);
318 pcap_perror(sock->pcap, "pcap_compile");
322 if ((err = pcap_setfilter(sock->pcap, &fp))) {
323 pcap_perror(sock->pcap, "pcap_setfilter");
330 pcap_close(sock->pcap);
336 int select_fd(int fd, unsigned timeout)
345 tv.tv_sec = timeout / 1000;
346 tv.tv_usec = 1000 * (timeout % 1000);
348 status = select(fd + 1, &fds, NULL, NULL, &tv);
350 sock_perror("select");
356 ssize_t ethsock_recv(struct ethsock *sock, void *buf, size_t len)
358 struct pcap_pkthdr* hdr;
359 const u_char *capbuf;
361 #ifdef NMRPFLASH_WINDOWS
365 ret = WaitForSingleObject(sock->handle, sock->timeout);
366 if (ret == WAIT_TIMEOUT) {
368 } else if (ret != WAIT_OBJECT_0) {
369 win_perror2("WaitForSingleObject", ret);
375 status = select_fd(sock->fd, sock->timeout);
378 } else if (status == 0) {
384 status = pcap_next_ex(sock->pcap, &hdr, &capbuf);
387 memcpy(buf, capbuf, MIN(len, hdr->caplen));
392 pcap_perror(sock->pcap, "pcap_next_ex");
395 fprintf(stderr, "pcap_next_ex: returned %d.\n", status);
400 int ethsock_send(struct ethsock *sock, void *buf, size_t len)
402 #ifdef NMRPFLASH_WINDOWS
403 if (pcap_sendpacket(sock->pcap, buf, len) == 0) {
406 pcap_perror(sock->pcap, "pcap_sendpacket");
410 if (pcap_inject(sock->pcap, buf, len) == len) {
413 pcap_perror(sock->pcap, "pcap_inject");
419 int ethsock_close(struct ethsock *sock)
421 pcap_close(sock->pcap);
426 inline int ethsock_set_timeout(struct ethsock *sock, unsigned msec)
428 sock->timeout = msec;
432 static bool is_ethernet(const char *intf)
435 char errbuf[PCAP_ERRBUF_SIZE];
438 if ((pcap = pcap_create(intf, errbuf))) {
439 if (pcap_activate(pcap) == 0) {
440 ret = (pcap_datalink(pcap) == DLT_EN10MB);
448 int ethsock_list_all(void)
450 pcap_if_t *devs, *dev;
453 unsigned dev_num = 0, dev_ok = 0;
454 #ifdef NMRPFLASH_WINDOWS
458 if (x_pcap_findalldevs(&devs) != 0) {
462 memset(hwaddr, 0, 6);
464 for (dev = devs; dev; dev = dev->next, ++dev_num) {
465 if (dev->flags & PCAP_IF_LOOPBACK) {
467 printf("%-15s (loopback device)\n", dev->name);
472 if (!is_ethernet(dev->name)) {
474 printf("%-15s (not an ethernet device)\n",
480 if (!get_hwaddr(hwaddr, dev->name)) {
482 printf("%-15s (failed to get hardware address)\n",
488 #ifndef NMRPFLASH_WINDOWS
489 printf("%-15s", dev->name);
491 /* Call this here so *_perror() calls don't happen within a line */
492 pretty = intf_get_pretty_name(dev->name);
495 printf("%s%u", NMRPFLASH_NETALIAS_PREFIX, dev_num);
497 printf("%s", dev->name);
501 for (addr = dev->addresses; addr; addr = addr->next) {
502 if (addr->addr->sa_family == AF_INET) {
504 inet_ntoa(((struct sockaddr_in*)addr->addr)->sin_addr));
510 printf(" %-15s", "0.0.0.0");
513 printf(" %02x:%02x:%02x:%02x:%02x:%02x", hwaddr[0], hwaddr[1],
514 hwaddr[2], hwaddr[3], hwaddr[4], hwaddr[5]);
516 #ifdef NMRPFLASH_WINDOWS
518 printf(" (%s)", pretty);
519 } else if (dev->description) {
520 printf(" (%s)", dev->description);
529 printf("No suitable network interfaces found.\n");