mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-18 13:55:27 +00:00
1fbbca66f9
* Adding support to scan all v4/v6 IPs * adding tests * metainput prototype * using new signature * fixing nil pointer * adding request context with metadata * removing log instruction * fixing merge conflicts * adding clone helpers * attempting to fix ipv6 square parenthesis wrap * fixing dialed ip info * fixing syntax * fixing output ip selection * adding integration tests * disabling test due to gh ipv6 issue * using ipv4 only due to GH limited networking * extending metainput marshaling * fixing hmap key * adding test for httpx integration * fixing lint error * reworking marshaling/id-calculation * adding ip version validation * improving handling non url targets * fixing condition check
70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
package contextargs
|
|
|
|
import (
|
|
"bytes"
|
|
"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
|
|
// CustomIP to use for connection
|
|
CustomIP string
|
|
}
|
|
|
|
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
|
|
}
|