mirror of https://github.com/daffainfo/nuclei.git
74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
package requests
|
|
|
|
import (
|
|
"bytes"
|
|
"compress/gzip"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"strings"
|
|
)
|
|
|
|
const (
|
|
markerParenthesisOpen = "{{"
|
|
markerParenthesisClose = "}}"
|
|
markerGeneral = "§"
|
|
)
|
|
|
|
func newReplacer(values map[string]interface{}) *strings.Replacer {
|
|
var replacerItems []string
|
|
for key, val := range values {
|
|
replacerItems = append(
|
|
replacerItems,
|
|
fmt.Sprintf("%s%s%s", markerParenthesisOpen, key, markerParenthesisClose),
|
|
fmt.Sprintf("%s", val),
|
|
fmt.Sprintf("%s%s%s", markerGeneral, key, markerGeneral),
|
|
fmt.Sprintf("%s", val),
|
|
)
|
|
}
|
|
|
|
return strings.NewReplacer(replacerItems...)
|
|
}
|
|
|
|
// HandleDecompression if the user specified a custom encoding (as golang transport doesn't do this automatically)
|
|
func HandleDecompression(r *HTTPRequest, bodyOrig []byte) (bodyDec []byte, err error) {
|
|
if r.Request == nil {
|
|
return bodyOrig, nil
|
|
}
|
|
|
|
encodingHeader := strings.ToLower(r.Request.Header.Get("Accept-Encoding"))
|
|
if encodingHeader == "gzip" {
|
|
gzipreader, err := gzip.NewReader(bytes.NewReader(bodyOrig))
|
|
if err != nil {
|
|
return bodyDec, err
|
|
}
|
|
defer gzipreader.Close()
|
|
|
|
bodyDec, err = ioutil.ReadAll(gzipreader)
|
|
if err != nil {
|
|
return bodyDec, err
|
|
}
|
|
|
|
return bodyDec, nil
|
|
}
|
|
|
|
return bodyOrig, nil
|
|
}
|
|
|
|
// ZipMapValues converts values from strings slices to flat string
|
|
func ZipMapValues(m map[string][]string) (m1 map[string]string) {
|
|
m1 = make(map[string]string)
|
|
for k, v := range m {
|
|
m1[k] = strings.Join(v, "")
|
|
}
|
|
return
|
|
}
|
|
|
|
// ExpandMapValues converts values from flat string to strings slice
|
|
func ExpandMapValues(m map[string]string) (m1 map[string][]string) {
|
|
m1 = make(map[string][]string)
|
|
for k, v := range m {
|
|
m1[k] = []string{v}
|
|
}
|
|
return
|
|
}
|