Add env to execute function.

1 files changed, 48 insertions, 0 deletions

diff --git a/test/tmpfile.h b/test/tmpfile.h
new file mode 100644
index 0000000..5d114d0
--- /dev/null
+++ b/test/tmpfile.h
@@ -0,0 +1,48 @@
+// -*- c++ -*-
+// Distributed under the BSD 2-Clause License.
+// See accompanying file LICENSE for details.
+#pragma once
+
+#include <cstdlib>
+#include <unistd.h>
+
+#ifdef _WIN32
+#define WIN32_LEAN_AND_MEAN
+#include <windows.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#endif
+
+struct tmp_file
+{
+	tmp_file(const std::string& data = {})
+	{
+		int fd;
+#ifdef _WIN32
+		char templ[] = "ctor_tmp_file-XXXXXX"; // buffer for filename
+		_mktemp_s(templ, sizeof(templ));
+		fd = open(templ, O_CREAT | O_RDWR);
+#else
+		char templ[] = "/tmp/ctor_tmp_file-XXXXXX"; // buffer for filename
+		fd = mkstemp(templ);
+#endif
+		filename = templ;
+		auto sz = write(fd, data.data(), data.size());
+		(void)sz;
+		close(fd);
+	}
+
+	~tmp_file()
+	{
+		unlink(filename.data());
+	}
+
+	const std::string& get() const
+	{
+		return filename;
+	}
+
+private:
+	std::string filename;
+};