Xinqi Bao's Git

refactor vol_perc to not depend on alsa libraries
[slstatus.git] / slstatus.c
1 /* See LICENSE file for copyright and license details. */
2
3 #include <err.h>
4 #include <fcntl.h>
5 #include <ifaddrs.h>
6 #include <limits.h>
7 #include <linux/wireless.h>
8 #include <locale.h>
9 #include <netdb.h>
10 #include <pwd.h>
11 #include <signal.h>
12 #include <stdarg.h>
13 #include <stdio.h>
14 #include <stdlib.h>
15 #include <string.h>
16 #include <sys/ioctl.h>
17 #include <sys/stat.h>
18 #include <sys/statvfs.h>
19 #include <sys/socket.h>
20 #include <sys/soundcard.h>
21 #include <sys/sysinfo.h>
22 #include <sys/types.h>
23 #include <sys/utsname.h>
24 #include <time.h>
25 #include <unistd.h>
26 #include <X11/Xlib.h>
27
28 #undef strlcat
29 #undef strlcpy
30
31 #include "extern/arg.h"
32 #include "extern/strlcat.h"
33 #include "extern/strlcpy.h"
34 #include "extern/concat.h"
35
36 struct arg {
37 char *(*func)();
38 const char *fmt;
39 const char *args;
40 };
41
42 static char *smprintf(const char *fmt, ...);
43 static char *battery_perc(const char *bat);
44 static char *battery_state(const char *bat);
45 static char *cpu_perc(void);
46 static char *datetime(const char *fmt);
47 static char *disk_free(const char *mnt);
48 static char *disk_perc(const char *mnt);
49 static char *disk_total(const char *mnt);
50 static char *disk_used(const char *mnt);
51 static char *entropy(void);
52 static char *gid(void);
53 static char *hostname(void);
54 static char *ip(const char *iface);
55 static char *load_avg(void);
56 static char *ram_free(void);
57 static char *ram_perc(void);
58 static char *ram_used(void);
59 static char *ram_total(void);
60 static char *run_command(const char *cmd);
61 static char *swap_free(void);
62 static char *swap_perc(void);
63 static char *swap_used(void);
64 static char *swap_total(void);
65 static char *temp(const char *file);
66 static char *uid(void);
67 static char *uptime(void);
68 static char *username(void);
69 static char *vol_perc(const char *card);
70 static char *wifi_perc(const char *iface);
71 static char *wifi_essid(const char *iface);
72 static char *kernel_release(void);
73 static void set_status(const char *str);
74 static void sighandler(const int signo);
75 static void usage(void);
76
77 char *argv0;
78 char concat[];
79 static unsigned short int delay;
80 static unsigned short int done;
81 static unsigned short int dflag, oflag;
82 static Display *dpy;
83
84 #include "config.h"
85
86 static char *
87 smprintf(const char *fmt, ...)
88 {
89 va_list ap;
90 char *ret;
91 int len;
92
93 va_start(ap, fmt);
94 len = vsnprintf(NULL, 0, fmt, ap);
95 va_end(ap);
96
97 ret = malloc(++len);
98 if (ret == NULL) {
99 err(1, "malloc");
100 }
101
102 va_start(ap, fmt);
103 vsnprintf(ret, len, fmt, ap);
104 va_end(ap);
105
106 return ret;
107 }
108
109 static char *
110 battery_perc(const char *bat)
111 {
112 int perc;
113 FILE *fp;
114
115 ccat(3, "/sys/class/power_supply/", bat, "/capacity");
116 fp = fopen(concat, "r");
117 if (fp == NULL) {
118 warn("Failed to open file %s", concat);
119 return smprintf(UNKNOWN_STR);
120 }
121 fscanf(fp, "%i", &perc);
122 fclose(fp);
123
124 return smprintf("%d%%", perc);
125 }
126
127 static char *
128 battery_state(const char *bat)
129 {
130 char state[12];
131 FILE *fp;
132
133 ccat(3, "/sys/class/power_supply/", bat, "/status");
134 fp = fopen(concat, "r");
135 if (fp == NULL) {
136 warn("Failed to open file %s", concat);
137 return smprintf(UNKNOWN_STR);
138 }
139 fscanf(fp, "%12s", state);
140 fclose(fp);
141
142 if (strcmp(state, "Charging") == 0) {
143 return smprintf("+");
144 } else if (strcmp(state, "Discharging") == 0) {
145 return smprintf("-");
146 } else if (strcmp(state, "Full") == 0) {
147 return smprintf("=");
148 } else {
149 return smprintf("?");
150 }
151 }
152
153 static char *
154 cpu_perc(void)
155 {
156 int perc;
157 long double a[4], b[4];
158 FILE *fp;
159
160 fp = fopen("/proc/stat", "r");
161 if (fp == NULL) {
162 warn("Failed to open file /proc/stat");
163 return smprintf(UNKNOWN_STR);
164 }
165 fscanf(fp, "%*s %Lf %Lf %Lf %Lf", &a[0], &a[1], &a[2], &a[3]);
166 fclose(fp);
167
168 delay = (UPDATE_INTERVAL - (UPDATE_INTERVAL - 1));
169 sleep(delay);
170
171 fp = fopen("/proc/stat", "r");
172 if (fp == NULL) {
173 warn("Failed to open file /proc/stat");
174 return smprintf(UNKNOWN_STR);
175 }
176 fscanf(fp, "%*s %Lf %Lf %Lf %Lf", &b[0], &b[1], &b[2], &b[3]);
177 fclose(fp);
178
179 perc = 100 * ((b[0]+b[1]+b[2]) - (a[0]+a[1]+a[2])) / ((b[0]+b[1]+b[2]+b[3]) - (a[0]+a[1]+a[2]+a[3]));
180 return smprintf("%d%%", perc);
181 }
182
183 static char *
184 datetime(const char *fmt)
185 {
186 time_t t;
187 char str[80];
188
189 t = time(NULL);
190 if (strftime(str, sizeof(str), fmt, localtime(&t)) == 0) {
191 return smprintf(UNKNOWN_STR);
192 }
193
194 return smprintf("%s", str);
195 }
196
197 static char *
198 disk_free(const char *mnt)
199 {
200 struct statvfs fs;
201
202 if (statvfs(mnt, &fs) < 0) {
203 warn("Failed to get filesystem info");
204 return smprintf(UNKNOWN_STR);
205 }
206
207 return smprintf("%f", (float)fs.f_bsize * (float)fs.f_bfree / 1024 / 1024 / 1024);
208 }
209
210 static char *
211 disk_perc(const char *mnt)
212 {
213 int perc;
214 struct statvfs fs;
215
216 if (statvfs(mnt, &fs) < 0) {
217 warn("Failed to get filesystem info");
218 return smprintf(UNKNOWN_STR);
219 }
220
221 perc = 100 * (1.0f - ((float)fs.f_bfree / (float)fs.f_blocks));
222
223 return smprintf("%d%%", perc);
224 }
225
226 static char *
227 disk_total(const char *mnt)
228 {
229 struct statvfs fs;
230
231 if (statvfs(mnt, &fs) < 0) {
232 warn("Failed to get filesystem info");
233 return smprintf(UNKNOWN_STR);
234 }
235
236 return smprintf("%f", (float)fs.f_bsize * (float)fs.f_blocks / 1024 / 1024 / 1024);
237 }
238
239 static char *
240 disk_used(const char *mnt)
241 {
242 struct statvfs fs;
243
244 if (statvfs(mnt, &fs) < 0) {
245 warn("Failed to get filesystem info");
246 return smprintf(UNKNOWN_STR);
247 }
248
249 return smprintf("%f", (float)fs.f_bsize * ((float)fs.f_blocks - (float)fs.f_bfree) / 1024 / 1024 / 1024);
250 }
251
252 static char *
253 entropy(void)
254 {
255 int num;
256 FILE *fp;
257
258 fp= fopen("/proc/sys/kernel/random/entropy_avail", "r");
259 if (fp == NULL) {
260 warn("Failed to open file /proc/sys/kernel/random/entropy_avail");
261 return smprintf(UNKNOWN_STR);
262 }
263 fscanf(fp, "%d", &num);
264 fclose(fp);
265
266 return smprintf("%d", num);
267 }
268
269 static char *
270 gid(void)
271 {
272 return smprintf("%d", getgid());
273 }
274
275 static char *
276 hostname(void)
277 {
278 char buf[HOST_NAME_MAX];
279
280 if (gethostname(buf, sizeof(buf)) == -1) {
281 warn("hostname");
282 return smprintf(UNKNOWN_STR);
283 }
284
285 return smprintf("%s", buf);
286 }
287
288 static char *
289 ip(const char *iface)
290 {
291 struct ifaddrs *ifaddr, *ifa;
292 int s;
293 char host[NI_MAXHOST];
294
295 if (getifaddrs(&ifaddr) == -1) {
296 warn("Failed to get IP address for interface %s", iface);
297 return smprintf(UNKNOWN_STR);
298 }
299
300 for (ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) {
301 if (ifa->ifa_addr == NULL) {
302 continue;
303 }
304 s = getnameinfo(ifa->ifa_addr, sizeof(struct sockaddr_in), host, NI_MAXHOST, NULL, 0, NI_NUMERICHOST);
305 if ((strcmp(ifa->ifa_name, iface) == 0) && (ifa->ifa_addr->sa_family == AF_INET)) {
306 if (s != 0) {
307 warnx("Failed to get IP address for interface %s", iface);
308 return smprintf(UNKNOWN_STR);
309 }
310 return smprintf("%s", host);
311 }
312 }
313
314 freeifaddrs(ifaddr);
315
316 return smprintf(UNKNOWN_STR);
317 }
318
319 static char *
320 load_avg(void)
321 {
322 double avgs[3];
323
324 if (getloadavg(avgs, 3) < 0) {
325 warnx("Failed to get the load avg");
326 return smprintf(UNKNOWN_STR);
327 }
328
329 return smprintf("%.2f %.2f %.2f", avgs[0], avgs[1], avgs[2]);
330 }
331
332 static char *
333 ram_free(void)
334 {
335 long free;
336 FILE *fp;
337
338 fp = fopen("/proc/meminfo", "r");
339 if (fp == NULL) {
340 warn("Failed to open file /proc/meminfo");
341 return smprintf(UNKNOWN_STR);
342 }
343 fscanf(fp, "MemFree: %ld kB\n", &free);
344 fclose(fp);
345
346 return smprintf("%f", (float)free / 1024 / 1024);
347 }
348
349 static char *
350 ram_perc(void)
351 {
352 long total, free, buffers, cached;
353 FILE *fp;
354
355 fp = fopen("/proc/meminfo", "r");
356 if (fp == NULL) {
357 warn("Failed to open file /proc/meminfo");
358 return smprintf(UNKNOWN_STR);
359 }
360 fscanf(fp, "MemTotal: %ld kB\n", &total);
361 fscanf(fp, "MemFree: %ld kB\n", &free);
362 fscanf(fp, "MemAvailable: %ld kB\nBuffers: %ld kB\n", &buffers, &buffers);
363 fscanf(fp, "Cached: %ld kB\n", &cached);
364 fclose(fp);
365
366 return smprintf("%d%%", 100 * ((total - free) - (buffers + cached)) / total);
367 }
368
369 static char *
370 ram_total(void)
371 {
372 long total;
373 FILE *fp;
374
375 fp = fopen("/proc/meminfo", "r");
376 if (fp == NULL) {
377 warn("Failed to open file /proc/meminfo");
378 return smprintf(UNKNOWN_STR);
379 }
380 fscanf(fp, "MemTotal: %ld kB\n", &total);
381 fclose(fp);
382
383 return smprintf("%f", (float)total / 1024 / 1024);
384 }
385
386 static char *
387 ram_used(void)
388 {
389 long free, total, buffers, cached;
390 FILE *fp;
391
392 fp = fopen("/proc/meminfo", "r");
393 if (fp == NULL) {
394 warn("Failed to open file /proc/meminfo");
395 return smprintf(UNKNOWN_STR);
396 }
397 fscanf(fp, "MemTotal: %ld kB\n", &total);
398 fscanf(fp, "MemFree: %ld kB\n", &free);
399 fscanf(fp, "MemAvailable: %ld kB\nBuffers: %ld kB\n", &buffers, &buffers);
400 fscanf(fp, "Cached: %ld kB\n", &cached);
401 fclose(fp);
402
403 return smprintf("%f", (float)(total - free - buffers - cached) / 1024 / 1024);
404 }
405
406 static char *
407 run_command(const char *cmd)
408 {
409 FILE *fp;
410 char buf[1024] = "n/a";
411
412 fp = popen(cmd, "r");
413 if (fp == NULL) {
414 warn("Failed to get command output for %s", cmd);
415 return smprintf(UNKNOWN_STR);
416 }
417 fgets(buf, sizeof(buf)-1, fp);
418 pclose(fp);
419
420 buf[strlen(buf)] = '\0';
421 strtok(buf, "\n");
422
423 return smprintf("%s", buf);
424 }
425
426 static char *
427 swap_free(void)
428 {
429 long total, free;
430 FILE *fp;
431 char buf[2048];
432 size_t bytes_read;
433 char *match;
434
435 fp = fopen("/proc/meminfo", "r");
436 if (fp == NULL) {
437 warn("Failed to open file /proc/meminfo");
438 return smprintf(UNKNOWN_STR);
439 }
440 bytes_read = fread(buf, sizeof(char), sizeof(buf), fp);
441 buf[bytes_read] = '\0';
442 fclose(fp);
443 if (bytes_read == 0 || bytes_read == sizeof(buf)) {
444 warn("Failed to read /proc/meminfo\n");
445 return smprintf(UNKNOWN_STR);
446 }
447
448 match = strstr(buf, "SwapTotal");
449 sscanf(match, "SwapTotal: %ld kB\n", &total);
450 if (total == 0) {
451 return smprintf(UNKNOWN_STR);
452 }
453
454 match = strstr(buf, "SwapFree");
455 sscanf(match, "SwapFree: %ld kB\n", &free);
456
457 return smprintf("%f", (float)free / 1024 / 1024);
458 }
459
460 static char *
461 swap_perc(void)
462 {
463 long total, free, cached;
464 FILE *fp;
465 char buf[2048];
466 size_t bytes_read;
467 char *match;
468
469 fp = fopen("/proc/meminfo", "r");
470 if (fp == NULL) {
471 warn("Failed to open file /proc/meminfo");
472 return smprintf(UNKNOWN_STR);
473 }
474 bytes_read = fread(buf, sizeof(char), sizeof(buf), fp);
475 buf[bytes_read] = '\0';
476 fclose(fp);
477 if (bytes_read == 0 || bytes_read == sizeof(buf)) {
478 warn("Failed to read /proc/meminfo\n");
479 return smprintf(UNKNOWN_STR);
480 }
481
482 match = strstr(buf, "SwapTotal");
483 sscanf(match, "SwapTotal: %ld kB\n", &total);
484 if (total == 0) {
485 return smprintf(UNKNOWN_STR);
486 }
487
488 match = strstr(buf, "SwapCached");
489 sscanf(match, "SwapCached: %ld kB\n", &cached);
490
491 match = strstr(buf, "SwapFree");
492 sscanf(match, "SwapFree: %ld kB\n", &free);
493
494 return smprintf("%d%%", 100 * (total - free - cached) / total);
495 }
496
497 static char *
498 swap_total(void)
499 {
500 long total;
501 FILE *fp;
502 char buf[2048];
503 size_t bytes_read;
504 char *match;
505
506 fp = fopen("/proc/meminfo", "r");
507 if (fp == NULL) {
508 warn("Failed to open file /proc/meminfo");
509 return smprintf(UNKNOWN_STR);
510 }
511 bytes_read = fread(buf, sizeof(char), sizeof(buf), fp);
512 buf[bytes_read] = '\0';
513 fclose(fp);
514 if (bytes_read == 0 || bytes_read == sizeof(buf)) {
515 warn("Failed to read /proc/meminfo\n");
516 return smprintf(UNKNOWN_STR);
517 }
518
519 match = strstr(buf, "SwapTotal");
520 sscanf(match, "SwapTotal: %ld kB\n", &total);
521 if (total == 0) {
522 return smprintf(UNKNOWN_STR);
523 }
524
525 return smprintf("%f", (float)total / 1024 / 1024);
526 }
527
528 static char *
529 swap_used(void)
530 {
531 long total, free, cached;
532 FILE *fp;
533 char buf[2048];
534 size_t bytes_read;
535 char *match;
536
537 fp = fopen("/proc/meminfo", "r");
538 if (fp == NULL) {
539 warn("Failed to open file /proc/meminfo");
540 return smprintf(UNKNOWN_STR);
541 }
542 bytes_read = fread(buf, sizeof(char), sizeof(buf), fp);
543 buf[bytes_read] = '\0';
544 fclose(fp);
545 if (bytes_read == 0 || bytes_read == sizeof(buf)) {
546 warn("Failed to read /proc/meminfo\n");
547 return smprintf(UNKNOWN_STR);
548 }
549
550 match = strstr(buf, "SwapTotal");
551 sscanf(match, "SwapTotal: %ld kB\n", &total);
552 if (total == 0) {
553 return smprintf(UNKNOWN_STR);
554 }
555
556 match = strstr(buf, "SwapCached");
557 sscanf(match, "SwapCached: %ld kB\n", &cached);
558
559 match = strstr(buf, "SwapFree");
560 sscanf(match, "SwapFree: %ld kB\n", &free);
561
562 return smprintf("%f", (float)(total - free - cached) / 1024 / 1024);
563 }
564
565 static char *
566 temp(const char *file)
567 {
568 int temp;
569 FILE *fp;
570
571 fp = fopen(file, "r");
572 if (fp == NULL) {
573 warn("Failed to open file %s", file);
574 return smprintf(UNKNOWN_STR);
575 }
576 fscanf(fp, "%d", &temp);
577 fclose(fp);
578
579 return smprintf("%d°C", temp / 1000);
580 }
581
582 static char *
583 uptime(void)
584 {
585 struct sysinfo info;
586 int h = 0;
587 int m = 0;
588
589 sysinfo(&info);
590 h = info.uptime / 3600;
591 m = (info.uptime - h * 3600 ) / 60;
592
593 return smprintf("%dh %dm", h, m);
594 }
595
596 static char *
597 username(void)
598 {
599 uid_t uid = geteuid();
600 struct passwd *pw = getpwuid(uid);
601
602 if (pw == NULL) {
603 warn("Failed to get username");
604 return smprintf(UNKNOWN_STR);
605 }
606
607 return smprintf("%s", pw->pw_name);
608 }
609
610 static char *
611 uid(void)
612 {
613 return smprintf("%d", geteuid());
614 }
615
616
617 static char *
618 vol_perc(const char *card)
619 {
620 unsigned int i;
621 int v, afd, devmask;
622 char *vnames[] = SOUND_DEVICE_NAMES;
623
624 afd = open(card, O_RDONLY);
625 if (afd < 0) {
626 warn("Cannot open %s", card);
627 return smprintf(UNKNOWN_STR);
628 }
629
630 ioctl(afd, MIXER_READ(SOUND_MIXER_DEVMASK), &devmask);
631 for (i = 0; i < (sizeof(vnames) / sizeof((vnames[0]))); i++)
632 if (devmask & (1 << i))
633 if (!strcmp("vol", vnames[i]))
634 ioctl(afd, MIXER_READ(i), &v);
635
636 close(afd);
637 if (v == 0)
638 return smprintf("mute");
639 return smprintf("%d%%", v & 0xff);
640 }
641
642 static char *
643 wifi_perc(const char *iface)
644 {
645 int perc;
646 char buf[255];
647 char *datastart;
648 char status[5];
649 FILE *fp;
650
651 ccat(3, "/sys/class/net/", iface, "/operstate");
652 fp = fopen(concat, "r");
653 if (fp == NULL) {
654 warn("Failed to open file %s", concat);
655 return smprintf(UNKNOWN_STR);
656 }
657 fgets(status, 5, fp);
658 fclose(fp);
659 if(strcmp(status, "up\n") != 0) {
660 return smprintf(UNKNOWN_STR);
661 }
662
663 fp = fopen("/proc/net/wireless", "r");
664 if (fp == NULL) {
665 warn("Failed to open file /proc/net/wireless");
666 return smprintf(UNKNOWN_STR);
667 }
668 ccat(2, iface, ":");
669 fgets(buf, sizeof(buf), fp);
670 fgets(buf, sizeof(buf), fp);
671 fgets(buf, sizeof(buf), fp);
672 fclose(fp);
673
674 datastart = strstr(buf, concat);
675 if (datastart != NULL) {
676 datastart = strstr(buf, ":");
677 sscanf(datastart + 1, " %*d %d %*d %*d %*d %*d %*d %*d %*d %*d", &perc);
678 }
679
680 return smprintf("%d%%", perc);
681 }
682
683 static char *
684 wifi_essid(const char *iface)
685 {
686 char id[IW_ESSID_MAX_SIZE+1];
687 int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
688 struct iwreq wreq;
689
690 memset(&wreq, 0, sizeof(struct iwreq));
691 wreq.u.essid.length = IW_ESSID_MAX_SIZE+1;
692 sprintf(wreq.ifr_name, iface);
693 if (sockfd == -1) {
694 warn("Failed to get ESSID for interface %s", iface);
695 return smprintf(UNKNOWN_STR);
696 }
697 wreq.u.essid.pointer = id;
698 if (ioctl(sockfd,SIOCGIWESSID, &wreq) == -1) {
699 warn("Failed to get ESSID for interface %s", iface);
700 return smprintf(UNKNOWN_STR);
701 }
702
703 close(sockfd);
704
705 if (strcmp((char *)wreq.u.essid.pointer, "") == 0)
706 return smprintf(UNKNOWN_STR);
707 else
708 return smprintf("%s", (char *)wreq.u.essid.pointer);
709 }
710
711 static char *
712 kernel_release(void)
713 {
714 struct utsname udata;
715 if (uname(&udata) < 0)
716 return smprintf(UNKNOWN_STR);
717
718 return smprintf("%s", udata.release);
719 }
720
721 static void
722 set_status(const char *str)
723 {
724 XStoreName(dpy, DefaultRootWindow(dpy), str);
725 XSync(dpy, False);
726 }
727
728 static void
729 sighandler(const int signo)
730 {
731 if (signo == SIGTERM || signo == SIGINT) {
732 done = 1;
733 }
734 }
735
736 static void
737 usage(void)
738 {
739 fprintf(stderr, "usage: %s [-dhov]\n", argv0);
740 exit(1);
741 }
742
743 int
744 main(int argc, char *argv[])
745 {
746 unsigned short int i;
747 char status_string[4096];
748 char *res, *element;
749 struct arg argument;
750 struct sigaction act;
751
752 ARGBEGIN {
753 case 'd':
754 dflag = 1;
755 break;
756 case 'o':
757 oflag = 1;
758 break;
759 case 'v':
760 printf("slstatus %s (C) 2016 slstatus engineers\n", VERSION);
761 return 0;
762 default:
763 usage();
764 } ARGEND
765
766 if (dflag && oflag) {
767 usage();
768 }
769 if (dflag && daemon(1, 1) < 0) {
770 err(1, "daemon");
771 }
772
773 memset(&act, 0, sizeof(act));
774 act.sa_handler = sighandler;
775 sigaction(SIGINT, &act, 0);
776 sigaction(SIGTERM, &act, 0);
777
778 if (!oflag) {
779 dpy = XOpenDisplay(NULL);
780 }
781
782 setlocale(LC_ALL, "");
783
784 while (!done) {
785 status_string[0] = '\0';
786
787 for (i = 0; i < sizeof(args) / sizeof(args[0]); ++i) {
788 argument = args[i];
789 if (argument.args == NULL) {
790 res = argument.func();
791 } else {
792 res = argument.func(argument.args);
793 }
794 element = smprintf(argument.fmt, res);
795 if (element == NULL) {
796 element = smprintf(UNKNOWN_STR);
797 warnx("Failed to format output");
798 }
799 strlcat(status_string, element, sizeof(status_string));
800 free(res);
801 free(element);
802 }
803
804 if (!oflag) {
805 set_status(status_string);
806 } else {
807 printf("%s\n", status_string);
808 }
809
810 /*
811 * subtract delay time spend in function
812 * calls from the actual global delay time
813 */
814 sleep(UPDATE_INTERVAL - delay);
815 delay = 0;
816 }
817
818 if (!oflag) {
819 set_status(NULL);
820 XCloseDisplay(dpy);
821 }
822
823 return 0;
824 }