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

Merge pull request #1156 from bettercap/otto_onExit

Browse source 
Implement onExit otto function calls when quitting the session or modules.
This commit is contained in:
buffermet 2025-03-01 23:12:51 +01:00 • committed by GitHub
commit 2c6f048cec
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 31 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
modules/dns_proxy/dns_proxy_base.go
View file

@ -14,6 +14,8 @@ import (
"github.com/evilsocket/islazy/log" "github.com/evilsocket/islazy/log"
"github.com/miekg/dns" "github.com/miekg/dns"
"github.com/robertkrimen/otto"
) )
const ( const (
@ -225,6 +227,14 @@ func (p *DNSProxy) Start() {
} }
func (p *DNSProxy) Stop() error { func (p *DNSProxy) Stop() error {
if p.Script != nil {
if p.Script.Plugin.HasFunc("onExit") {
if _, err := p.Script.Call("onExit"); err != nil {
log.Error("Error while executing onExit callback: %s", "\nTraceback:\n "+err.(*otto.Error).String())
}
}
}
if p.doRedirect && p.Redirection != nil { if p.doRedirect && p.Redirection != nil {
p.Debug("disabling redirection %s", p.Redirection.String()) p.Debug("disabling redirection %s", p.Redirection.String())
if err := p.Sess.Firewall.EnableRedirection(p.Redirection, false); err != nil { if err := p.Sess.Firewall.EnableRedirection(p.Redirection, false); err != nil {

10
modules/http_proxy/http_proxy_base.go
View file

@ -27,6 +27,8 @@ import (
"github.com/evilsocket/islazy/log" "github.com/evilsocket/islazy/log"
"github.com/evilsocket/islazy/str" "github.com/evilsocket/islazy/str"
"github.com/evilsocket/islazy/tui" "github.com/evilsocket/islazy/tui"
"github.com/robertkrimen/otto"
) )
const ( const (
@ -432,6 +434,14 @@ func (p *HTTPProxy) Start() {
} }
func (p *HTTPProxy) Stop() error { func (p *HTTPProxy) Stop() error {
if p.Script != nil {
if p.Script.Plugin.HasFunc("onExit") {
if _, err := p.Script.Call("onExit"); err != nil {
log.Error("Error while executing onExit callback: %s", "\nTraceback:\n "+err.(*otto.Error).String())
}
}
}
if p.doRedirect && p.Redirection != nil { if p.doRedirect && p.Redirection != nil {
p.Debug("disabling redirection %s", p.Redirection.String()) p.Debug("disabling redirection %s", p.Redirection.String())
if err := p.Sess.Firewall.EnableRedirection(p.Redirection, false); err != nil { if err := p.Sess.Firewall.EnableRedirection(p.Redirection, false); err != nil {

11
session/session_core_handlers.go
View file

@ -13,11 +13,14 @@ import (
"time" "time"
"github.com/bettercap/bettercap/v2/core" "github.com/bettercap/bettercap/v2/core"
"github.com/bettercap/bettercap/v2/log"
"github.com/bettercap/bettercap/v2/network" "github.com/bettercap/bettercap/v2/network"
"github.com/bettercap/readline" "github.com/bettercap/readline"
"github.com/evilsocket/islazy/str" "github.com/evilsocket/islazy/str"
"github.com/evilsocket/islazy/tui" "github.com/evilsocket/islazy/tui"
"github.com/robertkrimen/otto"
) )
func (s *Session) generalHelp() { func (s *Session) generalHelp() {
@ -155,6 +158,14 @@ func (s *Session) activeHandler(args []string, sess *Session) error {
} }
func (s *Session) exitHandler(args []string, sess *Session) error { func (s *Session) exitHandler(args []string, sess *Session) error {
if s.script != nil {
if s.script.Plugin.HasFunc("onExit") {
if _, err := s.script.Plugin.Call("onExit"); err != nil {
log.Error("Error while executing onExit callback: %s", "\nTraceback:\n "+err.(*otto.Error).String())
}
}
}
// notify any listener that the session is about to end // notify any listener that the session is about to end
s.Events.Add("session.stopped", nil) s.Events.Add("session.stopped", nil)