Xinqi Bao's Git
c4a9e92186d4abd4edb333dc1e5a468ae8c4fb5c
1 /* See LICENSE file for copyright and license details. */
8 #include <linux/wireless.h>
17 #include <sys/ioctl.h>
19 #include <sys/statvfs.h>
20 #include <sys/socket.h>
21 #include <sys/soundcard.h>
22 #include <sys/sysinfo.h>
23 #include <sys/types.h>
24 #include <sys/utsname.h>
32 const char *(*func
)();
37 static const char *bprintf(const char *fmt
, ...);
38 static const char *battery_perc(const char *bat
);
39 static const char *battery_power(const char *bat
);
40 static const char *battery_state(const char *bat
);
41 static const char *cpu_freq(void);
42 static const char *cpu_perc(void);
43 static const char *datetime(const char *fmt
);
44 static const char *disk_free(const char *mnt
);
45 static const char *disk_perc(const char *mnt
);
46 static const char *disk_total(const char *mnt
);
47 static const char *disk_used(const char *mnt
);
48 static const char *entropy(void);
49 static const char *gid(void);
50 static const char *hostname(void);
51 static const char *ip(const char *iface
);
52 static const char *kernel_release(void);
53 static const char *keyboard_indicators(void);
54 static const char *load_avg(void);
55 static const char *num_files(const char *dir
);
56 static const char *ram_free(void);
57 static const char *ram_perc(void);
58 static const char *ram_used(void);
59 static const char *ram_total(void);
60 static const char *run_command(const char *cmd
);
61 static const char *swap_free(void);
62 static const char *swap_perc(void);
63 static const char *swap_used(void);
64 static const char *swap_total(void);
65 static const char *temp(const char *file
);
66 static const char *uid(void);
67 static const char *uptime(void);
68 static const char *username(void);
69 static const char *vol_perc(const char *card
);
70 static const char *wifi_perc(const char *iface
);
71 static const char *wifi_essid(const char *iface
);
72 static void sighandler(const int signo
);
73 static void usage(void);
76 static unsigned short int delay
= 0;
77 static unsigned short int done
;
82 static char buf
[MAXLEN
];
85 bprintf(const char *fmt
, ...)
91 len
= vsnprintf(buf
, sizeof(buf
) - 1, fmt
, ap
);
94 if (len
>= sizeof(buf
))
95 buf
[sizeof(buf
)-1] = '\0';
101 battery_perc(const char *bat
)
107 snprintf(path
, sizeof(path
), "%s%s%s", "/sys/class/power_supply/", bat
, "/capacity");
108 fp
= fopen(path
, "r");
110 warn("Failed to open file %s", path
);
113 n
= fscanf(fp
, "%i", &perc
);
118 return bprintf("%d", perc
);
122 battery_power(const char *bat
)
128 snprintf(path
, sizeof(path
), "%s%s%s", "/sys/class/power_supply/", bat
, "/power_now");
129 fp
= fopen(path
, "r");
131 warn("Failed to open file %s", path
);
134 n
= fscanf(fp
, "%i", &watts
);
139 return bprintf("%d", (watts
+ 500000) / 1000000);
143 battery_state(const char *bat
)
151 { "Discharging", "-" },
157 char path
[PATH_MAX
], state
[12];
159 snprintf(path
, sizeof(path
), "%s%s%s", "/sys/class/power_supply/", bat
, "/status");
160 fp
= fopen(path
, "r");
162 warn("Failed to open file %s", path
);
165 n
= fscanf(fp
, "%12s", state
);
170 for (i
= 0; i
< sizeof(map
) / sizeof(*map
); i
++) {
171 if (!strcmp(map
[i
].state
, state
)) {
176 return (i
== sizeof(map
) / sizeof(*map
)) ? "?" : map
[i
].symbol
;
185 fp
= fopen("/sys/devices/system/cpu/cpu0/cpufreq/scaling_cur_freq", "r");
187 warn("Failed to open file /sys/devices/system/cpu/cpu0/cpufreq/scaling_cur_freq");
190 n
= fscanf(fp
, "%i", &freq
);
195 return bprintf("%d", (freq
+ 500) / 1000);
202 long double a
[4], b
[4];
205 fp
= fopen("/proc/stat", "r");
207 warn("Failed to open file /proc/stat");
210 n
= fscanf(fp
, "%*s %Lf %Lf %Lf %Lf", &a
[0], &a
[1], &a
[2], &a
[3]);
218 fp
= fopen("/proc/stat", "r");
220 warn("Failed to open file /proc/stat");
223 n
= fscanf(fp
, "%*s %Lf %Lf %Lf %Lf", &b
[0], &b
[1], &b
[2], &b
[3]);
228 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]));
229 return bprintf("%d", perc
);
233 datetime(const char *fmt
)
238 if (strftime(buf
, sizeof(buf
), fmt
, localtime(&t
)) == 0)
245 disk_free(const char *mnt
)
249 if (statvfs(mnt
, &fs
) < 0) {
250 warn("Failed to get filesystem info");
254 return bprintf("%f", (float)fs
.f_bsize
* (float)fs
.f_bfree
/ 1024 / 1024 / 1024);
258 disk_perc(const char *mnt
)
263 if (statvfs(mnt
, &fs
) < 0) {
264 warn("Failed to get filesystem info");
268 perc
= 100 * (1.0f
- ((float)fs
.f_bfree
/ (float)fs
.f_blocks
));
270 return bprintf("%d", perc
);
274 disk_total(const char *mnt
)
278 if (statvfs(mnt
, &fs
) < 0) {
279 warn("Failed to get filesystem info");
283 return bprintf("%f", (float)fs
.f_bsize
* (float)fs
.f_blocks
/ 1024 / 1024 / 1024);
287 disk_used(const char *mnt
)
291 if (statvfs(mnt
, &fs
) < 0) {
292 warn("Failed to get filesystem info");
296 return bprintf("%f", (float)fs
.f_bsize
* ((float)fs
.f_blocks
- (float)fs
.f_bfree
) / 1024 / 1024 / 1024);
305 fp
= fopen("/proc/sys/kernel/random/entropy_avail", "r");
307 warn("Failed to open file /proc/sys/kernel/random/entropy_avail");
310 n
= fscanf(fp
, "%d", &num
);
315 return bprintf("%d", num
);
321 return bprintf("%d", getgid());
327 if (gethostname(buf
, sizeof(buf
)) == -1) {
336 ip(const char *iface
)
338 struct ifaddrs
*ifaddr
, *ifa
;
340 char host
[NI_MAXHOST
];
342 if (getifaddrs(&ifaddr
) == -1) {
343 warn("Failed to get IP address for interface %s", iface
);
347 for (ifa
= ifaddr
; ifa
!= NULL
; ifa
= ifa
->ifa_next
) {
348 if (ifa
->ifa_addr
== NULL
) {
351 s
= getnameinfo(ifa
->ifa_addr
, sizeof(struct sockaddr_in
), host
, NI_MAXHOST
, NULL
, 0, NI_NUMERICHOST
);
352 if ((strcmp(ifa
->ifa_name
, iface
) == 0) && (ifa
->ifa_addr
->sa_family
== AF_INET
)) {
354 warnx("Failed to get IP address for interface %s", iface
);
357 return bprintf("%s", host
);
369 struct utsname udata
;
371 if (uname(&udata
) < 0) {
375 return bprintf("%s", udata
.release
);
379 keyboard_indicators(void)
381 Display
*dpy
= XOpenDisplay(NULL
);
382 XKeyboardState state
;
383 XGetKeyboardControl(dpy
, &state
);
386 switch (state
.led_mask
) {
403 if (getloadavg(avgs
, 3) < 0) {
404 warnx("Failed to get the load avg");
408 return bprintf("%.2f %.2f %.2f", avgs
[0], avgs
[1], avgs
[2]);
412 num_files(const char *dir
)
418 if ((fd
= opendir(dir
)) == NULL
) {
419 warn("Failed to get number of files in directory %s", dir
);
423 while ((dp
= readdir(fd
)) != NULL
) {
424 if (!strcmp(dp
->d_name
, ".") || !strcmp(dp
->d_name
, ".."))
425 continue; /* skip self and parent */
431 return bprintf("%d", num
);
441 fp
= fopen("/proc/meminfo", "r");
443 warn("Failed to open file /proc/meminfo");
446 n
= fscanf(fp
, "MemFree: %ld kB\n", &free
);
451 return bprintf("%f", (float)free
/ 1024 / 1024);
457 long total
, free
, buffers
, cached
;
460 fp
= fopen("/proc/meminfo", "r");
462 warn("Failed to open file /proc/meminfo");
465 if (fscanf(fp
, "MemTotal: %ld kB\n", &total
) != 1 ||
466 fscanf(fp
, "MemFree: %ld kB\n", &free
) != 1 ||
467 fscanf(fp
, "MemAvailable: %ld kB\nBuffers: %ld kB\n",
468 &buffers
, &buffers
) != 2 ||
469 fscanf(fp
, "Cached: %ld kB\n", &cached
) != 1)
473 return bprintf("%d", 100 * ((total
- free
) - (buffers
+ cached
)) / total
);
487 fp
= fopen("/proc/meminfo", "r");
489 warn("Failed to open file /proc/meminfo");
492 n
= fscanf(fp
, "MemTotal: %ld kB\n", &total
);
497 return bprintf("%f", (float)total
/ 1024 / 1024);
503 long free
, total
, buffers
, cached
;
506 fp
= fopen("/proc/meminfo", "r");
508 warn("Failed to open file /proc/meminfo");
511 if (fscanf(fp
, "MemTotal: %ld kB\n", &total
) != 1 ||
512 fscanf(fp
, "MemFree: %ld kB\n", &free
) != 1 ||
513 fscanf(fp
, "MemAvailable: %ld kB\nBuffers: %ld kB\n",
514 &buffers
, &buffers
) != 2 ||
515 fscanf(fp
, "Cached: %ld kB\n", &cached
) != 1)
519 return bprintf("%f", (float)(total
- free
- buffers
- cached
) / 1024 / 1024);
527 run_command(const char *cmd
)
532 fp
= popen(cmd
, "r");
534 warn("Failed to get command output for %s", cmd
);
537 p
= fgets(buf
, sizeof(buf
) - 1, fp
);
541 if ((p
= strrchr(buf
, '\n')) != NULL
)
544 return buf
[0] ? buf
: UNKNOWN_STR
;
555 fp
= fopen("/proc/meminfo", "r");
557 warn("Failed to open file /proc/meminfo");
561 if ((bytes_read
= fread(buf
, sizeof(char), sizeof(buf
) - 1, fp
)) == 0) {
562 warn("swap_free: read error");
568 if ((match
= strstr(buf
, "SwapTotal")) == NULL
)
570 sscanf(match
, "SwapTotal: %ld kB\n", &total
);
572 if ((match
= strstr(buf
, "SwapFree")) == NULL
)
574 sscanf(match
, "SwapFree: %ld kB\n", &free
);
576 return bprintf("%f", (float)free
/ 1024 / 1024);
582 long total
, free
, cached
;
587 fp
= fopen("/proc/meminfo", "r");
589 warn("Failed to open file /proc/meminfo");
593 if ((bytes_read
= fread(buf
, sizeof(char), sizeof(buf
) - 1, fp
)) == 0) {
594 warn("swap_perc: read error");
600 if ((match
= strstr(buf
, "SwapTotal")) == NULL
)
602 sscanf(match
, "SwapTotal: %ld kB\n", &total
);
604 if ((match
= strstr(buf
, "SwapCached")) == NULL
)
606 sscanf(match
, "SwapCached: %ld kB\n", &cached
);
608 if ((match
= strstr(buf
, "SwapFree")) == NULL
)
610 sscanf(match
, "SwapFree: %ld kB\n", &free
);
612 return bprintf("%d", 100 * (total
- free
- cached
) / total
);
623 fp
= fopen("/proc/meminfo", "r");
625 warn("Failed to open file /proc/meminfo");
628 if ((bytes_read
= fread(buf
, sizeof(char), sizeof(buf
) - 1, fp
)) == 0) {
629 warn("swap_total: read error");
635 if ((match
= strstr(buf
, "SwapTotal")) == NULL
)
637 sscanf(match
, "SwapTotal: %ld kB\n", &total
);
639 return bprintf("%f", (float)total
/ 1024 / 1024);
645 long total
, free
, cached
;
650 fp
= fopen("/proc/meminfo", "r");
652 warn("Failed to open file /proc/meminfo");
655 if ((bytes_read
= fread(buf
, sizeof(char), sizeof(buf
) - 1, fp
)) == 0) {
656 warn("swap_used: read error");
662 if ((match
= strstr(buf
, "SwapTotal")) == NULL
)
664 sscanf(match
, "SwapTotal: %ld kB\n", &total
);
666 if ((match
= strstr(buf
, "SwapCached")) == NULL
)
668 sscanf(match
, "SwapCached: %ld kB\n", &cached
);
670 if ((match
= strstr(buf
, "SwapFree")) == NULL
)
672 sscanf(match
, "SwapFree: %ld kB\n", &free
);
674 return bprintf("%f", (float)(total
- free
- cached
) / 1024 / 1024);
678 temp(const char *file
)
683 fp
= fopen(file
, "r");
685 warn("Failed to open file %s", file
);
688 n
= fscanf(fp
, "%d", &temp
);
693 return bprintf("%d", temp
/ 1000);
704 h
= info
.uptime
/ 3600;
705 m
= (info
.uptime
- h
* 3600 ) / 60;
707 return bprintf("%dh %dm", h
, m
);
713 struct passwd
*pw
= getpwuid(geteuid());
716 warn("Failed to get username");
720 return bprintf("%s", pw
->pw_name
);
726 return bprintf("%d", geteuid());
731 vol_perc(const char *card
)
735 char *vnames
[] = SOUND_DEVICE_NAMES
;
737 afd
= open(card
, O_RDONLY
| O_NONBLOCK
);
739 warn("Cannot open %s", card
);
743 if (ioctl(afd
, SOUND_MIXER_READ_DEVMASK
, &devmask
) == -1) {
744 warn("Cannot get volume for %s", card
);
748 for (i
= 0; i
< (sizeof(vnames
) / sizeof((vnames
[0]))); i
++) {
749 if (devmask
& (1 << i
) && !strcmp("vol", vnames
[i
])) {
750 if (ioctl(afd
, MIXER_READ(i
), &v
) == -1) {
751 warn("vol_perc: ioctl");
760 return bprintf("%d", v
& 0xff);
764 wifi_perc(const char *iface
)
772 snprintf(path
, sizeof(path
), "%s%s%s", "/sys/class/net/", iface
, "/operstate");
773 fp
= fopen(path
, "r");
775 warn("Failed to open file %s", path
);
778 p
= fgets(status
, 5, fp
);
780 if(!p
|| strcmp(status
, "up\n") != 0) {
784 fp
= fopen("/proc/net/wireless", "r");
786 warn("Failed to open file /proc/net/wireless");
790 for (i
= 0; i
< 3; i
++) {
791 if (!(p
= fgets(buf
, sizeof(buf
) - 1, fp
)))
798 if ((datastart
= strstr(buf
, iface
)) == NULL
)
801 datastart
= (datastart
+(strlen(iface
)+1));
802 sscanf(datastart
+ 1, " %*d %d %*d %*d %*d %*d %*d %*d %*d %*d", &perc
);
804 return bprintf("%d", perc
);
808 wifi_essid(const char *iface
)
810 static char id
[IW_ESSID_MAX_SIZE
+1];
811 int sockfd
= socket(AF_INET
, SOCK_DGRAM
, 0);
814 memset(&wreq
, 0, sizeof(struct iwreq
));
815 wreq
.u
.essid
.length
= IW_ESSID_MAX_SIZE
+1;
816 snprintf(wreq
.ifr_name
, sizeof(wreq
.ifr_name
), "%s", iface
);
819 warn("Failed to get ESSID for interface %s", iface
);
822 wreq
.u
.essid
.pointer
= id
;
823 if (ioctl(sockfd
,SIOCGIWESSID
, &wreq
) == -1) {
824 warn("Failed to get ESSID for interface %s", iface
);
830 if (strcmp(id
, "") == 0)
837 sighandler(const int signo
)
839 if (signo
== SIGTERM
|| signo
== SIGINT
) {
847 fprintf(stderr
, "usage: %s [-s]\n", argv0
);
852 main(int argc
, char *argv
[])
855 struct sigaction act
;
858 char status_string
[MAXLEN
];
873 memset(&act
, 0, sizeof(act
));
874 act
.sa_handler
= sighandler
;
875 sigaction(SIGINT
, &act
, 0);
876 sigaction(SIGTERM
, &act
, 0);
879 dpy
= XOpenDisplay(NULL
);
882 setlocale(LC_ALL
, "");
885 status_string
[0] = '\0';
887 for (element
= status_string
, i
= len
= 0;
888 i
< sizeof(args
) / sizeof(args
[0]);
889 ++i
, element
+= len
) {
891 len
= snprintf(element
, sizeof(status_string
)-1 - len
,
893 argument
.func(argument
.args
));
894 if (len
>= sizeof(status_string
)) {
895 status_string
[sizeof(status_string
)-1] = '\0';
901 printf("%s\n", status_string
);
903 XStoreName(dpy
, DefaultRootWindow(dpy
), status_string
);
907 if ((UPDATE_INTERVAL
- delay
) <= 0) {
911 sleep(UPDATE_INTERVAL
- delay
);
917 XStoreName(dpy
, DefaultRootWindow(dpy
), NULL
);