fergalmoran/DnsServer
1
0
Fork 0
mirror of https://github.com/fergalmoran/DnsServer.git synced 2025-12-31 13:59:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

PrimarySubDomainZone: updated code to use CommitAndIncrementSerial(). Implemented UpdateRecord().

Browse Source
This commit is contained in:
Shreyas Zare
2021-07-10 13:23:43 +05:30
parent 59b139f43e
commit 3da24cf3a2
1 changed files with 26 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
DnsServerCore/Dns/Zones/PrimarySubDomainZone.cs
View File

@@ -17,6 +17,7 @@ along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
using System;
using System.Collections.Generic;
using TechnitiumLibrary.Net.Dns;
@@ -33,7 +34,7 @@ namespace DnsServerCore.Dns.Zones
#region constructor
public PrimarySubDomainZone(PrimaryZone primaryZone, string name)
: base(name)
: base(primaryZone, name)
{
_primaryZone = primaryZone;
}
@@ -44,9 +45,10 @@ namespace DnsServerCore.Dns.Zones
public override void SetRecords(DnsResourceRecordType type, IReadOnlyList<DnsResourceRecord> records)
{
base.SetRecords(type, records);
if (!SetRecords(type, records, out IReadOnlyList<DnsResourceRecord> deletedRecords))
throw new DnsServerException("Failed to set records. Please try again.");
_primaryZone.IncrementSoaSerial();
_primaryZone.CommitAndIncrementSerial(deletedRecords, records);
_primaryZone.TriggerNotify();
}
@@ -54,15 +56,15 @@ namespace DnsServerCore.Dns.Zones
{
base.AddRecord(record);
_primaryZone.IncrementSoaSerial();
_primaryZone.CommitAndIncrementSerial(null, new DnsResourceRecord[] { record });
_primaryZone.TriggerNotify();
}
public override bool DeleteRecords(DnsResourceRecordType type)
{
if (base.DeleteRecords(type))
if (_entries.TryRemove(type, out IReadOnlyList<DnsResourceRecord> removedRecords))
{
_primaryZone.IncrementSoaSerial();
_primaryZone.CommitAndIncrementSerial(removedRecords);
_primaryZone.TriggerNotify();
return true;
@@ -71,11 +73,11 @@ namespace DnsServerCore.Dns.Zones
return false;
}
public override bool DeleteRecord(DnsResourceRecordType type, DnsResourceRecordData record)
public override bool DeleteRecord(DnsResourceRecordType type, DnsResourceRecordData rdata)
{
if (base.DeleteRecord(type, record))
if (DeleteRecord(type, rdata, out DnsResourceRecord deletedRecord))
{
_primaryZone.IncrementSoaSerial();
_primaryZone.CommitAndIncrementSerial(new DnsResourceRecord[] { deletedRecord });
_primaryZone.TriggerNotify();
return true;
@@ -84,6 +86,21 @@ namespace DnsServerCore.Dns.Zones
return false;
}
public override void UpdateRecord(DnsResourceRecord oldRecord, DnsResourceRecord newRecord)
{
if (oldRecord.Type == DnsResourceRecordType.SOA)
throw new InvalidOperationException("Cannot update record: use SetRecords() for " + oldRecord.Type.ToString() + " record");
if (oldRecord.Type != newRecord.Type)
throw new InvalidOperationException("Old and new record types do not match.");
DeleteRecord(oldRecord.Type, oldRecord.RDATA, out DnsResourceRecord deletedRecord);
base.AddRecord(newRecord);
_primaryZone.CommitAndIncrementSerial(new DnsResourceRecord[] { deletedRecord }, new DnsResourceRecord[] { newRecord });
_primaryZone.TriggerNotify();
}
#endregion
}
}