From 4f9229dfe609632a193ce515f8fa83b494948e48 Mon Sep 17 00:00:00 2001 From: jbzdarkid Date: Wed, 24 Oct 2018 19:02:59 -0700 Subject: Better errors (?) --- WitnessRandomizer/Memory.cpp | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) (limited to 'WitnessRandomizer') diff --git a/WitnessRandomizer/Memory.cpp b/WitnessRandomizer/Memory.cpp index f2bd8f1..7005455 100644 --- a/WitnessRandomizer/Memory.cpp +++ b/WitnessRandomizer/Memory.cpp @@ -14,13 +14,13 @@ Memory::Memory(const std::string& processName) { while (Process32Next(snapshot, &entry)) { if (processName == entry.szExeFile) { _handle = OpenProcess(PROCESS_ALL_ACCESS, FALSE, entry.th32ProcessID); - if (!_handle) { - std::cerr << "Couldn't find " << processName.c_str() << ". Is it open?" << std::endl; - exit(EXIT_FAILURE); - } break; } } + if (!_handle) { + OutputDebugStringA("Process is not open!"); + exit(EXIT_FAILURE); + } // Next, get the process base address DWORD numModules; @@ -38,7 +38,7 @@ Memory::Memory(const std::string& processName) { } } if (_baseAddress == 0) { - std::cerr << "Couldn't find base address!" << std::endl; + OutputDebugStringA("Couldn't find base address!"); exit(EXIT_FAILURE); } } @@ -50,9 +50,9 @@ Memory::~Memory() { // Private methods: void Memory::ThrowError() { - wchar_t message[256]; - FormatMessageW(4096, NULL, GetLastError(), 1024, message, 256, NULL); - std::cerr << message << std::endl; + std::string message(256, '\0'); + FormatMessageA(4096, NULL, GetLastError(), 1024, &message[0], message.length(), NULL); + OutputDebugStringA(message.c_str()); exit(EXIT_FAILURE); } -- cgit 1.4.1