github.com/projectdiscovery/nuclei/v2@v2.9.15/pkg/protocols/http/request_generator_test.go (about)

     1  package http
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/require"
     7  
     8  	"github.com/projectdiscovery/nuclei/v2/pkg/catalog/disk"
     9  	"github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/generators"
    10  )
    11  
    12  func TestRequestGeneratorPaths(t *testing.T) {
    13  	req := &Request{
    14  		Path: []string{"{{BaseURL}}/test", "{{BaseURL}}/test.php"},
    15  	}
    16  	generator := req.newGenerator(false)
    17  	var payloads []string
    18  	for {
    19  		raw, _, ok := generator.nextValue()
    20  		if !ok {
    21  			break
    22  		}
    23  		payloads = append(payloads, raw)
    24  	}
    25  	require.Equal(t, req.Path, payloads, "Could not get correct paths")
    26  }
    27  
    28  func TestRequestGeneratorClusterBombSingle(t *testing.T) {
    29  	var err error
    30  
    31  	req := &Request{
    32  		Payloads:   map[string]interface{}{"username": []string{"admin", "tomcat", "manager"}, "password": []string{"password", "test", "secret"}},
    33  		AttackType: generators.AttackTypeHolder{Value: generators.ClusterBombAttack},
    34  		Raw:        []string{`GET /{{username}}:{{password}} HTTP/1.1`},
    35  	}
    36  	catalogInstance := disk.NewCatalog("")
    37  	req.generator, err = generators.New(req.Payloads, req.AttackType.Value, "", false, catalogInstance, "")
    38  	require.Nil(t, err, "could not create generator")
    39  
    40  	generator := req.newGenerator(false)
    41  	var payloads []map[string]interface{}
    42  	for {
    43  		_, data, ok := generator.nextValue()
    44  		if !ok {
    45  			break
    46  		}
    47  		payloads = append(payloads, data)
    48  	}
    49  	require.Equal(t, 9, len(payloads), "Could not get correct number of payloads")
    50  }
    51  
    52  func TestRequestGeneratorClusterBombMultipleRaw(t *testing.T) {
    53  	var err error
    54  
    55  	req := &Request{
    56  		Payloads:   map[string]interface{}{"username": []string{"admin", "tomcat", "manager"}, "password": []string{"password", "test", "secret"}},
    57  		AttackType: generators.AttackTypeHolder{Value: generators.ClusterBombAttack},
    58  		Raw:        []string{`GET /{{username}}:{{password}} HTTP/1.1`, `GET /{{username}}@{{password}} HTTP/1.1`},
    59  	}
    60  	catalogInstance := disk.NewCatalog("")
    61  	req.generator, err = generators.New(req.Payloads, req.AttackType.Value, "", false, catalogInstance, "")
    62  	require.Nil(t, err, "could not create generator")
    63  
    64  	generator := req.newGenerator(false)
    65  	var payloads []map[string]interface{}
    66  	for {
    67  		_, data, ok := generator.nextValue()
    68  		if !ok {
    69  			break
    70  		}
    71  		payloads = append(payloads, data)
    72  	}
    73  	require.Equal(t, 18, len(payloads), "Could not get correct number of payloads")
    74  }