transport/http: fix partial request with haves. Fix #216. (#221)

1 files changed, 4 insertions, 3 deletions

diff --git a/plumbing/protocol/packp/uppackreq.go b/plumbing/protocol/packp/uppackreq.go
index 84e2b4e..4bb22d0 100644
--- a/plumbing/protocol/packp/uppackreq.go
+++ b/plumbing/protocol/packp/uppackreq.go
@@ -68,8 +68,9 @@ type UploadHaves struct {
 	Haves []plumbing.Hash
 }
 
-// Encode encodes the UploadHaves into the Writer.
-func (u *UploadHaves) Encode(w io.Writer) error {
+// Encode encodes the UploadHaves into the Writer. If flush is true, a flush
+// command will be encoded at the end of the writer content.
+func (u *UploadHaves) Encode(w io.Writer, flush bool) error {
 	e := pktline.NewEncoder(w)
 
 	plumbing.HashesSort(u.Haves)
@@ -87,7 +88,7 @@ func (u *UploadHaves) Encode(w io.Writer) error {
 		last = have
 	}
 
-	if len(u.Haves) != 0 {
+	if flush && len(u.Haves) != 0 {
 		if err := e.Flush(); err != nil {
 			return fmt.Errorf("sending flush-pkt after haves: %s", err)
 		}