packer/rpc: log when client closes mux

packer/rpc/client.go

@@ -3,6 +3,7 @@ package rpc
 import (
 	"github.com/mitchellh/packer/packer"
 	"io"
+	"log"
 	"net/rpc"
 )
 
@@ -12,10 +13,17 @@ import (
 type Client struct {
 	mux      *MuxConn
 	client   *rpc.Client
+	closeMux bool
 }
 
 func NewClient(rwc io.ReadWriteCloser) (*Client, error) {
-	return NewClientWithMux(NewMuxConn(rwc), 0)
+	result, err := NewClientWithMux(NewMuxConn(rwc), 0)
+	if err != nil {
+		return nil, err
+	}
+
+	result.closeMux = true
+	return result, err
 }
 
 func NewClientWithMux(mux *MuxConn, streamId uint32) (*Client, error) {
@@ -27,6 +35,7 @@ func NewClientWithMux(mux *MuxConn, streamId uint32) (*Client, error) {
 	return &Client{
 		mux:      mux,
 		client:   rpc.NewClient(clientConn),
+		closeMux: false,
 	}, nil
 }
 
@@ -35,6 +44,11 @@ func (c *Client) Close() error {
 		return err
 	}
 
+	if c.closeMux {
+		log.Printf("[WARN] Client is closing mux")
+		return c.mux.Close()
+	}
+
 	return nil
 }