github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-07-06 04:52:10 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added dropCallback to drop packets instead of just changing it

Browse source
This commit is contained in:
realgam3 2019-09-06 02:31:48 +03:00
parent 709232dba2
commit 11d2756283
3 changed files with 12 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

10
modules/tcp_proxy/tcp_proxy.go
View file

@ -8,6 +8,8 @@ import (
"github.com/bettercap/bettercap/firewall"
"github.com/bettercap/bettercap/session"
"github.com/robertkrimen/otto"
)
type TcpProxy struct {
@ -150,7 +152,7 @@ func (mod *TcpProxy) Configure() error {
return nil
}
func (mod *TcpProxy) doPipe(from, to net.Addr, src, dst io.ReadWriter, wg *sync.WaitGroup) {
func (mod *TcpProxy) doPipe(from, to net.Addr, src *net.TCPConn, dst io.ReadWriter, wg *sync.WaitGroup) {
defer wg.Done()
buff := make([]byte, 0xffff)
@ -165,7 +167,11 @@ func (mod *TcpProxy) doPipe(from, to net.Addr, src, dst io.ReadWriter, wg *sync.
b := buff[:n]
if mod.script != nil {
ret := mod.script.OnData(from, to, b)
ret := mod.script.OnData(from, to, b, func(call otto.FunctionCall) otto.Value {
mod.Debug("onData dropCallback called")
src.Close()
return otto.Value{}
})
if ret != nil {
nret := len(ret)