[SHELLBTRFS] Fix build?
[reactos.git] / dll / shellext / shellbtrfs / shellext.h
1 /* Copyright (c) Mark Harmstone 2016-17
2 *
3 * This file is part of WinBtrfs.
4 *
5 * WinBtrfs is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU Lesser General Public Licence as published by
7 * the Free Software Foundation, either version 3 of the Licence, or
8 * (at your option) any later version.
9 *
10 * WinBtrfs is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU Lesser General Public Licence for more details.
14 *
15 * You should have received a copy of the GNU Lesser General Public Licence
16 * along with WinBtrfs. If not, see <http://www.gnu.org/licenses/>. */
17
18 #pragma once
19
20 /* C++ backwards-compatibility */
21 #ifdef __REACTOS__
22 #if defined(_MSC_VER) && (_MSC_VER < 1900)
23 #define noexcept
24 #endif
25 #endif
26
27 #define ISOLATION_AWARE_ENABLED 1
28 #define STRSAFE_NO_DEPRECATE
29
30 #ifndef __REACTOS__
31 #define WINVER 0x0A00 // Windows 10
32 #define _WIN32_WINNT 0x0A00
33 #endif
34
35 #ifndef __REACTOS__
36 #include <windows.h>
37 #include <winternl.h>
38 #else
39 #define WIN32_NO_STATUS
40 #include <windef.h>
41 #include <winbase.h>
42 #include <strsafe.h>
43 #include <ndk/iofuncs.h>
44 #include <ndk/obfuncs.h>
45 #include <ndk/rtlfuncs.h>
46 #endif
47 #include <string>
48 #ifdef __REACTOS__
49 #define string_view string
50 #define wstring_view wstring
51 #endif
52 #include <vector>
53 #include <stdint.h>
54 #ifndef __REACTOS__
55 #include "../btrfs.h"
56 #include "../btrfsioctl.h"
57 #else
58 #include "btrfs.h"
59 #include "btrfsioctl.h"
60 #endif
61
62 using namespace std;
63
64 #ifndef __REACTOS__
65 #define STATUS_SUCCESS (NTSTATUS)0x00000000
66 #define STATUS_BUFFER_OVERFLOW (NTSTATUS)0x80000005
67 #define STATUS_END_OF_FILE (NTSTATUS)0xc0000011
68 #define STATUS_MORE_PROCESSING_REQUIRED (NTSTATUS)0xc0000016
69 #define STATUS_BUFFER_TOO_SMALL (NTSTATUS)0xc0000023
70 #define STATUS_DEVICE_NOT_READY (NTSTATUS)0xc00000a3
71 #define STATUS_CANNOT_DELETE (NTSTATUS)0xc0000121
72 #define STATUS_NOT_FOUND (NTSTATUS)0xc0000225
73 #endif
74
75 #define BLOCK_FLAG_DATA 0x001
76 #define BLOCK_FLAG_SYSTEM 0x002
77 #define BLOCK_FLAG_METADATA 0x004
78 #define BLOCK_FLAG_RAID0 0x008
79 #define BLOCK_FLAG_RAID1 0x010
80 #define BLOCK_FLAG_DUPLICATE 0x020
81 #define BLOCK_FLAG_RAID10 0x040
82 #define BLOCK_FLAG_RAID5 0x080
83 #define BLOCK_FLAG_RAID6 0x100
84
85 #define BTRFS_TYPE_FILE 1
86 #define BTRFS_TYPE_DIRECTORY 2
87 #define BTRFS_TYPE_CHARDEV 3
88 #define BTRFS_TYPE_BLOCKDEV 4
89 #define BTRFS_TYPE_FIFO 5
90 #define BTRFS_TYPE_SOCKET 6
91 #define BTRFS_TYPE_SYMLINK 7
92
93 #ifdef _MSC_VER
94 #define funcname __FUNCTION__
95 #else
96 #define funcname __func__
97 #endif
98
99 #ifdef _MSC_VER
100 #pragma warning(disable: 4800)
101 #endif
102
103 #ifdef __cplusplus
104 extern "C" {
105 #endif
106 #ifndef __REACTOS__
107 NTSYSCALLAPI NTSTATUS NTAPI NtFsControlFile(HANDLE FileHandle, HANDLE Event, PIO_APC_ROUTINE ApcRoutine, PVOID ApcContext,
108 PIO_STATUS_BLOCK IoStatusBlock, ULONG FsControlCode, PVOID InputBuffer, ULONG InputBufferLength,
109 PVOID OutputBuffer, ULONG OutputBufferLength);
110
111 NTSTATUS NTAPI NtReadFile(HANDLE FileHandle, HANDLE Event, PIO_APC_ROUTINE ApcRoutine, PVOID ApcContext, PIO_STATUS_BLOCK IoStatusBlock, PVOID Buffer,
112 ULONG Length, PLARGE_INTEGER ByteOffset, PULONG Key);
113
114 NTSTATUS WINAPI NtSetEaFile(HANDLE FileHandle, PIO_STATUS_BLOCK IoStatusBlock, PVOID Buffer, ULONG Length);
115
116 NTSTATUS WINAPI NtSetSecurityObject(HANDLE Handle, SECURITY_INFORMATION SecurityInformation, PSECURITY_DESCRIPTOR SecurityDescriptor);
117
118 NTSTATUS NTAPI NtQueryInformationFile(HANDLE hFile, PIO_STATUS_BLOCK io, PVOID ptr, ULONG len, FILE_INFORMATION_CLASS FileInformationClass);
119
120 NTSTATUS NTAPI NtSetInformationFile(HANDLE hFile, PIO_STATUS_BLOCK io, PVOID ptr, ULONG len, FILE_INFORMATION_CLASS FileInformationClass);
121
122 #ifdef _MSC_VER
123 #define FileBasicInformation (FILE_INFORMATION_CLASS)4
124 #define FileStandardInformation (FILE_INFORMATION_CLASS)5
125 #define FileDispositionInformation (FILE_INFORMATION_CLASS)13
126 #define FileEndOfFileInformation (FILE_INFORMATION_CLASS)20
127 #define FileStreamInformation (FILE_INFORMATION_CLASS)22
128
129 typedef enum _FSINFOCLASS {
130 FileFsVolumeInformation = 1,
131 FileFsLabelInformation,
132 FileFsSizeInformation,
133 FileFsDeviceInformation,
134 FileFsAttributeInformation,
135 FileFsControlInformation,
136 FileFsFullSizeInformation,
137 FileFsObjectIdInformation,
138 FileFsDriverPathInformation,
139 FileFsVolumeFlagsInformation,
140 FileFsSectorSizeInformation,
141 FileFsDataCopyInformation,
142 FileFsMetadataSizeInformation,
143 FileFsFullSizeInformationEx,
144 FileFsMaximumInformation
145 } FS_INFORMATION_CLASS, *PFS_INFORMATION_CLASS;
146
147 typedef struct _FILE_STREAM_INFORMATION {
148 ULONG NextEntryOffset;
149 ULONG StreamNameLength;
150 LARGE_INTEGER StreamSize;
151 LARGE_INTEGER StreamAllocationSize;
152 WCHAR StreamName[1];
153 } FILE_STREAM_INFORMATION, *PFILE_STREAM_INFORMATION;
154 #endif
155
156 NTSTATUS NTAPI NtQueryVolumeInformationFile(HANDLE FileHandle, PIO_STATUS_BLOCK IoStatusBlock, PVOID FsInformation, ULONG Length,
157 FS_INFORMATION_CLASS FsInformationClass);
158 #endif
159 #ifdef __cplusplus
160 }
161 #endif
162
163 #ifndef __REACTOS__
164 typedef struct _REPARSE_DATA_BUFFER {
165 ULONG ReparseTag;
166 USHORT ReparseDataLength;
167 USHORT Reserved;
168
169 union {
170 struct {
171 USHORT SubstituteNameOffset;
172 USHORT SubstituteNameLength;
173 USHORT PrintNameOffset;
174 USHORT PrintNameLength;
175 ULONG Flags;
176 WCHAR PathBuffer[1];
177 } SymbolicLinkReparseBuffer;
178
179 struct {
180 USHORT SubstituteNameOffset;
181 USHORT SubstituteNameLength;
182 USHORT PrintNameOffset;
183 USHORT PrintNameLength;
184 WCHAR PathBuffer[1];
185 } MountPointReparseBuffer;
186
187 struct {
188 UCHAR DataBuffer[1];
189 } GenericReparseBuffer;
190 };
191 } REPARSE_DATA_BUFFER, *PREPARSE_DATA_BUFFER;
192
193 #define SYMLINK_FLAG_RELATIVE 1
194 #endif
195
196 #ifndef FILE_SUPPORTS_BLOCK_REFCOUNTING
197
198 typedef struct _DUPLICATE_EXTENTS_DATA {
199 HANDLE FileHandle;
200 LARGE_INTEGER SourceFileOffset;
201 LARGE_INTEGER TargetFileOffset;
202 LARGE_INTEGER ByteCount;
203 } DUPLICATE_EXTENTS_DATA, *PDUPLICATE_EXTENTS_DATA;
204
205 #define FSCTL_DUPLICATE_EXTENTS_TO_FILE CTL_CODE(FILE_DEVICE_FILE_SYSTEM, 209, METHOD_BUFFERED, FILE_WRITE_ACCESS)
206
207 typedef struct _FSCTL_GET_INTEGRITY_INFORMATION_BUFFER {
208 WORD ChecksumAlgorithm;
209 WORD Reserved;
210 DWORD Flags;
211 DWORD ChecksumChunkSizeInBytes;
212 DWORD ClusterSizeInBytes;
213 } FSCTL_GET_INTEGRITY_INFORMATION_BUFFER, *PFSCTL_GET_INTEGRITY_INFORMATION_BUFFER;
214
215 typedef struct _FSCTL_SET_INTEGRITY_INFORMATION_BUFFER {
216 WORD ChecksumAlgorithm;
217 WORD Reserved;
218 DWORD Flags;
219 } FSCTL_SET_INTEGRITY_INFORMATION_BUFFER, *PFSCTL_SET_INTEGRITY_INFORMATION_BUFFER;
220
221 #define FSCTL_GET_INTEGRITY_INFORMATION CTL_CODE(FILE_DEVICE_FILE_SYSTEM, 159, METHOD_BUFFERED, FILE_ANY_ACCESS)
222 #define FSCTL_SET_INTEGRITY_INFORMATION CTL_CODE(FILE_DEVICE_FILE_SYSTEM, 160, METHOD_BUFFERED, FILE_READ_DATA | FILE_WRITE_DATA)
223
224 #endif
225
226 class win_handle {
227 public:
228 win_handle() {
229 #ifdef __REACTOS__
230 h = INVALID_HANDLE_VALUE;
231 #endif
232 }
233
234 win_handle(HANDLE nh) {
235 h = nh;
236 }
237
238 ~win_handle() {
239 if (h != INVALID_HANDLE_VALUE)
240 CloseHandle(h);
241 }
242
243 operator HANDLE() const {
244 return h;
245 }
246
247 win_handle& operator=(const HANDLE nh) {
248 if (h != INVALID_HANDLE_VALUE)
249 CloseHandle(h);
250
251 h = nh;
252
253 return *this;
254 }
255
256 HANDLE* operator&() {
257 return &h;
258 }
259
260 private:
261 #ifndef __REACTOS__
262 HANDLE h = INVALID_HANDLE_VALUE;
263 #else
264 HANDLE h;
265 #endif
266 };
267
268 class fff_handle {
269 public:
270 fff_handle() {
271 #ifdef __REACTOS__
272 h = INVALID_HANDLE_VALUE;
273 #endif
274 }
275
276 fff_handle(HANDLE nh) {
277 h = nh;
278 }
279
280 ~fff_handle() {
281 if (h != INVALID_HANDLE_VALUE)
282 FindClose(h);
283 }
284
285 operator HANDLE() const {
286 return h;
287 }
288
289 fff_handle& operator=(const HANDLE nh) {
290 if (h != INVALID_HANDLE_VALUE)
291 FindClose(h);
292
293 h = nh;
294
295 return *this;
296 }
297
298 HANDLE* operator&() {
299 return &h;
300 }
301
302 private:
303 #ifndef __REACTOS__
304 HANDLE h = INVALID_HANDLE_VALUE;
305 #else
306 HANDLE h;
307 #endif
308 };
309
310 class nt_handle {
311 public:
312 nt_handle() {
313 #ifdef __REACTOS__
314 h = INVALID_HANDLE_VALUE;
315 #endif
316 }
317
318 nt_handle(HANDLE nh) {
319 h = nh;
320 }
321
322 ~nt_handle() {
323 if (h != INVALID_HANDLE_VALUE)
324 NtClose(h);
325 }
326
327 operator HANDLE() const {
328 return h;
329 }
330
331 nt_handle& operator=(const HANDLE nh) {
332 if (h != INVALID_HANDLE_VALUE)
333 NtClose(h);
334
335 h = nh;
336
337 return *this;
338 }
339
340 HANDLE* operator&() {
341 return &h;
342 }
343
344 private:
345 #ifndef __REACTOS__
346 HANDLE h = INVALID_HANDLE_VALUE;
347 #else
348 HANDLE h;
349 #endif
350 };
351
352 class string_error : public exception {
353 public:
354 string_error(int resno, ...);
355
356 const char* what() const noexcept {
357 return msg.c_str();
358 }
359
360 private:
361 string msg;
362 };
363
364
365 class last_error : public exception {
366 public:
367 last_error(DWORD errnum);
368
369 const char* what() const noexcept {
370 return msg.c_str();
371 }
372
373 private:
374 string msg;
375 };
376
377 class ntstatus_error : public exception {
378 public:
379 ntstatus_error(NTSTATUS Status);
380
381 const char* what() const noexcept {
382 return msg.c_str();
383 }
384
385 NTSTATUS Status;
386
387 private:
388 string msg;
389 };
390
391 #ifdef __REACTOS__
392 inline wstring to_wstring(uint8_t a) { WCHAR buffer[16]; swprintf(buffer, L"%d", a); return wstring(buffer); }
393 inline wstring to_wstring(uint16_t a) { WCHAR buffer[16]; swprintf(buffer, L"%d", a); return wstring(buffer); }
394 inline wstring to_wstring(uint32_t a) { WCHAR buffer[32]; swprintf(buffer, L"%ld", a); return wstring(buffer); }
395 inline wstring to_wstring(uint64_t a) { WCHAR buffer[64]; swprintf(buffer, L"%I64d", a); return wstring(buffer); }
396 #endif
397
398 extern HMODULE module;
399 void format_size(uint64_t size, wstring& s, bool show_bytes);
400 void set_dpi_aware();
401 wstring format_message(ULONG last_error);
402 wstring format_ntstatus(NTSTATUS Status);
403 bool load_string(HMODULE module, UINT id, wstring& s);
404 void wstring_sprintf(wstring& s, wstring fmt, ...);
405 void command_line_to_args(LPWSTR cmdline, vector<wstring>& args);
406 wstring utf8_to_utf16(const string_view& utf8);
407 void error_message(HWND hwnd, const char* msg);