Skip to content

Commit

Permalink
os: don't wait for Close if the File was returned by NewFile
Browse files Browse the repository at this point in the history
os.NewFile doesn't put the fd into non-blocking mode.
In most cases, an *os.File returned by os.NewFile is in blocking mode.

Updates #7970
Updates #21856
Updates #23111

Change-Id: Iab08432e41f7ac1b5e25aaa8855d478adb7f98ed
Reviewed-on: https://go-review.googlesource.com/83995
Reviewed-by: Ian Lance Taylor <[email protected]>
  • Loading branch information
crvv authored and ianlancetaylor committed Dec 14, 2017
1 parent b944f91 commit e28a0d3
Show file tree
Hide file tree
Showing 2 changed files with 24 additions and 6 deletions.
1 change: 1 addition & 0 deletions src/internal/poll/fd_unix.go
Original file line number Diff line number Diff line change
Expand Up @@ -56,6 +56,7 @@ func (fd *FD) Init(net string, pollable bool) error {
fd.isFile = true
}
if !pollable {
fd.isBlocking = true
return nil
}
return fd.pd.init(fd)
Expand Down
29 changes: 23 additions & 6 deletions src/os/pipe_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ package os_test
import (
"fmt"
"internal/testenv"
"io"
"io/ioutil"
"os"
osexec "os/exec"
Expand Down Expand Up @@ -222,12 +223,28 @@ func TestReadNonblockingFd(t *testing.T) {
}
}

// Test that we don't let a blocking read prevent a close.
func TestCloseWithBlockingRead(t *testing.T) {
func TestCloseWithBlockingReadByNewFile(t *testing.T) {
var p [2]int
err := syscall.Pipe(p[:])
if err != nil {
t.Fatal(err)
}
// os.NewFile returns a blocking mode file.
testCloseWithBlockingRead(t, os.NewFile(uintptr(p[0]), "reader"), os.NewFile(uintptr(p[1]), "writer"))
}

func TestCloseWithBlockingReadByFd(t *testing.T) {
r, w, err := os.Pipe()
if err != nil {
t.Fatal(err)
}
// Calling Fd will put the file into blocking mode.
_ = r.Fd()
testCloseWithBlockingRead(t, r, w)
}

// Test that we don't let a blocking read prevent a close.
func testCloseWithBlockingRead(t *testing.T, r, w *os.File) {
defer r.Close()
defer w.Close()

Expand All @@ -248,18 +265,18 @@ func TestCloseWithBlockingRead(t *testing.T) {
close(c)
}(c1)

// Calling Fd will put the file into blocking mode.
_ = r.Fd()

wg.Add(1)
go func(c chan bool) {
defer wg.Done()
var b [1]byte
_, err = r.Read(b[:])
_, err := r.Read(b[:])
close(c)
if err == nil {
t.Error("I/O on closed pipe unexpectedly succeeded")
}
if err != io.EOF {
t.Errorf("got %v, expected io.EOF", err)
}
}(c2)

for c1 != nil || c2 != nil {
Expand Down

0 comments on commit e28a0d3

Please sign in to comment.