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