stream.odin 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  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. case .Read_At:
  27. when !(ODIN_OS == .FreeBSD || ODIN_OS == .OpenBSD) {
  28. n_int, os_err = read_at(fd, p, offset)
  29. n = i64(n_int)
  30. }
  31. case .Write:
  32. n_int, os_err = write(fd, p)
  33. n = i64(n_int)
  34. case .Write_At:
  35. when !(ODIN_OS == .FreeBSD || ODIN_OS == .OpenBSD) {
  36. n_int, os_err = write_at(fd, p, offset)
  37. n = i64(n_int)
  38. }
  39. case .Seek:
  40. n, os_err = seek(fd, offset, int(whence))
  41. case .Size:
  42. n, os_err = file_size(fd)
  43. case .Destroy:
  44. err = .Empty
  45. case .Query:
  46. when ODIN_OS == .FreeBSD || ODIN_OS == .OpenBSD {
  47. return io.query_utility({.Close, .Flush, .Read, .Write, .Seek, .Size, .Query})
  48. } else {
  49. return io.query_utility({.Close, .Flush, .Read, .Read_At, .Write, .Write_At, .Seek, .Size, .Query})
  50. }
  51. }
  52. if err == nil && os_err != 0 {
  53. when ODIN_OS == .Windows {
  54. if os_err == ERROR_HANDLE_EOF {
  55. return n, .EOF
  56. }
  57. }
  58. err = .Unknown
  59. }
  60. return
  61. }