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

Implement DNS proxy script onExit call.

Browse source
This commit is contained in:
☸️ 2024-11-23 16:02:28 +01:00 • committed by GitHub
parent 91f5213526
commit fdca49678e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 10 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 {