Textractor/host/textthread.h

63 lines
1.5 KiB
C
Raw Normal View History

#pragma once
// textthread.h
// 8/23/2013 jichi
// Branch: ITH/TextThread.h, rev 120
2018-07-18 16:18:43 -04:00
#include <Windows.h>
#include <string>
#include <vector>
#include <functional>
2018-08-21 22:43:30 -04:00
#include <mutex>
#include <thread>
2018-07-19 23:22:55 -04:00
struct ThreadParameter
2018-07-18 23:40:44 -04:00
{
2018-07-19 23:22:55 -04:00
DWORD pid; // jichi: 5/11/2014: The process ID
2018-08-21 22:43:30 -04:00
unsigned __int64 hook; // Artikash 6/6/2018: The insertion address of the hook
unsigned __int64 retn; // jichi 5/11/2014: The return address of the hook
2018-08-21 22:43:30 -04:00
unsigned __int64 spl; // jichi 5/11/2014: the processed split value of the hook paramete
2018-07-19 23:22:55 -04:00
// Artikash 5/31/2018: required for unordered_map to work with struct key
friend bool operator==(const ThreadParameter& one, const ThreadParameter& two)
{
return one.pid == two.pid && one.hook == two.hook && one.retn == two.retn && one.spl == two.spl;
}
};
class TextThread;
typedef std::function<std::wstring(TextThread*, std::wstring)> ThreadOutputCallback;
//extern DWORD split_time,repeat_count,global_filter,cyclic_remove;
2018-07-18 23:40:44 -04:00
class TextThread
{
public:
2018-08-21 22:43:30 -04:00
TextThread(ThreadParameter tp, DWORD status);
virtual ~TextThread();
2018-07-25 21:48:18 -07:00
virtual std::wstring GetStore();
2018-07-19 23:22:55 -04:00
ThreadParameter GetThreadParameter() { return tp; }
2018-08-21 22:43:30 -04:00
void RegisterOutputCallBack(ThreadOutputCallback cb) { Output = cb; }
2018-07-25 21:48:18 -07:00
void AddText(const BYTE *con, int len);
void AddSentence(std::wstring sentence);
private:
2018-08-21 22:43:30 -04:00
bool FlushSentenceBuffer();
2018-07-19 23:22:55 -04:00
std::vector<char> sentenceBuffer;
std::wstring storage;
2018-08-21 22:43:30 -04:00
std::recursive_mutex ttMutex;
2018-08-21 23:56:16 -04:00
HANDLE deletionEvent;
2018-08-21 22:43:30 -04:00
std::thread flushThread;
2018-08-21 23:56:16 -04:00
2018-08-21 12:41:19 -04:00
DWORD timestamp;
2018-08-21 23:56:16 -04:00
ThreadOutputCallback Output;
ThreadParameter tp;
2018-07-19 23:22:55 -04:00
DWORD status;
};
// EOF