mirror of https://github.com/daffainfo/nuclei.git
78 lines
2.5 KiB
Go
78 lines
2.5 KiB
Go
package file
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"testing"
|
|
|
|
"github.com/projectdiscovery/nuclei/v2/internal/testutils"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/operators"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/operators/extractors"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/operators/matchers"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/output"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestFileExecuteWithResults(t *testing.T) {
|
|
options := testutils.DefaultOptions
|
|
|
|
testutils.Init(options)
|
|
templateID := "testing-file"
|
|
request := &Request{
|
|
ID: templateID,
|
|
MaxSize: 1024,
|
|
NoRecursive: false,
|
|
Extensions: []string{"*"},
|
|
ExtensionAllowlist: []string{".lock"},
|
|
ExtensionDenylist: []string{".go"},
|
|
Operators: operators.Operators{
|
|
Matchers: []*matchers.Matcher{{
|
|
Name: "test",
|
|
Part: "raw",
|
|
Type: "word",
|
|
Words: []string{"1.1.1.1"},
|
|
}},
|
|
Extractors: []*extractors.Extractor{{
|
|
Part: "raw",
|
|
Type: "regex",
|
|
Regex: []string{"[0-9]+\\.[0-9]+\\.[0-9]+\\.[0-9]+"},
|
|
}},
|
|
},
|
|
}
|
|
executerOpts := testutils.NewMockExecuterOptions(options, &testutils.TemplateInfo{
|
|
ID: templateID,
|
|
Info: map[string]interface{}{"severity": "low", "name": "test"},
|
|
})
|
|
err := request.Compile(executerOpts)
|
|
require.Nil(t, err, "could not compile file request")
|
|
|
|
tempDir, err := ioutil.TempDir("", "test-*")
|
|
require.Nil(t, err, "could not create temporary directory")
|
|
defer os.RemoveAll(tempDir)
|
|
|
|
files := map[string]string{
|
|
"config.yaml": "TEST\r\n1.1.1.1\r\n",
|
|
}
|
|
for k, v := range files {
|
|
err = ioutil.WriteFile(path.Join(tempDir, k), []byte(v), 0777)
|
|
require.Nil(t, err, "could not write temporary file")
|
|
}
|
|
|
|
var finalEvent *output.InternalWrappedEvent
|
|
t.Run("valid", func(t *testing.T) {
|
|
metadata := make(output.InternalEvent)
|
|
previous := make(output.InternalEvent)
|
|
err := request.ExecuteWithResults(tempDir, metadata, previous, func(event *output.InternalWrappedEvent) {
|
|
finalEvent = event
|
|
})
|
|
require.Nil(t, err, "could not execute file request")
|
|
})
|
|
require.NotNil(t, finalEvent, "could not get event output from request")
|
|
require.Equal(t, 1, len(finalEvent.Results), "could not get correct number of results")
|
|
require.Equal(t, "test", finalEvent.Results[0].MatcherName, "could not get correct matcher name of results")
|
|
require.Equal(t, 1, len(finalEvent.Results[0].ExtractedResults), "could not get correct number of extracted results")
|
|
require.Equal(t, "1.1.1.1", finalEvent.Results[0].ExtractedResults[0], "could not get correct extracted results")
|
|
finalEvent = nil
|
|
}
|