Use io.Stream as the internal interface for os2.File

This commit is contained in:
gingerBill
2023-06-08 16:55:01 +01:00
parent 6c040497ef
commit 145a7a24e8
4 changed files with 131 additions and 104 deletions
+14 -28
View File
@@ -5,13 +5,8 @@ import "core:time"
import "core:runtime" import "core:runtime"
File :: struct { File :: struct {
impl: _File, impl: _File,
} stream: io.Stream,
Seek_From :: enum {
Start = 0, // seek relative to the origin of the file
Current = 1, // seek relative to the current offset
End = 2, // seek relative to the end
} }
File_Mode :: distinct u32 File_Mode :: distinct u32
@@ -72,45 +67,36 @@ fd :: proc(f: ^File) -> uintptr {
return _fd(f) return _fd(f)
} }
close :: proc(f: ^File) -> Error {
return _close(f)
}
name :: proc(f: ^File) -> string { name :: proc(f: ^File) -> string {
return _name(f) return _name(f)
} }
seek :: proc(f: ^File, offset: i64, whence: Seek_From) -> (ret: i64, err: Error) { close :: proc(f: ^File) -> Error {
return _seek(f, offset, whence) return io.close(f.stream)
}
seek :: proc(f: ^File, offset: i64, whence: io.Seek_From) -> (ret: i64, err: Error) {
return io.seek(f.stream, offset, whence)
} }
read :: proc(f: ^File, p: []byte) -> (n: int, err: Error) { read :: proc(f: ^File, p: []byte) -> (n: int, err: Error) {
return _read(f, p) return io.read(f.stream, p)
} }
read_at :: proc(f: ^File, p: []byte, offset: i64) -> (n: int, err: Error) { read_at :: proc(f: ^File, p: []byte, offset: i64) -> (n: int, err: Error) {
return _read_at(f, p, offset) return io.read_at(f.stream, p, offset)
}
read_from :: proc(f: ^File, r: io.Reader) -> (n: i64, err: Error) {
return _read_from(f, r)
} }
write :: proc(f: ^File, p: []byte) -> (n: int, err: Error) { write :: proc(f: ^File, p: []byte) -> (n: int, err: Error) {
return _write(f, p) return io.write(f.stream, p)
} }
write_at :: proc(f: ^File, p: []byte, offset: i64) -> (n: int, err: Error) { write_at :: proc(f: ^File, p: []byte, offset: i64) -> (n: int, err: Error) {
return _write_at(f, p, offset) return io.write_at(f.stream, p, offset)
}
write_to :: proc(f: ^File, w: io.Writer) -> (n: i64, err: Error) {
return _write_to(f, w)
} }
file_size :: proc(f: ^File) -> (n: i64, err: Error) { file_size :: proc(f: ^File) -> (n: i64, err: Error) {
return _file_size(f) return io.size(f.stream)
} }
@@ -119,7 +105,7 @@ sync :: proc(f: ^File) -> Error {
} }
flush :: proc(f: ^File) -> Error { flush :: proc(f: ^File) -> Error {
return _flush(f) return io.flush(f.stream)
} }
truncate :: proc(f: ^File, size: i64) -> Error { truncate :: proc(f: ^File, size: i64) -> Error {
+55 -11
View File
@@ -73,6 +73,10 @@ _new_file :: proc(fd: uintptr, _: string) -> ^File {
file.impl.fd = int(fd) file.impl.fd = int(fd)
file.impl.allocator = _file_allocator() file.impl.allocator = _file_allocator()
file.impl.name = _get_full_path(file.impl.fd, file.impl.allocator) file.impl.name = _get_full_path(file.impl.fd, file.impl.allocator)
file.stream = {
data = file,
procedure = _file_stream_proc,
}
return file return file
} }
@@ -102,7 +106,7 @@ _name :: proc(f: ^File) -> string {
return f.impl.name if f != nil else "" return f.impl.name if f != nil else ""
} }
_seek :: proc(f: ^File, offset: i64, whence: Seek_From) -> (ret: i64, err: Error) { _seek :: proc(f: ^File, offset: i64, whence: io.Seek_From) -> (ret: i64, err: Error) {
res := unix.sys_lseek(f.impl.fd, offset, int(whence)) res := unix.sys_lseek(f.impl.fd, offset, int(whence))
if res < 0 { if res < 0 {
return -1, _get_platform_error(int(res)) return -1, _get_platform_error(int(res))
@@ -139,11 +143,6 @@ _read_at :: proc(f: ^File, p: []byte, offset: i64) -> (n: int, err: Error) {
return return
} }
_read_from :: proc(f: ^File, r: io.Reader) -> (n: i64, err: Error) {
//TODO
return
}
_write :: proc(f: ^File, p: []byte) -> (n: int, err: Error) { _write :: proc(f: ^File, p: []byte) -> (n: int, err: Error) {
if len(p) == 0 { if len(p) == 0 {
return 0, nil return 0, nil
@@ -173,11 +172,6 @@ _write_at :: proc(f: ^File, p: []byte, offset: i64) -> (n: int, err: Error) {
return return
} }
_write_to :: proc(f: ^File, w: io.Writer) -> (n: i64, err: Error) {
//TODO
return
}
_file_size :: proc(f: ^File) -> (n: i64, err: Error) { _file_size :: proc(f: ^File) -> (n: i64, err: Error) {
s: _Stat = --- s: _Stat = ---
res := unix.sys_fstat(f.impl.fd, &s) res := unix.sys_fstat(f.impl.fd, &s)
@@ -366,3 +360,53 @@ _is_dir_fd :: proc(fd: int) -> bool {
_temp_name_to_cstring :: proc(name: string) -> (cname: cstring) { _temp_name_to_cstring :: proc(name: string) -> (cname: cstring) {
return strings.clone_to_cstring(name, context.temp_allocator) return strings.clone_to_cstring(name, context.temp_allocator)
} }
@(private="package")
_file_stream_proc :: proc(stream_data: rawptr, mode: io.Stream_Mode, p: []byte, offset: i64, whence: io.Seek_From) -> (n: i64, err: io.Error) {
f := (^File)(stream_data)
ferr: Error
i: int
switch mode {
case .Read:
i, ferr = _read(f, p)
n = i64(i)
err = error_to_io_error(ferr)
return
case .Read_At:
i, ferr = _read_at(f, p, offset)
n = i64(i)
err = error_to_io_error(ferr)
return
case .Write:
i, ferr = _write(f, p)
n = i64(i)
err = error_to_io_error(ferr)
return
case .Write_At:
i, ferr = _write_at(f, p, offset)
n = i64(i)
err = error_to_io_error(ferr)
return
case .Seek:
n, ferr = _seek(f, offset, whence)
err = error_to_io_error(ferr)
return
case .Size:
n, ferr = _file_size(f)
err = error_to_io_error(ferr)
return
case .Flush:
ferr = _flush(f)
err = error_to_io_error(ferr)
return
case .Close, .Destroy:
ferr = _close(f)
err = error_to_io_error(ferr)
return
case .Query:
return io.query_utility({.Read, .Read_At, .Write, .Write_At, .Seek, .Size, .Flush, .Close, .Destroy, .Query})
}
return 0, .Empty
}
+4 -54
View File
@@ -3,8 +3,10 @@ package os2
import "core:io" import "core:io"
to_stream :: proc(f: ^File) -> (s: io.Stream) { to_stream :: proc(f: ^File) -> (s: io.Stream) {
s.data = f if f != nil {
s.procedure = _file_stream_proc assert(f.stream.procedure != nil)
s = f.stream
}
return return
} }
@@ -19,55 +21,3 @@ error_to_io_error :: proc(ferr: Error) -> io.Error {
} }
return ferr.(io.Error) or_else .Unknown return ferr.(io.Error) or_else .Unknown
} }
@(private)
_file_stream_proc :: proc(stream_data: rawptr, mode: io.Stream_Mode, p: []byte, offset: i64, whence: io.Seek_From) -> (n: i64, err: io.Error) {
f := (^File)(stream_data)
ferr: Error
i: int
switch mode {
case .Read:
i, ferr = read(f, p)
n = i64(i)
err = error_to_io_error(ferr)
return
case .Read_At:
i, ferr = read_at(f, p, offset)
n = i64(i)
err = error_to_io_error(ferr)
return
case .Write:
i, ferr = write(f, p)
n = i64(i)
err = error_to_io_error(ferr)
return
case .Write_At:
i, ferr = write_at(f, p, offset)
n = i64(i)
err = error_to_io_error(ferr)
return
case .Seek:
n, ferr = seek(f, offset, Seek_From(whence))
err = error_to_io_error(ferr)
return
case .Size:
n, ferr = file_size(f)
err = error_to_io_error(ferr)
return
case .Flush:
ferr = flush(f)
err = error_to_io_error(ferr)
return
case .Close:
ferr = close(f)
err = error_to_io_error(ferr)
return
case .Query:
return io.query_utility({.Read, .Read_At, .Write, .Write_At, .Seek, .Size, .Flush, .Close, .Query})
case .Destroy:
return 0, .Empty
}
return 0, .Empty
}
+58 -11
View File
@@ -144,6 +144,11 @@ _new_file :: proc(handle: uintptr, name: string) -> ^File {
} }
f.impl.kind = kind f.impl.kind = kind
f.stream = {
data = f,
procedure = _file_stream_proc,
}
return f return f
} }
@@ -181,7 +186,7 @@ _name :: proc(f: ^File) -> string {
return f.impl.name if f != nil else "" return f.impl.name if f != nil else ""
} }
_seek :: proc(f: ^File, offset: i64, whence: Seek_From) -> (ret: i64, err: Error) { _seek :: proc(f: ^File, offset: i64, whence: io.Seek_From) -> (ret: i64, err: Error) {
handle := _handle(f) handle := _handle(f)
if handle == win32.INVALID_HANDLE { if handle == win32.INVALID_HANDLE {
return 0, .Invalid_File return 0, .Invalid_File
@@ -329,11 +334,6 @@ _read_at :: proc(f: ^File, p: []byte, offset: i64) -> (n: int, err: Error) {
return return
} }
_read_from :: proc(f: ^File, r: io.Reader) -> (n: i64, err: Error) {
// TODO(bill)
return
}
_write :: proc(f: ^File, p: []byte) -> (n: int, err: Error) { _write :: proc(f: ^File, p: []byte) -> (n: int, err: Error) {
if len(p) == 0 { if len(p) == 0 {
return return
@@ -397,11 +397,6 @@ _write_at :: proc(f: ^File, p: []byte, offset: i64) -> (n: int, err: Error) {
return return
} }
_write_to :: proc(f: ^File, w: io.Writer) -> (n: i64, err: Error) {
// TODO(bill)
return
}
_file_size :: proc(f: ^File) -> (n: i64, err: Error) { _file_size :: proc(f: ^File) -> (n: i64, err: Error) {
length: win32.LARGE_INTEGER length: win32.LARGE_INTEGER
handle := _handle(f) handle := _handle(f)
@@ -727,3 +722,55 @@ _is_dir :: proc(path: string) -> bool {
} }
return false return false
} }
@(private="package")
_file_stream_proc :: proc(stream_data: rawptr, mode: io.Stream_Mode, p: []byte, offset: i64, whence: io.Seek_From) -> (n: i64, err: io.Error) {
f := (^File)(stream_data)
ferr: Error
i: int
switch mode {
case .Read:
i, ferr = _read(f, p)
n = i64(i)
err = error_to_io_error(ferr)
return
case .Read_At:
i, ferr = _read_at(f, p, offset)
n = i64(i)
err = error_to_io_error(ferr)
return
case .Write:
i, ferr = _write(f, p)
n = i64(i)
err = error_to_io_error(ferr)
return
case .Write_At:
i, ferr = _write_at(f, p, offset)
n = i64(i)
err = error_to_io_error(ferr)
return
case .Seek:
n, ferr = _seek(f, offset, whence)
err = error_to_io_error(ferr)
return
case .Size:
n, ferr = _file_size(f)
err = error_to_io_error(ferr)
return
case .Flush:
ferr = _flush(f)
err = error_to_io_error(ferr)
return
case .Close:
ferr = _close(f)
err = error_to_io_error(ferr)
return
case .Query:
return io.query_utility({.Read, .Read_At, .Write, .Write_At, .Seek, .Size, .Flush, .Close, .Query})
case .Destroy:
return 0, .Empty
}
return 0, .Empty
}