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

refact: several fixes and optimizations to the proxy

Browse source
This commit is contained in:
evilsocket 2018-01-07 23:22:33 +01:00
commit 7248dd701a
3 changed files with 28 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

21
session/modules/http_proxy.go
View file

@ -9,6 +9,7 @@ import (
"github.com/elazarl/goproxy"
"github.com/evilsocket/bettercap-ng/core"
"github.com/evilsocket/bettercap-ng/firewall"
"github.com/evilsocket/bettercap-ng/session"
)
@ -25,6 +26,16 @@ type HttpProxy struct {
script *ProxyScript
}
func (p HttpProxy) logAction(req *http.Request, jsres *JSResponse) {
fmt.Printf("[%s] %s > '%s %s%s' | Sending %d bytes of spoofed response.\n",
core.Green("http.proxy"),
core.Bold(strings.Split(req.RemoteAddr, ":")[0]),
req.Method,
req.Host,
req.URL.Path,
len(jsres.Body))
}
func NewHttpProxy(s *session.Session) *HttpProxy {
p := &HttpProxy{
SessionModule: session.NewSessionModule(s),
@ -51,7 +62,6 @@ func NewHttpProxy(s *session.Session) *HttpProxy {
}))
proxy := goproxy.NewProxyHttpServer()
proxy.NonproxyHandler = http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
if p.doProxy(req) == true {
req.URL.Scheme = "http"
@ -65,9 +75,8 @@ func NewHttpProxy(s *session.Session) *HttpProxy {
if p.script != nil {
jsres := p.script.OnRequest(req)
if jsres != nil {
log.Infof("Sending %d bytes of spoofed response to %s.", len(jsres.Body), req.RemoteAddr)
resp := jsres.ToResponse(req)
return req, resp
p.logAction(req, jsres)
return req, jsres.ToResponse(req)
}
}
return req, nil
@ -77,8 +86,8 @@ func NewHttpProxy(s *session.Session) *HttpProxy {
if p.script != nil {
jsres := p.script.OnResponse(res)
if jsres != nil {
log.Infof("Sending %d bytes of spoofed response to %s.", len(jsres.Body), res.Request.RemoteAddr)
res = jsres.ToResponse(res.Request)
p.logAction(res.Request, jsres)
return jsres.ToResponse(res.Request)
}
}
return res