Home | History | Annotate | Line # | Download | only in mktemp
mktemp.c revision 1.11.12.1 
      1 /* $NetBSD: mktemp.c,v 1.11.12.1 2012/11/20 03:02:58 tls Exp $ */
      2 
      3 /*-
      4  * Copyright (c) 1994, 1995, 1996, 1998 Peter Wemm <peter (at) netplex.com.au>
      5  * All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
     17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
     20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     26  * SUCH DAMAGE.
     27  *
     28  */
     29 
     30 /*
     31  * This program was originally written long ago, originally for a non
     32  * BSD-like OS without mkstemp().  It's been modified over the years
     33  * to use mkstemp() rather than the original O_CREAT|O_EXCL/fstat/lstat
     34  * etc style hacks.
     35  * A cleanup, misc options and mkdtemp() calls were added to try and work
     36  * more like the OpenBSD version - which was first to publish the interface.
     37  */
     38 
     39 #if HAVE_NBTOOL_CONFIG_H
     40 #include "nbtool_config.h"
     41 #endif
     42 
     43 #include <sys/cdefs.h>
     44 #include <sys/types.h>
     45 #include <err.h>
     46 #include <paths.h>
     47 #include <stdio.h>
     48 #include <stdlib.h>
     49 #include <string.h>
     50 #include <unistd.h>
     51 
     52 #if defined(__RCSID) && !defined(__lint)
     53 __RCSID("$NetBSD: mktemp.c,v 1.11.12.1 2012/11/20 03:02:58 tls Exp $");
     54 #endif /* !__lint */
     55 
     56 static void usage(void) __dead;
     57 
     58 int
     59 main(int argc, char **argv)
     60 {
     61 	int c, fd, ret;
     62 	char *tmpdir;
     63 	const char *prefix;
     64 	char *name;
     65 	int dflag, qflag, tflag, uflag;
     66 
     67 	setprogname(*argv);
     68 	ret = dflag = qflag = tflag = uflag = 0;
     69 	tmpdir = NULL;
     70 	prefix = "mktemp";
     71 	name = NULL;
     72 
     73 	while ((c = getopt(argc, argv, "dp:qt:u")) != -1)
     74 		switch (c) {
     75 		case 'd':
     76 			dflag++;
     77 			break;
     78 
     79 		case 'p':
     80 			tmpdir = optarg;
     81 			break;
     82 
     83 		case 'q':
     84 			qflag++;
     85 			break;
     86 
     87 		case 't':
     88 			prefix = optarg;
     89 			tflag++;
     90 			break;
     91 
     92 		case 'u':
     93 			uflag++;
     94 			break;
     95 
     96 		default:
     97 			usage();
     98 		}
     99 
    100 	argc -= optind;
    101 	argv += optind;
    102 
    103 	if (tflag == 0 && argc < 1)
    104 		tflag = 1;
    105 
    106 	if (tflag) {
    107 		if (tmpdir == NULL)
    108 			tmpdir = getenv("TMPDIR");
    109 		if (tmpdir == NULL)
    110 			(void)asprintf(&name, "%s%s.XXXXXXXX", _PATH_TMP,
    111 			    prefix);
    112 		else
    113 			(void)asprintf(&name, "%s/%s.XXXXXXXX", tmpdir, prefix);
    114 		/* if this fails, the program is in big trouble already */
    115 		if (name == NULL) {
    116 			if (qflag)
    117 				return 1;
    118 			else
    119 				errx(1, "Cannot generate template");
    120 		}
    121 	} else if (argc < 1) {
    122 		usage();
    123 	}
    124 
    125 	/* generate all requested files */
    126 	while (name != NULL || argc > 0) {
    127 		if (name == NULL) {
    128 			if (tmpdir)
    129 				(void)asprintf(&name, "%s/%s",
    130 				    tmpdir, argv[0]);
    131 			else
    132 				name = strdup(argv[0]);
    133 			argv++;
    134 			argc--;
    135 		}
    136 
    137 		if (dflag) {
    138 			if (mkdtemp(name) == NULL) {
    139 				ret = 1;
    140 				if (!qflag)
    141 					warn("mkdtemp failed on %s", name);
    142 			} else {
    143 				(void)printf("%s\n", name);
    144 				if (uflag)
    145 					(void)rmdir(name);
    146 			}
    147 		} else {
    148 			fd = mkstemp(name);
    149 			if (fd < 0) {
    150 				ret = 1;
    151 				if (!qflag)
    152 					warn("mkstemp failed on %s", name);
    153 			} else {
    154 				(void)close(fd);
    155 				if (uflag)
    156 					(void)unlink(name);
    157 				(void)printf("%s\n", name);
    158 			}
    159 		}
    160 		if (name)
    161 			free(name);
    162 		name = NULL;
    163 	}
    164 	return ret;
    165 }
    166 
    167 static void
    168 usage(void)
    169 {
    170 	(void)fprintf(stderr,
    171 	    "Usage: %s [-dqu] [-p <tmpdir>] {-t prefix | template ...}\n",
    172 	    getprogname());
    173 	exit (1);
    174 }
    175