blob: 9a79d16bc50d080b4bc6dbb4a23cce4bea76cf0d (
plain)
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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
#include <filesystem>
#include <fstream>
std::filesystem::path storages_path;
std::filesystem::path hudpainter_path;
bool has_mo2;
// See below main() for implementation.
void request_directory(std::filesystem::path);
std::filesystem::path restrict_path(std::filesystem::path);
std::vector<std::filesystem::path> get_files();
std::string read_file(std::filesystem::path);
void write_file(std::filesystem::path);
// Expected path when not using MO2
int main()
{
// RedFileSystem plugin (DLL) is loaded.
has_mo2 = GetModuleHandle(TEXT("usvfs_x64.dll")) != nullptr;
// Setup paths
auto path = std::filesystem::absolute("."); // <GAME>\bin\x64
auto game_path = path.parent_path().parent_path(); // <GAME>
storages_path = game_path / "r6" / "storages"; // <GAME>\r6\storages
// Create storages directory if it is no present.
request_directory(storages_path); // <GAME>\r6\storages
// Game is running...
// HUD Painter request its storage endpoint.
hudpainter_path = storages_path / "HUDPainter";
request_directory(hudpainter_path); // <GAME>\r6\storages\HUDPainter
// HUD Painter request a file (present in archive).
auto default_path =
restrict_path("DEFAULT.json"); // <GAME>\r6\storages\HUDPainter\DEFAULT.json
// HUD Painter read file in `default_path` with success.
auto data = read_file(default_path); // <GAME>\r6\storages\HUDPainter\DEFAULT.json
// HUD Painter request another file.
auto test_path =
restrict_path("TEST.json"); // <GAME>\r6\storages\HUDPainter\TEST.json
// (Bug A) HUD Painter write file in `test_path`.
write_file(test_path); // <GAME>\r6\storages\HUDPainter\TEST.json
// file is created with workaround, not created otherwise.
// (Bug B) HUD Painter request list of files.
// TEST.json is successfully created when using workaround.
auto files = get_files(); // [0] <GAME>\r6\storages\HUDPainter\DEFAULT.json
// files.size() == 1
// Expect:
// files.size() == 2
//
// [0] <GAME>\r6\storages\HUDPainter\DEFAULT.json
// [1] <GAME>\r6\storages\HUDPainter\TEST.json
}
// Create directory if it is no present
void request_directory(std::filesystem::path path)
{
bool is_present = std::filesystem::exists(path);
if (is_present) {
return;
}
std::filesystem::create_directory(path);
}
// Resolve path for HUDPainter (security layer)
std::filesystem::path restrict_path(std::filesystem::path path)
{
auto real_path = std::filesystem::weakly_canonical(hudpainter_path / path);
// Workaround when using MO2
if (has_mo2) {
return real_path;
}
// Expected implementation without MO2
if (real_path.string().find(hudpainter_path.string() + "\\") != 0) {
// "throw" error
}
return real_path;
}
// List files in storage of HUDPainter.
std::vector<std::filesystem::path> get_files()
{
std::vector<std::filesystem::path> files;
auto entries = std::filesystem::directory_iterator(hudpainter_path);
for (const auto& entry : entries) {
if (entry.is_regular_file()) {
auto file_name = entry.path().filename();
auto file_path =
hudpainter_path /
file_name; // Culprit of bug B. When using entry.path() directly, it works.
files.emplace_back(file_path);
}
}
return files;
}
std::string read_file(std::filesystem::path path)
{
std::ifstream stream;
// With workaround:
// std::filesystem::create_directories(path.parent_path());
stream.open(path);
if (!stream.is_open()) {
return "";
}
std::stringstream data;
data << stream.rdbuf();
stream.close();
return data.str();
}
void write_file(std::filesystem::path path)
{
std::ofstream stream;
// With workaround:
// std::filesystem::create_directories(path.parent_path());
stream.open(path, std::ios_base::trunc);
if (!stream.is_open()) {
return;
}
stream << "<DATA>";
stream.close();
}
|
