Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
projectdiscovery
GitHub Repository: projectdiscovery/nuclei
 Path: blob/dev/pkg/protocols/dns/request_test.go
2070 views
1
package dns

2


3
import (

4
	"context"

5
	"testing"

6


7
	"github.com/stretchr/testify/require"

8


9
	"github.com/projectdiscovery/nuclei/v3/pkg/model"

10
	"github.com/projectdiscovery/nuclei/v3/pkg/model/types/severity"

11
	"github.com/projectdiscovery/nuclei/v3/pkg/operators"

12
	"github.com/projectdiscovery/nuclei/v3/pkg/operators/extractors"

13
	"github.com/projectdiscovery/nuclei/v3/pkg/operators/matchers"

14
	"github.com/projectdiscovery/nuclei/v3/pkg/output"

15
	"github.com/projectdiscovery/nuclei/v3/pkg/protocols/common/contextargs"

16
	"github.com/projectdiscovery/nuclei/v3/pkg/testutils"

17
)

18


19
func TestDNSExecuteWithResults(t *testing.T) {

20
	options := testutils.DefaultOptions

21


22
	recursion := true

23
	testutils.Init(options)

24
	templateID := "testing-dns"

25
	executerOpts := testutils.NewMockExecuterOptions(options, &testutils.TemplateInfo{

26
		ID:   templateID,

27
		Info: model.Info{SeverityHolder: severity.Holder{Severity: severity.Low}, Name: "test"},

28
	})

29
	request := &Request{

30
		RequestType: DNSRequestTypeHolder{DNSRequestType: A},

31
		Class:       "INET",

32
		Retries:     5,

33
		ID:          templateID,

34
		Recursion:   &recursion,

35
		Name:        "{{FQDN}}",

36
		Operators: operators.Operators{

37
			Matchers: []*matchers.Matcher{{

38
				Name:  "test",

39
				Part:  "raw",

40
				Type:  matchers.MatcherTypeHolder{MatcherType: matchers.WordsMatcher},

41
				Words: []string{"8.8.8.8"},

42
			}},

43
			Extractors: []*extractors.Extractor{{

44
				Part:  "raw",

45
				Type:  extractors.ExtractorTypeHolder{ExtractorType: extractors.RegexExtractor},

46
				Regex: []string{"[0-9]+\\.[0-9]+\\.[0-9]+\\.[0-9]+"},

47
			}},

48
		},

49
		options: executerOpts,

50
	}

51
	err := request.Compile(executerOpts)

52
	require.Nil(t, err, "could not compile dns request")

53


54
	var finalEvent *output.InternalWrappedEvent

55
	t.Run("domain-valid", func(t *testing.T) {

56
		metadata := make(output.InternalEvent)

57
		previous := make(output.InternalEvent)

58
		ctxArgs := contextargs.NewWithInput(context.Background(), "dns.google")

59
		err := request.ExecuteWithResults(ctxArgs, metadata, previous, func(event *output.InternalWrappedEvent) {

60
			finalEvent = event

61
		})

62
		require.Nil(t, err, "could not execute dns request")

63
	})

64
	require.NotNil(t, finalEvent, "could not get event output from request")

65
	require.Equal(t, 1, len(finalEvent.Results), "could not get correct number of results")

66
	require.Equal(t, "test", finalEvent.Results[0].MatcherName, "could not get correct matcher name of results")

67
	require.GreaterOrEqual(t, 2, len(finalEvent.Results[0].ExtractedResults), "could not get correct number of extracted results")

68
	require.Contains(t, finalEvent.Results[0].ExtractedResults, "8.8.8.8", "could not get correct extracted results")

69
	require.Contains(t, finalEvent.Results[0].ExtractedResults, "8.8.4.4", "could not get correct extracted results")

70
	finalEvent = nil

71
	// Note: changing url to domain is responsible at tmplexec package and is implemented there

72
}

73


74