slstatus.c

   1 /* See LICENSE file for copyright and license details. */
   2
   3 /* global libraries */
   4 #include <alsa/asoundlib.h>
   5 #include <stdarg.h>
   6 #include <stdio.h>
   7 #include <stdlib.h>
   8 #include <string.h>
   9 #include <time.h>
  10 #include <unistd.h>
  11 #include <X11/Xlib.h>
  12
  13 /* local libraries */
  14 #include "config.h"
  15
  16 /* functions */
  17 void setstatus(char *str);
  18 char *battery();
  19 char *cpu_temperature();
  20 char *cpu_usage();
  21 char *datetime();
  22 char *ram_usage();
  23 char *smprintf(char *fmt, ...);
  24 char *volume();
  25 char *wifi_signal();
  26
  27 /* global variables */
  28 static Display *dpy;
  29
  30 /* set statusbar (WM_NAME) */
  31 void
  32 setstatus(char *str)
  33 {
  34   XStoreName(dpy, DefaultRootWindow(dpy), str);
  35   XSync(dpy, False);
  36 }
  37
  38 /* battery percentage */
  39 char *
  40 battery()
  41 {
  42   int battery_now, battery_full, battery_perc;
  43   FILE *fp;
  44
  45   /* open battery now file */
  46   if (!(fp = fopen(batterynowfile, "r"))) {
  47     fprintf(stderr, "Error opening battery file.");
  48     exit(1);
  49   }
  50
  51   /* read value */
  52   fscanf(fp, "%i", &battery_now);
  53
  54   /* close battery now file */
  55   fclose(fp);
  56
  57   /* open battery full file */
  58   if (!(fp = fopen(batteryfullfile, "r"))) {
  59     fprintf(stderr, "Error opening battery file.");
  60     exit(1);
  61   }
  62
  63   /* read value */
  64   fscanf(fp, "%i", &battery_full);
  65
  66   /* close battery full file */
  67   fclose(fp);
  68
  69   /* calculate percent */
  70   battery_perc = battery_now / (battery_full / 100);
  71
  72   /* return batt_perc as string */
  73   return smprintf("%d%%", battery_perc);
  74 }
  75
  76 /* cpu temperature */
  77 char *
  78 cpu_temperature()
  79 {
  80   int temperature;
  81   FILE *fp;
  82
  83   /* open temperature file */
  84   if (!(fp = fopen(tempfile, "r"))) {
  85     fprintf(stderr, "Could not open temperature file.\n");
  86     exit(1);
  87   }
  88
  89   /* extract temperature */
  90   fscanf(fp, "%d", &temperature);
  91
  92   /* close temperature file */
  93   fclose(fp);
  94
  95   /* return temperature in degrees */
  96   return smprintf("%d°C", temperature / 1000);
  97 }
  98
  99 /* cpu percentage */
 100 char *
 101 cpu_usage()
 102 {
 103   int cpu_perc;
 104   long double a[4], b[4];
 105   FILE *fp;
 106
 107   /* open stat file */
 108   if (!(fp = fopen("/proc/stat","r"))) {
 109     fprintf(stderr, "Error opening stat file.");
 110     exit(1);
 111   }
 112
 113   /* read values */
 114   fscanf(fp, "%*s %Lf %Lf %Lf %Lf", &a[0], &a[1], &a[2], &a[3]);
 115
 116   /* close stat file */
 117   fclose(fp);
 118
 119   /* wait a second (for avg values) */
 120   sleep(1);
 121
 122   /* open stat file */
 123   if (!(fp = fopen("/proc/stat","r"))) {
 124     fprintf(stderr, "Error opening stat file.");
 125     exit(1);
 126   }
 127
 128   /* read values */
 129   fscanf(fp, "%*s %Lf %Lf %Lf %Lf", &b[0], &b[1], &b[2], &b[3]);
 130
 131   /* close stat file */
 132   fclose(fp);
 133
 134   /* calculate avg in this second */
 135   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]));
 136
 137   /* return cpu_perc as string */
 138   return smprintf("%d%%", cpu_perc);
 139 }
 140
 141 /* date and time */
 142 char *
 143 datetime()
 144 {
 145   time_t tm;
 146   size_t bufsize = 19;
 147   char *buf = malloc(bufsize);
 148
 149   /* get time in format */
 150   time(&tm);
 151   if(!strftime(buf, bufsize, timeformat, localtime(&tm))) {
 152     fprintf(stderr, "Strftime failed.\n");
 153     exit(1);
 154   }
 155
 156   /* return time */
 157   return buf;
 158 }
 159
 160 /* ram percentage */
 161 char *
 162 ram_usage()
 163 {
 164   int ram_perc;
 165   long total, free, available;
 166   FILE *fp;
 167
 168   /* open meminfo file */
 169   if (!(fp = fopen("/proc/meminfo", "r"))) {
 170     fprintf(stderr, "Error opening meminfo file.");
 171     exit(1);
 172   }
 173
 174   /* read the values */
 175   fscanf(fp, "MemTotal: %ld kB\n", &total);
 176   fscanf(fp, "MemFree: %ld kB\n", &free);
 177   fscanf(fp, "MemAvailable: %ld kB\n", &available);
 178
 179   /* close meminfo file */
 180   fclose(fp);
 181
 182   /* calculate percentage */
 183   ram_perc = 100 * (total - available) / total;
 184
 185   /* return ram_perc as string */
 186   return smprintf("%d%%",ram_perc);
 187 }
 188
 189 /* smprintf function */
 190 char *
 191 smprintf(char *fmt, ...)
 192 {
 193   va_list fmtargs;
 194   char *ret;
 195   int len;
 196
 197   va_start(fmtargs, fmt);
 198   len = vsnprintf(NULL, 0, fmt, fmtargs);
 199   va_end(fmtargs);
 200
 201   ret = malloc(++len);
 202   if (ret == NULL) {
 203     fprintf(stderr, "Malloc error.");
 204     exit(1);
 205   }
 206
 207   va_start(fmtargs, fmt);
 208   vsnprintf(ret, len, fmt, fmtargs);
 209   va_end(fmtargs);
 210
 211   return ret;
 212 }
 213
 214 /* alsa volume percentage */
 215 char *
 216 volume()
 217 {
 218   int mute = 0;
 219   long vol = 0, max = 0, min = 0;
 220
 221   /* get volume from alsa */
 222   snd_mixer_t *handle;
 223   snd_mixer_elem_t *pcm_mixer, *mas_mixer;
 224   snd_mixer_selem_id_t *vol_info, *mute_info;
 225   snd_mixer_open(&handle, 0);
 226   snd_mixer_attach(handle, soundcard);
 227   snd_mixer_selem_register(handle, NULL, NULL);
 228   snd_mixer_load(handle);
 229   snd_mixer_selem_id_malloc(&vol_info);
 230   snd_mixer_selem_id_malloc(&mute_info);
 231   snd_mixer_selem_id_set_name(vol_info, channel);
 232   snd_mixer_selem_id_set_name(mute_info, channel);
 233   pcm_mixer = snd_mixer_find_selem(handle, vol_info);
 234   mas_mixer = snd_mixer_find_selem(handle, mute_info);
 235   snd_mixer_selem_get_playback_volume_range((snd_mixer_elem_t *)pcm_mixer,
 236                                             &min, &max);
 237   snd_mixer_selem_get_playback_volume((snd_mixer_elem_t *)pcm_mixer,
 238                                       SND_MIXER_SCHN_MONO, &vol);
 239   snd_mixer_selem_get_playback_switch(mas_mixer, SND_MIXER_SCHN_MONO,
 240                                       &mute);
 241   if (vol_info)
 242     snd_mixer_selem_id_free(vol_info);
 243   if (mute_info)
 244     snd_mixer_selem_id_free(mute_info);
 245   if (handle)
 246     snd_mixer_close(handle);
 247
 248   /* return the string (mute) */
 249   if (!mute)
 250     return smprintf("mute");
 251   else
 252     return smprintf("%d%%", (vol * 100) / max);
 253 }
 254
 255 /* wifi percentage */
 256 char *
 257 wifi_signal()
 258 {
 259   int bufsize = 255;
 260   int strength;
 261   char buf[bufsize];
 262   char *datastart;
 263   char path_start[16] = "/sys/class/net/";
 264   char path_end[11] = "/operstate";
 265   char path[32];
 266   char status[5];
 267   char needle[sizeof wificard + 1];
 268   FILE *fp;
 269
 270   /* generate the path name */
 271
 272   memset(path, 0, sizeof path);
 273   strcat(path, path_start);
 274   strcat(path, wificard);
 275   strcat(path, path_end);
 276
 277   /* open wifi file */
 278   if(!(fp = fopen(path, "r"))) {
 279     fprintf(stderr, "Error opening wifi operstate file.");
 280     exit(1);
 281   }
 282
 283   /* read the status */
 284   fgets(status, 5, fp);
 285
 286   /* close wifi file */
 287   fclose(fp);
 288
 289   /* check if interface down */
 290   if(strcmp(status, "up\n") != 0){
 291     return "n/a";
 292   }
 293
 294   /* open wifi file */
 295   if (!(fp = fopen("/proc/net/wireless", "r"))) {
 296     fprintf(stderr, "Error opening wireless file.");
 297     exit(1);
 298   }
 299
 300   /* extract the signal strength */
 301   strcpy(needle, wificard);
 302   strcat(needle, ":");
 303   fgets(buf, bufsize, fp);
 304   fgets(buf, bufsize, fp);
 305   fgets(buf, bufsize, fp);
 306   if ((datastart = strstr(buf, needle)) != NULL) {
 307     datastart = strstr(buf, ":");
 308     sscanf(datastart + 1, " %*d   %d  %*d  %*d        %*d      %*d      %*d      %*d      %*d        %*d", &strength);
 309   }
 310
 311   /* close wifi file */
 312   fclose(fp);
 313
 314   /* return strength in percent */
 315   return smprintf("%d%%", strength);
 316 }
 317
 318 /* main function */
 319 int
 320 main()
 321 {
 322   char status[1024];
 323
 324   /* open display */
 325   if (!(dpy = XOpenDisplay(0x0))) {
 326     fprintf(stderr, "Cannot open display!\n");
 327     exit(1);
 328   }
 329
 330   char *pWifi_signal = NULL;
 331   char *pBattery = NULL;
 332   char *pCpu_usage = NULL;
 333   char *pCpu_temperature = NULL;
 334   char *pRam_usage = NULL;
 335   char *pVolume = NULL;
 336   char *pDatetime = NULL;
 337
 338   /* return status every second */
 339   for (;;) {
 340     pWifi_signal = wifi_signal();
 341     pBattery = battery();
 342     pCpu_usage = cpu_usage();
 343     pCpu_temperature = cpu_temperature();
 344     pRam_usage = ram_usage();
 345     pVolume = volume();
 346     pDatetime = datetime();
 347     sprintf(status, FORMATSTRING, ARGUMENTS);
 348     setstatus(status);
 349     free(pWifi_signal);
 350     free(pBattery);
 351     free(pCpu_usage);
 352     free(pCpu_temperature);
 353     free(pRam_usage);
 354     free(pVolume);
 355     free(pDatetime);
 356   }
 357
 358   /* close display */
 359   XCloseDisplay(dpy);
 360
 361   /* exit successfully */
 362   return 0;
 363 }