mirror of
https://github.com/bettercap/bettercap.git
synced 2024-09-18 14:40:21 -07:00
67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
package tcp_proxy
|
|
|
|
import (
|
|
"net"
|
|
"strings"
|
|
|
|
"github.com/bettercap/bettercap/v2/log"
|
|
"github.com/bettercap/bettercap/v2/session"
|
|
|
|
"github.com/evilsocket/islazy/plugin"
|
|
|
|
"github.com/robertkrimen/otto"
|
|
)
|
|
|
|
type TcpProxyScript struct {
|
|
*plugin.Plugin
|
|
doOnData bool
|
|
}
|
|
|
|
func LoadTcpProxyScript(path string, sess *session.Session) (err error, s *TcpProxyScript) {
|
|
log.Info("loading tcp proxy script %s ...", path)
|
|
|
|
plug, err := plugin.Load(path)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
// define session pointer
|
|
if err = plug.Set("env", sess.Env.Data); err != nil {
|
|
log.Error("error while defining environment: %+v", err)
|
|
return
|
|
}
|
|
|
|
// run onLoad if defined
|
|
if plug.HasFunc("onLoad") {
|
|
if _, err = plug.Call("onLoad"); err != nil {
|
|
log.Error("error while executing onLoad callback: %s", "\ntraceback:\n "+err.(*otto.Error).String())
|
|
return
|
|
}
|
|
}
|
|
|
|
s = &TcpProxyScript{
|
|
Plugin: plug,
|
|
doOnData: plug.HasFunc("onData"),
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *TcpProxyScript) OnData(from, to net.Addr, data []byte, callback func(call otto.FunctionCall) otto.Value) []byte {
|
|
if s.doOnData {
|
|
addrFrom := strings.Split(from.String(), ":")[0]
|
|
addrTo := strings.Split(to.String(), ":")[0]
|
|
|
|
if ret, err := s.Call("onData", addrFrom, addrTo, data, callback); err != nil {
|
|
log.Error("error while executing onData callback: %s", err)
|
|
return nil
|
|
} else if ret != nil {
|
|
array, ok := ret.([]byte)
|
|
if !ok {
|
|
log.Error("error while casting exported value to array of byte: value = %+v", ret)
|
|
}
|
|
return array
|
|
}
|
|
}
|
|
return nil
|
|
}
|