122 lines
3.5 KiB
C
122 lines
3.5 KiB
C
/* tempfile.c
|
|
* Routines to create temporary files
|
|
*
|
|
* Wireshark - Network traffic analyzer
|
|
* By Gerald Combs <gerald@wireshark.org>
|
|
* Copyright 1998 Gerald Combs
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <errno.h>
|
|
|
|
#include "tempfile.h"
|
|
#include "file_util.h"
|
|
|
|
/**
|
|
* Create a tempfile with the given prefix (e.g. "wireshark"). The path
|
|
* is created using g_file_open_tmp.
|
|
*
|
|
* @param tempdir [in] If not NULL, the directory in which to create the file.
|
|
* @param namebuf [in,out] If not NULL, receives the full path of the temp file.
|
|
* Must be freed.
|
|
* @param pfx [in] A prefix for the temporary file.
|
|
* @param sfx [in] A file extension for the temporary file. NULL can be passed
|
|
* if no file extension is needed
|
|
* @param err [out] Any error returned by g_file_open_tmp. May be NULL
|
|
* @return The file descriptor of the new tempfile, from mkstemps().
|
|
*/
|
|
int
|
|
create_tempfile(const char *tempdir, gchar **namebuf, const char *pfx, const char *sfx, GError **err)
|
|
{
|
|
int fd;
|
|
gchar *safe_pfx = NULL;
|
|
|
|
if (pfx) {
|
|
/* The characters in "delimiters" come from:
|
|
* https://docs.microsoft.com/en-us/windows/win32/fileio/naming-a-file#naming-conventions.
|
|
* Add to the list as necessary for other OS's.
|
|
*/
|
|
const gchar *delimiters = "<>:\"/\\|?*"
|
|
"\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a"
|
|
"\x0b\x0c\x0d\x0e\x0f\x10\x11\x12\x13\x14"
|
|
"\x15\x16\x17\x18\x19\x1a\x1b\x1c\x1d\x1e\x1f";
|
|
|
|
/* Sanitize the pfx to resolve bug 7877 */
|
|
safe_pfx = g_strdup(pfx);
|
|
safe_pfx = g_strdelimit(safe_pfx, delimiters, '-');
|
|
}
|
|
|
|
if (tempdir == NULL || tempdir[0] == '\0') {
|
|
/* Use OS default tempdir behaviour */
|
|
gchar* filetmpl = ws_strdup_printf("%sXXXXXX%s", safe_pfx ? safe_pfx : "", sfx ? sfx : "");
|
|
g_free(safe_pfx);
|
|
|
|
fd = g_file_open_tmp(filetmpl, namebuf, err);
|
|
g_free(filetmpl);
|
|
}
|
|
else {
|
|
/* User-specified tempdir.
|
|
* We don't get libc's help generating a random name here.
|
|
*/
|
|
const gchar alphabet[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789_-";
|
|
const gint32 a_len = 64;
|
|
gchar* filetmpl = NULL;
|
|
|
|
while(1) {
|
|
g_free(filetmpl);
|
|
filetmpl = ws_strdup_printf("%s%c%s%c%c%c%c%c%c%s",
|
|
tempdir,
|
|
G_DIR_SEPARATOR,
|
|
safe_pfx ? safe_pfx : "",
|
|
alphabet[g_random_int_range(0, a_len)],
|
|
alphabet[g_random_int_range(0, a_len)],
|
|
alphabet[g_random_int_range(0, a_len)],
|
|
alphabet[g_random_int_range(0, a_len)],
|
|
alphabet[g_random_int_range(0, a_len)],
|
|
alphabet[g_random_int_range(0, a_len)],
|
|
sfx ? sfx : "");
|
|
|
|
fd = ws_open(filetmpl, O_CREAT|O_EXCL|O_BINARY|O_WRONLY, 0600);
|
|
if (fd >= 0) {
|
|
break;
|
|
}
|
|
if (errno != EEXIST) {
|
|
g_set_error_literal(err, G_FILE_ERROR,
|
|
g_file_error_from_errno(errno), g_strerror(errno));
|
|
g_free(filetmpl);
|
|
filetmpl = NULL;
|
|
break;
|
|
}
|
|
/* Loop continues if error was EEXIST, meaning the file we tried
|
|
* to make already existed at the destination
|
|
*/
|
|
}
|
|
|
|
if (namebuf == NULL) {
|
|
g_free(filetmpl);
|
|
}
|
|
else {
|
|
*namebuf = filetmpl;
|
|
}
|
|
g_free(safe_pfx);
|
|
}
|
|
|
|
return fd;
|
|
}
|
|
|
|
/*
|
|
* Editor modelines - https://www.wireshark.org/tools/modelines.html
|
|
*
|
|
* Local Variables:
|
|
* c-basic-offset: 2
|
|
* tab-width: 8
|
|
* indent-tabs-mode: nil
|
|
* End:
|
|
*
|
|
* ex: set shiftwidth=2 tabstop=8 expandtab:
|
|
* :indentSize=2:tabSize=8:noTabs=true:
|
|
*/
|