]> git.sur5r.net Git - i3/i3status/blob - src/first_network_device.c
Implement support for memory-usage on Linux
[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 bool is_virtual(const char *ifname) {
54     char path[1024];
55     char *target = NULL;
56     bool is_virtual = false;
57
58     snprintf(path, sizeof(path), "/sys/class/net/%s", ifname);
59     if ((target = realpath(path, NULL))) {
60         if (BEGINS_WITH(target, "/sys/devices/virtual/")) {
61             is_virtual = true;
62         }
63     }
64
65     free(target);
66     return is_virtual;
67 }
68
69 static net_type_t iface_type(const char *ifname) {
70 #ifdef __linux__
71     char devtype[32];
72
73     if (!sysfs_devtype(devtype, sizeof(devtype), ifname))
74         return NET_TYPE_OTHER;
75
76     /* Default to Ethernet when no devtype is available */
77     if (!devtype[0])
78         return NET_TYPE_ETHERNET;
79
80     if (strcmp(devtype, "wlan") == 0)
81         return NET_TYPE_WIRELESS;
82
83     if (strcmp(devtype, "wwan") == 0)
84         return NET_TYPE_OTHER;
85
86     return NET_TYPE_OTHER;
87 #elif __OpenBSD__
88     /*
89      *First determine if the device is a wireless device by trying two ioctl(2)
90      * commands against it. If either succeeds we can be sure it's a wireless
91      * device.
92      * Otherwise we try another ioctl(2) to determine the interface media. If that
93      * fails it's not an ethernet device eiter.
94      */
95     struct ifreq ifr;
96     struct ieee80211_bssid bssid;
97     struct ieee80211_nwid nwid;
98     struct ifmediareq ifmr;
99
100     int s, ibssid, inwid;
101
102     if ((s = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
103         return NET_TYPE_OTHER;
104
105     memset(&ifr, 0, sizeof(ifr));
106     ifr.ifr_data = (caddr_t)&nwid;
107     (void)strlcpy(ifr.ifr_name, ifname, sizeof(ifr.ifr_name));
108     inwid = ioctl(s, SIOCG80211NWID, (caddr_t)&ifr);
109
110     memset(&bssid, 0, sizeof(bssid));
111     strlcpy(bssid.i_name, ifname, sizeof(bssid.i_name));
112     ibssid = ioctl(s, SIOCG80211BSSID, &bssid);
113
114     if (ibssid == 0 || inwid == 0) {
115         close(s);
116         return NET_TYPE_WIRELESS;
117     }
118
119     (void)memset(&ifmr, 0, sizeof(ifmr));
120     (void)strlcpy(ifmr.ifm_name, ifname, sizeof(ifmr.ifm_name));
121
122     if (ioctl(s, SIOCGIFMEDIA, (caddr_t)&ifmr) < 0) {
123         close(s);
124         return NET_TYPE_OTHER;
125     } else {
126         close(s);
127         return NET_TYPE_ETHERNET;
128     }
129 #else
130 #error Missing implementation to determine interface type.
131 #endif
132 }
133
134 const char *first_eth_interface(const net_type_t type) {
135     static char *interface = NULL;
136     struct ifaddrs *ifaddr, *addrp;
137     net_type_t iftype;
138
139     getifaddrs(&ifaddr);
140
141     if (ifaddr == NULL)
142         return NULL;
143
144     free(interface);
145     interface = NULL;
146     for (addrp = ifaddr;
147          addrp != NULL;
148          addrp = addrp->ifa_next) {
149         if (strncasecmp(LOOPBACK_DEV, addrp->ifa_name, strlen(LOOPBACK_DEV)) == 0)
150             continue;
151         if (addrp->ifa_addr == NULL)
152             continue;
153 #ifdef __OpenBSD__
154         if (addrp->ifa_addr->sa_family != AF_LINK)
155             continue;
156 #else
157         // Skip PF_PACKET addresses (MAC addresses), as they are present on any
158         // ethernet interface.
159         if (addrp->ifa_addr->sa_family != AF_INET &&
160             addrp->ifa_addr->sa_family != AF_INET6)
161             continue;
162 #endif /* !__OpenBSD__ */
163         // Skip this interface if it is a wireless interface.
164         iftype = iface_type(addrp->ifa_name);
165         if (iftype != type)
166             continue;
167         if (is_virtual(addrp->ifa_name))
168             continue;
169         interface = strdup(addrp->ifa_name);
170         break;
171     }
172
173     freeifaddrs(ifaddr);
174     return interface;
175 }