2020-01-23 21:04:46 +01:00
|
|
|
// Copyright (c) 2019-2020 Alexander Medvednikov. All rights reserved.
|
2019-06-23 04:21:30 +02:00
|
|
|
// Use of this source code is governed by an MIT license
|
|
|
|
// that can be found in the LICENSE file.
|
2019-06-22 21:53:22 +02:00
|
|
|
module sync
|
2019-07-24 17:36:20 +02:00
|
|
|
|
2020-06-22 19:23:42 +02:00
|
|
|
// TODO: The suggestion of using CriticalSection instead of mutex
|
|
|
|
// was discussed. Needs consideration.
|
|
|
|
|
2019-07-24 17:36:20 +02:00
|
|
|
// Mutex HANDLE
|
|
|
|
type MHANDLE voidptr
|
2019-06-22 21:53:22 +02:00
|
|
|
|
2019-10-25 16:24:40 +02:00
|
|
|
//[init_with=new_mutex] // TODO: implement support for this struct attribute, and disallow Mutex{} from outside the sync.new_mutex() function.
|
2020-05-10 07:58:54 +02:00
|
|
|
|
|
|
|
[ref_only]
|
2019-10-24 12:19:27 +02:00
|
|
|
pub struct Mutex {
|
2019-07-24 17:36:20 +02:00
|
|
|
mut:
|
2019-08-29 10:48:03 +02:00
|
|
|
mx MHANDLE // mutex handle
|
|
|
|
state MutexState // mutex state
|
|
|
|
cycle_wait i64 // waiting cycles (implemented only with atomic)
|
|
|
|
cycle_woken i64 // woken cycles ^
|
|
|
|
reader_sem u32 // reader semarphone
|
|
|
|
writer_sem u32 // writer semaphones
|
2019-06-22 21:53:22 +02:00
|
|
|
}
|
|
|
|
|
2019-08-29 10:48:03 +02:00
|
|
|
enum MutexState {
|
|
|
|
broken
|
|
|
|
waiting
|
|
|
|
released
|
|
|
|
abandoned
|
|
|
|
destroyed
|
|
|
|
}
|
2019-06-22 21:53:22 +02:00
|
|
|
|
2020-01-19 20:32:22 +01:00
|
|
|
pub fn new_mutex() &Mutex {
|
|
|
|
sm := &Mutex{}
|
2019-10-25 16:24:40 +02:00
|
|
|
unsafe {
|
2020-04-02 20:58:07 +02:00
|
|
|
mut m := sm
|
2020-05-27 05:42:48 +02:00
|
|
|
m.mx = MHANDLE(C.CreateMutex(0, false, 0))
|
2019-10-25 16:24:40 +02:00
|
|
|
if isnil(m.mx) {
|
|
|
|
m.state = .broken // handle broken and mutex state are broken
|
|
|
|
return sm
|
2020-04-02 20:58:07 +02:00
|
|
|
}
|
2019-10-25 16:24:40 +02:00
|
|
|
}
|
2020-01-19 20:32:22 +01:00
|
|
|
return sm
|
2019-10-25 16:24:40 +02:00
|
|
|
}
|
|
|
|
|
2020-05-17 13:51:18 +02:00
|
|
|
pub fn (mut m Mutex) lock() {
|
2019-08-29 10:48:03 +02:00
|
|
|
// if mutex handle not initalized
|
|
|
|
if isnil(m.mx) {
|
2020-05-27 05:42:48 +02:00
|
|
|
m.mx = MHANDLE(C.CreateMutex(0, false, 0))
|
2019-08-29 10:48:03 +02:00
|
|
|
if isnil(m.mx) {
|
|
|
|
m.state = .broken // handle broken and mutex state are broken
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
2020-04-02 20:58:07 +02:00
|
|
|
state := C.WaitForSingleObject(m.mx, C.INFINITE) // infinite wait
|
2019-10-30 15:29:05 +01:00
|
|
|
/* TODO fix match/enum combo
|
2019-08-29 10:48:03 +02:00
|
|
|
m.state = match state {
|
2020-04-02 20:58:07 +02:00
|
|
|
C.WAIT_ABANDONED { .abandoned }
|
|
|
|
C.WAIT_OBJECT_0 { .waiting }
|
2019-10-30 14:43:40 +01:00
|
|
|
else { .broken }
|
2019-08-29 10:48:03 +02:00
|
|
|
}
|
2019-10-30 15:29:05 +01:00
|
|
|
*/
|
2020-04-02 20:58:07 +02:00
|
|
|
if state == C.WAIT_ABANDONED {
|
|
|
|
m.state = .abandoned
|
|
|
|
// FIXME Use C constant instead
|
|
|
|
} else if state == 0 /* C.WAIT_OBJECT_0 */ {
|
|
|
|
m.state = .waiting
|
|
|
|
} else {
|
|
|
|
m.state = .broken
|
|
|
|
}
|
2019-06-22 21:53:22 +02:00
|
|
|
}
|
|
|
|
|
2020-05-17 13:51:18 +02:00
|
|
|
pub fn (mut m Mutex) unlock() {
|
2019-08-29 10:48:03 +02:00
|
|
|
if m.state == .waiting {
|
2020-04-02 20:58:07 +02:00
|
|
|
if C.ReleaseMutex(m.mx) {
|
2019-08-29 10:48:03 +02:00
|
|
|
m.state = .broken
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
m.state = .released
|
2019-07-25 06:25:25 +02:00
|
|
|
}
|
2019-07-24 17:36:20 +02:00
|
|
|
|
2020-05-17 13:51:18 +02:00
|
|
|
pub fn (mut m Mutex) destroy() {
|
2019-08-29 10:48:03 +02:00
|
|
|
if m.state == .waiting {
|
|
|
|
m.unlock() // unlock mutex before destroying
|
|
|
|
}
|
|
|
|
C.CloseHandle(m.mx) // destroy mutex
|
|
|
|
m.state = .destroyed // setting up reference to invalid state
|
2019-07-25 06:25:25 +02:00
|
|
|
}
|