summaryrefslogtreecommitdiff
path: root/src/lib9/tempdir_unix.c
blob: 3ce87751b2faed3a47c6c67ff5d612fa903df596 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
// Copyright 2013 The Go Authors.  All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

// +build darwin dragonfly freebsd linux netbsd openbsd

#include <u.h>
#include <dirent.h>
#include <sys/stat.h>
#define NOPLAN9DEFINES
#include <libc.h>

char*
mktempdir(void)
{
	char *tmp, *p;
	
	tmp = getenv("TMPDIR");
	if(tmp == nil || strlen(tmp) == 0)
		tmp = "/var/tmp";
	p = smprint("%s/go-link-XXXXXX", tmp);
	if(mkdtemp(p) == nil)
		return nil;
	return p;
}

void
removeall(char *p)
{
	DIR *d;
	struct dirent *dp;
	char *q;
	struct stat st;

	if(stat(p, &st) < 0)
		return;
	if(!S_ISDIR(st.st_mode)) {
		unlink(p);
		return;
	}

	d = opendir(p);
	while((dp = readdir(d)) != nil) {
		if(strcmp(dp->d_name, ".") == 0 || strcmp(dp->d_name, "..") == 0)
			continue;
		q = smprint("%s/%s", p, dp->d_name);
		removeall(q);
		free(q);
	}
	closedir(d);
	rmdir(p);
}