fix deadlock in cancelable stdin

This commit is contained in:
chzyer 2016-10-06 12:30:22 +08:00
parent 94eaec69a7
commit fa8d922787
1 changed files with 8 additions and 1 deletions

7
std.go
View File

@ -4,6 +4,7 @@ import (
"io"
"os"
"sync"
"sync/atomic"
)
var (
@ -69,6 +70,7 @@ type CancelableStdin struct {
r io.Reader
mutex sync.Mutex
stop chan struct{}
closed int32
notify chan struct{}
data []byte
read int
@ -101,6 +103,9 @@ loop:
func (c *CancelableStdin) Read(b []byte) (n int, err error) {
c.mutex.Lock()
defer c.mutex.Unlock()
if atomic.LoadInt32(&c.closed) == 1 {
return 0, io.EOF
}
c.data = b
c.notify <- struct{}{}
@ -113,6 +118,8 @@ func (c *CancelableStdin) Read(b []byte) (n int, err error) {
}
func (c *CancelableStdin) Close() error {
if atomic.CompareAndSwapInt32(&c.closed, 0, 1) {
close(c.stop)
}
return nil
}