From 5aca7b099c46a87a859f40110efce708200a4bc8 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Thu, 3 Feb 2022 17:55:00 +0800
Subject: ...

---
 src/common/platform/win/StreamConvert.cpp | 53 +++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 src/common/platform/win/StreamConvert.cpp

(limited to 'src/common/platform/win/StreamConvert.cpp')

diff --git a/src/common/platform/win/StreamConvert.cpp b/src/common/platform/win/StreamConvert.cpp
new file mode 100644
index 00000000..1d077573
--- /dev/null
+++ b/src/common/platform/win/StreamConvert.cpp
@@ -0,0 +1,53 @@
+#include "cru/common/platform/win/StreamConvert.hpp"
+#include "BrigdeComStream.hpp"
+#include "cru/common/Exception.hpp"
+#include "cru/common/io/MemoryStream.hpp"
+#include "cru/common/io/OpenFileFlag.hpp"
+#include "cru/common/platform/win/ComAutoInit.hpp"
+#include "cru/common/platform/win/Exception.hpp"
+#include "cru/common/platform/win/Win32FileStream.hpp"
+
+#include <shlwapi.h>
+#include <winnt.h>
+
+namespace cru::platform::win {
+IStream* ConvertStreamToComStream(io::Stream* stream) {
+  static ComAutoInit com_auto_init;
+
+  if (auto memory_stream = dynamic_cast<io::MemoryStream*>(stream)) {
+    return SHCreateMemStream(
+        reinterpret_cast<const BYTE*>(memory_stream->GetBuffer()),
+        memory_stream->GetSize());
+  } else if (auto file_stream = dynamic_cast<Win32FileStream*>(stream)) {
+    auto path = file_stream->GetPath();
+    auto flags = file_stream->GetOpenFileFlags();
+    DWORD grfMode = STGM_SHARE_DENY_NONE | STGM_FAILIFTHERE;
+    if (flags & io::OpenFileFlags::Read) {
+      if (flags & io::OpenFileFlags::Write) {
+        grfMode |= STGM_READWRITE;
+      } else {
+        grfMode |= STGM_READ;
+      }
+    } else {
+      if (flags & io::OpenFileFlags::Write) {
+        grfMode |= STGM_WRITE;
+      } else {
+        throw Exception(u"Stream must be readable or writable.");
+      }
+    }
+
+    IStream* result;
+
+    ThrowIfFailed(SHCreateStreamOnFileEx(
+        path.WinCStr(), grfMode, FILE_ATTRIBUTE_NORMAL, FALSE, NULL, &result));
+
+    LARGE_INTEGER position;
+    position.QuadPart = stream->Tell();
+    result->Seek(position, STREAM_SEEK_SET, NULL);
+
+    return result;
+  } else {
+    return new BridgeComStream(stream);
+  }
+}
+}  // namespace cru::platform::win
-- 
cgit v1.2.3