Use actual body length for BytesIn

2013-09-09 03:04:33 +01:00
parent 93310995c7
commit 67fb276a76
1 changed files with 7 additions and 3 deletions
--- a/lib/attack.go
+++ b/lib/attack.go
@@ -73,9 +73,13 @@ func hit(req *http.Request, res chan Result) {
 		Error:     err,
 	}
 	if err == nil {
-		result.BytesIn, result.Code = uint64(r.ContentLength), uint64(r.StatusCode)
-		if body, err := ioutil.ReadAll(r.Body); err != nil && (result.Code < 200 || result.Code >= 300) {
-			result.Error = errors.New(string(body))
+		result.Code = uint64(r.StatusCode)
+		if body, err := ioutil.ReadAll(r.Body); err != nil {
+			if result.Code < 200 || result.Code >= 300 {
+				result.Error = errors.New(string(body))
+			}
+		} else {
+			result.BytesIn = uint64(len(body))
 		}
 	}
 	res <- result