From fe8da4e5c8db28a07177acac3b096fa94f875394 Mon Sep 17 00:00:00 2001
From: Shreyas Zare <shreyas@technitium.com>
Date: Sat, 27 Jun 2020 11:58:37 +0530
Subject: [PATCH] ResolverPrefetchDnsCache: updated class to inherit
 ResolverDnsCache and allow using auth zone for recursive resolution.

---
 DnsServerCore/Dns/ResolverPrefetchDnsCache.cs | 16 +++++-----------
 1 file changed, 5 insertions(+), 11 deletions(-)

diff --git a/DnsServerCore/Dns/ResolverPrefetchDnsCache.cs b/DnsServerCore/Dns/ResolverPrefetchDnsCache.cs
index 899d61dc..6eda39b7 100644
--- a/DnsServerCore/Dns/ResolverPrefetchDnsCache.cs
+++ b/DnsServerCore/Dns/ResolverPrefetchDnsCache.cs
@@ -22,20 +22,19 @@ using TechnitiumLibrary.Net.Dns;
 
 namespace DnsServerCore.Dns
 {
-    class ResolverPrefetchDnsCache : IDnsCache
+    class ResolverPrefetchDnsCache : ResolverDnsCache
     {
         #region variables
 
-        readonly CacheZoneManager _cacheZoneManager;
         readonly DnsQuestionRecord _prefetchQuery;
 
         #endregion
 
         #region constructor
 
-        public ResolverPrefetchDnsCache(CacheZoneManager cacheZoneManager, DnsQuestionRecord prefetchQuery)
+        public ResolverPrefetchDnsCache(AuthZoneManager authZoneManager, CacheZoneManager cacheZoneManager, DnsQuestionRecord prefetchQuery)
+            : base(authZoneManager, cacheZoneManager)
         {
-            _cacheZoneManager = cacheZoneManager;
             _prefetchQuery = prefetchQuery;
         }
 
@@ -43,7 +42,7 @@ namespace DnsServerCore.Dns
 
         #region public
 
-        public DnsDatagram Query(DnsDatagram request, bool serveStale = false)
+        public override DnsDatagram Query(DnsDatagram request, bool serveStale = false)
         {
             if (_prefetchQuery.Equals(request.Question[0]))
             {
@@ -51,12 +50,7 @@ namespace DnsServerCore.Dns
                 return _cacheZoneManager.QueryClosestDelegation(request);
             }
 
-            return _cacheZoneManager.Query(request, serveStale);
-        }
-
-        public void CacheResponse(DnsDatagram response)
-        {
-            _cacheZoneManager.CacheResponse(response);
+            return base.Query(request, serveStale);
         }
 
         #endregion