diff --git a/modules/dns_proxy/dns_proxy_script.go b/modules/dns_proxy/dns_proxy_script.go index 4a608168..83dd6777 100644 --- a/modules/dns_proxy/dns_proxy_script.go +++ b/modules/dns_proxy/dns_proxy_script.go @@ -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 } }