1*0Sstevel@tonic-gate /* 2*0Sstevel@tonic-gate * CDDL HEADER START 3*0Sstevel@tonic-gate * 4*0Sstevel@tonic-gate * The contents of this file are subject to the terms of the 5*0Sstevel@tonic-gate * Common Development and Distribution License, Version 1.0 only 6*0Sstevel@tonic-gate * (the "License"). You may not use this file except in compliance 7*0Sstevel@tonic-gate * with the License. 8*0Sstevel@tonic-gate * 9*0Sstevel@tonic-gate * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE 10*0Sstevel@tonic-gate * or http://www.opensolaris.org/os/licensing. 11*0Sstevel@tonic-gate * See the License for the specific language governing permissions 12*0Sstevel@tonic-gate * and limitations under the License. 13*0Sstevel@tonic-gate * 14*0Sstevel@tonic-gate * When distributing Covered Code, include this CDDL HEADER in each 15*0Sstevel@tonic-gate * file and include the License file at usr/src/OPENSOLARIS.LICENSE. 16*0Sstevel@tonic-gate * If applicable, add the following below this CDDL HEADER, with the 17*0Sstevel@tonic-gate * fields enclosed by brackets "[]" replaced with your own identifying 18*0Sstevel@tonic-gate * information: Portions Copyright [yyyy] [name of copyright owner] 19*0Sstevel@tonic-gate * 20*0Sstevel@tonic-gate * CDDL HEADER END 21*0Sstevel@tonic-gate */ 22*0Sstevel@tonic-gate /* 23*0Sstevel@tonic-gate * Copyright 2004 Sun Microsystems, Inc. All rights reserved. 24*0Sstevel@tonic-gate * Use is subject to license terms. 25*0Sstevel@tonic-gate * 26*0Sstevel@tonic-gate * Create unique plain files or directories. 27*0Sstevel@tonic-gate */ 28*0Sstevel@tonic-gate 29*0Sstevel@tonic-gate #pragma ident "%Z%%M% %I% %E% SMI" 30*0Sstevel@tonic-gate 31*0Sstevel@tonic-gate #include <stdio.h> 32*0Sstevel@tonic-gate #include <stdlib.h> 33*0Sstevel@tonic-gate #include <unistd.h> 34*0Sstevel@tonic-gate #include <errno.h> 35*0Sstevel@tonic-gate #include <string.h> 36*0Sstevel@tonic-gate #include <locale.h> 37*0Sstevel@tonic-gate #include <sys/stat.h> 38*0Sstevel@tonic-gate 39*0Sstevel@tonic-gate static void 40*0Sstevel@tonic-gate usage(void) 41*0Sstevel@tonic-gate { 42*0Sstevel@tonic-gate (void) fprintf(stderr, 43*0Sstevel@tonic-gate gettext("Usage: mktemp [-dqtu] [-p prefix_dir] [template]\n")); 44*0Sstevel@tonic-gate exit(1); 45*0Sstevel@tonic-gate /* NOTREACHED */ 46*0Sstevel@tonic-gate } 47*0Sstevel@tonic-gate 48*0Sstevel@tonic-gate static int 49*0Sstevel@tonic-gate mktmpdir(char *tmpl) 50*0Sstevel@tonic-gate { 51*0Sstevel@tonic-gate char *tx = strdup(tmpl); 52*0Sstevel@tonic-gate int i; 53*0Sstevel@tonic-gate 54*0Sstevel@tonic-gate if (tx == NULL) { 55*0Sstevel@tonic-gate perror("strdup"); 56*0Sstevel@tonic-gate exit(1); 57*0Sstevel@tonic-gate } 58*0Sstevel@tonic-gate for (i = 0; i < 100; i++) { 59*0Sstevel@tonic-gate if (*mktemp(tmpl) == '\0') 60*0Sstevel@tonic-gate goto errout; 61*0Sstevel@tonic-gate 62*0Sstevel@tonic-gate if (mkdir(tmpl, 0700) == 0) { 63*0Sstevel@tonic-gate free(tx); 64*0Sstevel@tonic-gate return (0); 65*0Sstevel@tonic-gate } 66*0Sstevel@tonic-gate 67*0Sstevel@tonic-gate /* All other errors won't be fixed by trying a different name */ 68*0Sstevel@tonic-gate if (errno != EEXIST) 69*0Sstevel@tonic-gate goto errout; 70*0Sstevel@tonic-gate 71*0Sstevel@tonic-gate /* Restore template for next try */ 72*0Sstevel@tonic-gate (void) strcpy(tmpl, tx); 73*0Sstevel@tonic-gate } 74*0Sstevel@tonic-gate errout: 75*0Sstevel@tonic-gate (void) strcpy(tmpl, tx); 76*0Sstevel@tonic-gate free(tx); 77*0Sstevel@tonic-gate return (-1); 78*0Sstevel@tonic-gate } 79*0Sstevel@tonic-gate 80*0Sstevel@tonic-gate int 81*0Sstevel@tonic-gate main(int argc, char **argv) 82*0Sstevel@tonic-gate { 83*0Sstevel@tonic-gate int opt; 84*0Sstevel@tonic-gate char *prefix = NULL; 85*0Sstevel@tonic-gate boolean_t dounlink = B_FALSE; 86*0Sstevel@tonic-gate boolean_t domkdir = B_FALSE; 87*0Sstevel@tonic-gate boolean_t quiet = B_FALSE; 88*0Sstevel@tonic-gate boolean_t usetmpdir = B_FALSE; 89*0Sstevel@tonic-gate char template[] = "tmp.XXXXXX"; 90*0Sstevel@tonic-gate char *tmpl; 91*0Sstevel@tonic-gate 92*0Sstevel@tonic-gate (void) setlocale(LC_ALL, ""); 93*0Sstevel@tonic-gate 94*0Sstevel@tonic-gate #ifndef TEXT_DOMAIN 95*0Sstevel@tonic-gate #define TEXT_DOMAIN "SYS_TEST" 96*0Sstevel@tonic-gate #endif 97*0Sstevel@tonic-gate (void) textdomain(TEXT_DOMAIN); 98*0Sstevel@tonic-gate 99*0Sstevel@tonic-gate opterr = 0; 100*0Sstevel@tonic-gate 101*0Sstevel@tonic-gate while ((opt = getopt(argc, argv, "dqtup:")) != EOF) { 102*0Sstevel@tonic-gate switch (opt) { 103*0Sstevel@tonic-gate case 'd': 104*0Sstevel@tonic-gate domkdir = B_TRUE; 105*0Sstevel@tonic-gate break; 106*0Sstevel@tonic-gate case 'q': 107*0Sstevel@tonic-gate quiet = B_TRUE; 108*0Sstevel@tonic-gate break; 109*0Sstevel@tonic-gate case 'p': 110*0Sstevel@tonic-gate prefix = optarg; 111*0Sstevel@tonic-gate /* FALLTHROUGH - -p implies -t */ 112*0Sstevel@tonic-gate case 't': 113*0Sstevel@tonic-gate usetmpdir = B_TRUE; 114*0Sstevel@tonic-gate break; 115*0Sstevel@tonic-gate case 'u': 116*0Sstevel@tonic-gate dounlink = B_TRUE; 117*0Sstevel@tonic-gate break; 118*0Sstevel@tonic-gate default: 119*0Sstevel@tonic-gate usage(); 120*0Sstevel@tonic-gate } 121*0Sstevel@tonic-gate } 122*0Sstevel@tonic-gate argc -= optind; 123*0Sstevel@tonic-gate argv += optind; 124*0Sstevel@tonic-gate switch (argc) { 125*0Sstevel@tonic-gate case 0: 126*0Sstevel@tonic-gate tmpl = template; 127*0Sstevel@tonic-gate usetmpdir = B_TRUE; 128*0Sstevel@tonic-gate break; 129*0Sstevel@tonic-gate case 1: 130*0Sstevel@tonic-gate tmpl = argv[0]; 131*0Sstevel@tonic-gate break; 132*0Sstevel@tonic-gate default: 133*0Sstevel@tonic-gate usage(); 134*0Sstevel@tonic-gate } 135*0Sstevel@tonic-gate 136*0Sstevel@tonic-gate if (usetmpdir) { 137*0Sstevel@tonic-gate char *tmp = getenv("TMPDIR"); 138*0Sstevel@tonic-gate size_t len; 139*0Sstevel@tonic-gate 140*0Sstevel@tonic-gate if (strchr(tmpl, '/') != NULL) { 141*0Sstevel@tonic-gate (void) fprintf(stderr, 142*0Sstevel@tonic-gate gettext("mktemp: template argument specified " 143*0Sstevel@tonic-gate "with -t/-p option must not contain '/'" 144*0Sstevel@tonic-gate "\n")); 145*0Sstevel@tonic-gate return (1); 146*0Sstevel@tonic-gate } 147*0Sstevel@tonic-gate /* TMPDIR overrides -p so that scripts will honor $TMPDIR */ 148*0Sstevel@tonic-gate if (tmp != NULL) 149*0Sstevel@tonic-gate prefix = tmp; 150*0Sstevel@tonic-gate else if (prefix == NULL) 151*0Sstevel@tonic-gate prefix = "/tmp"; 152*0Sstevel@tonic-gate 153*0Sstevel@tonic-gate len = snprintf(NULL, 0, "%s/%s", prefix, tmpl) + 1; 154*0Sstevel@tonic-gate tmp = malloc(len); 155*0Sstevel@tonic-gate if (tmp == NULL) { 156*0Sstevel@tonic-gate perror("malloc"); 157*0Sstevel@tonic-gate return (1); 158*0Sstevel@tonic-gate } 159*0Sstevel@tonic-gate (void) snprintf(tmp, len, "%s/%s", prefix, tmpl); 160*0Sstevel@tonic-gate tmpl = tmp; 161*0Sstevel@tonic-gate } 162*0Sstevel@tonic-gate 163*0Sstevel@tonic-gate if (domkdir) { 164*0Sstevel@tonic-gate if (mktmpdir(tmpl) != 0) { 165*0Sstevel@tonic-gate if (!quiet) { 166*0Sstevel@tonic-gate (void) fprintf(stderr, 167*0Sstevel@tonic-gate gettext("mktemp: failed to create " 168*0Sstevel@tonic-gate "directory: %s\n"), tmpl); 169*0Sstevel@tonic-gate } 170*0Sstevel@tonic-gate return (1); 171*0Sstevel@tonic-gate } 172*0Sstevel@tonic-gate if (dounlink) 173*0Sstevel@tonic-gate (void) rmdir(tmpl); 174*0Sstevel@tonic-gate } else { 175*0Sstevel@tonic-gate if (mkstemp(tmpl) < 0) { 176*0Sstevel@tonic-gate if (!quiet) { 177*0Sstevel@tonic-gate (void) fprintf(stderr, 178*0Sstevel@tonic-gate gettext("mktemp: failed to create file: " 179*0Sstevel@tonic-gate "%s\n"), tmpl); 180*0Sstevel@tonic-gate } 181*0Sstevel@tonic-gate return (1); 182*0Sstevel@tonic-gate } 183*0Sstevel@tonic-gate if (dounlink) 184*0Sstevel@tonic-gate (void) unlink(tmpl); 185*0Sstevel@tonic-gate } 186*0Sstevel@tonic-gate (void) puts(tmpl); 187*0Sstevel@tonic-gate return (0); 188*0Sstevel@tonic-gate } 189