Skip to content

Commit

Permalink
Merge pull request #60 from cfsdes/master
Browse files Browse the repository at this point in the history
fix: Add HTTP/2 support
  • Loading branch information
j3ssie authored Jul 8, 2023
2 parents d824b0b + 44757fc commit 8d274e0
Show file tree
Hide file tree
Showing 2 changed files with 80 additions and 30 deletions.
109 changes: 79 additions & 30 deletions core/parser.go
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ import (
"path"
"path/filepath"
"strings"
"bytes"

"github.com/jaeles-project/jaeles/libs"
"github.com/thoas/go-funk"
Expand Down Expand Up @@ -451,45 +452,93 @@ func ParsePayloads(sign libs.Signature) []string {
func ParseBurpRequest(raw string) (req libs.Request) {
var realReq libs.Request
realReq.Raw = raw
reader := bufio.NewReader(strings.NewReader(raw))
parsedReq, err := http.ReadRequest(reader)
if err != nil {
return realReq
}
realReq.Method = parsedReq.Method
// URL part
if parsedReq.URL.Host == "" {
realReq.Host = parsedReq.Host
parsedReq.URL.Host = parsedReq.Host
}
if parsedReq.URL.Scheme == "" {
if parsedReq.Referer() == "" {
realReq.Scheme = "https"
parsedReq.URL.Scheme = "https"
} else {
u, err := url.Parse(parsedReq.Referer())
if err == nil {
realReq.Scheme = u.Scheme
parsedReq.URL.Scheme = u.Scheme
}

// create a scanner to read the request
scanner := bufio.NewScanner(strings.NewReader(raw))

// read the first request line that contains HTTP version, method and path
scanner.Scan()
requestLine := scanner.Text()
parts := strings.Split(requestLine, " ")
if len(parts) != 3 {
return
}
httpMethod := parts[0]
httpVersion := parts[2]
path := parts[1]

// create a new buffer for the request body
bodyBuffer := bytes.NewBuffer([]byte{})

// create a new map to hold the headers
headers := make([]map[string]string, 0)

// read the headers
for scanner.Scan() {
line := scanner.Text()
if line == "" {
break
}

parts := strings.Split(line, ": ")
if len(parts) != 2 {
return
}
headerName := parts[0]
headerValue := parts[1]

header := map[string]string{
headerName: headerValue,
}
headers = append(headers, header)
}

// read the request body
for scanner.Scan() {
bodyBuffer.Write(scanner.Bytes())
bodyBuffer.Write([]byte("\r\n"))
}
realReq.URL = parsedReq.URL.String()
realReq.Path = parsedReq.RequestURI
realReq.Headers = ParseHeaders(parsedReq.Header)

body, _ := ioutil.ReadAll(parsedReq.Body)
realReq.Body = string(body)
// net/http parse something weird here
if !strings.HasSuffix(raw, realReq.Body) {
if strings.Contains(raw, "\n\n") {
realReq.Body = strings.Split(raw, "\n\n")[1]
body := bodyBuffer.String()

var urlScheme string
referer := headersGet(headers, "Referer")
if referer != "" {
refURL, err := url.Parse(referer)
if err == nil {
urlScheme = refURL.Scheme
}
}

if urlScheme == "" {
urlScheme = "https"
}

url := urlScheme + "://" + headers[0]["Host"] + path

realReq.Body = body
realReq.Headers = headers
realReq.Path = path
realReq.URL = url
realReq.Method = httpMethod
realReq.Scheme = urlScheme
realReq.Proto = httpVersion

return realReq
}

// helper function to get header value
func headersGet(headers []map[string]string, headerName string) string {
for _, header := range headers {
for k, v := range header {
if strings.ToLower(k) == strings.ToLower(headerName) {
return v
}
}
}
return ""
}

// ParseHeaders parse header for sending method
func ParseHeaders(rawHeaders map[string][]string) []map[string]string {
var headers []map[string]string
Expand Down
1 change: 1 addition & 0 deletions libs/http.go
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@ type Request struct {
Port string
Path string
URL string
Proto string
Proxy string
Method string
Payload string
Expand Down

0 comments on commit 8d274e0

Please sign in to comment.