Browse Source

Implement esnprintf() and make formatted calls more efficient

Within the components, snprintf() was unchecked and had inefficient
calls in some places.
We implement esnprintf() that does all the dirty laundry for us
and use it exclusively now.
master
Laslo Hunhold 8 years ago
committed by Aaron Marcher
parent
commit
422cadfd5f
  1. 41
      components/battery.c
  2. 14
      components/netspeeds.c
  3. 12
      components/wifi.c
  4. 6
      slstatus.c
  5. 21
      util.c
  6. 1
      util.h

41
components/battery.c

@ -14,9 +14,16 @@
int perc; int perc;
char path[PATH_MAX]; char path[PATH_MAX];
snprintf(path, sizeof(path), "%s%s%s", "/sys/class/power_supply/", if (esnprintf(path, sizeof(path),
bat, "/capacity"); "/sys/class/power_supply/%s/capacity",
return (pscanf(path, "%d", &perc) == 1) ? bprintf("%d", perc) : NULL; bat) < 0) {
return NULL;
}
if (pscanf(path, "%d", &perc) != 1) {
return NULL;
}
return bprintf("%d", perc);
} }
const char * const char *
@ -32,8 +39,11 @@
size_t i; size_t i;
char path[PATH_MAX], state[12]; char path[PATH_MAX], state[12];
snprintf(path, sizeof(path), "%s%s%s", "/sys/class/power_supply/", if (esnprintf(path, sizeof(path),
bat, "/status"); "/sys/class/power_supply/%s/status",
bat) < 0) {
return NULL;
}
if (pscanf(path, "%12s", state) != 1) { if (pscanf(path, "%12s", state) != 1) {
return NULL; return NULL;
} }
@ -53,20 +63,29 @@
float timeleft; float timeleft;
char path[PATH_MAX], state[12]; char path[PATH_MAX], state[12];
snprintf(path, sizeof(path), "%s%s%s", "/sys/class/power_supply/", if (esnprintf(path, sizeof(path),
bat, "/status"); "/sys/class/power_supply/%s/status",
bat) < 0) {
return NULL;
}
if (pscanf(path, "%12s", state) != 1) { if (pscanf(path, "%12s", state) != 1) {
return NULL; return NULL;
} }
if (!strcmp(state, "Discharging")) { if (!strcmp(state, "Discharging")) {
snprintf(path, sizeof(path), "%s%s%s", "/sys/class/power_supply/", if (esnprintf(path, sizeof(path),
bat, "/charge_now"); "/sys/class/power_supply/%s/charge_now",
bat) < 0) {
return NULL;
}
if (pscanf(path, "%d", &charge_now) != 1) { if (pscanf(path, "%d", &charge_now) != 1) {
return NULL; return NULL;
} }
snprintf(path, sizeof(path), "%s%s%s", "/sys/class/power_supply/", if (esnprintf(path, sizeof(path),
bat, "/current_now"); "/sys/class/power_supply/%s/current_now",
bat) < 0) {
return NULL;
}
if (pscanf(path, "%d", &current_now) != 1) { if (pscanf(path, "%d", &current_now) != 1) {
return NULL; return NULL;
} }

14
components/netspeeds.c

@ -17,8 +17,11 @@
oldrxbytes = rxbytes; oldrxbytes = rxbytes;
snprintf(path, sizeof(path), if (esnprintf(path, sizeof(path),
"/sys/class/net/%s/statistics/rx_bytes", interface); "/sys/class/net/%s/statistics/rx_bytes",
interface) < 0) {
return NULL;
}
if (pscanf(path, "%llu", &rxbytes) != 1) { if (pscanf(path, "%llu", &rxbytes) != 1) {
return NULL; return NULL;
} }
@ -37,8 +40,11 @@
oldtxbytes = txbytes; oldtxbytes = txbytes;
snprintf(path, sizeof(path), if (esnprintf(path, sizeof(path),
"/sys/class/net/%s/statistics/tx_bytes", interface); "/sys/class/net/%s/statistics/tx_bytes",
interface) < 0) {
return NULL;
}
if (pscanf(path, "%llu", &txbytes) != 1) { if (pscanf(path, "%llu", &txbytes) != 1) {
return NULL; return NULL;
} }

12
components/wifi.c

@ -23,8 +23,11 @@
char status[5]; char status[5];
FILE *fp; FILE *fp;
snprintf(path, sizeof(path), "%s%s%s", "/sys/class/net/", iface, if (esnprintf(path, sizeof(path),
"/operstate"); "/sys/class/net/%s/operstate",
iface) < 0) {
return NULL;
}
if (!(fp = fopen(path, "r"))) { if (!(fp = fopen(path, "r"))) {
warn("fopen '%s':", path); warn("fopen '%s':", path);
return NULL; return NULL;
@ -69,7 +72,10 @@
memset(&wreq, 0, sizeof(struct iwreq)); memset(&wreq, 0, sizeof(struct iwreq));
wreq.u.essid.length = IW_ESSID_MAX_SIZE+1; wreq.u.essid.length = IW_ESSID_MAX_SIZE+1;
snprintf(wreq.ifr_name, sizeof(wreq.ifr_name), "%s", iface); if (esnprintf(wreq.ifr_name, sizeof(wreq.ifr_name),
"%s", iface) < 0) {
return NULL;
}
if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) { if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
warn("socket 'AF_INET':"); warn("socket 'AF_INET':");

6
slstatus.c

@ -91,12 +91,8 @@ main(int argc, char *argv[])
if (!(res = args[i].func(args[i].args))) { if (!(res = args[i].func(args[i].args))) {
res = unknown_str; res = unknown_str;
} }
if ((ret = snprintf(status + len, sizeof(status) - len, if ((ret = esnprintf(status + len, sizeof(status) - len,
args[i].fmt, res)) < 0) { args[i].fmt, res)) < 0) {
warn("snprintf:");
break;
} else if ((size_t)ret >= sizeof(status) - len) {
warn("snprintf: Output truncated");
break; break;
} }
len += ret; len += ret;

21
util.c

@ -48,6 +48,27 @@ die(const char *fmt, ...)
exit(1); exit(1);
} }
int
esnprintf(char *str, size_t size, const char *fmt, ...)
{
va_list ap;
int ret;
va_start(ap, fmt);
ret = vsnprintf(str, size, fmt, ap);
va_end(ap);
if (ret < 0) {
warn("snprintf:");
return -1;
} else if ((size_t)ret >= size) {
warn("snprintf: Output truncated");
return -1;
}
return ret;
}
const char * const char *
bprintf(const char *fmt, ...) bprintf(const char *fmt, ...)
{ {

1
util.h

@ -8,6 +8,7 @@ extern char *argv0;
void warn(const char *, ...); void warn(const char *, ...);
void die(const char *, ...); void die(const char *, ...);
int esnprintf(char *str, size_t size, const char *fmt, ...);
const char *bprintf(const char *fmt, ...); const char *bprintf(const char *fmt, ...);
const char *fmt_scaled(size_t); const char *fmt_scaled(size_t);
int pscanf(const char *path, const char *fmt, ...); int pscanf(const char *path, const char *fmt, ...);

Loading…
Cancel
Save