Home | History | Annotate | Line # | Download | only in mktemp
mktemp.c revision 1.2 
      1 /* $NetBSD: mktemp.c,v 1.2 1999/09/21 06:24:46 cgd 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 #include <sys/cdefs.h>
     40 #include <err.h>
     41 #include <paths.h>
     42 #include <stdio.h>
     43 #include <stdlib.h>
     44 #include <string.h>
     45 #include <unistd.h>
     46 
     47 #ifndef lint
     48 #if 0
     49 static const char rcsid[] =
     50 	"$FreeBSD: src/usr.bin/mktemp/mktemp.c,v 1.2 1998/05/05 06:13:47 charnier Exp $";
     51 #else
     52 __RCSID("$NetBSD: mktemp.c,v 1.2 1999/09/21 06:24:46 cgd Exp $");
     53 #endif
     54 #endif /* not lint */
     55 
     56 static void usage __P((void));
     57 
     58 int
     59 main(int argc, char **argv)
     60 {
     61 	int c, fd, ret;
     62 	char *tmpdir, *prefix;
     63 	char *name;
     64 	int dflag, qflag, tflag, uflag;
     65 
     66 	ret = dflag = qflag = tflag = uflag = 0;
     67 	prefix = "mktemp";
     68 	name = NULL;
     69 
     70 	while ((c = getopt(argc, argv, "dqt:u")) != -1)
     71 		switch (c) {
     72 		case 'd':
     73 			dflag++;
     74 			break;
     75 
     76 		case 'q':
     77 			qflag++;
     78 			break;
     79 
     80 		case 't':
     81 			prefix = optarg;
     82 			tflag++;
     83 			break;
     84 
     85 		case 'u':
     86 			uflag++;
     87 			break;
     88 
     89 		default:
     90 			usage();
     91 		}
     92 
     93 	argc -= optind;
     94 	argv += optind;
     95 
     96 	if (tflag) {
     97 		tmpdir = getenv("TMPDIR");
     98 		if (tmpdir == NULL)
     99 			asprintf(&name, "%s%s.XXXXXXXX", _PATH_TMP, prefix);
    100 		else
    101 			asprintf(&name, "%s/%s.XXXXXXXX", tmpdir, prefix);
    102 		/* if this fails, the program is in big trouble already */
    103 		if (name == NULL) {
    104 			if (qflag)
    105 				return (1);
    106 			else
    107 				errx(1, "cannot generate template");
    108 		}
    109 	} else if (argc < 1) {
    110 		usage();
    111 	}
    112 
    113 	/* generate all requested files */
    114 	while (name != NULL || argc > 0) {
    115 		if (name == NULL) {
    116 			name = strdup(argv[0]);
    117 			argv++;
    118 			argc--;
    119 		}
    120 
    121 		if (dflag) {
    122 			if (mkdtemp(name) == NULL) {
    123 				ret = 1;
    124 				if (!qflag)
    125 					warn("mkdtemp failed on %s", name);
    126 			} else {
    127 				printf("%s\n", name);
    128 				if (uflag)
    129 					rmdir(name);
    130 			}
    131 		} else {
    132 			fd = mkstemp(name);
    133 			if (fd < 0) {
    134 				ret = 1;
    135 				if (!qflag)
    136 					warn("mkstemp failed on %s", name);
    137 			} else {
    138 				close(fd);
    139 				if (uflag)
    140 					unlink(name);
    141 				printf("%s\n", name);
    142 			}
    143 		}
    144 		if (name)
    145 			free(name);
    146 		name = NULL;
    147 	}
    148 	return (ret);
    149 }
    150 
    151 static void
    152 usage()
    153 {
    154 	fprintf(stderr,
    155 		"usage: mktemp [-d] [-q] [-t prefix] [-u] [template ...]\n");
    156 	exit (1);
    157 }
    158