X-Git-Url: https://git.xinqibao.xyz/slstatus.git/blobdiff_plain/864f2246b5dfeb023137d8f5ebb599e4eb6aedda..046e273a7eb270f4b275c02546e4547d8cfc2a82:/slstatus.c diff --git a/slstatus.c b/slstatus.c index 962744a..214f667 100644 --- a/slstatus.c +++ b/slstatus.c @@ -1,5 +1,6 @@ /* See LICENSE file for copyright and license details. */ +/* global libraries */ #include #include #include @@ -9,98 +10,86 @@ #include #include +/* local libraries */ #include "config.h" -char *smprintf(char *fmt, ...); - +/* functions */ void setstatus(char *str); - -char *wifi_signal(); -char *battery(); -char *cpu_usage(); -char *cpu_temperature(); -char *ram_usage(); -char *volume(); -char *datetime(); - +char *smprintf(char *fmt, ...); +char *get_battery(); +char *get_cpu_temperature(); +char *get_cpu_usage(); +char *get_datetime(); +char *get_ram_usage(); +char *get_volume(); +char *get_wifi_signal(); + +/* global variables */ static Display *dpy; +/* set statusbar (WM_NAME) */ +void +setstatus(char *str) +{ + XStoreName(dpy, DefaultRootWindow(dpy), str); + XSync(dpy, False); +} + /* smprintf function */ char * smprintf(char *fmt, ...) { - va_list fmtargs; - char *ret; - int len; + va_list fmtargs; + char *ret = NULL; + va_start(fmtargs, fmt); + if (vasprintf(&ret, fmt, fmtargs) < 0) + return NULL; + va_end(fmtargs); + + return ret; +} + +/* battery percentage */ +char * +get_battery() +{ + int now, full, perc; + FILE *fp; + + /* open battery now file */ + if (!(fp = fopen(batterynowfile, "r"))) { + fprintf(stderr, "Error opening battery file."); + exit(1); + } - va_start(fmtargs, fmt); - len = vsnprintf(NULL, 0, fmt, fmtargs); - va_end(fmtargs); + /* read value */ + fscanf(fp, "%i", &now); - ret = malloc(++len); - if (ret == NULL) { - fprintf(stderr, "Malloc error."); - exit(1); - } + /* close battery now file */ + fclose(fp); - va_start(fmtargs, fmt); - vsnprintf(ret, len, fmt, fmtargs); - va_end(fmtargs); + /* open battery full file */ + if (!(fp = fopen(batteryfullfile, "r"))) { + fprintf(stderr, "Error opening battery file."); + exit(1); + } - return ret; -} + /* read value */ + fscanf(fp, "%i", &full); -/* set statusbar (WM_NAME) */ -void -setstatus(char *str) -{ - XStoreName(dpy, DefaultRootWindow(dpy), str); - XSync(dpy, False); -} + /* close battery full file */ + fclose(fp); -/* alsa volume percentage */ -char * -volume() -{ - int mute = 0; - long vol = 0, max = 0, min = 0; - - snd_mixer_t *handle; - snd_mixer_elem_t *pcm_mixer, *mas_mixer; - snd_mixer_selem_id_t *vol_info, *mute_info; - snd_mixer_open(&handle, 0); - snd_mixer_attach(handle, soundcard); - snd_mixer_selem_register(handle, NULL, NULL); - snd_mixer_load(handle); - snd_mixer_selem_id_malloc(&vol_info); - snd_mixer_selem_id_malloc(&mute_info); - snd_mixer_selem_id_set_name(vol_info, channel); - snd_mixer_selem_id_set_name(mute_info, channel); - pcm_mixer = snd_mixer_find_selem(handle, vol_info); - mas_mixer = snd_mixer_find_selem(handle, mute_info); - snd_mixer_selem_get_playback_volume_range((snd_mixer_elem_t *)pcm_mixer, - &min, &max); - snd_mixer_selem_get_playback_volume((snd_mixer_elem_t *)pcm_mixer, - SND_MIXER_SCHN_MONO, &vol); - snd_mixer_selem_get_playback_switch(mas_mixer, SND_MIXER_SCHN_MONO, - &mute); - - if (vol_info) - snd_mixer_selem_id_free(vol_info); - if (mute_info) - snd_mixer_selem_id_free(mute_info); - if (handle) - snd_mixer_close(handle); - - if (!mute) - return "mute"; - else - return smprintf("%d%%", (vol * 100) / max); + /* calculate percent */ + perc = now / (full / 100); + + /* return perc as string */ + return smprintf("%d%%", perc); } /* cpu temperature */ char * -cpu_temperature() +get_cpu_temperature() { int temperature; FILE *fp; @@ -111,192 +100,254 @@ cpu_temperature() exit(1); } - /* extract temperature, close file */ + /* extract temperature */ fscanf(fp, "%d", &temperature); + + /* close temperature file */ fclose(fp); /* return temperature in degrees */ return smprintf("%d°C", temperature / 1000); } -/* wifi percentage */ +/* cpu percentage */ char * -wifi_signal() +get_cpu_usage() { - int bufsize = 255; - int strength; - char buf[bufsize]; - char *datastart; - char path_start[16] = "/sys/class/net/"; - char path_end[11] = "/operstate"; - char path[32]; - FILE *fp; - - /* generate the path name */ - strcat(path, path_start); - strcat(path, wificard); - strcat(path, path_end); + int perc; + long double a[4], b[4]; + FILE *fp; - /* open wifi file, extract status, close file */ - if(!(fp = fopen(path, "r"))) { - fprintf(stderr, "Error opening wifi operstate file."); + /* open stat file */ + if (!(fp = fopen("/proc/stat","r"))) { + fprintf(stderr, "Error opening stat file."); exit(1); } - char status[5]; - fgets(status, 5, fp); + + /* read values */ + fscanf(fp, "%*s %Lf %Lf %Lf %Lf", &a[0], &a[1], &a[2], &a[3]); + + /* close stat file */ fclose(fp); - /* check if interface down */ - if(strcmp(status, "up\n") != 0){ - return "n/a"; - } + /* wait a second (for avg values) */ + sleep(1); - /* open wifi file */ - if (!(fp = fopen("/proc/net/wireless", "r"))) { - fprintf(stderr, "Error opening wireless file."); + /* open stat file */ + if (!(fp = fopen("/proc/stat","r"))) { + fprintf(stderr, "Error opening stat file."); exit(1); } - /* extract the signal strength and close the file */ - fgets(buf, bufsize, fp); - fgets(buf, bufsize, fp); - fgets(buf, bufsize, fp); - if ((datastart = strstr(buf, "wlp3s0:")) != NULL) { - datastart = strstr(buf, ":"); - sscanf(datastart + 1, " %*d %d %*d %*d %*d %*d %*d %*d %*d %*d", - &strength); - } - fclose(fp); - - /* return strength in percent */ - return smprintf("%d%%", strength); -} - -/* battery percentage */ -char * -battery() -{ - int batt_now; - int batt_full; - int batt_perc; - FILE *fp; + /* read values */ + fscanf(fp, "%*s %Lf %Lf %Lf %Lf", &b[0], &b[1], &b[2], &b[3]); - /* open battery now file, extract and close */ - if (!(fp = fopen(batterynowfile, "r"))) { - fprintf(stderr, "Error opening battery file."); - exit(1); - } - fscanf(fp, "%i", &batt_now); - fclose(fp); - - /* extract battery full file, extract and close */ - if (!(fp = fopen(batteryfullfile, "r"))) { - fprintf(stderr, "Error opening battery file."); - exit(1); - } - fscanf(fp, "%i", &batt_full); - fclose(fp); + /* close stat file */ + fclose(fp); - /* calculate percent */ - batt_perc = batt_now / (batt_full / 100); + /* calculate avg in this second */ + 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])); - /* return percent */ - return smprintf("%d%%", batt_perc); + /* return perc as string */ + return smprintf("%d%%", perc); } /* date and time */ char * -datetime() +get_datetime() { - time_t tm; - size_t bufsize = 19; - char *buf = malloc(bufsize); + time_t tm; + size_t bufsize = 64; + char *buf = malloc(bufsize); /* get time in format */ - time(&tm); - if(!strftime(buf, bufsize, timeformat, localtime(&tm))) { - fprintf(stderr, "Strftime failed.\n"); - exit(1); - } + time(&tm); + if(!strftime(buf, bufsize, timeformat, localtime(&tm))) { + fprintf(stderr, "Strftime failed.\n"); + exit(1); + } /* return time */ - return buf; + return smprintf("%s", buf); } -/* cpu percentage */ +/* ram percentage */ char * -cpu_usage() +get_ram_usage() { + int perc; + long total, free, buffers, cached; FILE *fp; - long double a[4], b[4], cpu_perc; - /* open stat file, read and close, do same after 1 second */ - if (!(fp = fopen("/proc/stat","r"))) { - fprintf(stderr, "Error opening stat file."); - exit(1); - } - fscanf(fp,"%*s %Lf %Lf %Lf %Lf",&a[0],&a[1],&a[2],&a[3]); - fclose(fp); - sleep(1); - if (!(fp = fopen("/proc/stat","r"))) { - fprintf(stderr, "Error opening stat file."); + /* open meminfo file */ + if (!(fp = fopen("/proc/meminfo", "r"))) { + fprintf(stderr, "Error opening meminfo file."); exit(1); } - fscanf(fp,"%*s %Lf %Lf %Lf %Lf",&b[0],&b[1],&b[2],&b[3]); + + /* read the values */ + fscanf(fp, "MemTotal: %ld kB\n", &total); + fscanf(fp, "MemFree: %ld kB\n", &free); + fscanf(fp, "MemAvailable: %ld kB\nBuffers: %ld kB\n", &buffers, &buffers); + fscanf(fp, "Cached: %ld kB\n", &cached); + + /* close meminfo file */ fclose(fp); - /* calculate average in 1 second */ - cpu_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])); + /* calculate percentage */ + perc = 100 * ((total - free) - (buffers + cached)) / total; + + /* return perc as string */ + return smprintf("%d%%", perc); +} - /* return avg cpu percentage */ - return smprintf("%d%%", (int)cpu_perc); +/* alsa volume percentage */ +char * +get_volume() +{ + int mute = 0; + long vol = 0, max = 0, min = 0; + + /* get volume from alsa */ + snd_mixer_t *handle; + snd_mixer_elem_t *pcm_mixer, *mas_mixer; + snd_mixer_selem_id_t *vol_info, *mute_info; + snd_mixer_open(&handle, 0); + snd_mixer_attach(handle, soundcard); + snd_mixer_selem_register(handle, NULL, NULL); + snd_mixer_load(handle); + snd_mixer_selem_id_malloc(&vol_info); + snd_mixer_selem_id_malloc(&mute_info); + snd_mixer_selem_id_set_name(vol_info, channel); + snd_mixer_selem_id_set_name(mute_info, channel); + pcm_mixer = snd_mixer_find_selem(handle, vol_info); + mas_mixer = snd_mixer_find_selem(handle, mute_info); + snd_mixer_selem_get_playback_volume_range((snd_mixer_elem_t *)pcm_mixer, &min, &max); + snd_mixer_selem_get_playback_volume((snd_mixer_elem_t *)pcm_mixer, SND_MIXER_SCHN_MONO, &vol); + snd_mixer_selem_get_playback_switch(mas_mixer, SND_MIXER_SCHN_MONO, &mute); + if (vol_info) + snd_mixer_selem_id_free(vol_info); + if (mute_info) + snd_mixer_selem_id_free(mute_info); + if (handle) + snd_mixer_close(handle); + + /* return the string (mute) */ + if (!mute) + return smprintf("mute"); + else + return smprintf("%d%%", (vol * 100) / max); } -/* ram percentage */ +/* wifi percentage */ char * -ram_usage() +get_wifi_signal() { + int bufsize = 255; + int strength; + char buf[bufsize]; + char *datastart; + char path_start[16] = "/sys/class/net/"; + char path_end[11] = "/operstate"; + char path[32]; + char status[5]; + char needle[sizeof wificard + 1]; FILE *fp; - long total, free, available; - int ram_perc; - /* read meminfo file, extract and close */ - if (!(fp = fopen("/proc/meminfo", "r"))) { - fprintf(stderr, "Error opening meminfo file."); + /* generate the path name */ + memset(path, 0, sizeof path); + strcat(path, path_start); + strcat(path, wificard); + strcat(path, path_end); + + /* open wifi file */ + if(!(fp = fopen(path, "r"))) { + fprintf(stderr, "Error opening wifi operstate file."); exit(1); } - fscanf(fp, "MemTotal: %ld kB\n", &total); - fscanf(fp, "MemFree: %ld kB\n", &free); - fscanf(fp, "MemAvailable: %ld kB\n", &available); + + /* read the status */ + fgets(status, 5, fp); + + /* close wifi file */ fclose(fp); - /* calculate percentage */ - ram_perc = 100 * (total - available) / total; + /* check if interface down */ + if(strcmp(status, "up\n") != 0){ + return "n/a"; + } - /* return in percent */ - return smprintf("%d%%",ram_perc); + /* open wifi file */ + if (!(fp = fopen("/proc/net/wireless", "r"))) { + fprintf(stderr, "Error opening wireless file."); + exit(1); + } + + /* extract the signal strength */ + strcpy(needle, wificard); + strcat(needle, ":"); + fgets(buf, bufsize, fp); + fgets(buf, bufsize, fp); + fgets(buf, bufsize, fp); + if ((datastart = strstr(buf, needle)) != NULL) { + datastart = strstr(buf, ":"); + sscanf(datastart + 1, " %*d %d %*d %*d %*d %*d %*d %*d %*d %*d", &strength); + } + + /* close wifi file */ + fclose(fp); + + /* return strength in percent */ + return smprintf("%d%%", strength); } +/* main function */ int main() { char status[1024]; + char *battery = NULL; + char *cpu_temperature = NULL; + char *cpu_usage = NULL; + char *datetime = NULL; + char *ram_usage = NULL; + char *volume = NULL; + char *wifi_signal = NULL; /* open display */ - if (( dpy = XOpenDisplay(0x0)) == NULL ) { + if (!(dpy = XOpenDisplay(0x0))) { fprintf(stderr, "Cannot open display!\n"); exit(1); } - /* return status every second */ - for (;;) { - sprintf(status, FORMATSTRING, ARGUMENTS); - setstatus(status); - } + /* return status every second */ + for (;;) { + /* assign the values */ + battery = get_battery(); + cpu_temperature = get_cpu_temperature(); + cpu_usage = get_cpu_usage(); + datetime = get_datetime(); + ram_usage = get_ram_usage(); + volume = get_volume(); + wifi_signal = get_wifi_signal(); + + /* return the status */ + sprintf(status, FORMATSTRING, ARGUMENTS); + setstatus(status); + + /* free the values */ + free(battery); + free(cpu_temperature); + free(cpu_usage); + free(datetime); + free(ram_usage); + free(volume); + free(wifi_signal); + } /* close display */ XCloseDisplay(dpy); /* exit successfully */ - return 0; + return 0; }