...

1 files changed, 125 insertions, 0 deletions

diff --git a/src/common/io/UnixFileStream.cpp b/src/common/io/UnixFileStream.cpp
new file mode 100644
index 00000000..23c11dad
--- /dev/null
+++ b/src/common/io/UnixFileStream.cpp
@@ -0,0 +1,125 @@
+#include "cru/common/io/UnixFileStream.hpp"
+#include "cru/common/ErrnoException.hpp"
+#include "cru/common/io/OpenFileFlag.hpp"
+
+#include <fcntl.h>
+#include <unistd.h>
+
+namespace cru::io {
+
+namespace {
+int MapOpenFileFlag(OpenFileFlag flag) {
+  int result = 0;
+  if (flag & OpenFileFlags::Read) {
+    if (flag & OpenFileFlags::Write) {
+      result |= O_RDWR;
+    } else {
+      result |= O_RDONLY;
+    }
+  } else {
+    if (flag & OpenFileFlags::Write) {
+      result |= O_WRONLY;
+    } else {
+      throw Exception(u"Invalid open file flag.");
+    }
+  }
+
+  if (flag | OpenFileFlags::Append) {
+    result |= O_APPEND;
+  }
+
+  if (flag | OpenFileFlags::Create) {
+    result |= O_CREAT;
+  }
+
+  if (flag | OpenFileFlags::ThrowOnExist) {
+    result |= O_EXCL;
+  }
+
+  return result;
+}
+
+int MapSeekOrigin(Stream::SeekOrigin origin) {
+  switch (origin) {
+    case Stream::SeekOrigin::Begin:
+      return SEEK_SET;
+    case Stream::SeekOrigin::Current:
+      return SEEK_CUR;
+    case Stream::SeekOrigin::End:
+      return SEEK_END;
+    default:
+      throw Exception(u"Invalid seek origin.");
+  }
+}
+}  // namespace
+
+UnixFileStream::UnixFileStream(String path, OpenFileFlag flags) {
+  file_descriptor_ = ::open(path.ToUtf8().c_str(), MapOpenFileFlag(flags));
+  if (file_descriptor_ == -1) {
+    throw ErrnoException(u"Failed to open file.");
+  }
+}
+
+bool UnixFileStream::CanSeek() {
+  CheckClosed();
+  return true;
+}
+
+Index UnixFileStream::Tell() {
+  CheckClosed();
+  auto result = ::lseek(file_descriptor_, 0, SEEK_CUR);
+  if (result == -1) {
+    throw ErrnoException(u"Failed to get file position.");
+  }
+  return result;
+}
+
+void UnixFileStream::Seek(Index offset, SeekOrigin origin) {
+  CheckClosed();
+  int result = ::lseek(file_descriptor_, offset, MapSeekOrigin(origin));
+  if (result == -1) {
+    throw ErrnoException(u"Failed to seek file.");
+  }
+}
+
+bool UnixFileStream::CanRead() {
+  CheckClosed();
+  return flags & OpenFileFlags::Read;
+}
+
+Index UnixFileStream::Read(std::byte *buffer, Index offset, Index size) {
+  CheckClosed();
+  auto result = ::read(file_descriptor_, buffer + offset, size);
+  if (result == -1) {
+    throw ErrnoException(u"Failed to read file.");
+  }
+  return result;
+}
+
+bool UnixFileStream::CanWrite() {
+  CheckClosed();
+  return flags & OpenFileFlags::Write;
+}
+
+Index UnixFileStream::Write(const std::byte *buffer, Index offset, Index size) {
+  CheckClosed();
+  auto result = ::write(file_descriptor_, buffer + offset, size);
+  if (result == -1) {
+    throw ErrnoException(u"Failed to write file.");
+  }
+  return result;
+}
+
+void UnixFileStream::Close() {
+  if (closed_) return;
+  if (::close(file_descriptor_) == -1) {
+    throw ErrnoException(u"Failed to close file.");
+  }
+}
+
+void UnixFileStream::CheckClosed() {
+  if (closed_) {
+    throw Exception(u"File is closed.");
+  }
+}
+}  // namespace cru::io