github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-20 21:43:18 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #1176 from bettercap/reduce_overhead

Browse source 
Reduce overhead for proxied HTTP/DNS packets
This commit is contained in:
Simone Margaritelli 2025-03-13 14:07:43 +01:00 • committed by GitHub
commit fc02767e72
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 48 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

8
modules/dns_proxy/dns_proxy_js_query.go
View file

@ -355,3 +355,11 @@ func (j *JSQuery) WasModified() bool {
// check if any of the fields has been changed
return j.NewHash() != j.refHash
}
func (j *JSQuery) CheckIfModifiedAndUpdateHash() bool {
// check if query was changed and update its hash
newHash := j.NewHash()
wasModified := j.refHash != newHash
j.refHash = newHash
return wasModified
}

9
modules/dns_proxy/dns_proxy_script.go
View file

@ -84,11 +84,9 @@ func (s *DnsProxyScript) OnRequest(req *dns.Msg, clientIP string) (jsreq, jsres
if _, err := s.Call("onRequest", jsreq, jsres); err != nil {
log.Error("%s", err)
return nil, nil
} else if jsreq.WasModified() {
jsreq.UpdateHash()
} else if jsreq.CheckIfModifiedAndUpdateHash() {
return jsreq, nil
} else if jsres.WasModified() {
jsres.UpdateHash()
} else if jsres.CheckIfModifiedAndUpdateHash() {
return nil, jsres
}
}
@ -104,8 +102,7 @@ func (s *DnsProxyScript) OnResponse(req, res *dns.Msg, clientIP string) (jsreq,
if _, err := s.Call("onResponse", jsreq, jsres); err != nil {
log.Error("%s", err)
return nil, nil
} else if jsres.WasModified() {
jsres.UpdateHash()
} else if jsres.CheckIfModifiedAndUpdateHash() {
return nil, jsres
}
}