contrib/gcc/choose-temp.c

   1 /* Utility to pick a temporary filename prefix.
   2    Copyright (C) 1996, 1997, 1998 Free Software Foundation, Inc.
   3
   4 This file is part of the libiberty library.
   5 Libiberty is free software; you can redistribute it and/or
   6 modify it under the terms of the GNU Library General Public
   7 License as published by the Free Software Foundation; either
   8 version 2 of the License, or (at your option) any later version.
   9
  10 Libiberty is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13 Library General Public License for more details.
  14
  15 You should have received a copy of the GNU Library General Public
  16 License along with libiberty; see the file COPYING.LIB.  If not,
  17 write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18 Boston, MA 02111-1307, USA.  */
  19
  20 /* This file exports two functions: choose_temp_base and make_temp_file.  */
  21
  22 /* This file lives in at least two places: libiberty and gcc.
  23    Don't change one without the other.  */
  24
  25 /* $FreeBSD: src/contrib/gcc/choose-temp.c,v 1.3 1999/11/04 10:23:25 obrien Exp $ */
  26 /* $DragonFly: src/contrib/gcc/Attic/choose-temp.c,v 1.2 2003/06/17 04:23:59 dillon Exp $ */
  27
  28 #ifdef HAVE_CONFIG_H
  29 #include "config.h"
  30 #endif
  31
  32 #include <stdio.h>      /* May get P_tmpdir.  */
  33 #include <sys/types.h>
  34 #ifdef HAVE_UNISTD_H
  35 #include <unistd.h>
  36 #endif
  37 #ifdef HAVE_STDLIB_H
  38 #include <stdlib.h>
  39 #endif
  40 #ifdef HAVE_SYS_FILE_H
  41 #include <sys/file.h>   /* May get R_OK, etc. on some systems.  */
  42 #endif
  43
  44 #ifndef R_OK
  45 #define R_OK 4
  46 #define W_OK 2
  47 #define X_OK 1
  48 #endif
  49
  50 #include "libiberty.h"
  51 extern int mkstemps ();
  52
  53 #ifndef IN_GCC
  54 #if defined (__MSDOS__) || (defined (_WIN32) && ! defined (__CYGWIN__) && ! defined (_UWIN))
  55 #define DIR_SEPARATOR '\\'
  56 #endif
  57 #endif
  58
  59 #ifndef DIR_SEPARATOR
  60 #define DIR_SEPARATOR '/'
  61 #endif
  62
  63 /* On MSDOS, write temp files in current dir
  64    because there's no place else we can expect to use.  */
  65 /* ??? Although the current directory is tried as a last resort,
  66    this is left in so that on MSDOS it is preferred to /tmp on the
  67    off chance that someone requires this, since that was the previous
  68    behaviour.  */
  69 #ifdef __MSDOS__
  70 #ifndef P_tmpdir
  71 #define P_tmpdir "."
  72 #endif
  73 #endif
  74
  75 /* Name of temporary file.
  76    mktemp requires 6 trailing X's.  */
  77 #define TEMP_FILE "ccXXXXXX"
  78
  79 /* Subroutine of choose_temp_base.
  80    If BASE is non-NULL, return it.
  81    Otherwise it checks if DIR is a usable directory.
  82    If success, DIR is returned.
  83    Otherwise NULL is returned.  */
  84
  85 static char *
  86 try (dir, base)
  87      char *dir, *base;
  88 {
  89   if (base != 0)
  90     return base;
  91   if (dir != 0
  92       && access (dir, R_OK | W_OK | X_OK) == 0)
  93     return dir;
  94   return 0;
  95 }
  96
  97 /* Return a prefix for temporary file names or NULL if unable to find one.
  98    The current directory is chosen if all else fails so the program is
  99    exited if a temporary directory can't be found (mktemp fails).
 100    The buffer for the result is obtained with xmalloc.
 101
 102    This function is provided for backwards compatability only.  It use
 103    is not recommended.  */
 104
 105 char *
 106 choose_temp_base ()
 107 {
 108   char *base = 0;
 109   char *temp_filename;
 110   int len;
 111   static char tmp[] = { DIR_SEPARATOR, 't', 'm', 'p', 0 };
 112   static char usrtmp[] = { DIR_SEPARATOR, 'u', 's', 'r', DIR_SEPARATOR, 't', 'm', 'p', 0 };
 113
 114   base = try (getenv ("TMPDIR"), base);
 115   base = try (getenv ("TMP"), base);
 116   base = try (getenv ("TEMP"), base);
 117
 118   base = try (tmp, base);
 119
 120 #ifdef P_tmpdir
 121   base = try (P_tmpdir, base);
 122 #endif
 123
 124   /* Try /usr/tmp even though it usually doesn't exist on FreeBSD.  */
 125   base = try (usrtmp, base);
 126
 127   /* If all else fails, use the current directory!  */
 128   if (base == 0)
 129     base = ".";
 130
 131   len = strlen (base);
 132   temp_filename = xmalloc (len + 1 /*DIR_SEPARATOR*/
 133                            + strlen (TEMP_FILE) + 1);
 134   strcpy (temp_filename, base);
 135
 136   if (len != 0
 137       && temp_filename[len-1] != '/'
 138       && temp_filename[len-1] != DIR_SEPARATOR)
 139     temp_filename[len++] = DIR_SEPARATOR;
 140   strcpy (temp_filename + len, TEMP_FILE);
 141
 142   mktemp (temp_filename);
 143   if (strlen (temp_filename) == 0)
 144     abort ();
 145   return temp_filename;
 146 }
 147 /* Return a temporary file name (as a string) or NULL if unable to create
 148    one.  */
 149
 150 char *
 151 make_temp_file (suffix)
 152      char *suffix;
 153 {
 154   char *base = 0;
 155   char *temp_filename;
 156   int base_len, suffix_len;
 157   int fd;
 158   static char tmp[] = { DIR_SEPARATOR, 't', 'm', 'p', 0 };
 159   static char usrtmp[] = { DIR_SEPARATOR, 'u', 's', 'r', DIR_SEPARATOR, 't', 'm', 'p', 0 };
 160
 161   base = try (getenv ("TMPDIR"), base);
 162   base = try (getenv ("TMP"), base);
 163   base = try (getenv ("TEMP"), base);
 164
 165   base = try (tmp, base);
 166
 167 #ifdef P_tmpdir
 168   base = try (P_tmpdir, base);
 169 #endif
 170
 171   /* Try /usr/tmp even though it usually doesn't exist on FreeBSD.  */
 172   base = try (usrtmp, base);
 173
 174   /* If all else fails, use the current directory!  */
 175   if (base == 0)
 176     base = ".";
 177
 178   base_len = strlen (base);
 179
 180   if (suffix)
 181     suffix_len = strlen (suffix);
 182   else
 183     suffix_len = 0;
 184
 185   temp_filename = xmalloc (base_len + 1 /*DIR_SEPARATOR*/
 186                            + strlen (TEMP_FILE)
 187                            + suffix_len + 1);
 188   strcpy (temp_filename, base);
 189
 190   if (base_len != 0
 191       && temp_filename[base_len-1] != '/'
 192       && temp_filename[base_len-1] != DIR_SEPARATOR)
 193     temp_filename[base_len++] = DIR_SEPARATOR;
 194   strcpy (temp_filename + base_len, TEMP_FILE);
 195
 196   if (suffix)
 197     strcat (temp_filename, suffix);
 198
 199   fd = mkstemps (temp_filename, suffix_len);
 200   /* If mkstemps failed, then something bad is happening.  Maybe we should
 201      issue a message about a possible security attack in progress?  */
 202   if (fd == -1)
 203     abort ();
 204   /* Similarly if we can not close the file.  */
 205   if (close (fd))
 206     abort ();
 207   return temp_filename;
 208 }