blob: 25eaa0d1e087c6945461c191bfba85259bc66152 [file] [log] [blame]
Alexey Samsonov298dd7c2012-06-05 07:46:31 +00001//===-- sanitizer_win.cc --------------------------------------------------===//
Alexey Samsonovdde1f112012-06-05 07:05:10 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file is shared between AddressSanitizer and ThreadSanitizer
11// run-time libraries and implements windows-specific functions from
12// sanitizer_libc.h.
13//===----------------------------------------------------------------------===//
Evgeniy Stepanov0af67232013-03-19 14:33:38 +000014
15#include "sanitizer_platform.h"
16#if SANITIZER_WINDOWS
17
Dmitry Vyukov0ff6d2d2012-11-06 13:19:59 +000018#define WIN32_LEAN_AND_MEAN
19#define NOGDI
Alexey Samsonovdde1f112012-06-05 07:05:10 +000020#include <windows.h>
Timur Iskhodzhanov19853dd2014-07-11 11:57:41 +000021#include <dbghelp.h>
22#include <io.h>
Timur Iskhodzhanova04b33b2014-12-26 14:28:32 +000023#include <psapi.h>
Timur Iskhodzhanov19853dd2014-07-11 11:57:41 +000024#include <stdlib.h>
Alexey Samsonovdde1f112012-06-05 07:05:10 +000025
Alexey Samsonov201aa362012-06-06 09:43:32 +000026#include "sanitizer_common.h"
Alexey Samsonovdde1f112012-06-05 07:05:10 +000027#include "sanitizer_libc.h"
Dmitry Vyukovff198092013-01-14 14:28:06 +000028#include "sanitizer_mutex.h"
Sergey Matveevaf179b82013-05-08 12:45:55 +000029#include "sanitizer_placement_new.h"
30#include "sanitizer_stacktrace.h"
Alexey Samsonovdde1f112012-06-05 07:05:10 +000031
Alexey Samsonovdde1f112012-06-05 07:05:10 +000032namespace __sanitizer {
33
Peter Collingbourne6f4be192013-05-08 14:43:49 +000034#include "sanitizer_syscall_generic.inc"
35
Alexey Samsonov4b1f1032012-06-07 07:13:46 +000036// --------------------- sanitizer_common.h
Kostya Serebryanyf22c6972012-11-23 15:38:49 +000037uptr GetPageSize() {
Reid Kleckner0765fbc2016-02-18 17:58:22 +000038 SYSTEM_INFO si;
39 GetSystemInfo(&si);
40 return si.dwPageSize;
Kostya Serebryanyf22c6972012-11-23 15:38:49 +000041}
42
43uptr GetMmapGranularity() {
Reid Kleckner0765fbc2016-02-18 17:58:22 +000044 SYSTEM_INFO si;
45 GetSystemInfo(&si);
46 return si.dwAllocationGranularity;
Kostya Serebryanyf22c6972012-11-23 15:38:49 +000047}
48
Timur Iskhodzhanov4245f782013-07-16 09:47:39 +000049uptr GetMaxVirtualAddress() {
50 SYSTEM_INFO si;
51 GetSystemInfo(&si);
52 return (uptr)si.lpMaximumApplicationAddress;
53}
54
Alexey Samsonovae9b18b2012-11-09 14:45:30 +000055bool FileExists(const char *filename) {
Reid Klecknere96833e2015-08-10 23:40:27 +000056 return ::GetFileAttributesA(filename) != INVALID_FILE_ATTRIBUTES;
Alexey Samsonovae9b18b2012-11-09 14:45:30 +000057}
58
Peter Collingbourneffaf2ea2013-05-17 16:56:53 +000059uptr internal_getpid() {
Alexey Samsonovee072902012-06-06 09:26:25 +000060 return GetProcessId(GetCurrentProcess());
61}
62
Timur Iskhodzhanov2dee3dd2013-03-25 22:04:29 +000063// In contrast to POSIX, on Windows GetCurrentThreadId()
64// returns a system-unique identifier.
65uptr GetTid() {
Alexey Samsonov70afb912012-06-15 06:37:34 +000066 return GetCurrentThreadId();
67}
68
Timur Iskhodzhanov2dee3dd2013-03-25 22:04:29 +000069uptr GetThreadSelf() {
70 return GetTid();
71}
72
Dmitry Vyukovaa8fa602014-09-01 11:44:59 +000073#if !SANITIZER_GO
Alexey Samsonovcf4d3a02012-06-07 07:32:00 +000074void GetThreadStackTopAndBottom(bool at_initialization, uptr *stack_top,
Alexey Samsonov4b1f1032012-06-07 07:13:46 +000075 uptr *stack_bottom) {
76 CHECK(stack_top);
77 CHECK(stack_bottom);
78 MEMORY_BASIC_INFORMATION mbi;
Kostya Serebryany98390d02012-06-20 15:19:17 +000079 CHECK_NE(VirtualQuery(&mbi /* on stack */, &mbi, sizeof(mbi)), 0);
Alexey Samsonov4b1f1032012-06-07 07:13:46 +000080 // FIXME: is it possible for the stack to not be a single allocation?
81 // Are these values what ASan expects to get (reserved, not committed;
82 // including stack guard page) ?
83 *stack_top = (uptr)mbi.BaseAddress + mbi.RegionSize;
84 *stack_bottom = (uptr)mbi.AllocationBase;
85}
Dmitry Vyukovaa8fa602014-09-01 11:44:59 +000086#endif // #if !SANITIZER_GO
Alexey Samsonov4b1f1032012-06-07 07:13:46 +000087
Anna Zaks8d225202015-11-20 18:42:01 +000088void *MmapOrDie(uptr size, const char *mem_type, bool raw_report) {
Alexey Samsonovee072902012-06-06 09:26:25 +000089 void *rv = VirtualAlloc(0, size, MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE);
Reid Kleckner7d0f22e2015-08-12 23:55:38 +000090 if (rv == 0)
Anna Zaks8d225202015-11-20 18:42:01 +000091 ReportMmapFailureAndDie(size, mem_type, "allocate",
92 GetLastError(), raw_report);
Alexey Samsonovee072902012-06-06 09:26:25 +000093 return rv;
94}
95
96void UnmapOrDie(void *addr, uptr size) {
Timur Iskhodzhanov89608af2015-03-31 16:39:20 +000097 if (!size || !addr)
98 return;
99
Alexey Samsonove95e29c2012-06-06 15:47:40 +0000100 if (VirtualFree(addr, size, MEM_DECOMMIT) == 0) {
Timur Iskhodzhanov364b8b82014-03-19 08:23:00 +0000101 Report("ERROR: %s failed to "
102 "deallocate 0x%zx (%zd) bytes at address %p (error code: %d)\n",
103 SanitizerToolName, size, size, addr, GetLastError());
Alexey Samsonov40d5b772012-06-06 16:15:07 +0000104 CHECK("unable to unmap" && 0);
Alexey Samsonove95e29c2012-06-06 15:47:40 +0000105 }
Alexey Samsonovee072902012-06-06 09:26:25 +0000106}
107
Evgeniy Stepanov8e9c70b2015-05-29 22:31:28 +0000108void *MmapFixedNoReserve(uptr fixed_addr, uptr size, const char *name) {
Kostya Serebryany98066282012-12-13 05:36:00 +0000109 // FIXME: is this really "NoReserve"? On Win32 this does not matter much,
110 // but on Win64 it does.
Evgeniy Stepanov8e9c70b2015-05-29 22:31:28 +0000111 (void)name; // unsupported
Dmitry Vyukov0ff6d2d2012-11-06 13:19:59 +0000112 void *p = VirtualAlloc((LPVOID)fixed_addr, size,
113 MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE);
114 if (p == 0)
Timur Iskhodzhanov364b8b82014-03-19 08:23:00 +0000115 Report("ERROR: %s failed to "
116 "allocate %p (%zd) bytes at %p (error code: %d)\n",
117 SanitizerToolName, size, size, fixed_addr, GetLastError());
Dmitry Vyukov0ff6d2d2012-11-06 13:19:59 +0000118 return p;
Alexey Samsonovc70d1082012-06-14 14:42:58 +0000119}
120
Kostya Serebryany98066282012-12-13 05:36:00 +0000121void *MmapFixedOrDie(uptr fixed_addr, uptr size) {
Kostya Serebryanya1670872012-12-13 05:51:02 +0000122 return MmapFixedNoReserve(fixed_addr, size);
Kostya Serebryany98066282012-12-13 05:36:00 +0000123}
124
Kostya Serebryany57bfdb02013-12-13 15:03:49 +0000125void *MmapNoReserveOrDie(uptr size, const char *mem_type) {
126 // FIXME: make this really NoReserve?
127 return MmapOrDie(size, mem_type);
128}
129
Evgeniy Stepanov8e9c70b2015-05-29 22:31:28 +0000130void *MmapNoAccess(uptr fixed_addr, uptr size, const char *name) {
131 (void)name; // unsupported
Timur Iskhodzhanove5935ef2015-02-02 15:04:23 +0000132 void *res = VirtualAlloc((LPVOID)fixed_addr, size,
133 MEM_RESERVE | MEM_COMMIT, PAGE_NOACCESS);
134 if (res == 0)
135 Report("WARNING: %s failed to "
136 "mprotect %p (%zd) bytes at %p (error code: %d)\n",
137 SanitizerToolName, size, size, fixed_addr, GetLastError());
138 return res;
Alexey Samsonovc70d1082012-06-14 14:42:58 +0000139}
140
Timur Iskhodzhanovea1f3322015-04-10 15:02:19 +0000141bool MprotectNoAccess(uptr addr, uptr size) {
142 DWORD old_protection;
143 return VirtualProtect((LPVOID)addr, size, PAGE_NOACCESS, &old_protection);
144}
145
146
Timur Iskhodzhanov167f9e42013-02-08 12:02:00 +0000147void FlushUnneededShadowMemory(uptr addr, uptr size) {
148 // This is almost useless on 32-bits.
Yury Gribov8f848ff2015-02-03 10:15:15 +0000149 // FIXME: add madvise-analog when we move to 64-bits.
Timur Iskhodzhanov167f9e42013-02-08 12:02:00 +0000150}
151
Kostya Serebryanyc6338ac2015-01-21 02:05:31 +0000152void NoHugePagesInRegion(uptr addr, uptr size) {
153 // FIXME: probably similar to FlushUnneededShadowMemory.
154}
155
Yury Gribov8f848ff2015-02-03 10:15:15 +0000156void DontDumpShadowMemory(uptr addr, uptr length) {
157 // This is almost useless on 32-bits.
158 // FIXME: add madvise-analog when we move to 64-bits.
159}
160
Alexey Samsonov40e51282012-06-15 07:29:14 +0000161bool MemoryRangeIsAvailable(uptr range_start, uptr range_end) {
Timur Iskhodzhanova04b33b2014-12-26 14:28:32 +0000162 MEMORY_BASIC_INFORMATION mbi;
163 CHECK(VirtualQuery((void *)range_start, &mbi, sizeof(mbi)));
Timur Iskhodzhanove5935ef2015-02-02 15:04:23 +0000164 return mbi.Protect == PAGE_NOACCESS &&
Timur Iskhodzhanova04b33b2014-12-26 14:28:32 +0000165 (uptr)mbi.BaseAddress + mbi.RegionSize >= range_end;
Alexey Samsonov40e51282012-06-15 07:29:14 +0000166}
167
Alexey Samsonov961276a2012-07-03 08:24:14 +0000168void *MapFileToMemory(const char *file_name, uptr *buff_size) {
169 UNIMPLEMENTED();
170}
171
Daniel Sandersadf1fcc2015-07-31 11:29:25 +0000172void *MapWritableFileToMemory(void *addr, uptr size, fd_t fd, OFF_T offset) {
Evgeniy Stepanova00ff192014-05-28 08:26:24 +0000173 UNIMPLEMENTED();
174}
175
Alexey Samsonov7d238542013-03-14 11:29:06 +0000176static const int kMaxEnvNameLength = 128;
Dmitry Vyukove979c542013-06-10 10:02:02 +0000177static const DWORD kMaxEnvValueLength = 32767;
Alexey Samsonov83e76222013-03-14 11:10:23 +0000178
179namespace {
180
181struct EnvVariable {
182 char name[kMaxEnvNameLength];
183 char value[kMaxEnvValueLength];
184};
185
186} // namespace
187
188static const int kEnvVariables = 5;
189static EnvVariable env_vars[kEnvVariables];
190static int num_env_vars;
191
Alexey Samsonov0c53a382012-06-14 14:07:21 +0000192const char *GetEnv(const char *name) {
Alexey Samsonov83e76222013-03-14 11:10:23 +0000193 // Note: this implementation caches the values of the environment variables
194 // and limits their quantity.
195 for (int i = 0; i < num_env_vars; i++) {
196 if (0 == internal_strcmp(name, env_vars[i].name))
197 return env_vars[i].value;
198 }
199 CHECK_LT(num_env_vars, kEnvVariables);
200 DWORD rv = GetEnvironmentVariableA(name, env_vars[num_env_vars].value,
201 kMaxEnvValueLength);
202 if (rv > 0 && rv < kMaxEnvValueLength) {
203 CHECK_LT(internal_strlen(name), kMaxEnvNameLength);
204 internal_strncpy(env_vars[num_env_vars].name, name, kMaxEnvNameLength);
205 num_env_vars++;
206 return env_vars[num_env_vars - 1].value;
207 }
Alexey Samsonov0c53a382012-06-14 14:07:21 +0000208 return 0;
209}
210
Alexey Samsonov58a3c582012-06-18 08:44:30 +0000211const char *GetPwd() {
212 UNIMPLEMENTED();
213}
214
Alexey Samsonov9211bd32013-02-18 07:17:12 +0000215u32 GetUid() {
216 UNIMPLEMENTED();
217}
218
Timur Iskhodzhanov64fc8e42014-12-30 14:44:12 +0000219namespace {
220struct ModuleInfo {
Timur Iskhodzhanovb97bcc42015-04-06 12:49:30 +0000221 const char *filepath;
Timur Iskhodzhanov64fc8e42014-12-30 14:44:12 +0000222 uptr base_address;
223 uptr end_address;
224};
225
Dmitry Vyukov8455cf02015-11-12 16:29:24 +0000226#ifndef SANITIZER_GO
Timur Iskhodzhanov64fc8e42014-12-30 14:44:12 +0000227int CompareModulesBase(const void *pl, const void *pr) {
Timur Iskhodzhanovb97bcc42015-04-06 12:49:30 +0000228 const ModuleInfo *l = (ModuleInfo *)pl, *r = (ModuleInfo *)pr;
229 if (l->base_address < r->base_address)
Timur Iskhodzhanov64fc8e42014-12-30 14:44:12 +0000230 return -1;
Timur Iskhodzhanovb97bcc42015-04-06 12:49:30 +0000231 return l->base_address > r->base_address;
Timur Iskhodzhanov64fc8e42014-12-30 14:44:12 +0000232}
Dmitry Vyukov8455cf02015-11-12 16:29:24 +0000233#endif
Timur Iskhodzhanova023e062014-12-30 15:30:19 +0000234} // namespace
Timur Iskhodzhanov64fc8e42014-12-30 14:44:12 +0000235
Dmitry Vyukovb3381fa2015-02-16 13:51:17 +0000236#ifndef SANITIZER_GO
Alexey Samsonovae1e1712012-06-15 06:08:19 +0000237void DumpProcessMap() {
Timur Iskhodzhanova04b33b2014-12-26 14:28:32 +0000238 Report("Dumping process modules:\n");
Timur Iskhodzhanovb97bcc42015-04-06 12:49:30 +0000239 InternalScopedBuffer<LoadedModule> modules(kMaxNumberOfModules);
240 uptr num_modules =
241 GetListOfModules(modules.data(), kMaxNumberOfModules, nullptr);
Timur Iskhodzhanova04b33b2014-12-26 14:28:32 +0000242
Timur Iskhodzhanovb97bcc42015-04-06 12:49:30 +0000243 InternalScopedBuffer<ModuleInfo> module_infos(num_modules);
244 for (size_t i = 0; i < num_modules; ++i) {
245 module_infos[i].filepath = modules[i].full_name();
246 module_infos[i].base_address = modules[i].base_address();
Peter Collingbourned2f24b62016-01-15 02:51:26 +0000247 module_infos[i].end_address = modules[i].ranges().front()->end;
Timur Iskhodzhanova04b33b2014-12-26 14:28:32 +0000248 }
Timur Iskhodzhanovb97bcc42015-04-06 12:49:30 +0000249 qsort(module_infos.data(), num_modules, sizeof(ModuleInfo),
250 CompareModulesBase);
Timur Iskhodzhanov64fc8e42014-12-30 14:44:12 +0000251
252 for (size_t i = 0; i < num_modules; ++i) {
Timur Iskhodzhanovb97bcc42015-04-06 12:49:30 +0000253 const ModuleInfo &mi = module_infos[i];
Timur Iskhodzhanov64fc8e42014-12-30 14:44:12 +0000254 if (mi.end_address != 0) {
255 Printf("\t%p-%p %s\n", mi.base_address, mi.end_address,
Timur Iskhodzhanovb97bcc42015-04-06 12:49:30 +0000256 mi.filepath[0] ? mi.filepath : "[no name]");
Timur Iskhodzhanov1931b622015-04-06 12:54:06 +0000257 } else if (mi.filepath[0]) {
Timur Iskhodzhanovb97bcc42015-04-06 12:49:30 +0000258 Printf("\t??\?-??? %s\n", mi.filepath);
Timur Iskhodzhanova04b33b2014-12-26 14:28:32 +0000259 } else {
260 Printf("\t???\n");
261 }
262 }
Alexey Samsonovae1e1712012-06-15 06:08:19 +0000263}
Dmitry Vyukovb3381fa2015-02-16 13:51:17 +0000264#endif
Alexey Samsonovae1e1712012-06-15 06:08:19 +0000265
Alexey Samsonov34e2b282014-08-12 22:31:19 +0000266void DisableCoreDumperIfNecessary() {
Timur Iskhodzhanov7d5c81d2014-05-06 08:21:50 +0000267 // Do nothing.
Alexey Samsonovae1e1712012-06-15 06:08:19 +0000268}
269
Alexey Samsonov97ca3062012-09-17 09:12:39 +0000270void ReExec() {
271 UNIMPLEMENTED();
272}
273
Sergey Matveev6cb47a082014-05-19 12:53:03 +0000274void PrepareForSandboxing(__sanitizer_sandbox_arguments *args) {
Timur Iskhodzhanovd0be0a92015-04-23 12:58:11 +0000275#if !SANITIZER_GO
276 CovPrepareForSandboxing(args);
277#endif
Alexander Potapenko1746f552012-12-10 13:10:40 +0000278}
279
Alexey Samsonov97ca3062012-09-17 09:12:39 +0000280bool StackSizeIsUnlimited() {
281 UNIMPLEMENTED();
Alexey Samsonov97ca3062012-09-17 09:12:39 +0000282}
283
284void SetStackSizeLimitInBytes(uptr limit) {
285 UNIMPLEMENTED();
286}
287
Alexey Samsonov34e2b282014-08-12 22:31:19 +0000288bool AddressSpaceIsUnlimited() {
289 UNIMPLEMENTED();
290}
291
292void SetAddressSpaceUnlimited() {
293 UNIMPLEMENTED();
294}
295
Anna Zaks22490492015-02-27 03:12:19 +0000296bool IsPathSeparator(const char c) {
297 return c == '\\' || c == '/';
298}
299
300bool IsAbsolutePath(const char *path) {
301 UNIMPLEMENTED();
302}
303
Alexey Samsonov70afb912012-06-15 06:37:34 +0000304void SleepForSeconds(int seconds) {
305 Sleep(seconds * 1000);
306}
307
Alexey Samsonov58a3c582012-06-18 08:44:30 +0000308void SleepForMillis(int millis) {
309 Sleep(millis);
310}
311
Dmitry Vyukove979c542013-06-10 10:02:02 +0000312u64 NanoTime() {
313 return 0;
314}
315
Alexey Samsonov70afb912012-06-15 06:37:34 +0000316void Abort() {
Timur Iskhodzhanovb8373bc2014-12-26 12:25:54 +0000317 if (::IsDebuggerPresent())
318 __debugbreak();
319 internal__exit(3);
Alexey Samsonov70afb912012-06-15 06:37:34 +0000320}
321
Reid Kleckner646386e2015-08-03 19:51:18 +0000322// Read the file to extract the ImageBase field from the PE header. If ASLR is
323// disabled and this virtual address is available, the loader will typically
324// load the image at this address. Therefore, we call it the preferred base. Any
325// addresses in the DWARF typically assume that the object has been loaded at
326// this address.
327static uptr GetPreferredBase(const char *modname) {
328 fd_t fd = OpenFile(modname, RdOnly, nullptr);
329 if (fd == kInvalidFd)
330 return 0;
331 FileCloser closer(fd);
332
333 // Read just the DOS header.
334 IMAGE_DOS_HEADER dos_header;
335 uptr bytes_read;
336 if (!ReadFromFile(fd, &dos_header, sizeof(dos_header), &bytes_read) ||
337 bytes_read != sizeof(dos_header))
338 return 0;
339
340 // The file should start with the right signature.
341 if (dos_header.e_magic != IMAGE_DOS_SIGNATURE)
342 return 0;
343
344 // The layout at e_lfanew is:
345 // "PE\0\0"
346 // IMAGE_FILE_HEADER
347 // IMAGE_OPTIONAL_HEADER
348 // Seek to e_lfanew and read all that data.
349 char buf[4 + sizeof(IMAGE_FILE_HEADER) + sizeof(IMAGE_OPTIONAL_HEADER)];
350 if (::SetFilePointer(fd, dos_header.e_lfanew, nullptr, FILE_BEGIN) ==
351 INVALID_SET_FILE_POINTER)
352 return 0;
353 if (!ReadFromFile(fd, &buf[0], sizeof(buf), &bytes_read) ||
354 bytes_read != sizeof(buf))
355 return 0;
356
357 // Check for "PE\0\0" before the PE header.
358 char *pe_sig = &buf[0];
359 if (internal_memcmp(pe_sig, "PE\0\0", 4) != 0)
360 return 0;
361
362 // Skip over IMAGE_FILE_HEADER. We could do more validation here if we wanted.
363 IMAGE_OPTIONAL_HEADER *pe_header =
364 (IMAGE_OPTIONAL_HEADER *)(pe_sig + 4 + sizeof(IMAGE_FILE_HEADER));
365
366 // Check for more magic in the PE header.
367 if (pe_header->Magic != IMAGE_NT_OPTIONAL_HDR_MAGIC)
368 return 0;
369
370 // Finally, return the ImageBase.
371 return (uptr)pe_header->ImageBase;
372}
373
Dmitry Vyukovb8da7942015-11-12 18:50:19 +0000374#ifndef SANITIZER_GO
Alexey Samsonov7a36e612013-09-10 14:36:16 +0000375uptr GetListOfModules(LoadedModule *modules, uptr max_modules,
376 string_predicate_t filter) {
Timur Iskhodzhanovb97bcc42015-04-06 12:49:30 +0000377 HANDLE cur_process = GetCurrentProcess();
378
379 // Query the list of modules. Start by assuming there are no more than 256
380 // modules and retry if that's not sufficient.
381 HMODULE *hmodules = 0;
382 uptr modules_buffer_size = sizeof(HMODULE) * 256;
383 DWORD bytes_required;
384 while (!hmodules) {
385 hmodules = (HMODULE *)MmapOrDie(modules_buffer_size, __FUNCTION__);
386 CHECK(EnumProcessModules(cur_process, hmodules, modules_buffer_size,
387 &bytes_required));
388 if (bytes_required > modules_buffer_size) {
389 // Either there turned out to be more than 256 hmodules, or new hmodules
390 // could have loaded since the last try. Retry.
391 UnmapOrDie(hmodules, modules_buffer_size);
392 hmodules = 0;
393 modules_buffer_size = bytes_required;
394 }
395 }
396
397 // |num_modules| is the number of modules actually present,
398 // |count| is the number of modules we return.
399 size_t nun_modules = bytes_required / sizeof(HMODULE),
400 count = 0;
401 for (size_t i = 0; i < nun_modules && count < max_modules; ++i) {
402 HMODULE handle = hmodules[i];
403 MODULEINFO mi;
404 if (!GetModuleInformation(cur_process, handle, &mi, sizeof(mi)))
405 continue;
406
Reid Kleckner646386e2015-08-03 19:51:18 +0000407 // Get the UTF-16 path and convert to UTF-8.
408 wchar_t modname_utf16[kMaxPathLength];
409 int modname_utf16_len =
410 GetModuleFileNameW(handle, modname_utf16, kMaxPathLength);
411 if (modname_utf16_len == 0)
412 modname_utf16[0] = '\0';
413 char module_name[kMaxPathLength];
414 int module_name_len =
415 ::WideCharToMultiByte(CP_UTF8, 0, modname_utf16, modname_utf16_len + 1,
416 &module_name[0], kMaxPathLength, NULL, NULL);
417 module_name[module_name_len] = '\0';
Timur Iskhodzhanovb97bcc42015-04-06 12:49:30 +0000418
419 if (filter && !filter(module_name))
420 continue;
421
422 uptr base_address = (uptr)mi.lpBaseOfDll;
423 uptr end_address = (uptr)mi.lpBaseOfDll + mi.SizeOfImage;
Reid Kleckner646386e2015-08-03 19:51:18 +0000424
425 // Adjust the base address of the module so that we get a VA instead of an
426 // RVA when computing the module offset. This helps llvm-symbolizer find the
427 // right DWARF CU. In the common case that the image is loaded at it's
428 // preferred address, we will now print normal virtual addresses.
429 uptr preferred_base = GetPreferredBase(&module_name[0]);
430 uptr adjusted_base = base_address - preferred_base;
431
Timur Iskhodzhanovb97bcc42015-04-06 12:49:30 +0000432 LoadedModule *cur_module = &modules[count];
Reid Kleckner646386e2015-08-03 19:51:18 +0000433 cur_module->set(module_name, adjusted_base);
Timur Iskhodzhanovb97bcc42015-04-06 12:49:30 +0000434 // We add the whole module as one single address range.
435 cur_module->addAddressRange(base_address, end_address, /*executable*/ true);
436 count++;
437 }
438 UnmapOrDie(hmodules, modules_buffer_size);
439
440 return count;
Alexey Samsonov7a36e612013-09-10 14:36:16 +0000441};
442
Timur Iskhodzhanovad3ec822015-04-02 14:48:08 +0000443// We can't use atexit() directly at __asan_init time as the CRT is not fully
444// initialized at this point. Place the functions into a vector and use
445// atexit() as soon as it is ready for use (i.e. after .CRT$XIC initializers).
446InternalMmapVectorNoCtor<void (*)(void)> atexit_functions;
447
Alexey Samsonov70afb912012-06-15 06:37:34 +0000448int Atexit(void (*function)(void)) {
Timur Iskhodzhanovad3ec822015-04-02 14:48:08 +0000449 atexit_functions.push_back(function);
450 return 0;
Alexey Samsonov70afb912012-06-15 06:37:34 +0000451}
Timur Iskhodzhanovad3ec822015-04-02 14:48:08 +0000452
453static int RunAtexit() {
454 int ret = 0;
455 for (uptr i = 0; i < atexit_functions.size(); ++i) {
456 ret |= atexit(atexit_functions[i]);
457 }
458 return ret;
459}
460
461#pragma section(".CRT$XID", long, read) // NOLINT
Reid Klecknere4fabc92015-08-13 16:40:54 +0000462__declspec(allocate(".CRT$XID")) int (*__run_atexit)() = RunAtexit;
Dmitry Vyukov0ff6d2d2012-11-06 13:19:59 +0000463#endif
Alexey Samsonov70afb912012-06-15 06:37:34 +0000464
Alexey Samsonov4b1f1032012-06-07 07:13:46 +0000465// ------------------ sanitizer_libc.h
Timur Iskhodzhanov864308a2015-04-09 12:37:05 +0000466fd_t OpenFile(const char *filename, FileAccessMode mode, error_t *last_error) {
Reid Kleckner646386e2015-08-03 19:51:18 +0000467 fd_t res;
468 if (mode == RdOnly) {
469 res = CreateFile(filename, GENERIC_READ,
470 FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
471 nullptr, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, nullptr);
472 } else if (mode == WrOnly) {
473 res = CreateFile(filename, GENERIC_WRITE, 0, nullptr, CREATE_ALWAYS,
474 FILE_ATTRIBUTE_NORMAL, nullptr);
475 } else {
Timur Iskhodzhanov007435c2015-04-09 15:25:21 +0000476 UNIMPLEMENTED();
Reid Kleckner646386e2015-08-03 19:51:18 +0000477 }
Timur Iskhodzhanov007435c2015-04-09 15:25:21 +0000478 CHECK(res != kStdoutFd || kStdoutFd == kInvalidFd);
479 CHECK(res != kStderrFd || kStderrFd == kInvalidFd);
Timur Iskhodzhanov8a673682015-04-23 12:57:29 +0000480 if (res == kInvalidFd && last_error)
481 *last_error = GetLastError();
Timur Iskhodzhanov007435c2015-04-09 15:25:21 +0000482 return res;
Alexey Samsonov03c8b842012-06-05 08:32:53 +0000483}
484
Timur Iskhodzhanov864308a2015-04-09 12:37:05 +0000485void CloseFile(fd_t fd) {
Timur Iskhodzhanov007435c2015-04-09 15:25:21 +0000486 CloseHandle(fd);
Alexey Samsonovdde1f112012-06-05 07:05:10 +0000487}
488
Timur Iskhodzhanov2b391692015-04-09 13:38:14 +0000489bool ReadFromFile(fd_t fd, void *buff, uptr buff_size, uptr *bytes_read,
490 error_t *error_p) {
Reid Kleckner646386e2015-08-03 19:51:18 +0000491 CHECK(fd != kInvalidFd);
Hans Wennborg92e64122015-08-05 17:55:26 +0000492
493 // bytes_read can't be passed directly to ReadFile:
494 // uptr is unsigned long long on 64-bit Windows.
495 unsigned long num_read_long;
496
497 bool success = ::ReadFile(fd, buff, buff_size, &num_read_long, nullptr);
Reid Kleckner646386e2015-08-03 19:51:18 +0000498 if (!success && error_p)
499 *error_p = GetLastError();
Hans Wennborg92e64122015-08-05 17:55:26 +0000500 if (bytes_read)
501 *bytes_read = num_read_long;
Reid Kleckner646386e2015-08-03 19:51:18 +0000502 return success;
Timur Iskhodzhanov2b391692015-04-09 13:38:14 +0000503}
504
Timur Iskhodzhanovc2c9ea52015-04-08 17:42:57 +0000505bool SupportsColoredOutput(fd_t fd) {
506 // FIXME: support colored output.
507 return false;
508}
509
Timur Iskhodzhanove8a6fbb2015-04-09 14:11:25 +0000510bool WriteToFile(fd_t fd, const void *buff, uptr buff_size, uptr *bytes_written,
511 error_t *error_p) {
Timur Iskhodzhanov007435c2015-04-09 15:25:21 +0000512 CHECK(fd != kInvalidFd);
Timur Iskhodzhanovaeefb6a2014-02-04 23:28:30 +0000513
Reid Klecknere96833e2015-08-10 23:40:27 +0000514 // Handle null optional parameters.
515 error_t dummy_error;
516 error_p = error_p ? error_p : &dummy_error;
517 uptr dummy_bytes_written;
518 bytes_written = bytes_written ? bytes_written : &dummy_bytes_written;
519
520 // Initialize output parameters in case we fail.
521 *error_p = 0;
522 *bytes_written = 0;
523
524 // Map the conventional Unix fds 1 and 2 to Windows handles. They might be
525 // closed, in which case this will fail.
526 if (fd == kStdoutFd || fd == kStderrFd) {
527 fd = GetStdHandle(fd == kStdoutFd ? STD_OUTPUT_HANDLE : STD_ERROR_HANDLE);
528 if (fd == 0) {
529 *error_p = ERROR_INVALID_HANDLE;
530 return false;
531 }
Timur Iskhodzhanovaeefb6a2014-02-04 23:28:30 +0000532 }
533
Reid Klecknere96833e2015-08-10 23:40:27 +0000534 DWORD bytes_written_32;
535 if (!WriteFile(fd, buff, buff_size, &bytes_written_32, 0)) {
536 *error_p = GetLastError();
Timur Iskhodzhanov007435c2015-04-09 15:25:21 +0000537 return false;
538 } else {
Reid Klecknere96833e2015-08-10 23:40:27 +0000539 *bytes_written = bytes_written_32;
Timur Iskhodzhanove8a6fbb2015-04-09 14:11:25 +0000540 return true;
541 }
Alexey Samsonov03c8b842012-06-05 08:32:53 +0000542}
543
Timur Iskhodzhanova6600a92015-04-09 14:45:17 +0000544bool RenameFile(const char *oldpath, const char *newpath, error_t *error_p) {
545 UNIMPLEMENTED();
546}
547
Peter Collingbourne6f4be192013-05-08 14:43:49 +0000548uptr internal_sched_yield() {
Dmitry Vyukov0ff6d2d2012-11-06 13:19:59 +0000549 Sleep(0);
550 return 0;
Alexey Samsonov58a3c582012-06-18 08:44:30 +0000551}
552
Alexey Samsonovaadd1f22013-02-20 13:54:32 +0000553void internal__exit(int exitcode) {
Timur Iskhodzhanov895392d2013-11-26 09:40:39 +0000554 ExitProcess(exitcode);
Alexey Samsonovaadd1f22013-02-20 13:54:32 +0000555}
556
Evgeniy Stepanova00ff192014-05-28 08:26:24 +0000557uptr internal_ftruncate(fd_t fd, uptr size) {
558 UNIMPLEMENTED();
559}
560
Kostya Serebryany6c54a6b2014-12-09 01:22:59 +0000561uptr GetRSS() {
562 return 0;
563}
564
Kostya Serebryany43eb7732014-12-16 19:13:01 +0000565void *internal_start_thread(void (*func)(void *arg), void *arg) { return 0; }
Hans Wennborg7dd94572014-12-16 20:46:05 +0000566void internal_join_thread(void *th) { }
Kostya Serebryany43eb7732014-12-16 19:13:01 +0000567
Dmitry Vyukovf22982b2013-01-14 07:51:39 +0000568// ---------------------- BlockingMutex ---------------- {{{1
Dmitry Vyukovfa67ed42013-02-04 10:42:38 +0000569const uptr LOCK_UNINITIALIZED = 0;
570const uptr LOCK_READY = (uptr)-1;
Dmitry Vyukovf22982b2013-01-14 07:51:39 +0000571
572BlockingMutex::BlockingMutex(LinkerInitialized li) {
573 // FIXME: see comments in BlockingMutex::Lock() for the details.
574 CHECK(li == LINKER_INITIALIZED || owner_ == LOCK_UNINITIALIZED);
575
576 CHECK(sizeof(CRITICAL_SECTION) <= sizeof(opaque_storage_));
577 InitializeCriticalSection((LPCRITICAL_SECTION)opaque_storage_);
578 owner_ = LOCK_READY;
579}
580
Alexey Samsonova097f7b2013-03-14 13:30:56 +0000581BlockingMutex::BlockingMutex() {
582 CHECK(sizeof(CRITICAL_SECTION) <= sizeof(opaque_storage_));
583 InitializeCriticalSection((LPCRITICAL_SECTION)opaque_storage_);
584 owner_ = LOCK_READY;
585}
586
Dmitry Vyukovf22982b2013-01-14 07:51:39 +0000587void BlockingMutex::Lock() {
588 if (owner_ == LOCK_UNINITIALIZED) {
589 // FIXME: hm, global BlockingMutex objects are not initialized?!?
590 // This might be a side effect of the clang+cl+link Frankenbuild...
591 new(this) BlockingMutex((LinkerInitialized)(LINKER_INITIALIZED + 1));
592
593 // FIXME: If it turns out the linker doesn't invoke our
594 // constructors, we should probably manually Lock/Unlock all the global
595 // locks while we're starting in one thread to avoid double-init races.
596 }
597 EnterCriticalSection((LPCRITICAL_SECTION)opaque_storage_);
Dmitry Vyukov7981ea82013-02-04 08:07:45 +0000598 CHECK_EQ(owner_, LOCK_READY);
Dmitry Vyukovf22982b2013-01-14 07:51:39 +0000599 owner_ = GetThreadSelf();
600}
601
602void BlockingMutex::Unlock() {
Dmitry Vyukov7981ea82013-02-04 08:07:45 +0000603 CHECK_EQ(owner_, GetThreadSelf());
Dmitry Vyukovf22982b2013-01-14 07:51:39 +0000604 owner_ = LOCK_READY;
605 LeaveCriticalSection((LPCRITICAL_SECTION)opaque_storage_);
606}
607
Alexey Samsonovdb7d9652013-03-11 15:45:20 +0000608void BlockingMutex::CheckLocked() {
609 CHECK_EQ(owner_, GetThreadSelf());
610}
611
Evgeniy Stepanov5697b582013-03-13 08:19:53 +0000612uptr GetTlsSize() {
613 return 0;
614}
615
616void InitTlsSize() {
617}
618
Sergey Matveev954c6ef12013-05-07 14:41:43 +0000619void GetThreadStackAndTls(bool main, uptr *stk_addr, uptr *stk_size,
620 uptr *tls_addr, uptr *tls_size) {
Dmitry Vyukovb278f122013-06-10 10:30:54 +0000621#ifdef SANITIZER_GO
622 *stk_addr = 0;
623 *stk_size = 0;
624 *tls_addr = 0;
625 *tls_size = 0;
626#else
Sergey Matveev954c6ef12013-05-07 14:41:43 +0000627 uptr stack_top, stack_bottom;
628 GetThreadStackTopAndBottom(main, &stack_top, &stack_bottom);
629 *stk_addr = stack_bottom;
630 *stk_size = stack_top - stack_bottom;
631 *tls_addr = 0;
632 *tls_size = 0;
Dmitry Vyukovb278f122013-06-10 10:30:54 +0000633#endif
Sergey Matveev954c6ef12013-05-07 14:41:43 +0000634}
635
Dmitry Vyukovaa8fa602014-09-01 11:44:59 +0000636#if !SANITIZER_GO
Evgeniy Stepanov8eb82042015-01-22 13:47:12 +0000637void BufferedStackTrace::SlowUnwindStack(uptr pc, u32 max_depth) {
Alexey Samsonov3e8467b2014-03-04 12:21:28 +0000638 CHECK_GE(max_depth, 2);
Sergey Matveevaf179b82013-05-08 12:45:55 +0000639 // FIXME: CaptureStackBackTrace might be too slow for us.
640 // FIXME: Compare with StackWalk64.
641 // FIXME: Look at LLVMUnhandledExceptionFilter in Signals.inc
Timur Iskhodzhanov1f1c7ec2013-11-09 13:59:12 +0000642 size = CaptureStackBackTrace(2, Min(max_depth, kStackTraceMax),
643 (void**)trace, 0);
Timur Iskhodzhanov89a346c2013-12-10 08:30:39 +0000644 if (size == 0)
645 return;
646
Sergey Matveevaf179b82013-05-08 12:45:55 +0000647 // Skip the RTL frames by searching for the PC in the stacktrace.
Timur Iskhodzhanov1f1c7ec2013-11-09 13:59:12 +0000648 uptr pc_location = LocatePcInTrace(pc);
649 PopStackFrames(pc_location);
Sergey Matveevaf179b82013-05-08 12:45:55 +0000650}
651
Alexey Samsonov9c859272014-10-26 03:35:14 +0000652void BufferedStackTrace::SlowUnwindStackWithContext(uptr pc, void *context,
Evgeniy Stepanov8eb82042015-01-22 13:47:12 +0000653 u32 max_depth) {
Timur Iskhodzhanov19853dd2014-07-11 11:57:41 +0000654 CONTEXT ctx = *(CONTEXT *)context;
655 STACKFRAME64 stack_frame;
656 memset(&stack_frame, 0, sizeof(stack_frame));
657 size = 0;
658#if defined(_WIN64)
659 int machine_type = IMAGE_FILE_MACHINE_AMD64;
660 stack_frame.AddrPC.Offset = ctx.Rip;
661 stack_frame.AddrFrame.Offset = ctx.Rbp;
662 stack_frame.AddrStack.Offset = ctx.Rsp;
663#else
664 int machine_type = IMAGE_FILE_MACHINE_I386;
665 stack_frame.AddrPC.Offset = ctx.Eip;
666 stack_frame.AddrFrame.Offset = ctx.Ebp;
667 stack_frame.AddrStack.Offset = ctx.Esp;
668#endif
669 stack_frame.AddrPC.Mode = AddrModeFlat;
670 stack_frame.AddrFrame.Mode = AddrModeFlat;
671 stack_frame.AddrStack.Mode = AddrModeFlat;
672 while (StackWalk64(machine_type, GetCurrentProcess(), GetCurrentThread(),
673 &stack_frame, &ctx, NULL, &SymFunctionTableAccess64,
674 &SymGetModuleBase64, NULL) &&
675 size < Min(max_depth, kStackTraceMax)) {
Hans Wennborgf89a3f862014-10-26 19:27:02 +0000676 trace_buffer[size++] = (uptr)stack_frame.AddrPC.Offset;
Timur Iskhodzhanov19853dd2014-07-11 11:57:41 +0000677 }
Evgeniy Stepanove5a447d2014-02-11 13:57:17 +0000678}
Dmitry Vyukovaa8fa602014-09-01 11:44:59 +0000679#endif // #if !SANITIZER_GO
Evgeniy Stepanove5a447d2014-02-11 13:57:17 +0000680
Alexey Samsonov3a41ed12014-12-11 18:30:25 +0000681void ReportFile::Write(const char *buffer, uptr length) {
682 SpinMutexLock l(mu);
683 ReopenIfNecessary();
Timur Iskhodzhanove8a6fbb2015-04-09 14:11:25 +0000684 if (!WriteToFile(fd, buffer, length)) {
Reid Klecknerd483c072013-09-05 03:19:57 +0000685 // stderr may be closed, but we may be able to print to the debugger
686 // instead. This is the case when launching a program from Visual Studio,
687 // and the following routine should write to its console.
688 OutputDebugStringA(buffer);
689 }
690}
691
Alexander Potapenkod8d490e2014-01-28 11:12:29 +0000692void SetAlternateSignalStack() {
693 // FIXME: Decide what to do on Windows.
694}
695
696void UnsetAlternateSignalStack() {
697 // FIXME: Decide what to do on Windows.
698}
699
Alexander Potapenkoea4a0db2014-01-31 15:11:11 +0000700void InstallDeadlySignalHandlers(SignalHandlerType handler) {
701 (void)handler;
Alexander Potapenko789e3e12014-01-31 13:10:07 +0000702 // FIXME: Decide what to do on Windows.
703}
704
Anna Zaksc77a0802016-02-02 02:01:17 +0000705bool IsHandledDeadlySignal(int signum) {
Alexander Potapenko789e3e12014-01-31 13:10:07 +0000706 // FIXME: Decide what to do on Windows.
707 return false;
708}
709
Alexey Samsonov1947bf92014-09-17 17:56:15 +0000710bool IsAccessibleMemoryRange(uptr beg, uptr size) {
Peter Collingbournea68d90f2015-07-02 22:08:38 +0000711 SYSTEM_INFO si;
712 GetNativeSystemInfo(&si);
713 uptr page_size = si.dwPageSize;
714 uptr page_mask = ~(page_size - 1);
715
716 for (uptr page = beg & page_mask, end = (beg + size - 1) & page_mask;
717 page <= end;) {
718 MEMORY_BASIC_INFORMATION info;
719 if (VirtualQuery((LPCVOID)page, &info, sizeof(info)) != sizeof(info))
720 return false;
721
722 if (info.Protect == 0 || info.Protect == PAGE_NOACCESS ||
723 info.Protect == PAGE_EXECUTE)
724 return false;
725
726 if (info.RegionSize == 0)
727 return false;
728
729 page += info.RegionSize;
730 }
731
Alexey Samsonov1947bf92014-09-17 17:56:15 +0000732 return true;
733}
734
Dmitry Vyukovdf01bdc2015-03-02 17:45:18 +0000735SignalContext SignalContext::Create(void *siginfo, void *context) {
Evgeniy Stepanovaa42f292016-02-09 00:28:57 +0000736 EXCEPTION_RECORD *exception_record = (EXCEPTION_RECORD *)siginfo;
737 CONTEXT *context_record = (CONTEXT *)context;
Dmitry Vyukovdf01bdc2015-03-02 17:45:18 +0000738
739 uptr pc = (uptr)exception_record->ExceptionAddress;
740#ifdef _WIN64
741 uptr bp = (uptr)context_record->Rbp;
742 uptr sp = (uptr)context_record->Rsp;
743#else
744 uptr bp = (uptr)context_record->Ebp;
745 uptr sp = (uptr)context_record->Esp;
746#endif
747 uptr access_addr = exception_record->ExceptionInformation[1];
748
Reid Klecknerceda8832016-02-11 16:44:35 +0000749 // The contents of this array are documented at
750 // https://msdn.microsoft.com/en-us/library/windows/desktop/aa363082(v=vs.85).aspx
751 // The first element indicates read as 0, write as 1, or execute as 8. The
752 // second element is the faulting address.
753 WriteFlag write_flag = SignalContext::UNKNOWN;
754 switch (exception_record->ExceptionInformation[0]) {
755 case 0: write_flag = SignalContext::READ; break;
756 case 1: write_flag = SignalContext::WRITE; break;
757 case 8: write_flag = SignalContext::UNKNOWN; break;
758 }
759 bool is_memory_access = write_flag != SignalContext::UNKNOWN;
Kostya Serebryany2b9be252016-02-04 02:02:09 +0000760 return SignalContext(context, access_addr, pc, sp, bp, is_memory_access,
Evgeniy Stepanov0d7839d2016-02-08 23:01:06 +0000761 write_flag);
Dmitry Vyukovdf01bdc2015-03-02 17:45:18 +0000762}
763
Yury Gribovc019a572015-06-04 07:29:43 +0000764uptr ReadBinaryName(/*out*/char *buf, uptr buf_len) {
765 // FIXME: Actually implement this function.
766 CHECK_GT(buf_len, 0);
767 buf[0] = 0;
768 return 0;
769}
770
Evgeniy Stepanov30257172015-07-28 21:01:42 +0000771uptr ReadLongProcessName(/*out*/char *buf, uptr buf_len) {
772 return ReadBinaryName(buf, buf_len);
773}
774
Adhemerval Zanella05636862015-09-11 13:55:00 +0000775void CheckVMASize() {
776 // Do nothing.
777}
778
Kuba Brecka65aa45e2015-12-03 10:39:43 +0000779void DisableReexec() {
780 // No need to re-exec on Windows.
781}
782
783void MaybeReexec() {
784 // No need to re-exec on Windows.
785}
786
Maxim Ostapenko1965cc62016-01-18 07:55:12 +0000787char **GetArgv() {
788 // FIXME: Actually implement this function.
789 return 0;
790}
791
Mike Aizatsky4a933162016-01-26 20:10:01 +0000792pid_t StartSubprocess(const char *program, const char *const argv[],
793 fd_t stdin_fd, fd_t stdout_fd, fd_t stderr_fd) {
794 // FIXME: implement on this platform
795 // Should be implemented based on
796 // SymbolizerProcess::StarAtSymbolizerSubprocess
797 // from lib/sanitizer_common/sanitizer_symbolizer_win.cc.
798 return -1;
799}
800
801bool IsProcessRunning(pid_t pid) {
802 // FIXME: implement on this platform.
803 return false;
804}
805
Mike Aizatsky26542992016-01-27 23:51:36 +0000806int WaitForProcess(pid_t pid) { return -1; }
807
Alexey Samsonovdde1f112012-06-05 07:05:10 +0000808} // namespace __sanitizer
809
810#endif // _WIN32