OpenSolaris Launch

2 files changed, 231 insertions, 0 deletions

diff --git a/usr/src/cmd/mktemp/Makefile b/usr/src/cmd/mktemp/Makefile
new file mode 100644
index 0000000000..4e9ddcd47b
--- /dev/null
+++ b/usr/src/cmd/mktemp/Makefile
@@ -0,0 +1,43 @@
+#
+# CDDL HEADER START
+#
+# The contents of this file are subject to the terms of the
+# Common Development and Distribution License, Version 1.0 only
+# (the "License").  You may not use this file except in compliance
+# with the License.
+#
+# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
+# or http://www.opensolaris.org/os/licensing.
+# See the License for the specific language governing permissions
+# and limitations under the License.
+#
+# When distributing Covered Code, include this CDDL HEADER in each
+# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
+# If applicable, add the following below this CDDL HEADER, with the
+# fields enclosed by brackets "[]" replaced with your own identifying
+# information: Portions Copyright [yyyy] [name of copyright owner]
+#
+# CDDL HEADER END
+#
+#
+# ident	"%Z%%M%	%I%	%E% SMI"
+#
+# Copyright 2004 Sun Microsystems, Inc.  All rights reserved.
+# Use is subject to license terms.
+#
+
+PROG= mktemp
+
+include ../Makefile.cmd
+
+.KEEP_STATE:
+
+all: $(PROG) 
+
+install: all $(ROOTPROG)
+
+clean:
+
+lint:	lint_PROG
+
+include ../Makefile.targ
diff --git a/usr/src/cmd/mktemp/mktemp.c b/usr/src/cmd/mktemp/mktemp.c
new file mode 100644
index 0000000000..d7982b892b
--- /dev/null
+++ b/usr/src/cmd/mktemp/mktemp.c
@@ -0,0 +1,188 @@
+/*
+ * CDDL HEADER START
+ *
+ * The contents of this file are subject to the terms of the
+ * Common Development and Distribution License, Version 1.0 only
+ * (the "License").  You may not use this file except in compliance
+ * with the License.
+ *
+ * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
+ * or http://www.opensolaris.org/os/licensing.
+ * See the License for the specific language governing permissions
+ * and limitations under the License.
+ *
+ * When distributing Covered Code, include this CDDL HEADER in each
+ * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
+ * If applicable, add the following below this CDDL HEADER, with the
+ * fields enclosed by brackets "[]" replaced with your own identifying
+ * information: Portions Copyright [yyyy] [name of copyright owner]
+ *
+ * CDDL HEADER END
+ */
+/*
+ * Copyright 2004 Sun Microsystems, Inc.  All rights reserved.
+ * Use is subject to license terms.
+ *
+ * Create unique plain files or directories.
+ */
+
+#pragma ident	"%Z%%M%	%I%	%E% SMI"
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <errno.h>
+#include <string.h>
+#include <locale.h>
+#include <sys/stat.h>
+
+static void
+usage(void)
+{
+	(void) fprintf(stderr,
+	    gettext("Usage: mktemp [-dqtu] [-p prefix_dir] [template]\n"));
+	exit(1);
+	/* NOTREACHED */
+}
+
+static int
+mktmpdir(char *tmpl)
+{
+	char *tx = strdup(tmpl);
+	int i;
+
+	if (tx == NULL) {
+		perror("strdup");
+		exit(1);
+	}
+	for (i = 0; i < 100; i++) {
+		if (*mktemp(tmpl) == '\0')
+			goto errout;
+
+		if (mkdir(tmpl, 0700) == 0) {
+			free(tx);
+			return (0);
+		}
+
+		/* All other errors won't be fixed by trying a different name */
+		if (errno != EEXIST)
+			goto errout;
+
+		/* Restore template for next try */
+		(void) strcpy(tmpl, tx);
+	}
+errout:
+	(void) strcpy(tmpl, tx);
+	free(tx);
+	return (-1);
+}
+
+int
+main(int argc, char **argv)
+{
+	int opt;
+	char *prefix = NULL;
+	boolean_t dounlink = B_FALSE;
+	boolean_t domkdir = B_FALSE;
+	boolean_t quiet = B_FALSE;
+	boolean_t usetmpdir = B_FALSE;
+	char template[] = "tmp.XXXXXX";
+	char *tmpl;
+
+	(void) setlocale(LC_ALL, "");
+
+#ifndef TEXT_DOMAIN
+#define	TEXT_DOMAIN "SYS_TEST"
+#endif
+	(void) textdomain(TEXT_DOMAIN);
+
+	opterr = 0;
+
+	while ((opt = getopt(argc, argv, "dqtup:")) != EOF) {
+		switch (opt) {
+		case 'd':
+			domkdir = B_TRUE;
+			break;
+		case 'q':
+			quiet = B_TRUE;
+			break;
+		case 'p':
+			prefix = optarg;
+			/* FALLTHROUGH - -p implies -t */
+		case 't':
+			usetmpdir = B_TRUE;
+			break;
+		case 'u':
+			dounlink = B_TRUE;
+			break;
+		default:
+			usage();
+		}
+	}
+	argc -= optind;
+	argv += optind;
+	switch (argc) {
+	case 0:
+		tmpl = template;
+		usetmpdir = B_TRUE;
+		break;
+	case 1:
+		tmpl = argv[0];
+		break;
+	default:
+		usage();
+	}
+
+	if (usetmpdir) {
+		char *tmp = getenv("TMPDIR");
+		size_t len;
+
+		if (strchr(tmpl, '/') != NULL) {
+			(void) fprintf(stderr,
+			    gettext("mktemp: template argument specified "
+			    "with -t/-p option must not contain '/'"
+			    "\n"));
+			return (1);
+		}
+		/* TMPDIR overrides -p so that scripts will honor $TMPDIR */
+		if (tmp != NULL)
+			prefix = tmp;
+		else if (prefix == NULL)
+			prefix = "/tmp";
+
+		len = snprintf(NULL, 0, "%s/%s", prefix, tmpl) + 1;
+		tmp = malloc(len);
+		if (tmp == NULL) {
+			perror("malloc");
+			return (1);
+		}
+		(void) snprintf(tmp, len, "%s/%s", prefix, tmpl);
+		tmpl = tmp;
+	}
+
+	if (domkdir) {
+		if (mktmpdir(tmpl) != 0) {
+			if (!quiet) {
+				(void) fprintf(stderr,
+				    gettext("mktemp: failed to create "
+				    "directory: %s\n"), tmpl);
+			}
+			return (1);
+		}
+		if (dounlink)
+			(void) rmdir(tmpl);
+	} else {
+		if (mkstemp(tmpl) < 0) {
+			if (!quiet) {
+				(void) fprintf(stderr,
+				    gettext("mktemp: failed to create file: "
+				    "%s\n"), tmpl);
+			}
+			return (1);
+		}
+		if (dounlink)
+			(void) unlink(tmpl);
+	}
+	(void) puts(tmpl);
+	return (0);
+}