vlib: prepare os for -Wimpure-v
Delyan Angelov
parent
14d55d26a3
commit
bd07177ef0
857
vlib/os/os.v
857
vlib/os/os.v
|
|
@ -3,136 +3,25 @@
|
|||
// that can be found in the LICENSE file.
|
||||
module os
|
||||
|
||||
pub struct Result {
|
||||
pub:
|
||||
exit_code int
|
||||
output string
|
||||
// stderr string // TODO
|
||||
}
|
||||
|
||||
pub const (
|
||||
args = []string{}
|
||||
max_path_len = 4096
|
||||
wd_at_startup = getwd()
|
||||
)
|
||||
|
||||
// read_bytes returns all bytes read from file in `path`.
|
||||
pub fn read_bytes(path string) ?[]byte {
|
||||
mut fp := vfopen(path, 'rb') ?
|
||||
cseek := C.fseek(fp, 0, C.SEEK_END)
|
||||
if cseek != 0 {
|
||||
return error('fseek failed')
|
||||
}
|
||||
fsize := C.ftell(fp)
|
||||
if fsize < 0 {
|
||||
return error('ftell failed')
|
||||
}
|
||||
C.rewind(fp)
|
||||
mut res := []byte{len: fsize}
|
||||
nr_read_elements := C.fread(res.data, fsize, 1, fp)
|
||||
if nr_read_elements == 0 && fsize > 0 {
|
||||
return error('fread failed')
|
||||
}
|
||||
C.fclose(fp)
|
||||
return res[0..nr_read_elements * fsize]
|
||||
}
|
||||
|
||||
// read_file reads the file in `path` and returns the contents.
|
||||
pub fn read_file(path string) ?string {
|
||||
mode := 'rb'
|
||||
mut fp := vfopen(path, mode) ?
|
||||
defer {
|
||||
C.fclose(fp)
|
||||
}
|
||||
cseek := C.fseek(fp, 0, C.SEEK_END)
|
||||
if cseek != 0 {
|
||||
return error('fseek failed')
|
||||
}
|
||||
fsize := C.ftell(fp)
|
||||
if fsize < 0 {
|
||||
return error('ftell failed')
|
||||
}
|
||||
// C.fseek(fp, 0, SEEK_SET) // same as `C.rewind(fp)` below
|
||||
C.rewind(fp)
|
||||
unsafe {
|
||||
mut str := malloc(fsize + 1)
|
||||
nelements := C.fread(str, fsize, 1, fp)
|
||||
if nelements == 0 && fsize > 0 {
|
||||
free(str)
|
||||
return error('fread failed')
|
||||
}
|
||||
str[fsize] = 0
|
||||
return str.vstring_with_len(fsize)
|
||||
}
|
||||
}
|
||||
|
||||
// ***************************** OS ops ************************
|
||||
// file_size returns the size of the file located in `path`.
|
||||
pub fn file_size(path string) int {
|
||||
mut s := C.stat{}
|
||||
unsafe {
|
||||
$if windows {
|
||||
$if tinyc {
|
||||
C.stat(charptr(path.str), &s)
|
||||
} $else {
|
||||
C._wstat(path.to_wide(), voidptr(&s))
|
||||
}
|
||||
} $else {
|
||||
C.stat(charptr(path.str), &s)
|
||||
}
|
||||
}
|
||||
return s.st_size
|
||||
}
|
||||
|
||||
// mv moves files or folders from `src` to `dst`.
|
||||
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)))
|
||||
}
|
||||
$if windows {
|
||||
w_src := src.replace('/', '\\')
|
||||
w_dst := rdst.replace('/', '\\')
|
||||
C._wrename(w_src.to_wide(), w_dst.to_wide())
|
||||
} $else {
|
||||
C.rename(charptr(src.str), charptr(rdst.str))
|
||||
}
|
||||
}
|
||||
|
||||
// cp copies files or folders from `src` to `dst`.
|
||||
pub fn cp(src string, dst string) ? {
|
||||
$if windows {
|
||||
w_src := src.replace('/', '\\')
|
||||
w_dst := dst.replace('/', '\\')
|
||||
if C.CopyFile(w_src.to_wide(), w_dst.to_wide(), false) == 0 {
|
||||
result := C.GetLastError()
|
||||
return error_with_code('failed to copy $src to $dst', int(result))
|
||||
}
|
||||
} $else {
|
||||
fp_from := C.open(charptr(src.str), C.O_RDONLY)
|
||||
if fp_from < 0 { // Check if file opened
|
||||
return error_with_code('cp: failed to open $src', int(fp_from))
|
||||
}
|
||||
fp_to := C.open(charptr(dst.str), C.O_WRONLY | C.O_CREAT | C.O_TRUNC, C.S_IWUSR | C.S_IRUSR)
|
||||
if fp_to < 0 { // Check if file opened (permissions problems ...)
|
||||
C.close(fp_from)
|
||||
return error_with_code('cp (permission): failed to write to $dst (fp_to: $fp_to)',
|
||||
int(fp_to))
|
||||
}
|
||||
mut buf := [1024]byte{}
|
||||
mut count := 0
|
||||
for {
|
||||
// FIXME: use sizeof, bug: 'os__buf' undeclared
|
||||
// count = C.read(fp_from, buf, sizeof(buf))
|
||||
count = C.read(fp_from, buf, 1024)
|
||||
if count == 0 {
|
||||
break
|
||||
}
|
||||
if C.write(fp_to, buf, count) < 0 {
|
||||
return error_with_code('cp: failed to write to $dst', int(-1))
|
||||
}
|
||||
}
|
||||
from_attr := C.stat{}
|
||||
unsafe {C.stat(charptr(src.str), &from_attr)}
|
||||
if C.chmod(charptr(dst.str), from_attr.st_mode) < 0 {
|
||||
return error_with_code('failed to set permissions for $dst', int(-1))
|
||||
}
|
||||
C.close(fp_to)
|
||||
C.close(fp_from)
|
||||
}
|
||||
}
|
||||
const (
|
||||
f_ok = 0
|
||||
x_ok = 1
|
||||
w_ok = 2
|
||||
r_ok = 4
|
||||
)
|
||||
|
||||
[deprecated]
|
||||
pub fn cp_r(osource_path string, odest_path string, overwrite bool) ? {
|
||||
|
|
@ -185,39 +74,6 @@ pub fn mv_by_cp(source string, target string) ? {
|
|||
rm(source) ?
|
||||
}
|
||||
|
||||
// vfopen returns an opened C file, given its path and open mode.
|
||||
// NB: os.vfopen is useful for compatibility with C libraries, that expect `FILE *`.
|
||||
// If you write pure V code, os.create or os.open are more convenient.
|
||||
pub fn vfopen(path string, mode string) ?&C.FILE {
|
||||
if path.len == 0 {
|
||||
return error('vfopen called with ""')
|
||||
}
|
||||
mut fp := voidptr(0)
|
||||
$if windows {
|
||||
fp = C._wfopen(path.to_wide(), mode.to_wide())
|
||||
} $else {
|
||||
fp = C.fopen(charptr(path.str), charptr(mode.str))
|
||||
}
|
||||
if isnil(fp) {
|
||||
return error('failed to open file "$path"')
|
||||
} else {
|
||||
return fp
|
||||
}
|
||||
}
|
||||
|
||||
// fileno returns the file descriptor of an opened C file.
|
||||
pub fn fileno(cfile voidptr) int {
|
||||
$if windows {
|
||||
return C._fileno(cfile)
|
||||
} $else {
|
||||
mut cfile_casted := &C.FILE(0) // FILE* cfile_casted = 0;
|
||||
cfile_casted = cfile
|
||||
// Required on FreeBSD/OpenBSD/NetBSD as stdio.h defines fileno(..) with a macro
|
||||
// that performs a field access on its argument without casting from void*.
|
||||
return C.fileno(cfile_casted)
|
||||
}
|
||||
}
|
||||
|
||||
// read_lines reads the file in `path` into an array of lines.
|
||||
pub fn read_lines(path string) ?[]string {
|
||||
buf := read_file(path) ?
|
||||
|
|
@ -236,175 +92,6 @@ fn read_ulines(path string) ?[]ustring {
|
|||
return ulines
|
||||
}
|
||||
|
||||
// open_append opens `path` file for appending.
|
||||
pub fn open_append(path string) ?File {
|
||||
mut file := File{}
|
||||
$if windows {
|
||||
wpath := path.replace('/', '\\').to_wide()
|
||||
mode := 'ab'
|
||||
file = File{
|
||||
cfile: C._wfopen(wpath, mode.to_wide())
|
||||
}
|
||||
} $else {
|
||||
cpath := path.str
|
||||
file = File{
|
||||
cfile: C.fopen(charptr(cpath), 'ab')
|
||||
}
|
||||
}
|
||||
if isnil(file.cfile) {
|
||||
return error('failed to create(append) file "$path"')
|
||||
}
|
||||
file.is_opened = true
|
||||
return file
|
||||
}
|
||||
|
||||
// open_file can be used to open or create a file with custom flags and permissions and returns a `File` object.
|
||||
pub fn open_file(path string, mode string, options ...int) ?File {
|
||||
mut flags := 0
|
||||
for m in mode {
|
||||
match m {
|
||||
`r` { flags |= o_rdonly }
|
||||
`w` { flags |= o_create | o_trunc }
|
||||
`b` { flags |= o_binary }
|
||||
`a` { flags |= o_create | o_append }
|
||||
`s` { flags |= o_sync }
|
||||
`n` { flags |= o_nonblock }
|
||||
`c` { flags |= o_noctty }
|
||||
`+` { flags |= o_rdwr }
|
||||
else {}
|
||||
}
|
||||
}
|
||||
mut permission := 0o666
|
||||
if options.len > 0 {
|
||||
permission = options[0]
|
||||
}
|
||||
$if windows {
|
||||
if permission < 0o600 {
|
||||
permission = 0x0100
|
||||
} else {
|
||||
permission = 0x0100 | 0x0080
|
||||
}
|
||||
}
|
||||
mut p := path
|
||||
$if windows {
|
||||
p = path.replace('/', '\\')
|
||||
}
|
||||
fd := C.open(charptr(p.str), flags, permission)
|
||||
if fd == -1 {
|
||||
return error(posix_get_error_msg(C.errno))
|
||||
}
|
||||
cfile := C.fdopen(fd, charptr(mode.str))
|
||||
if isnil(cfile) {
|
||||
return error('Failed to open or create file "$path"')
|
||||
}
|
||||
return File{
|
||||
cfile: cfile
|
||||
fd: fd
|
||||
is_opened: true
|
||||
}
|
||||
}
|
||||
|
||||
// vpopen system starts the specified command, waits for it to complete, and returns its code.
|
||||
fn vpopen(path string) voidptr {
|
||||
// *C.FILE {
|
||||
$if windows {
|
||||
mode := 'rb'
|
||||
wpath := path.to_wide()
|
||||
return C._wpopen(wpath, mode.to_wide())
|
||||
} $else {
|
||||
cpath := path.str
|
||||
return C.popen(charptr(cpath), 'r')
|
||||
}
|
||||
}
|
||||
|
||||
fn posix_wait4_to_exit_status(waitret int) (int, bool) {
|
||||
$if windows {
|
||||
return waitret, false
|
||||
} $else {
|
||||
mut ret := 0
|
||||
mut is_signaled := true
|
||||
// (see man system, man 2 waitpid: C macro WEXITSTATUS section)
|
||||
if C.WIFEXITED(waitret) {
|
||||
ret = C.WEXITSTATUS(waitret)
|
||||
is_signaled = false
|
||||
} else if C.WIFSIGNALED(waitret) {
|
||||
ret = C.WTERMSIG(waitret)
|
||||
is_signaled = true
|
||||
}
|
||||
return ret, is_signaled
|
||||
}
|
||||
}
|
||||
|
||||
// posix_get_error_msg return error code representation in string.
|
||||
pub fn posix_get_error_msg(code int) string {
|
||||
ptr_text := C.strerror(code) // voidptr?
|
||||
if ptr_text == 0 {
|
||||
return ''
|
||||
}
|
||||
return tos3(ptr_text)
|
||||
}
|
||||
|
||||
// vpclose will close a file pointer opened with `vpopen`.
|
||||
fn vpclose(f voidptr) int {
|
||||
$if windows {
|
||||
return C._pclose(f)
|
||||
} $else {
|
||||
ret, _ := posix_wait4_to_exit_status(C.pclose(f))
|
||||
return ret
|
||||
}
|
||||
}
|
||||
|
||||
pub struct Result {
|
||||
pub:
|
||||
exit_code int
|
||||
output string
|
||||
// stderr string // TODO
|
||||
}
|
||||
|
||||
// system works like `exec`, but only returns a return code.
|
||||
pub fn system(cmd string) int {
|
||||
// if cmd.contains(';') || cmd.contains('&&') || cmd.contains('||') || cmd.contains('\n') {
|
||||
// TODO remove panic
|
||||
// panic(';, &&, || and \\n are not allowed in shell commands')
|
||||
// }
|
||||
mut ret := 0
|
||||
$if windows {
|
||||
// overcome bug in system & _wsystem (cmd) when first char is quote `"`
|
||||
wcmd := if cmd.len > 1 && cmd[0] == `"` && cmd[1] != `"` { '"$cmd"' } else { cmd }
|
||||
unsafe {
|
||||
ret = C._wsystem(wcmd.to_wide())
|
||||
}
|
||||
} $else {
|
||||
$if ios {
|
||||
unsafe {
|
||||
arg := [c'/bin/sh', c'-c', byteptr(cmd.str), 0]
|
||||
pid := 0
|
||||
ret = C.posix_spawn(&pid, '/bin/sh', 0, 0, arg.data, 0)
|
||||
status := 0
|
||||
ret = C.waitpid(pid, &status, 0)
|
||||
if C.WIFEXITED(status) {
|
||||
ret = C.WEXITSTATUS(status)
|
||||
}
|
||||
}
|
||||
} $else {
|
||||
unsafe {
|
||||
ret = C.system(charptr(cmd.str))
|
||||
}
|
||||
}
|
||||
}
|
||||
if ret == -1 {
|
||||
print_c_errno()
|
||||
}
|
||||
$if !windows {
|
||||
pret, is_signaled := posix_wait4_to_exit_status(ret)
|
||||
if is_signaled {
|
||||
println('Terminated by signal ${ret:2d} (' + sigint_to_signal_name(pret) + ')')
|
||||
}
|
||||
ret = pret
|
||||
}
|
||||
return ret
|
||||
}
|
||||
|
||||
// sigint_to_signal_name will translate `si` signal integer code to it's string code representation.
|
||||
pub fn sigint_to_signal_name(si int) string {
|
||||
// POSIX signals:
|
||||
|
|
@ -444,106 +131,12 @@ pub fn sigint_to_signal_name(si int) string {
|
|||
return 'unknown'
|
||||
}
|
||||
|
||||
const (
|
||||
f_ok = 0
|
||||
x_ok = 1
|
||||
w_ok = 2
|
||||
r_ok = 4
|
||||
)
|
||||
|
||||
// exists returns true if `path` (file or directory) exists.
|
||||
pub fn exists(path string) bool {
|
||||
$if windows {
|
||||
p := path.replace('/', '\\')
|
||||
return C._waccess(p.to_wide(), f_ok) != -1
|
||||
} $else {
|
||||
return C.access(charptr(path.str), f_ok) != -1
|
||||
}
|
||||
}
|
||||
|
||||
// is_executable returns `true` if `path` is executable.
|
||||
pub fn is_executable(path string) bool {
|
||||
$if windows {
|
||||
// NB: https://docs.microsoft.com/en-us/cpp/c-runtime-library/reference/access-waccess?view=vs-2019
|
||||
// i.e. there is no X bit there, the modes can be:
|
||||
// 00 Existence only
|
||||
// 02 Write-only
|
||||
// 04 Read-only
|
||||
// 06 Read and write
|
||||
p := real_path(path)
|
||||
return (exists(p) && p.ends_with('.exe'))
|
||||
}
|
||||
$if solaris {
|
||||
statbuf := C.stat{}
|
||||
unsafe {
|
||||
if C.stat(charptr(path.str), &statbuf) != 0 {
|
||||
return false
|
||||
}
|
||||
}
|
||||
return (int(statbuf.st_mode) & (s_ixusr | s_ixgrp | s_ixoth)) != 0
|
||||
}
|
||||
return C.access(charptr(path.str), x_ok) != -1
|
||||
}
|
||||
|
||||
// is_writable returns `true` if `path` is writable.
|
||||
pub fn is_writable(path string) bool {
|
||||
$if windows {
|
||||
p := path.replace('/', '\\')
|
||||
return C._waccess(p.to_wide(), w_ok) != -1
|
||||
} $else {
|
||||
return C.access(charptr(path.str), w_ok) != -1
|
||||
}
|
||||
}
|
||||
|
||||
// is_readable returns `true` if `path` is readable.
|
||||
pub fn is_readable(path string) bool {
|
||||
$if windows {
|
||||
p := path.replace('/', '\\')
|
||||
return C._waccess(p.to_wide(), r_ok) != -1
|
||||
} $else {
|
||||
return C.access(charptr(path.str), r_ok) != -1
|
||||
}
|
||||
}
|
||||
|
||||
[deprecated]
|
||||
pub fn file_exists(_path string) bool {
|
||||
eprintln('warning: `os.file_exists` has been deprecated, use `os.exists` instead')
|
||||
return exists(_path)
|
||||
}
|
||||
|
||||
// rm removes file in `path`.
|
||||
pub fn rm(path string) ? {
|
||||
$if windows {
|
||||
rc := C._wremove(path.to_wide())
|
||||
if rc == -1 {
|
||||
// TODO: proper error as soon as it's supported on windows
|
||||
return error('Failed to remove "$path"')
|
||||
}
|
||||
} $else {
|
||||
rc := C.remove(charptr(path.str))
|
||||
if rc == -1 {
|
||||
return error(posix_get_error_msg(C.errno))
|
||||
}
|
||||
}
|
||||
// C.unlink(path.cstr())
|
||||
}
|
||||
|
||||
// rmdir removes a specified directory.
|
||||
pub fn rmdir(path string) ? {
|
||||
$if windows {
|
||||
rc := C.RemoveDirectory(path.to_wide())
|
||||
if rc == 0 {
|
||||
// https://docs.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-removedirectorya - 0 is failure
|
||||
return error('Failed to remove "$path"')
|
||||
}
|
||||
} $else {
|
||||
rc := C.rmdir(charptr(path.str))
|
||||
if rc == -1 {
|
||||
return error(posix_get_error_msg(C.errno))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
[deprecated]
|
||||
pub fn rmdir_recursive(path string) {
|
||||
eprintln('warning: `os.rmdir_recursive` has been deprecated, use `os.rmdir_all` instead')
|
||||
|
|
@ -572,13 +165,6 @@ pub fn is_dir_empty(path string) bool {
|
|||
return items.len == 0
|
||||
}
|
||||
|
||||
// print_c_errno will print the current value of `C.errno`.
|
||||
fn print_c_errno() {
|
||||
e := C.errno
|
||||
se := tos_clone(byteptr(C.strerror(C.errno)))
|
||||
println('errno=$e err=$se')
|
||||
}
|
||||
|
||||
// file_ext will return the part after the last occurence of `.` in `path`.
|
||||
// The `.` is included.
|
||||
pub fn file_ext(path string) string {
|
||||
|
|
@ -642,79 +228,6 @@ pub fn get_line() string {
|
|||
}
|
||||
}
|
||||
|
||||
// get_raw_line returns a one-line string from stdin along with '\n' if there is any.
|
||||
pub fn get_raw_line() string {
|
||||
$if windows {
|
||||
unsafe {
|
||||
max_line_chars := 256
|
||||
buf := malloc(max_line_chars * 2)
|
||||
h_input := C.GetStdHandle(std_input_handle)
|
||||
mut bytes_read := 0
|
||||
if is_atty(0) > 0 {
|
||||
C.ReadConsole(h_input, buf, max_line_chars * 2, C.LPDWORD(&bytes_read),
|
||||
0)
|
||||
return string_from_wide2(&u16(buf), bytes_read)
|
||||
}
|
||||
mut offset := 0
|
||||
for {
|
||||
pos := buf + offset
|
||||
res := C.ReadFile(h_input, pos, 1, C.LPDWORD(&bytes_read), 0)
|
||||
if !res || bytes_read == 0 {
|
||||
break
|
||||
}
|
||||
if *pos == `\n` || *pos == `\r` {
|
||||
offset++
|
||||
break
|
||||
}
|
||||
offset++
|
||||
}
|
||||
return buf.vstring_with_len(offset)
|
||||
}
|
||||
} $else {
|
||||
max := size_t(0)
|
||||
mut buf := charptr(0)
|
||||
nr_chars := C.getline(&buf, &max, C.stdin)
|
||||
// defer { unsafe{ free(buf) } }
|
||||
if nr_chars == 0 || nr_chars == -1 {
|
||||
return ''
|
||||
}
|
||||
return tos3(buf)
|
||||
// res := tos_clone(buf)
|
||||
// return res
|
||||
}
|
||||
}
|
||||
|
||||
// get_raw_stdin will get the raw input from stdin.
|
||||
pub fn get_raw_stdin() []byte {
|
||||
$if windows {
|
||||
unsafe {
|
||||
block_bytes := 512
|
||||
mut buf := malloc(block_bytes)
|
||||
h_input := C.GetStdHandle(std_input_handle)
|
||||
mut bytes_read := 0
|
||||
mut offset := 0
|
||||
for {
|
||||
pos := buf + offset
|
||||
res := C.ReadFile(h_input, pos, block_bytes, C.LPDWORD(&bytes_read), 0)
|
||||
offset += bytes_read
|
||||
if !res {
|
||||
break
|
||||
}
|
||||
buf = v_realloc(buf, offset + block_bytes + (block_bytes - bytes_read))
|
||||
}
|
||||
C.CloseHandle(h_input)
|
||||
return array{
|
||||
element_size: 1
|
||||
data: voidptr(buf)
|
||||
len: offset
|
||||
cap: offset
|
||||
}
|
||||
}
|
||||
} $else {
|
||||
panic('get_raw_stdin not implemented on this platform...')
|
||||
}
|
||||
}
|
||||
|
||||
// get_lines returns an array of strings read from from stdin.
|
||||
// reading is stopped when an empty line is read.
|
||||
pub fn get_lines() []string {
|
||||
|
|
@ -782,7 +295,7 @@ pub fn user_os() string {
|
|||
return 'unknown'
|
||||
}
|
||||
|
||||
// home_dir returns path to user's home directory.
|
||||
// home_dir returns path to the user's home directory.
|
||||
pub fn home_dir() string {
|
||||
$if windows {
|
||||
return getenv('USERPROFILE')
|
||||
|
|
@ -808,129 +321,6 @@ pub fn write_file_array(path string, buffer array) ? {
|
|||
f.close()
|
||||
}
|
||||
|
||||
// read_file_array reads an array of `T` values from file `path`.
|
||||
pub fn read_file_array<T>(path string) []T {
|
||||
a := T{}
|
||||
tsize := int(sizeof(a))
|
||||
// prepare for reading, get current file size
|
||||
mut fp := vfopen(path, 'rb') or { return array{} }
|
||||
C.fseek(fp, 0, C.SEEK_END)
|
||||
fsize := C.ftell(fp)
|
||||
C.rewind(fp)
|
||||
// read the actual data from the file
|
||||
len := fsize / tsize
|
||||
buf := malloc(fsize)
|
||||
C.fread(buf, fsize, 1, fp)
|
||||
C.fclose(fp)
|
||||
return array{
|
||||
element_size: tsize
|
||||
data: buf
|
||||
len: len
|
||||
cap: len
|
||||
}
|
||||
}
|
||||
|
||||
pub fn on_segfault(f voidptr) {
|
||||
$if windows {
|
||||
return
|
||||
}
|
||||
$if macos {
|
||||
C.printf('TODO')
|
||||
/*
|
||||
mut sa := C.sigaction{}
|
||||
C.memset(&sa, 0, sizeof(C.sigaction_size))
|
||||
C.sigemptyset(&sa.sa_mask)
|
||||
sa.sa_sigaction = f
|
||||
sa.sa_flags = C.SA_SIGINFO
|
||||
C.sigaction(C.SIGSEGV, &sa, 0)
|
||||
*/
|
||||
}
|
||||
}
|
||||
|
||||
// executable returns the path name of the executable that started the current
|
||||
// process.
|
||||
pub fn executable() string {
|
||||
$if linux {
|
||||
mut result := vcalloc(max_path_len)
|
||||
count := C.readlink('/proc/self/exe', charptr(result), max_path_len)
|
||||
if count < 0 {
|
||||
eprintln('os.executable() failed at reading /proc/self/exe to get exe path')
|
||||
return executable_fallback()
|
||||
}
|
||||
return unsafe {result.vstring()}
|
||||
}
|
||||
$if windows {
|
||||
max := 512
|
||||
size := max * 2 // max_path_len * sizeof(wchar_t)
|
||||
mut result := &u16(vcalloc(size))
|
||||
len := C.GetModuleFileName(0, result, max)
|
||||
// determine if the file is a windows symlink
|
||||
attrs := C.GetFileAttributesW(result)
|
||||
is_set := attrs & 0x400 // FILE_ATTRIBUTE_REPARSE_POINT
|
||||
if is_set != 0 { // it's a windows symlink
|
||||
// gets handle with GENERIC_READ, FILE_SHARE_READ, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0
|
||||
file := C.CreateFile(result, 0x80000000, 1, 0, 3, 0x80, 0)
|
||||
if file != voidptr(-1) {
|
||||
final_path := &u16(vcalloc(size))
|
||||
// https://docs.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-getfinalpathnamebyhandlew
|
||||
final_len := C.GetFinalPathNameByHandleW(file, final_path, size, 0)
|
||||
if final_len < size {
|
||||
ret := string_from_wide2(final_path, final_len)
|
||||
// remove '\\?\' from beginning (see link above)
|
||||
return ret[4..]
|
||||
} else {
|
||||
eprintln('os.executable() saw that the executable file path was too long')
|
||||
}
|
||||
}
|
||||
C.CloseHandle(file)
|
||||
}
|
||||
return string_from_wide2(result, len)
|
||||
}
|
||||
$if macos {
|
||||
mut result := vcalloc(max_path_len)
|
||||
pid := C.getpid()
|
||||
ret := proc_pidpath(pid, result, max_path_len)
|
||||
if ret <= 0 {
|
||||
eprintln('os.executable() failed at calling proc_pidpath with pid: $pid . proc_pidpath returned $ret ')
|
||||
return executable_fallback()
|
||||
}
|
||||
return unsafe {result.vstring()}
|
||||
}
|
||||
$if freebsd {
|
||||
mut result := vcalloc(max_path_len)
|
||||
mib := [1 /* CTL_KERN */, 14 /* KERN_PROC */, 12 /* KERN_PROC_PATHNAME */, -1]
|
||||
size := max_path_len
|
||||
unsafe {C.sysctl(mib.data, 4, result, &size, 0, 0)}
|
||||
return unsafe {result.vstring()}
|
||||
}
|
||||
// "Sadly there is no way to get the full path of the executed file in OpenBSD."
|
||||
$if openbsd {
|
||||
}
|
||||
$if solaris {
|
||||
}
|
||||
$if haiku {
|
||||
}
|
||||
$if netbsd {
|
||||
mut result := vcalloc(max_path_len)
|
||||
count := C.readlink('/proc/curproc/exe', charptr(result), max_path_len)
|
||||
if count < 0 {
|
||||
eprintln('os.executable() failed at reading /proc/curproc/exe to get exe path')
|
||||
return executable_fallback()
|
||||
}
|
||||
return unsafe {result.vstring_with_len(count)}
|
||||
}
|
||||
$if dragonfly {
|
||||
mut result := vcalloc(max_path_len)
|
||||
count := C.readlink('/proc/curproc/file', charptr(result), max_path_len)
|
||||
if count < 0 {
|
||||
eprintln('os.executable() failed at reading /proc/curproc/file to get exe path')
|
||||
return executable_fallback()
|
||||
}
|
||||
return unsafe {result.vstring_with_len(count)}
|
||||
}
|
||||
return executable_fallback()
|
||||
}
|
||||
|
||||
// executable_fallback is used when there is not a more platform specific and accurate implementation.
|
||||
// It relies on path manipulation of os.args[0] and os.wd_at_startup, so it may not work properly in
|
||||
// all cases, but it should be better, than just using os.args[0] directly.
|
||||
|
|
@ -993,114 +383,11 @@ pub fn dir_exists(path string) bool {
|
|||
return is_dir(path)
|
||||
}
|
||||
|
||||
// is_dir returns a `bool` indicating whether the given `path` is a directory.
|
||||
pub fn is_dir(path string) bool {
|
||||
$if windows {
|
||||
w_path := path.replace('/', '\\')
|
||||
attr := C.GetFileAttributesW(w_path.to_wide())
|
||||
if attr == u32(C.INVALID_FILE_ATTRIBUTES) {
|
||||
return false
|
||||
}
|
||||
if int(attr) & C.FILE_ATTRIBUTE_DIRECTORY != 0 {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
} $else {
|
||||
statbuf := C.stat{}
|
||||
if unsafe {C.stat(charptr(path.str), &statbuf)} != 0 {
|
||||
return false
|
||||
}
|
||||
// ref: https://code.woboq.org/gcc/include/sys/stat.h.html
|
||||
val := int(statbuf.st_mode) & s_ifmt
|
||||
return val == s_ifdir
|
||||
}
|
||||
}
|
||||
|
||||
// is_file returns a `bool` indicating whether the given `path` is a file.
|
||||
pub fn is_file(path string) bool {
|
||||
return exists(path) && !is_dir(path)
|
||||
}
|
||||
|
||||
// is_link returns a boolean indicating whether `path` is a link.
|
||||
pub fn is_link(path string) bool {
|
||||
$if windows {
|
||||
return false // TODO
|
||||
} $else {
|
||||
statbuf := C.stat{}
|
||||
if C.lstat(charptr(path.str), &statbuf) != 0 {
|
||||
return false
|
||||
}
|
||||
return int(statbuf.st_mode) & s_ifmt == s_iflnk
|
||||
}
|
||||
}
|
||||
|
||||
// chdir changes the current working directory to the new directory in `path`.
|
||||
pub fn chdir(path string) {
|
||||
$if windows {
|
||||
C._wchdir(path.to_wide())
|
||||
} $else {
|
||||
C.chdir(charptr(path.str))
|
||||
}
|
||||
}
|
||||
|
||||
// getwd returns the absolute path of the current directory.
|
||||
pub fn getwd() string {
|
||||
$if windows {
|
||||
max := 512 // max_path_len * sizeof(wchar_t)
|
||||
buf := &u16(vcalloc(max * 2))
|
||||
if C._wgetcwd(buf, max) == 0 {
|
||||
return ''
|
||||
}
|
||||
return string_from_wide(buf)
|
||||
} $else {
|
||||
buf := vcalloc(512)
|
||||
if C.getcwd(charptr(buf), 512) == 0 {
|
||||
return ''
|
||||
}
|
||||
return unsafe {buf.vstring()}
|
||||
}
|
||||
}
|
||||
|
||||
// real_path returns the full absolute path for fpath, with all relative ../../, symlinks and so on resolved.
|
||||
// See http://pubs.opengroup.org/onlinepubs/9699919799/functions/realpath.html
|
||||
// Also https://insanecoding.blogspot.com/2007/11/pathmax-simply-isnt.html
|
||||
// and https://insanecoding.blogspot.com/2007/11/implementing-realpath-in-c.html
|
||||
// NB: this particular rabbit hole is *deep* ...
|
||||
pub fn real_path(fpath string) string {
|
||||
mut fullpath := vcalloc(max_path_len)
|
||||
mut ret := charptr(0)
|
||||
$if windows {
|
||||
ret = charptr(C._fullpath(charptr(fullpath), charptr(fpath.str), max_path_len))
|
||||
if ret == 0 {
|
||||
return fpath
|
||||
}
|
||||
} $else {
|
||||
ret = charptr(C.realpath(charptr(fpath.str), charptr(fullpath)))
|
||||
if ret == 0 {
|
||||
return fpath
|
||||
}
|
||||
}
|
||||
res := unsafe {fullpath.vstring()}
|
||||
return normalize_drive_letter(res)
|
||||
}
|
||||
|
||||
fn normalize_drive_letter(path string) string {
|
||||
// normalize_drive_letter is needed, because a path like c:\nv\.bin (note the small `c`)
|
||||
// in %PATH is NOT recognized by cmd.exe (and probably other programs too)...
|
||||
// Capital drive letters do work fine.
|
||||
$if !windows {
|
||||
return path
|
||||
}
|
||||
if path.len > 2 &&
|
||||
path[0] >= `a` && path[0] <= `z` && path[1] == `:` && path[2] == path_separator[0] {
|
||||
unsafe {
|
||||
x := &path.str[0]
|
||||
(*x) = *x - 32
|
||||
}
|
||||
}
|
||||
return path
|
||||
}
|
||||
|
||||
// is_abs_path returns `true` if `path` is absolute.
|
||||
pub fn is_abs_path(path string) bool {
|
||||
$if windows {
|
||||
|
|
@ -1160,46 +447,6 @@ pub fn walk(path string, f fn (string)) {
|
|||
return
|
||||
}
|
||||
|
||||
// signal will assign `handler` callback to be called when `signum` signal is recieved.
|
||||
pub fn signal(signum int, handler voidptr) {
|
||||
unsafe {C.signal(signum, handler)}
|
||||
}
|
||||
|
||||
// fork will fork the current system process and return the pid of the fork.
|
||||
pub fn fork() int {
|
||||
mut pid := -1
|
||||
$if !windows {
|
||||
pid = C.fork()
|
||||
}
|
||||
$if windows {
|
||||
panic('os.fork not supported in windows') // TODO
|
||||
}
|
||||
return pid
|
||||
}
|
||||
|
||||
// wait blocks the calling process until one of its child processes exits or a signal is received.
|
||||
// After child process terminates, parent continues its execution after wait system call instruction.
|
||||
pub fn wait() int {
|
||||
mut pid := -1
|
||||
$if !windows {
|
||||
pid = C.wait(0)
|
||||
}
|
||||
$if windows {
|
||||
panic('os.wait not supported in windows') // TODO
|
||||
}
|
||||
return pid
|
||||
}
|
||||
|
||||
// file_last_mod_unix returns the "last modified" time stamp of file in `path`.
|
||||
pub fn file_last_mod_unix(path string) int {
|
||||
attr := C.stat{}
|
||||
// # struct stat attr;
|
||||
unsafe {C.stat(charptr(path.str), &attr)}
|
||||
// # stat(path.str, &attr);
|
||||
return attr.st_mtime
|
||||
// # return attr.st_mtime ;
|
||||
}
|
||||
|
||||
// log will print "os.log: "+`s` ...
|
||||
pub fn log(s string) {
|
||||
println('os.log: ' + s)
|
||||
|
|
@ -1211,11 +458,6 @@ pub fn flush_stdout() {
|
|||
flush()
|
||||
}
|
||||
|
||||
// flush will flush the stdout buffer.
|
||||
pub fn flush() {
|
||||
C.fflush(C.stdout)
|
||||
}
|
||||
|
||||
// mkdir_all will create a valid full path of all directories given in `path`.
|
||||
pub fn mkdir_all(path string) ? {
|
||||
mut p := if path.starts_with(path_separator) { path_separator } else { '' }
|
||||
|
|
@ -1304,16 +546,6 @@ pub fn vmodules_paths() []string {
|
|||
return list
|
||||
}
|
||||
|
||||
// chmod change file access attributes of `path` to `mode`.
|
||||
// Octals like `0o600` can be used.
|
||||
pub fn chmod(path string, mode int) {
|
||||
C.chmod(charptr(path.str), mode)
|
||||
}
|
||||
|
||||
pub const (
|
||||
wd_at_startup = getwd()
|
||||
)
|
||||
|
||||
// resource_abs_path returns an absolute path, for the given `path`.
|
||||
// (the path is expected to be relative to the executable program)
|
||||
// See https://discordapp.com/channels/592103645835821068/592294828432424960/630806741373943808
|
||||
|
|
@ -1328,65 +560,6 @@ pub fn resource_abs_path(path string) string {
|
|||
return real_path(join_path(base_path, path))
|
||||
}
|
||||
|
||||
// open tries to open a file for reading and returns back a read-only `File` object.
|
||||
pub fn open(path string) ?File {
|
||||
/*
|
||||
$if linux {
|
||||
$if !android {
|
||||
fd := C.syscall(sys_open, path.str, 511)
|
||||
if fd == -1 {
|
||||
return error('failed to open file "$path"')
|
||||
}
|
||||
return File{
|
||||
fd: fd
|
||||
is_opened: true
|
||||
}
|
||||
}
|
||||
}
|
||||
*/
|
||||
cfile := vfopen(path, 'rb') ?
|
||||
fd := fileno(cfile)
|
||||
return File{
|
||||
cfile: cfile
|
||||
fd: fd
|
||||
is_opened: true
|
||||
}
|
||||
}
|
||||
|
||||
// create creates or opens a file at a specified location and returns a write-only `File` object.
|
||||
pub fn create(path string) ?File {
|
||||
/*
|
||||
// NB: android/termux/bionic is also a kind of linux,
|
||||
// but linux syscalls there sometimes fail,
|
||||
// while the libc version should work.
|
||||
$if linux {
|
||||
$if !android {
|
||||
//$if macos {
|
||||
// fd = C.syscall(398, path.str, 0x601, 0x1b6)
|
||||
//}
|
||||
//$if linux {
|
||||
fd = C.syscall(sys_creat, path.str, 511)
|
||||
//}
|
||||
if fd == -1 {
|
||||
return error('failed to create file "$path"')
|
||||
}
|
||||
file = File{
|
||||
fd: fd
|
||||
is_opened: true
|
||||
}
|
||||
return file
|
||||
}
|
||||
}
|
||||
*/
|
||||
cfile := vfopen(path, 'wb') ?
|
||||
fd := fileno(cfile)
|
||||
return File{
|
||||
cfile: cfile
|
||||
fd: fd
|
||||
is_opened: true
|
||||
}
|
||||
}
|
||||
|
||||
pub struct Uname {
|
||||
pub mut:
|
||||
sysname string
|
||||
|
|
|
|||
825
vlib/os/os_c.v
825
vlib/os/os_c.v
|
|
@ -42,3 +42,828 @@ mut:
|
|||
struct C.dirent {
|
||||
d_name byteptr
|
||||
}
|
||||
|
||||
// read_bytes returns all bytes read from file in `path`.
|
||||
pub fn read_bytes(path string) ?[]byte {
|
||||
mut fp := vfopen(path, 'rb') ?
|
||||
cseek := C.fseek(fp, 0, C.SEEK_END)
|
||||
if cseek != 0 {
|
||||
return error('fseek failed')
|
||||
}
|
||||
fsize := C.ftell(fp)
|
||||
if fsize < 0 {
|
||||
return error('ftell failed')
|
||||
}
|
||||
C.rewind(fp)
|
||||
mut res := []byte{len: fsize}
|
||||
nr_read_elements := C.fread(res.data, fsize, 1, fp)
|
||||
if nr_read_elements == 0 && fsize > 0 {
|
||||
return error('fread failed')
|
||||
}
|
||||
C.fclose(fp)
|
||||
return res[0..nr_read_elements * fsize]
|
||||
}
|
||||
|
||||
// read_file reads the file in `path` and returns the contents.
|
||||
pub fn read_file(path string) ?string {
|
||||
mode := 'rb'
|
||||
mut fp := vfopen(path, mode) ?
|
||||
defer {
|
||||
C.fclose(fp)
|
||||
}
|
||||
cseek := C.fseek(fp, 0, C.SEEK_END)
|
||||
if cseek != 0 {
|
||||
return error('fseek failed')
|
||||
}
|
||||
fsize := C.ftell(fp)
|
||||
if fsize < 0 {
|
||||
return error('ftell failed')
|
||||
}
|
||||
// C.fseek(fp, 0, SEEK_SET) // same as `C.rewind(fp)` below
|
||||
C.rewind(fp)
|
||||
unsafe {
|
||||
mut str := malloc(fsize + 1)
|
||||
nelements := C.fread(str, fsize, 1, fp)
|
||||
if nelements == 0 && fsize > 0 {
|
||||
free(str)
|
||||
return error('fread failed')
|
||||
}
|
||||
str[fsize] = 0
|
||||
return str.vstring_with_len(fsize)
|
||||
}
|
||||
}
|
||||
|
||||
// ***************************** OS ops ************************
|
||||
// file_size returns the size of the file located in `path`.
|
||||
pub fn file_size(path string) int {
|
||||
mut s := C.stat{}
|
||||
unsafe {
|
||||
$if windows {
|
||||
$if tinyc {
|
||||
C.stat(charptr(path.str), &s)
|
||||
} $else {
|
||||
C._wstat(path.to_wide(), voidptr(&s))
|
||||
}
|
||||
} $else {
|
||||
C.stat(charptr(path.str), &s)
|
||||
}
|
||||
}
|
||||
return s.st_size
|
||||
}
|
||||
|
||||
// mv moves files or folders from `src` to `dst`.
|
||||
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)))
|
||||
}
|
||||
$if windows {
|
||||
w_src := src.replace('/', '\\')
|
||||
w_dst := rdst.replace('/', '\\')
|
||||
C._wrename(w_src.to_wide(), w_dst.to_wide())
|
||||
} $else {
|
||||
C.rename(charptr(src.str), charptr(rdst.str))
|
||||
}
|
||||
}
|
||||
|
||||
// cp copies files or folders from `src` to `dst`.
|
||||
pub fn cp(src string, dst string) ? {
|
||||
$if windows {
|
||||
w_src := src.replace('/', '\\')
|
||||
w_dst := dst.replace('/', '\\')
|
||||
if C.CopyFile(w_src.to_wide(), w_dst.to_wide(), false) == 0 {
|
||||
result := C.GetLastError()
|
||||
return error_with_code('failed to copy $src to $dst', int(result))
|
||||
}
|
||||
} $else {
|
||||
fp_from := C.open(charptr(src.str), C.O_RDONLY)
|
||||
if fp_from < 0 { // Check if file opened
|
||||
return error_with_code('cp: failed to open $src', int(fp_from))
|
||||
}
|
||||
fp_to := C.open(charptr(dst.str), C.O_WRONLY | C.O_CREAT | C.O_TRUNC, C.S_IWUSR | C.S_IRUSR)
|
||||
if fp_to < 0 { // Check if file opened (permissions problems ...)
|
||||
C.close(fp_from)
|
||||
return error_with_code('cp (permission): failed to write to $dst (fp_to: $fp_to)',
|
||||
int(fp_to))
|
||||
}
|
||||
mut buf := [1024]byte{}
|
||||
mut count := 0
|
||||
for {
|
||||
// FIXME: use sizeof, bug: 'os__buf' undeclared
|
||||
// count = C.read(fp_from, buf, sizeof(buf))
|
||||
count = C.read(fp_from, buf, 1024)
|
||||
if count == 0 {
|
||||
break
|
||||
}
|
||||
if C.write(fp_to, buf, count) < 0 {
|
||||
return error_with_code('cp: failed to write to $dst', int(-1))
|
||||
}
|
||||
}
|
||||
from_attr := C.stat{}
|
||||
unsafe {C.stat(charptr(src.str), &from_attr)}
|
||||
if C.chmod(charptr(dst.str), from_attr.st_mode) < 0 {
|
||||
return error_with_code('failed to set permissions for $dst', int(-1))
|
||||
}
|
||||
C.close(fp_to)
|
||||
C.close(fp_from)
|
||||
}
|
||||
}
|
||||
|
||||
// vfopen returns an opened C file, given its path and open mode.
|
||||
// NB: os.vfopen is useful for compatibility with C libraries, that expect `FILE *`.
|
||||
// If you write pure V code, os.create or os.open are more convenient.
|
||||
pub fn vfopen(path string, mode string) ?&C.FILE {
|
||||
if path.len == 0 {
|
||||
return error('vfopen called with ""')
|
||||
}
|
||||
mut fp := voidptr(0)
|
||||
$if windows {
|
||||
fp = C._wfopen(path.to_wide(), mode.to_wide())
|
||||
} $else {
|
||||
fp = C.fopen(charptr(path.str), charptr(mode.str))
|
||||
}
|
||||
if isnil(fp) {
|
||||
return error('failed to open file "$path"')
|
||||
} else {
|
||||
return fp
|
||||
}
|
||||
}
|
||||
|
||||
// fileno returns the file descriptor of an opened C file.
|
||||
pub fn fileno(cfile voidptr) int {
|
||||
$if windows {
|
||||
return C._fileno(cfile)
|
||||
} $else {
|
||||
mut cfile_casted := &C.FILE(0) // FILE* cfile_casted = 0;
|
||||
cfile_casted = cfile
|
||||
// Required on FreeBSD/OpenBSD/NetBSD as stdio.h defines fileno(..) with a macro
|
||||
// that performs a field access on its argument without casting from void*.
|
||||
return C.fileno(cfile_casted)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// open_append opens `path` file for appending.
|
||||
pub fn open_append(path string) ?File {
|
||||
mut file := File{}
|
||||
$if windows {
|
||||
wpath := path.replace('/', '\\').to_wide()
|
||||
mode := 'ab'
|
||||
file = File{
|
||||
cfile: C._wfopen(wpath, mode.to_wide())
|
||||
}
|
||||
} $else {
|
||||
cpath := path.str
|
||||
file = File{
|
||||
cfile: C.fopen(charptr(cpath), 'ab')
|
||||
}
|
||||
}
|
||||
if isnil(file.cfile) {
|
||||
return error('failed to create(append) file "$path"')
|
||||
}
|
||||
file.is_opened = true
|
||||
return file
|
||||
}
|
||||
|
||||
// open_file can be used to open or create a file with custom flags and permissions and returns a `File` object.
|
||||
pub fn open_file(path string, mode string, options ...int) ?File {
|
||||
mut flags := 0
|
||||
for m in mode {
|
||||
match m {
|
||||
`r` { flags |= o_rdonly }
|
||||
`w` { flags |= o_create | o_trunc }
|
||||
`b` { flags |= o_binary }
|
||||
`a` { flags |= o_create | o_append }
|
||||
`s` { flags |= o_sync }
|
||||
`n` { flags |= o_nonblock }
|
||||
`c` { flags |= o_noctty }
|
||||
`+` { flags |= o_rdwr }
|
||||
else {}
|
||||
}
|
||||
}
|
||||
mut permission := 0o666
|
||||
if options.len > 0 {
|
||||
permission = options[0]
|
||||
}
|
||||
$if windows {
|
||||
if permission < 0o600 {
|
||||
permission = 0x0100
|
||||
} else {
|
||||
permission = 0x0100 | 0x0080
|
||||
}
|
||||
}
|
||||
mut p := path
|
||||
$if windows {
|
||||
p = path.replace('/', '\\')
|
||||
}
|
||||
fd := C.open(charptr(p.str), flags, permission)
|
||||
if fd == -1 {
|
||||
return error(posix_get_error_msg(C.errno))
|
||||
}
|
||||
cfile := C.fdopen(fd, charptr(mode.str))
|
||||
if isnil(cfile) {
|
||||
return error('Failed to open or create file "$path"')
|
||||
}
|
||||
return File{
|
||||
cfile: cfile
|
||||
fd: fd
|
||||
is_opened: true
|
||||
}
|
||||
}
|
||||
|
||||
// vpopen system starts the specified command, waits for it to complete, and returns its code.
|
||||
fn vpopen(path string) voidptr {
|
||||
// *C.FILE {
|
||||
$if windows {
|
||||
mode := 'rb'
|
||||
wpath := path.to_wide()
|
||||
return C._wpopen(wpath, mode.to_wide())
|
||||
} $else {
|
||||
cpath := path.str
|
||||
return C.popen(charptr(cpath), 'r')
|
||||
}
|
||||
}
|
||||
|
||||
fn posix_wait4_to_exit_status(waitret int) (int, bool) {
|
||||
$if windows {
|
||||
return waitret, false
|
||||
} $else {
|
||||
mut ret := 0
|
||||
mut is_signaled := true
|
||||
// (see man system, man 2 waitpid: C macro WEXITSTATUS section)
|
||||
if C.WIFEXITED(waitret) {
|
||||
ret = C.WEXITSTATUS(waitret)
|
||||
is_signaled = false
|
||||
} else if C.WIFSIGNALED(waitret) {
|
||||
ret = C.WTERMSIG(waitret)
|
||||
is_signaled = true
|
||||
}
|
||||
return ret, is_signaled
|
||||
}
|
||||
}
|
||||
|
||||
// posix_get_error_msg return error code representation in string.
|
||||
pub fn posix_get_error_msg(code int) string {
|
||||
ptr_text := C.strerror(code) // voidptr?
|
||||
if ptr_text == 0 {
|
||||
return ''
|
||||
}
|
||||
return tos3(ptr_text)
|
||||
}
|
||||
|
||||
// vpclose will close a file pointer opened with `vpopen`.
|
||||
fn vpclose(f voidptr) int {
|
||||
$if windows {
|
||||
return C._pclose(f)
|
||||
} $else {
|
||||
ret, _ := posix_wait4_to_exit_status(C.pclose(f))
|
||||
return ret
|
||||
}
|
||||
}
|
||||
|
||||
// system works like `exec`, but only returns a return code.
|
||||
pub fn system(cmd string) int {
|
||||
// if cmd.contains(';') || cmd.contains('&&') || cmd.contains('||') || cmd.contains('\n') {
|
||||
// TODO remove panic
|
||||
// panic(';, &&, || and \\n are not allowed in shell commands')
|
||||
// }
|
||||
mut ret := 0
|
||||
$if windows {
|
||||
// overcome bug in system & _wsystem (cmd) when first char is quote `"`
|
||||
wcmd := if cmd.len > 1 && cmd[0] == `"` && cmd[1] != `"` { '"$cmd"' } else { cmd }
|
||||
unsafe {
|
||||
ret = C._wsystem(wcmd.to_wide())
|
||||
}
|
||||
} $else {
|
||||
$if ios {
|
||||
unsafe {
|
||||
arg := [c'/bin/sh', c'-c', byteptr(cmd.str), 0]
|
||||
pid := 0
|
||||
ret = C.posix_spawn(&pid, '/bin/sh', 0, 0, arg.data, 0)
|
||||
status := 0
|
||||
ret = C.waitpid(pid, &status, 0)
|
||||
if C.WIFEXITED(status) {
|
||||
ret = C.WEXITSTATUS(status)
|
||||
}
|
||||
}
|
||||
} $else {
|
||||
unsafe {
|
||||
ret = C.system(charptr(cmd.str))
|
||||
}
|
||||
}
|
||||
}
|
||||
if ret == -1 {
|
||||
print_c_errno()
|
||||
}
|
||||
$if !windows {
|
||||
pret, is_signaled := posix_wait4_to_exit_status(ret)
|
||||
if is_signaled {
|
||||
println('Terminated by signal ${ret:2d} (' + sigint_to_signal_name(pret) + ')')
|
||||
}
|
||||
ret = pret
|
||||
}
|
||||
return ret
|
||||
}
|
||||
|
||||
// exists returns true if `path` (file or directory) exists.
|
||||
pub fn exists(path string) bool {
|
||||
$if windows {
|
||||
p := path.replace('/', '\\')
|
||||
return C._waccess(p.to_wide(), f_ok) != -1
|
||||
} $else {
|
||||
return C.access(charptr(path.str), f_ok) != -1
|
||||
}
|
||||
}
|
||||
|
||||
// is_executable returns `true` if `path` is executable.
|
||||
pub fn is_executable(path string) bool {
|
||||
$if windows {
|
||||
// NB: https://docs.microsoft.com/en-us/cpp/c-runtime-library/reference/access-waccess?view=vs-2019
|
||||
// i.e. there is no X bit there, the modes can be:
|
||||
// 00 Existence only
|
||||
// 02 Write-only
|
||||
// 04 Read-only
|
||||
// 06 Read and write
|
||||
p := real_path(path)
|
||||
return (exists(p) && p.ends_with('.exe'))
|
||||
}
|
||||
$if solaris {
|
||||
statbuf := C.stat{}
|
||||
unsafe {
|
||||
if C.stat(charptr(path.str), &statbuf) != 0 {
|
||||
return false
|
||||
}
|
||||
}
|
||||
return (int(statbuf.st_mode) & (s_ixusr | s_ixgrp | s_ixoth)) != 0
|
||||
}
|
||||
return C.access(charptr(path.str), x_ok) != -1
|
||||
}
|
||||
|
||||
// is_writable returns `true` if `path` is writable.
|
||||
pub fn is_writable(path string) bool {
|
||||
$if windows {
|
||||
p := path.replace('/', '\\')
|
||||
return C._waccess(p.to_wide(), w_ok) != -1
|
||||
} $else {
|
||||
return C.access(charptr(path.str), w_ok) != -1
|
||||
}
|
||||
}
|
||||
|
||||
// is_readable returns `true` if `path` is readable.
|
||||
pub fn is_readable(path string) bool {
|
||||
$if windows {
|
||||
p := path.replace('/', '\\')
|
||||
return C._waccess(p.to_wide(), r_ok) != -1
|
||||
} $else {
|
||||
return C.access(charptr(path.str), r_ok) != -1
|
||||
}
|
||||
}
|
||||
|
||||
// rm removes file in `path`.
|
||||
pub fn rm(path string) ? {
|
||||
$if windows {
|
||||
rc := C._wremove(path.to_wide())
|
||||
if rc == -1 {
|
||||
// TODO: proper error as soon as it's supported on windows
|
||||
return error('Failed to remove "$path"')
|
||||
}
|
||||
} $else {
|
||||
rc := C.remove(charptr(path.str))
|
||||
if rc == -1 {
|
||||
return error(posix_get_error_msg(C.errno))
|
||||
}
|
||||
}
|
||||
// C.unlink(path.cstr())
|
||||
}
|
||||
|
||||
// rmdir removes a specified directory.
|
||||
pub fn rmdir(path string) ? {
|
||||
$if windows {
|
||||
rc := C.RemoveDirectory(path.to_wide())
|
||||
if rc == 0 {
|
||||
// https://docs.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-removedirectorya - 0 is failure
|
||||
return error('Failed to remove "$path"')
|
||||
}
|
||||
} $else {
|
||||
rc := C.rmdir(charptr(path.str))
|
||||
if rc == -1 {
|
||||
return error(posix_get_error_msg(C.errno))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// print_c_errno will print the current value of `C.errno`.
|
||||
fn print_c_errno() {
|
||||
e := C.errno
|
||||
se := tos_clone(byteptr(C.strerror(C.errno)))
|
||||
println('errno=$e err=$se')
|
||||
}
|
||||
|
||||
// get_raw_line returns a one-line string from stdin along with '\n' if there is any.
|
||||
pub fn get_raw_line() string {
|
||||
$if windows {
|
||||
unsafe {
|
||||
max_line_chars := 256
|
||||
buf := malloc(max_line_chars * 2)
|
||||
h_input := C.GetStdHandle(std_input_handle)
|
||||
mut bytes_read := 0
|
||||
if is_atty(0) > 0 {
|
||||
C.ReadConsole(h_input, buf, max_line_chars * 2, C.LPDWORD(&bytes_read),
|
||||
0)
|
||||
return string_from_wide2(&u16(buf), bytes_read)
|
||||
}
|
||||
mut offset := 0
|
||||
for {
|
||||
pos := buf + offset
|
||||
res := C.ReadFile(h_input, pos, 1, C.LPDWORD(&bytes_read), 0)
|
||||
if !res || bytes_read == 0 {
|
||||
break
|
||||
}
|
||||
if *pos == `\n` || *pos == `\r` {
|
||||
offset++
|
||||
break
|
||||
}
|
||||
offset++
|
||||
}
|
||||
return buf.vstring_with_len(offset)
|
||||
}
|
||||
} $else {
|
||||
max := size_t(0)
|
||||
mut buf := charptr(0)
|
||||
nr_chars := C.getline(&buf, &max, C.stdin)
|
||||
// defer { unsafe{ free(buf) } }
|
||||
if nr_chars == 0 || nr_chars == -1 {
|
||||
return ''
|
||||
}
|
||||
return tos3(buf)
|
||||
// res := tos_clone(buf)
|
||||
// return res
|
||||
}
|
||||
}
|
||||
|
||||
// get_raw_stdin will get the raw input from stdin.
|
||||
pub fn get_raw_stdin() []byte {
|
||||
$if windows {
|
||||
unsafe {
|
||||
block_bytes := 512
|
||||
mut buf := malloc(block_bytes)
|
||||
h_input := C.GetStdHandle(std_input_handle)
|
||||
mut bytes_read := 0
|
||||
mut offset := 0
|
||||
for {
|
||||
pos := buf + offset
|
||||
res := C.ReadFile(h_input, pos, block_bytes, C.LPDWORD(&bytes_read), 0)
|
||||
offset += bytes_read
|
||||
if !res {
|
||||
break
|
||||
}
|
||||
buf = v_realloc(buf, offset + block_bytes + (block_bytes - bytes_read))
|
||||
}
|
||||
C.CloseHandle(h_input)
|
||||
return array{
|
||||
element_size: 1
|
||||
data: voidptr(buf)
|
||||
len: offset
|
||||
cap: offset
|
||||
}
|
||||
}
|
||||
} $else {
|
||||
panic('get_raw_stdin not implemented on this platform...')
|
||||
}
|
||||
}
|
||||
|
||||
// read_file_array reads an array of `T` values from file `path`.
|
||||
pub fn read_file_array<T>(path string) []T {
|
||||
a := T{}
|
||||
tsize := int(sizeof(a))
|
||||
// prepare for reading, get current file size
|
||||
mut fp := vfopen(path, 'rb') or { return array{} }
|
||||
C.fseek(fp, 0, C.SEEK_END)
|
||||
fsize := C.ftell(fp)
|
||||
C.rewind(fp)
|
||||
// read the actual data from the file
|
||||
len := fsize / tsize
|
||||
buf := malloc(fsize)
|
||||
C.fread(buf, fsize, 1, fp)
|
||||
C.fclose(fp)
|
||||
return array{
|
||||
element_size: tsize
|
||||
data: buf
|
||||
len: len
|
||||
cap: len
|
||||
}
|
||||
}
|
||||
|
||||
pub fn on_segfault(f voidptr) {
|
||||
$if windows {
|
||||
return
|
||||
}
|
||||
$if macos {
|
||||
C.printf('TODO')
|
||||
/*
|
||||
mut sa := C.sigaction{}
|
||||
C.memset(&sa, 0, sizeof(C.sigaction_size))
|
||||
C.sigemptyset(&sa.sa_mask)
|
||||
sa.sa_sigaction = f
|
||||
sa.sa_flags = C.SA_SIGINFO
|
||||
C.sigaction(C.SIGSEGV, &sa, 0)
|
||||
*/
|
||||
}
|
||||
}
|
||||
|
||||
// executable returns the path name of the executable that started the current
|
||||
// process.
|
||||
pub fn executable() string {
|
||||
$if linux {
|
||||
mut result := vcalloc(max_path_len)
|
||||
count := C.readlink('/proc/self/exe', charptr(result), max_path_len)
|
||||
if count < 0 {
|
||||
eprintln('os.executable() failed at reading /proc/self/exe to get exe path')
|
||||
return executable_fallback()
|
||||
}
|
||||
return unsafe {result.vstring()}
|
||||
}
|
||||
$if windows {
|
||||
max := 512
|
||||
size := max * 2 // max_path_len * sizeof(wchar_t)
|
||||
mut result := &u16(vcalloc(size))
|
||||
len := C.GetModuleFileName(0, result, max)
|
||||
// determine if the file is a windows symlink
|
||||
attrs := C.GetFileAttributesW(result)
|
||||
is_set := attrs & 0x400 // FILE_ATTRIBUTE_REPARSE_POINT
|
||||
if is_set != 0 { // it's a windows symlink
|
||||
// gets handle with GENERIC_READ, FILE_SHARE_READ, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0
|
||||
file := C.CreateFile(result, 0x80000000, 1, 0, 3, 0x80, 0)
|
||||
if file != voidptr(-1) {
|
||||
final_path := &u16(vcalloc(size))
|
||||
// https://docs.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-getfinalpathnamebyhandlew
|
||||
final_len := C.GetFinalPathNameByHandleW(file, final_path, size, 0)
|
||||
if final_len < size {
|
||||
ret := string_from_wide2(final_path, final_len)
|
||||
// remove '\\?\' from beginning (see link above)
|
||||
return ret[4..]
|
||||
} else {
|
||||
eprintln('os.executable() saw that the executable file path was too long')
|
||||
}
|
||||
}
|
||||
C.CloseHandle(file)
|
||||
}
|
||||
return string_from_wide2(result, len)
|
||||
}
|
||||
$if macos {
|
||||
mut result := vcalloc(max_path_len)
|
||||
pid := C.getpid()
|
||||
ret := proc_pidpath(pid, result, max_path_len)
|
||||
if ret <= 0 {
|
||||
eprintln('os.executable() failed at calling proc_pidpath with pid: $pid . proc_pidpath returned $ret ')
|
||||
return executable_fallback()
|
||||
}
|
||||
return unsafe {result.vstring()}
|
||||
}
|
||||
$if freebsd {
|
||||
mut result := vcalloc(max_path_len)
|
||||
mib := [1 /* CTL_KERN */, 14 /* KERN_PROC */, 12 /* KERN_PROC_PATHNAME */, -1]
|
||||
size := max_path_len
|
||||
unsafe {C.sysctl(mib.data, 4, result, &size, 0, 0)}
|
||||
return unsafe {result.vstring()}
|
||||
}
|
||||
// "Sadly there is no way to get the full path of the executed file in OpenBSD."
|
||||
$if openbsd {
|
||||
}
|
||||
$if solaris {
|
||||
}
|
||||
$if haiku {
|
||||
}
|
||||
$if netbsd {
|
||||
mut result := vcalloc(max_path_len)
|
||||
count := C.readlink('/proc/curproc/exe', charptr(result), max_path_len)
|
||||
if count < 0 {
|
||||
eprintln('os.executable() failed at reading /proc/curproc/exe to get exe path')
|
||||
return executable_fallback()
|
||||
}
|
||||
return unsafe {result.vstring_with_len(count)}
|
||||
}
|
||||
$if dragonfly {
|
||||
mut result := vcalloc(max_path_len)
|
||||
count := C.readlink('/proc/curproc/file', charptr(result), max_path_len)
|
||||
if count < 0 {
|
||||
eprintln('os.executable() failed at reading /proc/curproc/file to get exe path')
|
||||
return executable_fallback()
|
||||
}
|
||||
return unsafe {result.vstring_with_len(count)}
|
||||
}
|
||||
return executable_fallback()
|
||||
}
|
||||
|
||||
// is_dir returns a `bool` indicating whether the given `path` is a directory.
|
||||
pub fn is_dir(path string) bool {
|
||||
$if windows {
|
||||
w_path := path.replace('/', '\\')
|
||||
attr := C.GetFileAttributesW(w_path.to_wide())
|
||||
if attr == u32(C.INVALID_FILE_ATTRIBUTES) {
|
||||
return false
|
||||
}
|
||||
if int(attr) & C.FILE_ATTRIBUTE_DIRECTORY != 0 {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
} $else {
|
||||
statbuf := C.stat{}
|
||||
if unsafe {C.stat(charptr(path.str), &statbuf)} != 0 {
|
||||
return false
|
||||
}
|
||||
// ref: https://code.woboq.org/gcc/include/sys/stat.h.html
|
||||
val := int(statbuf.st_mode) & s_ifmt
|
||||
return val == s_ifdir
|
||||
}
|
||||
}
|
||||
|
||||
// is_link returns a boolean indicating whether `path` is a link.
|
||||
pub fn is_link(path string) bool {
|
||||
$if windows {
|
||||
return false // TODO
|
||||
} $else {
|
||||
statbuf := C.stat{}
|
||||
if C.lstat(charptr(path.str), &statbuf) != 0 {
|
||||
return false
|
||||
}
|
||||
return int(statbuf.st_mode) & s_ifmt == s_iflnk
|
||||
}
|
||||
}
|
||||
|
||||
// chdir changes the current working directory to the new directory in `path`.
|
||||
pub fn chdir(path string) {
|
||||
$if windows {
|
||||
C._wchdir(path.to_wide())
|
||||
} $else {
|
||||
C.chdir(charptr(path.str))
|
||||
}
|
||||
}
|
||||
|
||||
// getwd returns the absolute path of the current directory.
|
||||
pub fn getwd() string {
|
||||
$if windows {
|
||||
max := 512 // max_path_len * sizeof(wchar_t)
|
||||
buf := &u16(vcalloc(max * 2))
|
||||
if C._wgetcwd(buf, max) == 0 {
|
||||
return ''
|
||||
}
|
||||
return string_from_wide(buf)
|
||||
} $else {
|
||||
buf := vcalloc(512)
|
||||
if C.getcwd(charptr(buf), 512) == 0 {
|
||||
return ''
|
||||
}
|
||||
return unsafe {buf.vstring()}
|
||||
}
|
||||
}
|
||||
|
||||
// real_path returns the full absolute path for fpath, with all relative ../../, symlinks and so on resolved.
|
||||
// See http://pubs.opengroup.org/onlinepubs/9699919799/functions/realpath.html
|
||||
// Also https://insanecoding.blogspot.com/2007/11/pathmax-simply-isnt.html
|
||||
// and https://insanecoding.blogspot.com/2007/11/implementing-realpath-in-c.html
|
||||
// NB: this particular rabbit hole is *deep* ...
|
||||
pub fn real_path(fpath string) string {
|
||||
mut fullpath := vcalloc(max_path_len)
|
||||
mut ret := charptr(0)
|
||||
$if windows {
|
||||
ret = charptr(C._fullpath(charptr(fullpath), charptr(fpath.str), max_path_len))
|
||||
if ret == 0 {
|
||||
return fpath
|
||||
}
|
||||
} $else {
|
||||
ret = charptr(C.realpath(charptr(fpath.str), charptr(fullpath)))
|
||||
if ret == 0 {
|
||||
return fpath
|
||||
}
|
||||
}
|
||||
res := unsafe {fullpath.vstring()}
|
||||
return normalize_drive_letter(res)
|
||||
}
|
||||
|
||||
fn normalize_drive_letter(path string) string {
|
||||
// normalize_drive_letter is needed, because a path like c:\nv\.bin (note the small `c`)
|
||||
// in %PATH is NOT recognized by cmd.exe (and probably other programs too)...
|
||||
// Capital drive letters do work fine.
|
||||
$if !windows {
|
||||
return path
|
||||
}
|
||||
if path.len > 2 &&
|
||||
path[0] >= `a` && path[0] <= `z` && path[1] == `:` && path[2] == path_separator[0] {
|
||||
unsafe {
|
||||
x := &path.str[0]
|
||||
(*x) = *x - 32
|
||||
}
|
||||
}
|
||||
return path
|
||||
}
|
||||
|
||||
// signal will assign `handler` callback to be called when `signum` signal is recieved.
|
||||
pub fn signal(signum int, handler voidptr) {
|
||||
unsafe {C.signal(signum, handler)}
|
||||
}
|
||||
|
||||
// fork will fork the current system process and return the pid of the fork.
|
||||
pub fn fork() int {
|
||||
mut pid := -1
|
||||
$if !windows {
|
||||
pid = C.fork()
|
||||
}
|
||||
$if windows {
|
||||
panic('os.fork not supported in windows') // TODO
|
||||
}
|
||||
return pid
|
||||
}
|
||||
|
||||
// wait blocks the calling process until one of its child processes exits or a signal is received.
|
||||
// After child process terminates, parent continues its execution after wait system call instruction.
|
||||
pub fn wait() int {
|
||||
mut pid := -1
|
||||
$if !windows {
|
||||
pid = C.wait(0)
|
||||
}
|
||||
$if windows {
|
||||
panic('os.wait not supported in windows') // TODO
|
||||
}
|
||||
return pid
|
||||
}
|
||||
|
||||
// file_last_mod_unix returns the "last modified" time stamp of file in `path`.
|
||||
pub fn file_last_mod_unix(path string) int {
|
||||
attr := C.stat{}
|
||||
// # struct stat attr;
|
||||
unsafe {C.stat(charptr(path.str), &attr)}
|
||||
// # stat(path.str, &attr);
|
||||
return attr.st_mtime
|
||||
// # return attr.st_mtime ;
|
||||
}
|
||||
|
||||
// flush will flush the stdout buffer.
|
||||
pub fn flush() {
|
||||
C.fflush(C.stdout)
|
||||
}
|
||||
|
||||
// chmod change file access attributes of `path` to `mode`.
|
||||
// Octals like `0o600` can be used.
|
||||
pub fn chmod(path string, mode int) {
|
||||
C.chmod(charptr(path.str), mode)
|
||||
}
|
||||
|
||||
// open tries to open a file for reading and returns back a read-only `File` object.
|
||||
pub fn open(path string) ?File {
|
||||
/*
|
||||
$if linux {
|
||||
$if !android {
|
||||
fd := C.syscall(sys_open, path.str, 511)
|
||||
if fd == -1 {
|
||||
return error('failed to open file "$path"')
|
||||
}
|
||||
return File{
|
||||
fd: fd
|
||||
is_opened: true
|
||||
}
|
||||
}
|
||||
}
|
||||
*/
|
||||
cfile := vfopen(path, 'rb') ?
|
||||
fd := fileno(cfile)
|
||||
return File{
|
||||
cfile: cfile
|
||||
fd: fd
|
||||
is_opened: true
|
||||
}
|
||||
}
|
||||
|
||||
// create creates or opens a file at a specified location and returns a write-only `File` object.
|
||||
pub fn create(path string) ?File {
|
||||
/*
|
||||
// NB: android/termux/bionic is also a kind of linux,
|
||||
// but linux syscalls there sometimes fail,
|
||||
// while the libc version should work.
|
||||
$if linux {
|
||||
$if !android {
|
||||
//$if macos {
|
||||
// fd = C.syscall(398, path.str, 0x601, 0x1b6)
|
||||
//}
|
||||
//$if linux {
|
||||
fd = C.syscall(sys_creat, path.str, 511)
|
||||
//}
|
||||
if fd == -1 {
|
||||
return error('failed to create file "$path"')
|
||||
}
|
||||
file = File{
|
||||
fd: fd
|
||||
is_opened: true
|
||||
}
|
||||
return file
|
||||
}
|
||||
}
|
||||
*/
|
||||
cfile := vfopen(path, 'wb') ?
|
||||
fd := fileno(cfile)
|
||||
return File{
|
||||
cfile: cfile
|
||||
fd: fd
|
||||
is_opened: true
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue