Skip to content

Commit 2cfa300

Browse files
authored
Merge pull request #248 from DrDaveD/envar_null
Null terminate envar result, and don't lose track of malloc
2 parents c034a4a + f52c9fa commit 2cfa300

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/util/util.c

+3-1
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@
4343

4444

4545
char *envar(char *name, char *allowed, int len) {
46-
char *ret = (char *) malloc(len);
46+
char *ret;
4747
char *env = getenv(name); // Flawfinder: ignore
4848
int count;
4949

@@ -62,6 +62,7 @@ char *envar(char *name, char *allowed, int len) {
6262
}
6363

6464
singularity_message(DEBUG, "Checking environment variable has allowed characters: %s\n", name);
65+
ret = (char *) malloc(len+1);
6566
for(count=0; count <= len && env[count] != '\0'; count++) {
6667
int test_char = env[count];
6768
int c, success = 0;
@@ -81,6 +82,7 @@ char *envar(char *name, char *allowed, int len) {
8182
}
8283
ret[count] = test_char;
8384
}
85+
ret[count] = '\0';
8486

8587
singularity_message(VERBOSE2, "Obtained input from environment '%s' = '%s'\n", name, ret);
8688
return(ret);

0 commit comments

Comments
 (0)