1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00

os: re-add the leak in os.join_path (the os.join_path(x, ...arr) case should be handled by V). Add a memleak free os.join_path_single version.

This commit is contained in:
Delyan Angelov 2021-11-22 14:40:55 +02:00
parent 5e1782bf9c
commit 7a0dc60d04
No known key found for this signature in database
GPG Key ID: 66886C0F12D595ED
7 changed files with 60 additions and 39 deletions

View File

@ -41,8 +41,8 @@ const (
)
const (
tfolder = os.join_path(os.temp_dir(), 'os_file_test')
tfile = os.join_path(tfolder, 'test_file')
tfolder = os.join_path_single(os.temp_dir(), 'os_file_test')
tfile = os.join_path_single(tfolder, 'test_file')
)
fn testsuite_begin() ? {

View File

@ -191,7 +191,7 @@ pub fn file_size(path string) u64 {
pub fn mv(src string, dst string) ? {
mut rdst := dst
if is_dir(rdst) {
rdst = join_path(rdst.trim_right(path_separator), file_name(src.trim_right(path_separator)))
rdst = join_path_single(rdst.trim_right(path_separator), file_name(src.trim_right(path_separator)))
}
$if windows {
w_src := src.replace('/', '\\')

View File

@ -40,7 +40,8 @@ pub fn cp_all(src string, dst string, overwrite bool) ? {
// single file copy
if !is_dir(source_path) {
adjusted_path := if is_dir(dest_path) {
join_path(dest_path, file_name(source_path))
fname := file_name(source_path)
join_path_single(dest_path, fname)
} else {
dest_path
}
@ -62,8 +63,8 @@ pub fn cp_all(src string, dst string, overwrite bool) ? {
}
files := ls(source_path) ?
for file in files {
sp := join_path(source_path, file)
dp := join_path(dest_path, file)
sp := join_path_single(source_path, file)
dp := join_path_single(dest_path, file)
if is_dir(sp) {
if !exists(dp) {
mkdir(dp) ?
@ -136,7 +137,7 @@ pub fn rmdir_all(path string) ? {
mut ret_err := ''
items := ls(path) ?
for item in items {
fullpath := join_path(path, item)
fullpath := join_path_single(path, item)
if is_dir(fullpath) && !is_link(fullpath) {
rmdir_all(fullpath) or { ret_err = err.msg }
} else {
@ -379,7 +380,7 @@ fn executable_fallback() string {
if !is_abs_path(exepath) {
rexepath := exepath.replace_each(['/', path_separator, r'\', path_separator])
if rexepath.contains(path_separator) {
exepath = join_path(os.wd_at_startup, exepath)
exepath = join_path_single(os.wd_at_startup, exepath)
} else {
// no choice but to try to walk the PATH folders :-| ...
foundpath := find_abs_path_of_executable(exepath) or { '' }
@ -414,7 +415,7 @@ pub fn find_abs_path_of_executable(exepath string) ?string {
path := getenv('PATH')
paths := path.split(path_delimiter)
for p in paths {
found_abs_path := join_path(p, exepath)
found_abs_path := join_path_single(p, exepath)
if exists(found_abs_path) && is_executable(found_abs_path) {
res = found_abs_path
break
@ -452,9 +453,8 @@ pub fn is_abs_path(path string) bool {
// join_path returns a path as string from input string parameter(s).
[manualfree]
pub fn join_path(base string, dirs ...string) string {
defer {
unsafe { dirs.free() }
}
// TODO: fix freeing of `dirs` when the passed arguments are variadic,
// but do not free the arr, when `os.join_path(base, ...arr)` is called.
mut sb := strings.new_builder(base.len + dirs.len * 50)
defer {
unsafe { sb.free() }
@ -471,6 +471,26 @@ pub fn join_path(base string, dirs ...string) string {
return sb.str()
}
// join_path_single appends the `elem` after `base`, using a platform specific
// path_separator.
[manualfree]
pub fn join_path_single(base string, elem string) string {
// TODO: deprecate this and make it `return os.join_path(base, elem)`,
// when freeing variadic args vs ...arr is solved in the compiler
mut sb := strings.new_builder(base.len + elem.len + 1)
defer {
unsafe { sb.free() }
}
sbase := base.trim_right('\\/')
defer {
unsafe { sbase.free() }
}
sb.write_string(sbase)
sb.write_string(path_separator)
sb.write_string(elem)
return sb.str()
}
// walk_ext returns a recursive list of all files in `path` ending with `ext`.
pub fn walk_ext(path string, ext string) []string {
mut res := []string{}
@ -587,7 +607,7 @@ pub fn cache_dir() string {
return xdg_cache_home
}
}
cdir := join_path(home_dir(), '.cache')
cdir := join_path_single(home_dir(), '.cache')
if !is_dir(cdir) && !is_link(cdir) {
mkdir(cdir) or { panic(err) }
}
@ -629,7 +649,7 @@ pub fn temp_dir() string {
fn default_vmodules_path() string {
hdir := home_dir()
res := join_path(hdir, '.vmodules')
res := join_path_single(hdir, '.vmodules')
return res
}
@ -684,7 +704,7 @@ pub fn resource_abs_path(path string) string {
unsafe { base_path.free() }
base_path = vresource
}
fp := join_path(base_path, path)
fp := join_path_single(base_path, path)
res := real_path(fp)
unsafe {
fp.free()

View File

@ -480,7 +480,7 @@ pub fn is_writable_folder(folder string) ?bool {
if !is_dir(folder) {
return error('`folder` is not a folder')
}
tmp_perm_check := join_path(folder, 'XXXXXX')
tmp_perm_check := join_path_single(folder, 'XXXXXX')
defer {
unsafe { tmp_perm_check.free() }
}

View File

@ -106,9 +106,9 @@ fn test_create_file() ? {
fn test_is_file() {
// Setup
work_dir := os.join_path(os.getwd(), 'is_file_test')
work_dir := os.join_path_single(os.getwd(), 'is_file_test')
os.mkdir_all(work_dir) or { panic(err) }
tfile := os.join_path(work_dir, 'tmp_file')
tfile := os.join_path_single(work_dir, 'tmp_file')
// Test things that shouldn't be a file
assert os.is_file(work_dir) == false
assert os.is_file('non-existent_file.tmp') == false
@ -120,7 +120,7 @@ fn test_is_file() {
$if windows {
assert true
} $else {
dsymlink := os.join_path(work_dir, 'dir_symlink')
dsymlink := os.join_path_single(work_dir, 'dir_symlink')
os.symlink(work_dir, dsymlink) or { panic(err) }
assert os.is_file(dsymlink) == false
}
@ -128,7 +128,7 @@ fn test_is_file() {
$if windows {
assert true
} $else {
fsymlink := os.join_path(work_dir, 'file_symlink')
fsymlink := os.join_path_single(work_dir, 'file_symlink')
os.symlink(tfile, fsymlink) or { panic(err) }
assert os.is_file(fsymlink)
}
@ -309,47 +309,47 @@ fn test_cp() {
}
fn test_mv() {
work_dir := os.join_path(os.getwd(), 'mv_test')
work_dir := os.join_path_single(os.getwd(), 'mv_test')
os.mkdir_all(work_dir) or { panic(err) }
// Setup test files
tfile1 := os.join_path(work_dir, 'file')
tfile2 := os.join_path(work_dir, 'file.test')
tfile3 := os.join_path(work_dir, 'file.3')
tfile1 := os.join_path_single(work_dir, 'file')
tfile2 := os.join_path_single(work_dir, 'file.test')
tfile3 := os.join_path_single(work_dir, 'file.3')
tfile_content := 'temporary file'
os.write_file(tfile1, tfile_content) or { panic(err) }
os.write_file(tfile2, tfile_content) or { panic(err) }
// Setup test dirs
tdir1 := os.join_path(work_dir, 'dir')
tdir2 := os.join_path(work_dir, 'dir2')
tdir3 := os.join_path(work_dir, 'dir3')
tdir1 := os.join_path_single(work_dir, 'dir')
tdir2 := os.join_path_single(work_dir, 'dir2')
tdir3 := os.join_path_single(work_dir, 'dir3')
os.mkdir(tdir1) or { panic(err) }
os.mkdir(tdir2) or { panic(err) }
// Move file with no extension to dir
os.mv(tfile1, tdir1) or { panic(err) }
mut expected := os.join_path(tdir1, 'file')
mut expected := os.join_path_single(tdir1, 'file')
assert os.exists(expected)
assert !os.is_dir(expected)
// Move dir with contents to other dir
os.mv(tdir1, tdir2) or { panic(err) }
expected = os.join_path(tdir2, 'dir')
expected = os.join_path_single(tdir2, 'dir')
assert os.exists(expected)
assert os.is_dir(expected)
expected = os.join_path(tdir2, 'dir', 'file')
assert os.exists(expected)
assert !os.is_dir(expected)
// Move dir with contents to other dir (by renaming)
os.mv(os.join_path(tdir2, 'dir'), tdir3) or { panic(err) }
os.mv(os.join_path_single(tdir2, 'dir'), tdir3) or { panic(err) }
expected = tdir3
assert os.exists(expected)
assert os.is_dir(expected)
assert os.is_dir_empty(tdir2)
// Move file with extension to dir
os.mv(tfile2, tdir2) or { panic(err) }
expected = os.join_path(tdir2, 'file.test')
expected = os.join_path_single(tdir2, 'file.test')
assert os.exists(expected)
assert !os.is_dir(expected)
// Move file to dir (by renaming)
os.mv(os.join_path(tdir2, 'file.test'), tfile3) or { panic(err) }
os.mv(os.join_path_single(tdir2, 'file.test'), tfile3) or { panic(err) }
expected = tfile3
assert os.exists(expected)
assert !os.is_dir(expected)
@ -375,7 +375,7 @@ fn test_cp_all() {
// regression test for executive runs with overwrite := true
os.cp_all('ex', './', true) or { panic(err) }
os.cp_all('ex', 'nonexisting', true) or { panic(err) }
assert os.exists(os.join_path('nonexisting', 'ex1.txt'))
assert os.exists(os.join_path_single('nonexisting', 'ex1.txt'))
}
fn test_realpath_of_empty_string_works() {
@ -398,7 +398,7 @@ fn test_realpath_non_existing() {
fn test_realpath_existing() {
existing_file_name := 'existing_file.txt'
existing_file := os.join_path(os.temp_dir(), existing_file_name)
existing_file := os.join_path_single(os.temp_dir(), existing_file_name)
os.rm(existing_file) or {}
os.write_file(existing_file, 'abc') or {}
assert os.exists(existing_file)
@ -843,7 +843,7 @@ fn test_expand_tilde_to_home() {
}
fn test_execute() ? {
print0script := os.join_path(tfolder, 'print0.v')
print0script := os.join_path_single(tfolder, 'print0.v')
// The output of the next command contains a 0 byte in the middle.
// Nevertheless, the execute function *should* return a string that
// contains it.

View File

@ -503,7 +503,8 @@ pub fn is_writable_folder(folder string) ?bool {
if !is_dir(folder) {
return error('`folder` is not a folder')
}
tmp_perm_check := join_path(folder, 'tmp_perm_check_pid_' + getpid().str())
tmp_folder_name := 'tmp_perm_check_pid_' + getpid().str()
tmp_perm_check := join_path_single(folder, tmp_folder_name)
mut f := open_file(tmp_perm_check, 'w+', 0o700) or {
return error('cannot write to folder $folder: $err')
}

View File

@ -7,8 +7,8 @@ fn main() {
println(os.args)
println(os.wd_at_startup)
//
fullpath := os.join_path('abc', 'xyz', 'def')
println(fullpath)
// fullpath := os.join_path('abc', 'xyz', 'def')
// println(fullpath)
//
x := 'abc'
t := x.trim_right('/')
@ -30,7 +30,7 @@ fn main() {
// exeparent_folder := os.dir(exe_realpath)
// println(exeparent_folder)
cdir := os.join_path(os.home_dir(), '.cache')
cdir := os.join_path_single(os.home_dir(), '.cache')
println(cdir)
wd_realpath := os.real_path(os.wd_at_startup)