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)
99 /* battery percentage */
101 battery_perc(const char *battery
)
104 char batterynowfile
[64] = "";
105 char batteryfullfile
[64] = "";
108 /* generate battery nowfile path */
109 strlcat(batterynowfile
, batterypath
, sizeof(batterynowfile
));
110 strlcat(batterynowfile
, battery
, sizeof(batterynowfile
));
111 strlcat(batterynowfile
, "/", sizeof(batterynowfile
));
112 strlcat(batterynowfile
, batterynow
, sizeof(batterynowfile
));
114 /* generate battery fullfile path */
115 strlcat(batteryfullfile
, batterypath
, sizeof(batteryfullfile
));
116 strlcat(batteryfullfile
, battery
, sizeof(batteryfullfile
));
117 strlcat(batteryfullfile
, "/", sizeof(batteryfullfile
));
118 strlcat(batteryfullfile
, batteryfull
, sizeof(batteryfullfile
));
120 /* open battery now file */
121 if (!(fp
= fopen(batterynowfile
, "r"))) {
122 fprintf(stderr
, "Error opening battery file: %s.\n", batterynowfile
);
123 return smprintf(unknowntext
);
127 fscanf(fp
, "%i", &now
);
129 /* close battery now file */
132 /* open battery full file */
133 if (!(fp
= fopen(batteryfullfile
, "r"))) {
134 fprintf(stderr
, "Error opening battery file.\n");
135 return smprintf(unknowntext
);
139 fscanf(fp
, "%i", &full
);
141 /* close battery full file */
144 /* calculate percent */
145 perc
= now
/ (full
/ 100);
147 /* return perc as string */
148 return smprintf("%d%%", perc
);
153 cpu_perc(const char *null
)
156 long double a
[4], b
[4];
160 if (!(fp
= fopen("/proc/stat","r"))) {
161 fprintf(stderr
, "Error opening stat file.\n");
162 return smprintf(unknowntext
);
166 fscanf(fp
, "%*s %Lf %Lf %Lf %Lf", &a
[0], &a
[1], &a
[2], &a
[3]);
168 /* close stat file */
171 /* wait a second (for avg values) */
175 if (!(fp
= fopen("/proc/stat","r"))) {
176 fprintf(stderr
, "Error opening stat file.\n");
177 return smprintf(unknowntext
);
181 fscanf(fp
, "%*s %Lf %Lf %Lf %Lf", &b
[0], &b
[1], &b
[2], &b
[3]);
183 /* close stat file */
186 /* calculate avg in this second */
187 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]));
189 /* return perc as string */
190 return smprintf("%d%%", perc
);
195 datetime(const char *timeformat
)
199 char *buf
= malloc(bufsize
);
201 fprintf(stderr
, "Failed to get date/time.\n");
202 return smprintf(unknowntext
);
205 /* get time in format */
207 setlocale(LC_TIME
, "");
208 if (!strftime(buf
, bufsize
, timeformat
, localtime(&tm
))) {
209 setlocale(LC_TIME
, "C");
211 fprintf(stderr
, "Strftime failed.\n");
212 return smprintf(unknowntext
);
215 setlocale(LC_TIME
, "C");
217 char *ret
= smprintf("%s", buf
);
224 disk_free(const char *mountpoint
)
228 /* try to open mountpoint */
229 if (statvfs(mountpoint
, &fs
) < 0) {
230 fprintf(stderr
, "Could not get filesystem info.\n");
231 return smprintf(unknowntext
);
235 return smprintf("%f", (float)fs
.f_bsize
* (float)fs
.f_bfree
/ 1024 / 1024 / 1024);
238 /* disk usage percentage */
240 disk_perc(const char *mountpoint
)
245 /* try to open mountpoint */
246 if (statvfs(mountpoint
, &fs
) < 0) {
247 fprintf(stderr
, "Could not get filesystem info.\n");
248 return smprintf(unknowntext
);
251 /* calculate percent */
252 perc
= 100 * (1.0f
- ((float)fs
.f_bfree
/ (float)fs
.f_blocks
));
255 return smprintf("%d%%", perc
);
260 disk_total(const char *mountpoint
)
264 /* try to open mountpoint */
265 if (statvfs(mountpoint
, &fs
) < 0) {
266 fprintf(stderr
, "Could not get filesystem info.\n");
267 return smprintf(unknowntext
);
271 return smprintf("%f", (float)fs
.f_bsize
* (float)fs
.f_blocks
/ 1024 / 1024 / 1024);
276 disk_used(const char *mountpoint
)
280 /* try to open mountpoint */
281 if (statvfs(mountpoint
, &fs
) < 0) {
282 fprintf(stderr
, "Could not get filesystem info.\n");
283 return smprintf(unknowntext
);
287 return smprintf("%f", (float)fs
.f_bsize
* ((float)fs
.f_blocks
- (float)fs
.f_bfree
) / 1024 / 1024 / 1024);
290 /* entropy available */
292 entropy(const char *null
)
297 /* open entropy file */
298 if (!(fp
= fopen("/proc/sys/kernel/random/entropy_avail", "r"))) {
299 fprintf(stderr
, "Could not open entropy file.\n");
300 return smprintf(unknowntext
);
303 /* extract entropy */
304 fscanf(fp
, "%d", &entropy
);
306 /* close entropy file */
310 return smprintf("%d", entropy
);
315 gid(const char *null
)
319 if ((gid
= getgid()) < 0) {
320 fprintf(stderr
, "Could no get gid.\n");
321 return smprintf(unknowntext
);
323 return smprintf("%d", gid
);
326 return smprintf(unknowntext
);
331 hostname(const char *null
)
333 char hostname
[HOST_NAME_MAX
];
336 /* open hostname file */
337 if (!(fp
= fopen("/proc/sys/kernel/hostname", "r"))) {
338 fprintf(stderr
, "Could not open hostname file.\n");
339 return smprintf(unknowntext
);
342 /* extract hostname */
343 fscanf(fp
, "%s\n", hostname
);
345 /* close hostname file */
349 return smprintf("%s", hostname
);
354 ip(const char *interface
)
356 struct ifaddrs
*ifaddr
, *ifa
;
358 char host
[NI_MAXHOST
];
360 /* check if getting ip address works */
361 if (getifaddrs(&ifaddr
) == -1) {
362 fprintf(stderr
, "Error getting IP address.\n");
363 return smprintf(unknowntext
);
366 /* get the ip address */
367 for (ifa
= ifaddr
; ifa
!= NULL
; ifa
= ifa
->ifa_next
) {
368 if (ifa
->ifa_addr
== NULL
)
371 s
= getnameinfo(ifa
->ifa_addr
, sizeof(struct sockaddr_in
), host
, NI_MAXHOST
, NULL
, 0, NI_NUMERICHOST
);
373 if ((strcmp(ifa
->ifa_name
, interface
) == 0) && (ifa
->ifa_addr
->sa_family
== AF_INET
)) {
375 fprintf(stderr
, "Error getting IP address.\n");
376 return smprintf(unknowntext
);
378 return smprintf("%s", host
);
382 /* free the address */
385 return smprintf(unknowntext
);
390 load_avg(const char *null
)
394 /* try to get load avg */
395 if (getloadavg(avgs
, 3) < 0) {
396 fprintf(stderr
, "Error getting load avg.\n");
397 return smprintf(unknowntext
);
401 return smprintf("%.2f %.2f %.2f", avgs
[0], avgs
[1], avgs
[2]);
406 ram_free(const char *null
)
411 /* open meminfo file */
412 if (!(fp
= fopen("/proc/meminfo", "r"))) {
413 fprintf(stderr
, "Error opening meminfo file.\n");
414 return smprintf(unknowntext
);
417 /* read the values */
418 fscanf(fp
, "MemFree: %ld kB\n", &free
);
420 /* close meminfo file */
423 /* return free ram as string */
424 return smprintf("%f", (float)free
/ 1024 / 1024);
429 ram_perc(const char *null
)
432 long total
, free
, buffers
, cached
;
435 /* open meminfo file */
436 if (!(fp
= fopen("/proc/meminfo", "r"))) {
437 fprintf(stderr
, "Error opening meminfo file.\n");
438 return smprintf(unknowntext
);
441 /* read the values */
442 fscanf(fp
, "MemTotal: %ld kB\n", &total
);
443 fscanf(fp
, "MemFree: %ld kB\n", &free
);
444 fscanf(fp
, "MemAvailable: %ld kB\nBuffers: %ld kB\n", &buffers
, &buffers
);
445 fscanf(fp
, "Cached: %ld kB\n", &cached
);
447 /* close meminfo file */
450 /* calculate percentage */
451 perc
= 100 * ((total
- free
) - (buffers
+ cached
)) / total
;
453 /* return perc as string */
454 return smprintf("%d%%", perc
);
459 ram_total(const char *null
)
464 /* open meminfo file */
465 if (!(fp
= fopen("/proc/meminfo", "r"))) {
466 fprintf(stderr
, "Error opening meminfo file.\n");
467 return smprintf(unknowntext
);
470 /* read the values */
471 fscanf(fp
, "MemTotal: %ld kB\n", &total
);
473 /* close meminfo file */
476 /* return total ram as string */
477 return smprintf("%f", (float)total
/ 1024 / 1024);
482 ram_used(const char *null
)
484 long free
, total
, buffers
, cached
, used
;
487 /* open meminfo file */
488 if (!(fp
= fopen("/proc/meminfo", "r"))) {
489 fprintf(stderr
, "Error opening meminfo file.\n");
490 return smprintf(unknowntext
);
493 /* read the values */
494 fscanf(fp
, "MemTotal: %ld kB\n", &total
);
495 fscanf(fp
, "MemFree: %ld kB\n", &free
);
496 fscanf(fp
, "MemAvailable: %ld kB\nBuffers: %ld kB\n", &buffers
, &buffers
);
497 fscanf(fp
, "Cached: %ld kB\n", &cached
);
499 /* close meminfo file */
503 used
= total
- free
- buffers
- cached
;
505 /* return used ram as string */
506 return smprintf("%f", (float)used
/ 1024 / 1024);
509 /* custom shell command */
511 run_command(const char* command
)
517 /* try to open the command output */
518 if (!(fp
= popen(command
, "r"))) {
519 fprintf(stderr
, "Could not get command output for: %s.\n", command
);
520 return smprintf(unknowntext
);
523 /* get command output text, save it to buffer */
524 fgets(buffer
, sizeof(buffer
) - 1, fp
);
529 /* add nullchar at the end */
530 for (int i
= 0 ; i
!= sizeof(buffer
); i
++) {
531 if (buffer
[i
] == '\0') {
537 buffer
[strlen(buffer
) - 1] = '\0';
539 /* return the output */
540 return smprintf("%s", buffer
);
545 temp(const char *file
)
550 /* open temperature file */
551 if (!(fp
= fopen(file
, "r"))) {
552 fprintf(stderr
, "Could not open temperature file.\n");
553 return smprintf(unknowntext
);
556 /* extract temperature */
557 fscanf(fp
, "%d", &temperature
);
559 /* close temperature file */
562 /* return temperature in degrees */
563 return smprintf("%d°C", temperature
/ 1000);
568 uptime(const char *null
)
576 hours
= info
.uptime
/ 3600;
577 minutes
= (info
.uptime
- hours
* 3600 ) / 60;
580 return smprintf("%dh %dm", hours
, minutes
);
585 username(const char *null
)
587 register struct passwd
*pw
;
594 /* if it worked, return */
596 return smprintf("%s", pw
->pw_name
);
598 fprintf(stderr
, "Could not get username.\n");
599 return smprintf(unknowntext
);
602 return smprintf(unknowntext
);
607 uid(const char *null
)
614 /* if it worked, return */
616 return smprintf("%d", uid
);
618 fprintf(stderr
, "Could not get uid.\n");
619 return smprintf(unknowntext
);
622 return smprintf(unknowntext
);
626 /* alsa volume percentage */
628 vol_perc(const char *soundcard
)
631 long vol
= 0, max
= 0, min
= 0;
633 snd_mixer_elem_t
*pcm_mixer
, *mas_mixer
;
634 snd_mixer_selem_id_t
*vol_info
, *mute_info
;
636 /* open everything */
637 snd_mixer_open(&handle
, 0);
638 snd_mixer_attach(handle
, soundcard
);
639 snd_mixer_selem_register(handle
, NULL
, NULL
);
640 snd_mixer_load(handle
);
642 /* prepare everything */
643 snd_mixer_selem_id_malloc(&vol_info
);
644 snd_mixer_selem_id_malloc(&mute_info
);
646 if (vol_info
== NULL
|| mute_info
== NULL
) {
647 fprintf(stderr
, "Could not get alsa volume.\n");
648 return smprintf(unknowntext
);
650 snd_mixer_selem_id_set_name(vol_info
, channel
);
651 snd_mixer_selem_id_set_name(mute_info
, channel
);
652 pcm_mixer
= snd_mixer_find_selem(handle
, vol_info
);
653 mas_mixer
= snd_mixer_find_selem(handle
, mute_info
);
656 snd_mixer_selem_get_playback_volume_range((snd_mixer_elem_t
*)pcm_mixer
, &min
, &max
);
657 snd_mixer_selem_get_playback_volume((snd_mixer_elem_t
*)pcm_mixer
, SND_MIXER_SCHN_MONO
, &vol
);
658 snd_mixer_selem_get_playback_switch(mas_mixer
, SND_MIXER_SCHN_MONO
, &mute
);
662 snd_mixer_selem_id_free(vol_info
);
664 snd_mixer_selem_id_free(mute_info
);
666 snd_mixer_close(handle
);
668 /* return the string (mute) */
670 return smprintf("mute");
672 return smprintf("%d%%", (vol
* 100) / max
);
675 /* wifi percentage */
677 wifi_perc(const char *wificard
)
685 char needle
[sizeof wificard
+ 1];
688 /* generate the path name */
689 memset(path
, 0, sizeof path
);
690 strlcat(path
, "/sys/class/net/", sizeof(path
));
691 strlcat(path
, wificard
, sizeof(path
));
692 strlcat(path
, "/operstate", sizeof(path
));
695 if(!(fp
= fopen(path
, "r"))) {
696 fprintf(stderr
, "Error opening wifi operstate file.\n");
697 return smprintf(unknowntext
);
700 /* read the status */
701 fgets(status
, 5, fp
);
703 /* close wifi file */
706 /* check if interface down */
707 if(strcmp(status
, "up\n") != 0)
708 return smprintf(unknowntext
);
711 if (!(fp
= fopen("/proc/net/wireless", "r"))) {
712 fprintf(stderr
, "Error opening wireless file.\n");
713 return smprintf(unknowntext
);
716 /* extract the signal strength */
717 strlcpy(needle
, wificard
, sizeof(needle
));
718 strlcat(needle
, ":", sizeof(needle
));
719 fgets(buf
, bufsize
, fp
);
720 fgets(buf
, bufsize
, fp
);
721 fgets(buf
, bufsize
, fp
);
722 if ((datastart
= strstr(buf
, needle
)) != NULL
) {
723 datastart
= strstr(buf
, ":");
724 sscanf(datastart
+ 1, " %*d %d %*d %*d %*d %*d %*d %*d %*d %*d", &strength
);
727 /* close wifi file */
730 /* return strength in percent */
731 return smprintf("%d%%", strength
);
736 wifi_essid(const char *wificard
)
738 char id
[IW_ESSID_MAX_SIZE
+1];
743 memset(&wreq
, 0, sizeof(struct iwreq
));
744 wreq
.u
.essid
.length
= IW_ESSID_MAX_SIZE
+1;
746 /* set the interface */
747 sprintf(wreq
.ifr_name
, wificard
);
750 if((sockfd
= socket(AF_INET
, SOCK_DGRAM
, 0)) == -1) {
751 fprintf(stderr
, "Cannot open socket for interface: %s\n", wificard
);
752 return smprintf(unknowntext
);
754 wreq
.u
.essid
.pointer
= id
;
755 if (ioctl(sockfd
,SIOCGIWESSID
, &wreq
) == -1) {
756 fprintf(stderr
, "Get ESSID ioctl failed for interface %s\n", wificard
);
757 return smprintf(unknowntext
);
760 /* return the essid */
761 if (strcmp((char *)wreq
.u
.essid
.pointer
, "") == 0)
762 return smprintf(unknowntext
);
764 return smprintf("%s", (char *)wreq
.u
.essid
.pointer
);
771 char status_string
[1024];
774 /* try to open display */
775 if (!(dpy
= XOpenDisplay(0x0))) {
776 fprintf(stderr
, "Cannot open display!\n");
780 /* return status every interval */
782 /* clear the string */
783 memset(status_string
, 0, sizeof(status_string
));
785 /* generate status_string */
786 for (size_t i
= 0; i
< sizeof(args
) / sizeof(args
[0]); ++i
) {
788 char *res
= argument
.func(argument
.args
);
789 char *element
= smprintf(argument
.format
, res
);
790 if (element
== NULL
) {
791 element
= smprintf(unknowntext
);
792 fprintf(stderr
, "Failed to format output.\n");
794 strlcat(status_string
, element
, sizeof(status_string
));
799 /* return the statusbar */
800 setstatus(status_string
);
802 /* wait, "update_interval - 1" because of get_cpu_usage() which uses 1 second */
803 sleep(update_interval
-1);
809 /* exit successfully */