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

new: now request object can be changed during OnRequest handlers (closes #135)

Browse source
This commit is contained in:
evilsocket 2018-03-05 16:48:45 +01:00
commit 07fe790ad1
3 changed files with 111 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

31
modules/http_proxy_base.go
View file

@ -77,9 +77,12 @@ func NewHTTPProxy(s *session.Session) *HTTPProxy {
p.Proxy.OnRequest().DoFunc(func(req *http.Request, ctx *goproxy.ProxyCtx) (*http.Request, *http.Response) {
log.Debug("(%s) < %s %s %s%s", core.Green(p.Name), req.RemoteAddr, req.Method, req.Host, req.URL.Path)
if p.Script != nil {
jsres := p.Script.OnRequest(req)
if jsres != nil {
p.logAction(req, jsres)
jsreq, jsres := p.Script.OnRequest(req)
if jsreq != nil {
p.logRequestAction(req, jsreq)
return jsreq.ToRequest(), nil
} else if jsres != nil {
p.logResponseAction(req, jsres)
return req, jsres.ToResponse(req)
}
}
@ -91,9 +94,9 @@ func NewHTTPProxy(s *session.Session) *HTTPProxy {
req := res.Request
log.Debug("(%s) > %s %s %s%s", core.Green(p.Name), req.RemoteAddr, req.Method, req.Host, req.URL.Path)
if p.Script != nil {
jsres := p.Script.OnResponse(res)
_, jsres := p.Script.OnResponse(res)
if jsres != nil {
p.logAction(res.Request, jsres)
p.logResponseAction(res.Request, jsres)
return jsres.ToResponse(res.Request)
}
}
@ -104,7 +107,23 @@ func NewHTTPProxy(s *session.Session) *HTTPProxy {
return p
}
func (p *HTTPProxy) logAction(req *http.Request, jsres *JSResponse) {
func (p *HTTPProxy) logRequestAction(req *http.Request, jsreq *JSRequest) {
p.sess.Events.Add(p.Name+".spoofed-request", struct {
To string
Method string
Host string
Path string
Size int
}{
strings.Split(req.RemoteAddr, ":")[0],
jsreq.Method,
jsreq.Hostname,
jsreq.Path,
len(jsreq.Body),
})
}
func (p *HTTPProxy) logResponseAction(req *http.Request, jsres *JSResponse) {
p.sess.Events.Add(p.Name+".spoofed-response", struct {
To string
Method string