Check for allocation failure

Pointed out by Michael Reed.
This commit is contained in:
Michael McConville 2016-01-02 12:20:40 -05:00
parent ae5c01f485
commit c1b3289219
1 changed files with 2 additions and 1 deletions

View File

@ -139,7 +139,8 @@ char *OpenBSDProcessList_readProcessName(kvm_t* kd, struct kinfo_proc* kproc, in
for (i = 0; arg[i] != NULL; i++) {
len += strlen(arg[i]) + 1;
}
buf = s = malloc(len);
if ((buf = s = malloc(len)) == NULL)
err(1, NULL);
for (i = 0; arg[i] != NULL; i++) {
n = strlcpy(buf, arg[i], (s + len) - buf);
buf += n;