1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
#include "cru/base/platform/win/Win32FileStream.h"
#include "Win32FileStreamPrivate.h"
#include "cru/base/io/OpenFileFlag.h"
#include "cru/base/platform/win/Exception.h"
#include <Windows.h>
#include <coml2api.h>
#include <shlwapi.h>
#include <winnt.h>
#include <filesystem>
namespace cru::platform::win {
using namespace cru::io;
Win32FileStream::Win32FileStream(String path, OpenFileFlag flags)
: Stream(true, true, true),
path_(std::move(path)),
flags_(flags),
p_(new details::Win32FileStreamPrivate()) {
DWORD grfMode = STGM_SHARE_DENY_NONE;
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.");
}
}
if (flags & io::OpenFileFlags::Truncate) {
grfMode |= STGM_CREATE;
}
IStream* stream;
ThrowIfFailed(SHCreateStreamOnFileEx(
path_.WinCStr(), grfMode, FILE_ATTRIBUTE_NORMAL,
flags & io::OpenFileFlags::Create ? TRUE : FALSE, NULL, &stream));
p_->stream_ = stream;
}
Win32FileStream::~Win32FileStream() {
DoClose();
delete p_;
}
Index Win32FileStream::DoSeek(Index offset, SeekOrigin origin) {
DWORD dwOrigin = 0;
if (origin == SeekOrigin::Current) {
dwOrigin = STREAM_SEEK_CUR;
} else if (origin == SeekOrigin::End) {
dwOrigin = STREAM_SEEK_END;
} else {
dwOrigin = STREAM_SEEK_SET;
}
LARGE_INTEGER n_offset;
n_offset.QuadPart = offset;
ULARGE_INTEGER n_new_offset;
ThrowIfFailed(p_->stream_->Seek(n_offset, dwOrigin, &n_new_offset));
return n_new_offset.QuadPart;
}
Index Win32FileStream::DoRead(std::byte* buffer, Index offset, Index size) {
ULONG n_read;
ThrowIfFailed(p_->stream_->Read(buffer + offset, size, &n_read));
return n_read;
}
Index Win32FileStream::DoWrite(const std::byte* buffer, Index offset,
Index size) {
if (size < 0) {
throw Exception(u"Size must be greater than 0.");
}
CheckClosed();
ULONG n_written;
ThrowIfFailed(p_->stream_->Write(buffer + offset, size, &n_written));
return n_written;
}
void Win32FileStream::DoClose() {
CRU_STREAM_BEGIN_CLOSE
if (p_->stream_) {
p_->stream_->Release();
p_->stream_ = nullptr;
}
}
} // namespace cru::platform::win
|
