stream.odin 1.4 KB

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