Modify MasterFile to be Extensible (#62)

* Modify MasterFile to be extensible Co-authored-by: malcomvetter <malcomvetter@gmail.com> Co-authored-by: Mirza Kapetanovic <mirza.kapetanovic@gmail.com>
kapetan · Nov 27, 2021 · 8899fc6 · 8899fc6
1 parent 3334793
commit 8899fc6
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/DNS/Server/MasterFile.cs b/DNS/Server/MasterFile.cs
@@ -11,9 +11,9 @@
 
 namespace DNS.Server {
     public class MasterFile : IRequestResolver {
-        private static readonly TimeSpan DEFAULT_TTL = new TimeSpan(0);
+        protected static readonly TimeSpan DEFAULT_TTL = new TimeSpan(0);
 
-        private static bool Matches(Domain domain, Domain entry) {
+        protected static bool Matches(Domain domain, Domain entry) {
             string[] labels = entry.ToString().Split('.');
             string[] patterns = new string[labels.Length];
 
@@ -26,14 +26,14 @@ private static bool Matches(Domain domain, Domain entry) {
             return re.IsMatch(domain.ToString());
         }
 
-        private static void Merge<T>(IList<T> l1, IList<T> l2) {
+        protected static void Merge<T>(IList<T> l1, IList<T> l2) {
             foreach (T obj in l2) {
                 l1.Add(obj);
             }
         }
 
-        private IList<IResourceRecord> entries = new List<IResourceRecord>();
-        private TimeSpan ttl = DEFAULT_TTL;
+        protected IList<IResourceRecord> entries = new List<IResourceRecord>();
+        protected TimeSpan ttl = DEFAULT_TTL;
 
         public MasterFile(TimeSpan ttl) {
             this.ttl = ttl;
@@ -113,11 +113,11 @@ public void AddServiceResourceRecord(string domain, ushort priority, ushort weig
             return Task.FromResult(response);
         }
 
-        private IList<IResourceRecord> Get(Domain domain, RecordType type) {
+        protected IList<IResourceRecord> Get(Domain domain, RecordType type) {
             return entries.Where(e => Matches(domain, e.Name) && (e.Type == type || type == RecordType.ANY)).ToList();
         }
 
-        private IList<IResourceRecord> Get(Question question) {
+        protected IList<IResourceRecord> Get(Question question) {
             return Get(question.Name, question.Type);
         }
     }