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
|
/*
* Copyright © 2023 Michael Smith <mikesmiffy128@gmail.com>
*
* Permission to use, copy, modify, and/or distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED “AS IS” AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
* REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
* AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
* INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
* LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
* OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
* PERFORMANCE OF THIS SOFTWARE.
*/
/* Windows-specific definitions for os.h - don't include this directly! */
#define IMPORT __declspec(dllimport) // only needed for variables
#define EXPORT __declspec(dllexport)
typedef unsigned short os_char;
#define _OS_CAT(L, x) L##x
#define OS_LIT(x) _OS_CAT(L, x)
#define os_snprintf _snwprintf
#define os_strchr wcschr
#define os_strcmp wcscmp
#define os_strcpy wcscpy
#define os_strlen wcslen
#define strncasecmp _strnicmp // stupid!
#define os_fopen _wfopen
// yuck :(
#define _os_open3(path, flags, mode) _wopen((path), (flags) | _O_BINARY, (mode))
#define _os_open2(path, flags) _wopen((path), (flags) | _O_BINARY)
#define _os_open(a, b, c, x, ...) x
#define os_open(...) _os_open(__VA_ARGS__, _os_open3, _os_open2, _)(__VA_ARGS__)
#define os_access _waccess
// ucrt defines __stat64 to _stat64. we want _wstat64 to be the actual function
#define _stat64(path, buf) _wstat64(path, buf)
#define os_stat _stat64
#define os_mkdir _wmkdir
#define os_unlink _wunlink
#define os_getenv _wgetenv
#define os_getcwd _wgetcwd
#define OS_DLPREFIX ""
#define OS_DLSUFFIX ".dll"
#define OS_MAIN wmain
static inline void *os_dlopen(const unsigned short *name) {
return LoadLibraryW(name);
}
static inline void *os_dlhandle(const unsigned short *name) {
return GetModuleHandleW(name);
}
static inline void *os_dlsym(void *m, const char *s) {
return (void *)GetProcAddress(m, s);
}
static inline int os_dlfile(void *m, unsigned short *buf, int sz) {
unsigned int n = GetModuleFileNameW(m, buf, sz);
if (n == 0 || n == sz) return -1;
// get the canonical capitalisation, as for some reason GetModuleFileName()
// returns all lowercase. this doesn't really matter except it looks nicer
GetLongPathNameW(buf, buf, n + 1);
// the drive letter will also be lower case, if it is an actual drive letter
// of course. it should be; I'm not gonna lose sleep over UNC paths and such
if (buf[0] >= L'a' && buf[0] <= L'z' && buf[1] == L':') buf[0] &= ~32u;
return n;
}
static inline bool os_mprot(void *addr, int len, int fl) {
unsigned long old;
return !!VirtualProtect(addr, len, fl, &old);
}
// SystemFunction036 is the *real* name of "RtlGenRandom," and is also
// incorrectly defined in system headers. Yay.
int __stdcall SystemFunction036(void *buf, unsigned long sz);
static inline void os_randombytes(void *buf, int sz) {
// if this call fails, the system is doomed, so loop until success and pray.
while (!SystemFunction036(buf, sz));
}
/* Further Windows-specific workarounds because Windows is great */
#ifndef PATH_MAX
// XXX: win32/crt has this dumb 260 limit even though the actual kernel imposes
// no limit (though apparently NTFS has a limit of 65535). Theoretically we
// could do some memes with UNC paths to extend it to at least have parity with
// Unix PATH_MAX (4096), but for now we just kind of accept that Windows is a
// disaster.
#define PATH_MAX MAX_PATH
#endif
// why does Windows prefix so many things with underscores?
#define alloca _alloca
#define read _read
#define write _write
#define close _close
#define fdopen _fdopen
#define dup _dup
#define dup2 _dup2
#define strdup _strdup
#define fstat _fstat64
#define lseek _lseeki64
#define O_RDONLY _O_RDONLY
#define O_RDWR _O_RDWR
#define O_CLOEXEC _O_NOINHERIT // and why did they rename this!?
#define O_CREAT _O_CREAT
#define O_EXCL _O_EXCL
// missing access() flags
#define F_OK 0
#define R_OK 4
#define W_OK 2
#define X_OK R_OK // there's no actual X bit, just pretend
// windows doesn't define this for some reason!? note: add others as needed...
#define S_ISDIR(mode) (((mode) & S_IFMT) == S_IFDIR)
// just dump this boilerplate here as well, I spose
#define OS_WINDOWS_ERROR(arrayname) \
FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM, 0, GetLastError(), \
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), arrayname, \
sizeof(arrayname), 0)
// vi: sw=4 ts=4 noet tw=80 cc=80
|
