client.go

@@ -159,10 +159,7 @@ func UseFstat(value bool) ClientOption {
 }
 
 // CopyStderrTo specifies a writer to which the standard error of the remote sftp-server command should be written.
-//
-// The writer passed in will not be automatically closed.
-// It is the responsibility of the caller to coordinate closure of any writers.
-func CopyStderrTo(wr io.Writer) ClientOption {
+func CopyStderrTo(wr io.WriteCloser) ClientOption {
 	return func(c *Client) error {
 		c.stderrTo = wr
 		return nil
@@ -177,7 +174,7 @@ func CopyStderrTo(wr io.Writer) ClientOption {
 type Client struct {
 	clientConn
 
-	stderrTo io.Writer
+	stderrTo io.WriteCloser
 
 	ext map[string]string // Extensions (name -> data).
 
@@ -217,17 +214,17 @@ func NewClient(conn *ssh.Client, opts ...ClientOption) (*Client, error) {
 		return nil, err
 	}
 
-	return newClientPipe(pr, perr, pw, s.Wait, opts...)
+	return newClientPipe(pr, pw, perr, s.Wait, opts...)
 }
 
 // NewClientPipe creates a new SFTP client given a Reader and a WriteCloser.
 // This can be used for connecting to an SFTP server over TCP/TLS or by using
 // the system's ssh client program (e.g. via exec.Command).
 func NewClientPipe(rd io.Reader, wr io.WriteCloser, opts ...ClientOption) (*Client, error) {
-	return newClientPipe(rd, nil, wr, nil, opts...)
+	return newClientPipe(rd, wr, nil, nil, opts...)
 }
 
-func newClientPipe(rd, stderr io.Reader, wr io.WriteCloser, wait func() error, opts ...ClientOption) (*Client, error) {
+func newClientPipe(rd io.Reader, wr io.WriteCloser, stderr io.Reader, wait func() error, opts ...ClientOption) (*Client, error) {
 	c := &Client{
 		clientConn: clientConn{
 			conn: conn{
@@ -259,10 +256,13 @@ func newClientPipe(rd, stderr io.Reader, wr io.WriteCloser, wait func() error, o
 		}
 
 		go func() {
-			// DO NOT close the writer!
-			// Programs may pass in `os.Stderr` to write the remote stderr to,
-			// and the program may continue after disconnect by reconnecting.
-			// But if we've closed their stderr, then we just messed everything up.
+			defer func() {
+				if closer, ok := wr.(io.Closer); ok {
+					if err := closer.Close(); err != nil {
+						debug("error closing stderrTo: %v", err)
+					}
+				}
+			}()
 
 			if _, err := io.Copy(wr, stderr); err != nil {
 				debug("error copying stderr: %v", err)

conn.go

@@ -22,7 +22,7 @@ type conn struct {
 // For the client mode just pass 0.
 // It returns io.EOF if the connection is closed and
 // there are no more packets to read.
-func (c *conn) recvPacket(orderID uint32) (fxp, []byte, error) {
+func (c *conn) recvPacket(orderID uint32) (uint8, []byte, error) {
 	return recvPacket(c, c.alloc, orderID)
 }
 
@@ -142,7 +142,7 @@ func (c *clientConn) getChannel(sid uint32) (chan<- result, bool) {
 
 // result captures the result of receiving the a packet from the server
 type result struct {
-	typ  fxp
+	typ  byte
 	data []byte
 	err  error
 }
@@ -152,7 +152,7 @@ type idmarshaler interface {
 	encoding.BinaryMarshaler
 }
 
-func (c *clientConn) sendPacket(ctx context.Context, ch chan result, p idmarshaler) (fxp, []byte, error) {
+func (c *clientConn) sendPacket(ctx context.Context, ch chan result, p idmarshaler) (byte, []byte, error) {
 	if cap(ch) < 1 {
 		ch = make(chan result, 1)
 	}

packet.go

@@ -304,22 +304,16 @@ func sendPacket(w io.Writer, m encoding.BinaryMarshaler) error {
 	return nil
 }
 
-func recvPacket(r io.Reader, alloc *allocator, orderID uint32) (fxp, []byte, error) {
+func recvPacket(r io.Reader, alloc *allocator, orderID uint32) (uint8, []byte, error) {
 	var b []byte
 	if alloc != nil {
 		b = alloc.GetPage(orderID)
 	} else {
 		b = make([]byte, 4)
 	}
-
-	if n, err := io.ReadFull(r, b[:4]); err != nil {
-		if err == io.EOF {
-			return 0, nil, err
-		}
-
-		return 0, nil, fmt.Errorf("error reading packet length: %d of 4: %w", n, err)
+	if _, err := io.ReadFull(r, b[:4]); err != nil {
+		return 0, nil, err
 	}
-
 	length, _ := unmarshalUint32(b)
 	if length > maxMsgLength {
 		debug("recv packet %d bytes too long", length)
@@ -329,39 +323,24 @@ func recvPacket(r io.Reader, alloc *allocator, orderID uint32) (fxp, []byte, err
 		debug("recv packet of 0 bytes too short")
 		return 0, nil, errShortPacket
 	}
-
 	if alloc == nil {
 		b = make([]byte, length)
 	}
-
-	n, err := io.ReadFull(r, b[:length])
-	b = b[:n]
-
-	if err != nil {
-		debug("recv packet error: %d of %d bytes: %x", n, length, b)
-
+	if _, err := io.ReadFull(r, b[:length]); err != nil {
 		// ReadFull only returns EOF if it has read no bytes.
 		// In this case, that means a partial packet, and thus unexpected.
 		if err == io.EOF {
 			err = io.ErrUnexpectedEOF
 		}
-
-		if n == 0 {
-			return 0, nil, fmt.Errorf("error reading packet body: %d of %d: %w", n, length, err)
-		}
-
-		return 0, nil, fmt.Errorf("error reading packet body: %d of %d: (%s) %w", n, length, fxp(b[0]), err)
+		debug("recv packet %d bytes: err %v", length, err)
+		return 0, nil, err
 	}
-
-	typ, payload := fxp(b[0]), b[1:n]
-
 	if debugDumpRxPacketBytes {
-		debug("recv packet: %s %d bytes %x", typ, length, payload)
+		debug("recv packet: %s %d bytes %x", fxp(b[0]), length, b[1:length])
 	} else if debugDumpRxPacket {
-		debug("recv packet: %s %d bytes", typ, length)
+		debug("recv packet: %s %d bytes", fxp(b[0]), length)
 	}
-
-	return typ, payload, nil
+	return b[0], b[1:length], nil
 }
 
 type extensionPair struct {

packet_test.go

@@ -468,7 +468,7 @@ func TestRecvPacket(t *testing.T) {
 	var recvPacketTests = []struct {
 		b []byte
 
-		want    fxp
+		want    uint8
 		body    []byte
 		wantErr error
 	}{

request-server.go

@@ -148,7 +148,7 @@ func (rs *RequestServer) serveLoop(pktChan chan<- orderedRequest) error {
 
 	var err error
 	var pkt requestPacket
-	var pktType fxp
+	var pktType uint8
 	var pktBytes []byte
 
 	for {
@@ -158,7 +158,7 @@ func (rs *RequestServer) serveLoop(pktChan chan<- orderedRequest) error {
 			return err
 		}
 
-		pkt, err = makePacket(rxPacket{pktType, pktBytes})
+		pkt, err = makePacket(rxPacket{fxp(pktType), pktBytes})
 		if err != nil {
 			switch {
 			case errors.Is(err, errUnknownExtendedPacket):

server.go

@@ -390,7 +390,7 @@ func (svr *Server) Serve() error {
 
 	var err error
 	var pkt requestPacket
-	var pktType fxp
+	var pktType uint8
 	var pktBytes []byte
 	for {
 		pktType, pktBytes, err = svr.serverConn.recvPacket(svr.pktMgr.getNextOrderID())
@@ -403,7 +403,7 @@ func (svr *Server) Serve() error {
 			break
 		}
 
-		pkt, err = makePacket(rxPacket{pktType, pktBytes})
+		pkt, err = makePacket(rxPacket{fxp(pktType), pktBytes})
 		if err != nil {
 			switch {
 			case errors.Is(err, errUnknownExtendedPacket):

sftp.go

@@ -184,15 +184,15 @@ func (f fx) String() string {
 }
 
 type unexpectedPacketErr struct {
-	want, got fxp
+	want, got uint8
 }
 
 func (u *unexpectedPacketErr) Error() string {
-	return fmt.Sprintf("sftp: unexpected packet: want %v, got %v", u.want, u.got)
+	return fmt.Sprintf("sftp: unexpected packet: want %v, got %v", fxp(u.want), fxp(u.got))
 }
 
-func unimplementedPacketErr(u fxp) error {
-	return fmt.Errorf("sftp: unimplemented packet type: got %v", u)
+func unimplementedPacketErr(u uint8) error {
+	return fmt.Errorf("sftp: unimplemented packet type: got %v", fxp(u))
 }
 
 type unexpectedIDErr struct{ want, got uint32 }