blob: e5d1e09ec96f88853fc6f5fe4f7002b46aab51ef (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
#pragma once
#include <windows.h>
#include "ThreadFunctions.h"
#include "threadpool_types.h"
#include <vector>
class ThreadID : private ThreadFunctions
{
public:
static DWORD CALLBACK thread_func_stub(LPVOID param);
ThreadID(ThreadFunctions *t_f, HANDLE killswitch, HANDLE global_functions_semaphore, ThreadPoolTypes::HandleList &inherited_handles, volatile LONG *thread_count, HANDLE _max_load_event, int _reserved, int _com_type);
~ThreadID();
void Kill();
/* Try and Wait must be paired!!! */
bool TryAddHandle(HANDLE new_handle);
void WaitAddHandle(HANDLE new_handle);
void AddHandle(HANDLE new_handle);
/* Try and Wait must be paired!!! */
bool TryRemoveHandle(HANDLE handle);
void WaitRemoveHandle(HANDLE handle);
void RemoveHandle(HANDLE handle);
using ThreadFunctions::QueueFunction;
bool IsReserved() const;
bool IsReleased() const;
bool CanRunCOM(int flags) const;
void Reserve(); // re-reserves a released thread
void Release(); // release a reversed thread
private:
void RemoveHandle_Internal(HANDLE handle);
DWORD CALLBACK ThreadFunction();
int reserved;
ThreadFunctions *global_functions;
volatile LONG *num_threads_available;
int com_type;
bool released;
ThreadFunctions local_functions;
// list of handles we're waiting on
typedef std::vector<HANDLE> HandleList;
HandleList wait_handles;
CRITICAL_SECTION handle_lock;
// handles we create/own
HANDLE threadHandle;
HANDLE wakeHandle;
// handles given to us
HANDLE max_load_event;
};
|
