Xinqi Bao's Git

wifi_perc: Fix file descriptor leak
[slstatus.git] / components / wifi.c
1 /* See LICENSE file for copyright and license details. */
2 #if defined(__linux__)
3 #include <errno.h>
4 #include <ifaddrs.h>
5 #include <linux/wireless.h>
6 #include <sys/socket.h>
7 #include <stdio.h>
8 #include <limits.h>
9 #include <string.h>
10 #include <sys/ioctl.h>
11 #include <unistd.h>
12
13 #include "../util.h"
14
15 const char *
16 wifi_perc(const char *iface)
17 {
18 int i, cur;
19 float perc;
20 int total = 70; /* the max of /proc/net/wireless */
21 char *p, *datastart;
22 char path[PATH_MAX];
23 char status[5];
24 FILE *fp;
25
26 snprintf(path, sizeof(path), "%s%s%s", "/sys/class/net/", iface,
27 "/operstate");
28 if (!(fp = fopen(path, "r"))) {
29 fprintf(stderr, "fopen '%s': %s\n", path,
30 strerror(errno));
31 return NULL;
32 }
33 p = fgets(status, 5, fp);
34 fclose(fp);
35 if(!p || strcmp(status, "up\n") != 0) {
36 return NULL;
37 }
38
39 if (!(fp = fopen("/proc/net/wireless", "r"))) {
40 fprintf(stderr, "fopen '/proc/net/wireless': %s\n",
41 strerror(errno));
42 return NULL;
43 }
44
45 for (i = 0; i < 3; i++) {
46 if (!(p = fgets(buf, sizeof(buf) - 1, fp)))
47 break;
48 }
49 fclose(fp);
50 if (i < 2 || !p)
51 return NULL;
52
53 if ((datastart = strstr(buf, iface)) == NULL)
54 return NULL;
55
56 datastart = (datastart+(strlen(iface)+1));
57 sscanf(datastart + 1, " %*d %d %*d %*d\t\t %*d\t "
58 "%*d\t\t%*d\t\t %*d\t %*d\t\t %*d", &cur);
59
60 perc = (float)cur / total * 100.0;
61
62 return bprintf("%.0f", perc);
63 }
64
65 const char *
66 wifi_essid(const char *iface)
67 {
68 static char id[IW_ESSID_MAX_SIZE+1];
69 int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
70 struct iwreq wreq;
71
72 memset(&wreq, 0, sizeof(struct iwreq));
73 wreq.u.essid.length = IW_ESSID_MAX_SIZE+1;
74 snprintf(wreq.ifr_name, sizeof(wreq.ifr_name), "%s", iface);
75
76 if (sockfd == -1) {
77 fprintf(stderr, "socket 'AF_INET': %s\n",
78 strerror(errno));
79 return NULL;
80 }
81 wreq.u.essid.pointer = id;
82 if (ioctl(sockfd,SIOCGIWESSID, &wreq) == -1) {
83 fprintf(stderr, "ioctl 'SIOCGIWESSID': %s\n",
84 strerror(errno));
85 close(sockfd);
86 return NULL;
87 }
88
89 close(sockfd);
90
91 if (strcmp(id, "") == 0)
92 return NULL;
93
94 return id;
95 }
96 #elif defined(__OpenBSD__)
97 /* unimplemented */
98 #endif