github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-21 05:53:20 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update http_proxy_js_request.go

Browse source
This commit is contained in:
yungtravla 2018-03-21 19:47:19 +10:00 • committed by GitHub
commit 78ae5095c4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

22
modules/http_proxy_js_request.go
View file

@ -34,11 +34,11 @@ func NewJSRequest(req *http.Request) *JSRequest {
headers := make([]JSHeader, 0) headers := make([]JSHeader, 0)
cType := "" cType := ""
for key, values := range req.Header { for name, values := range req.Header {
for _, value := range values { for _, value := range values {
headers = append(headers, JSHeader{key, value}) headers = append(headers, JSHeader{name, value})
if key == "Content-Type" { if name == "Content-Type" {
cType = value cType = value
} }
} }
@ -88,7 +88,7 @@ func (j *JSRequest) WasModified() bool {
return false return false
} }
func (j *JSRequest) Header(name, deflt string) string { func (j *JSRequest) GetHeader(name, deflt string) string {
name = strings.ToLower(name) name = strings.ToLower(name)
for _, h := range j.Headers { for _, h := range j.Headers {
if name == strings.ToLower(h.Name) { if name == strings.ToLower(h.Name) {
@ -98,6 +98,20 @@ func (j *JSRequest) Header(name, deflt string) string {
return deflt return deflt
} }
func (j *JSRequest) SetHeader(name, value string) {
name = strings.ToLower(name)
found := false
for _, h := range j.Headers {
if name == strings.ToLower(h.Name) {
found = true
h.Value = value
}
}
if found == false {
j.Headers = append(j.Headers, JSHeader{name, value})
}
}
func (j *JSRequest) ReadBody() string { func (j *JSRequest) ReadBody() string {
raw, err := ioutil.ReadAll(j.req.Body) raw, err := ioutil.ReadAll(j.req.Body)
if err != nil { if err != nil {