20#include <winpr/config.h>
23#include <winpr/sysinfo.h>
24#include <winpr/pool.h>
25#include <winpr/library.h>
29#ifdef WINPR_THREAD_POOL
32static INIT_ONCE init_once_module = INIT_ONCE_STATIC_INIT;
33static PTP_POOL(WINAPI* pCreateThreadpool)(PVOID reserved);
34static VOID(WINAPI* pCloseThreadpool)(PTP_POOL ptpp);
35static BOOL(WINAPI* pSetThreadpoolThreadMinimum)(PTP_POOL ptpp, DWORD cthrdMic);
36static VOID(WINAPI* pSetThreadpoolThreadMaximum)(PTP_POOL ptpp, DWORD cthrdMost);
38static BOOL CALLBACK init_module(
PINIT_ONCE once, PVOID param, PVOID* context)
40 HMODULE kernel32 = LoadLibraryA(
"kernel32.dll");
43 pCreateThreadpool = GetProcAddressAs(kernel32,
"CreateThreadpool",
void*);
44 pCloseThreadpool = GetProcAddressAs(kernel32,
"CloseThreadpool",
void*);
45 pSetThreadpoolThreadMinimum =
46 GetProcAddressAs(kernel32,
"SetThreadpoolThreadMinimum",
void*);
47 pSetThreadpoolThreadMaximum =
48 GetProcAddressAs(kernel32,
"SetThreadpoolThreadMaximum",
void*);
54static TP_POOL DEFAULT_POOL = {
63static DWORD WINAPI thread_pool_work_func(LPVOID arg)
69 PTP_CALLBACK_INSTANCE callbackInstance = NULL;
73 events[0] = pool->TerminateEvent;
74 events[1] = Queue_Event(pool->PendingQueue);
78 status = WaitForMultipleObjects(2, events, FALSE, INFINITE);
80 if (status == WAIT_OBJECT_0)
83 if (status != (WAIT_OBJECT_0 + 1))
86 callbackInstance = (PTP_CALLBACK_INSTANCE)Queue_Dequeue(pool->PendingQueue);
90 work = callbackInstance->Work;
91 work->WorkCallback(callbackInstance, work->CallbackParameter, work);
92 CountdownEvent_Signal(pool->WorkComplete, 1);
93 free(callbackInstance);
101static void threads_close(
void* thread)
103 (void)WaitForSingleObject(thread, INFINITE);
104 (void)CloseHandle(thread);
107static BOOL InitializeThreadpool(PTP_POOL pool)
115 if (!(pool->PendingQueue = Queue_New(TRUE, -1, -1)))
118 if (!(pool->WorkComplete = CountdownEvent_New(0)))
121 if (!(pool->TerminateEvent = CreateEvent(NULL, TRUE, FALSE, NULL)))
124 if (!(pool->Threads = ArrayList_New(TRUE)))
127 obj = ArrayList_Object(pool->Threads);
128 obj->fnObjectFree = threads_close;
130#if !defined(WINPR_THREADPOOL_DEFAULT_MIN_COUNT)
131#error "WINPR_THREADPOOL_DEFAULT_MIN_COUNT must be defined"
133#if !defined(WINPR_THREADPOOL_DEFAULT_MAX_COUNT)
134#error "WINPR_THREADPOOL_DEFAULT_MAX_COUNT must be defined"
138 GetSystemInfo(&info);
140 DWORD min = info.dwNumberOfProcessors;
141 DWORD max = info.dwNumberOfProcessors;
142 if (info.dwNumberOfProcessors < WINPR_THREADPOOL_DEFAULT_MIN_COUNT)
143 min = WINPR_THREADPOOL_DEFAULT_MIN_COUNT;
144 if (info.dwNumberOfProcessors > WINPR_THREADPOOL_DEFAULT_MAX_COUNT)
145 max = WINPR_THREADPOOL_DEFAULT_MAX_COUNT;
149 if (!SetThreadpoolThreadMinimum(pool, min))
152 SetThreadpoolThreadMaximum(pool, max);
160PTP_POOL GetDefaultThreadpool(
void)
162 PTP_POOL pool = &DEFAULT_POOL;
164 if (!InitializeThreadpool(pool))
170PTP_POOL winpr_CreateThreadpool(PVOID reserved)
172 PTP_POOL pool = NULL;
174 InitOnceExecuteOnce(&init_once_module, init_module, NULL, NULL);
175 if (pCreateThreadpool)
176 return pCreateThreadpool(reserved);
178 WINPR_UNUSED(reserved);
180 if (!(pool = (PTP_POOL)calloc(1,
sizeof(TP_POOL))))
183 if (!InitializeThreadpool(pool))
185 winpr_CloseThreadpool(pool);
192VOID winpr_CloseThreadpool(PTP_POOL ptpp)
195 InitOnceExecuteOnce(&init_once_module, init_module, NULL, NULL);
196 if (pCloseThreadpool)
198 pCloseThreadpool(ptpp);
202 (void)SetEvent(ptpp->TerminateEvent);
204 ArrayList_Free(ptpp->Threads);
205 Queue_Free(ptpp->PendingQueue);
206 CountdownEvent_Free(ptpp->WorkComplete);
207 (void)CloseHandle(ptpp->TerminateEvent);
210 TP_POOL empty = { 0 };
214 if (ptpp != &DEFAULT_POOL)
218BOOL winpr_SetThreadpoolThreadMinimum(PTP_POOL ptpp, DWORD cthrdMic)
222 InitOnceExecuteOnce(&init_once_module, init_module, NULL, NULL);
223 if (pSetThreadpoolThreadMinimum)
224 return pSetThreadpoolThreadMinimum(ptpp, cthrdMic);
226 ptpp->Minimum = cthrdMic;
228 ArrayList_Lock(ptpp->Threads);
229 while (ArrayList_Count(ptpp->Threads) < ptpp->Minimum)
231 HANDLE thread = CreateThread(NULL, 0, thread_pool_work_func, (
void*)ptpp, 0, NULL);
235 if (!ArrayList_Append(ptpp->Threads, thread))
237 (void)CloseHandle(thread);
244 ArrayList_Unlock(ptpp->Threads);
249VOID winpr_SetThreadpoolThreadMaximum(PTP_POOL ptpp, DWORD cthrdMost)
252 InitOnceExecuteOnce(&init_once_module, init_module, NULL, NULL);
253 if (pSetThreadpoolThreadMaximum)
255 pSetThreadpoolThreadMaximum(ptpp, cthrdMost);
259 ptpp->Maximum = cthrdMost;
261 ArrayList_Lock(ptpp->Threads);
262 if (ArrayList_Count(ptpp->Threads) > ptpp->Maximum)
264 (void)SetEvent(ptpp->TerminateEvent);
265 ArrayList_Clear(ptpp->Threads);
266 (void)ResetEvent(ptpp->TerminateEvent);
268 ArrayList_Unlock(ptpp->Threads);
269 winpr_SetThreadpoolThreadMinimum(ptpp, ptpp->Minimum);
This struct contains function pointer to initialize/free objects.