stream.odin 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  1. package os
  2. import "core:io"
  3. stream_from_handle :: proc(fd: Handle) -> io.Stream {
  4. s: io.Stream
  5. s.data = rawptr(uintptr(fd))
  6. s.procedure = _file_stream_proc
  7. return s
  8. }
  9. @(private)
  10. _file_stream_proc :: proc(stream_data: rawptr, mode: io.Stream_Mode, p: []byte, offset: i64, whence: io.Seek_From) -> (n: i64, err: io.Error) {
  11. fd := Handle(uintptr(stream_data))
  12. n_int: int
  13. os_err: Errno
  14. switch mode {
  15. case .Close:
  16. close(fd)
  17. case .Flush:
  18. when ODIN_OS == .Windows {
  19. flush(fd)
  20. } else {
  21. // TOOD(bill): other operating systems
  22. }
  23. case .Read:
  24. n_int, os_err = read(fd, p)
  25. n = i64(n_int)
  26. if n == 0 && os_err == 0 {
  27. err = .EOF
  28. }
  29. case .Read_At:
  30. when !(ODIN_OS == .FreeBSD || ODIN_OS == .OpenBSD || ODIN_OS == .Haiku) {
  31. n_int, os_err = read_at(fd, p, offset)
  32. n = i64(n_int)
  33. if n == 0 && os_err == 0 {
  34. err = .EOF
  35. }
  36. }
  37. case .Write:
  38. n_int, os_err = write(fd, p)
  39. n = i64(n_int)
  40. if n == 0 && os_err == 0 {
  41. err = .EOF
  42. }
  43. case .Write_At:
  44. when !(ODIN_OS == .FreeBSD || ODIN_OS == .OpenBSD || ODIN_OS == .Haiku) {
  45. n_int, os_err = write_at(fd, p, offset)
  46. n = i64(n_int)
  47. if n == 0 && os_err == 0 {
  48. err = .EOF
  49. }
  50. }
  51. case .Seek:
  52. n, os_err = seek(fd, offset, int(whence))
  53. case .Size:
  54. n, os_err = file_size(fd)
  55. case .Destroy:
  56. err = .Empty
  57. case .Query:
  58. when ODIN_OS == .FreeBSD || ODIN_OS == .OpenBSD || ODIN_OS == .Haiku {
  59. return io.query_utility({.Close, .Flush, .Read, .Write, .Seek, .Size, .Query})
  60. } else {
  61. return io.query_utility({.Close, .Flush, .Read, .Read_At, .Write, .Write_At, .Seek, .Size, .Query})
  62. }
  63. }
  64. if err == nil && os_err != 0 {
  65. when ODIN_OS == .Windows {
  66. if os_err == ERROR_HANDLE_EOF {
  67. return n, .EOF
  68. }
  69. }
  70. err = .Unknown
  71. }
  72. return
  73. }