mirror of https://github.com/daffainfo/nuclei.git
341 lines
12 KiB
Go
341 lines
12 KiB
Go
package file
|
|
|
|
import (
|
|
"bufio"
|
|
"encoding/hex"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/docker/go-units"
|
|
"github.com/mholt/archiver"
|
|
"github.com/pkg/errors"
|
|
"github.com/remeh/sizedwaitgroup"
|
|
|
|
"github.com/projectdiscovery/gologger"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/operators"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/operators/matchers"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/output"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/protocols"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/contextargs"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/helpers/eventcreator"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/helpers/responsehighlighter"
|
|
templateTypes "github.com/projectdiscovery/nuclei/v2/pkg/templates/types"
|
|
sliceutil "github.com/projectdiscovery/utils/slice"
|
|
)
|
|
|
|
var _ protocols.Request = &Request{}
|
|
|
|
// Type returns the type of the protocol request
|
|
func (request *Request) Type() templateTypes.ProtocolType {
|
|
return templateTypes.FileProtocol
|
|
}
|
|
|
|
type FileMatch struct {
|
|
Data string
|
|
Line int
|
|
ByteIndex int
|
|
Match bool
|
|
Extract bool
|
|
Expr string
|
|
Raw string
|
|
}
|
|
|
|
var errEmptyResult = errors.New("Empty result")
|
|
|
|
// ExecuteWithResults executes the protocol requests and returns results instead of writing them.
|
|
func (request *Request) ExecuteWithResults(input *contextargs.Context, metadata, previous output.InternalEvent, callback protocols.OutputEventCallback) error {
|
|
wg := sizedwaitgroup.New(request.options.Options.BulkSize)
|
|
err := request.getInputPaths(input.MetaInput.Input, func(filePath string) {
|
|
wg.Add()
|
|
func(filePath string) {
|
|
defer wg.Done()
|
|
archiveReader, _ := archiver.ByExtension(filePath)
|
|
switch {
|
|
case archiveReader != nil:
|
|
switch archiveInstance := archiveReader.(type) {
|
|
case archiver.Walker:
|
|
err := archiveInstance.Walk(filePath, func(file archiver.File) error {
|
|
if !request.validatePath("/", file.Name(), true) {
|
|
return nil
|
|
}
|
|
// every new file in the compressed multi-file archive counts 1
|
|
request.options.Progress.AddToTotal(1)
|
|
archiveFileName := filepath.Join(filePath, file.Name())
|
|
event, fileMatches, err := request.processReader(file.ReadCloser, archiveFileName, input.MetaInput.Input, file.Size(), previous)
|
|
if err != nil {
|
|
if errors.Is(err, errEmptyResult) {
|
|
// no matches but one file elaborated
|
|
request.options.Progress.IncrementRequests()
|
|
return nil
|
|
}
|
|
gologger.Error().Msgf("%s\n", err)
|
|
// error while elaborating the file
|
|
request.options.Progress.IncrementFailedRequestsBy(1)
|
|
return err
|
|
}
|
|
defer file.Close()
|
|
dumpResponse(event, request.options, fileMatches, filePath)
|
|
callback(event)
|
|
// file elaborated and matched
|
|
request.options.Progress.IncrementRequests()
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
gologger.Error().Msgf("%s\n", err)
|
|
return
|
|
}
|
|
case archiver.Decompressor:
|
|
// compressed archive - contains only one file => increments the counter by 1
|
|
request.options.Progress.AddToTotal(1)
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
gologger.Error().Msgf("%s\n", err)
|
|
// error while elaborating the file
|
|
request.options.Progress.IncrementFailedRequestsBy(1)
|
|
return
|
|
}
|
|
defer file.Close()
|
|
fileStat, _ := file.Stat()
|
|
tmpFileOut, err := os.CreateTemp("", "")
|
|
if err != nil {
|
|
gologger.Error().Msgf("%s\n", err)
|
|
// error while elaborating the file
|
|
request.options.Progress.IncrementFailedRequestsBy(1)
|
|
return
|
|
}
|
|
defer tmpFileOut.Close()
|
|
defer os.RemoveAll(tmpFileOut.Name())
|
|
if err := archiveInstance.Decompress(file, tmpFileOut); err != nil {
|
|
gologger.Error().Msgf("%s\n", err)
|
|
// error while elaborating the file
|
|
request.options.Progress.IncrementFailedRequestsBy(1)
|
|
return
|
|
}
|
|
_ = tmpFileOut.Sync()
|
|
// rewind the file
|
|
_, _ = tmpFileOut.Seek(0, 0)
|
|
event, fileMatches, err := request.processReader(tmpFileOut, filePath, input.MetaInput.Input, fileStat.Size(), previous)
|
|
if err != nil {
|
|
if errors.Is(err, errEmptyResult) {
|
|
// no matches but one file elaborated
|
|
request.options.Progress.IncrementRequests()
|
|
return
|
|
}
|
|
gologger.Error().Msgf("%s\n", err)
|
|
// error while elaborating the file
|
|
request.options.Progress.IncrementFailedRequestsBy(1)
|
|
return
|
|
}
|
|
dumpResponse(event, request.options, fileMatches, filePath)
|
|
callback(event)
|
|
// file elaborated and matched
|
|
request.options.Progress.IncrementRequests()
|
|
}
|
|
default:
|
|
// normal file - increments the counter by 1
|
|
request.options.Progress.AddToTotal(1)
|
|
event, fileMatches, err := request.processFile(filePath, input.MetaInput.Input, previous)
|
|
if err != nil {
|
|
if errors.Is(err, errEmptyResult) {
|
|
// no matches but one file elaborated
|
|
request.options.Progress.IncrementRequests()
|
|
return
|
|
}
|
|
gologger.Error().Msgf("%s\n", err)
|
|
// error while elaborating the file
|
|
request.options.Progress.IncrementFailedRequestsBy(1)
|
|
return
|
|
}
|
|
dumpResponse(event, request.options, fileMatches, filePath)
|
|
callback(event)
|
|
// file elaborated and matched
|
|
request.options.Progress.IncrementRequests()
|
|
}
|
|
}(filePath)
|
|
})
|
|
|
|
wg.Wait()
|
|
if err != nil {
|
|
request.options.Output.Request(request.options.TemplatePath, input.MetaInput.Input, request.Type().String(), err)
|
|
request.options.Progress.IncrementFailedRequestsBy(1)
|
|
return errors.Wrap(err, "could not send file request")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (request *Request) processFile(filePath, input string, previousInternalEvent output.InternalEvent) (*output.InternalWrappedEvent, []FileMatch, error) {
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
return nil, nil, errors.Errorf("Could not open file path %s: %s\n", filePath, err)
|
|
}
|
|
defer file.Close()
|
|
|
|
stat, err := file.Stat()
|
|
if err != nil {
|
|
return nil, nil, errors.Errorf("Could not stat file path %s: %s\n", filePath, err)
|
|
}
|
|
if stat.Size() >= request.maxSize {
|
|
maxSizeString := units.HumanSize(float64(request.maxSize))
|
|
gologger.Verbose().Msgf("Limiting %s processed data to %s bytes: exceeded max size\n", filePath, maxSizeString)
|
|
}
|
|
|
|
return request.processReader(file, filePath, input, stat.Size(), previousInternalEvent)
|
|
}
|
|
|
|
func (request *Request) processReader(reader io.Reader, filePath, input string, totalBytes int64, previousInternalEvent output.InternalEvent) (*output.InternalWrappedEvent, []FileMatch, error) {
|
|
fileReader := io.LimitReader(reader, request.maxSize)
|
|
fileMatches, opResult := request.findMatchesWithReader(fileReader, input, filePath, totalBytes, previousInternalEvent)
|
|
if opResult == nil && len(fileMatches) == 0 {
|
|
return nil, nil, errEmptyResult
|
|
}
|
|
|
|
// build event structure to interface with internal logic
|
|
return request.buildEvent(input, filePath, fileMatches, opResult, previousInternalEvent), fileMatches, nil
|
|
}
|
|
|
|
func (request *Request) findMatchesWithReader(reader io.Reader, input, filePath string, totalBytes int64, previous output.InternalEvent) ([]FileMatch, *operators.Result) {
|
|
var bytesCount, linesCount, wordsCount int
|
|
isResponseDebug := request.options.Options.Debug || request.options.Options.DebugResponse
|
|
totalBytesString := units.BytesSize(float64(totalBytes))
|
|
|
|
scanner := bufio.NewScanner(reader)
|
|
buffer := []byte{}
|
|
if request.CompiledOperators.GetMatchersCondition() == matchers.ANDCondition {
|
|
scanner.Buffer(buffer, int(defaultMaxReadSize))
|
|
scanner.Split(func(data []byte, atEOF bool) (advance int, token []byte, err error) {
|
|
defaultMaxReadSizeInt := int(defaultMaxReadSize)
|
|
if len(data) > defaultMaxReadSizeInt {
|
|
return defaultMaxReadSizeInt, data[0:defaultMaxReadSizeInt], nil
|
|
}
|
|
if !atEOF {
|
|
return 0, nil, nil
|
|
}
|
|
return len(data), data, bufio.ErrFinalToken
|
|
})
|
|
} else {
|
|
scanner.Buffer(buffer, int(chunkSize))
|
|
}
|
|
|
|
var fileMatches []FileMatch
|
|
var opResult *operators.Result
|
|
for scanner.Scan() {
|
|
lineContent := scanner.Text()
|
|
n := len(lineContent)
|
|
|
|
// update counters
|
|
currentBytes := bytesCount + n
|
|
processedBytes := units.BytesSize(float64(currentBytes))
|
|
|
|
gologger.Verbose().Msgf("[%s] Processing file %s chunk %s/%s", request.options.TemplateID, filePath, processedBytes, totalBytesString)
|
|
dslMap := request.responseToDSLMap(lineContent, input, filePath)
|
|
for k, v := range previous {
|
|
dslMap[k] = v
|
|
}
|
|
discardEvent := eventcreator.CreateEvent(request, dslMap, isResponseDebug)
|
|
newOpResult := discardEvent.OperatorsResult
|
|
if newOpResult != nil {
|
|
if opResult == nil {
|
|
opResult = newOpResult
|
|
} else {
|
|
opResult.Merge(newOpResult)
|
|
}
|
|
if newOpResult.Matched || newOpResult.Extracted {
|
|
if newOpResult.Extracts != nil {
|
|
for expr, extracts := range newOpResult.Extracts {
|
|
for _, extract := range extracts {
|
|
fileMatches = append(fileMatches, FileMatch{
|
|
Data: extract,
|
|
Extract: true,
|
|
Line: linesCount + 1,
|
|
ByteIndex: bytesCount,
|
|
Expr: expr,
|
|
Raw: lineContent,
|
|
})
|
|
}
|
|
}
|
|
}
|
|
if newOpResult.Matches != nil {
|
|
for expr, matches := range newOpResult.Matches {
|
|
for _, match := range matches {
|
|
fileMatches = append(fileMatches, FileMatch{
|
|
Data: match,
|
|
Match: true,
|
|
Line: linesCount + 1,
|
|
ByteIndex: bytesCount,
|
|
Expr: expr,
|
|
Raw: lineContent,
|
|
})
|
|
}
|
|
}
|
|
}
|
|
for _, outputExtract := range newOpResult.OutputExtracts {
|
|
fileMatches = append(fileMatches, FileMatch{
|
|
Data: outputExtract,
|
|
Match: true,
|
|
Line: linesCount + 1,
|
|
ByteIndex: bytesCount,
|
|
Expr: outputExtract,
|
|
Raw: lineContent,
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
currentLinesCount := 1 + strings.Count(lineContent, "\n")
|
|
linesCount += currentLinesCount
|
|
wordsCount += strings.Count(lineContent, " ")
|
|
bytesCount = currentBytes
|
|
}
|
|
return fileMatches, opResult
|
|
}
|
|
|
|
func (request *Request) buildEvent(input, filePath string, fileMatches []FileMatch, operatorResult *operators.Result, previous output.InternalEvent) *output.InternalWrappedEvent {
|
|
exprLines := make(map[string][]int)
|
|
exprBytes := make(map[string][]int)
|
|
internalEvent := request.responseToDSLMap("", input, filePath)
|
|
for k, v := range previous {
|
|
internalEvent[k] = v
|
|
}
|
|
for _, fileMatch := range fileMatches {
|
|
exprLines[fileMatch.Expr] = append(exprLines[fileMatch.Expr], fileMatch.Line)
|
|
exprBytes[fileMatch.Expr] = append(exprBytes[fileMatch.Expr], fileMatch.ByteIndex)
|
|
}
|
|
|
|
event := eventcreator.CreateEventWithOperatorResults(request, internalEvent, operatorResult)
|
|
// Annotate with line numbers if asked by the user
|
|
if request.options.Options.ShowMatchLine {
|
|
for _, result := range event.Results {
|
|
switch {
|
|
case result.MatcherName != "":
|
|
result.Lines = exprLines[result.MatcherName]
|
|
case result.ExtractorName != "":
|
|
result.Lines = exprLines[result.ExtractorName]
|
|
default:
|
|
for _, extractedResult := range result.ExtractedResults {
|
|
result.Lines = append(result.Lines, exprLines[extractedResult]...)
|
|
}
|
|
}
|
|
result.Lines = sliceutil.Dedupe(result.Lines)
|
|
}
|
|
}
|
|
return event
|
|
}
|
|
|
|
func dumpResponse(event *output.InternalWrappedEvent, requestOptions *protocols.ExecuterOptions, filematches []FileMatch, filePath string) {
|
|
cliOptions := requestOptions.Options
|
|
if cliOptions.Debug || cliOptions.DebugResponse {
|
|
for _, fileMatch := range filematches {
|
|
lineContent := fileMatch.Raw
|
|
hexDump := false
|
|
if responsehighlighter.HasBinaryContent(lineContent) {
|
|
hexDump = true
|
|
lineContent = hex.Dump([]byte(lineContent))
|
|
}
|
|
highlightedResponse := responsehighlighter.Highlight(event.OperatorsResult, lineContent, cliOptions.NoColor, hexDump)
|
|
gologger.Debug().Msgf("[%s] Dumped match/extract file snippet for %s at line %d\n\n%s", requestOptions.TemplateID, filePath, fileMatch.Line, highlightedResponse)
|
|
}
|
|
}
|
|
}
|