Browse Source

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

gingerBill 2 years ago
parent
commit
145a7a24e8
4 changed files with 131 additions and 104 deletions
  1. 14 28
      core/os/os2/file.odin
  2. 55 11
      core/os/os2/file_linux.odin
  3. 4 54
      core/os/os2/file_stream.odin
  4. 58 11
      core/os/os2/file_windows.odin

+ 14 - 28
core/os/os2/file.odin

@@ -5,13 +5,8 @@ import "core:time"
 import "core:runtime"
 import "core:runtime"
 
 
 File :: struct {
 File :: struct {
-	impl: _File,
-}
-
-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
+	impl:   _File,
+	stream: io.Stream,
 }
 }
 
 
 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) {
-	return _seek(f, offset, whence)
+close :: proc(f: ^File) -> Error {
+	return io.close(f.stream)
 }
 }
 
 
-read :: proc(f: ^File, p: []byte) -> (n: int, err: Error) {
-	return _read(f, p)
+seek :: proc(f: ^File, offset: i64, whence: io.Seek_From) -> (ret: i64, err: Error) {
+	return io.seek(f.stream, offset, whence)
 }
 }
 
 
-read_at :: proc(f: ^File, p: []byte, offset: i64) -> (n: int, err: Error) {
-	return _read_at(f, p, offset)
+read :: proc(f: ^File, p: []byte) -> (n: int, err: Error) {
+	return io.read(f.stream, p)
 }
 }
 
 
-read_from :: proc(f: ^File, r: io.Reader) -> (n: i64, err: Error) {
-	return _read_from(f, r)
+read_at :: proc(f: ^File, p: []byte, offset: i64) -> (n: int, err: Error) {
+	return io.read_at(f.stream, p, offset)
 }
 }
 
 
 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)
-}
-
-write_to :: proc(f: ^File, w: io.Writer) -> (n: i64, err: Error) {
-	return _write_to(f, w)
+	return io.write_at(f.stream, p, offset)
 }
 }
 
 
 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
core/os/os2/file_linux.odin

@@ -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
core/os/os2/file_stream.odin

@@ -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
-	s.procedure = _file_stream_proc
+	if f != nil {
+		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
core/os/os2/file_windows.odin

@@ -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
+}
+