asar coverage - build #174

src/asar/
File: src/asar/virtualfile.cpp
Date: 2024-01-28 00:31:05
Lines:
125/142
88.0%
Functions:
27/27
100.0%
Branches:
94/153
61.4%
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 1712 virtual ~virtual_file()
10 1712 {
11 1712 }
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 38 memory_file(const void* data, size_t length)
24 40 : m_data(data), m_length(length)
25 {
26 38 }
27
28 156 virtual ~memory_file()
29 76 {
30 76 close();
31 156 }
32
33 78 virtual void close()
34 {
35 78 }
36
37 40 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 38 times.
 40 if(pos > m_length) return 0;
40
41 40 int diff = (int)(pos + num_bytes) - (int)m_length;
42 40 num_bytes -= diff < 0 ? 0 : (unsigned int)diff;
43
44 40 memcpy(out_buffer, (const char*)m_data + pos, num_bytes);
45 40 return num_bytes;
46 }
47
48 40 virtual size_t get_size()
49 {
50 40 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 818 physical_file()
62 1633 : m_file_handle(InvalidFileHandle)
63 {
64 818 }
65
66 6532 virtual ~physical_file()
67 3266 {
68 3266 close();
69 6532 }
70
71 1633 virtual_file_error open(const string& path)
72 {
73
1/2
✓ Branch 0 taken 1633 times.
✗ Branch 1 not taken.
 1633 if (path != "")
74 {
75 // randomdude999: checking this before file regularity to improve error messages
76
4/4
✓ Branch 0 taken 842 times.
✓ Branch 1 taken 791 times.
✓ Branch 2 taken 24 times.
✓ Branch 3 taken 794 times.
 1633 if(!file_exists((const char*)path)) return vfe_doesnt_exist;
77
4/4
✓ Branch 0 taken 1567 times.
✓ Branch 1 taken 18 times.
✓ Branch 2 taken 18 times.
✓ Branch 3 taken 776 times.
 1585 if(!check_is_regular_file((const char*)path)) return vfe_not_regular_file;
78
79 1549 FileOpenError error = FileOpenError_None;
80
81
1/2
✓ Branch 0 taken 776 times.
✗ Branch 1 not taken.
 1549 m_file_handle = open_file((const char*)path, FileOpenMode_Read, &error);
82
83
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1549 times.
 1549 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 776 return vfe_none;
100 }
101
102 ✗ return vfe_doesnt_exist;
103 }
104
105 3182 virtual void close()
106 {
107
2/2
✓ Branch 0 taken 1549 times.
✓ Branch 1 taken 1633 times.
 3182 if (m_file_handle != InvalidFileHandle)
108 {
109 1549 close_file(m_file_handle);
110 1549 m_file_handle = InvalidFileHandle;
111 }
112 3182 }
113
114 1591 virtual size_t read(void* out_buffer, size_t pos, size_t num_bytes)
115 {
116 1591 set_file_pos(m_file_handle, (uint64_t)pos);
117 1591 return (size_t)read_file(m_file_handle, out_buffer, (uint32_t)num_bytes);
118 }
119
120 1549 virtual size_t get_size()
121 {
122 1549 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 747 void virtual_filesystem::initialize(const char** include_paths, size_t num_include_paths)
134 {
135 747 m_include_paths.reset();
136
137
2/2
✓ Branch 0 taken 1223 times.
✓ Branch 1 taken 747 times.
 1970 for (size_t i = 0; i < num_include_paths; ++i)
138 {
139 1223 m_include_paths[(int)i] = include_paths[i];
140 }
141
142 747 m_last_error = vfe_none;
143 747 m_memory_files.reset();
144 747 }
145
146 747 void virtual_filesystem::destroy()
147 {
148 747 m_include_paths.reset();
149 747 }
150
151
152 1673 virtual_file_handle virtual_filesystem::open_file(const char* path, const char* base_path)
153 {
154 1673 m_last_error = vfe_none;
155
156
1/2
✓ Branch 0 taken 856 times.
✗ Branch 1 not taken.
 1673 string absolutepath = create_absolute_path(base_path, path);
157
158
1/2
✓ Branch 0 taken 856 times.
✗ Branch 1 not taken.
 1673 virtual_file_type vft = get_file_type_from_path(absolutepath);
159
160
2/2
✓ Branch 0 taken 820 times.
✓ Branch 1 taken 853 times.
 1673 if (vft != vft_memory_file)
161 {
162
1/2
✓ Branch 0 taken 1633 times.
✗ Branch 1 not taken.
 1633 asar_throw_warning(0, warning_id_check_memory_file, path, (int)warning_id_check_memory_file);
163 }
164
165
2/3
✓ Branch 0 taken 1633 times.
✓ Branch 1 taken 38 times.
✗ Branch 2 not taken.
 1671 switch (vft)
166 {
167 1633 case vft_physical_file:
168 {
169
1/2
✓ Branch 0 taken 1633 times.
✗ Branch 1 not taken.
 1633 physical_file* new_file = new physical_file;
170
171
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 818 times.
 818 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 1633 times.
✗ Branch 1 not taken.
 1633 virtual_file_error error = new_file->open(absolutepath);
178
179
2/2
✓ Branch 0 taken 84 times.
✓ Branch 1 taken 1549 times.
 1633 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 776 return static_cast<virtual_file_handle>(new_file);
187 }
188
189 38 case vft_memory_file:
190 {
191
2/4
✓ Branch 0 taken 40 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 38 times.
✗ Branch 3 not taken.
 40 if(m_memory_files.exists(absolutepath)) {
192
1/2
✓ Branch 0 taken 38 times.
✗ Branch 1 not taken.
 40 memory_buffer mem_buf = m_memory_files.find(absolutepath);
193
1/2
✓ Branch 0 taken 40 times.
✗ Branch 1 not taken.
 40 memory_file* new_file = new memory_file(mem_buf.data, mem_buf.length);
194 38 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 1673 }
207
208 1589 void virtual_filesystem::close_file(virtual_file_handle file_handle)
209 {
210
1/2
✓ Branch 0 taken 1589 times.
✗ Branch 1 not taken.
 1589 if (file_handle != INVALID_VIRTUAL_FILE_HANDLE)
211 {
212 814 virtual_file* file = static_cast<virtual_file*>(file_handle);
213
214 1589 file->close();
215
216
1/2
✓ Branch 0 taken 814 times.
✗ Branch 1 not taken.
 1589 delete file;
217 }
218 1589 }
219
220
221
222 1631 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 1631 times.
✗ Branch 1 not taken.
 1631 if (file_handle != INVALID_VIRTUAL_FILE_HANDLE)
225 {
226 835 virtual_file* file = static_cast<virtual_file*>(file_handle);
227
228 1631 return file->read(out_buffer, pos, num_bytes);
229 }
230
231 ✗ return 0u;
232 }
233
234 1589 size_t virtual_filesystem::get_file_size(virtual_file_handle file_handle)
235 {
236
1/2
✓ Branch 0 taken 1589 times.
✗ Branch 1 not taken.
 1589 if (file_handle != INVALID_VIRTUAL_FILE_HANDLE)
237 {
238 814 virtual_file* file = static_cast<virtual_file*>(file_handle);
239
240 1589 return file->get_size();
241 }
242
243 ✗ return 0u;
244 }
245
246
247 1673 virtual_filesystem::virtual_file_type virtual_filesystem::get_file_type_from_path(const char* path)
248 {
249
2/2
✓ Branch 0 taken 853 times.
✓ Branch 1 taken 820 times.
 1673 if(m_memory_files.exists(path)) {
250 38 return vft_memory_file;
251 } else {
252 1633 return vft_physical_file;
253 }
254 }
255
256 269 void virtual_filesystem::add_memory_file(const char* name, const void* buffer, size_t length) {
257 147 memory_buffer mem_buf = { buffer, length };
258
1/2
✓ Branch 0 taken 147 times.
✗ Branch 1 not taken.
 269 string normalized_path = normalize_path(name);
259
1/2
✓ Branch 0 taken 269 times.
✗ Branch 1 not taken.
 269 m_memory_files.remove(normalized_path);
260
1/2
✓ Branch 0 taken 269 times.
✗ Branch 1 not taken.
 269 m_memory_files.create(normalized_path) = mem_buf;
261 269 }
262
263 12232 bool virtual_filesystem::is_path_absolute(const char* path)
264 {
265 12232 return path_is_absolute(path);
266 }
267
268 30464 string virtual_filesystem::create_absolute_path(const char* base, const char* target)
269 {
270
10/10
✓ Branch 0 taken 27654 times.
✓ Branch 1 taken 2810 times.
✓ Branch 2 taken 24687 times.
✓ Branch 3 taken 2967 times.
✓ Branch 4 taken 24670 times.
✓ Branch 5 taken 17 times.
✓ Branch 6 taken 13 times.
✓ Branch 7 taken 9246 times.
✓ Branch 8 taken 2986 times.
✓ Branch 9 taken 9246 times.
 30464 if (is_path_absolute(target) || base == nullptr || base[0] == '\0')
271 {
272
1/2
✓ Branch 0 taken 2986 times.
✗ Branch 1 not taken.
 5807 return normalize_path(target);
273 }
274
275
1/2
✓ Branch 0 taken 9246 times.
✗ Branch 1 not taken.
 9246 string path_to_use = "";
276
1/2
✓ Branch 0 taken 9246 times.
✗ Branch 1 not taken.
 9246 string test_path = "";
277
278
2/4
✓ Branch 0 taken 24657 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 24657 times.
✗ Branch 3 not taken.
 24657 test_path = normalize_path(target);
279
280 // First check if path is absolute
281
2/4
✓ Branch 0 taken 24657 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 24657 times.
 24657 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 9246 times.
✗ Branch 1 not taken.
 9246 if (base != nullptr)
292 {
293
3/6
✓ Branch 0 taken 24657 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 24657 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 9246 times.
✗ Branch 5 not taken.
 24657 test_path = create_combined_path(dir(base), target);
294 }
295
296
10/12
✓ Branch 0 taken 24657 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 24657 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 24654 times.
✓ Branch 5 taken 3 times.
✓ Branch 6 taken 9294 times.
✓ Branch 7 taken 15360 times.
✓ Branch 8 taken 9183 times.
✓ Branch 9 taken 60 times.
✓ Branch 10 taken 9186 times.
✓ Branch 11 taken 60 times.
 24657 if (test_path != "" && (m_memory_files.exists(test_path) || file_exists(test_path)))
297 {
298
1/2
✓ Branch 0 taken 9186 times.
✗ Branch 1 not taken.
 9186 path_to_use = test_path;
299 }
300 else
301 {
302 // Finally check if path exists relative to any include path
303 60 bool found = false;
304
2/2
✓ Branch 0 taken 214 times.
✓ Branch 1 taken 90 times.
 304 for (int i = 0; i < m_include_paths.count; ++i)
305 {
306
3/6
✓ Branch 0 taken 214 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 116 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 116 times.
✗ Branch 5 not taken.
 214 test_path = create_combined_path(m_include_paths[i], target);
307
308
9/10
✓ Branch 0 taken 214 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 205 times.
✓ Branch 3 taken 9 times.
✓ Branch 4 taken 199 times.
✓ Branch 5 taken 6 times.
✓ Branch 6 taken 6 times.
✓ Branch 7 taken 101 times.
✓ Branch 8 taken 15 times.
✓ Branch 9 taken 101 times.
 214 if (m_memory_files.exists(test_path) || file_exists(test_path))
309 {
310 15 found = true;
311
1/2
✓ Branch 0 taken 15 times.
✗ Branch 1 not taken.
 15 path_to_use = test_path;
312 15 break;
313 }
314 }
315
316
2/2
✓ Branch 0 taken 45 times.
✓ Branch 1 taken 15 times.
 60 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
1/2
✓ Branch 0 taken 9246 times.
✗ Branch 1 not taken.
 9246 return path_to_use;
326 24657 }
327