mirror of
https://github.com/neovim/neovim.git
synced 2025-10-06 09:56:31 +00:00
Implement os_mkdtemp
on top of uv_fs_mkdtemp
This commit is contained in:

committed by
Thiago de Arruda

parent
19f44fda8b
commit
5e42b406a5
@@ -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.
|
||||
|
@@ -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);
|
||||
}
|
||||
|
Reference in New Issue
Block a user