Xinqi Bao's Git
1 /* See LICENSE file for copyright and license details. */
4 #include <alsa/asoundlib.h>
9 #include <linux/wireless.h>
17 #include <sys/ioctl.h>
19 #include <sys/statvfs.h>
20 #include <sys/socket.h>
21 #include <sys/sysinfo.h>
22 #include <sys/types.h>
33 /* statusbar configuration type and struct */
34 typedef char *(*op_fun
) (const char *);
41 /* function declarations */
42 void setstatus(const char *str
);
43 char *smprintf(const char *fmt
, ...);
44 char *battery_perc(const char *battery
);
45 char *cpu_perc(const char *null
);
46 char *datetime(const char *timeformat
);
47 char *disk_free(const char *mountpoint
);
48 char *disk_perc(const char *mountpoint
);
49 char *disk_total(const char *mountpoint
);
50 char *disk_used(const char *mountpoint
);
51 char *entropy(const char *null
);
52 char *gid(const char *null
);
53 char *hostname(const char *null
);
54 char *ip(const char *interface
);
55 char *load_avg(const char *null
);
56 char *ram_free(const char *null
);
57 char *ram_perc(const char *null
);
58 char *ram_used(const char *null
);
59 char *ram_total(const char *null
);
60 char *run_command(const char *command
);
61 char *temp(const char *file
);
62 char *uid(const char *null
);
63 char *uptime(const char *null
);
64 char *username(const char *null
);
65 char *vol_perc(const char *soundcard
);
66 char *wifi_perc(const char *wificard
);
67 char *wifi_essid(const char *wificard
);
69 /* global variables */
72 /* configuration header */
77 setstatus(const char *str
)
79 /* set WM_NAME via X11 */
80 XStoreName(dpy
, DefaultRootWindow(dpy
), str
);
84 /* smprintf function */
86 smprintf(const char *fmt
, ...)
91 va_start(fmtargs
, fmt
);
92 if (vasprintf(&ret
, fmt
, fmtargs
) < 0) {
100 /* battery percentage */
102 battery_perc(const char *battery
)
105 char batterynowfile
[64] = "";
106 char batteryfullfile
[64] = "";
109 /* generate battery nowfile path */
110 strlcat(batterynowfile
, batterypath
, sizeof(batterynowfile
));
111 strlcat(batterynowfile
, battery
, sizeof(batterynowfile
));
112 strlcat(batterynowfile
, "/", sizeof(batterynowfile
));
113 strlcat(batterynowfile
, batterynow
, sizeof(batterynowfile
));
115 /* generate battery fullfile path */
116 strlcat(batteryfullfile
, batterypath
, sizeof(batteryfullfile
));
117 strlcat(batteryfullfile
, battery
, sizeof(batteryfullfile
));
118 strlcat(batteryfullfile
, "/", sizeof(batteryfullfile
));
119 strlcat(batteryfullfile
, batteryfull
, sizeof(batteryfullfile
));
121 /* open battery now file */
122 if (!(fp
= fopen(batterynowfile
, "r"))) {
123 fprintf(stderr
, "Error opening battery file: %s.\n", batterynowfile
);
124 return smprintf(unknowntext
);
128 fscanf(fp
, "%i", &now
);
130 /* close battery now file */
133 /* open battery full file */
134 if (!(fp
= fopen(batteryfullfile
, "r"))) {
135 fprintf(stderr
, "Error opening battery file.\n");
136 return smprintf(unknowntext
);
140 fscanf(fp
, "%i", &full
);
142 /* close battery full file */
145 /* calculate percent */
146 perc
= now
/ (full
/ 100);
148 /* return perc as string */
149 return smprintf("%d%%", perc
);
154 cpu_perc(const char *null
)
157 long double a
[4], b
[4];
161 if (!(fp
= fopen("/proc/stat","r"))) {
162 fprintf(stderr
, "Error opening stat file.\n");
163 return smprintf(unknowntext
);
167 fscanf(fp
, "%*s %Lf %Lf %Lf %Lf", &a
[0], &a
[1], &a
[2], &a
[3]);
169 /* close stat file */
172 /* wait a second (for avg values) */
176 if (!(fp
= fopen("/proc/stat","r"))) {
177 fprintf(stderr
, "Error opening stat file.\n");
178 return smprintf(unknowntext
);
182 fscanf(fp
, "%*s %Lf %Lf %Lf %Lf", &b
[0], &b
[1], &b
[2], &b
[3]);
184 /* close stat file */
187 /* calculate avg in this second */
188 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]));
190 /* return perc as string */
191 return smprintf("%d%%", perc
);
196 datetime(const char *timeformat
)
200 char *buf
= malloc(bufsize
);
202 fprintf(stderr
, "Failed to get date/time.\n");
203 return smprintf(unknowntext
);
206 /* get time in format */
208 setlocale(LC_TIME
, "");
209 if (!strftime(buf
, bufsize
, timeformat
, localtime(&tm
))) {
210 setlocale(LC_TIME
, "C");
212 fprintf(stderr
, "Strftime failed.\n");
213 return smprintf(unknowntext
);
216 setlocale(LC_TIME
, "C");
218 char *ret
= smprintf("%s", buf
);
225 disk_free(const char *mountpoint
)
229 /* try to open mountpoint */
230 if (statvfs(mountpoint
, &fs
) < 0) {
231 fprintf(stderr
, "Could not get filesystem info.\n");
232 return smprintf(unknowntext
);
236 return smprintf("%f", (float)fs
.f_bsize
* (float)fs
.f_bfree
/ 1024 / 1024 / 1024);
239 /* disk usage percentage */
241 disk_perc(const char *mountpoint
)
246 /* try to open mountpoint */
247 if (statvfs(mountpoint
, &fs
) < 0) {
248 fprintf(stderr
, "Could not get filesystem info.\n");
249 return smprintf(unknowntext
);
252 /* calculate percent */
253 perc
= 100 * (1.0f
- ((float)fs
.f_bfree
/ (float)fs
.f_blocks
));
256 return smprintf("%d%%", perc
);
261 disk_total(const char *mountpoint
)
265 /* try to open mountpoint */
266 if (statvfs(mountpoint
, &fs
) < 0) {
267 fprintf(stderr
, "Could not get filesystem info.\n");
268 return smprintf(unknowntext
);
272 return smprintf("%f", (float)fs
.f_bsize
* (float)fs
.f_blocks
/ 1024 / 1024 / 1024);
277 disk_used(const char *mountpoint
)
281 /* try to open mountpoint */
282 if (statvfs(mountpoint
, &fs
) < 0) {
283 fprintf(stderr
, "Could not get filesystem info.\n");
284 return smprintf(unknowntext
);
288 return smprintf("%f", (float)fs
.f_bsize
* ((float)fs
.f_blocks
- (float)fs
.f_bfree
) / 1024 / 1024 / 1024);
291 /* entropy available */
293 entropy(const char *null
)
298 /* open entropy file */
299 if (!(fp
= fopen("/proc/sys/kernel/random/entropy_avail", "r"))) {
300 fprintf(stderr
, "Could not open entropy file.\n");
301 return smprintf(unknowntext
);
304 /* extract entropy */
305 fscanf(fp
, "%d", &entropy
);
307 /* close entropy file */
311 return smprintf("%d", entropy
);
316 gid(const char *null
)
320 if ((gid
= getgid()) < 0) {
321 fprintf(stderr
, "Could no get gid.\n");
322 return smprintf(unknowntext
);
324 return smprintf("%d", gid
);
327 return smprintf(unknowntext
);
332 hostname(const char *null
)
334 char hostname
[HOST_NAME_MAX
];
337 /* open hostname file */
338 if (!(fp
= fopen("/proc/sys/kernel/hostname", "r"))) {
339 fprintf(stderr
, "Could not open hostname file.\n");
340 return smprintf(unknowntext
);
343 /* extract hostname */
344 fscanf(fp
, "%s\n", hostname
);
346 /* close hostname file */
350 return smprintf("%s", hostname
);
355 ip(const char *interface
)
357 struct ifaddrs
*ifaddr
, *ifa
;
359 char host
[NI_MAXHOST
];
361 /* check if getting ip address works */
362 if (getifaddrs(&ifaddr
) == -1) {
363 fprintf(stderr
, "Error getting IP address.\n");
364 return smprintf(unknowntext
);
367 /* get the ip address */
368 for (ifa
= ifaddr
; ifa
!= NULL
; ifa
= ifa
->ifa_next
) {
369 if (ifa
->ifa_addr
== NULL
) {
373 s
= getnameinfo(ifa
->ifa_addr
, sizeof(struct sockaddr_in
), host
, NI_MAXHOST
, NULL
, 0, NI_NUMERICHOST
);
375 if ((strcmp(ifa
->ifa_name
, interface
) == 0) && (ifa
->ifa_addr
->sa_family
== AF_INET
)) {
377 fprintf(stderr
, "Error getting IP address.\n");
378 return smprintf(unknowntext
);
380 return smprintf("%s", host
);
384 /* free the address */
387 return smprintf(unknowntext
);
392 load_avg(const char *null
)
396 /* try to get load avg */
397 if (getloadavg(avgs
, 3) < 0) {
398 fprintf(stderr
, "Error getting load avg.\n");
399 return smprintf(unknowntext
);
403 return smprintf("%.2f %.2f %.2f", avgs
[0], avgs
[1], avgs
[2]);
408 ram_free(const char *null
)
413 /* open meminfo file */
414 if (!(fp
= fopen("/proc/meminfo", "r"))) {
415 fprintf(stderr
, "Error opening meminfo file.\n");
416 return smprintf(unknowntext
);
419 /* read the values */
420 fscanf(fp
, "MemFree: %ld kB\n", &free
);
422 /* close meminfo file */
425 /* return free ram as string */
426 return smprintf("%f", (float)free
/ 1024 / 1024);
431 ram_perc(const char *null
)
434 long total
, free
, buffers
, cached
;
437 /* open meminfo file */
438 if (!(fp
= fopen("/proc/meminfo", "r"))) {
439 fprintf(stderr
, "Error opening meminfo file.\n");
440 return smprintf(unknowntext
);
443 /* read the values */
444 fscanf(fp
, "MemTotal: %ld kB\n", &total
);
445 fscanf(fp
, "MemFree: %ld kB\n", &free
);
446 fscanf(fp
, "MemAvailable: %ld kB\nBuffers: %ld kB\n", &buffers
, &buffers
);
447 fscanf(fp
, "Cached: %ld kB\n", &cached
);
449 /* close meminfo file */
452 /* calculate percentage */
453 perc
= 100 * ((total
- free
) - (buffers
+ cached
)) / total
;
455 /* return perc as string */
456 return smprintf("%d%%", perc
);
461 ram_total(const char *null
)
466 /* open meminfo file */
467 if (!(fp
= fopen("/proc/meminfo", "r"))) {
468 fprintf(stderr
, "Error opening meminfo file.\n");
469 return smprintf(unknowntext
);
472 /* read the values */
473 fscanf(fp
, "MemTotal: %ld kB\n", &total
);
475 /* close meminfo file */
478 /* return total ram as string */
479 return smprintf("%f", (float)total
/ 1024 / 1024);
484 ram_used(const char *null
)
486 long free
, total
, buffers
, cached
, used
;
489 /* open meminfo file */
490 if (!(fp
= fopen("/proc/meminfo", "r"))) {
491 fprintf(stderr
, "Error opening meminfo file.\n");
492 return smprintf(unknowntext
);
495 /* read the values */
496 fscanf(fp
, "MemTotal: %ld kB\n", &total
);
497 fscanf(fp
, "MemFree: %ld kB\n", &free
);
498 fscanf(fp
, "MemAvailable: %ld kB\nBuffers: %ld kB\n", &buffers
, &buffers
);
499 fscanf(fp
, "Cached: %ld kB\n", &cached
);
501 /* close meminfo file */
505 used
= total
- free
- buffers
- cached
;
507 /* return used ram as string */
508 return smprintf("%f", (float)used
/ 1024 / 1024);
511 /* custom shell command */
513 run_command(const char* command
)
519 /* try to open the command output */
520 if (!(fp
= popen(command
, "r"))) {
521 fprintf(stderr
, "Could not get command output for: %s.\n", command
);
522 return smprintf(unknowntext
);
525 /* get command output text, save it to buffer */
526 fgets(buffer
, sizeof(buffer
) - 1, fp
);
531 /* add nullchar at the end */
532 for (int i
= 0 ; i
!= sizeof(buffer
); i
++) {
533 if (buffer
[i
] == '\0') {
539 buffer
[strlen(buffer
) - 1] = '\0';
542 /* return the output */
543 return smprintf("%s", buffer
);
548 temp(const char *file
)
553 /* open temperature file */
554 if (!(fp
= fopen(file
, "r"))) {
555 fprintf(stderr
, "Could not open temperature file.\n");
556 return smprintf(unknowntext
);
559 /* extract temperature */
560 fscanf(fp
, "%d", &temperature
);
562 /* close temperature file */
565 /* return temperature in degrees */
566 return smprintf("%d°C", temperature
/ 1000);
571 uptime(const char *null
)
579 hours
= info
.uptime
/ 3600;
580 minutes
= (info
.uptime
- hours
* 3600 ) / 60;
583 return smprintf("%dh %dm", hours
, minutes
);
588 username(const char *null
)
590 register struct passwd
*pw
;
597 /* if it worked, return */
599 return smprintf("%s", pw
->pw_name
);
601 fprintf(stderr
, "Could not get username.\n");
602 return smprintf(unknowntext
);
605 return smprintf(unknowntext
);
610 uid(const char *null
)
617 /* if it worked, return */
619 return smprintf("%d", uid
);
621 fprintf(stderr
, "Could not get uid.\n");
622 return smprintf(unknowntext
);
625 return smprintf(unknowntext
);
629 /* alsa volume percentage */
631 vol_perc(const char *soundcard
)
634 long vol
= 0, max
= 0, min
= 0;
636 snd_mixer_elem_t
*pcm_mixer
, *mas_mixer
;
637 snd_mixer_selem_id_t
*vol_info
, *mute_info
;
639 /* open everything */
640 snd_mixer_open(&handle
, 0);
641 snd_mixer_attach(handle
, soundcard
);
642 snd_mixer_selem_register(handle
, NULL
, NULL
);
643 snd_mixer_load(handle
);
645 /* prepare everything */
646 snd_mixer_selem_id_malloc(&vol_info
);
647 snd_mixer_selem_id_malloc(&mute_info
);
649 if (vol_info
== NULL
|| mute_info
== NULL
) {
650 fprintf(stderr
, "Could not get alsa volume.\n");
651 return smprintf(unknowntext
);
653 snd_mixer_selem_id_set_name(vol_info
, channel
);
654 snd_mixer_selem_id_set_name(mute_info
, channel
);
655 pcm_mixer
= snd_mixer_find_selem(handle
, vol_info
);
656 mas_mixer
= snd_mixer_find_selem(handle
, mute_info
);
659 snd_mixer_selem_get_playback_volume_range((snd_mixer_elem_t
*)pcm_mixer
, &min
, &max
);
660 snd_mixer_selem_get_playback_volume((snd_mixer_elem_t
*)pcm_mixer
, SND_MIXER_SCHN_MONO
, &vol
);
661 snd_mixer_selem_get_playback_switch(mas_mixer
, SND_MIXER_SCHN_MONO
, &mute
);
665 snd_mixer_selem_id_free(vol_info
);
668 snd_mixer_selem_id_free(mute_info
);
671 snd_mixer_close(handle
);
674 /* return the string (mute) */
676 return smprintf("mute");
678 return smprintf("%d%%", (vol
* 100) / max
);
682 /* wifi percentage */
684 wifi_perc(const char *wificard
)
692 char needle
[sizeof wificard
+ 1];
695 /* generate the path name */
696 memset(path
, 0, sizeof path
);
697 strlcat(path
, "/sys/class/net/", sizeof(path
));
698 strlcat(path
, wificard
, sizeof(path
));
699 strlcat(path
, "/operstate", sizeof(path
));
702 if(!(fp
= fopen(path
, "r"))) {
703 fprintf(stderr
, "Error opening wifi operstate file.\n");
704 return smprintf(unknowntext
);
707 /* read the status */
708 fgets(status
, 5, fp
);
710 /* close wifi file */
713 /* check if interface down */
714 if(strcmp(status
, "up\n") != 0) {
715 return smprintf(unknowntext
);
719 if (!(fp
= fopen("/proc/net/wireless", "r"))) {
720 fprintf(stderr
, "Error opening wireless file.\n");
721 return smprintf(unknowntext
);
724 /* extract the signal strength */
725 strlcpy(needle
, wificard
, sizeof(needle
));
726 strlcat(needle
, ":", sizeof(needle
));
727 fgets(buf
, bufsize
, fp
);
728 fgets(buf
, bufsize
, fp
);
729 fgets(buf
, bufsize
, fp
);
730 if ((datastart
= strstr(buf
, needle
)) != NULL
) {
731 datastart
= strstr(buf
, ":");
732 sscanf(datastart
+ 1, " %*d %d %*d %*d %*d %*d %*d %*d %*d %*d", &strength
);
735 /* close wifi file */
738 /* return strength in percent */
739 return smprintf("%d%%", strength
);
744 wifi_essid(const char *wificard
)
746 char id
[IW_ESSID_MAX_SIZE
+1];
751 memset(&wreq
, 0, sizeof(struct iwreq
));
752 wreq
.u
.essid
.length
= IW_ESSID_MAX_SIZE
+1;
754 /* set the interface */
755 sprintf(wreq
.ifr_name
, wificard
);
758 if((sockfd
= socket(AF_INET
, SOCK_DGRAM
, 0)) == -1) {
759 fprintf(stderr
, "Cannot open socket for interface: %s\n", wificard
);
760 return smprintf(unknowntext
);
762 wreq
.u
.essid
.pointer
= id
;
763 if (ioctl(sockfd
,SIOCGIWESSID
, &wreq
) == -1) {
764 fprintf(stderr
, "Get ESSID ioctl failed for interface %s\n", wificard
);
765 return smprintf(unknowntext
);
768 /* return the essid */
769 if (strcmp((char *)wreq
.u
.essid
.pointer
, "") == 0) {
770 return smprintf(unknowntext
);
772 return smprintf("%s", (char *)wreq
.u
.essid
.pointer
);
780 char status_string
[1024];
783 /* try to open display */
784 if (!(dpy
= XOpenDisplay(0x0))) {
785 fprintf(stderr
, "Cannot open display!\n");
789 /* return status every interval */
791 /* clear the string */
792 memset(status_string
, 0, sizeof(status_string
));
794 /* generate status_string */
795 for (size_t i
= 0; i
< sizeof(args
) / sizeof(args
[0]); ++i
) {
797 char *res
= argument
.func(argument
.args
);
798 char *element
= smprintf(argument
.format
, res
);
799 if (element
== NULL
) {
800 element
= smprintf(unknowntext
);
801 fprintf(stderr
, "Failed to format output.\n");
803 strlcat(status_string
, element
, sizeof(status_string
));
808 /* return the statusbar */
809 setstatus(status_string
);
811 /* wait, "update_interval - 1" because of get_cpu_usage() which uses 1 second */
812 sleep(update_interval
-1);
818 /* exit successfully */