use volatile load/store for mutex state

Author: soypat

src/sync/mutex.go

@@ -3,37 +3,39 @@ package sync
 import (
 	"internal/task"
 	_ "unsafe"
+
+	"github.com/tinygo-org/tinygo/src/runtime/volatile"
 )
 
 type Mutex struct {
-	locked  bool
+	state   uint8 // Set to non-zero if locked.
 	blocked task.Stack
 }
 
 //go:linkname scheduleTask runtime.runqueuePushBack
 func scheduleTask(*task.Task)
 
 func (m *Mutex) Lock() {
-	if m.locked {
+	if m.islocked() {
 		// Push self onto stack of blocked tasks, and wait to be resumed.
 		m.blocked.Push(task.Current())
 		task.Pause()
 		return
 	}
 
-	m.locked = true
+	m.setlock(true)
 }
 
 func (m *Mutex) Unlock() {
-	if !m.locked {
+	if !m.islocked() {
 		panic("sync: unlock of unlocked Mutex")
 	}
 
 	// Wake up a blocked task, if applicable.
 	if t := m.blocked.Pop(); t != nil {
 		scheduleTask(t)
 	} else {
-		m.locked = false
+		m.setlock(false)
 	}
 }
 
@@ -43,13 +45,28 @@ func (m *Mutex) Unlock() {
 // and use of TryLock is often a sign of a deeper problem
 // in a particular use of mutexes.
 func (m *Mutex) TryLock() bool {
-	if m.locked {
+	if m.islocked() {
 		return false
 	}
 	m.Lock()
 	return true
 }
 
+func (m *Mutex) islocked() bool {
+	return volatile.LoadUint8(&m.state) != 0
+}
+
+func (m *Mutex) setlock(b bool) {
+	volatile.StoreUint8(&m.state, boolToU8(b))
+}
+
+func boolToU8(b bool) uint8 {
+	if b {
+		return 1
+	}
+	return 0
+}
+
 type RWMutex struct {
 	// waitingWriters are all of the tasks waiting for write locks.
 	waitingWriters task.Stack