nuclei/pkg/templates/templates_utils.go at dev · stringsbuilder/nuclei · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
package templates

import "github.com/projectdiscovery/nuclei/v3/pkg/protocols"

// HasRequest returns true if the given requests slice is non-empty.
//
// If n is provided, it checks for more than n requests.
func HasRequest[T protocols.Request](requests []T, n ...int) bool {
	if len(n) > 0 {
		return len(requests) > n[0]
	}

	return len(requests) > 0
}

// HasDNSRequest returns true if the template has a DNS request.
//
// If n is provided, it checks for more than n requests.
func (t *Template) HasDNSRequest(n ...int) bool {
	return HasRequest(t.RequestsDNS, n...)
}

// HasFileRequest returns true if the template has a File request.
//
// If n is provided, it checks for more than n requests.
func (t *Template) HasFileRequest(n ...int) bool {
	return HasRequest(t.RequestsFile, n...)
}

// HasHTTPRequest returns true if the template has an HTTP request.
//
// If n is provided, it checks for more than n requests.
func (t *Template) HasHTTPRequest(n ...int) bool {
	return HasRequest(t.RequestsHTTP, n...)
}

// HasHeadlessRequest returns true if the template has a Headless protocol
// request.
//
// If n is provided, it checks for more than n requests.
func (t *Template) HasHeadlessRequest(n ...int) bool {
	return HasRequest(t.RequestsHeadless, n...)
}

// HasNetworkRequest returns true if the template has a Network protocol
// request.
//
// If n is provided, it checks for more than n requests.
func (t *Template) HasNetworkRequest(n ...int) bool {
	return HasRequest(t.RequestsNetwork, n...)
}

// HasSSLRequest returns true if the template has an SSL request.
//
// If n is provided, it checks for more than n requests.
func (t *Template) HasSSLRequest(n ...int) bool {
	return HasRequest(t.RequestsSSL, n...)
}

// HasWebsocketRequest returns true if the template has a Websocket protocol
// request.
//
// If n is provided, it checks for more than n requests.
func (t *Template) HasWebsocketRequest(n ...int) bool {
	return HasRequest(t.RequestsWebsocket, n...)
}

// HasWHOISRequest returns true if the template has a WHOIS request.
//
// If n is provided, it checks for more than n requests.
func (t *Template) HasWHOISRequest(n ...int) bool {
	return HasRequest(t.RequestsWHOIS, n...)
}

// HasCodeRequest returns true if the template has a Code request.
//
// If n is provided, it checks for more than n requests.
func (t *Template) HasCodeRequest(n ...int) bool {
	return HasRequest(t.RequestsCode, n...)
}

// HasJavascriptRequest returns true if the template has a Javascript protocol
// request.
//
// If n is provided, it checks for more than n requests.
func (t *Template) HasJavascriptRequest(n ...int) bool {
	return HasRequest(t.RequestsJavascript, n...)
}

// HasQueueRequests returns true if the template has queued requests.
//
// Queued requests contain all template requests in order (both protocol &
// request order).
//
// If n is provided, it checks for more than n requests.
func (t *Template) HasQueueRequests(n ...int) bool {
	return HasRequest(t.RequestsQueue, n...)
}

// HasWorkflows returns true if the template has workflows defined.
func (t *Template) HasWorkflows() bool {
	return len(t.Workflows) > 0
}

// IsFuzzableRequest returns true if the template has at least one request with
// fuzzing configured.
//
// Currently, it checks across HTTP and Headless requests.
func (t *Template) IsFuzzableRequest() bool {
	if t.HasHTTPRequest() {
		for _, request := range t.RequestsHTTP {
			if request.HasFuzzing() {
				return true
			}
		}
	}

	if t.HasHeadlessRequest() {
		for _, request := range t.RequestsHeadless {
			if request.HasFuzzing() {
				return true
			}
		}
	}

	return false
}

// IsFlowTemplate returns true if the template has a flow defined.
func (template *Template) IsFlowTemplate() bool {
	return template.Flow != "" && len(template.Flow) > 0
}

// IsGlobalMatchersTemplate returns true if the template has global matchers
// defined.
func (template *Template) IsGlobalMatchersTemplate() bool {
	return template.Options != nil &&
		template.Options.GlobalMatchers != nil &&
		template.Options.GlobalMatchers.HasMatchers()
}