mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-18 16:07:23 +00:00
f7fe99f806
* add flow logic * progress * working POC * fix string slice normalization issue in variables * update * fix nil panic * remove poll() * load file with sandbox and more * fix failing integration tests * JS: log: print in vardump format * fix missing id in protocols * fix proto prefix in template context * flow: add unit tests * conditional flow support using flow * fix proto callbacks + more unit tests * adds integration test * conditional flow: check if req has any matchers * fix lint error * deprecate iterate-all+ missing multi-proto implementation * fix ip input in raw request * JS: feat dedupe object+ more builtin funcs * feat: hide protocol result using hide * feat: async execution * complete async execution support * fix condition-flow without any matchers * refactor: template executer package (tmplexec) * flow executor working * fix data race in templateCtx * templateCtx redesign * fix failing unit test * add multiprotocol support to deprecated syntax * fix race condition in utils & tlsx * add documentation in flow package * remove regions.txt file * fix minor issue with self contained templates * fix typos of copilot * dep + misc update * fix reqID: use req.Type instead of template.Type --------- Co-authored-by: sandeep <8293321+ehsandeep@users.noreply.github.com>
89 lines
2.4 KiB
Go
89 lines
2.4 KiB
Go
package contextargs
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/md5"
|
|
"fmt"
|
|
"strings"
|
|
|
|
jsoniter "github.com/json-iterator/go"
|
|
)
|
|
|
|
// MetaInput represents a target with metadata (TODO: replace with https://github.com/projectdiscovery/metainput)
|
|
type MetaInput struct {
|
|
// Input represent the target
|
|
Input string `json:"input,omitempty"`
|
|
// CustomIP to use for connection
|
|
CustomIP string `json:"customIP,omitempty"`
|
|
// hash of the input
|
|
hash string `json:"-"`
|
|
}
|
|
|
|
func (metaInput *MetaInput) marshalToBuffer() (bytes.Buffer, error) {
|
|
var b bytes.Buffer
|
|
err := jsoniter.NewEncoder(&b).Encode(metaInput)
|
|
return b, err
|
|
}
|
|
|
|
// ID returns a unique id/hash for metainput
|
|
func (metaInput *MetaInput) ID() string {
|
|
if metaInput.CustomIP != "" {
|
|
return fmt.Sprintf("%s-%s", metaInput.Input, metaInput.CustomIP)
|
|
}
|
|
return metaInput.Input
|
|
}
|
|
|
|
func (metaInput *MetaInput) MarshalString() (string, error) {
|
|
b, err := metaInput.marshalToBuffer()
|
|
return b.String(), err
|
|
}
|
|
|
|
func (metaInput *MetaInput) MustMarshalString() string {
|
|
marshaled, _ := metaInput.MarshalString()
|
|
return marshaled
|
|
}
|
|
|
|
func (metaInput *MetaInput) MarshalBytes() ([]byte, error) {
|
|
b, err := metaInput.marshalToBuffer()
|
|
return b.Bytes(), err
|
|
}
|
|
|
|
func (metaInput *MetaInput) MustMarshalBytes() []byte {
|
|
marshaled, _ := metaInput.MarshalBytes()
|
|
return marshaled
|
|
}
|
|
|
|
func (metaInput *MetaInput) Unmarshal(data string) error {
|
|
return jsoniter.NewDecoder(strings.NewReader(data)).Decode(metaInput)
|
|
}
|
|
|
|
func (metaInput *MetaInput) Clone() *MetaInput {
|
|
return &MetaInput{
|
|
Input: metaInput.Input,
|
|
CustomIP: metaInput.CustomIP,
|
|
}
|
|
}
|
|
|
|
func (metaInput *MetaInput) PrettyPrint() string {
|
|
if metaInput.CustomIP != "" {
|
|
return fmt.Sprintf("%s [%s]", metaInput.Input, metaInput.CustomIP)
|
|
}
|
|
return metaInput.Input
|
|
}
|
|
|
|
// GetScanHash returns a unique hash that represents a scan by hashing (metainput + templateId)
|
|
func (metaInput *MetaInput) GetScanHash(templateId string) string {
|
|
// there may be some cases where metainput is changed ex: while executing self-contained template etc
|
|
// but that totally changes the scanID/hash so to avoid that we compute hash only once
|
|
// and reuse it for all subsequent calls
|
|
if metaInput.hash == "" {
|
|
metaInput.hash = getMd5Hash(templateId + ":" + metaInput.Input + ":" + metaInput.CustomIP)
|
|
}
|
|
return metaInput.hash
|
|
}
|
|
|
|
func getMd5Hash(data string) string {
|
|
bin := md5.Sum([]byte(data))
|
|
return string(bin[:])
|
|
}
|