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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
#include "cru/common/platform/win/Win32FileStream.h"
#include "Win32FileStreamPrivate.h"
#include "cru/common/io/OpenFileFlag.h"
#include "cru/common/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)
: 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() {
Close();
delete p_;
}
bool Win32FileStream::CanSeek() { return true; }
Index Win32FileStream::Seek(Index offset, SeekOrigin origin) {
CheckClosed();
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;
}
bool Win32FileStream::CanRead() { return true; }
Index Win32FileStream::Read(std::byte* buffer, Index offset, Index size) {
if (size < 0) {
throw Exception(u"Size must be greater than 0.");
}
CheckClosed();
ULONG n_read;
ThrowIfFailed(p_->stream_->Read(buffer + offset, size, &n_read));
return n_read;
}
bool Win32FileStream::CanWrite() { return true; }
Index Win32FileStream::Write(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::Close() {
if (closed_) return;
if (p_->stream_) {
p_->stream_->Release();
p_->stream_ = nullptr;
}
closed_ = true;
}
void Win32FileStream::CheckClosed() {
if (closed_) throw Exception(u"Stream is closed.");
}
} // namespace cru::platform::win
|
