github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap.git synced 2025-03-12 04:36:03 -07:00
Code Issues Packages Projects Releases Wiki Activity

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

10
modules/dns_proxy/dns_proxy_base.go

@ -14,6 +14,8 @@ import (
"github.com/evilsocket/islazy/log"
"github.com/miekg/dns"
"github.com/robertkrimen/otto"
)
const (
@ -225,6 +227,14 @@ func (p *DNSProxy) Start() {
}
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 {
p.Debug("disabling redirection %s", p.Redirection.String())
if err := p.Sess.Firewall.EnableRedirection(p.Redirection, false); err != nil {