Implement os_mkdtemp on top of uv_fs_mkdtemp

This commit is contained in:
Pavel Platto
2014-08-05 16:54:32 +03:00
committed by Thiago de Arruda
parent 19f44fda8b
commit 5e42b406a5
2 changed files with 28 additions and 29 deletions

View File

@@ -3,13 +3,6 @@
#include <assert.h>
// TODO(hinidu): remove after implementing `os_mkdtemp` on top of libuv
#ifdef WIN32
# include <io.h>
#else
# include <stdlib.h>
#endif
#include "nvim/os/os.h"
#include "nvim/ascii.h"
#include "nvim/memory.h"
@@ -293,18 +286,21 @@ int os_mkdir(const char *path, int32_t mode)
}
/// Create a unique temporary directory.
/// TODO(hinidu): Implement on top of libuv. ref #850
///
/// @param[in,out] template Template of the path to the directory with XXXXXX
/// @param[in] template Template of the path to the directory with XXXXXX
/// which would be replaced by random chars.
/// @return Pointer to changed `template` for success, `NULL` for failure.
char *os_mkdtemp(char *template)
/// @param[out] path Path to created directory for success, undefined for
/// failure.
/// @return `0` for success, non-zero for failure.
int os_mkdtemp(const char *template, char *path)
{
#ifdef WIN32
return _mktemp(template) && os_mkdir(template, 0700) == 0 ? template : NULL;
#else
return mkdtemp(template);
#endif
uv_fs_t request;
int result = uv_fs_mkdtemp(uv_default_loop(), &request, template, NULL);
if (result == kLibuvSuccess) {
strcpy(path, request.path);
}
uv_fs_req_cleanup(&request);
return result;
}
/// Remove a directory.

View File

@@ -28,26 +28,29 @@ static void vim_maketempdir(void)
{
static const char *temp_dirs[] = TEMP_DIR_NAMES;
// Try the entries in `TEMP_DIR_NAMES` to create the temp directory.
char_u itmp[TEMP_FILE_PATH_MAXLEN];
char_u template[TEMP_FILE_PATH_MAXLEN];
char_u path[TEMP_FILE_PATH_MAXLEN];
for (size_t i = 0; i < sizeof(temp_dirs) / sizeof(char *); ++i) {
// Expand environment variables, leave room for "/nvimXXXXXX/999999999"
expand_env((char_u *)temp_dirs[i], itmp, TEMP_FILE_PATH_MAXLEN - 22);
if (!os_isdir(itmp)) { // directory doesn't exist
expand_env((char_u *)temp_dirs[i], template, TEMP_FILE_PATH_MAXLEN - 22);
if (!os_isdir(template)) { // directory doesn't exist
continue;
}
add_pathsep(itmp);
add_pathsep(template);
// Concatenate with temporary directory name pattern
STRCAT(itmp, "nvimXXXXXX");
if (!os_mkdtemp((char *)itmp)) {
STRCAT(template, "nvimXXXXXX");
if (os_mkdtemp((const char *)template, (char *)path) != 0) {
continue;
}
if (vim_settempdir(itmp)) {
if (vim_settempdir(path)) {
// Successfully created and set temporary directory so stop trying.
break;
} else {
// Couldn't set `vim_tempdir` to itmp so remove created directory.
os_rmdir((char *)itmp);
// Couldn't set `vim_tempdir` to `path` so remove created directory.
os_rmdir((char *)path);
}
}
}
@@ -128,8 +131,8 @@ char_u *vim_tempname(void)
// There is no need to check if the file exists, because we own the directory
// and nobody else creates a file in it.
char_u itmp[TEMP_FILE_PATH_MAXLEN];
snprintf((char *)itmp, TEMP_FILE_PATH_MAXLEN,
char_u template[TEMP_FILE_PATH_MAXLEN];
snprintf((char *)template, TEMP_FILE_PATH_MAXLEN,
"%s%" PRIu32, tempdir, temp_count++);
return vim_strsave(itmp);
return vim_strsave(template);
}