forked from OSchip/llvm-project
124 lines
3.3 KiB
C++
124 lines
3.3 KiB
C++
//===-- sanitizer_win.cc --------------------------------------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file is shared between AddressSanitizer and ThreadSanitizer
|
|
// run-time libraries and implements windows-specific functions from
|
|
// sanitizer_libc.h.
|
|
//===----------------------------------------------------------------------===//
|
|
#ifdef _WIN32
|
|
#include <windows.h>
|
|
|
|
#include "sanitizer_common.h"
|
|
#include "sanitizer_libc.h"
|
|
|
|
namespace __sanitizer {
|
|
|
|
// --------------------- sanitizer_common.h
|
|
int GetPid() {
|
|
return GetProcessId(GetCurrentProcess());
|
|
}
|
|
|
|
void GetThreadStackTopAndBottom(bool at_initialization, uptr *stack_top,
|
|
uptr *stack_bottom) {
|
|
CHECK(stack_top);
|
|
CHECK(stack_bottom);
|
|
MEMORY_BASIC_INFORMATION mbi;
|
|
CHECK(VirtualQuery(&mbi /* on stack */, &mbi, sizeof(mbi)) != 0);
|
|
// FIXME: is it possible for the stack to not be a single allocation?
|
|
// Are these values what ASan expects to get (reserved, not committed;
|
|
// including stack guard page) ?
|
|
*stack_top = (uptr)mbi.BaseAddress + mbi.RegionSize;
|
|
*stack_bottom = (uptr)mbi.AllocationBase;
|
|
}
|
|
|
|
|
|
void *MmapOrDie(uptr size, const char *mem_type) {
|
|
void *rv = VirtualAlloc(0, size, MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE);
|
|
if (rv == 0) {
|
|
Report("ERROR: Failed to allocate 0x%zx (%zd) bytes of %s\n",
|
|
size, size, mem_type);
|
|
CHECK("unable to mmap" && 0);
|
|
}
|
|
return rv;
|
|
}
|
|
|
|
void UnmapOrDie(void *addr, uptr size) {
|
|
if (VirtualFree(addr, size, MEM_DECOMMIT) == 0) {
|
|
Report("ERROR: Failed to deallocate 0x%zx (%zd) bytes at address %p\n",
|
|
size, size, addr);
|
|
CHECK("unable to unmap" && 0);
|
|
}
|
|
}
|
|
|
|
const char *GetEnv(const char *name) {
|
|
static char env_buffer[32767] = {};
|
|
|
|
// Note: this implementation stores the result in a static buffer so we only
|
|
// allow it to be called just once.
|
|
static bool called_once = false;
|
|
if (called_once)
|
|
UNIMPLEMENTED();
|
|
called_once = true;
|
|
|
|
DWORD rv = GetEnvironmentVariableA(name, env_buffer, sizeof(env_buffer));
|
|
if (rv > 0 && rv < sizeof(env_buffer))
|
|
return env_buffer;
|
|
return 0;
|
|
}
|
|
|
|
// ------------------ sanitizer_libc.h
|
|
void *internal_mmap(void *addr, uptr length, int prot, int flags,
|
|
int fd, u64 offset) {
|
|
UNIMPLEMENTED();
|
|
}
|
|
|
|
int internal_munmap(void *addr, uptr length) {
|
|
UNIMPLEMENTED();
|
|
}
|
|
|
|
int internal_close(fd_t fd) {
|
|
UNIMPLEMENTED();
|
|
}
|
|
|
|
fd_t internal_open(const char *filename, bool write) {
|
|
UNIMPLEMENTED();
|
|
}
|
|
|
|
uptr internal_read(fd_t fd, void *buf, uptr count) {
|
|
UNIMPLEMENTED();
|
|
}
|
|
|
|
uptr internal_write(fd_t fd, const void *buf, uptr count) {
|
|
if (fd != 2)
|
|
UNIMPLEMENTED();
|
|
HANDLE err = GetStdHandle(STD_ERROR_HANDLE);
|
|
if (err == 0)
|
|
return 0; // FIXME: this might not work on some apps.
|
|
DWORD ret;
|
|
if (!WriteFile(err, buf, count, &ret, 0))
|
|
return 0;
|
|
return ret;
|
|
}
|
|
|
|
uptr internal_filesize(fd_t fd) {
|
|
UNIMPLEMENTED();
|
|
}
|
|
|
|
int internal_dup2(int oldfd, int newfd) {
|
|
UNIMPLEMENTED();
|
|
}
|
|
|
|
int internal_sscanf(const char *str, const char *format, ...) {
|
|
UNIMPLEMENTED();
|
|
}
|
|
|
|
} // namespace __sanitizer
|
|
|
|
#endif // _WIN32
|