2851 lines
91 KiB
C
2851 lines
91 KiB
C
/*
|
|
* Concurrency namespace implementation
|
|
*
|
|
* Copyright 2017 Piotr Caban
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include <stdarg.h>
|
|
#include <stdbool.h>
|
|
|
|
#include "windef.h"
|
|
#include "winternl.h"
|
|
#include "wine/debug.h"
|
|
#include "msvcrt.h"
|
|
#include "cxx.h"
|
|
|
|
#if _MSVCR_VER >= 100
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(msvcrt);
|
|
|
|
typedef exception cexception;
|
|
CREATE_EXCEPTION_OBJECT(cexception)
|
|
|
|
static int context_id = -1;
|
|
static int scheduler_id = -1;
|
|
|
|
typedef enum {
|
|
SchedulerKind,
|
|
MaxConcurrency,
|
|
MinConcurrency,
|
|
TargetOversubscriptionFactor,
|
|
LocalContextCacheSize,
|
|
ContextStackSize,
|
|
ContextPriority,
|
|
SchedulingProtocol,
|
|
DynamicProgressFeedback,
|
|
WinRTInitialization,
|
|
last_policy_id
|
|
} PolicyElementKey;
|
|
|
|
typedef struct {
|
|
struct _policy_container {
|
|
unsigned int policies[last_policy_id];
|
|
} *policy_container;
|
|
} SchedulerPolicy;
|
|
|
|
typedef struct {
|
|
const vtable_ptr *vtable;
|
|
} Context;
|
|
#define call_Context_GetId(this) CALL_VTBL_FUNC(this, 0, \
|
|
unsigned int, (const Context*), (this))
|
|
#define call_Context_GetVirtualProcessorId(this) CALL_VTBL_FUNC(this, 4, \
|
|
unsigned int, (const Context*), (this))
|
|
#define call_Context_GetScheduleGroupId(this) CALL_VTBL_FUNC(this, 8, \
|
|
unsigned int, (const Context*), (this))
|
|
#define call_Context_dtor(this, flags) CALL_VTBL_FUNC(this, 20, \
|
|
Context*, (Context*, unsigned int), (this, flags))
|
|
|
|
typedef struct {
|
|
Context *context;
|
|
} _Context;
|
|
|
|
union allocator_cache_entry {
|
|
struct _free {
|
|
int depth;
|
|
union allocator_cache_entry *next;
|
|
} free;
|
|
struct _alloc {
|
|
int bucket;
|
|
char mem[1];
|
|
} alloc;
|
|
};
|
|
|
|
struct scheduler_list {
|
|
struct Scheduler *scheduler;
|
|
struct scheduler_list *next;
|
|
};
|
|
|
|
typedef struct {
|
|
Context context;
|
|
struct scheduler_list scheduler;
|
|
unsigned int id;
|
|
union allocator_cache_entry *allocator_cache[8];
|
|
} ExternalContextBase;
|
|
extern const vtable_ptr ExternalContextBase_vtable;
|
|
static void ExternalContextBase_ctor(ExternalContextBase*);
|
|
|
|
typedef struct Scheduler {
|
|
const vtable_ptr *vtable;
|
|
} Scheduler;
|
|
#define call_Scheduler_Id(this) CALL_VTBL_FUNC(this, 4, unsigned int, (const Scheduler*), (this))
|
|
#define call_Scheduler_GetNumberOfVirtualProcessors(this) CALL_VTBL_FUNC(this, 8, unsigned int, (const Scheduler*), (this))
|
|
#define call_Scheduler_GetPolicy(this,policy) CALL_VTBL_FUNC(this, 12, \
|
|
SchedulerPolicy*, (Scheduler*,SchedulerPolicy*), (this,policy))
|
|
#define call_Scheduler_Reference(this) CALL_VTBL_FUNC(this, 16, unsigned int, (Scheduler*), (this))
|
|
#define call_Scheduler_Release(this) CALL_VTBL_FUNC(this, 20, unsigned int, (Scheduler*), (this))
|
|
#define call_Scheduler_RegisterShutdownEvent(this,event) CALL_VTBL_FUNC(this, 24, void, (Scheduler*,HANDLE), (this,event))
|
|
#define call_Scheduler_Attach(this) CALL_VTBL_FUNC(this, 28, void, (Scheduler*), (this))
|
|
#if _MSVCR_VER > 100
|
|
#define call_Scheduler_CreateScheduleGroup_loc(this,placement) CALL_VTBL_FUNC(this, 32, \
|
|
/*ScheduleGroup*/void*, (Scheduler*,/*location*/void*), (this,placement))
|
|
#define call_Scheduler_CreateScheduleGroup(this) CALL_VTBL_FUNC(this, 36, /*ScheduleGroup*/void*, (Scheduler*), (this))
|
|
#define call_Scheduler_ScheduleTask_loc(this,proc,data,placement) CALL_VTBL_FUNC(this, 40, \
|
|
void, (Scheduler*,void (__cdecl*)(void*),void*,/*location*/void*), (this,proc,data,placement))
|
|
#define call_Scheduler_ScheduleTask(this,proc,data) CALL_VTBL_FUNC(this, 44, \
|
|
void, (Scheduler*,void (__cdecl*)(void*),void*), (this,proc,data))
|
|
#define call_Scheduler_IsAvailableLocation(this,placement) CALL_VTBL_FUNC(this, 48, \
|
|
bool, (Scheduler*,const /*location*/void*), (this,placement))
|
|
#else
|
|
#define call_Scheduler_CreateScheduleGroup(this) CALL_VTBL_FUNC(this, 32, /*ScheduleGroup*/void*, (Scheduler*), (this))
|
|
#define call_Scheduler_ScheduleTask(this,proc,data) CALL_VTBL_FUNC(this, 36, \
|
|
void, (Scheduler*,void (__cdecl*)(void*),void*), (this,proc,data))
|
|
#endif
|
|
|
|
typedef struct {
|
|
Scheduler scheduler;
|
|
LONG ref;
|
|
unsigned int id;
|
|
unsigned int virt_proc_no;
|
|
SchedulerPolicy policy;
|
|
int shutdown_count;
|
|
int shutdown_size;
|
|
HANDLE *shutdown_events;
|
|
CRITICAL_SECTION cs;
|
|
} ThreadScheduler;
|
|
extern const vtable_ptr ThreadScheduler_vtable;
|
|
|
|
typedef struct {
|
|
Scheduler *scheduler;
|
|
} _Scheduler;
|
|
|
|
typedef struct {
|
|
char empty;
|
|
} _CurrentScheduler;
|
|
|
|
typedef enum
|
|
{
|
|
SPINWAIT_INIT,
|
|
SPINWAIT_SPIN,
|
|
SPINWAIT_YIELD,
|
|
SPINWAIT_DONE
|
|
} SpinWait_state;
|
|
|
|
typedef void (__cdecl *yield_func)(void);
|
|
|
|
typedef struct
|
|
{
|
|
ULONG spin;
|
|
ULONG unknown;
|
|
SpinWait_state state;
|
|
yield_func yield_func;
|
|
} SpinWait;
|
|
|
|
/* keep in sync with msvcp90/msvcp90.h */
|
|
typedef struct cs_queue
|
|
{
|
|
struct cs_queue *next;
|
|
#if _MSVCR_VER >= 110
|
|
BOOL free;
|
|
int unknown;
|
|
#endif
|
|
} cs_queue;
|
|
|
|
typedef struct
|
|
{
|
|
ULONG_PTR unk_thread_id;
|
|
cs_queue unk_active;
|
|
#if _MSVCR_VER >= 110
|
|
void *unknown[2];
|
|
#else
|
|
void *unknown[1];
|
|
#endif
|
|
cs_queue *head;
|
|
void *tail;
|
|
} critical_section;
|
|
|
|
typedef struct
|
|
{
|
|
critical_section *cs;
|
|
union {
|
|
cs_queue q;
|
|
struct {
|
|
void *unknown[4];
|
|
int unknown2[2];
|
|
} unknown;
|
|
} lock;
|
|
} critical_section_scoped_lock;
|
|
|
|
typedef struct
|
|
{
|
|
critical_section cs;
|
|
} _NonReentrantPPLLock;
|
|
|
|
typedef struct
|
|
{
|
|
_NonReentrantPPLLock *lock;
|
|
union {
|
|
cs_queue q;
|
|
struct {
|
|
void *unknown[4];
|
|
int unknown2[2];
|
|
} unknown;
|
|
} wait;
|
|
} _NonReentrantPPLLock__Scoped_lock;
|
|
|
|
typedef struct
|
|
{
|
|
critical_section cs;
|
|
LONG count;
|
|
LONG owner;
|
|
} _ReentrantPPLLock;
|
|
|
|
typedef struct
|
|
{
|
|
_ReentrantPPLLock *lock;
|
|
union {
|
|
cs_queue q;
|
|
struct {
|
|
void *unknown[4];
|
|
int unknown2[2];
|
|
} unknown;
|
|
} wait;
|
|
} _ReentrantPPLLock__Scoped_lock;
|
|
|
|
#define EVT_RUNNING (void*)1
|
|
#define EVT_WAITING NULL
|
|
|
|
struct thread_wait;
|
|
typedef struct thread_wait_entry
|
|
{
|
|
struct thread_wait *wait;
|
|
struct thread_wait_entry *next;
|
|
struct thread_wait_entry *prev;
|
|
} thread_wait_entry;
|
|
|
|
typedef struct thread_wait
|
|
{
|
|
void *signaled;
|
|
int pending_waits;
|
|
thread_wait_entry entries[1];
|
|
} thread_wait;
|
|
|
|
typedef struct
|
|
{
|
|
thread_wait_entry *waiters;
|
|
INT_PTR signaled;
|
|
critical_section cs;
|
|
} event;
|
|
|
|
#if _MSVCR_VER >= 110
|
|
typedef struct cv_queue {
|
|
struct cv_queue *next;
|
|
BOOL expired;
|
|
} cv_queue;
|
|
|
|
typedef struct {
|
|
/* cv_queue structure is not binary compatible */
|
|
cv_queue *queue;
|
|
critical_section lock;
|
|
} _Condition_variable;
|
|
#endif
|
|
|
|
typedef struct rwl_queue
|
|
{
|
|
struct rwl_queue *next;
|
|
} rwl_queue;
|
|
|
|
#define WRITER_WAITING 0x80000000
|
|
/* FIXME: reader_writer_lock structure is not binary compatible
|
|
* it can't exceed 28/56 bytes */
|
|
typedef struct
|
|
{
|
|
LONG count;
|
|
LONG thread_id;
|
|
rwl_queue active;
|
|
rwl_queue *writer_head;
|
|
rwl_queue *writer_tail;
|
|
rwl_queue *reader_head;
|
|
} reader_writer_lock;
|
|
|
|
typedef struct {
|
|
reader_writer_lock *lock;
|
|
} reader_writer_lock_scoped_lock;
|
|
|
|
typedef struct {
|
|
CRITICAL_SECTION cs;
|
|
} _ReentrantBlockingLock;
|
|
|
|
typedef exception improper_lock;
|
|
extern const vtable_ptr improper_lock_vtable;
|
|
|
|
typedef exception improper_scheduler_attach;
|
|
extern const vtable_ptr improper_scheduler_attach_vtable;
|
|
|
|
typedef exception improper_scheduler_detach;
|
|
extern const vtable_ptr improper_scheduler_detach_vtable;
|
|
|
|
typedef exception invalid_scheduler_policy_key;
|
|
extern const vtable_ptr invalid_scheduler_policy_key_vtable;
|
|
|
|
typedef exception invalid_scheduler_policy_thread_specification;
|
|
extern const vtable_ptr invalid_scheduler_policy_thread_specification_vtable;
|
|
|
|
typedef exception invalid_scheduler_policy_value;
|
|
extern const vtable_ptr invalid_scheduler_policy_value_vtable;
|
|
|
|
typedef struct {
|
|
exception e;
|
|
HRESULT hr;
|
|
} scheduler_resource_allocation_error;
|
|
extern const vtable_ptr scheduler_resource_allocation_error_vtable;
|
|
|
|
enum ConcRT_EventType
|
|
{
|
|
CONCRT_EVENT_GENERIC,
|
|
CONCRT_EVENT_START,
|
|
CONCRT_EVENT_END,
|
|
CONCRT_EVENT_BLOCK,
|
|
CONCRT_EVENT_UNBLOCK,
|
|
CONCRT_EVENT_YIELD,
|
|
CONCRT_EVENT_ATTACH,
|
|
CONCRT_EVENT_DETACH
|
|
};
|
|
|
|
static int context_tls_index = TLS_OUT_OF_INDEXES;
|
|
|
|
static CRITICAL_SECTION default_scheduler_cs;
|
|
static CRITICAL_SECTION_DEBUG default_scheduler_cs_debug =
|
|
{
|
|
0, 0, &default_scheduler_cs,
|
|
{ &default_scheduler_cs_debug.ProcessLocksList, &default_scheduler_cs_debug.ProcessLocksList },
|
|
0, 0, { (DWORD_PTR)(__FILE__ ": default_scheduler_cs") }
|
|
};
|
|
static CRITICAL_SECTION default_scheduler_cs = { &default_scheduler_cs_debug, -1, 0, 0, 0, 0 };
|
|
static SchedulerPolicy default_scheduler_policy;
|
|
static ThreadScheduler *default_scheduler;
|
|
|
|
static HANDLE keyed_event;
|
|
|
|
static void create_default_scheduler(void);
|
|
|
|
/* ??0improper_lock@Concurrency@@QAE@PBD@Z */
|
|
/* ??0improper_lock@Concurrency@@QEAA@PEBD@Z */
|
|
DEFINE_THISCALL_WRAPPER(improper_lock_ctor_str, 8)
|
|
improper_lock* __thiscall improper_lock_ctor_str(improper_lock *this, const char *str)
|
|
{
|
|
TRACE("(%p %p)\n", this, str);
|
|
return __exception_ctor(this, str, &improper_lock_vtable);
|
|
}
|
|
|
|
/* ??0improper_lock@Concurrency@@QAE@XZ */
|
|
/* ??0improper_lock@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(improper_lock_ctor, 4)
|
|
improper_lock* __thiscall improper_lock_ctor(improper_lock *this)
|
|
{
|
|
return improper_lock_ctor_str(this, NULL);
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(improper_lock_copy_ctor,8)
|
|
improper_lock * __thiscall improper_lock_copy_ctor(improper_lock *this, const improper_lock *rhs)
|
|
{
|
|
TRACE("(%p %p)\n", this, rhs);
|
|
return __exception_copy_ctor(this, rhs, &improper_lock_vtable);
|
|
}
|
|
|
|
/* ??0improper_scheduler_attach@Concurrency@@QAE@PBD@Z */
|
|
/* ??0improper_scheduler_attach@Concurrency@@QEAA@PEBD@Z */
|
|
DEFINE_THISCALL_WRAPPER(improper_scheduler_attach_ctor_str, 8)
|
|
improper_scheduler_attach* __thiscall improper_scheduler_attach_ctor_str(
|
|
improper_scheduler_attach *this, const char *str)
|
|
{
|
|
TRACE("(%p %p)\n", this, str);
|
|
return __exception_ctor(this, str, &improper_scheduler_attach_vtable);
|
|
}
|
|
|
|
/* ??0improper_scheduler_attach@Concurrency@@QAE@XZ */
|
|
/* ??0improper_scheduler_attach@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(improper_scheduler_attach_ctor, 4)
|
|
improper_scheduler_attach* __thiscall improper_scheduler_attach_ctor(
|
|
improper_scheduler_attach *this)
|
|
{
|
|
return improper_scheduler_attach_ctor_str(this, NULL);
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(improper_scheduler_attach_copy_ctor,8)
|
|
improper_scheduler_attach * __thiscall improper_scheduler_attach_copy_ctor(
|
|
improper_scheduler_attach * _this, const improper_scheduler_attach * rhs)
|
|
{
|
|
TRACE("(%p %p)\n", _this, rhs);
|
|
return __exception_copy_ctor(_this, rhs, &improper_scheduler_attach_vtable);
|
|
}
|
|
|
|
/* ??0improper_scheduler_detach@Concurrency@@QAE@PBD@Z */
|
|
/* ??0improper_scheduler_detach@Concurrency@@QEAA@PEBD@Z */
|
|
DEFINE_THISCALL_WRAPPER(improper_scheduler_detach_ctor_str, 8)
|
|
improper_scheduler_detach* __thiscall improper_scheduler_detach_ctor_str(
|
|
improper_scheduler_detach *this, const char *str)
|
|
{
|
|
TRACE("(%p %p)\n", this, str);
|
|
return __exception_ctor(this, str, &improper_scheduler_detach_vtable);
|
|
}
|
|
|
|
/* ??0improper_scheduler_detach@Concurrency@@QAE@XZ */
|
|
/* ??0improper_scheduler_detach@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(improper_scheduler_detach_ctor, 4)
|
|
improper_scheduler_detach* __thiscall improper_scheduler_detach_ctor(
|
|
improper_scheduler_detach *this)
|
|
{
|
|
return improper_scheduler_detach_ctor_str(this, NULL);
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(improper_scheduler_detach_copy_ctor,8)
|
|
improper_scheduler_detach * __thiscall improper_scheduler_detach_copy_ctor(
|
|
improper_scheduler_detach * _this, const improper_scheduler_detach * rhs)
|
|
{
|
|
TRACE("(%p %p)\n", _this, rhs);
|
|
return __exception_copy_ctor(_this, rhs, &improper_scheduler_detach_vtable);
|
|
}
|
|
|
|
/* ??0invalid_scheduler_policy_key@Concurrency@@QAE@PBD@Z */
|
|
/* ??0invalid_scheduler_policy_key@Concurrency@@QEAA@PEBD@Z */
|
|
DEFINE_THISCALL_WRAPPER(invalid_scheduler_policy_key_ctor_str, 8)
|
|
invalid_scheduler_policy_key* __thiscall invalid_scheduler_policy_key_ctor_str(
|
|
invalid_scheduler_policy_key *this, const char *str)
|
|
{
|
|
TRACE("(%p %p)\n", this, str);
|
|
return __exception_ctor(this, str, &invalid_scheduler_policy_key_vtable);
|
|
}
|
|
|
|
/* ??0invalid_scheduler_policy_key@Concurrency@@QAE@XZ */
|
|
/* ??0invalid_scheduler_policy_key@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(invalid_scheduler_policy_key_ctor, 4)
|
|
invalid_scheduler_policy_key* __thiscall invalid_scheduler_policy_key_ctor(
|
|
invalid_scheduler_policy_key *this)
|
|
{
|
|
return invalid_scheduler_policy_key_ctor_str(this, NULL);
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(invalid_scheduler_policy_key_copy_ctor,8)
|
|
invalid_scheduler_policy_key * __thiscall invalid_scheduler_policy_key_copy_ctor(
|
|
invalid_scheduler_policy_key * _this, const invalid_scheduler_policy_key * rhs)
|
|
{
|
|
TRACE("(%p %p)\n", _this, rhs);
|
|
return __exception_copy_ctor(_this, rhs, &invalid_scheduler_policy_key_vtable);
|
|
}
|
|
|
|
/* ??0invalid_scheduler_policy_thread_specification@Concurrency@@QAE@PBD@Z */
|
|
/* ??0invalid_scheduler_policy_thread_specification@Concurrency@@QEAA@PEBD@Z */
|
|
DEFINE_THISCALL_WRAPPER(invalid_scheduler_policy_thread_specification_ctor_str, 8)
|
|
invalid_scheduler_policy_thread_specification* __thiscall invalid_scheduler_policy_thread_specification_ctor_str(
|
|
invalid_scheduler_policy_thread_specification *this, const char *str)
|
|
{
|
|
TRACE("(%p %p)\n", this, str);
|
|
return __exception_ctor(this, str, &invalid_scheduler_policy_thread_specification_vtable);
|
|
}
|
|
|
|
/* ??0invalid_scheduler_policy_thread_specification@Concurrency@@QAE@XZ */
|
|
/* ??0invalid_scheduler_policy_thread_specification@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(invalid_scheduler_policy_thread_specification_ctor, 4)
|
|
invalid_scheduler_policy_thread_specification* __thiscall invalid_scheduler_policy_thread_specification_ctor(
|
|
invalid_scheduler_policy_thread_specification *this)
|
|
{
|
|
return invalid_scheduler_policy_thread_specification_ctor_str(this, NULL);
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(invalid_scheduler_policy_thread_specification_copy_ctor,8)
|
|
invalid_scheduler_policy_thread_specification * __thiscall invalid_scheduler_policy_thread_specification_copy_ctor(
|
|
invalid_scheduler_policy_thread_specification * _this, const invalid_scheduler_policy_thread_specification * rhs)
|
|
{
|
|
TRACE("(%p %p)\n", _this, rhs);
|
|
return __exception_copy_ctor(_this, rhs, &invalid_scheduler_policy_thread_specification_vtable);
|
|
}
|
|
|
|
/* ??0invalid_scheduler_policy_value@Concurrency@@QAE@PBD@Z */
|
|
/* ??0invalid_scheduler_policy_value@Concurrency@@QEAA@PEBD@Z */
|
|
DEFINE_THISCALL_WRAPPER(invalid_scheduler_policy_value_ctor_str, 8)
|
|
invalid_scheduler_policy_value* __thiscall invalid_scheduler_policy_value_ctor_str(
|
|
invalid_scheduler_policy_value *this, const char *str)
|
|
{
|
|
TRACE("(%p %p)\n", this, str);
|
|
return __exception_ctor(this, str, &invalid_scheduler_policy_value_vtable);
|
|
}
|
|
|
|
/* ??0invalid_scheduler_policy_value@Concurrency@@QAE@XZ */
|
|
/* ??0invalid_scheduler_policy_value@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(invalid_scheduler_policy_value_ctor, 4)
|
|
invalid_scheduler_policy_value* __thiscall invalid_scheduler_policy_value_ctor(
|
|
invalid_scheduler_policy_value *this)
|
|
{
|
|
return invalid_scheduler_policy_value_ctor_str(this, NULL);
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(invalid_scheduler_policy_value_copy_ctor,8)
|
|
invalid_scheduler_policy_value * __thiscall invalid_scheduler_policy_value_copy_ctor(
|
|
invalid_scheduler_policy_value * _this, const invalid_scheduler_policy_value * rhs)
|
|
{
|
|
TRACE("(%p %p)\n", _this, rhs);
|
|
return __exception_copy_ctor(_this, rhs, &invalid_scheduler_policy_value_vtable);
|
|
}
|
|
|
|
/* ??0scheduler_resource_allocation_error@Concurrency@@QAE@PBDJ@Z */
|
|
/* ??0scheduler_resource_allocation_error@Concurrency@@QEAA@PEBDJ@Z */
|
|
DEFINE_THISCALL_WRAPPER(scheduler_resource_allocation_error_ctor_name, 12)
|
|
scheduler_resource_allocation_error* __thiscall scheduler_resource_allocation_error_ctor_name(
|
|
scheduler_resource_allocation_error *this, const char *name, HRESULT hr)
|
|
{
|
|
TRACE("(%p %s %x)\n", this, wine_dbgstr_a(name), hr);
|
|
__exception_ctor(&this->e, name, &scheduler_resource_allocation_error_vtable);
|
|
this->hr = hr;
|
|
return this;
|
|
}
|
|
|
|
/* ??0scheduler_resource_allocation_error@Concurrency@@QAE@J@Z */
|
|
/* ??0scheduler_resource_allocation_error@Concurrency@@QEAA@J@Z */
|
|
DEFINE_THISCALL_WRAPPER(scheduler_resource_allocation_error_ctor, 8)
|
|
scheduler_resource_allocation_error* __thiscall scheduler_resource_allocation_error_ctor(
|
|
scheduler_resource_allocation_error *this, HRESULT hr)
|
|
{
|
|
return scheduler_resource_allocation_error_ctor_name(this, NULL, hr);
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(scheduler_resource_allocation_error_copy_ctor,8)
|
|
scheduler_resource_allocation_error* __thiscall scheduler_resource_allocation_error_copy_ctor(
|
|
scheduler_resource_allocation_error *this,
|
|
const scheduler_resource_allocation_error *rhs)
|
|
{
|
|
TRACE("(%p,%p)\n", this, rhs);
|
|
|
|
if (!rhs->e.do_free)
|
|
memcpy(this, rhs, sizeof(*this));
|
|
else
|
|
scheduler_resource_allocation_error_ctor_name(this, rhs->e.name, rhs->hr);
|
|
return this;
|
|
}
|
|
|
|
/* ?get_error_code@scheduler_resource_allocation_error@Concurrency@@QBEJXZ */
|
|
/* ?get_error_code@scheduler_resource_allocation_error@Concurrency@@QEBAJXZ */
|
|
DEFINE_THISCALL_WRAPPER(scheduler_resource_allocation_error_get_error_code, 4)
|
|
HRESULT __thiscall scheduler_resource_allocation_error_get_error_code(
|
|
const scheduler_resource_allocation_error *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
return this->hr;
|
|
}
|
|
|
|
DEFINE_RTTI_DATA1(improper_lock, 0, &cexception_rtti_base_descriptor,
|
|
".?AVimproper_lock@Concurrency@@")
|
|
DEFINE_RTTI_DATA1(improper_scheduler_attach, 0, &cexception_rtti_base_descriptor,
|
|
".?AVimproper_scheduler_attach@Concurrency@@")
|
|
DEFINE_RTTI_DATA1(improper_scheduler_detach, 0, &cexception_rtti_base_descriptor,
|
|
".?AVimproper_scheduler_detach@Concurrency@@")
|
|
DEFINE_RTTI_DATA1(invalid_scheduler_policy_key, 0, &cexception_rtti_base_descriptor,
|
|
".?AVinvalid_scheduler_policy_key@Concurrency@@")
|
|
DEFINE_RTTI_DATA1(invalid_scheduler_policy_thread_specification, 0, &cexception_rtti_base_descriptor,
|
|
".?AVinvalid_scheduler_policy_thread_specification@Concurrency@@")
|
|
DEFINE_RTTI_DATA1(invalid_scheduler_policy_value, 0, &cexception_rtti_base_descriptor,
|
|
".?AVinvalid_scheduler_policy_value@Concurrency@@")
|
|
DEFINE_RTTI_DATA1(scheduler_resource_allocation_error, 0, &cexception_rtti_base_descriptor,
|
|
".?AVscheduler_resource_allocation_error@Concurrency@@")
|
|
|
|
DEFINE_CXX_DATA1(improper_lock, &cexception_cxx_type_info, cexception_dtor)
|
|
DEFINE_CXX_DATA1(improper_scheduler_attach, &cexception_cxx_type_info, cexception_dtor)
|
|
DEFINE_CXX_DATA1(improper_scheduler_detach, &cexception_cxx_type_info, cexception_dtor)
|
|
DEFINE_CXX_DATA1(invalid_scheduler_policy_key, &cexception_cxx_type_info, cexception_dtor)
|
|
DEFINE_CXX_DATA1(invalid_scheduler_policy_thread_specification, &cexception_cxx_type_info, cexception_dtor)
|
|
DEFINE_CXX_DATA1(invalid_scheduler_policy_value, &cexception_cxx_type_info, cexception_dtor)
|
|
DEFINE_CXX_DATA1(scheduler_resource_allocation_error, &cexception_cxx_type_info, cexception_dtor)
|
|
|
|
__ASM_BLOCK_BEGIN(concurrency_exception_vtables)
|
|
__ASM_VTABLE(improper_lock,
|
|
VTABLE_ADD_FUNC(cexception_vector_dtor)
|
|
VTABLE_ADD_FUNC(cexception_what));
|
|
__ASM_VTABLE(improper_scheduler_attach,
|
|
VTABLE_ADD_FUNC(cexception_vector_dtor)
|
|
VTABLE_ADD_FUNC(cexception_what));
|
|
__ASM_VTABLE(improper_scheduler_detach,
|
|
VTABLE_ADD_FUNC(cexception_vector_dtor)
|
|
VTABLE_ADD_FUNC(cexception_what));
|
|
__ASM_VTABLE(invalid_scheduler_policy_key,
|
|
VTABLE_ADD_FUNC(cexception_vector_dtor)
|
|
VTABLE_ADD_FUNC(cexception_what));
|
|
__ASM_VTABLE(invalid_scheduler_policy_thread_specification,
|
|
VTABLE_ADD_FUNC(cexception_vector_dtor)
|
|
VTABLE_ADD_FUNC(cexception_what));
|
|
__ASM_VTABLE(invalid_scheduler_policy_value,
|
|
VTABLE_ADD_FUNC(cexception_vector_dtor)
|
|
VTABLE_ADD_FUNC(cexception_what));
|
|
__ASM_VTABLE(scheduler_resource_allocation_error,
|
|
VTABLE_ADD_FUNC(cexception_vector_dtor)
|
|
VTABLE_ADD_FUNC(cexception_what));
|
|
__ASM_BLOCK_END
|
|
|
|
static Context* try_get_current_context(void)
|
|
{
|
|
if (context_tls_index == TLS_OUT_OF_INDEXES)
|
|
return NULL;
|
|
return TlsGetValue(context_tls_index);
|
|
}
|
|
|
|
static Context* get_current_context(void)
|
|
{
|
|
Context *ret;
|
|
|
|
if (context_tls_index == TLS_OUT_OF_INDEXES) {
|
|
int tls_index = TlsAlloc();
|
|
if (tls_index == TLS_OUT_OF_INDEXES) {
|
|
scheduler_resource_allocation_error e;
|
|
scheduler_resource_allocation_error_ctor_name(&e, NULL,
|
|
HRESULT_FROM_WIN32(GetLastError()));
|
|
_CxxThrowException(&e.e, &scheduler_resource_allocation_error_exception_type);
|
|
}
|
|
|
|
if(InterlockedCompareExchange(&context_tls_index, tls_index, TLS_OUT_OF_INDEXES) != TLS_OUT_OF_INDEXES)
|
|
TlsFree(tls_index);
|
|
}
|
|
|
|
ret = TlsGetValue(context_tls_index);
|
|
if (!ret) {
|
|
ExternalContextBase *context = operator_new(sizeof(ExternalContextBase));
|
|
ExternalContextBase_ctor(context);
|
|
TlsSetValue(context_tls_index, context);
|
|
ret = &context->context;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
static Scheduler* try_get_current_scheduler(void)
|
|
{
|
|
ExternalContextBase *context = (ExternalContextBase*)try_get_current_context();
|
|
|
|
if (!context)
|
|
return NULL;
|
|
|
|
if (context->context.vtable != &ExternalContextBase_vtable) {
|
|
ERR("unknown context set\n");
|
|
return NULL;
|
|
}
|
|
return context->scheduler.scheduler;
|
|
}
|
|
|
|
static Scheduler* get_current_scheduler(void)
|
|
{
|
|
ExternalContextBase *context = (ExternalContextBase*)get_current_context();
|
|
|
|
if (context->context.vtable != &ExternalContextBase_vtable) {
|
|
ERR("unknown context set\n");
|
|
return NULL;
|
|
}
|
|
return context->scheduler.scheduler;
|
|
}
|
|
|
|
/* ?CurrentContext@Context@Concurrency@@SAPAV12@XZ */
|
|
/* ?CurrentContext@Context@Concurrency@@SAPEAV12@XZ */
|
|
Context* __cdecl Context_CurrentContext(void)
|
|
{
|
|
TRACE("()\n");
|
|
return get_current_context();
|
|
}
|
|
|
|
/* ?Id@Context@Concurrency@@SAIXZ */
|
|
unsigned int __cdecl Context_Id(void)
|
|
{
|
|
Context *ctx = try_get_current_context();
|
|
TRACE("()\n");
|
|
return ctx ? call_Context_GetId(ctx) : -1;
|
|
}
|
|
|
|
/* ?Block@Context@Concurrency@@SAXXZ */
|
|
void __cdecl Context_Block(void)
|
|
{
|
|
FIXME("()\n");
|
|
}
|
|
|
|
/* ?Yield@Context@Concurrency@@SAXXZ */
|
|
/* ?_Yield@_Context@details@Concurrency@@SAXXZ */
|
|
void __cdecl Context_Yield(void)
|
|
{
|
|
FIXME("()\n");
|
|
}
|
|
|
|
/* ?_SpinYield@Context@Concurrency@@SAXXZ */
|
|
void __cdecl Context__SpinYield(void)
|
|
{
|
|
FIXME("()\n");
|
|
}
|
|
|
|
/* ?IsCurrentTaskCollectionCanceling@Context@Concurrency@@SA_NXZ */
|
|
bool __cdecl Context_IsCurrentTaskCollectionCanceling(void)
|
|
{
|
|
FIXME("()\n");
|
|
return FALSE;
|
|
}
|
|
|
|
/* ?Oversubscribe@Context@Concurrency@@SAX_N@Z */
|
|
void __cdecl Context_Oversubscribe(bool begin)
|
|
{
|
|
FIXME("(%x)\n", begin);
|
|
}
|
|
|
|
/* ?ScheduleGroupId@Context@Concurrency@@SAIXZ */
|
|
unsigned int __cdecl Context_ScheduleGroupId(void)
|
|
{
|
|
Context *ctx = try_get_current_context();
|
|
TRACE("()\n");
|
|
return ctx ? call_Context_GetScheduleGroupId(ctx) : -1;
|
|
}
|
|
|
|
/* ?VirtualProcessorId@Context@Concurrency@@SAIXZ */
|
|
unsigned int __cdecl Context_VirtualProcessorId(void)
|
|
{
|
|
Context *ctx = try_get_current_context();
|
|
TRACE("()\n");
|
|
return ctx ? call_Context_GetVirtualProcessorId(ctx) : -1;
|
|
}
|
|
|
|
#if _MSVCR_VER > 100
|
|
/* ?_CurrentContext@_Context@details@Concurrency@@SA?AV123@XZ */
|
|
_Context *__cdecl _Context__CurrentContext(_Context *ret)
|
|
{
|
|
TRACE("(%p)\n", ret);
|
|
ret->context = Context_CurrentContext();
|
|
return ret;
|
|
}
|
|
#endif
|
|
|
|
DEFINE_THISCALL_WRAPPER(ExternalContextBase_GetId, 4)
|
|
unsigned int __thiscall ExternalContextBase_GetId(const ExternalContextBase *this)
|
|
{
|
|
TRACE("(%p)->()\n", this);
|
|
return this->id;
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ExternalContextBase_GetVirtualProcessorId, 4)
|
|
unsigned int __thiscall ExternalContextBase_GetVirtualProcessorId(const ExternalContextBase *this)
|
|
{
|
|
FIXME("(%p)->() stub\n", this);
|
|
return -1;
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ExternalContextBase_GetScheduleGroupId, 4)
|
|
unsigned int __thiscall ExternalContextBase_GetScheduleGroupId(const ExternalContextBase *this)
|
|
{
|
|
FIXME("(%p)->() stub\n", this);
|
|
return -1;
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ExternalContextBase_Unblock, 4)
|
|
void __thiscall ExternalContextBase_Unblock(ExternalContextBase *this)
|
|
{
|
|
FIXME("(%p)->() stub\n", this);
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ExternalContextBase_IsSynchronouslyBlocked, 4)
|
|
bool __thiscall ExternalContextBase_IsSynchronouslyBlocked(const ExternalContextBase *this)
|
|
{
|
|
FIXME("(%p)->() stub\n", this);
|
|
return FALSE;
|
|
}
|
|
|
|
static void ExternalContextBase_dtor(ExternalContextBase *this)
|
|
{
|
|
struct scheduler_list *scheduler_cur, *scheduler_next;
|
|
union allocator_cache_entry *next, *cur;
|
|
int i;
|
|
|
|
/* TODO: move the allocator cache to scheduler so it can be reused */
|
|
for(i=0; i<ARRAY_SIZE(this->allocator_cache); i++) {
|
|
for(cur = this->allocator_cache[i]; cur; cur=next) {
|
|
next = cur->free.next;
|
|
operator_delete(cur);
|
|
}
|
|
}
|
|
|
|
if (this->scheduler.scheduler) {
|
|
call_Scheduler_Release(this->scheduler.scheduler);
|
|
|
|
for(scheduler_cur=this->scheduler.next; scheduler_cur; scheduler_cur=scheduler_next) {
|
|
scheduler_next = scheduler_cur->next;
|
|
call_Scheduler_Release(scheduler_cur->scheduler);
|
|
operator_delete(scheduler_cur);
|
|
}
|
|
}
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ExternalContextBase_vector_dtor, 8)
|
|
Context* __thiscall ExternalContextBase_vector_dtor(ExternalContextBase *this, unsigned int flags)
|
|
{
|
|
TRACE("(%p %x)\n", this, flags);
|
|
if(flags & 2) {
|
|
/* we have an array, with the number of elements stored before the first object */
|
|
INT_PTR i, *ptr = (INT_PTR *)this-1;
|
|
|
|
for(i=*ptr-1; i>=0; i--)
|
|
ExternalContextBase_dtor(this+i);
|
|
operator_delete(ptr);
|
|
} else {
|
|
ExternalContextBase_dtor(this);
|
|
if(flags & 1)
|
|
operator_delete(this);
|
|
}
|
|
|
|
return &this->context;
|
|
}
|
|
|
|
static void ExternalContextBase_ctor(ExternalContextBase *this)
|
|
{
|
|
TRACE("(%p)->()\n", this);
|
|
|
|
memset(this, 0, sizeof(*this));
|
|
this->context.vtable = &ExternalContextBase_vtable;
|
|
this->id = InterlockedIncrement(&context_id);
|
|
|
|
create_default_scheduler();
|
|
this->scheduler.scheduler = &default_scheduler->scheduler;
|
|
call_Scheduler_Reference(&default_scheduler->scheduler);
|
|
}
|
|
|
|
/* ?Alloc@Concurrency@@YAPAXI@Z */
|
|
/* ?Alloc@Concurrency@@YAPEAX_K@Z */
|
|
void * CDECL Concurrency_Alloc(size_t size)
|
|
{
|
|
ExternalContextBase *context = (ExternalContextBase*)get_current_context();
|
|
union allocator_cache_entry *p;
|
|
|
|
size += FIELD_OFFSET(union allocator_cache_entry, alloc.mem);
|
|
if (size < sizeof(*p))
|
|
size = sizeof(*p);
|
|
|
|
if (context->context.vtable != &ExternalContextBase_vtable) {
|
|
p = operator_new(size);
|
|
p->alloc.bucket = -1;
|
|
}else {
|
|
int i;
|
|
|
|
C_ASSERT(sizeof(union allocator_cache_entry) <= 1 << 4);
|
|
for(i=0; i<ARRAY_SIZE(context->allocator_cache); i++)
|
|
if (1 << (i+4) >= size) break;
|
|
|
|
if(i==ARRAY_SIZE(context->allocator_cache)) {
|
|
p = operator_new(size);
|
|
p->alloc.bucket = -1;
|
|
}else if (context->allocator_cache[i]) {
|
|
p = context->allocator_cache[i];
|
|
context->allocator_cache[i] = p->free.next;
|
|
p->alloc.bucket = i;
|
|
}else {
|
|
p = operator_new(1 << (i+4));
|
|
p->alloc.bucket = i;
|
|
}
|
|
}
|
|
|
|
TRACE("(%Iu) returning %p\n", size, p->alloc.mem);
|
|
return p->alloc.mem;
|
|
}
|
|
|
|
/* ?Free@Concurrency@@YAXPAX@Z */
|
|
/* ?Free@Concurrency@@YAXPEAX@Z */
|
|
void CDECL Concurrency_Free(void* mem)
|
|
{
|
|
union allocator_cache_entry *p = (union allocator_cache_entry*)((char*)mem-FIELD_OFFSET(union allocator_cache_entry, alloc.mem));
|
|
ExternalContextBase *context = (ExternalContextBase*)get_current_context();
|
|
int bucket = p->alloc.bucket;
|
|
|
|
TRACE("(%p)\n", mem);
|
|
|
|
if (context->context.vtable != &ExternalContextBase_vtable) {
|
|
operator_delete(p);
|
|
}else {
|
|
if(bucket >= 0 && bucket < ARRAY_SIZE(context->allocator_cache) &&
|
|
(!context->allocator_cache[bucket] || context->allocator_cache[bucket]->free.depth < 20)) {
|
|
p->free.next = context->allocator_cache[bucket];
|
|
p->free.depth = p->free.next ? p->free.next->free.depth+1 : 0;
|
|
context->allocator_cache[bucket] = p;
|
|
}else {
|
|
operator_delete(p);
|
|
}
|
|
}
|
|
}
|
|
|
|
/* ?SetPolicyValue@SchedulerPolicy@Concurrency@@QAEIW4PolicyElementKey@2@I@Z */
|
|
/* ?SetPolicyValue@SchedulerPolicy@Concurrency@@QEAAIW4PolicyElementKey@2@I@Z */
|
|
DEFINE_THISCALL_WRAPPER(SchedulerPolicy_SetPolicyValue, 12)
|
|
unsigned int __thiscall SchedulerPolicy_SetPolicyValue(SchedulerPolicy *this,
|
|
PolicyElementKey policy, unsigned int val)
|
|
{
|
|
unsigned int ret;
|
|
|
|
TRACE("(%p %d %d)\n", this, policy, val);
|
|
|
|
if (policy == MinConcurrency) {
|
|
invalid_scheduler_policy_key e;
|
|
invalid_scheduler_policy_key_ctor_str(&e, "MinConcurrency");
|
|
_CxxThrowException(&e, &invalid_scheduler_policy_key_exception_type);
|
|
}
|
|
if (policy == MaxConcurrency) {
|
|
invalid_scheduler_policy_key e;
|
|
invalid_scheduler_policy_key_ctor_str(&e, "MaxConcurrency");
|
|
_CxxThrowException(&e, &invalid_scheduler_policy_key_exception_type);
|
|
}
|
|
if (policy >= last_policy_id) {
|
|
invalid_scheduler_policy_key e;
|
|
invalid_scheduler_policy_key_ctor_str(&e, "Invalid policy");
|
|
_CxxThrowException(&e, &invalid_scheduler_policy_key_exception_type);
|
|
}
|
|
|
|
switch(policy) {
|
|
case SchedulerKind:
|
|
if (val) {
|
|
invalid_scheduler_policy_value e;
|
|
invalid_scheduler_policy_value_ctor_str(&e, "SchedulerKind");
|
|
_CxxThrowException(&e, &invalid_scheduler_policy_value_exception_type);
|
|
}
|
|
break;
|
|
case TargetOversubscriptionFactor:
|
|
if (!val) {
|
|
invalid_scheduler_policy_value e;
|
|
invalid_scheduler_policy_value_ctor_str(&e, "TargetOversubscriptionFactor");
|
|
_CxxThrowException(&e, &invalid_scheduler_policy_value_exception_type);
|
|
}
|
|
break;
|
|
case ContextPriority:
|
|
if (((int)val < -7 /* THREAD_PRIORITY_REALTIME_LOWEST */
|
|
|| val > 6 /* THREAD_PRIORITY_REALTIME_HIGHEST */)
|
|
&& val != THREAD_PRIORITY_IDLE && val != THREAD_PRIORITY_TIME_CRITICAL
|
|
&& val != INHERIT_THREAD_PRIORITY) {
|
|
invalid_scheduler_policy_value e;
|
|
invalid_scheduler_policy_value_ctor_str(&e, "ContextPriority");
|
|
_CxxThrowException(&e, &invalid_scheduler_policy_value_exception_type);
|
|
}
|
|
break;
|
|
case SchedulingProtocol:
|
|
case DynamicProgressFeedback:
|
|
case WinRTInitialization:
|
|
if (val != 0 && val != 1) {
|
|
invalid_scheduler_policy_value e;
|
|
invalid_scheduler_policy_value_ctor_str(&e, "SchedulingProtocol");
|
|
_CxxThrowException(&e, &invalid_scheduler_policy_value_exception_type);
|
|
}
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
ret = this->policy_container->policies[policy];
|
|
this->policy_container->policies[policy] = val;
|
|
return ret;
|
|
}
|
|
|
|
/* ?SetConcurrencyLimits@SchedulerPolicy@Concurrency@@QAEXII@Z */
|
|
/* ?SetConcurrencyLimits@SchedulerPolicy@Concurrency@@QEAAXII@Z */
|
|
DEFINE_THISCALL_WRAPPER(SchedulerPolicy_SetConcurrencyLimits, 12)
|
|
void __thiscall SchedulerPolicy_SetConcurrencyLimits(SchedulerPolicy *this,
|
|
unsigned int min_concurrency, unsigned int max_concurrency)
|
|
{
|
|
TRACE("(%p %d %d)\n", this, min_concurrency, max_concurrency);
|
|
|
|
if (min_concurrency > max_concurrency) {
|
|
invalid_scheduler_policy_thread_specification e;
|
|
invalid_scheduler_policy_thread_specification_ctor_str(&e, NULL);
|
|
_CxxThrowException(&e, &invalid_scheduler_policy_thread_specification_exception_type);
|
|
}
|
|
if (!max_concurrency) {
|
|
invalid_scheduler_policy_value e;
|
|
invalid_scheduler_policy_value_ctor_str(&e, "MaxConcurrency");
|
|
_CxxThrowException(&e, &invalid_scheduler_policy_value_exception_type);
|
|
}
|
|
|
|
this->policy_container->policies[MinConcurrency] = min_concurrency;
|
|
this->policy_container->policies[MaxConcurrency] = max_concurrency;
|
|
}
|
|
|
|
/* ?GetPolicyValue@SchedulerPolicy@Concurrency@@QBEIW4PolicyElementKey@2@@Z */
|
|
/* ?GetPolicyValue@SchedulerPolicy@Concurrency@@QEBAIW4PolicyElementKey@2@@Z */
|
|
DEFINE_THISCALL_WRAPPER(SchedulerPolicy_GetPolicyValue, 8)
|
|
unsigned int __thiscall SchedulerPolicy_GetPolicyValue(
|
|
const SchedulerPolicy *this, PolicyElementKey policy)
|
|
{
|
|
TRACE("(%p %d)\n", this, policy);
|
|
|
|
if (policy >= last_policy_id) {
|
|
invalid_scheduler_policy_key e;
|
|
invalid_scheduler_policy_key_ctor_str(&e, "Invalid policy");
|
|
_CxxThrowException(&e, &invalid_scheduler_policy_key_exception_type);
|
|
}
|
|
return this->policy_container->policies[policy];
|
|
}
|
|
|
|
/* ??0SchedulerPolicy@Concurrency@@QAE@XZ */
|
|
/* ??0SchedulerPolicy@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(SchedulerPolicy_ctor, 4)
|
|
SchedulerPolicy* __thiscall SchedulerPolicy_ctor(SchedulerPolicy *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
this->policy_container = operator_new(sizeof(*this->policy_container));
|
|
/* TODO: default values can probably be affected by CurrentScheduler */
|
|
this->policy_container->policies[SchedulerKind] = 0;
|
|
this->policy_container->policies[MaxConcurrency] = -1;
|
|
this->policy_container->policies[MinConcurrency] = 1;
|
|
this->policy_container->policies[TargetOversubscriptionFactor] = 1;
|
|
this->policy_container->policies[LocalContextCacheSize] = 8;
|
|
this->policy_container->policies[ContextStackSize] = 0;
|
|
this->policy_container->policies[ContextPriority] = THREAD_PRIORITY_NORMAL;
|
|
this->policy_container->policies[SchedulingProtocol] = 0;
|
|
this->policy_container->policies[DynamicProgressFeedback] = 1;
|
|
return this;
|
|
}
|
|
|
|
/* ??0SchedulerPolicy@Concurrency@@QAA@IZZ */
|
|
/* ??0SchedulerPolicy@Concurrency@@QEAA@_KZZ */
|
|
/* TODO: don't leak policy_container on exception */
|
|
SchedulerPolicy* WINAPIV SchedulerPolicy_ctor_policies(
|
|
SchedulerPolicy *this, size_t n, ...)
|
|
{
|
|
unsigned int min_concurrency, max_concurrency;
|
|
__ms_va_list valist;
|
|
size_t i;
|
|
|
|
TRACE("(%p %Iu)\n", this, n);
|
|
|
|
SchedulerPolicy_ctor(this);
|
|
min_concurrency = this->policy_container->policies[MinConcurrency];
|
|
max_concurrency = this->policy_container->policies[MaxConcurrency];
|
|
|
|
__ms_va_start(valist, n);
|
|
for(i=0; i<n; i++) {
|
|
PolicyElementKey policy = va_arg(valist, PolicyElementKey);
|
|
unsigned int val = va_arg(valist, unsigned int);
|
|
|
|
if(policy == MinConcurrency)
|
|
min_concurrency = val;
|
|
else if(policy == MaxConcurrency)
|
|
max_concurrency = val;
|
|
else
|
|
SchedulerPolicy_SetPolicyValue(this, policy, val);
|
|
}
|
|
__ms_va_end(valist);
|
|
|
|
SchedulerPolicy_SetConcurrencyLimits(this, min_concurrency, max_concurrency);
|
|
return this;
|
|
}
|
|
|
|
/* ??4SchedulerPolicy@Concurrency@@QAEAAV01@ABV01@@Z */
|
|
/* ??4SchedulerPolicy@Concurrency@@QEAAAEAV01@AEBV01@@Z */
|
|
DEFINE_THISCALL_WRAPPER(SchedulerPolicy_op_assign, 8)
|
|
SchedulerPolicy* __thiscall SchedulerPolicy_op_assign(
|
|
SchedulerPolicy *this, const SchedulerPolicy *rhs)
|
|
{
|
|
TRACE("(%p %p)\n", this, rhs);
|
|
memcpy(this->policy_container->policies, rhs->policy_container->policies,
|
|
sizeof(this->policy_container->policies));
|
|
return this;
|
|
}
|
|
|
|
/* ??0SchedulerPolicy@Concurrency@@QAE@ABV01@@Z */
|
|
/* ??0SchedulerPolicy@Concurrency@@QEAA@AEBV01@@Z */
|
|
DEFINE_THISCALL_WRAPPER(SchedulerPolicy_copy_ctor, 8)
|
|
SchedulerPolicy* __thiscall SchedulerPolicy_copy_ctor(
|
|
SchedulerPolicy *this, const SchedulerPolicy *rhs)
|
|
{
|
|
TRACE("(%p %p)\n", this, rhs);
|
|
SchedulerPolicy_ctor(this);
|
|
return SchedulerPolicy_op_assign(this, rhs);
|
|
}
|
|
|
|
/* ??1SchedulerPolicy@Concurrency@@QAE@XZ */
|
|
/* ??1SchedulerPolicy@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(SchedulerPolicy_dtor, 4)
|
|
void __thiscall SchedulerPolicy_dtor(SchedulerPolicy *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
operator_delete(this->policy_container);
|
|
}
|
|
|
|
static void ThreadScheduler_dtor(ThreadScheduler *this)
|
|
{
|
|
int i;
|
|
|
|
if(this->ref != 0) WARN("ref = %d\n", this->ref);
|
|
SchedulerPolicy_dtor(&this->policy);
|
|
|
|
for(i=0; i<this->shutdown_count; i++)
|
|
SetEvent(this->shutdown_events[i]);
|
|
operator_delete(this->shutdown_events);
|
|
|
|
this->cs.DebugInfo->Spare[0] = 0;
|
|
DeleteCriticalSection(&this->cs);
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ThreadScheduler_Id, 4)
|
|
unsigned int __thiscall ThreadScheduler_Id(const ThreadScheduler *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
return this->id;
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ThreadScheduler_GetNumberOfVirtualProcessors, 4)
|
|
unsigned int __thiscall ThreadScheduler_GetNumberOfVirtualProcessors(const ThreadScheduler *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
return this->virt_proc_no;
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ThreadScheduler_GetPolicy, 8)
|
|
SchedulerPolicy* __thiscall ThreadScheduler_GetPolicy(
|
|
const ThreadScheduler *this, SchedulerPolicy *ret)
|
|
{
|
|
TRACE("(%p %p)\n", this, ret);
|
|
return SchedulerPolicy_copy_ctor(ret, &this->policy);
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ThreadScheduler_Reference, 4)
|
|
unsigned int __thiscall ThreadScheduler_Reference(ThreadScheduler *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
return InterlockedIncrement(&this->ref);
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ThreadScheduler_Release, 4)
|
|
unsigned int __thiscall ThreadScheduler_Release(ThreadScheduler *this)
|
|
{
|
|
unsigned int ret = InterlockedDecrement(&this->ref);
|
|
|
|
TRACE("(%p)\n", this);
|
|
|
|
if(!ret) {
|
|
ThreadScheduler_dtor(this);
|
|
operator_delete(this);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ThreadScheduler_RegisterShutdownEvent, 8)
|
|
void __thiscall ThreadScheduler_RegisterShutdownEvent(ThreadScheduler *this, HANDLE event)
|
|
{
|
|
HANDLE *shutdown_events;
|
|
int size;
|
|
|
|
TRACE("(%p %p)\n", this, event);
|
|
|
|
EnterCriticalSection(&this->cs);
|
|
|
|
size = this->shutdown_size ? this->shutdown_size * 2 : 1;
|
|
shutdown_events = operator_new(size * sizeof(*shutdown_events));
|
|
memcpy(shutdown_events, this->shutdown_events,
|
|
this->shutdown_count * sizeof(*shutdown_events));
|
|
operator_delete(this->shutdown_events);
|
|
this->shutdown_size = size;
|
|
this->shutdown_events = shutdown_events;
|
|
this->shutdown_events[this->shutdown_count++] = event;
|
|
|
|
LeaveCriticalSection(&this->cs);
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ThreadScheduler_Attach, 4)
|
|
void __thiscall ThreadScheduler_Attach(ThreadScheduler *this)
|
|
{
|
|
ExternalContextBase *context = (ExternalContextBase*)get_current_context();
|
|
|
|
TRACE("(%p)\n", this);
|
|
|
|
if(context->context.vtable != &ExternalContextBase_vtable) {
|
|
ERR("unknown context set\n");
|
|
return;
|
|
}
|
|
|
|
if(context->scheduler.scheduler == &this->scheduler) {
|
|
improper_scheduler_attach e;
|
|
improper_scheduler_attach_ctor_str(&e, NULL);
|
|
_CxxThrowException(&e, &improper_scheduler_attach_exception_type);
|
|
}
|
|
|
|
if(context->scheduler.scheduler) {
|
|
struct scheduler_list *l = operator_new(sizeof(*l));
|
|
*l = context->scheduler;
|
|
context->scheduler.next = l;
|
|
}
|
|
context->scheduler.scheduler = &this->scheduler;
|
|
ThreadScheduler_Reference(this);
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ThreadScheduler_CreateScheduleGroup_loc, 8)
|
|
/*ScheduleGroup*/void* __thiscall ThreadScheduler_CreateScheduleGroup_loc(
|
|
ThreadScheduler *this, /*location*/void *placement)
|
|
{
|
|
FIXME("(%p %p) stub\n", this, placement);
|
|
return NULL;
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ThreadScheduler_CreateScheduleGroup, 4)
|
|
/*ScheduleGroup*/void* __thiscall ThreadScheduler_CreateScheduleGroup(ThreadScheduler *this)
|
|
{
|
|
FIXME("(%p) stub\n", this);
|
|
return NULL;
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ThreadScheduler_ScheduleTask_loc, 16)
|
|
void __thiscall ThreadScheduler_ScheduleTask_loc(ThreadScheduler *this,
|
|
void (__cdecl *proc)(void*), void* data, /*location*/void *placement)
|
|
{
|
|
FIXME("(%p %p %p %p) stub\n", this, proc, data, placement);
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ThreadScheduler_ScheduleTask, 12)
|
|
void __thiscall ThreadScheduler_ScheduleTask(ThreadScheduler *this,
|
|
void (__cdecl *proc)(void*), void* data)
|
|
{
|
|
FIXME("(%p %p %p) stub\n", this, proc, data);
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ThreadScheduler_IsAvailableLocation, 8)
|
|
bool __thiscall ThreadScheduler_IsAvailableLocation(
|
|
const ThreadScheduler *this, const /*location*/void *placement)
|
|
{
|
|
FIXME("(%p %p) stub\n", this, placement);
|
|
return FALSE;
|
|
}
|
|
|
|
DEFINE_THISCALL_WRAPPER(ThreadScheduler_vector_dtor, 8)
|
|
Scheduler* __thiscall ThreadScheduler_vector_dtor(ThreadScheduler *this, unsigned int flags)
|
|
{
|
|
TRACE("(%p %x)\n", this, flags);
|
|
if(flags & 2) {
|
|
/* we have an array, with the number of elements stored before the first object */
|
|
INT_PTR i, *ptr = (INT_PTR *)this-1;
|
|
|
|
for(i=*ptr-1; i>=0; i--)
|
|
ThreadScheduler_dtor(this+i);
|
|
operator_delete(ptr);
|
|
} else {
|
|
ThreadScheduler_dtor(this);
|
|
if(flags & 1)
|
|
operator_delete(this);
|
|
}
|
|
|
|
return &this->scheduler;
|
|
}
|
|
|
|
static ThreadScheduler* ThreadScheduler_ctor(ThreadScheduler *this,
|
|
const SchedulerPolicy *policy)
|
|
{
|
|
SYSTEM_INFO si;
|
|
|
|
TRACE("(%p)->()\n", this);
|
|
|
|
this->scheduler.vtable = &ThreadScheduler_vtable;
|
|
this->ref = 1;
|
|
this->id = InterlockedIncrement(&scheduler_id);
|
|
SchedulerPolicy_copy_ctor(&this->policy, policy);
|
|
|
|
GetSystemInfo(&si);
|
|
this->virt_proc_no = SchedulerPolicy_GetPolicyValue(&this->policy, MaxConcurrency);
|
|
if(this->virt_proc_no > si.dwNumberOfProcessors)
|
|
this->virt_proc_no = si.dwNumberOfProcessors;
|
|
|
|
this->shutdown_count = this->shutdown_size = 0;
|
|
this->shutdown_events = NULL;
|
|
|
|
InitializeCriticalSection(&this->cs);
|
|
this->cs.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": ThreadScheduler");
|
|
return this;
|
|
}
|
|
|
|
/* ?Create@Scheduler@Concurrency@@SAPAV12@ABVSchedulerPolicy@2@@Z */
|
|
/* ?Create@Scheduler@Concurrency@@SAPEAV12@AEBVSchedulerPolicy@2@@Z */
|
|
Scheduler* __cdecl Scheduler_Create(const SchedulerPolicy *policy)
|
|
{
|
|
ThreadScheduler *ret;
|
|
|
|
TRACE("(%p)\n", policy);
|
|
|
|
ret = operator_new(sizeof(*ret));
|
|
return &ThreadScheduler_ctor(ret, policy)->scheduler;
|
|
}
|
|
|
|
/* ?ResetDefaultSchedulerPolicy@Scheduler@Concurrency@@SAXXZ */
|
|
void __cdecl Scheduler_ResetDefaultSchedulerPolicy(void)
|
|
{
|
|
TRACE("()\n");
|
|
|
|
EnterCriticalSection(&default_scheduler_cs);
|
|
if(default_scheduler_policy.policy_container)
|
|
SchedulerPolicy_dtor(&default_scheduler_policy);
|
|
SchedulerPolicy_ctor(&default_scheduler_policy);
|
|
LeaveCriticalSection(&default_scheduler_cs);
|
|
}
|
|
|
|
/* ?SetDefaultSchedulerPolicy@Scheduler@Concurrency@@SAXABVSchedulerPolicy@2@@Z */
|
|
/* ?SetDefaultSchedulerPolicy@Scheduler@Concurrency@@SAXAEBVSchedulerPolicy@2@@Z */
|
|
void __cdecl Scheduler_SetDefaultSchedulerPolicy(const SchedulerPolicy *policy)
|
|
{
|
|
TRACE("(%p)\n", policy);
|
|
|
|
EnterCriticalSection(&default_scheduler_cs);
|
|
if(!default_scheduler_policy.policy_container)
|
|
SchedulerPolicy_copy_ctor(&default_scheduler_policy, policy);
|
|
else
|
|
SchedulerPolicy_op_assign(&default_scheduler_policy, policy);
|
|
LeaveCriticalSection(&default_scheduler_cs);
|
|
}
|
|
|
|
/* ?Create@CurrentScheduler@Concurrency@@SAXABVSchedulerPolicy@2@@Z */
|
|
/* ?Create@CurrentScheduler@Concurrency@@SAXAEBVSchedulerPolicy@2@@Z */
|
|
void __cdecl CurrentScheduler_Create(const SchedulerPolicy *policy)
|
|
{
|
|
Scheduler *scheduler;
|
|
|
|
TRACE("(%p)\n", policy);
|
|
|
|
scheduler = Scheduler_Create(policy);
|
|
call_Scheduler_Attach(scheduler);
|
|
}
|
|
|
|
/* ?Detach@CurrentScheduler@Concurrency@@SAXXZ */
|
|
void __cdecl CurrentScheduler_Detach(void)
|
|
{
|
|
ExternalContextBase *context = (ExternalContextBase*)try_get_current_context();
|
|
|
|
TRACE("()\n");
|
|
|
|
if(!context) {
|
|
improper_scheduler_detach e;
|
|
improper_scheduler_detach_ctor_str(&e, NULL);
|
|
_CxxThrowException(&e, &improper_scheduler_detach_exception_type);
|
|
}
|
|
|
|
if(context->context.vtable != &ExternalContextBase_vtable) {
|
|
ERR("unknown context set\n");
|
|
return;
|
|
}
|
|
|
|
if(!context->scheduler.next) {
|
|
improper_scheduler_detach e;
|
|
improper_scheduler_detach_ctor_str(&e, NULL);
|
|
_CxxThrowException(&e, &improper_scheduler_detach_exception_type);
|
|
}
|
|
|
|
call_Scheduler_Release(context->scheduler.scheduler);
|
|
if(!context->scheduler.next) {
|
|
context->scheduler.scheduler = NULL;
|
|
}else {
|
|
struct scheduler_list *entry = context->scheduler.next;
|
|
context->scheduler.scheduler = entry->scheduler;
|
|
context->scheduler.next = entry->next;
|
|
operator_delete(entry);
|
|
}
|
|
}
|
|
|
|
static void create_default_scheduler(void)
|
|
{
|
|
if(default_scheduler)
|
|
return;
|
|
|
|
EnterCriticalSection(&default_scheduler_cs);
|
|
if(!default_scheduler) {
|
|
ThreadScheduler *scheduler;
|
|
|
|
if(!default_scheduler_policy.policy_container)
|
|
SchedulerPolicy_ctor(&default_scheduler_policy);
|
|
|
|
scheduler = operator_new(sizeof(*scheduler));
|
|
ThreadScheduler_ctor(scheduler, &default_scheduler_policy);
|
|
default_scheduler = scheduler;
|
|
}
|
|
LeaveCriticalSection(&default_scheduler_cs);
|
|
}
|
|
|
|
/* ?Get@CurrentScheduler@Concurrency@@SAPAVScheduler@2@XZ */
|
|
/* ?Get@CurrentScheduler@Concurrency@@SAPEAVScheduler@2@XZ */
|
|
Scheduler* __cdecl CurrentScheduler_Get(void)
|
|
{
|
|
TRACE("()\n");
|
|
return get_current_scheduler();
|
|
}
|
|
|
|
#if _MSVCR_VER > 100
|
|
/* ?CreateScheduleGroup@CurrentScheduler@Concurrency@@SAPAVScheduleGroup@2@AAVlocation@2@@Z */
|
|
/* ?CreateScheduleGroup@CurrentScheduler@Concurrency@@SAPEAVScheduleGroup@2@AEAVlocation@2@@Z */
|
|
/*ScheduleGroup*/void* __cdecl CurrentScheduler_CreateScheduleGroup_loc(/*location*/void *placement)
|
|
{
|
|
TRACE("(%p)\n", placement);
|
|
return call_Scheduler_CreateScheduleGroup_loc(get_current_scheduler(), placement);
|
|
}
|
|
#endif
|
|
|
|
/* ?CreateScheduleGroup@CurrentScheduler@Concurrency@@SAPAVScheduleGroup@2@XZ */
|
|
/* ?CreateScheduleGroup@CurrentScheduler@Concurrency@@SAPEAVScheduleGroup@2@XZ */
|
|
/*ScheduleGroup*/void* __cdecl CurrentScheduler_CreateScheduleGroup(void)
|
|
{
|
|
TRACE("()\n");
|
|
return call_Scheduler_CreateScheduleGroup(get_current_scheduler());
|
|
}
|
|
|
|
/* ?GetNumberOfVirtualProcessors@CurrentScheduler@Concurrency@@SAIXZ */
|
|
unsigned int __cdecl CurrentScheduler_GetNumberOfVirtualProcessors(void)
|
|
{
|
|
Scheduler *scheduler = try_get_current_scheduler();
|
|
|
|
TRACE("()\n");
|
|
|
|
if(!scheduler)
|
|
return -1;
|
|
return call_Scheduler_GetNumberOfVirtualProcessors(scheduler);
|
|
}
|
|
|
|
/* ?GetPolicy@CurrentScheduler@Concurrency@@SA?AVSchedulerPolicy@2@XZ */
|
|
SchedulerPolicy* __cdecl CurrentScheduler_GetPolicy(SchedulerPolicy *policy)
|
|
{
|
|
TRACE("(%p)\n", policy);
|
|
return call_Scheduler_GetPolicy(get_current_scheduler(), policy);
|
|
}
|
|
|
|
/* ?Id@CurrentScheduler@Concurrency@@SAIXZ */
|
|
unsigned int __cdecl CurrentScheduler_Id(void)
|
|
{
|
|
Scheduler *scheduler = try_get_current_scheduler();
|
|
|
|
TRACE("()\n");
|
|
|
|
if(!scheduler)
|
|
return -1;
|
|
return call_Scheduler_Id(scheduler);
|
|
}
|
|
|
|
#if _MSVCR_VER > 100
|
|
/* ?IsAvailableLocation@CurrentScheduler@Concurrency@@SA_NABVlocation@2@@Z */
|
|
/* ?IsAvailableLocation@CurrentScheduler@Concurrency@@SA_NAEBVlocation@2@@Z */
|
|
bool __cdecl CurrentScheduler_IsAvailableLocation(const /*location*/void *placement)
|
|
{
|
|
Scheduler *scheduler = try_get_current_scheduler();
|
|
|
|
TRACE("(%p)\n", placement);
|
|
|
|
if(!scheduler)
|
|
return FALSE;
|
|
return call_Scheduler_IsAvailableLocation(scheduler, placement);
|
|
}
|
|
#endif
|
|
|
|
/* ?RegisterShutdownEvent@CurrentScheduler@Concurrency@@SAXPAX@Z */
|
|
/* ?RegisterShutdownEvent@CurrentScheduler@Concurrency@@SAXPEAX@Z */
|
|
void __cdecl CurrentScheduler_RegisterShutdownEvent(HANDLE event)
|
|
{
|
|
TRACE("(%p)\n", event);
|
|
call_Scheduler_RegisterShutdownEvent(get_current_scheduler(), event);
|
|
}
|
|
|
|
#if _MSVCR_VER > 100
|
|
/* ?ScheduleTask@CurrentScheduler@Concurrency@@SAXP6AXPAX@Z0AAVlocation@2@@Z */
|
|
/* ?ScheduleTask@CurrentScheduler@Concurrency@@SAXP6AXPEAX@Z0AEAVlocation@2@@Z */
|
|
void __cdecl CurrentScheduler_ScheduleTask_loc(void (__cdecl *proc)(void*),
|
|
void *data, /*location*/void *placement)
|
|
{
|
|
TRACE("(%p %p %p)\n", proc, data, placement);
|
|
call_Scheduler_ScheduleTask_loc(get_current_scheduler(), proc, data, placement);
|
|
}
|
|
#endif
|
|
|
|
/* ?ScheduleTask@CurrentScheduler@Concurrency@@SAXP6AXPAX@Z0@Z */
|
|
/* ?ScheduleTask@CurrentScheduler@Concurrency@@SAXP6AXPEAX@Z0@Z */
|
|
void __cdecl CurrentScheduler_ScheduleTask(void (__cdecl *proc)(void*), void *data)
|
|
{
|
|
TRACE("(%p %p)\n", proc, data);
|
|
call_Scheduler_ScheduleTask(get_current_scheduler(), proc, data);
|
|
}
|
|
|
|
/* ??0_Scheduler@details@Concurrency@@QAE@PAVScheduler@2@@Z */
|
|
/* ??0_Scheduler@details@Concurrency@@QEAA@PEAVScheduler@2@@Z */
|
|
DEFINE_THISCALL_WRAPPER(_Scheduler_ctor_sched, 8)
|
|
_Scheduler* __thiscall _Scheduler_ctor_sched(_Scheduler *this, Scheduler *scheduler)
|
|
{
|
|
TRACE("(%p %p)\n", this, scheduler);
|
|
|
|
this->scheduler = scheduler;
|
|
return this;
|
|
}
|
|
|
|
/* ??_F_Scheduler@details@Concurrency@@QAEXXZ */
|
|
/* ??_F_Scheduler@details@Concurrency@@QEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(_Scheduler_ctor, 4)
|
|
_Scheduler* __thiscall _Scheduler_ctor(_Scheduler *this)
|
|
{
|
|
return _Scheduler_ctor_sched(this, NULL);
|
|
}
|
|
|
|
/* ?_GetScheduler@_Scheduler@details@Concurrency@@QAEPAVScheduler@3@XZ */
|
|
/* ?_GetScheduler@_Scheduler@details@Concurrency@@QEAAPEAVScheduler@3@XZ */
|
|
DEFINE_THISCALL_WRAPPER(_Scheduler__GetScheduler, 4)
|
|
Scheduler* __thiscall _Scheduler__GetScheduler(_Scheduler *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
return this->scheduler;
|
|
}
|
|
|
|
/* ?_Reference@_Scheduler@details@Concurrency@@QAEIXZ */
|
|
/* ?_Reference@_Scheduler@details@Concurrency@@QEAAIXZ */
|
|
DEFINE_THISCALL_WRAPPER(_Scheduler__Reference, 4)
|
|
unsigned int __thiscall _Scheduler__Reference(_Scheduler *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
return call_Scheduler_Reference(this->scheduler);
|
|
}
|
|
|
|
/* ?_Release@_Scheduler@details@Concurrency@@QAEIXZ */
|
|
/* ?_Release@_Scheduler@details@Concurrency@@QEAAIXZ */
|
|
DEFINE_THISCALL_WRAPPER(_Scheduler__Release, 4)
|
|
unsigned int __thiscall _Scheduler__Release(_Scheduler *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
return call_Scheduler_Release(this->scheduler);
|
|
}
|
|
|
|
/* ?_Get@_CurrentScheduler@details@Concurrency@@SA?AV_Scheduler@23@XZ */
|
|
_Scheduler* __cdecl _CurrentScheduler__Get(_Scheduler *ret)
|
|
{
|
|
TRACE("()\n");
|
|
return _Scheduler_ctor_sched(ret, get_current_scheduler());
|
|
}
|
|
|
|
/* ?_GetNumberOfVirtualProcessors@_CurrentScheduler@details@Concurrency@@SAIXZ */
|
|
unsigned int __cdecl _CurrentScheduler__GetNumberOfVirtualProcessors(void)
|
|
{
|
|
TRACE("()\n");
|
|
get_current_scheduler();
|
|
return CurrentScheduler_GetNumberOfVirtualProcessors();
|
|
}
|
|
|
|
/* ?_Id@_CurrentScheduler@details@Concurrency@@SAIXZ */
|
|
unsigned int __cdecl _CurrentScheduler__Id(void)
|
|
{
|
|
TRACE("()\n");
|
|
get_current_scheduler();
|
|
return CurrentScheduler_Id();
|
|
}
|
|
|
|
/* ?_ScheduleTask@_CurrentScheduler@details@Concurrency@@SAXP6AXPAX@Z0@Z */
|
|
/* ?_ScheduleTask@_CurrentScheduler@details@Concurrency@@SAXP6AXPEAX@Z0@Z */
|
|
void __cdecl _CurrentScheduler__ScheduleTask(void (__cdecl *proc)(void*), void *data)
|
|
{
|
|
TRACE("(%p %p)\n", proc, data);
|
|
CurrentScheduler_ScheduleTask(proc, data);
|
|
}
|
|
|
|
/* ?_Value@_SpinCount@details@Concurrency@@SAIXZ */
|
|
unsigned int __cdecl SpinCount__Value(void)
|
|
{
|
|
static unsigned int val = -1;
|
|
|
|
TRACE("()\n");
|
|
|
|
if(val == -1) {
|
|
SYSTEM_INFO si;
|
|
|
|
GetSystemInfo(&si);
|
|
val = si.dwNumberOfProcessors>1 ? 4000 : 0;
|
|
}
|
|
|
|
return val;
|
|
}
|
|
|
|
/* ??0?$_SpinWait@$00@details@Concurrency@@QAE@P6AXXZ@Z */
|
|
/* ??0?$_SpinWait@$00@details@Concurrency@@QEAA@P6AXXZ@Z */
|
|
DEFINE_THISCALL_WRAPPER(SpinWait_ctor_yield, 8)
|
|
SpinWait* __thiscall SpinWait_ctor_yield(SpinWait *this, yield_func yf)
|
|
{
|
|
TRACE("(%p %p)\n", this, yf);
|
|
|
|
this->state = SPINWAIT_INIT;
|
|
this->unknown = 1;
|
|
this->yield_func = yf;
|
|
return this;
|
|
}
|
|
|
|
/* ??0?$_SpinWait@$0A@@details@Concurrency@@QAE@P6AXXZ@Z */
|
|
/* ??0?$_SpinWait@$0A@@details@Concurrency@@QEAA@P6AXXZ@Z */
|
|
DEFINE_THISCALL_WRAPPER(SpinWait_ctor, 8)
|
|
SpinWait* __thiscall SpinWait_ctor(SpinWait *this, yield_func yf)
|
|
{
|
|
TRACE("(%p %p)\n", this, yf);
|
|
|
|
this->state = SPINWAIT_INIT;
|
|
this->unknown = 0;
|
|
this->yield_func = yf;
|
|
return this;
|
|
}
|
|
|
|
/* ??_F?$_SpinWait@$00@details@Concurrency@@QAEXXZ */
|
|
/* ??_F?$_SpinWait@$00@details@Concurrency@@QEAAXXZ */
|
|
/* ??_F?$_SpinWait@$0A@@details@Concurrency@@QAEXXZ */
|
|
/* ??_F?$_SpinWait@$0A@@details@Concurrency@@QEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(SpinWait_dtor, 4)
|
|
void __thiscall SpinWait_dtor(SpinWait *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
}
|
|
|
|
/* ?_DoYield@?$_SpinWait@$00@details@Concurrency@@IAEXXZ */
|
|
/* ?_DoYield@?$_SpinWait@$00@details@Concurrency@@IEAAXXZ */
|
|
/* ?_DoYield@?$_SpinWait@$0A@@details@Concurrency@@IAEXXZ */
|
|
/* ?_DoYield@?$_SpinWait@$0A@@details@Concurrency@@IEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(SpinWait__DoYield, 4)
|
|
void __thiscall SpinWait__DoYield(SpinWait *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
if(this->unknown)
|
|
this->yield_func();
|
|
}
|
|
|
|
/* ?_NumberOfSpins@?$_SpinWait@$00@details@Concurrency@@IAEKXZ */
|
|
/* ?_NumberOfSpins@?$_SpinWait@$00@details@Concurrency@@IEAAKXZ */
|
|
/* ?_NumberOfSpins@?$_SpinWait@$0A@@details@Concurrency@@IAEKXZ */
|
|
/* ?_NumberOfSpins@?$_SpinWait@$0A@@details@Concurrency@@IEAAKXZ */
|
|
DEFINE_THISCALL_WRAPPER(SpinWait__NumberOfSpins, 4)
|
|
ULONG __thiscall SpinWait__NumberOfSpins(SpinWait *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
return 1;
|
|
}
|
|
|
|
/* ?_SetSpinCount@?$_SpinWait@$00@details@Concurrency@@QAEXI@Z */
|
|
/* ?_SetSpinCount@?$_SpinWait@$00@details@Concurrency@@QEAAXI@Z */
|
|
/* ?_SetSpinCount@?$_SpinWait@$0A@@details@Concurrency@@QAEXI@Z */
|
|
/* ?_SetSpinCount@?$_SpinWait@$0A@@details@Concurrency@@QEAAXI@Z */
|
|
DEFINE_THISCALL_WRAPPER(SpinWait__SetSpinCount, 8)
|
|
void __thiscall SpinWait__SetSpinCount(SpinWait *this, unsigned int spin)
|
|
{
|
|
TRACE("(%p %d)\n", this, spin);
|
|
|
|
this->spin = spin;
|
|
this->state = spin ? SPINWAIT_SPIN : SPINWAIT_YIELD;
|
|
}
|
|
|
|
/* ?_Reset@?$_SpinWait@$00@details@Concurrency@@IAEXXZ */
|
|
/* ?_Reset@?$_SpinWait@$00@details@Concurrency@@IEAAXXZ */
|
|
/* ?_Reset@?$_SpinWait@$0A@@details@Concurrency@@IAEXXZ */
|
|
/* ?_Reset@?$_SpinWait@$0A@@details@Concurrency@@IEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(SpinWait__Reset, 4)
|
|
void __thiscall SpinWait__Reset(SpinWait *this)
|
|
{
|
|
SpinWait__SetSpinCount(this, SpinCount__Value());
|
|
}
|
|
|
|
/* ?_ShouldSpinAgain@?$_SpinWait@$00@details@Concurrency@@IAE_NXZ */
|
|
/* ?_ShouldSpinAgain@?$_SpinWait@$00@details@Concurrency@@IEAA_NXZ */
|
|
/* ?_ShouldSpinAgain@?$_SpinWait@$0A@@details@Concurrency@@IAE_NXZ */
|
|
/* ?_ShouldSpinAgain@?$_SpinWait@$0A@@details@Concurrency@@IEAA_NXZ */
|
|
DEFINE_THISCALL_WRAPPER(SpinWait__ShouldSpinAgain, 4)
|
|
bool __thiscall SpinWait__ShouldSpinAgain(SpinWait *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
this->spin--;
|
|
return this->spin > 0;
|
|
}
|
|
|
|
/* ?_SpinOnce@?$_SpinWait@$00@details@Concurrency@@QAE_NXZ */
|
|
/* ?_SpinOnce@?$_SpinWait@$00@details@Concurrency@@QEAA_NXZ */
|
|
/* ?_SpinOnce@?$_SpinWait@$0A@@details@Concurrency@@QAE_NXZ */
|
|
/* ?_SpinOnce@?$_SpinWait@$0A@@details@Concurrency@@QEAA_NXZ */
|
|
DEFINE_THISCALL_WRAPPER(SpinWait__SpinOnce, 4)
|
|
bool __thiscall SpinWait__SpinOnce(SpinWait *this)
|
|
{
|
|
switch(this->state) {
|
|
case SPINWAIT_INIT:
|
|
SpinWait__Reset(this);
|
|
/* fall through */
|
|
case SPINWAIT_SPIN:
|
|
InterlockedDecrement((LONG*)&this->spin);
|
|
if(!this->spin)
|
|
this->state = this->unknown ? SPINWAIT_YIELD : SPINWAIT_DONE;
|
|
return TRUE;
|
|
case SPINWAIT_YIELD:
|
|
this->state = SPINWAIT_DONE;
|
|
this->yield_func();
|
|
return TRUE;
|
|
default:
|
|
SpinWait__Reset(this);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
/* ??0critical_section@Concurrency@@QAE@XZ */
|
|
/* ??0critical_section@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(critical_section_ctor, 4)
|
|
critical_section* __thiscall critical_section_ctor(critical_section *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
if(!keyed_event) {
|
|
HANDLE event;
|
|
|
|
NtCreateKeyedEvent(&event, GENERIC_READ|GENERIC_WRITE, NULL, 0);
|
|
if(InterlockedCompareExchangePointer(&keyed_event, event, NULL) != NULL)
|
|
NtClose(event);
|
|
}
|
|
|
|
this->unk_thread_id = 0;
|
|
this->head = this->tail = NULL;
|
|
return this;
|
|
}
|
|
|
|
/* ??1critical_section@Concurrency@@QAE@XZ */
|
|
/* ??1critical_section@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(critical_section_dtor, 4)
|
|
void __thiscall critical_section_dtor(critical_section *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
}
|
|
|
|
static void __cdecl spin_wait_yield(void)
|
|
{
|
|
Sleep(0);
|
|
}
|
|
|
|
static inline void spin_wait_for_next_cs(cs_queue *q)
|
|
{
|
|
SpinWait sw;
|
|
|
|
if(q->next) return;
|
|
|
|
SpinWait_ctor(&sw, &spin_wait_yield);
|
|
SpinWait__Reset(&sw);
|
|
while(!q->next)
|
|
SpinWait__SpinOnce(&sw);
|
|
SpinWait_dtor(&sw);
|
|
}
|
|
|
|
static inline void cs_set_head(critical_section *cs, cs_queue *q)
|
|
{
|
|
cs->unk_thread_id = GetCurrentThreadId();
|
|
cs->unk_active.next = q->next;
|
|
cs->head = &cs->unk_active;
|
|
}
|
|
|
|
static inline void cs_lock(critical_section *cs, cs_queue *q)
|
|
{
|
|
cs_queue *last;
|
|
|
|
if(cs->unk_thread_id == GetCurrentThreadId()) {
|
|
improper_lock e;
|
|
improper_lock_ctor_str(&e, "Already locked");
|
|
_CxxThrowException(&e, &improper_lock_exception_type);
|
|
}
|
|
|
|
memset(q, 0, sizeof(*q));
|
|
last = InterlockedExchangePointer(&cs->tail, q);
|
|
if(last) {
|
|
last->next = q;
|
|
NtWaitForKeyedEvent(keyed_event, q, 0, NULL);
|
|
}
|
|
|
|
cs_set_head(cs, q);
|
|
if(InterlockedCompareExchangePointer(&cs->tail, &cs->unk_active, q) != q) {
|
|
spin_wait_for_next_cs(q);
|
|
cs->unk_active.next = q->next;
|
|
}
|
|
}
|
|
|
|
/* ?lock@critical_section@Concurrency@@QAEXXZ */
|
|
/* ?lock@critical_section@Concurrency@@QEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(critical_section_lock, 4)
|
|
void __thiscall critical_section_lock(critical_section *this)
|
|
{
|
|
cs_queue q;
|
|
|
|
TRACE("(%p)\n", this);
|
|
cs_lock(this, &q);
|
|
}
|
|
|
|
/* ?try_lock@critical_section@Concurrency@@QAE_NXZ */
|
|
/* ?try_lock@critical_section@Concurrency@@QEAA_NXZ */
|
|
DEFINE_THISCALL_WRAPPER(critical_section_try_lock, 4)
|
|
bool __thiscall critical_section_try_lock(critical_section *this)
|
|
{
|
|
cs_queue q;
|
|
|
|
TRACE("(%p)\n", this);
|
|
|
|
if(this->unk_thread_id == GetCurrentThreadId())
|
|
return FALSE;
|
|
|
|
memset(&q, 0, sizeof(q));
|
|
if(!InterlockedCompareExchangePointer(&this->tail, &q, NULL)) {
|
|
cs_set_head(this, &q);
|
|
if(InterlockedCompareExchangePointer(&this->tail, &this->unk_active, &q) != &q) {
|
|
spin_wait_for_next_cs(&q);
|
|
this->unk_active.next = q.next;
|
|
}
|
|
return TRUE;
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
/* ?unlock@critical_section@Concurrency@@QAEXXZ */
|
|
/* ?unlock@critical_section@Concurrency@@QEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(critical_section_unlock, 4)
|
|
void __thiscall critical_section_unlock(critical_section *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
this->unk_thread_id = 0;
|
|
this->head = NULL;
|
|
if(InterlockedCompareExchangePointer(&this->tail, NULL, &this->unk_active)
|
|
== &this->unk_active) return;
|
|
spin_wait_for_next_cs(&this->unk_active);
|
|
|
|
#if _MSVCR_VER >= 110
|
|
while(1) {
|
|
cs_queue *next;
|
|
|
|
if(!InterlockedExchange(&this->unk_active.next->free, TRUE))
|
|
break;
|
|
|
|
next = this->unk_active.next;
|
|
if(InterlockedCompareExchangePointer(&this->tail, NULL, next) == next) {
|
|
HeapFree(GetProcessHeap(), 0, next);
|
|
return;
|
|
}
|
|
spin_wait_for_next_cs(next);
|
|
|
|
this->unk_active.next = next->next;
|
|
HeapFree(GetProcessHeap(), 0, next);
|
|
}
|
|
#endif
|
|
|
|
NtReleaseKeyedEvent(keyed_event, this->unk_active.next, 0, NULL);
|
|
}
|
|
|
|
/* ?native_handle@critical_section@Concurrency@@QAEAAV12@XZ */
|
|
/* ?native_handle@critical_section@Concurrency@@QEAAAEAV12@XZ */
|
|
DEFINE_THISCALL_WRAPPER(critical_section_native_handle, 4)
|
|
critical_section* __thiscall critical_section_native_handle(critical_section *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
return this;
|
|
}
|
|
|
|
#if _MSVCR_VER >= 110
|
|
/* ?try_lock_for@critical_section@Concurrency@@QAE_NI@Z */
|
|
/* ?try_lock_for@critical_section@Concurrency@@QEAA_NI@Z */
|
|
DEFINE_THISCALL_WRAPPER(critical_section_try_lock_for, 8)
|
|
bool __thiscall critical_section_try_lock_for(
|
|
critical_section *this, unsigned int timeout)
|
|
{
|
|
cs_queue *q, *last;
|
|
|
|
TRACE("(%p %d)\n", this, timeout);
|
|
|
|
if(this->unk_thread_id == GetCurrentThreadId()) {
|
|
improper_lock e;
|
|
improper_lock_ctor_str(&e, "Already locked");
|
|
_CxxThrowException(&e, &improper_lock_exception_type);
|
|
}
|
|
|
|
if(!(q = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*q))))
|
|
return critical_section_try_lock(this);
|
|
|
|
last = InterlockedExchangePointer(&this->tail, q);
|
|
if(last) {
|
|
LARGE_INTEGER to;
|
|
NTSTATUS status;
|
|
FILETIME ft;
|
|
|
|
last->next = q;
|
|
GetSystemTimeAsFileTime(&ft);
|
|
to.QuadPart = ((LONGLONG)ft.dwHighDateTime<<32) +
|
|
ft.dwLowDateTime + (LONGLONG)timeout*10000;
|
|
status = NtWaitForKeyedEvent(keyed_event, q, 0, &to);
|
|
if(status == STATUS_TIMEOUT) {
|
|
if(!InterlockedExchange(&q->free, TRUE))
|
|
return FALSE;
|
|
/* A thread has signaled the event and is block waiting. */
|
|
/* We need to catch the event to wake the thread. */
|
|
NtWaitForKeyedEvent(keyed_event, q, 0, NULL);
|
|
}
|
|
}
|
|
|
|
cs_set_head(this, q);
|
|
if(InterlockedCompareExchangePointer(&this->tail, &this->unk_active, q) != q) {
|
|
spin_wait_for_next_cs(q);
|
|
this->unk_active.next = q->next;
|
|
}
|
|
|
|
HeapFree(GetProcessHeap(), 0, q);
|
|
return TRUE;
|
|
}
|
|
#endif
|
|
|
|
/* ??0scoped_lock@critical_section@Concurrency@@QAE@AAV12@@Z */
|
|
/* ??0scoped_lock@critical_section@Concurrency@@QEAA@AEAV12@@Z */
|
|
DEFINE_THISCALL_WRAPPER(critical_section_scoped_lock_ctor, 8)
|
|
critical_section_scoped_lock* __thiscall critical_section_scoped_lock_ctor(
|
|
critical_section_scoped_lock *this, critical_section *cs)
|
|
{
|
|
TRACE("(%p %p)\n", this, cs);
|
|
this->cs = cs;
|
|
cs_lock(this->cs, &this->lock.q);
|
|
return this;
|
|
}
|
|
|
|
/* ??1scoped_lock@critical_section@Concurrency@@QAE@XZ */
|
|
/* ??1scoped_lock@critical_section@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(critical_section_scoped_lock_dtor, 4)
|
|
void __thiscall critical_section_scoped_lock_dtor(critical_section_scoped_lock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
critical_section_unlock(this->cs);
|
|
}
|
|
|
|
/* ??0_NonReentrantPPLLock@details@Concurrency@@QAE@XZ */
|
|
/* ??0_NonReentrantPPLLock@details@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(_NonReentrantPPLLock_ctor, 4)
|
|
_NonReentrantPPLLock* __thiscall _NonReentrantPPLLock_ctor(_NonReentrantPPLLock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
critical_section_ctor(&this->cs);
|
|
return this;
|
|
}
|
|
|
|
/* ?_Acquire@_NonReentrantPPLLock@details@Concurrency@@QAEXPAX@Z */
|
|
/* ?_Acquire@_NonReentrantPPLLock@details@Concurrency@@QEAAXPEAX@Z */
|
|
DEFINE_THISCALL_WRAPPER(_NonReentrantPPLLock__Acquire, 8)
|
|
void __thiscall _NonReentrantPPLLock__Acquire(_NonReentrantPPLLock *this, cs_queue *q)
|
|
{
|
|
TRACE("(%p %p)\n", this, q);
|
|
cs_lock(&this->cs, q);
|
|
}
|
|
|
|
/* ?_Release@_NonReentrantPPLLock@details@Concurrency@@QAEXXZ */
|
|
/* ?_Release@_NonReentrantPPLLock@details@Concurrency@@QEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(_NonReentrantPPLLock__Release, 4)
|
|
void __thiscall _NonReentrantPPLLock__Release(_NonReentrantPPLLock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
critical_section_unlock(&this->cs);
|
|
}
|
|
|
|
/* ??0_Scoped_lock@_NonReentrantPPLLock@details@Concurrency@@QAE@AAV123@@Z */
|
|
/* ??0_Scoped_lock@_NonReentrantPPLLock@details@Concurrency@@QEAA@AEAV123@@Z */
|
|
DEFINE_THISCALL_WRAPPER(_NonReentrantPPLLock__Scoped_lock_ctor, 8)
|
|
_NonReentrantPPLLock__Scoped_lock* __thiscall _NonReentrantPPLLock__Scoped_lock_ctor(
|
|
_NonReentrantPPLLock__Scoped_lock *this, _NonReentrantPPLLock *lock)
|
|
{
|
|
TRACE("(%p %p)\n", this, lock);
|
|
|
|
this->lock = lock;
|
|
_NonReentrantPPLLock__Acquire(this->lock, &this->wait.q);
|
|
return this;
|
|
}
|
|
|
|
/* ??1_Scoped_lock@_NonReentrantPPLLock@details@Concurrency@@QAE@XZ */
|
|
/* ??1_Scoped_lock@_NonReentrantPPLLock@details@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(_NonReentrantPPLLock__Scoped_lock_dtor, 4)
|
|
void __thiscall _NonReentrantPPLLock__Scoped_lock_dtor(_NonReentrantPPLLock__Scoped_lock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
_NonReentrantPPLLock__Release(this->lock);
|
|
}
|
|
|
|
/* ??0_ReentrantPPLLock@details@Concurrency@@QAE@XZ */
|
|
/* ??0_ReentrantPPLLock@details@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(_ReentrantPPLLock_ctor, 4)
|
|
_ReentrantPPLLock* __thiscall _ReentrantPPLLock_ctor(_ReentrantPPLLock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
critical_section_ctor(&this->cs);
|
|
this->count = 0;
|
|
this->owner = -1;
|
|
return this;
|
|
}
|
|
|
|
/* ?_Acquire@_ReentrantPPLLock@details@Concurrency@@QAEXPAX@Z */
|
|
/* ?_Acquire@_ReentrantPPLLock@details@Concurrency@@QEAAXPEAX@Z */
|
|
DEFINE_THISCALL_WRAPPER(_ReentrantPPLLock__Acquire, 8)
|
|
void __thiscall _ReentrantPPLLock__Acquire(_ReentrantPPLLock *this, cs_queue *q)
|
|
{
|
|
TRACE("(%p %p)\n", this, q);
|
|
|
|
if(this->owner == GetCurrentThreadId()) {
|
|
this->count++;
|
|
return;
|
|
}
|
|
|
|
cs_lock(&this->cs, q);
|
|
this->count++;
|
|
this->owner = GetCurrentThreadId();
|
|
}
|
|
|
|
/* ?_Release@_ReentrantPPLLock@details@Concurrency@@QAEXXZ */
|
|
/* ?_Release@_ReentrantPPLLock@details@Concurrency@@QEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(_ReentrantPPLLock__Release, 4)
|
|
void __thiscall _ReentrantPPLLock__Release(_ReentrantPPLLock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
this->count--;
|
|
if(this->count)
|
|
return;
|
|
|
|
this->owner = -1;
|
|
critical_section_unlock(&this->cs);
|
|
}
|
|
|
|
/* ??0_Scoped_lock@_ReentrantPPLLock@details@Concurrency@@QAE@AAV123@@Z */
|
|
/* ??0_Scoped_lock@_ReentrantPPLLock@details@Concurrency@@QEAA@AEAV123@@Z */
|
|
DEFINE_THISCALL_WRAPPER(_ReentrantPPLLock__Scoped_lock_ctor, 8)
|
|
_ReentrantPPLLock__Scoped_lock* __thiscall _ReentrantPPLLock__Scoped_lock_ctor(
|
|
_ReentrantPPLLock__Scoped_lock *this, _ReentrantPPLLock *lock)
|
|
{
|
|
TRACE("(%p %p)\n", this, lock);
|
|
|
|
this->lock = lock;
|
|
_ReentrantPPLLock__Acquire(this->lock, &this->wait.q);
|
|
return this;
|
|
}
|
|
|
|
/* ??1_Scoped_lock@_ReentrantPPLLock@details@Concurrency@@QAE@XZ */
|
|
/* ??1_Scoped_lock@_ReentrantPPLLock@details@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(_ReentrantPPLLock__Scoped_lock_dtor, 4)
|
|
void __thiscall _ReentrantPPLLock__Scoped_lock_dtor(_ReentrantPPLLock__Scoped_lock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
_ReentrantPPLLock__Release(this->lock);
|
|
}
|
|
|
|
/* ?_GetConcurrency@details@Concurrency@@YAIXZ */
|
|
unsigned int __cdecl _GetConcurrency(void)
|
|
{
|
|
static unsigned int val = -1;
|
|
|
|
TRACE("()\n");
|
|
|
|
if(val == -1) {
|
|
SYSTEM_INFO si;
|
|
|
|
GetSystemInfo(&si);
|
|
val = si.dwNumberOfProcessors;
|
|
}
|
|
|
|
return val;
|
|
}
|
|
|
|
static inline PLARGE_INTEGER evt_timeout(PLARGE_INTEGER pTime, unsigned int timeout)
|
|
{
|
|
if(timeout == COOPERATIVE_TIMEOUT_INFINITE) return NULL;
|
|
pTime->QuadPart = (ULONGLONG)timeout * -10000;
|
|
return pTime;
|
|
}
|
|
|
|
static void evt_add_queue(thread_wait_entry **head, thread_wait_entry *entry)
|
|
{
|
|
entry->next = *head;
|
|
entry->prev = NULL;
|
|
if(*head) (*head)->prev = entry;
|
|
*head = entry;
|
|
}
|
|
|
|
static void evt_remove_queue(thread_wait_entry **head, thread_wait_entry *entry)
|
|
{
|
|
if(entry == *head)
|
|
*head = entry->next;
|
|
else if(entry->prev)
|
|
entry->prev->next = entry->next;
|
|
if(entry->next) entry->next->prev = entry->prev;
|
|
}
|
|
|
|
static size_t evt_end_wait(thread_wait *wait, event **events, int count)
|
|
{
|
|
size_t i, ret = COOPERATIVE_WAIT_TIMEOUT;
|
|
|
|
for(i = 0; i < count; i++) {
|
|
critical_section_lock(&events[i]->cs);
|
|
if(events[i] == wait->signaled) ret = i;
|
|
evt_remove_queue(&events[i]->waiters, &wait->entries[i]);
|
|
critical_section_unlock(&events[i]->cs);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static inline int evt_transition(void **state, void *from, void *to)
|
|
{
|
|
return InterlockedCompareExchangePointer(state, to, from) == from;
|
|
}
|
|
|
|
static size_t evt_wait(thread_wait *wait, event **events, int count, bool wait_all, unsigned int timeout)
|
|
{
|
|
int i;
|
|
NTSTATUS status;
|
|
LARGE_INTEGER ntto;
|
|
|
|
wait->signaled = EVT_RUNNING;
|
|
wait->pending_waits = wait_all ? count : 1;
|
|
for(i = 0; i < count; i++) {
|
|
wait->entries[i].wait = wait;
|
|
|
|
critical_section_lock(&events[i]->cs);
|
|
evt_add_queue(&events[i]->waiters, &wait->entries[i]);
|
|
if(events[i]->signaled) {
|
|
if(!InterlockedDecrement(&wait->pending_waits)) {
|
|
wait->signaled = events[i];
|
|
critical_section_unlock(&events[i]->cs);
|
|
|
|
return evt_end_wait(wait, events, i+1);
|
|
}
|
|
}
|
|
critical_section_unlock(&events[i]->cs);
|
|
}
|
|
|
|
if(!timeout)
|
|
return evt_end_wait(wait, events, count);
|
|
|
|
if(!evt_transition(&wait->signaled, EVT_RUNNING, EVT_WAITING))
|
|
return evt_end_wait(wait, events, count);
|
|
|
|
status = NtWaitForKeyedEvent(keyed_event, wait, 0, evt_timeout(&ntto, timeout));
|
|
|
|
if(status && !evt_transition(&wait->signaled, EVT_WAITING, EVT_RUNNING))
|
|
NtWaitForKeyedEvent(keyed_event, wait, 0, NULL);
|
|
|
|
return evt_end_wait(wait, events, count);
|
|
}
|
|
|
|
/* ??0event@Concurrency@@QAE@XZ */
|
|
/* ??0event@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(event_ctor, 4)
|
|
event* __thiscall event_ctor(event *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
this->waiters = NULL;
|
|
this->signaled = FALSE;
|
|
critical_section_ctor(&this->cs);
|
|
|
|
return this;
|
|
}
|
|
|
|
/* ??1event@Concurrency@@QAE@XZ */
|
|
/* ??1event@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(event_dtor, 4)
|
|
void __thiscall event_dtor(event *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
critical_section_dtor(&this->cs);
|
|
if(this->waiters)
|
|
ERR("there's a wait on destroyed event\n");
|
|
}
|
|
|
|
/* ?reset@event@Concurrency@@QAEXXZ */
|
|
/* ?reset@event@Concurrency@@QEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(event_reset, 4)
|
|
void __thiscall event_reset(event *this)
|
|
{
|
|
thread_wait_entry *entry;
|
|
|
|
TRACE("(%p)\n", this);
|
|
|
|
critical_section_lock(&this->cs);
|
|
if(this->signaled) {
|
|
this->signaled = FALSE;
|
|
for(entry=this->waiters; entry; entry = entry->next)
|
|
InterlockedIncrement(&entry->wait->pending_waits);
|
|
}
|
|
critical_section_unlock(&this->cs);
|
|
}
|
|
|
|
/* ?set@event@Concurrency@@QAEXXZ */
|
|
/* ?set@event@Concurrency@@QEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(event_set, 4)
|
|
void __thiscall event_set(event *this)
|
|
{
|
|
thread_wait_entry *wakeup = NULL;
|
|
thread_wait_entry *entry, *next;
|
|
|
|
TRACE("(%p)\n", this);
|
|
|
|
critical_section_lock(&this->cs);
|
|
if(!this->signaled) {
|
|
this->signaled = TRUE;
|
|
for(entry=this->waiters; entry; entry=next) {
|
|
next = entry->next;
|
|
if(!InterlockedDecrement(&entry->wait->pending_waits)) {
|
|
if(InterlockedExchangePointer(&entry->wait->signaled, this) == EVT_WAITING) {
|
|
evt_remove_queue(&this->waiters, entry);
|
|
evt_add_queue(&wakeup, entry);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
critical_section_unlock(&this->cs);
|
|
|
|
for(entry=wakeup; entry; entry=next) {
|
|
next = entry->next;
|
|
entry->next = entry->prev = NULL;
|
|
NtReleaseKeyedEvent(keyed_event, entry->wait, 0, NULL);
|
|
}
|
|
}
|
|
|
|
/* ?wait@event@Concurrency@@QAEII@Z */
|
|
/* ?wait@event@Concurrency@@QEAA_KI@Z */
|
|
DEFINE_THISCALL_WRAPPER(event_wait, 8)
|
|
size_t __thiscall event_wait(event *this, unsigned int timeout)
|
|
{
|
|
thread_wait wait;
|
|
size_t signaled;
|
|
|
|
TRACE("(%p %u)\n", this, timeout);
|
|
|
|
critical_section_lock(&this->cs);
|
|
signaled = this->signaled;
|
|
critical_section_unlock(&this->cs);
|
|
|
|
if(!timeout) return signaled ? 0 : COOPERATIVE_WAIT_TIMEOUT;
|
|
return signaled ? 0 : evt_wait(&wait, &this, 1, FALSE, timeout);
|
|
}
|
|
|
|
/* ?wait_for_multiple@event@Concurrency@@SAIPAPAV12@I_NI@Z */
|
|
/* ?wait_for_multiple@event@Concurrency@@SA_KPEAPEAV12@_K_NI@Z */
|
|
int __cdecl event_wait_for_multiple(event **events, size_t count, bool wait_all, unsigned int timeout)
|
|
{
|
|
thread_wait *wait;
|
|
size_t ret;
|
|
|
|
TRACE("(%p %Iu %d %u)\n", events, count, wait_all, timeout);
|
|
|
|
if(count == 0)
|
|
return 0;
|
|
|
|
wait = operator_new(FIELD_OFFSET(thread_wait, entries[count]));
|
|
ret = evt_wait(wait, events, count, wait_all, timeout);
|
|
operator_delete(wait);
|
|
|
|
return ret;
|
|
}
|
|
|
|
#if _MSVCR_VER >= 110
|
|
|
|
/* ??0_Condition_variable@details@Concurrency@@QAE@XZ */
|
|
/* ??0_Condition_variable@details@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(_Condition_variable_ctor, 4)
|
|
_Condition_variable* __thiscall _Condition_variable_ctor(_Condition_variable *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
this->queue = NULL;
|
|
critical_section_ctor(&this->lock);
|
|
return this;
|
|
}
|
|
|
|
/* ??1_Condition_variable@details@Concurrency@@QAE@XZ */
|
|
/* ??1_Condition_variable@details@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(_Condition_variable_dtor, 4)
|
|
void __thiscall _Condition_variable_dtor(_Condition_variable *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
while(this->queue) {
|
|
cv_queue *next = this->queue->next;
|
|
if(!this->queue->expired)
|
|
ERR("there's an active wait\n");
|
|
HeapFree(GetProcessHeap(), 0, this->queue);
|
|
this->queue = next;
|
|
}
|
|
critical_section_dtor(&this->lock);
|
|
}
|
|
|
|
/* ?wait@_Condition_variable@details@Concurrency@@QAEXAAVcritical_section@3@@Z */
|
|
/* ?wait@_Condition_variable@details@Concurrency@@QEAAXAEAVcritical_section@3@@Z */
|
|
DEFINE_THISCALL_WRAPPER(_Condition_variable_wait, 8)
|
|
void __thiscall _Condition_variable_wait(_Condition_variable *this, critical_section *cs)
|
|
{
|
|
cv_queue q;
|
|
|
|
TRACE("(%p, %p)\n", this, cs);
|
|
|
|
critical_section_lock(&this->lock);
|
|
q.next = this->queue;
|
|
q.expired = FALSE;
|
|
this->queue = &q;
|
|
critical_section_unlock(&this->lock);
|
|
|
|
critical_section_unlock(cs);
|
|
NtWaitForKeyedEvent(keyed_event, &q, 0, NULL);
|
|
critical_section_lock(cs);
|
|
}
|
|
|
|
/* ?wait_for@_Condition_variable@details@Concurrency@@QAE_NAAVcritical_section@3@I@Z */
|
|
/* ?wait_for@_Condition_variable@details@Concurrency@@QEAA_NAEAVcritical_section@3@I@Z */
|
|
DEFINE_THISCALL_WRAPPER(_Condition_variable_wait_for, 12)
|
|
bool __thiscall _Condition_variable_wait_for(_Condition_variable *this,
|
|
critical_section *cs, unsigned int timeout)
|
|
{
|
|
LARGE_INTEGER to;
|
|
NTSTATUS status;
|
|
FILETIME ft;
|
|
cv_queue *q;
|
|
|
|
TRACE("(%p %p %d)\n", this, cs, timeout);
|
|
|
|
q = operator_new(sizeof(cv_queue));
|
|
critical_section_lock(&this->lock);
|
|
q->next = this->queue;
|
|
q->expired = FALSE;
|
|
this->queue = q;
|
|
critical_section_unlock(&this->lock);
|
|
|
|
critical_section_unlock(cs);
|
|
|
|
GetSystemTimeAsFileTime(&ft);
|
|
to.QuadPart = ((LONGLONG)ft.dwHighDateTime << 32) +
|
|
ft.dwLowDateTime + (LONGLONG)timeout * 10000;
|
|
status = NtWaitForKeyedEvent(keyed_event, q, 0, &to);
|
|
if(status == STATUS_TIMEOUT) {
|
|
if(!InterlockedExchange(&q->expired, TRUE)) {
|
|
critical_section_lock(cs);
|
|
return FALSE;
|
|
}
|
|
else
|
|
NtWaitForKeyedEvent(keyed_event, q, 0, 0);
|
|
}
|
|
|
|
operator_delete(q);
|
|
critical_section_lock(cs);
|
|
return TRUE;
|
|
}
|
|
|
|
/* ?notify_one@_Condition_variable@details@Concurrency@@QAEXXZ */
|
|
/* ?notify_one@_Condition_variable@details@Concurrency@@QEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(_Condition_variable_notify_one, 4)
|
|
void __thiscall _Condition_variable_notify_one(_Condition_variable *this)
|
|
{
|
|
cv_queue *node;
|
|
|
|
TRACE("(%p)\n", this);
|
|
|
|
if(!this->queue)
|
|
return;
|
|
|
|
while(1) {
|
|
critical_section_lock(&this->lock);
|
|
node = this->queue;
|
|
if(!node) {
|
|
critical_section_unlock(&this->lock);
|
|
return;
|
|
}
|
|
this->queue = node->next;
|
|
critical_section_unlock(&this->lock);
|
|
|
|
if(!InterlockedExchange(&node->expired, TRUE)) {
|
|
NtReleaseKeyedEvent(keyed_event, node, 0, NULL);
|
|
return;
|
|
} else {
|
|
HeapFree(GetProcessHeap(), 0, node);
|
|
}
|
|
}
|
|
}
|
|
|
|
/* ?notify_all@_Condition_variable@details@Concurrency@@QAEXXZ */
|
|
/* ?notify_all@_Condition_variable@details@Concurrency@@QEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(_Condition_variable_notify_all, 4)
|
|
void __thiscall _Condition_variable_notify_all(_Condition_variable *this)
|
|
{
|
|
cv_queue *ptr;
|
|
|
|
TRACE("(%p)\n", this);
|
|
|
|
if(!this->queue)
|
|
return;
|
|
|
|
critical_section_lock(&this->lock);
|
|
ptr = this->queue;
|
|
this->queue = NULL;
|
|
critical_section_unlock(&this->lock);
|
|
|
|
while(ptr) {
|
|
cv_queue *next = ptr->next;
|
|
|
|
if(!InterlockedExchange(&ptr->expired, TRUE))
|
|
NtReleaseKeyedEvent(keyed_event, ptr, 0, NULL);
|
|
else
|
|
HeapFree(GetProcessHeap(), 0, ptr);
|
|
ptr = next;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
/* ??0reader_writer_lock@Concurrency@@QAE@XZ */
|
|
/* ??0reader_writer_lock@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(reader_writer_lock_ctor, 4)
|
|
reader_writer_lock* __thiscall reader_writer_lock_ctor(reader_writer_lock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
if (!keyed_event) {
|
|
HANDLE event;
|
|
|
|
NtCreateKeyedEvent(&event, GENERIC_READ|GENERIC_WRITE, NULL, 0);
|
|
if (InterlockedCompareExchangePointer(&keyed_event, event, NULL) != NULL)
|
|
NtClose(event);
|
|
}
|
|
|
|
memset(this, 0, sizeof(*this));
|
|
return this;
|
|
}
|
|
|
|
/* ??1reader_writer_lock@Concurrency@@QAE@XZ */
|
|
/* ??1reader_writer_lock@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(reader_writer_lock_dtor, 4)
|
|
void __thiscall reader_writer_lock_dtor(reader_writer_lock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
if (this->thread_id != 0 || this->count)
|
|
WARN("destroying locked reader_writer_lock\n");
|
|
}
|
|
|
|
static inline void spin_wait_for_next_rwl(rwl_queue *q)
|
|
{
|
|
SpinWait sw;
|
|
|
|
if(q->next) return;
|
|
|
|
SpinWait_ctor(&sw, &spin_wait_yield);
|
|
SpinWait__Reset(&sw);
|
|
while(!q->next)
|
|
SpinWait__SpinOnce(&sw);
|
|
SpinWait_dtor(&sw);
|
|
}
|
|
|
|
/* ?lock@reader_writer_lock@Concurrency@@QAEXXZ */
|
|
/* ?lock@reader_writer_lock@Concurrency@@QEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(reader_writer_lock_lock, 4)
|
|
void __thiscall reader_writer_lock_lock(reader_writer_lock *this)
|
|
{
|
|
rwl_queue q = { NULL }, *last;
|
|
|
|
TRACE("(%p)\n", this);
|
|
|
|
if (this->thread_id == GetCurrentThreadId()) {
|
|
improper_lock e;
|
|
improper_lock_ctor_str(&e, "Already locked");
|
|
_CxxThrowException(&e, &improper_lock_exception_type);
|
|
}
|
|
|
|
last = InterlockedExchangePointer((void**)&this->writer_tail, &q);
|
|
if (last) {
|
|
last->next = &q;
|
|
NtWaitForKeyedEvent(keyed_event, &q, 0, NULL);
|
|
} else {
|
|
this->writer_head = &q;
|
|
if (InterlockedOr(&this->count, WRITER_WAITING))
|
|
NtWaitForKeyedEvent(keyed_event, &q, 0, NULL);
|
|
}
|
|
|
|
this->thread_id = GetCurrentThreadId();
|
|
this->writer_head = &this->active;
|
|
this->active.next = NULL;
|
|
if (InterlockedCompareExchangePointer((void**)&this->writer_tail, &this->active, &q) != &q) {
|
|
spin_wait_for_next_rwl(&q);
|
|
this->active.next = q.next;
|
|
}
|
|
}
|
|
|
|
/* ?lock_read@reader_writer_lock@Concurrency@@QAEXXZ */
|
|
/* ?lock_read@reader_writer_lock@Concurrency@@QEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(reader_writer_lock_lock_read, 4)
|
|
void __thiscall reader_writer_lock_lock_read(reader_writer_lock *this)
|
|
{
|
|
rwl_queue q;
|
|
|
|
TRACE("(%p)\n", this);
|
|
|
|
if (this->thread_id == GetCurrentThreadId()) {
|
|
improper_lock e;
|
|
improper_lock_ctor_str(&e, "Already locked as writer");
|
|
_CxxThrowException(&e, &improper_lock_exception_type);
|
|
}
|
|
|
|
do {
|
|
q.next = this->reader_head;
|
|
} while(InterlockedCompareExchangePointer((void**)&this->reader_head, &q, q.next) != q.next);
|
|
|
|
if (!q.next) {
|
|
rwl_queue *head;
|
|
LONG count;
|
|
|
|
while (!((count = this->count) & WRITER_WAITING))
|
|
if (InterlockedCompareExchange(&this->count, count+1, count) == count) break;
|
|
|
|
if (count & WRITER_WAITING)
|
|
NtWaitForKeyedEvent(keyed_event, &q, 0, NULL);
|
|
|
|
head = InterlockedExchangePointer((void**)&this->reader_head, NULL);
|
|
while(head && head != &q) {
|
|
rwl_queue *next = head->next;
|
|
InterlockedIncrement(&this->count);
|
|
NtReleaseKeyedEvent(keyed_event, head, 0, NULL);
|
|
head = next;
|
|
}
|
|
} else {
|
|
NtWaitForKeyedEvent(keyed_event, &q, 0, NULL);
|
|
}
|
|
}
|
|
|
|
/* ?try_lock@reader_writer_lock@Concurrency@@QAE_NXZ */
|
|
/* ?try_lock@reader_writer_lock@Concurrency@@QEAA_NXZ */
|
|
DEFINE_THISCALL_WRAPPER(reader_writer_lock_try_lock, 4)
|
|
bool __thiscall reader_writer_lock_try_lock(reader_writer_lock *this)
|
|
{
|
|
rwl_queue q = { NULL };
|
|
|
|
TRACE("(%p)\n", this);
|
|
|
|
if (this->thread_id == GetCurrentThreadId())
|
|
return FALSE;
|
|
|
|
if (InterlockedCompareExchangePointer((void**)&this->writer_tail, &q, NULL))
|
|
return FALSE;
|
|
this->writer_head = &q;
|
|
if (!InterlockedCompareExchange(&this->count, WRITER_WAITING, 0)) {
|
|
this->thread_id = GetCurrentThreadId();
|
|
this->writer_head = &this->active;
|
|
this->active.next = NULL;
|
|
if (InterlockedCompareExchangePointer((void**)&this->writer_tail, &this->active, &q) != &q) {
|
|
spin_wait_for_next_rwl(&q);
|
|
this->active.next = q.next;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
if (InterlockedCompareExchangePointer((void**)&this->writer_tail, NULL, &q) == &q)
|
|
return FALSE;
|
|
spin_wait_for_next_rwl(&q);
|
|
this->writer_head = q.next;
|
|
if (!InterlockedOr(&this->count, WRITER_WAITING)) {
|
|
this->thread_id = GetCurrentThreadId();
|
|
this->writer_head = &this->active;
|
|
this->active.next = q.next;
|
|
return TRUE;
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
/* ?try_lock_read@reader_writer_lock@Concurrency@@QAE_NXZ */
|
|
/* ?try_lock_read@reader_writer_lock@Concurrency@@QEAA_NXZ */
|
|
DEFINE_THISCALL_WRAPPER(reader_writer_lock_try_lock_read, 4)
|
|
bool __thiscall reader_writer_lock_try_lock_read(reader_writer_lock *this)
|
|
{
|
|
LONG count;
|
|
|
|
TRACE("(%p)\n", this);
|
|
|
|
while (!((count = this->count) & WRITER_WAITING))
|
|
if (InterlockedCompareExchange(&this->count, count+1, count) == count) return TRUE;
|
|
return FALSE;
|
|
}
|
|
|
|
/* ?unlock@reader_writer_lock@Concurrency@@QAEXXZ */
|
|
/* ?unlock@reader_writer_lock@Concurrency@@QEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(reader_writer_lock_unlock, 4)
|
|
void __thiscall reader_writer_lock_unlock(reader_writer_lock *this)
|
|
{
|
|
LONG count;
|
|
rwl_queue *head, *next;
|
|
|
|
TRACE("(%p)\n", this);
|
|
|
|
if ((count = this->count) & ~WRITER_WAITING) {
|
|
count = InterlockedDecrement(&this->count);
|
|
if (count != WRITER_WAITING)
|
|
return;
|
|
NtReleaseKeyedEvent(keyed_event, this->writer_head, 0, NULL);
|
|
return;
|
|
}
|
|
|
|
this->thread_id = 0;
|
|
next = this->writer_head->next;
|
|
if (next) {
|
|
NtReleaseKeyedEvent(keyed_event, next, 0, NULL);
|
|
return;
|
|
}
|
|
InterlockedAnd(&this->count, ~WRITER_WAITING);
|
|
head = InterlockedExchangePointer((void**)&this->reader_head, NULL);
|
|
while (head) {
|
|
next = head->next;
|
|
InterlockedIncrement(&this->count);
|
|
NtReleaseKeyedEvent(keyed_event, head, 0, NULL);
|
|
head = next;
|
|
}
|
|
|
|
if (InterlockedCompareExchangePointer((void**)&this->writer_tail, NULL, this->writer_head) == this->writer_head)
|
|
return;
|
|
InterlockedOr(&this->count, WRITER_WAITING);
|
|
}
|
|
|
|
/* ??0scoped_lock@reader_writer_lock@Concurrency@@QAE@AAV12@@Z */
|
|
/* ??0scoped_lock@reader_writer_lock@Concurrency@@QEAA@AEAV12@@Z */
|
|
DEFINE_THISCALL_WRAPPER(reader_writer_lock_scoped_lock_ctor, 8)
|
|
reader_writer_lock_scoped_lock* __thiscall reader_writer_lock_scoped_lock_ctor(
|
|
reader_writer_lock_scoped_lock *this, reader_writer_lock *lock)
|
|
{
|
|
TRACE("(%p %p)\n", this, lock);
|
|
|
|
this->lock = lock;
|
|
reader_writer_lock_lock(lock);
|
|
return this;
|
|
}
|
|
|
|
/* ??1scoped_lock@reader_writer_lock@Concurrency@@QAE@XZ */
|
|
/* ??1scoped_lock@reader_writer_lock@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(reader_writer_lock_scoped_lock_dtor, 4)
|
|
void __thiscall reader_writer_lock_scoped_lock_dtor(reader_writer_lock_scoped_lock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
reader_writer_lock_unlock(this->lock);
|
|
}
|
|
|
|
/* ??0scoped_lock_read@reader_writer_lock@Concurrency@@QAE@AAV12@@Z */
|
|
/* ??0scoped_lock_read@reader_writer_lock@Concurrency@@QEAA@AEAV12@@Z */
|
|
DEFINE_THISCALL_WRAPPER(reader_writer_lock_scoped_lock_read_ctor, 8)
|
|
reader_writer_lock_scoped_lock* __thiscall reader_writer_lock_scoped_lock_read_ctor(
|
|
reader_writer_lock_scoped_lock *this, reader_writer_lock *lock)
|
|
{
|
|
TRACE("(%p %p)\n", this, lock);
|
|
|
|
this->lock = lock;
|
|
reader_writer_lock_lock_read(lock);
|
|
return this;
|
|
}
|
|
|
|
/* ??1scoped_lock_read@reader_writer_lock@Concurrency@@QAE@XZ */
|
|
/* ??1scoped_lock_read@reader_writer_lock@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(reader_writer_lock_scoped_lock_read_dtor, 4)
|
|
void __thiscall reader_writer_lock_scoped_lock_read_dtor(reader_writer_lock_scoped_lock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
reader_writer_lock_unlock(this->lock);
|
|
}
|
|
|
|
/* ??0_ReentrantBlockingLock@details@Concurrency@@QAE@XZ */
|
|
/* ??0_ReentrantBlockingLock@details@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(_ReentrantBlockingLock_ctor, 4)
|
|
_ReentrantBlockingLock* __thiscall _ReentrantBlockingLock_ctor(_ReentrantBlockingLock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
InitializeCriticalSection(&this->cs);
|
|
this->cs.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": _ReentrantBlockingLock");
|
|
return this;
|
|
}
|
|
|
|
/* ??1_ReentrantBlockingLock@details@Concurrency@@QAE@XZ */
|
|
/* ??1_ReentrantBlockingLock@details@Concurrency@@QEAA@XZ */
|
|
DEFINE_THISCALL_WRAPPER(_ReentrantBlockingLock_dtor, 4)
|
|
void __thiscall _ReentrantBlockingLock_dtor(_ReentrantBlockingLock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
|
|
this->cs.DebugInfo->Spare[0] = 0;
|
|
DeleteCriticalSection(&this->cs);
|
|
}
|
|
|
|
/* ?_Acquire@_ReentrantBlockingLock@details@Concurrency@@QAEXXZ */
|
|
/* ?_Acquire@_ReentrantBlockingLock@details@Concurrency@@QEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(_ReentrantBlockingLock__Acquire, 4)
|
|
void __thiscall _ReentrantBlockingLock__Acquire(_ReentrantBlockingLock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
EnterCriticalSection(&this->cs);
|
|
}
|
|
|
|
/* ?_Release@_ReentrantBlockingLock@details@Concurrency@@QAEXXZ */
|
|
/* ?_Release@_ReentrantBlockingLock@details@Concurrency@@QEAAXXZ */
|
|
DEFINE_THISCALL_WRAPPER(_ReentrantBlockingLock__Release, 4)
|
|
void __thiscall _ReentrantBlockingLock__Release(_ReentrantBlockingLock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
LeaveCriticalSection(&this->cs);
|
|
}
|
|
|
|
/* ?_TryAcquire@_ReentrantBlockingLock@details@Concurrency@@QAE_NXZ */
|
|
/* ?_TryAcquire@_ReentrantBlockingLock@details@Concurrency@@QEAA_NXZ */
|
|
DEFINE_THISCALL_WRAPPER(_ReentrantBlockingLock__TryAcquire, 4)
|
|
bool __thiscall _ReentrantBlockingLock__TryAcquire(_ReentrantBlockingLock *this)
|
|
{
|
|
TRACE("(%p)\n", this);
|
|
return TryEnterCriticalSection(&this->cs);
|
|
}
|
|
|
|
/* ?wait@Concurrency@@YAXI@Z */
|
|
void __cdecl Concurrency_wait(unsigned int time)
|
|
{
|
|
static int once;
|
|
|
|
if (!once++) FIXME("(%d) stub!\n", time);
|
|
|
|
Sleep(time);
|
|
}
|
|
|
|
#if _MSVCR_VER>=110
|
|
/* ?_Trace_agents@Concurrency@@YAXW4Agents_EventType@1@_JZZ */
|
|
void WINAPIV _Trace_agents(/*enum Concurrency::Agents_EventType*/int type, __int64 id, ...)
|
|
{
|
|
FIXME("(%d %s)\n", type, wine_dbgstr_longlong(id));
|
|
}
|
|
#endif
|
|
|
|
/* ?_Trace_ppl_function@Concurrency@@YAXABU_GUID@@EW4ConcRT_EventType@1@@Z */
|
|
/* ?_Trace_ppl_function@Concurrency@@YAXAEBU_GUID@@EW4ConcRT_EventType@1@@Z */
|
|
void __cdecl _Trace_ppl_function(const GUID *guid, unsigned char level, enum ConcRT_EventType type)
|
|
{
|
|
FIXME("(%s %u %i) stub\n", debugstr_guid(guid), level, type);
|
|
}
|
|
|
|
#ifdef __ASM_USE_THISCALL_WRAPPER
|
|
|
|
#define DEFINE_VTBL_WRAPPER(off) \
|
|
__ASM_GLOBAL_FUNC(vtbl_wrapper_ ## off, \
|
|
"popl %eax\n\t" \
|
|
"popl %ecx\n\t" \
|
|
"pushl %eax\n\t" \
|
|
"movl 0(%ecx), %eax\n\t" \
|
|
"jmp *" #off "(%eax)\n\t")
|
|
|
|
DEFINE_VTBL_WRAPPER(0);
|
|
DEFINE_VTBL_WRAPPER(4);
|
|
DEFINE_VTBL_WRAPPER(8);
|
|
DEFINE_VTBL_WRAPPER(12);
|
|
DEFINE_VTBL_WRAPPER(16);
|
|
DEFINE_VTBL_WRAPPER(20);
|
|
DEFINE_VTBL_WRAPPER(24);
|
|
DEFINE_VTBL_WRAPPER(28);
|
|
DEFINE_VTBL_WRAPPER(32);
|
|
DEFINE_VTBL_WRAPPER(36);
|
|
DEFINE_VTBL_WRAPPER(40);
|
|
DEFINE_VTBL_WRAPPER(44);
|
|
DEFINE_VTBL_WRAPPER(48);
|
|
|
|
#endif
|
|
|
|
DEFINE_RTTI_DATA0(Context, 0, ".?AVContext@Concurrency@@")
|
|
DEFINE_RTTI_DATA1(ContextBase, 0, &Context_rtti_base_descriptor, ".?AVContextBase@details@Concurrency@@")
|
|
DEFINE_RTTI_DATA2(ExternalContextBase, 0, &ContextBase_rtti_base_descriptor,
|
|
&Context_rtti_base_descriptor, ".?AVExternalContextBase@details@Concurrency@@")
|
|
DEFINE_RTTI_DATA0(Scheduler, 0, ".?AVScheduler@Concurrency@@")
|
|
DEFINE_RTTI_DATA1(SchedulerBase, 0, &Scheduler_rtti_base_descriptor, ".?AVSchedulerBase@details@Concurrency@@")
|
|
DEFINE_RTTI_DATA2(ThreadScheduler, 0, &SchedulerBase_rtti_base_descriptor,
|
|
&Scheduler_rtti_base_descriptor, ".?AVThreadScheduler@details@Concurrency@@")
|
|
|
|
__ASM_BLOCK_BEGIN(concurrency_vtables)
|
|
__ASM_VTABLE(ExternalContextBase,
|
|
VTABLE_ADD_FUNC(ExternalContextBase_GetId)
|
|
VTABLE_ADD_FUNC(ExternalContextBase_GetVirtualProcessorId)
|
|
VTABLE_ADD_FUNC(ExternalContextBase_GetScheduleGroupId)
|
|
VTABLE_ADD_FUNC(ExternalContextBase_Unblock)
|
|
VTABLE_ADD_FUNC(ExternalContextBase_IsSynchronouslyBlocked)
|
|
VTABLE_ADD_FUNC(ExternalContextBase_vector_dtor));
|
|
__ASM_VTABLE(ThreadScheduler,
|
|
VTABLE_ADD_FUNC(ThreadScheduler_vector_dtor)
|
|
VTABLE_ADD_FUNC(ThreadScheduler_Id)
|
|
VTABLE_ADD_FUNC(ThreadScheduler_GetNumberOfVirtualProcessors)
|
|
VTABLE_ADD_FUNC(ThreadScheduler_GetPolicy)
|
|
VTABLE_ADD_FUNC(ThreadScheduler_Reference)
|
|
VTABLE_ADD_FUNC(ThreadScheduler_Release)
|
|
VTABLE_ADD_FUNC(ThreadScheduler_RegisterShutdownEvent)
|
|
VTABLE_ADD_FUNC(ThreadScheduler_Attach)
|
|
#if _MSVCR_VER > 100
|
|
VTABLE_ADD_FUNC(ThreadScheduler_CreateScheduleGroup_loc)
|
|
#endif
|
|
VTABLE_ADD_FUNC(ThreadScheduler_CreateScheduleGroup)
|
|
#if _MSVCR_VER > 100
|
|
VTABLE_ADD_FUNC(ThreadScheduler_ScheduleTask_loc)
|
|
#endif
|
|
VTABLE_ADD_FUNC(ThreadScheduler_ScheduleTask)
|
|
#if _MSVCR_VER > 100
|
|
VTABLE_ADD_FUNC(ThreadScheduler_IsAvailableLocation)
|
|
#endif
|
|
);
|
|
__ASM_BLOCK_END
|
|
|
|
void msvcrt_init_concurrency(void *base)
|
|
{
|
|
#ifdef __x86_64__
|
|
init_cexception_rtti(base);
|
|
init_improper_lock_rtti(base);
|
|
init_improper_scheduler_attach_rtti(base);
|
|
init_improper_scheduler_detach_rtti(base);
|
|
init_invalid_scheduler_policy_key_rtti(base);
|
|
init_invalid_scheduler_policy_thread_specification_rtti(base);
|
|
init_invalid_scheduler_policy_value_rtti(base);
|
|
init_scheduler_resource_allocation_error_rtti(base);
|
|
init_Context_rtti(base);
|
|
init_ContextBase_rtti(base);
|
|
init_ExternalContextBase_rtti(base);
|
|
init_Scheduler_rtti(base);
|
|
init_SchedulerBase_rtti(base);
|
|
init_ThreadScheduler_rtti(base);
|
|
|
|
init_cexception_cxx_type_info(base);
|
|
init_improper_lock_cxx(base);
|
|
init_improper_scheduler_attach_cxx(base);
|
|
init_improper_scheduler_detach_cxx(base);
|
|
init_invalid_scheduler_policy_key_cxx(base);
|
|
init_invalid_scheduler_policy_thread_specification_cxx(base);
|
|
init_invalid_scheduler_policy_value_cxx(base);
|
|
init_scheduler_resource_allocation_error_cxx(base);
|
|
#endif
|
|
}
|
|
|
|
void msvcrt_free_concurrency(void)
|
|
{
|
|
if (context_tls_index != TLS_OUT_OF_INDEXES)
|
|
TlsFree(context_tls_index);
|
|
if(default_scheduler_policy.policy_container)
|
|
SchedulerPolicy_dtor(&default_scheduler_policy);
|
|
if(default_scheduler) {
|
|
ThreadScheduler_dtor(default_scheduler);
|
|
operator_delete(default_scheduler);
|
|
}
|
|
|
|
if(keyed_event)
|
|
NtClose(keyed_event);
|
|
}
|
|
|
|
void msvcrt_free_scheduler_thread(void)
|
|
{
|
|
Context *context = try_get_current_context();
|
|
if (!context) return;
|
|
call_Context_dtor(context, 1);
|
|
}
|
|
|
|
#endif /* _MSVCR_VER >= 100 */
|