lhark/slstatus
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

check for fgets/fscanf return values

Browse source
This commit is contained in:
Aaron Marcher 2017-06-13 00:06:56 +02:00
parent 597cdc6b5c
commit 1c8aa5318f
No known key found for this signature in database
GPG key ID: 74B048E5C2474F9A
1 changed files with 75 additions and 35 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

110
slstatus.c
View file

@ -99,7 +99,7 @@ bprintf(const char *fmt, ...)
static const char * static const char *
battery_perc(const char *bat) battery_perc(const char *bat)
{ {
int perc; int n, perc;
char path[PATH_MAX]; char path[PATH_MAX];
FILE *fp; FILE *fp;
@ -109,8 +109,10 @@ battery_perc(const char *bat)
warn("Failed to open file %s", path); warn("Failed to open file %s", path);
return UNKNOWN_STR; return UNKNOWN_STR;
} }
fscanf(fp, "%i", &perc); n = fscanf(fp, "%i", &perc);
fclose(fp); fclose(fp);
if (n != 1)
return UNKNOWN_STR;
return bprintf("%d", perc); return bprintf("%d", perc);
} }
@ -120,7 +122,7 @@ battery_power(const char *bat)
{ {
char path[PATH_MAX]; char path[PATH_MAX];
FILE *fp; FILE *fp;
int watts; int n, watts;
snprintf(path, sizeof(path), "%s%s%s", "/sys/class/power_supply/", bat, "/power_now"); snprintf(path, sizeof(path), "%s%s%s", "/sys/class/power_supply/", bat, "/power_now");
fp = fopen(path, "r"); fp = fopen(path, "r");
@ -128,8 +130,10 @@ battery_power(const char *bat)
warn("Failed to open file %s", path); warn("Failed to open file %s", path);
return UNKNOWN_STR; return UNKNOWN_STR;
} }
fscanf(fp, "%i", &watts); n = fscanf(fp, "%i", &watts);
fclose(fp); fclose(fp);
if (n != 1)
return UNKNOWN_STR;
return bprintf("%d", (watts + 500000) / 1000000); return bprintf("%d", (watts + 500000) / 1000000);
} }
@ -140,6 +144,7 @@ battery_state(const char *bat)
char path[PATH_MAX]; char path[PATH_MAX];
char state[12]; char state[12];
FILE *fp; FILE *fp;
int n;
snprintf(path, sizeof(path), "%s%s%s", "/sys/class/power_supply/", bat, "/status"); snprintf(path, sizeof(path), "%s%s%s", "/sys/class/power_supply/", bat, "/status");
fp = fopen(path, "r"); fp = fopen(path, "r");
@ -147,8 +152,10 @@ battery_state(const char *bat)
warn("Failed to open file %s", path); warn("Failed to open file %s", path);
return UNKNOWN_STR; return UNKNOWN_STR;
} }
fscanf(fp, "%12s", state); n = fscanf(fp, "%12s", state);
fclose(fp); fclose(fp);
if (n != 1)
return UNKNOWN_STR;
if (strcmp(state, "Charging") == 0) { if (strcmp(state, "Charging") == 0) {
return "+"; return "+";
@ -166,7 +173,7 @@ battery_state(const char *bat)
static const char * static const char *
cpu_freq(void) cpu_freq(void)
{ {
int freq; int n, freq;
FILE *fp; FILE *fp;
fp = fopen("/sys/devices/system/cpu/cpu0/cpufreq/scaling_cur_freq", "r"); fp = fopen("/sys/devices/system/cpu/cpu0/cpufreq/scaling_cur_freq", "r");
@ -174,8 +181,10 @@ cpu_freq(void)
warn("Failed to open file /sys/devices/system/cpu/cpu0/cpufreq/scaling_cur_freq"); warn("Failed to open file /sys/devices/system/cpu/cpu0/cpufreq/scaling_cur_freq");
return UNKNOWN_STR; return UNKNOWN_STR;
} }
fscanf(fp, "%i", &freq); n = fscanf(fp, "%i", &freq);
fclose(fp); fclose(fp);
if (n != 1)
return UNKNOWN_STR;
return bprintf("%d", (freq + 500) / 1000); return bprintf("%d", (freq + 500) / 1000);
} }
@ -183,7 +192,7 @@ cpu_freq(void)
static const char * static const char *
cpu_perc(void) cpu_perc(void)
{ {
int perc; int n, perc;
long double a[4], b[4]; long double a[4], b[4];
FILE *fp; FILE *fp;
@ -192,8 +201,10 @@ cpu_perc(void)
warn("Failed to open file /proc/stat"); warn("Failed to open file /proc/stat");
return UNKNOWN_STR; return UNKNOWN_STR;
} }
fscanf(fp, "%*s %Lf %Lf %Lf %Lf", &a[0], &a[1], &a[2], &a[3]); n = fscanf(fp, "%*s %Lf %Lf %Lf %Lf", &a[0], &a[1], &a[2], &a[3]);
fclose(fp); fclose(fp);
if (n != 4)
return UNKNOWN_STR;
delay++; delay++;
sleep(delay); sleep(delay);
@ -203,8 +214,10 @@ cpu_perc(void)
warn("Failed to open file /proc/stat"); warn("Failed to open file /proc/stat");
return UNKNOWN_STR; return UNKNOWN_STR;
} }
fscanf(fp, "%*s %Lf %Lf %Lf %Lf", &b[0], &b[1], &b[2], &b[3]); n = fscanf(fp, "%*s %Lf %Lf %Lf %Lf", &b[0], &b[1], &b[2], &b[3]);
fclose(fp); fclose(fp);
if (n != 4)
return UNKNOWN_STR;
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])); 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 bprintf("%d", perc); return bprintf("%d", perc);
@ -280,7 +293,7 @@ disk_used(const char *mnt)
static const char * static const char *
entropy(void) entropy(void)
{ {
int num; int n, num;
FILE *fp; FILE *fp;
fp= fopen("/proc/sys/kernel/random/entropy_avail", "r"); fp= fopen("/proc/sys/kernel/random/entropy_avail", "r");
@ -288,8 +301,10 @@ entropy(void)
warn("Failed to open file /proc/sys/kernel/random/entropy_avail"); warn("Failed to open file /proc/sys/kernel/random/entropy_avail");
return UNKNOWN_STR; return UNKNOWN_STR;
} }
fscanf(fp, "%d", &num); n = fscanf(fp, "%d", &num);
fclose(fp); fclose(fp);
if (n != 1)
return UNKNOWN_STR;
return bprintf("%d", num); return bprintf("%d", num);
} }
@ -392,14 +407,17 @@ ram_free(void)
{ {
long free; long free;
FILE *fp; FILE *fp;
int n;
fp = fopen("/proc/meminfo", "r"); fp = fopen("/proc/meminfo", "r");
if (fp == NULL) { if (fp == NULL) {
warn("Failed to open file /proc/meminfo"); warn("Failed to open file /proc/meminfo");
return UNKNOWN_STR; return UNKNOWN_STR;
} }
fscanf(fp, "MemFree: %ld kB\n", &free); n = fscanf(fp, "MemFree: %ld kB\n", &free);
fclose(fp); fclose(fp);
if (n != 1)
return UNKNOWN_STR;
return bprintf("%f", (float)free / 1024 / 1024); return bprintf("%f", (float)free / 1024 / 1024);
} }
@ -415,13 +433,19 @@ ram_perc(void)
warn("Failed to open file /proc/meminfo"); warn("Failed to open file /proc/meminfo");
return UNKNOWN_STR; return UNKNOWN_STR;
} }
fscanf(fp, "MemTotal: %ld kB\n", &total); if (fscanf(fp, "MemTotal: %ld kB\n", &total) != 1 ||
fscanf(fp, "MemFree: %ld kB\n", &free); fscanf(fp, "MemFree: %ld kB\n", &free) != 1 ||
fscanf(fp, "MemAvailable: %ld kB\nBuffers: %ld kB\n", &buffers, &buffers); fscanf(fp, "MemAvailable: %ld kB\nBuffers: %ld kB\n",
fscanf(fp, "Cached: %ld kB\n", &cached); &buffers, &buffers) != 2 ||
fscanf(fp, "Cached: %ld kB\n", &cached) != 1)
goto scanerr;
fclose(fp); fclose(fp);
return bprintf("%d", 100 * ((total - free) - (buffers + cached)) / total); return bprintf("%d", 100 * ((total - free) - (buffers + cached)) / total);
scanerr:
fclose(fp);
return UNKNOWN_STR;
} }
static const char * static const char *
@ -429,14 +453,17 @@ ram_total(void)
{ {
long total; long total;
FILE *fp; FILE *fp;
int n;
fp = fopen("/proc/meminfo", "r"); fp = fopen("/proc/meminfo", "r");
if (fp == NULL) { if (fp == NULL) {
warn("Failed to open file /proc/meminfo"); warn("Failed to open file /proc/meminfo");
return UNKNOWN_STR; return UNKNOWN_STR;
} }
fscanf(fp, "MemTotal: %ld kB\n", &total); n = fscanf(fp, "MemTotal: %ld kB\n", &total);
fclose(fp); fclose(fp);
if (n != 1)
return UNKNOWN_STR;
return bprintf("%f", (float)total / 1024 / 1024); return bprintf("%f", (float)total / 1024 / 1024);
} }
@ -452,19 +479,25 @@ ram_used(void)
warn("Failed to open file /proc/meminfo"); warn("Failed to open file /proc/meminfo");
return UNKNOWN_STR; return UNKNOWN_STR;
} }
fscanf(fp, "MemTotal: %ld kB\n", &total); if (fscanf(fp, "MemTotal: %ld kB\n", &total) != 1 ||
fscanf(fp, "MemFree: %ld kB\n", &free); fscanf(fp, "MemFree: %ld kB\n", &free) != 1 ||
fscanf(fp, "MemAvailable: %ld kB\nBuffers: %ld kB\n", &buffers, &buffers); fscanf(fp, "MemAvailable: %ld kB\nBuffers: %ld kB\n",
fscanf(fp, "Cached: %ld kB\n", &cached); &buffers, &buffers) != 2 ||
fscanf(fp, "Cached: %ld kB\n", &cached) != 1)
goto scanerr;
fclose(fp); fclose(fp);
return bprintf("%f", (float)(total - free - buffers - cached) / 1024 / 1024); return bprintf("%f", (float)(total - free - buffers - cached) / 1024 / 1024);
scanerr:
fclose(fp);
return UNKNOWN_STR;
} }
static const char * static const char *
run_command(const char *cmd) run_command(const char *cmd)
{ {
char *nlptr; char *p;
FILE *fp; FILE *fp;
fp = popen(cmd, "r"); fp = popen(cmd, "r");
@ -472,10 +505,12 @@ run_command(const char *cmd)
warn("Failed to get command output for %s", cmd); warn("Failed to get command output for %s", cmd);
return UNKNOWN_STR; return UNKNOWN_STR;
} }
fgets(buf, sizeof(buf) - 1, fp); p = fgets(buf, sizeof(buf) - 1, fp);
pclose(fp); pclose(fp);
if ((nlptr = strrchr(buf, '\n')) != NULL) if (!p)
nlptr[0] = '\0'; return UNKNOWN_STR;
if ((p = strrchr(buf, '\n')) != NULL)
p[0] = '\0';
return buf[0] ? buf : UNKNOWN_STR; return buf[0] ? buf : UNKNOWN_STR;
} }
@ -613,7 +648,7 @@ swap_used(void)
static const char * static const char *
temp(const char *file) temp(const char *file)
{ {
int temp; int n, temp;
FILE *fp; FILE *fp;
fp = fopen(file, "r"); fp = fopen(file, "r");
@ -621,8 +656,10 @@ temp(const char *file)
warn("Failed to open file %s", file); warn("Failed to open file %s", file);
return UNKNOWN_STR; return UNKNOWN_STR;
} }
fscanf(fp, "%d", &temp); n = fscanf(fp, "%d", &temp);
fclose(fp); fclose(fp);
if (n != 1)
return UNKNOWN_STR;
return bprintf("%d", temp / 1000); return bprintf("%d", temp / 1000);
} }
@ -697,8 +734,8 @@ vol_perc(const char *card)
static const char * static const char *
wifi_perc(const char *iface) wifi_perc(const char *iface)
{ {
int perc; int i, perc;
char *datastart; char *p, *datastart;
char path[PATH_MAX]; char path[PATH_MAX];
char status[5]; char status[5];
FILE *fp; FILE *fp;
@ -709,9 +746,9 @@ wifi_perc(const char *iface)
warn("Failed to open file %s", path); warn("Failed to open file %s", path);
return UNKNOWN_STR; return UNKNOWN_STR;
} }
fgets(status, 5, fp); p = fgets(status, 5, fp);
fclose(fp); fclose(fp);
if(strcmp(status, "up\n") != 0) { if(!p || strcmp(status, "up\n") != 0) {
return UNKNOWN_STR; return UNKNOWN_STR;
} }
@ -721,10 +758,13 @@ wifi_perc(const char *iface)
return UNKNOWN_STR; return UNKNOWN_STR;
} }
fgets(buf, sizeof(buf) - 1, fp); for (i = 0; i < 3; i++) {
fgets(buf, sizeof(buf) - 1, fp); if (!(p = fgets(buf, sizeof(buf) - 1, fp)))
fgets(buf, sizeof(buf) - 1, fp); break;
}
fclose(fp); fclose(fp);
if (i < 2 || !p)
return UNKNOWN_STR;
if ((datastart = strstr(buf, iface)) == NULL) if ((datastart = strstr(buf, iface)) == NULL)
return UNKNOWN_STR; return UNKNOWN_STR;