Skip to content

Commit dc89b21

Browse files
authored
[std/locks]close #7998(complete condition variables) (#17711)
* close #7998 * workaround genode * Update lib/system/syslocks.nim
1 parent cedbc70 commit dc89b21

File tree

5 files changed

+60
-45
lines changed

5 files changed

+60
-45
lines changed

lib/core/locks.nim

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,13 +66,18 @@ proc deinitCond*(cond: var Cond) {.inline.} =
6666
deinitSysCond(cond)
6767

6868
proc wait*(cond: var Cond, lock: var Lock) {.inline.} =
69-
## waits on the condition variable `cond`.
69+
## Waits on the condition variable `cond`.
7070
waitSysCond(cond, lock)
7171

7272
proc signal*(cond: var Cond) {.inline.} =
73-
## sends a signal to the condition variable `cond`.
73+
## Sends a signal to the condition variable `cond`.
7474
signalSysCond(cond)
7575

76+
proc broadcast*(cond: var Cond) {.inline.} =
77+
## Unblocks all threads currently blocked on the
78+
## specified condition variable `cond`.
79+
broadcastSysCond(cond)
80+
7681
template withLock*(a: Lock, body: untyped) =
7782
## Acquires the given lock, executes the statements in body and
7883
## releases the lock after the statements finish executing.

lib/genode_cpp/syslocks.h

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,11 @@ struct Nim::SysCond
7171
{
7272
_semaphore.up();
7373
}
74+
75+
void broadcastSysCond()
76+
{
77+
_semaphore.up();
78+
}
7479
};
7580

7681
#endif

lib/system/syslocks.nim

Lines changed: 32 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,9 @@ when defined(windows):
2424
LockSemaphore: int
2525
SpinCount: int
2626

27-
SysCond = Handle
27+
RTL_CONDITION_VARIABLE {.importc: "RTL_CONDITION_VARIABLE", header: "synchapi.h".} = object
28+
thePtr {.importc: "ptr".} : Handle
29+
SysCond = RTL_CONDITION_VARIABLE
2830

2931
proc initSysLock(L: var SysLock) {.importc: "InitializeCriticalSection",
3032
header: "<windows.h>".}
@@ -48,30 +50,29 @@ when defined(windows):
4850
proc deinitSys(L: var SysLock) {.importc: "DeleteCriticalSection",
4951
header: "<windows.h>".}
5052

51-
proc createEvent(lpEventAttributes: pointer,
52-
bManualReset, bInitialState: int32,
53-
lpName: cstring): SysCond {.stdcall, noSideEffect,
54-
dynlib: "kernel32", importc: "CreateEventA".}
53+
proc initializeConditionVariable(
54+
conditionVariable: var SysCond
55+
) {.stdcall, noSideEffect, dynlib: "kernel32", importc: "InitializeConditionVariable".}
5556

56-
proc closeHandle(hObject: Handle) {.stdcall, noSideEffect,
57-
dynlib: "kernel32", importc: "CloseHandle".}
58-
proc waitForSingleObject(hHandle: Handle, dwMilliseconds: int32): int32 {.
59-
stdcall, dynlib: "kernel32", importc: "WaitForSingleObject", noSideEffect.}
57+
proc sleepConditionVariableCS(
58+
conditionVariable: var SysCond,
59+
PCRITICAL_SECTION: var SysLock,
60+
dwMilliseconds: int
61+
): int32 {.stdcall, noSideEffect, dynlib: "kernel32", importc: "SleepConditionVariableCS".}
6062

61-
proc signalSysCond(hEvent: SysCond) {.stdcall, noSideEffect,
62-
dynlib: "kernel32", importc: "SetEvent".}
63+
64+
proc signalSysCond(hEvent: var SysCond) {.stdcall, noSideEffect,
65+
dynlib: "kernel32", importc: "WakeConditionVariable".}
66+
67+
proc broadcastSysCond(hEvent: var SysCond) {.stdcall, noSideEffect,
68+
dynlib: "kernel32", importc: "WakeAllConditionVariable".}
6369

6470
proc initSysCond(cond: var SysCond) {.inline.} =
65-
cond = createEvent(nil, 0'i32, 0'i32, nil)
71+
initializeConditionVariable(cond)
6672
proc deinitSysCond(cond: var SysCond) {.inline.} =
67-
closeHandle(cond)
73+
discard
6874
proc waitSysCond(cond: var SysCond, lock: var SysLock) =
69-
releaseSys(lock)
70-
discard waitForSingleObject(cond, -1'i32)
71-
acquireSys(lock)
72-
73-
proc waitSysCondWindows(cond: var SysCond) =
74-
discard waitForSingleObject(cond, -1'i32)
75+
discard sleepConditionVariableCS(cond, lock, -1'i32)
7576

7677
elif defined(genode):
7778
const
@@ -94,6 +95,8 @@ elif defined(genode):
9495
noSideEffect, importcpp.}
9596
proc signalSysCond(cond: var SysCond) {.
9697
noSideEffect, importcpp.}
98+
proc broadcastSysCond(cond: var SysCond) {.
99+
noSideEffect, importcpp.}
97100

98101
else:
99102
type
@@ -181,7 +184,7 @@ else:
181184
releaseSysAux(L)
182185

183186
when insideRLocksModule:
184-
let SysLockType_Reentrant{.importc: "PTHREAD_MUTEX_RECURSIVE",
187+
let SysLockType_Reentrant {.importc: "PTHREAD_MUTEX_RECURSIVE",
185188
header: "<pthread.h>".}: SysLockType
186189
proc initSysLockAttr(a: var SysLockAttr) {.
187190
importc: "pthread_mutexattr_init", header: "<pthread.h>", noSideEffect.}
@@ -194,10 +197,12 @@ else:
194197
proc deinitSysCondAux(cond: var SysCondObj) {.noSideEffect,
195198
importc: "pthread_cond_destroy", header: "<pthread.h>".}
196199

197-
proc waitSysCondAux(cond: var SysCondObj, lock: var SysLockObj) {.
200+
proc waitSysCondAux(cond: var SysCondObj, lock: var SysLockObj): cint {.
198201
importc: "pthread_cond_wait", header: "<pthread.h>", noSideEffect.}
199202
proc signalSysCondAux(cond: var SysCondObj) {.
200203
importc: "pthread_cond_signal", header: "<pthread.h>", noSideEffect.}
204+
proc broadcastSysCondAux(cond: var SysCondObj) {.
205+
importc: "pthread_cond_broadcast", header: "<pthread.h>", noSideEffect.}
201206

202207
when defined(ios):
203208
proc initSysCond(cond: var SysCond, cond_attr: ptr SysCondAttr = nil) =
@@ -209,18 +214,22 @@ else:
209214
c_free(cond)
210215

211216
template waitSysCond(cond: var SysCond, lock: var SysLock) =
212-
waitSysCondAux(cond[], lock[])
217+
discard waitSysCondAux(cond[], lock[])
213218
template signalSysCond(cond: var SysCond) =
214219
signalSysCondAux(cond[])
220+
template broadcastSysCond(cond: var SysCond) =
221+
broadcastSysCondAux(cond[])
215222
else:
216223
template initSysCond(cond: var SysCond, cond_attr: ptr SysCondAttr = nil) =
217224
initSysCondAux(cond, cond_attr)
218225
template deinitSysCond(cond: var SysCond) =
219226
deinitSysCondAux(cond)
220227

221228
template waitSysCond(cond: var SysCond, lock: var SysLock) =
222-
waitSysCondAux(cond, lock)
229+
discard waitSysCondAux(cond, lock)
223230
template signalSysCond(cond: var SysCond) =
224231
signalSysCondAux(cond)
232+
template broadcastSysCond(cond: var SysCond) =
233+
broadcastSysCondAux(cond)
225234

226235
{.pop.}

lib/system/sysspawn.nim

Lines changed: 13 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -20,34 +20,28 @@ when not declared(NimString):
2020
type
2121
CondVar = object
2222
c: SysCond
23-
when defined(posix):
24-
stupidLock: SysLock
25-
counter: int
23+
stupidLock: SysLock
24+
counter: int
2625

2726
proc createCondVar(): CondVar =
2827
initSysCond(result.c)
29-
when defined(posix):
30-
initSysLock(result.stupidLock)
31-
#acquireSys(result.stupidLock)
28+
initSysLock(result.stupidLock)
29+
#acquireSys(result.stupidLock)
3230

3331
proc destroyCondVar(c: var CondVar) {.inline.} =
3432
deinitSysCond(c.c)
3533

3634
proc await(cv: var CondVar) =
37-
when defined(posix):
38-
acquireSys(cv.stupidLock)
39-
while cv.counter <= 0:
40-
waitSysCond(cv.c, cv.stupidLock)
41-
dec cv.counter
42-
releaseSys(cv.stupidLock)
43-
else:
44-
waitSysCondWindows(cv.c)
35+
acquireSys(cv.stupidLock)
36+
while cv.counter <= 0:
37+
waitSysCond(cv.c, cv.stupidLock)
38+
dec cv.counter
39+
releaseSys(cv.stupidLock)
4540

4641
proc signal(cv: var CondVar) =
47-
when defined(posix):
48-
acquireSys(cv.stupidLock)
49-
inc cv.counter
50-
releaseSys(cv.stupidLock)
42+
acquireSys(cv.stupidLock)
43+
inc cv.counter
44+
releaseSys(cv.stupidLock)
5145
signalSysCond(cv.c)
5246

5347
type
@@ -57,8 +51,7 @@ type
5751

5852
proc createFastCondVar(): FastCondVar =
5953
initSysCond(result.slow.c)
60-
when defined(posix):
61-
initSysLock(result.slow.stupidLock)
54+
initSysLock(result.slow.stupidLock)
6255
#acquireSys(result.slow.stupidLock)
6356
result.event = false
6457
result.slowPath = false

lib/system/threadlocalstorage.nim

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,9 @@ when defined(windows):
1717
proc winResumeThread(hThread: SysThread): int32 {.
1818
stdcall, dynlib: "kernel32", importc: "ResumeThread".}
1919

20+
proc waitForSingleObject(hHandle: SysThread, dwMilliseconds: int32): int32 {.
21+
stdcall, dynlib: "kernel32", importc: "WaitForSingleObject".}
22+
2023
proc waitForMultipleObjects(nCount: int32,
2124
lpHandles: ptr SysThread,
2225
bWaitAll: int32,

0 commit comments

Comments
 (0)