]> git.sur5r.net Git - i3/i3status/blob - src/first_network_device.c
Merge pull request #202 from jasperla/openbsd_fdleak
[i3/i3status] / src / first_network_device.c
1 // vim:ts=4:sw=4:expandtab
2 #include <sys/socket.h>
3 #include <sys/stat.h>
4 #include <stdlib.h>
5 #include <ifaddrs.h>
6 #ifdef __OpenBSD__
7 #include <sys/types.h>
8 #include <sys/sockio.h>
9 #include <sys/ioctl.h>
10 #include <net/if.h>
11 #include <net80211/ieee80211.h>
12 #include <net80211/ieee80211_ioctl.h>
13 #endif
14
15 #include "i3status.h"
16
17 #ifdef __linux__
18 #define LOOPBACK_DEV "lo"
19 #else
20 #define LOOPBACK_DEV "lo0"
21 #endif
22
23 static bool sysfs_devtype(char *dest, size_t n, const char *ifnam) {
24     FILE *fp;
25     char buf[1024];
26
27     snprintf(buf, sizeof(buf), "/sys/class/net/%s/uevent", ifnam);
28     if ((fp = fopen(buf, "r")) == NULL)
29         return false;
30
31     dest[0] = '\0';
32
33     while (fgets(buf, sizeof(buf), fp)) {
34         size_t slen;
35         char *s;
36
37         slen = strlen(buf);
38         /* Line is too long to fit in the buffer */
39         if (buf[slen - 1] != '\n' && !feof(fp))
40             break;
41         if ((s = strchr(buf, '='))) {
42             if (strncmp(buf, "DEVTYPE", s - buf))
43                 continue;
44             buf[slen - 1] = '\0';
45             strncpy(dest, ++s, n);
46             break;
47         }
48     }
49     fclose(fp);
50     return true;
51 }
52
53 static net_type_t iface_type(const char *ifname) {
54 #ifdef __linux__
55     char devtype[32];
56
57     if (!sysfs_devtype(devtype, sizeof(devtype), ifname))
58         return NET_TYPE_OTHER;
59
60     if (!devtype[0])
61         return NET_TYPE_ETHERNET;
62
63     if (strcmp(devtype, "wlan") == 0)
64         return NET_TYPE_WIRELESS;
65
66     if (strcmp(devtype, "wwan") == 0)
67         return NET_TYPE_OTHER;
68
69     return NET_TYPE_OTHER;
70 #elif __OpenBSD__
71     /*
72      *First determine if the device is a wireless device by trying two ioctl(2)
73      * commands against it. If either succeeds we can be sure it's a wireless
74      * device.
75      * Otherwise we try another ioctl(2) to determine the interface media. If that
76      * fails it's not an ethernet device eiter.
77      */
78     struct ifreq ifr;
79     struct ieee80211_bssid bssid;
80     struct ieee80211_nwid nwid;
81     struct ifmediareq ifmr;
82
83     int s, ibssid, inwid;
84
85     if ((s = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
86         return NET_TYPE_OTHER;
87
88     memset(&ifr, 0, sizeof(ifr));
89     ifr.ifr_data = (caddr_t)&nwid;
90     (void)strlcpy(ifr.ifr_name, ifname, sizeof(ifr.ifr_name));
91     inwid = ioctl(s, SIOCG80211NWID, (caddr_t)&ifr);
92
93     memset(&bssid, 0, sizeof(bssid));
94     strlcpy(bssid.i_name, ifname, sizeof(bssid.i_name));
95     ibssid = ioctl(s, SIOCG80211BSSID, &bssid);
96
97     if (ibssid == 0 || inwid == 0) {
98         close(s);
99         return NET_TYPE_WIRELESS;
100     }
101
102     (void)memset(&ifmr, 0, sizeof(ifmr));
103     (void)strlcpy(ifmr.ifm_name, ifname, sizeof(ifmr.ifm_name));
104
105     if (ioctl(s, SIOCGIFMEDIA, (caddr_t)&ifmr) < 0) {
106         close(s);
107         return NET_TYPE_OTHER;
108     } else {
109         close(s);
110         return NET_TYPE_ETHERNET;
111     }
112 #else
113 #error Missing implementation to determine interface type.
114 #endif
115 }
116
117 const char *first_eth_interface(const net_type_t type) {
118     static char *interface = NULL;
119     struct ifaddrs *ifaddr, *addrp;
120     net_type_t iftype;
121
122     getifaddrs(&ifaddr);
123
124     if (ifaddr == NULL)
125         return NULL;
126
127     free(interface);
128     interface = NULL;
129     for (addrp = ifaddr;
130          addrp != NULL;
131          addrp = addrp->ifa_next) {
132         if (strncasecmp(LOOPBACK_DEV, addrp->ifa_name, strlen(LOOPBACK_DEV)) == 0)
133             continue;
134         if (addrp->ifa_addr == NULL)
135             continue;
136 #ifdef __OpenBSD__
137         if (addrp->ifa_addr->sa_family != AF_LINK)
138             continue;
139 #else
140         // Skip PF_PACKET addresses (MAC addresses), as they are present on any
141         // ethernet interface.
142         if (addrp->ifa_addr->sa_family != AF_INET &&
143             addrp->ifa_addr->sa_family != AF_INET6)
144             continue;
145 #endif /* !__OpenBSD__ */
146         // Skip this interface if it is a wireless interface.
147         iftype = iface_type(addrp->ifa_name);
148         if (iftype != type)
149             continue;
150         interface = strdup(addrp->ifa_name);
151         break;
152     }
153
154     freeifaddrs(ifaddr);
155     return interface;
156 }