128 lines
2.7 KiB
Go
128 lines
2.7 KiB
Go
package libgobuster
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"crypto/rand"
|
|
"fmt"
|
|
"math/big"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
func httpServerB(b *testing.B, content string) *httptest.Server {
|
|
b.Helper()
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
fmt.Fprint(w, content)
|
|
}))
|
|
return ts
|
|
}
|
|
|
|
func httpServerT(t *testing.T, content string) *httptest.Server {
|
|
t.Helper()
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
fmt.Fprint(w, content)
|
|
}))
|
|
return ts
|
|
}
|
|
|
|
func randomString(length int) (string, error) {
|
|
var letter = []byte("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")
|
|
letterLen := len(letter)
|
|
|
|
b := make([]byte, length)
|
|
for i := range b {
|
|
n, err := rand.Int(rand.Reader, big.NewInt(int64(letterLen)))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
b[i] = letter[n.Int64()]
|
|
}
|
|
return string(b), nil
|
|
}
|
|
|
|
func TestRequest(t *testing.T) {
|
|
t.Parallel()
|
|
ret, err := randomString(100)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
h := httpServerT(t, ret)
|
|
defer h.Close()
|
|
var o HTTPOptions
|
|
c, err := NewHTTPClient(&o)
|
|
if err != nil {
|
|
t.Fatalf("Got Error: %v", err)
|
|
}
|
|
status, length, _, body, err := c.Request(context.Background(), h.URL, RequestOptions{ReturnBody: true})
|
|
if err != nil {
|
|
t.Fatalf("Got Error: %v", err)
|
|
}
|
|
if status != 200 {
|
|
t.Fatalf("Invalid status returned: %d", status)
|
|
}
|
|
if length != int64(len(ret)) {
|
|
t.Fatalf("Invalid length returned: %d", length)
|
|
}
|
|
if body == nil || !bytes.Equal(body, []byte(ret)) {
|
|
t.Fatalf("Invalid body returned: %d", body)
|
|
}
|
|
}
|
|
|
|
func BenchmarkRequestWithoutBody(b *testing.B) {
|
|
r, err := randomString(10000)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
h := httpServerB(b, r)
|
|
defer h.Close()
|
|
var o HTTPOptions
|
|
c, err := NewHTTPClient(&o)
|
|
if err != nil {
|
|
b.Fatalf("Got Error: %v", err)
|
|
}
|
|
for x := 0; x < b.N; x++ {
|
|
_, _, _, _, err := c.Request(context.Background(), h.URL, RequestOptions{ReturnBody: false})
|
|
if err != nil {
|
|
b.Fatalf("Got Error: %v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkRequestWitBody(b *testing.B) {
|
|
r, err := randomString(10000)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
h := httpServerB(b, r)
|
|
defer h.Close()
|
|
var o HTTPOptions
|
|
c, err := NewHTTPClient(&o)
|
|
if err != nil {
|
|
b.Fatalf("Got Error: %v", err)
|
|
}
|
|
for x := 0; x < b.N; x++ {
|
|
_, _, _, _, err := c.Request(context.Background(), h.URL, RequestOptions{ReturnBody: true})
|
|
if err != nil {
|
|
b.Fatalf("Got Error: %v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkNewHTTPClient(b *testing.B) {
|
|
r, err := randomString(500)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
h := httpServerB(b, r)
|
|
defer h.Close()
|
|
var o HTTPOptions
|
|
for x := 0; x < b.N; x++ {
|
|
_, err := NewHTTPClient(&o)
|
|
if err != nil {
|
|
b.Fatalf("Got Error: %v", err)
|
|
}
|
|
}
|
|
}
|