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>
27 /* statusbar configuration type and struct */
28 typedef char *(*op_fun
) (const char *);
35 /* function declarations */
36 void setstatus(const char *str
);
37 char *smprintf(const char *fmt
, ...);
38 char *battery_perc(const char *battery
);
39 char *cpu_perc(const char *null
);
40 char *datetime(const char *timeformat
);
41 char *disk_free(const char *mountpoint
);
42 char *disk_perc(const char *mountpoint
);
43 char *disk_total(const char *mountpoint
);
44 char *disk_used(const char *mountpoint
);
45 char *entropy(const char *null
);
46 char *gid(const char *null
);
47 char *hostname(const char *null
);
48 char *ip(const char *interface
);
49 char *load_avg(const char *null
);
50 char *ram_free(const char *null
);
51 char *ram_perc(const char *null
);
52 char *ram_used(const char *null
);
53 char *ram_total(const char *null
);
54 char *run_command(const char *command
);
55 char *temp(const char *file
);
56 char *uid(const char *null
);
57 char *uptime(const char *null
);
58 char *username(const char *null
);
59 char *vol_perc(const char *soundcard
);
60 char *wifi_perc(const char *wificard
);
61 char *wifi_essid(const char *wificard
);
63 /* global variables */
66 /* configuration header */
71 setstatus(const char *str
)
73 /* set WM_NAME via X11 */
74 XStoreName(dpy
, DefaultRootWindow(dpy
), str
);
78 /* smprintf function */
80 smprintf(const char *fmt
, ...)
85 va_start(fmtargs
, fmt
);
86 if (vasprintf(&ret
, fmt
, fmtargs
) < 0) {
94 /* battery percentage */
96 battery_perc(const char *battery
)
99 char batterynowfile
[64] = "";
100 char batteryfullfile
[64] = "";
103 /* generate battery nowfile path */
104 strcat(batterynowfile
, batterypath
);
105 strcat(batterynowfile
, battery
);
106 strcat(batterynowfile
, "/");
107 strcat(batterynowfile
, batterynow
);
109 /* generate battery fullfile path */
110 strcat(batteryfullfile
, batterypath
);
111 strcat(batteryfullfile
, battery
);
112 strcat(batteryfullfile
, "/");
113 strcat(batteryfullfile
, batteryfull
);
115 /* open battery now file */
116 if (!(fp
= fopen(batterynowfile
, "r"))) {
117 fprintf(stderr
, "Error opening battery file: %s.\n", batterynowfile
);
118 return smprintf(unknowntext
);
122 fscanf(fp
, "%i", &now
);
124 /* close battery now file */
127 /* open battery full file */
128 if (!(fp
= fopen(batteryfullfile
, "r"))) {
129 fprintf(stderr
, "Error opening battery file.\n");
130 return smprintf(unknowntext
);
134 fscanf(fp
, "%i", &full
);
136 /* close battery full file */
139 /* calculate percent */
140 perc
= now
/ (full
/ 100);
142 /* return perc as string */
143 return smprintf("%d%%", perc
);
148 cpu_perc(const char *null
)
151 long double a
[4], b
[4];
155 if (!(fp
= fopen("/proc/stat","r"))) {
156 fprintf(stderr
, "Error opening stat file.\n");
157 return smprintf(unknowntext
);
161 fscanf(fp
, "%*s %Lf %Lf %Lf %Lf", &a
[0], &a
[1], &a
[2], &a
[3]);
163 /* close stat file */
166 /* wait a second (for avg values) */
170 if (!(fp
= fopen("/proc/stat","r"))) {
171 fprintf(stderr
, "Error opening stat file.\n");
172 return smprintf(unknowntext
);
176 fscanf(fp
, "%*s %Lf %Lf %Lf %Lf", &b
[0], &b
[1], &b
[2], &b
[3]);
178 /* close stat file */
181 /* calculate avg in this second */
182 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]));
184 /* return perc as string */
185 return smprintf("%d%%", perc
);
190 datetime(const char *timeformat
)
194 char *buf
= malloc(bufsize
);
196 fprintf(stderr
, "Failed to get date/time.\n");
197 return smprintf(unknowntext
);
200 /* get time in format */
202 setlocale(LC_TIME
, "");
203 if (!strftime(buf
, bufsize
, timeformat
, localtime(&tm
))) {
204 setlocale(LC_TIME
, "C");
206 fprintf(stderr
, "Strftime failed.\n");
207 return smprintf(unknowntext
);
210 setlocale(LC_TIME
, "C");
212 char *ret
= smprintf("%s", buf
);
219 disk_free(const char *mountpoint
)
223 /* try to open mountpoint */
224 if (statvfs(mountpoint
, &fs
) < 0) {
225 fprintf(stderr
, "Could not get filesystem info.\n");
226 return smprintf(unknowntext
);
230 return smprintf("%f", (float)fs
.f_bsize
* (float)fs
.f_bfree
/ 1024 / 1024 / 1024);
233 /* disk usage percentage */
235 disk_perc(const char *mountpoint
)
240 /* try to open mountpoint */
241 if (statvfs(mountpoint
, &fs
) < 0) {
242 fprintf(stderr
, "Could not get filesystem info.\n");
243 return smprintf(unknowntext
);
246 /* calculate percent */
247 perc
= 100 * (1.0f
- ((float)fs
.f_bfree
/ (float)fs
.f_blocks
));
250 return smprintf("%d%%", perc
);
255 disk_total(const char *mountpoint
)
259 /* try to open mountpoint */
260 if (statvfs(mountpoint
, &fs
) < 0) {
261 fprintf(stderr
, "Could not get filesystem info.\n");
262 return smprintf(unknowntext
);
266 return smprintf("%f", (float)fs
.f_bsize
* (float)fs
.f_blocks
/ 1024 / 1024 / 1024);
271 disk_used(const char *mountpoint
)
275 /* try to open mountpoint */
276 if (statvfs(mountpoint
, &fs
) < 0) {
277 fprintf(stderr
, "Could not get filesystem info.\n");
278 return smprintf(unknowntext
);
282 return smprintf("%f", (float)fs
.f_bsize
* ((float)fs
.f_blocks
- (float)fs
.f_bfree
) / 1024 / 1024 / 1024);
285 /* entropy available */
287 entropy(const char *null
)
292 /* open entropy file */
293 if (!(fp
= fopen("/proc/sys/kernel/random/entropy_avail", "r"))) {
294 fprintf(stderr
, "Could not open entropy file.\n");
295 return smprintf(unknowntext
);
298 /* extract entropy */
299 fscanf(fp
, "%d", &entropy
);
301 /* close entropy file */
305 return smprintf("%d", entropy
);
310 gid(const char *null
)
314 if ((gid
= getgid()) < 0) {
315 fprintf(stderr
, "Could no get gid.\n");
316 return smprintf(unknowntext
);
318 return smprintf("%d", gid
);
321 return smprintf(unknowntext
);
326 hostname(const char *null
)
328 char hostname
[HOST_NAME_MAX
];
331 /* open hostname file */
332 if (!(fp
= fopen("/proc/sys/kernel/hostname", "r"))) {
333 fprintf(stderr
, "Could not open hostname file.\n");
334 return smprintf(unknowntext
);
337 /* extract hostname */
338 fscanf(fp
, "%s\n", hostname
);
340 /* close hostname file */
344 return smprintf("%s", hostname
);
349 ip(const char *interface
)
351 struct ifaddrs
*ifaddr
, *ifa
;
353 char host
[NI_MAXHOST
];
355 /* check if getting ip address works */
356 if (getifaddrs(&ifaddr
) == -1) {
357 fprintf(stderr
, "Error getting IP address.\n");
358 return smprintf(unknowntext
);
361 /* get the ip address */
362 for (ifa
= ifaddr
; ifa
!= NULL
; ifa
= ifa
->ifa_next
) {
363 if (ifa
->ifa_addr
== NULL
) {
367 s
= getnameinfo(ifa
->ifa_addr
, sizeof(struct sockaddr_in
), host
, NI_MAXHOST
, NULL
, 0, NI_NUMERICHOST
);
369 if ((strcmp(ifa
->ifa_name
, interface
) == 0) && (ifa
->ifa_addr
->sa_family
== AF_INET
)) {
371 fprintf(stderr
, "Error getting IP address.\n");
372 return smprintf(unknowntext
);
374 return smprintf("%s", host
);
378 /* free the address */
381 return smprintf(unknowntext
);
386 load_avg(const char *null
)
390 /* try to get load avg */
391 if (getloadavg(avgs
, 3) < 0) {
392 fprintf(stderr
, "Error getting load avg.\n");
393 return smprintf(unknowntext
);
397 return smprintf("%.2f %.2f %.2f", avgs
[0], avgs
[1], avgs
[2]);
402 ram_free(const char *null
)
407 /* open meminfo file */
408 if (!(fp
= fopen("/proc/meminfo", "r"))) {
409 fprintf(stderr
, "Error opening meminfo file.\n");
410 return smprintf(unknowntext
);
413 /* read the values */
414 fscanf(fp
, "MemFree: %ld kB\n", &free
);
416 /* close meminfo file */
419 /* return free ram as string */
420 return smprintf("%f", (float)free
/ 1024 / 1024);
425 ram_perc(const char *null
)
428 long total
, free
, buffers
, cached
;
431 /* open meminfo file */
432 if (!(fp
= fopen("/proc/meminfo", "r"))) {
433 fprintf(stderr
, "Error opening meminfo file.\n");
434 return smprintf(unknowntext
);
437 /* read the values */
438 fscanf(fp
, "MemTotal: %ld kB\n", &total
);
439 fscanf(fp
, "MemFree: %ld kB\n", &free
);
440 fscanf(fp
, "MemAvailable: %ld kB\nBuffers: %ld kB\n", &buffers
, &buffers
);
441 fscanf(fp
, "Cached: %ld kB\n", &cached
);
443 /* close meminfo file */
446 /* calculate percentage */
447 perc
= 100 * ((total
- free
) - (buffers
+ cached
)) / total
;
449 /* return perc as string */
450 return smprintf("%d%%", perc
);
455 ram_total(const char *null
)
460 /* open meminfo file */
461 if (!(fp
= fopen("/proc/meminfo", "r"))) {
462 fprintf(stderr
, "Error opening meminfo file.\n");
463 return smprintf(unknowntext
);
466 /* read the values */
467 fscanf(fp
, "MemTotal: %ld kB\n", &total
);
469 /* close meminfo file */
472 /* return total ram as string */
473 return smprintf("%f", (float)total
/ 1024 / 1024);
478 ram_used(const char *null
)
480 long free
, total
, buffers
, cached
, used
;
483 /* open meminfo file */
484 if (!(fp
= fopen("/proc/meminfo", "r"))) {
485 fprintf(stderr
, "Error opening meminfo file.\n");
486 return smprintf(unknowntext
);
489 /* read the values */
490 fscanf(fp
, "MemTotal: %ld kB\n", &total
);
491 fscanf(fp
, "MemFree: %ld kB\n", &free
);
492 fscanf(fp
, "MemAvailable: %ld kB\nBuffers: %ld kB\n", &buffers
, &buffers
);
493 fscanf(fp
, "Cached: %ld kB\n", &cached
);
495 /* close meminfo file */
499 used
= total
- free
- buffers
- cached
;
501 /* return used ram as string */
502 return smprintf("%f", (float)used
/ 1024 / 1024);
505 /* custom shell command */
507 run_command(const char* command
)
513 /* try to open the command output */
514 if (!(fp
= popen(command
, "r"))) {
515 fprintf(stderr
, "Could not get command output for: %s.\n", command
);
516 return smprintf(unknowntext
);
519 /* get command output text, save it to buffer */
520 fgets(buffer
, sizeof(buffer
) - 1, fp
);
525 /* add nullchar at the end */
526 for (int i
= 0 ; i
!= sizeof(buffer
); i
++) {
527 if (buffer
[i
] == '\0') {
533 buffer
[strlen(buffer
) - 1] = '\0';
536 /* return the output */
537 return smprintf("%s", buffer
);
542 temp(const char *file
)
547 /* open temperature file */
548 if (!(fp
= fopen(file
, "r"))) {
549 fprintf(stderr
, "Could not open temperature file.\n");
550 return smprintf(unknowntext
);
553 /* extract temperature */
554 fscanf(fp
, "%d", &temperature
);
556 /* close temperature file */
559 /* return temperature in degrees */
560 return smprintf("%d°C", temperature
/ 1000);
565 uptime(const char *null
)
573 hours
= info
.uptime
/ 3600;
574 minutes
= (info
.uptime
- hours
* 3600 ) / 60;
577 return smprintf("%dh %dm", hours
, minutes
);
582 username(const char *null
)
584 register struct passwd
*pw
;
591 /* if it worked, return */
593 return smprintf("%s", pw
->pw_name
);
595 fprintf(stderr
, "Could not get username.\n");
596 return smprintf(unknowntext
);
599 return smprintf(unknowntext
);
604 uid(const char *null
)
611 /* if it worked, return */
613 return smprintf("%d", uid
);
615 fprintf(stderr
, "Could not get uid.\n");
616 return smprintf(unknowntext
);
619 return smprintf(unknowntext
);
623 /* alsa volume percentage */
625 vol_perc(const char *soundcard
)
628 long vol
= 0, max
= 0, min
= 0;
630 snd_mixer_elem_t
*pcm_mixer
, *mas_mixer
;
631 snd_mixer_selem_id_t
*vol_info
, *mute_info
;
633 /* open everything */
634 snd_mixer_open(&handle
, 0);
635 snd_mixer_attach(handle
, soundcard
);
636 snd_mixer_selem_register(handle
, NULL
, NULL
);
637 snd_mixer_load(handle
);
639 /* prepare everything */
640 snd_mixer_selem_id_malloc(&vol_info
);
641 snd_mixer_selem_id_malloc(&mute_info
);
643 if (vol_info
== NULL
|| mute_info
== NULL
) {
644 fprintf(stderr
, "Could not get alsa volume.\n");
645 return smprintf(unknowntext
);
647 snd_mixer_selem_id_set_name(vol_info
, channel
);
648 snd_mixer_selem_id_set_name(mute_info
, channel
);
649 pcm_mixer
= snd_mixer_find_selem(handle
, vol_info
);
650 mas_mixer
= snd_mixer_find_selem(handle
, mute_info
);
653 snd_mixer_selem_get_playback_volume_range((snd_mixer_elem_t
*)pcm_mixer
, &min
, &max
);
654 snd_mixer_selem_get_playback_volume((snd_mixer_elem_t
*)pcm_mixer
, SND_MIXER_SCHN_MONO
, &vol
);
655 snd_mixer_selem_get_playback_switch(mas_mixer
, SND_MIXER_SCHN_MONO
, &mute
);
659 snd_mixer_selem_id_free(vol_info
);
662 snd_mixer_selem_id_free(mute_info
);
665 snd_mixer_close(handle
);
668 /* return the string (mute) */
670 return smprintf("mute");
672 return smprintf("%d%%", (vol
* 100) / max
);
676 /* wifi percentage */
678 wifi_perc(const char *wificard
)
686 char needle
[sizeof wificard
+ 1];
689 /* generate the path name */
690 memset(path
, 0, sizeof path
);
691 strcat(path
, "/sys/class/net/");
692 strcat(path
, wificard
);
693 strcat(path
, "/operstate");
696 if(!(fp
= fopen(path
, "r"))) {
697 fprintf(stderr
, "Error opening wifi operstate file.\n");
698 return smprintf(unknowntext
);
701 /* read the status */
702 fgets(status
, 5, fp
);
704 /* close wifi file */
707 /* check if interface down */
708 if(strcmp(status
, "up\n") != 0) {
709 return smprintf(unknowntext
);
713 if (!(fp
= fopen("/proc/net/wireless", "r"))) {
714 fprintf(stderr
, "Error opening wireless file.\n");
715 return smprintf(unknowntext
);
718 /* extract the signal strength */
719 strcpy(needle
, wificard
);
721 fgets(buf
, bufsize
, fp
);
722 fgets(buf
, bufsize
, fp
);
723 fgets(buf
, bufsize
, fp
);
724 if ((datastart
= strstr(buf
, needle
)) != NULL
) {
725 datastart
= strstr(buf
, ":");
726 sscanf(datastart
+ 1, " %*d %d %*d %*d %*d %*d %*d %*d %*d %*d", &strength
);
729 /* close wifi file */
732 /* return strength in percent */
733 return smprintf("%d%%", strength
);
738 wifi_essid(const char *wificard
)
740 char id
[IW_ESSID_MAX_SIZE
+1];
745 memset(&wreq
, 0, sizeof(struct iwreq
));
746 wreq
.u
.essid
.length
= IW_ESSID_MAX_SIZE
+1;
748 /* set the interface */
749 sprintf(wreq
.ifr_name
, wificard
);
752 if((sockfd
= socket(AF_INET
, SOCK_DGRAM
, 0)) == -1) {
753 fprintf(stderr
, "Cannot open socket for interface: %s\n", wificard
);
754 return smprintf(unknowntext
);
756 wreq
.u
.essid
.pointer
= id
;
757 if (ioctl(sockfd
,SIOCGIWESSID
, &wreq
) == -1) {
758 fprintf(stderr
, "Get ESSID ioctl failed for interface %s\n", wificard
);
759 return smprintf(unknowntext
);
762 /* return the essid */
763 if (strcmp((char *)wreq
.u
.essid
.pointer
, "") == 0) {
764 return smprintf(unknowntext
);
766 return smprintf("%s", (char *)wreq
.u
.essid
.pointer
);
774 char status_string
[1024];
777 /* try to open display */
778 if (!(dpy
= XOpenDisplay(0x0))) {
779 fprintf(stderr
, "Cannot open display!\n");
783 /* return status every interval */
785 /* clear the string */
786 memset(status_string
, 0, sizeof(status_string
));
788 /* generate status_string */
789 for (size_t i
= 0; i
< sizeof(args
) / sizeof(args
[0]); ++i
) {
791 char *res
= argument
.func(argument
.args
);
792 char *element
= smprintf(argument
.format
, res
);
793 if (element
== NULL
) {
794 element
= smprintf(unknowntext
);
795 fprintf(stderr
, "Failed to format output.\n");
797 strcat(status_string
, element
);
802 /* return the statusbar */
803 setstatus(status_string
);
805 /* wait, "update_interval - 1" because of get_cpu_usage() which uses 1 second */
806 sleep(update_interval
-1);
812 /* exit successfully */