1
0
mirror of https://github.com/golang/go synced 2024-09-30 08:18:40 -06:00

net: return io.ErrClosedPipe when possible from net.Pipe

The previous implementation of net.Pipe was just a thin wrapper around
io.Pipe and did not wrap any of the io.Pipe errors as net.Errors.
As a result of Hyrum's law, users have come to depend on the fact that
net.Pipe returns io.ErrClosedPipe when the pipe is closed.
Thus, we preserve this behavior to avoid regressing such use cases.

Change-Id: I06b387877b944c1c08527601f58983872b7557b4
Reviewed-on: https://go-review.googlesource.com/81777
Run-TryBot: Joe Tsai <thebrokentoaster@gmail.com>
TryBot-Result: Gobot Gobot <gobot@golang.org>
Reviewed-by: Ian Lance Taylor <iant@golang.org>
This commit is contained in:
Joe Tsai 2017-12-04 11:16:51 -08:00 committed by Joe Tsai
parent 49fec9b488
commit 8f2a9267c8
2 changed files with 45 additions and 27 deletions

View File

@ -78,19 +78,11 @@ func isClosedChan(c <-chan struct{}) bool {
} }
} }
type pipeError struct { type timeoutError struct{}
errStr string
timeout bool
}
func (pe pipeError) Error() string { return pe.errStr } func (timeoutError) Error() string { return "deadline exceeded" }
func (pe pipeError) Timeout() bool { return pe.timeout } func (timeoutError) Timeout() bool { return true }
func (pe pipeError) Temporary() bool { return pe.timeout } func (timeoutError) Temporary() bool { return true }
var (
errDeadline = pipeError{"deadline exceeded", true}
errClosed = pipeError{"closed connection", false}
)
type pipeAddr struct{} type pipeAddr struct{}
@ -153,7 +145,7 @@ func (*pipe) RemoteAddr() Addr { return pipeAddr{} }
func (p *pipe) Read(b []byte) (int, error) { func (p *pipe) Read(b []byte) (int, error) {
n, err := p.read(b) n, err := p.read(b)
if err != nil && err != io.EOF { if err != nil && err != io.EOF && err != io.ErrClosedPipe {
err = &OpError{Op: "read", Net: "pipe", Err: err} err = &OpError{Op: "read", Net: "pipe", Err: err}
} }
return n, err return n, err
@ -162,11 +154,11 @@ func (p *pipe) Read(b []byte) (int, error) {
func (p *pipe) read(b []byte) (n int, err error) { func (p *pipe) read(b []byte) (n int, err error) {
switch { switch {
case isClosedChan(p.localDone): case isClosedChan(p.localDone):
return 0, errClosed return 0, io.ErrClosedPipe
case isClosedChan(p.remoteDone): case isClosedChan(p.remoteDone):
return 0, io.EOF return 0, io.EOF
case isClosedChan(p.readDeadline.wait()): case isClosedChan(p.readDeadline.wait()):
return 0, errDeadline return 0, timeoutError{}
} }
select { select {
@ -175,17 +167,17 @@ func (p *pipe) read(b []byte) (n int, err error) {
p.rdTx <- nr p.rdTx <- nr
return nr, nil return nr, nil
case <-p.localDone: case <-p.localDone:
return 0, errClosed return 0, io.ErrClosedPipe
case <-p.remoteDone: case <-p.remoteDone:
return 0, io.EOF return 0, io.EOF
case <-p.readDeadline.wait(): case <-p.readDeadline.wait():
return 0, errDeadline return 0, timeoutError{}
} }
} }
func (p *pipe) Write(b []byte) (int, error) { func (p *pipe) Write(b []byte) (int, error) {
n, err := p.write(b) n, err := p.write(b)
if err != nil { if err != nil && err != io.ErrClosedPipe {
err = &OpError{Op: "write", Net: "pipe", Err: err} err = &OpError{Op: "write", Net: "pipe", Err: err}
} }
return n, err return n, err
@ -194,11 +186,11 @@ func (p *pipe) Write(b []byte) (int, error) {
func (p *pipe) write(b []byte) (n int, err error) { func (p *pipe) write(b []byte) (n int, err error) {
switch { switch {
case isClosedChan(p.localDone): case isClosedChan(p.localDone):
return 0, errClosed return 0, io.ErrClosedPipe
case isClosedChan(p.remoteDone): case isClosedChan(p.remoteDone):
return 0, errClosed return 0, io.ErrClosedPipe
case isClosedChan(p.writeDeadline.wait()): case isClosedChan(p.writeDeadline.wait()):
return 0, errDeadline return 0, timeoutError{}
} }
p.wrMu.Lock() // Ensure entirety of b is written together p.wrMu.Lock() // Ensure entirety of b is written together
@ -210,11 +202,11 @@ func (p *pipe) write(b []byte) (n int, err error) {
b = b[nw:] b = b[nw:]
n += nw n += nw
case <-p.localDone: case <-p.localDone:
return n, errClosed return n, io.ErrClosedPipe
case <-p.remoteDone: case <-p.remoteDone:
return n, errClosed return n, io.ErrClosedPipe
case <-p.writeDeadline.wait(): case <-p.writeDeadline.wait():
return n, errDeadline return n, timeoutError{}
} }
} }
return n, nil return n, nil
@ -222,7 +214,7 @@ func (p *pipe) write(b []byte) (n int, err error) {
func (p *pipe) SetDeadline(t time.Time) error { func (p *pipe) SetDeadline(t time.Time) error {
if isClosedChan(p.localDone) || isClosedChan(p.remoteDone) { if isClosedChan(p.localDone) || isClosedChan(p.remoteDone) {
return &OpError{Op: "set", Net: "pipe", Err: errClosed} return io.ErrClosedPipe
} }
p.readDeadline.set(t) p.readDeadline.set(t)
p.writeDeadline.set(t) p.writeDeadline.set(t)
@ -231,7 +223,7 @@ func (p *pipe) SetDeadline(t time.Time) error {
func (p *pipe) SetReadDeadline(t time.Time) error { func (p *pipe) SetReadDeadline(t time.Time) error {
if isClosedChan(p.localDone) || isClosedChan(p.remoteDone) { if isClosedChan(p.localDone) || isClosedChan(p.remoteDone) {
return &OpError{Op: "set", Net: "pipe", Err: errClosed} return io.ErrClosedPipe
} }
p.readDeadline.set(t) p.readDeadline.set(t)
return nil return nil
@ -239,7 +231,7 @@ func (p *pipe) SetReadDeadline(t time.Time) error {
func (p *pipe) SetWriteDeadline(t time.Time) error { func (p *pipe) SetWriteDeadline(t time.Time) error {
if isClosedChan(p.localDone) || isClosedChan(p.remoteDone) { if isClosedChan(p.localDone) || isClosedChan(p.remoteDone) {
return &OpError{Op: "set", Net: "pipe", Err: errClosed} return io.ErrClosedPipe
} }
p.writeDeadline.set(t) p.writeDeadline.set(t)
return nil return nil

View File

@ -5,8 +5,10 @@
package net_test package net_test
import ( import (
"io"
"net" "net"
"testing" "testing"
"time"
"golang_org/x/net/nettest" "golang_org/x/net/nettest"
) )
@ -21,3 +23,27 @@ func TestPipe(t *testing.T) {
return return
}) })
} }
func TestPipeCloseError(t *testing.T) {
c1, c2 := net.Pipe()
c1.Close()
if _, err := c1.Read(nil); err != io.ErrClosedPipe {
t.Errorf("c1.Read() = %v, want io.ErrClosedPipe", err)
}
if _, err := c1.Write(nil); err != io.ErrClosedPipe {
t.Errorf("c1.Write() = %v, want io.ErrClosedPipe", err)
}
if err := c1.SetDeadline(time.Time{}); err != io.ErrClosedPipe {
t.Errorf("c1.SetDeadline() = %v, want io.ErrClosedPipe", err)
}
if _, err := c2.Read(nil); err != io.EOF {
t.Errorf("c2.Read() = %v, want io.EOF", err)
}
if _, err := c2.Write(nil); err != io.ErrClosedPipe {
t.Errorf("c2.Write() = %v, want io.ErrClosedPipe", err)
}
if err := c2.SetDeadline(time.Time{}); err != io.ErrClosedPipe {
t.Errorf("c2.SetDeadline() = %v, want io.ErrClosedPipe", err)
}
}