asar coverage - build #262

src/asar/
File: src/asar/virtualfile.cpp
Date: 2025-02-27 19:01:43
Lines:
125/142
88.0%
Functions:
27/27
100.0%
Branches:
89/147
60.5%
Line Branch Exec Source
1 #include <errno.h>
2 #include "virtualfile.h"
3 #include "platform/file-helpers.h"
4 #include "warnings.h"
5
6 class virtual_file
7 {
8 public:
9 1806 virtual ~virtual_file()
10 1806 {
11 1806 }
12
13 virtual void close() = 0;
14
15 virtual size_t read(void* out_buffer, size_t pos, size_t num_bytes) = 0;
16
17 virtual size_t get_size() = 0;
18 };
19
20 class memory_file : public virtual_file
21 {
22 public:
23 54 memory_file(const void* data, size_t length)
24 56 : m_data(data), m_length(length)
25 {
26 54 }
27
28 220 virtual ~memory_file()
29 108 {
30 108 close();
31 220 }
32
33 110 virtual void close()
34 {
35 110 }
36
37 56 virtual size_t read(void* out_buffer, size_t pos, size_t num_bytes)
38 {
39
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 54 times.
 56 if(pos > m_length) return 0;
40
41 56 int diff = (int)(pos + num_bytes) - (int)m_length;
42 56 num_bytes -= diff < 0 ? 0 : (unsigned int)diff;
43
44 56 memcpy(out_buffer, (const char*)m_data + pos, num_bytes);
45 56 return num_bytes;
46 }
47
48 56 virtual size_t get_size()
49 {
50 56 return m_length;
51 }
52
53 private:
54 const void* m_data;
55 size_t m_length;
56 };
57
58 class physical_file : public virtual_file
59 {
60 public:
61 849 physical_file()
62 1691 : m_file_handle(InvalidFileHandle)
63 {
64 849 }
65
66 6764 virtual ~physical_file()
67 3382 {
68 3382 close();
69 6764 }
70
71 1691 virtual_file_error open(const string& path)
72 {
73
1/2
✓ Branch 0 taken 1691 times.
✗ Branch 1 not taken.
 1691 if (path != "")
74 {
75 // randomdude999: checking this before file regularity to improve error messages
76
4/4
✓ Branch 0 taken 873 times.
✓ Branch 1 taken 818 times.
✓ Branch 2 taken 24 times.
✓ Branch 3 taken 825 times.
 1691 if(!file_exists((const char*)path)) return vfe_doesnt_exist;
77
4/4
✓ Branch 0 taken 1625 times.
✓ Branch 1 taken 18 times.
✓ Branch 2 taken 18 times.
✓ Branch 3 taken 807 times.
 1643 if(!check_is_regular_file((const char*)path)) return vfe_not_regular_file;
78
79 1607 FileOpenError error = FileOpenError_None;
80
81
1/2
✓ Branch 0 taken 807 times.
✗ Branch 1 not taken.
 1607 m_file_handle = open_file((const char*)path, FileOpenMode_Read, &error);
82
83
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1607 times.
 1607 if (m_file_handle == InvalidFileHandle)
84 {
85 ✗ if (error == FileOpenError_NotFound)
86 {
87 ✗ return vfe_doesnt_exist;
88 }
89 ✗ else if (error == FileOpenError_AccessDenied)
90 {
91 ✗ return vfe_access_denied;
92 }
93 else
94 {
95 ✗ return vfe_unknown;
96 }
97 }
98
99 807 return vfe_none;
100 }
101
102 ✗ return vfe_doesnt_exist;
103 }
104
105 3298 virtual void close()
106 {
107
2/2
✓ Branch 0 taken 1607 times.
✓ Branch 1 taken 1691 times.
 3298 if (m_file_handle != InvalidFileHandle)
108 {
109 1607 close_file(m_file_handle);
110 1607 m_file_handle = InvalidFileHandle;
111 }
112 3298 }
113
114 1643 virtual size_t read(void* out_buffer, size_t pos, size_t num_bytes)
115 {
116 1643 set_file_pos(m_file_handle, (uint64_t)pos);
117 1643 return (size_t)read_file(m_file_handle, out_buffer, (uint32_t)num_bytes);
118 }
119
120 1607 virtual size_t get_size()
121 {
122 1607 return (size_t)get_file_size(m_file_handle);
123 }
124
125 private:
126 friend class virtual_filesystem;
127
128 FileHandleType m_file_handle;
129 };
130
131
132
133 787 void virtual_filesystem::initialize(const char** include_paths, size_t num_include_paths)
134 {
135 787 m_include_paths.reset();
136
137
2/2
✓ Branch 0 taken 1277 times.
✓ Branch 1 taken 787 times.
 2064 for (size_t i = 0; i < num_include_paths; ++i)
138 {
139 1277 m_include_paths[(int)i] = include_paths[i];
140 }
141
142 787 m_last_error = vfe_none;
143 409 m_memory_files.reset();
144 787 }
145
146 787 void virtual_filesystem::destroy()
147 {
148 787 m_include_paths.reset();
149 787 }
150
151
152 1747 virtual_file_handle virtual_filesystem::open_file(const char* path, const char* base_path)
153 {
154 1747 m_last_error = vfe_none;
155
156
1/2
✓ Branch 0 taken 903 times.
✗ Branch 1 not taken.
 1747 string absolutepath = create_absolute_path(base_path, path);
157
158
1/2
✓ Branch 0 taken 1747 times.
✗ Branch 1 not taken.
 1747 virtual_file_type vft = get_file_type_from_path(absolutepath);
159
160
2/2
✓ Branch 0 taken 851 times.
✓ Branch 1 taken 896 times.
 1747 if (vft != vft_memory_file)
161 {
162
1/2
✓ Branch 0 taken 1691 times.
✗ Branch 1 not taken.
 1691 asar_throw_warning(0, warning_id_check_memory_file, path, (int)warning_id_check_memory_file);
163 }
164
165
2/3
✓ Branch 0 taken 1691 times.
✓ Branch 1 taken 54 times.
✗ Branch 2 not taken.
 1745 switch (vft)
166 {
167 1691 case vft_physical_file:
168 {
169
1/2
✓ Branch 0 taken 1691 times.
✗ Branch 1 not taken.
 1691 physical_file* new_file = new physical_file;
170
171
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 849 times.
 849 if (new_file == nullptr)
172 {
173 ✗ m_last_error = vfe_unknown;
174 ✗ return INVALID_VIRTUAL_FILE_HANDLE;
175 }
176
177
1/2
✓ Branch 0 taken 1691 times.
✗ Branch 1 not taken.
 1691 virtual_file_error error = new_file->open(absolutepath);
178
179
2/2
✓ Branch 0 taken 84 times.
✓ Branch 1 taken 1607 times.
 1691 if (error != vfe_none)
180 {
181
1/2
✓ Branch 0 taken 42 times.
✗ Branch 1 not taken.
 84 delete new_file;
182 84 m_last_error = error;
183 84 return INVALID_VIRTUAL_FILE_HANDLE;
184 }
185
186 807 return static_cast<virtual_file_handle>(new_file);
187 }
188
189 54 case vft_memory_file:
190 {
191
2/4
✓ Branch 0 taken 56 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 56 times.
✗ Branch 3 not taken.
 56 if(m_memory_files.exists(absolutepath)) {
192
1/2
✓ Branch 0 taken 56 times.
✗ Branch 1 not taken.
 56 memory_buffer mem_buf = m_memory_files.find(absolutepath);
193
1/2
✓ Branch 0 taken 56 times.
✗ Branch 1 not taken.
 56 memory_file* new_file = new memory_file(mem_buf.data, mem_buf.length);
194 54 return static_cast<virtual_file_handle>(new_file);
195 } else {
196 ✗ m_last_error = vfe_doesnt_exist;
197 ✗ return INVALID_VIRTUAL_FILE_HANDLE;
198 }
199 }
200
201 ✗ default:
202 // We should not get here
203 ✗ m_last_error = vfe_unknown;
204 ✗ return INVALID_VIRTUAL_FILE_HANDLE;
205 }
206 1747 }
207
208 1663 void virtual_filesystem::close_file(virtual_file_handle file_handle)
209 {
210
1/2
✓ Branch 0 taken 1663 times.
✗ Branch 1 not taken.
 1663 if (file_handle != INVALID_VIRTUAL_FILE_HANDLE)
211 {
212 861 virtual_file* file = static_cast<virtual_file*>(file_handle);
213
214 1663 file->close();
215
216
1/2
✓ Branch 0 taken 861 times.
✗ Branch 1 not taken.
 1663 delete file;
217 }
218 1663 }
219
220
221
222 1699 size_t virtual_filesystem::read_file(virtual_file_handle file_handle, void* out_buffer, size_t pos, size_t num_bytes)
223 {
224
1/2
✓ Branch 0 taken 1699 times.
✗ Branch 1 not taken.
 1699 if (file_handle != INVALID_VIRTUAL_FILE_HANDLE)
225 {
226 879 virtual_file* file = static_cast<virtual_file*>(file_handle);
227
228 1699 return file->read(out_buffer, pos, num_bytes);
229 }
230
231 ✗ return 0u;
232 }
233
234 1663 size_t virtual_filesystem::get_file_size(virtual_file_handle file_handle)
235 {
236
1/2
✓ Branch 0 taken 1663 times.
✗ Branch 1 not taken.
 1663 if (file_handle != INVALID_VIRTUAL_FILE_HANDLE)
237 {
238 861 virtual_file* file = static_cast<virtual_file*>(file_handle);
239
240 1663 return file->get_size();
241 }
242
243 ✗ return 0u;
244 }
245
246
247 1747 virtual_filesystem::virtual_file_type virtual_filesystem::get_file_type_from_path(const char* path)
248 {
249
2/2
✓ Branch 0 taken 896 times.
✓ Branch 1 taken 851 times.
 1747 if(m_memory_files.exists(path)) {
250 54 return vft_memory_file;
251 } else {
252 1691 return vft_physical_file;
253 }
254 }
255
256 288 void virtual_filesystem::add_memory_file(const char* name, const void* buffer, size_t length) {
257 161 memory_buffer mem_buf = { buffer, length };
258
1/2
✓ Branch 0 taken 161 times.
✗ Branch 1 not taken.
 288 string normalized_path = normalize_path(name);
259
1/2
✓ Branch 0 taken 288 times.
✗ Branch 1 not taken.
 288 m_memory_files.remove(normalized_path);
260
1/2
✓ Branch 0 taken 288 times.
✗ Branch 1 not taken.
 288 m_memory_files.create(normalized_path) = mem_buf;
261 288 }
262
263 13473 bool virtual_filesystem::is_path_absolute(const char* path)
264 {
265 13473 return path_is_absolute(path);
266 }
267
268 31840 string virtual_filesystem::create_absolute_path(const char* base, const char* target)
269 {
270
10/10
✓ Branch 0 taken 28916 times.
✓ Branch 1 taken 2924 times.
✓ Branch 2 taken 25755 times.
✓ Branch 3 taken 3161 times.
✓ Branch 4 taken 25738 times.
✓ Branch 5 taken 17 times.
✓ Branch 6 taken 13 times.
✓ Branch 7 taken 10293 times.
✓ Branch 8 taken 3180 times.
✓ Branch 9 taken 10293 times.
 31840 if (is_path_absolute(target) || base == nullptr || base[0] == '\0')
271 {
272
1/2
✓ Branch 0 taken 3180 times.
✗ Branch 1 not taken.
 6115 return normalize_path(target);
273 }
274
275
1/2
✓ Branch 0 taken 10293 times.
✗ Branch 1 not taken.
 10293 string path_to_use = "";
276
1/2
✓ Branch 0 taken 10293 times.
✗ Branch 1 not taken.
 10293 string test_path = "";
277
278
2/4
✓ Branch 0 taken 25725 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 15432 times.
✗ Branch 3 not taken.
 25725 test_path = normalize_path(target);
279
280 // First check if path is absolute
281
2/4
✓ Branch 0 taken 25725 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 25725 times.
 25725 if (path_is_absolute(test_path))
282 {
283 ✗ if (m_memory_files.exists(test_path) || file_exists(test_path))
284 {
285 ✗ path_to_use = test_path;
286 }
287 }
288 else
289 {
290 // Now check if path exists relative to the base path
291
1/2
✓ Branch 0 taken 10293 times.
✗ Branch 1 not taken.
 10293 if (base != nullptr)
292 {
293
2/4
✓ Branch 0 taken 25725 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 25725 times.
✗ Branch 3 not taken.
 25725 test_path = create_combined_path(dir(base), target);
294 }
295
296
9/12
✓ Branch 0 taken 25725 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 25725 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 25722 times.
✓ Branch 5 taken 3 times.
✓ Branch 6 taken 25722 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 10278 times.
✓ Branch 9 taken 15444 times.
✓ Branch 10 taken 10230 times.
✓ Branch 11 taken 63 times.
 25725 if (test_path != "" && (m_memory_files.exists(test_path) || file_exists(test_path)))
297 {
298
1/2
✓ Branch 0 taken 10230 times.
✗ Branch 1 not taken.
 10230 path_to_use = test_path;
299 }
300 else
301 {
302 // Finally check if path exists relative to any include path
303 63 bool found = false;
304
2/2
✓ Branch 0 taken 235 times.
✓ Branch 1 taken 90 times.
 325 for (int i = 0; i < m_include_paths.count; ++i)
305 {
306
2/4
✓ Branch 0 taken 235 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 137 times.
✗ Branch 3 not taken.
 235 test_path = create_combined_path(m_include_paths[i], target);
307
308
8/10
✓ Branch 0 taken 235 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 223 times.
✓ Branch 3 taken 12 times.
✓ Branch 4 taken 223 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 98 times.
✓ Branch 7 taken 125 times.
✓ Branch 8 taken 18 times.
✓ Branch 9 taken 119 times.
 235 if (m_memory_files.exists(test_path) || file_exists(test_path))
309 {
310 18 found = true;
311
1/2
✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
 18 path_to_use = test_path;
312 18 break;
313 }
314 }
315
316
2/2
✓ Branch 0 taken 45 times.
✓ Branch 1 taken 18 times.
 63 if (!found)
317 {
318 // Reset our path so that we don't return an empty one
319 // (that will do some weird shit and fuck up error messages)
320
1/2
✓ Branch 0 taken 45 times.
✗ Branch 1 not taken.
 45 path_to_use = target;
321 }
322 }
323 }
324
325 10293 return path_to_use;
326 25725 }
327