diff options
Diffstat (limited to 'tempfile.c')
| -rw-r--r-- | tempfile.c | 161 |
1 files changed, 161 insertions, 0 deletions
diff --git a/tempfile.c b/tempfile.c new file mode 100644 index 0000000000..d82a3d3924 --- /dev/null +++ b/tempfile.c @@ -0,0 +1,161 @@ +/* tempfile.c + * Routines to create temporary files + * + * $Id$ + * + * Ethereal - Network traffic analyzer + * By Gerald Combs <gerald@ethereal.com> + * Copyright 1998 Gerald Combs + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. + */ + +#ifdef HAVE_CONFIG_H +# include "config.h" +#endif + +#include <glib.h> + +#include <stdlib.h> +#include <string.h> +#include <stdio.h> +#include <errno.h> + +#ifdef HAVE_UNISTD_H +#include <unistd.h> +#endif + +#include "file_util.h" + +#if GLIB_MAJOR_VERSION > 2 || (GLIB_MAJOR_VERSION == 2 && GLIB_MINOR_VERSION >= 6) +#include <glib/gstdio.h> /* available since GLib 2.6 only! */ + +/* GLib2.6 or above, using new wrapper functions */ +#define eth_mkstemp g_mkstemp +#else +#define eth_mkstemp mkstemp +#endif + +/* + * This has to come after the include of <pcap.h>, as the include of + * <pcap.h> might cause <winsock2.h> to be included, and if we've + * already included <winsock.h> as a result of including <windows.h>, + * we get a bunch of redefinitions. + */ +#ifdef HAVE_WINDOWS_H +#include <windows.h> +#endif + +#include "tempfile.h" + +static const char * +setup_tmpdir(const char *dir) +{ + size_t len = strlen(dir); + char *newdir; + + /* Append path separator if necessary */ + if (len != 0 && dir[len - 1] == G_DIR_SEPARATOR) { + return dir; + } + else { + newdir = g_malloc(len + 2); + strcpy(newdir, dir); + strcat(newdir, G_DIR_SEPARATOR_S); + return newdir; + } +} + +static int +try_tempfile(char *namebuf, int namebuflen, const char *dir, const char *pfx) +{ + static const char suffix[] = "XXXXXXXXXX"; + int namelen = strlen(dir) + strlen(pfx) + sizeof suffix; + int old_umask; + int tmp_fd; + + if (namebuflen < namelen) { + /* Stick in a truncated name, so that if this error is + reported with the file name, you at least get + something. */ + g_snprintf(namebuf, namebuflen, "%s%s%s", dir, pfx, suffix); + errno = ENAMETOOLONG; + return -1; + } + strcpy(namebuf, dir); + strcat(namebuf, pfx); + strcat(namebuf, suffix); + + /* The Single UNIX Specification doesn't say that "mkstemp()" + creates the temporary file with mode rw-------, so we + won't assume that all UNIXes will do so; instead, we set + the umask to 0077 to take away all group and other + permissions, attempt to create the file, and then put + the umask back. */ + old_umask = umask(0077); + tmp_fd = eth_mkstemp(namebuf); + umask(old_umask); + return tmp_fd; +} + +static const char *tmpdir = NULL; +#ifdef _WIN32 +static const char *temp = NULL; +#endif +static const char *E_tmpdir; + +#ifndef P_tmpdir +#define P_tmpdir "/var/tmp" +#endif + +int +create_tempfile(char *namebuf, int namebuflen, const char *pfx) +{ + char *dir; + int fd; + static gboolean initialized; + + if (!initialized) { + if ((dir = getenv("TMPDIR")) != NULL) + tmpdir = setup_tmpdir(dir); +#ifdef _WIN32 + if ((dir = getenv("TEMP")) != NULL) + temp = setup_tmpdir(dir); +#endif + + E_tmpdir = setup_tmpdir(P_tmpdir); + initialized = TRUE; + } + + if (tmpdir != NULL) { + fd = try_tempfile(namebuf, namebuflen, tmpdir, pfx); + if (fd != -1) + return fd; + } + +#ifdef _WIN32 + if (temp != NULL) { + fd = try_tempfile(namebuf, namebuflen, temp, pfx); + if (fd != -1) + return fd; + } +#endif + + fd = try_tempfile(namebuf, namebuflen, E_tmpdir, pfx); + if (fd != -1) + return fd; + + return try_tempfile(namebuf, namebuflen, G_DIR_SEPARATOR_S "tmp", pfx); +} |