changeset 575:953c22158c2e

Addition of mktemp
author Elie De Brauwer <eliedebrauwer@gmail.com>
date Tue, 01 May 2012 11:45:45 +0200
parents d8effa95a5f0
children 4058eacd4fbc
files toys/mktemp.c
diffstat 1 files changed, 56 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/toys/mktemp.c	Tue May 01 11:45:45 2012 +0200
@@ -0,0 +1,56 @@
+/* vi: set sw=4 ts=4:
+ *
+ * mktemp.c - Create a temporary file or directory.
+ *
+ * Copyright 2012 Elie De Brauwer <eliedebrauwer@gmail.com>
+ *
+ * Not in SUSv4.
+
+USE_MKTEMP(NEWTOY(mktemp, ">1(directory)d(tmpdir)p:", TOYFLAG_BIN))
+
+config MKTEMP
+	bool "mktemp"
+	default y
+	help
+	  usage: mktemp [OPTION] [TEMPLATE]
+
+	  Safely create a temporary file or directory and print its name.
+	  TEMPLATE should end in 6 consecutive X's, the default
+	  template is tmp.XXXXXX and the default directory is /tmp/.
+	  -d, --directory        Create a directory, instead of a file
+	  -p DIR, --tmpdir=DIR   Use DIR as a base path
+
+*/
+
+#include "toys.h"
+
+DEFINE_GLOBALS(
+	char * tmpdir;
+)
+#define TT this.mktemp
+
+void mktemp_main(void)
+{
+	int  p_flag = (toys.optflags & 1);
+	int  d_flag = (toys.optflags & 2) >> 1;
+	char * result;
+
+	int size = snprintf(toybuf, sizeof(toybuf)-1, "%s/%s",
+			(p_flag && TT.tmpdir)?TT.tmpdir:"/tmp/",
+			(toys.optargs[0])?toys.optargs[0]:"tmp.XXXXXX");
+	toybuf[size] = 0;
+
+	if (d_flag) {
+		if (mkdtemp(toybuf) == NULL)
+			perror_exit("Failed to create temporary directory");
+	} else {
+		if (mkstemp(toybuf) == -1)
+			perror_exit("Failed to create temporary file");
+	}
+
+	result = realpath(toybuf, NULL);
+	xputs(result);
+
+	if (CFG_TOYBOX_FREE)
+		free(result);
+}