mirror of
https://github.com/Ed94/Odin.git
synced 2026-06-16 19:02:23 -07:00
gingerBill
319 lines
8.0 KiB
Odin
319 lines
8.0 KiB
Odin
package os
|
|
|
|
import "core:time"
|
|
import win32 "core:sys/windows"
|
|
|
|
@(private)
|
|
full_path_from_name :: proc(name: string, allocator := context.allocator) -> (path: string, err: Errno) {
|
|
context.allocator = allocator
|
|
|
|
name := name
|
|
if name == "" {
|
|
name = "."
|
|
}
|
|
p := win32.utf8_to_utf16(name, context.temp_allocator)
|
|
buf := make([dynamic]u16, 100)
|
|
defer delete(buf)
|
|
for {
|
|
n := win32.GetFullPathNameW(raw_data(p), u32(len(buf)), raw_data(buf), nil)
|
|
if n == 0 {
|
|
return "", Errno(win32.GetLastError())
|
|
}
|
|
if n <= u32(len(buf)) {
|
|
return win32.utf16_to_utf8(buf[:n], allocator), ERROR_NONE
|
|
}
|
|
resize(&buf, len(buf)*2)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
@(private)
|
|
_stat :: proc(name: string, create_file_attributes: u32, allocator := context.allocator) -> (fi: File_Info, e: Errno) {
|
|
if len(name) == 0 {
|
|
return {}, ERROR_PATH_NOT_FOUND
|
|
}
|
|
|
|
context.allocator = allocator
|
|
|
|
|
|
wname := win32.utf8_to_wstring(fix_long_path(name), context.temp_allocator)
|
|
fa: win32.WIN32_FILE_ATTRIBUTE_DATA
|
|
ok := win32.GetFileAttributesExW(wname, win32.GetFileExInfoStandard, &fa)
|
|
if ok && fa.dwFileAttributes & win32.FILE_ATTRIBUTE_REPARSE_POINT == 0 {
|
|
// Not a symlink
|
|
return file_info_from_win32_file_attribute_data(&fa, name)
|
|
}
|
|
|
|
err := 0 if ok else win32.GetLastError()
|
|
|
|
if err == win32.ERROR_SHARING_VIOLATION {
|
|
fd: win32.WIN32_FIND_DATAW
|
|
sh := win32.FindFirstFileW(wname, &fd)
|
|
if sh == win32.INVALID_HANDLE_VALUE {
|
|
e = Errno(win32.GetLastError())
|
|
return
|
|
}
|
|
win32.FindClose(sh)
|
|
|
|
return file_info_from_win32_find_data(&fd, name)
|
|
}
|
|
|
|
h := win32.CreateFileW(wname, 0, 0, nil, win32.OPEN_EXISTING, create_file_attributes, nil)
|
|
if h == win32.INVALID_HANDLE_VALUE {
|
|
e = Errno(win32.GetLastError())
|
|
return
|
|
}
|
|
defer win32.CloseHandle(h)
|
|
return file_info_from_get_file_information_by_handle(name, h)
|
|
}
|
|
|
|
|
|
lstat :: proc(name: string, allocator := context.allocator) -> (File_Info, Errno) {
|
|
attrs := win32.FILE_FLAG_BACKUP_SEMANTICS
|
|
attrs |= win32.FILE_FLAG_OPEN_REPARSE_POINT
|
|
return _stat(name, attrs, allocator)
|
|
}
|
|
|
|
stat :: proc(name: string, allocator := context.allocator) -> (File_Info, Errno) {
|
|
attrs := win32.FILE_FLAG_BACKUP_SEMANTICS
|
|
return _stat(name, attrs, allocator)
|
|
}
|
|
|
|
fstat :: proc(fd: Handle, allocator := context.allocator) -> (File_Info, Errno) {
|
|
if fd == 0 {
|
|
return {}, ERROR_INVALID_HANDLE
|
|
}
|
|
context.allocator = allocator
|
|
|
|
path, err := cleanpath_from_handle(fd)
|
|
if err != ERROR_NONE {
|
|
return {}, err
|
|
}
|
|
|
|
h := win32.HANDLE(fd)
|
|
switch win32.GetFileType(h) {
|
|
case win32.FILE_TYPE_PIPE, win32.FILE_TYPE_CHAR:
|
|
fi: File_Info
|
|
fi.fullpath = path
|
|
fi.name = basename(path)
|
|
fi.mode |= file_type_mode(h)
|
|
return fi, ERROR_NONE
|
|
}
|
|
|
|
return file_info_from_get_file_information_by_handle(path, h)
|
|
}
|
|
|
|
|
|
@(private)
|
|
cleanpath_strip_prefix :: proc(buf: []u16) -> []u16 {
|
|
buf := buf
|
|
N := 0
|
|
for c, i in buf {
|
|
if c == 0 { break }
|
|
N = i+1
|
|
}
|
|
buf = buf[:N]
|
|
|
|
if len(buf) >= 4 {
|
|
if buf[0] == '\\' &&
|
|
buf[1] == '\\' &&
|
|
buf[2] == '?' &&
|
|
buf[3] == '\\' {
|
|
buf = buf[4:]
|
|
}
|
|
}
|
|
return buf
|
|
}
|
|
|
|
@(private)
|
|
cleanpath_from_handle :: proc(fd: Handle) -> (string, Errno) {
|
|
if fd == 0 {
|
|
return "", ERROR_INVALID_HANDLE
|
|
}
|
|
h := win32.HANDLE(fd)
|
|
|
|
MAX_PATH := win32.DWORD(260) + 1
|
|
buf: []u16
|
|
for {
|
|
buf = make([]u16, MAX_PATH, context.temp_allocator)
|
|
err := win32.GetFinalPathNameByHandleW(h, raw_data(buf), MAX_PATH, 0)
|
|
switch Errno(err) {
|
|
case ERROR_PATH_NOT_FOUND, ERROR_INVALID_PARAMETER:
|
|
return "", Errno(err)
|
|
case ERROR_NOT_ENOUGH_MEMORY:
|
|
MAX_PATH = MAX_PATH*2 + 1
|
|
continue
|
|
}
|
|
break
|
|
}
|
|
return cleanpath_from_buf(buf), ERROR_NONE
|
|
}
|
|
@(private)
|
|
cleanpath_from_handle_u16 :: proc(fd: Handle) -> ([]u16, Errno) {
|
|
if fd == 0 {
|
|
return nil, ERROR_INVALID_HANDLE
|
|
}
|
|
h := win32.HANDLE(fd)
|
|
|
|
MAX_PATH := win32.DWORD(260) + 1
|
|
buf: []u16
|
|
for {
|
|
buf = make([]u16, MAX_PATH, context.temp_allocator)
|
|
err := win32.GetFinalPathNameByHandleW(h, raw_data(buf), MAX_PATH, 0)
|
|
switch Errno(err) {
|
|
case ERROR_PATH_NOT_FOUND, ERROR_INVALID_PARAMETER:
|
|
return nil, Errno(err)
|
|
case ERROR_NOT_ENOUGH_MEMORY:
|
|
MAX_PATH = MAX_PATH*2 + 1
|
|
continue
|
|
}
|
|
break
|
|
}
|
|
return cleanpath_strip_prefix(buf), ERROR_NONE
|
|
}
|
|
@(private)
|
|
cleanpath_from_buf :: proc(buf: []u16) -> string {
|
|
buf := buf
|
|
buf = cleanpath_strip_prefix(buf)
|
|
return win32.utf16_to_utf8(buf, context.allocator)
|
|
}
|
|
|
|
@(private)
|
|
basename :: proc(name: string) -> (base: string) {
|
|
name := name
|
|
if len(name) > 3 && name[:3] == `\\?` {
|
|
name = name[3:]
|
|
}
|
|
|
|
if len(name) == 2 && name[1] == ':' {
|
|
return "."
|
|
} else if len(name) > 2 && name[1] == ':' {
|
|
name = name[2:]
|
|
}
|
|
i := len(name)-1
|
|
|
|
for ; i > 0 && (name[i] == '/' || name[i] == '\\'); i -= 1 {
|
|
name = name[:i]
|
|
}
|
|
for i -= 1; i >= 0; i -= 1 {
|
|
if name[i] == '/' || name[i] == '\\' {
|
|
name = name[i+1:]
|
|
break
|
|
}
|
|
}
|
|
return name
|
|
}
|
|
|
|
@(private)
|
|
file_type_mode :: proc(h: win32.HANDLE) -> File_Mode {
|
|
switch win32.GetFileType(h) {
|
|
case win32.FILE_TYPE_PIPE:
|
|
return File_Mode_Named_Pipe
|
|
case win32.FILE_TYPE_CHAR:
|
|
return File_Mode_Device | File_Mode_Char_Device
|
|
}
|
|
return 0
|
|
}
|
|
|
|
|
|
@(private)
|
|
file_mode_from_file_attributes :: proc(FileAttributes: win32.DWORD, h: win32.HANDLE, ReparseTag: win32.DWORD) -> (mode: File_Mode) {
|
|
if FileAttributes & win32.FILE_ATTRIBUTE_READONLY != 0 {
|
|
mode |= 0o444
|
|
} else {
|
|
mode |= 0o666
|
|
}
|
|
|
|
is_sym := false
|
|
if FileAttributes & win32.FILE_ATTRIBUTE_REPARSE_POINT == 0 {
|
|
is_sym = false
|
|
} else {
|
|
is_sym = ReparseTag == win32.IO_REPARSE_TAG_SYMLINK || ReparseTag == win32.IO_REPARSE_TAG_MOUNT_POINT
|
|
}
|
|
|
|
if is_sym {
|
|
mode |= File_Mode_Sym_Link
|
|
} else {
|
|
if FileAttributes & win32.FILE_ATTRIBUTE_DIRECTORY != 0 {
|
|
mode |= 0o111 | File_Mode_Dir
|
|
}
|
|
|
|
if h != nil {
|
|
mode |= file_type_mode(h)
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
@(private)
|
|
file_info_from_win32_file_attribute_data :: proc(d: ^win32.WIN32_FILE_ATTRIBUTE_DATA, name: string) -> (fi: File_Info, e: Errno) {
|
|
fi.size = i64(d.nFileSizeHigh)<<32 + i64(d.nFileSizeLow)
|
|
|
|
fi.mode |= file_mode_from_file_attributes(d.dwFileAttributes, nil, 0)
|
|
fi.is_dir = fi.mode & File_Mode_Dir != 0
|
|
|
|
fi.creation_time = time.unix(0, win32.FILETIME_as_unix_nanoseconds(d.ftCreationTime))
|
|
fi.modification_time = time.unix(0, win32.FILETIME_as_unix_nanoseconds(d.ftLastWriteTime))
|
|
fi.access_time = time.unix(0, win32.FILETIME_as_unix_nanoseconds(d.ftLastAccessTime))
|
|
|
|
fi.fullpath, e = full_path_from_name(name)
|
|
fi.name = basename(fi.fullpath)
|
|
|
|
return
|
|
}
|
|
|
|
@(private)
|
|
file_info_from_win32_find_data :: proc(d: ^win32.WIN32_FIND_DATAW, name: string) -> (fi: File_Info, e: Errno) {
|
|
fi.size = i64(d.nFileSizeHigh)<<32 + i64(d.nFileSizeLow)
|
|
|
|
fi.mode |= file_mode_from_file_attributes(d.dwFileAttributes, nil, 0)
|
|
fi.is_dir = fi.mode & File_Mode_Dir != 0
|
|
|
|
fi.creation_time = time.unix(0, win32.FILETIME_as_unix_nanoseconds(d.ftCreationTime))
|
|
fi.modification_time = time.unix(0, win32.FILETIME_as_unix_nanoseconds(d.ftLastWriteTime))
|
|
fi.access_time = time.unix(0, win32.FILETIME_as_unix_nanoseconds(d.ftLastAccessTime))
|
|
|
|
fi.fullpath, e = full_path_from_name(name)
|
|
fi.name = basename(fi.fullpath)
|
|
|
|
return
|
|
}
|
|
|
|
@(private)
|
|
file_info_from_get_file_information_by_handle :: proc(path: string, h: win32.HANDLE) -> (File_Info, Errno) {
|
|
d: win32.BY_HANDLE_FILE_INFORMATION
|
|
if !win32.GetFileInformationByHandle(h, &d) {
|
|
err := Errno(win32.GetLastError())
|
|
return {}, err
|
|
|
|
}
|
|
|
|
ti: win32.FILE_ATTRIBUTE_TAG_INFO
|
|
if !win32.GetFileInformationByHandleEx(h, .FileAttributeTagInfo, &ti, size_of(ti)) {
|
|
err := win32.GetLastError()
|
|
if err != u32(ERROR_INVALID_PARAMETER) {
|
|
return {}, Errno(err)
|
|
}
|
|
// Indicate this is a symlink on FAT file systems
|
|
ti.ReparseTag = 0
|
|
}
|
|
|
|
fi: File_Info
|
|
|
|
fi.fullpath = path
|
|
fi.name = basename(path)
|
|
fi.size = i64(d.nFileSizeHigh)<<32 + i64(d.nFileSizeLow)
|
|
|
|
fi.mode |= file_mode_from_file_attributes(ti.FileAttributes, h, ti.ReparseTag)
|
|
fi.is_dir = fi.mode & File_Mode_Dir != 0
|
|
|
|
fi.creation_time = time.unix(0, win32.FILETIME_as_unix_nanoseconds(d.ftCreationTime))
|
|
fi.modification_time = time.unix(0, win32.FILETIME_as_unix_nanoseconds(d.ftLastWriteTime))
|
|
fi.access_time = time.unix(0, win32.FILETIME_as_unix_nanoseconds(d.ftLastAccessTime))
|
|
|
|
|
|
return fi, ERROR_NONE
|
|
}
|