remove a race from the hashing write()

Calling load() and store() separately leaves a lot of opportunity for a
race to occur.

This changes the logic to use compare_exchange() and will only return if
the previous value is absolutely not 1 at the point of comparison.

I also don't think this needs to be SeqCst, so relaxing that requirement
a little.
This commit is contained in:
Gary Pennington 2023-06-14 15:23:12 +01:00
parent 517f219066
commit 785be453f6

View File

@ -50,14 +50,17 @@ impl HokmaLock {
pub fn write(&'static self) -> WhenTheHokmaSuppression {
loop {
let previous = self.lock.load(Ordering::SeqCst);
self.lock.store(1, Ordering::SeqCst);
if previous != 1 {
return WhenTheHokmaSuppression {
hokma: self,
state: previous,
};
// We are only interested in error results
if let Err(previous) = self
.lock
.compare_exchange(1, 1, Ordering::Acquire, Ordering::Relaxed)
{
if previous != 1 {
return WhenTheHokmaSuppression {
hokma: self,
state: previous,
};
}
}
}
}