Xinqi Bao's Git
1 /* See LICENSE file for copyright and license details. */
7 #include <linux/wireless.h>
16 #include <sys/ioctl.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>
31 #include "extern/arg.h"
32 #include "extern/strlcat.h"
33 #include "extern/strlcpy.h"
34 #include "extern/concat.h"
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);
79 static unsigned short int delay
;
80 static unsigned short int done
;
81 static unsigned short int dflag
, oflag
;
87 smprintf(const char *fmt
, ...)
94 len
= vsnprintf(NULL
, 0, fmt
, ap
);
103 vsnprintf(ret
, len
, fmt
, ap
);
110 battery_perc(const char *bat
)
115 ccat(3, "/sys/class/power_supply/", bat
, "/capacity");
116 fp
= fopen(concat
, "r");
118 warn("Failed to open file %s", concat
);
119 return smprintf(UNKNOWN_STR
);
121 fscanf(fp
, "%i", &perc
);
124 return smprintf("%d%%", perc
);
128 battery_state(const char *bat
)
133 ccat(3, "/sys/class/power_supply/", bat
, "/status");
134 fp
= fopen(concat
, "r");
136 warn("Failed to open file %s", concat
);
137 return smprintf(UNKNOWN_STR
);
139 fscanf(fp
, "%12s", state
);
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("=");
149 return smprintf("?");
157 long double a
[4], b
[4];
160 fp
= fopen("/proc/stat", "r");
162 warn("Failed to open file /proc/stat");
163 return smprintf(UNKNOWN_STR
);
165 fscanf(fp
, "%*s %Lf %Lf %Lf %Lf", &a
[0], &a
[1], &a
[2], &a
[3]);
168 delay
= (UPDATE_INTERVAL
- (UPDATE_INTERVAL
- 1));
171 fp
= fopen("/proc/stat", "r");
173 warn("Failed to open file /proc/stat");
174 return smprintf(UNKNOWN_STR
);
176 fscanf(fp
, "%*s %Lf %Lf %Lf %Lf", &b
[0], &b
[1], &b
[2], &b
[3]);
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
);
184 datetime(const char *fmt
)
190 if (strftime(str
, sizeof(str
), fmt
, localtime(&t
)) == 0) {
191 return smprintf(UNKNOWN_STR
);
194 return smprintf("%s", str
);
198 disk_free(const char *mnt
)
202 if (statvfs(mnt
, &fs
) < 0) {
203 warn("Failed to get filesystem info");
204 return smprintf(UNKNOWN_STR
);
207 return smprintf("%f", (float)fs
.f_bsize
* (float)fs
.f_bfree
/ 1024 / 1024 / 1024);
211 disk_perc(const char *mnt
)
216 if (statvfs(mnt
, &fs
) < 0) {
217 warn("Failed to get filesystem info");
218 return smprintf(UNKNOWN_STR
);
221 perc
= 100 * (1.0f
- ((float)fs
.f_bfree
/ (float)fs
.f_blocks
));
223 return smprintf("%d%%", perc
);
227 disk_total(const char *mnt
)
231 if (statvfs(mnt
, &fs
) < 0) {
232 warn("Failed to get filesystem info");
233 return smprintf(UNKNOWN_STR
);
236 return smprintf("%f", (float)fs
.f_bsize
* (float)fs
.f_blocks
/ 1024 / 1024 / 1024);
240 disk_used(const char *mnt
)
244 if (statvfs(mnt
, &fs
) < 0) {
245 warn("Failed to get filesystem info");
246 return smprintf(UNKNOWN_STR
);
249 return smprintf("%f", (float)fs
.f_bsize
* ((float)fs
.f_blocks
- (float)fs
.f_bfree
) / 1024 / 1024 / 1024);
258 fp
= fopen("/proc/sys/kernel/random/entropy_avail", "r");
260 warn("Failed to open file /proc/sys/kernel/random/entropy_avail");
261 return smprintf(UNKNOWN_STR
);
263 fscanf(fp
, "%d", &num
);
266 return smprintf("%d", num
);
272 return smprintf("%d", getgid());
278 char buf
[HOST_NAME_MAX
];
280 if (gethostname(buf
, sizeof(buf
)) == -1) {
282 return smprintf(UNKNOWN_STR
);
285 return smprintf("%s", buf
);
289 ip(const char *iface
)
291 struct ifaddrs
*ifaddr
, *ifa
;
293 char host
[NI_MAXHOST
];
295 if (getifaddrs(&ifaddr
) == -1) {
296 warn("Failed to get IP address for interface %s", iface
);
297 return smprintf(UNKNOWN_STR
);
300 for (ifa
= ifaddr
; ifa
!= NULL
; ifa
= ifa
->ifa_next
) {
301 if (ifa
->ifa_addr
== NULL
) {
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
)) {
307 warnx("Failed to get IP address for interface %s", iface
);
308 return smprintf(UNKNOWN_STR
);
310 return smprintf("%s", host
);
316 return smprintf(UNKNOWN_STR
);
324 if (getloadavg(avgs
, 3) < 0) {
325 warnx("Failed to get the load avg");
326 return smprintf(UNKNOWN_STR
);
329 return smprintf("%.2f %.2f %.2f", avgs
[0], avgs
[1], avgs
[2]);
338 fp
= fopen("/proc/meminfo", "r");
340 warn("Failed to open file /proc/meminfo");
341 return smprintf(UNKNOWN_STR
);
343 fscanf(fp
, "MemFree: %ld kB\n", &free
);
346 return smprintf("%f", (float)free
/ 1024 / 1024);
352 long total
, free
, buffers
, cached
;
355 fp
= fopen("/proc/meminfo", "r");
357 warn("Failed to open file /proc/meminfo");
358 return smprintf(UNKNOWN_STR
);
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
);
366 return smprintf("%d%%", 100 * ((total
- free
) - (buffers
+ cached
)) / total
);
375 fp
= fopen("/proc/meminfo", "r");
377 warn("Failed to open file /proc/meminfo");
378 return smprintf(UNKNOWN_STR
);
380 fscanf(fp
, "MemTotal: %ld kB\n", &total
);
383 return smprintf("%f", (float)total
/ 1024 / 1024);
389 long free
, total
, buffers
, cached
;
392 fp
= fopen("/proc/meminfo", "r");
394 warn("Failed to open file /proc/meminfo");
395 return smprintf(UNKNOWN_STR
);
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
);
403 return smprintf("%f", (float)(total
- free
- buffers
- cached
) / 1024 / 1024);
407 run_command(const char *cmd
)
410 char buf
[1024] = "n/a";
412 fp
= popen(cmd
, "r");
414 warn("Failed to get command output for %s", cmd
);
415 return smprintf(UNKNOWN_STR
);
417 fgets(buf
, sizeof(buf
)-1, fp
);
420 buf
[strlen(buf
)] = '\0';
423 return smprintf("%s", buf
);
435 fp
= fopen("/proc/meminfo", "r");
437 warn("Failed to open file /proc/meminfo");
438 return smprintf(UNKNOWN_STR
);
440 bytes_read
= fread(buf
, sizeof(char), sizeof(buf
), fp
);
441 buf
[bytes_read
] = '\0';
443 if (bytes_read
== 0 || bytes_read
== sizeof(buf
)) {
444 warn("Failed to read /proc/meminfo\n");
445 return smprintf(UNKNOWN_STR
);
448 match
= strstr(buf
, "SwapTotal");
449 sscanf(match
, "SwapTotal: %ld kB\n", &total
);
451 return smprintf(UNKNOWN_STR
);
454 match
= strstr(buf
, "SwapFree");
455 sscanf(match
, "SwapFree: %ld kB\n", &free
);
457 return smprintf("%f", (float)free
/ 1024 / 1024);
463 long total
, free
, cached
;
469 fp
= fopen("/proc/meminfo", "r");
471 warn("Failed to open file /proc/meminfo");
472 return smprintf(UNKNOWN_STR
);
474 bytes_read
= fread(buf
, sizeof(char), sizeof(buf
), fp
);
475 buf
[bytes_read
] = '\0';
477 if (bytes_read
== 0 || bytes_read
== sizeof(buf
)) {
478 warn("Failed to read /proc/meminfo\n");
479 return smprintf(UNKNOWN_STR
);
482 match
= strstr(buf
, "SwapTotal");
483 sscanf(match
, "SwapTotal: %ld kB\n", &total
);
485 return smprintf(UNKNOWN_STR
);
488 match
= strstr(buf
, "SwapCached");
489 sscanf(match
, "SwapCached: %ld kB\n", &cached
);
491 match
= strstr(buf
, "SwapFree");
492 sscanf(match
, "SwapFree: %ld kB\n", &free
);
494 return smprintf("%d%%", 100 * (total
- free
- cached
) / total
);
506 fp
= fopen("/proc/meminfo", "r");
508 warn("Failed to open file /proc/meminfo");
509 return smprintf(UNKNOWN_STR
);
511 bytes_read
= fread(buf
, sizeof(char), sizeof(buf
), fp
);
512 buf
[bytes_read
] = '\0';
514 if (bytes_read
== 0 || bytes_read
== sizeof(buf
)) {
515 warn("Failed to read /proc/meminfo\n");
516 return smprintf(UNKNOWN_STR
);
519 match
= strstr(buf
, "SwapTotal");
520 sscanf(match
, "SwapTotal: %ld kB\n", &total
);
522 return smprintf(UNKNOWN_STR
);
525 return smprintf("%f", (float)total
/ 1024 / 1024);
531 long total
, free
, cached
;
537 fp
= fopen("/proc/meminfo", "r");
539 warn("Failed to open file /proc/meminfo");
540 return smprintf(UNKNOWN_STR
);
542 bytes_read
= fread(buf
, sizeof(char), sizeof(buf
), fp
);
543 buf
[bytes_read
] = '\0';
545 if (bytes_read
== 0 || bytes_read
== sizeof(buf
)) {
546 warn("Failed to read /proc/meminfo\n");
547 return smprintf(UNKNOWN_STR
);
550 match
= strstr(buf
, "SwapTotal");
551 sscanf(match
, "SwapTotal: %ld kB\n", &total
);
553 return smprintf(UNKNOWN_STR
);
556 match
= strstr(buf
, "SwapCached");
557 sscanf(match
, "SwapCached: %ld kB\n", &cached
);
559 match
= strstr(buf
, "SwapFree");
560 sscanf(match
, "SwapFree: %ld kB\n", &free
);
562 return smprintf("%f", (float)(total
- free
- cached
) / 1024 / 1024);
566 temp(const char *file
)
571 fp
= fopen(file
, "r");
573 warn("Failed to open file %s", file
);
574 return smprintf(UNKNOWN_STR
);
576 fscanf(fp
, "%d", &temp
);
579 return smprintf("%d°C", temp
/ 1000);
590 h
= info
.uptime
/ 3600;
591 m
= (info
.uptime
- h
* 3600 ) / 60;
593 return smprintf("%dh %dm", h
, m
);
599 uid_t uid
= geteuid();
600 struct passwd
*pw
= getpwuid(uid
);
603 warn("Failed to get username");
604 return smprintf(UNKNOWN_STR
);
607 return smprintf("%s", pw
->pw_name
);
613 return smprintf("%d", geteuid());
618 vol_perc(const char *card
)
622 char *vnames
[] = SOUND_DEVICE_NAMES
;
624 afd
= open(card
, O_RDONLY
);
626 warn("Cannot open %s", card
);
627 return smprintf(UNKNOWN_STR
);
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
);
638 return smprintf("mute");
639 return smprintf("%d%%", v
& 0xff);
643 wifi_perc(const char *iface
)
651 ccat(3, "/sys/class/net/", iface
, "/operstate");
652 fp
= fopen(concat
, "r");
654 warn("Failed to open file %s", concat
);
655 return smprintf(UNKNOWN_STR
);
657 fgets(status
, 5, fp
);
659 if(strcmp(status
, "up\n") != 0) {
660 return smprintf(UNKNOWN_STR
);
663 fp
= fopen("/proc/net/wireless", "r");
665 warn("Failed to open file /proc/net/wireless");
666 return smprintf(UNKNOWN_STR
);
669 fgets(buf
, sizeof(buf
), fp
);
670 fgets(buf
, sizeof(buf
), fp
);
671 fgets(buf
, sizeof(buf
), fp
);
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
);
680 return smprintf("%d%%", perc
);
684 wifi_essid(const char *iface
)
686 char id
[IW_ESSID_MAX_SIZE
+1];
687 int sockfd
= socket(AF_INET
, SOCK_DGRAM
, 0);
690 memset(&wreq
, 0, sizeof(struct iwreq
));
691 wreq
.u
.essid
.length
= IW_ESSID_MAX_SIZE
+1;
692 sprintf(wreq
.ifr_name
, iface
);
694 warn("Failed to get ESSID for interface %s", iface
);
695 return smprintf(UNKNOWN_STR
);
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
);
705 if (strcmp((char *)wreq
.u
.essid
.pointer
, "") == 0)
706 return smprintf(UNKNOWN_STR
);
708 return smprintf("%s", (char *)wreq
.u
.essid
.pointer
);
714 struct utsname udata
;
715 if (uname(&udata
) < 0)
716 return smprintf(UNKNOWN_STR
);
718 return smprintf("%s", udata
.release
);
722 set_status(const char *str
)
724 XStoreName(dpy
, DefaultRootWindow(dpy
), str
);
729 sighandler(const int signo
)
731 if (signo
== SIGTERM
|| signo
== SIGINT
) {
739 fprintf(stderr
, "usage: %s [-dhov]\n", argv0
);
744 main(int argc
, char *argv
[])
746 unsigned short int i
;
747 char status_string
[4096];
750 struct sigaction act
;
760 printf("slstatus %s (C) 2016 slstatus engineers\n", VERSION
);
766 if (dflag
&& oflag
) {
769 if (dflag
&& daemon(1, 1) < 0) {
773 memset(&act
, 0, sizeof(act
));
774 act
.sa_handler
= sighandler
;
775 sigaction(SIGINT
, &act
, 0);
776 sigaction(SIGTERM
, &act
, 0);
779 dpy
= XOpenDisplay(NULL
);
782 setlocale(LC_ALL
, "");
785 status_string
[0] = '\0';
787 for (i
= 0; i
< sizeof(args
) / sizeof(args
[0]); ++i
) {
789 if (argument
.args
== NULL
) {
790 res
= argument
.func();
792 res
= argument
.func(argument
.args
);
794 element
= smprintf(argument
.fmt
, res
);
795 if (element
== NULL
) {
796 element
= smprintf(UNKNOWN_STR
);
797 warnx("Failed to format output");
799 strlcat(status_string
, element
, sizeof(status_string
));
805 set_status(status_string
);
807 printf("%s\n", status_string
);
811 * subtract delay time spend in function
812 * calls from the actual global delay time
814 sleep(UPDATE_INTERVAL
- delay
);