forked from Public-Mirror/Textractor
62 lines
1.4 KiB
C++
62 lines
1.4 KiB
C++
|
// windbg/util.cc
|
||
|
|
// 1/27/2013 jichi
|
||
|
|
#include "windbg/util.h"
|
||
|
|
#include <tlhelp32.h>
|
||
|
|
#include <boost/foreach.hpp>
|
||
|
|
#include <list>
|
||
|
|
|
||
|
|
WINDBG_BEGIN_NAMESPACE
|
||
|
|
|
||
|
|
class ThreadsSuspenderPrivate
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
std::list<HANDLE> threads;
|
||
|
|
};
|
||
|
|
|
||
|
|
ThreadsSuspender::ThreadsSuspender(bool autoSuspend)
|
||
|
|
: d_(new D)
|
||
|
|
{ if (autoSuspend) suspend(); }
|
||
|
|
|
||
|
|
ThreadsSuspender::~ThreadsSuspender()
|
||
|
|
{
|
||
|
|
resume();
|
||
|
|
delete d_;
|
||
|
|
}
|
||
|
|
|
||
|
|
void ThreadsSuspender::suspend()
|
||
|
|
{
|
||
|
|
HANDLE hSnap = ::CreateToolhelp32Snapshot(TH32CS_SNAPTHREAD, 0);
|
||
|
|
if (hSnap == INVALID_HANDLE_VALUE)
|
||
|
|
return;
|
||
|
|
THREADENTRY32 entry;
|
||
|
|
entry.dwSize = sizeof(entry);
|
||
|
|
DWORD pid = ::GetCurrentProcessId();
|
||
|
|
DWORD tid = ::GetCurrentThreadId();
|
||
|
|
if (::Thread32First(hSnap, &entry))
|
||
|
|
do if (entry.dwSize >= 4 * sizeof(DWORD) && entry.th32OwnerProcessID == pid && entry.th32ThreadID != tid) {
|
||
|
|
if (HANDLE hThread = ::OpenThread(THREAD_SUSPEND_RESUME, 0, entry.th32ThreadID)) {
|
||
|
|
if (::SuspendThread(hThread) != DWORD(-1))
|
||
|
|
d_->threads.push_back(hThread);
|
||
|
|
else
|
||
|
|
::CloseHandle(hThread);
|
||
|
|
}
|
||
|
|
entry.dwSize = sizeof(entry);
|
||
|
|
} while (::Thread32Next(hSnap, &entry));
|
||
|
|
::CloseHandle(hSnap);
|
||
|
|
}
|
||
|
|
|
||
|
|
void ThreadsSuspender::resume()
|
||
|
|
{
|
||
|
|
if (!d_->threads.empty()) {
|
||
|
|
BOOST_FOREACH (HANDLE hThread, d_->threads) {
|
||
|
|
::ResumeThread(hThread);
|
||
|
|
::CloseHandle(hThread);
|
||
|
|
}
|
||
|
|
d_->threads.clear();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
WINDBG_END_NAMESPACE
|
||
|
|
|
||
|
|
// EOF
|