Merge pull request #6 from Mic92/master

support for TMPDIR variable
This commit is contained in:
Luca Bruno 2016-08-29 11:14:07 +01:00 committed by GitHub
commit 809dda7f0a

12
main.c
View File

@ -48,7 +48,17 @@ int main(int argc, char *argv[]) {
usage(argv[0]);
}
char template[] = "/tmp/nixXXXXXX";
char *tmpdir = getenv("TMPDIR");
if (!tmpdir) {
tmpdir = "/tmp";
}
char template[PATH_MAX];
int needed = snprintf(template, PATH_MAX, "%s/nixXXXXXX", tmpdir);
if (needed < 0) {
err_exit("TMPDIR too long: '%s'", tmpdir);
}
char *rootdir = mkdtemp(template);
if (!rootdir) {
err_exit("mkdtemp(%s)", template);