Merge pull request #415 from mighe/dev

Fixed raspi adaptor race conditions

platforms/raspi/raspi_adaptor.go

@@ -11,6 +11,7 @@ import (
 	"gobot.io/x/gobot"
 	"gobot.io/x/gobot/drivers/i2c"
 	"gobot.io/x/gobot/sysfs"
+	"sync"
 )
 
 var readFile = func() ([]byte, error) {
@@ -19,6 +20,7 @@ var readFile = func() ([]byte, error) {
 
 // Adaptor is the Gobot Adaptor for the Raspberry Pi
 type Adaptor struct {
+	mutex         *sync.Mutex
 	name          string
 	revision      string
 	digitalPins   map[int]*sysfs.DigitalPin
@@ -30,6 +32,7 @@ type Adaptor struct {
 // NewAdaptor creates a Raspi Adaptor
 func NewAdaptor() *Adaptor {
 	r := &Adaptor{
+		mutex:       &sync.Mutex{},
 		name:        gobot.DefaultName("RaspberryPi"),
 		digitalPins: make(map[int]*sysfs.DigitalPin),
 		pwmPins:     make(map[int]*PWMPin),
@@ -55,10 +58,20 @@ func NewAdaptor() *Adaptor {
 }
 
 // Name returns the Adaptor's name
-func (r *Adaptor) Name() string { return r.name }
+func (r *Adaptor) Name() string {
+	r.mutex.Lock()
+	defer r.mutex.Unlock()
+
+	return r.name
+}
 
 // SetName sets the Adaptor's name
-func (r *Adaptor) SetName(n string) { r.name = n }
+func (r *Adaptor) SetName(n string) {
+	r.mutex.Lock()
+	defer r.mutex.Unlock()
+
+	r.name = n
+}
 
 // Connect starts connection with board and creates
 // digitalPins and pwmPins adaptor maps
@@ -68,6 +81,9 @@ func (r *Adaptor) Connect() (err error) {
 
 // Finalize closes connection to board and pins
 func (r *Adaptor) Finalize() (err error) {
+	r.mutex.Lock()
+	defer r.mutex.Unlock()
+
 	for _, pin := range r.digitalPins {
 		if pin != nil {
 			if perr := pin.Unexport(); err != nil {
@@ -100,18 +116,31 @@ func (r *Adaptor) DigitalPin(pin string, dir string) (sysfsPin sysfs.DigitalPinn
 		return
 	}
 
-	if r.digitalPins[i] == nil {
-		r.digitalPins[i] = sysfs.NewDigitalPin(i)
-		if err = r.digitalPins[i].Export(); err != nil {
+	currentPin, err := r.getExportedDigitalPin(i, dir)
+
+	if err != nil {
+		return
+	}
+
+	if err = currentPin.Direction(dir); err != nil {
+		return
+	}
+
+	return currentPin, nil
+}
+
+func (r *Adaptor) getExportedDigitalPin(translatedPin int, dir string) (sysfsPin sysfs.DigitalPinner, err error) {
+	r.mutex.Lock()
+	defer r.mutex.Unlock()
+
+	if r.digitalPins[translatedPin] == nil {
+		r.digitalPins[translatedPin] = sysfs.NewDigitalPin(translatedPin)
+		if err = r.digitalPins[translatedPin].Export(); err != nil {
 			return
 		}
 	}
 
-	if err = r.digitalPins[i].Direction(dir); err != nil {
-		return
-	}
-
-	return r.digitalPins[i], nil
+	return r.digitalPins[translatedPin], nil
 }
 
 // DigitalRead reads digital value from pin
@@ -138,10 +167,21 @@ func (r *Adaptor) GetConnection(address int, bus int) (connection i2c.Connection
 	if (bus < 0) || (bus > 1) {
 		return nil, fmt.Errorf("Bus number %d out of range", bus)
 	}
+
+	device, err := r.getI2cBus(bus)
+
+	return i2c.NewConnection(device, address), err
+}
+
+func (r *Adaptor) getI2cBus(bus int) (_ sysfs.I2cDevice, err error) {
+	r.mutex.Lock()
+	defer r.mutex.Unlock()
+
 	if r.i2cBuses[bus] == nil {
 		r.i2cBuses[bus], err = sysfs.NewI2cDevice(fmt.Sprintf("/dev/i2c-%d", bus))
 	}
-	return i2c.NewConnection(r.i2cBuses[bus], address), err
+
+	return r.i2cBuses[bus], err
 }
 
 // GetDefaultBus returns the default i2c bus for this platform
@@ -156,6 +196,9 @@ func (r *Adaptor) PWMPin(pin string) (raspiPWMPin sysfs.PWMPinner, err error) {
 		return
 	}
 
+	r.mutex.Lock()
+	defer r.mutex.Unlock()
+
 	if r.pwmPins[i] == nil {
 		r.pwmPins[i] = NewPWMPin(strconv.Itoa(i))
 	}

platforms/raspi/raspi_adaptor_test.go

@@ -10,6 +10,9 @@ import (
 	"gobot.io/x/gobot/drivers/i2c"
 	"gobot.io/x/gobot/gobottest"
 	"gobot.io/x/gobot/sysfs"
+	"runtime"
+	"strconv"
+	"sync"
 )
 
 // make sure that this Adaptor fullfills all the required interfaces
@@ -174,3 +177,52 @@ func TestAdaptorI2c(t *testing.T) {
 
 	gobottest.Assert(t, a.GetDefaultBus(), 1)
 }
+
+func TestAdaptorDigitalPinConcurrency(t *testing.T) {
+
+	oldProcs := runtime.GOMAXPROCS(0)
+	runtime.GOMAXPROCS(8)
+
+	for retry := 0; retry < 20; retry++ {
+
+		a := initTestAdaptor()
+		var wg sync.WaitGroup
+		wg.Add(20)
+
+		for i := 0; i < 20; i++ {
+			pinAsString := strconv.Itoa(i)
+			go func() {
+				defer wg.Done()
+				a.DigitalPin(pinAsString, sysfs.IN)
+			}()
+		}
+
+		wg.Wait()
+	}
+
+	runtime.GOMAXPROCS(oldProcs)
+
+}
+
+func TestAdaptorPWMPin(t *testing.T) {
+	a := initTestAdaptor()
+
+	gobottest.Assert(t, len(a.pwmPins), 0)
+
+	firstSysPin, err := a.PWMPin("35")
+
+	gobottest.Assert(t, err, nil)
+	gobottest.Assert(t, len(a.pwmPins), 1)
+
+	secondSysPin, err := a.PWMPin("35")
+
+	gobottest.Assert(t, err, nil)
+	gobottest.Assert(t, len(a.pwmPins), 1)
+	gobottest.Assert(t, firstSysPin, secondSysPin)
+
+	otherSysPin, err := a.PWMPin("36")
+
+	gobottest.Assert(t, err, nil)
+	gobottest.Assert(t, len(a.pwmPins), 2)
+	gobottest.Refute(t, firstSysPin, otherSysPin)
+}