X-Git-Url: https://git.ozlabs.org/?p=ccan;a=blobdiff_plain;f=tools%2Ftools.c;h=2da37234e2dd034ea8d5dd0f108eadcb8caf86bf;hp=d50a67fd18256b0ab3668018d84e0db429b85477;hb=3612661714e86333ceacca7314959a5ed938dc6a;hpb=61088f5c752c555172e2ab6cf93a7967f79f3f2c

diff --git a/tools/tools.c b/tools/tools.c
index d50a67fd..2da37234 100644
--- a/tools/tools.c
+++ b/tools/tools.c
@@ -1,11 +1,17 @@
 #include <ccan/talloc/talloc.h>
 #include <ccan/grab_file/grab_file.h>
+#include <sys/stat.h>
+#include <sys/types.h>
 #include <string.h>
 #include <unistd.h>
 #include <stdarg.h>
 #include <errno.h>
+#include <err.h>
 #include "tools.h"
 
+static char *tmpdir = NULL;
+static unsigned int count;
+
 char *talloc_basename(const void *ctx, const char *dir)
 {
 	char *p = strrchr(dir, '/');
@@ -42,6 +48,7 @@ char *talloc_getcwd(const void *ctx)
 	return cwd;
 }
 
+/* Returns output if command fails. */
 char *run_command(const void *ctx, const char *fmt, ...)
 {
 	va_list ap;
@@ -67,3 +74,35 @@ char *run_command(const void *ctx, const char *fmt, ...)
 	talloc_free(cmd);
 	return NULL;
 }
+
+static int unlink_all(char *dir)
+{
+	char cmd[strlen(dir) + sizeof("rm -rf ")];
+	sprintf(cmd, "rm -rf %s", dir);
+//	system(cmd);
+	return 0;
+}
+
+char *temp_file(const void *ctx, const char *extension)
+{
+	/* For first call, create dir. */
+	while (!tmpdir) {
+		tmpdir = getenv("TMPDIR");
+		if (!tmpdir)
+			tmpdir = "/tmp";
+		tmpdir = talloc_asprintf(talloc_autofree_context(),
+					 "%s/ccanlint-%u.%lu",
+					 tmpdir, getpid(), random());
+		if (mkdir(tmpdir, 0700) != 0) {
+			if (errno == EEXIST) {
+				talloc_free(tmpdir);
+				tmpdir = NULL;
+				continue;
+			}
+			err(1, "mkdir %s failed", tmpdir);
+		}
+		talloc_set_destructor(tmpdir, unlink_all);
+	}
+
+	return talloc_asprintf(ctx, "%s/%u%s", tmpdir, count++, extension);
+}