tun2: read entire response body

This commit is contained in:
Cadey Ratio 2017-10-01 08:59:49 -07:00
parent 3fefd31625
commit c2ac2e3055
No known key found for this signature in database
GPG Key ID: D607EE27C2E7F89A
1 changed files with 11 additions and 0 deletions

View File

@ -2,7 +2,9 @@ package tun2
import (
"bufio"
"bytes"
"context"
"io/ioutil"
"net"
"net/http"
"time"
@ -138,6 +140,15 @@ func (c *Connection) RoundTrip(req *http.Request) (*http.Response, error) {
if err != nil {
return nil, errors.Wrap(err, ErrCantReadResponse.Error())
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return nil, errors.Wrap(err, "can't read response body")
}
resp.Body = ioutil.NopCloser(bytes.NewBuffer(body))
resp.ContentLength = int64(len(body))
return resp, nil
}