// textthread.cc // 8/24/2013 jichi // Branch IHF/TextThread.cpp, rev 133 #include "textthread.h" #include "host.h" #include "const.h" TextThread::TextThread(ThreadParam tp, DWORD status) : handle(ThreadCounter++), name(Host::GetHookName(tp.pid, tp.hook)), tp(tp), status(status) {} TextThread::~TextThread() { SetEvent(cancelFlushEvent); flusher.join(); CloseHandle(cancelFlushEvent); } std::wstring TextThread::GetStore() { LOCK(ttMutex); return storage; } void TextThread::Flush() { std::wstring sentence; { LOCK(ttMutex); sentence = buffer; buffer.clear(); } AddSentence(sentence); } void TextThread::AddSentence(std::wstring sentence) { // Dispatch to extensions occurs here. Don't hold mutex! Extensions might take a while! if (Output) sentence = Output(this, sentence); LOCK(ttMutex); storage.append(sentence); } void TextThread::AddText(const BYTE* data, int len) { if (buffer.size() > MaxBufferSize) Flush(); SetEvent(cancelFlushEvent); flusher.join(); flusher = std::thread([&] { ResetEvent(cancelFlushEvent); if (WaitForSingleObject(cancelFlushEvent, FlushDelay) == WAIT_TIMEOUT) Flush(); }); LOCK(ttMutex); buffer += status & USING_UNICODE ? std::wstring((wchar_t*)data, len / 2) : StringToWideString(std::string((char*)data, len), status & USING_UTF8 ? CP_UTF8 : SHIFT_JIS); } // EOF