Files
DnsServer/DnsServerCore/Dns/Zones/AuthZoneInfo.cs
2022-09-18 17:27:30 +05:30

694 lines
21 KiB
C#

/*
Technitium DNS Server
Copyright (C) 2022 Shreyas Zare (shreyas@technitium.com)
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
using DnsServerCore.Dns.Dnssec;
using DnsServerCore.Dns.ResourceRecords;
using System;
using System.Collections.Generic;
using System.IO;
using System.Net;
using System.Threading.Tasks;
using TechnitiumLibrary.IO;
using TechnitiumLibrary.Net;
using TechnitiumLibrary.Net.Dns;
using TechnitiumLibrary.Net.Dns.ResourceRecords;
namespace DnsServerCore.Dns.Zones
{
public enum AuthZoneType : byte
{
Unknown = 0,
Primary = 1,
Secondary = 2,
Stub = 3,
Forwarder = 4
}
public sealed class AuthZoneInfo : IComparable<AuthZoneInfo>
{
#region variables
readonly ApexZone _apexZone;
readonly string _name;
readonly AuthZoneType _type;
readonly bool _disabled;
readonly AuthZoneTransfer _zoneTransfer;
readonly IReadOnlyCollection<IPAddress> _zoneTransferNameServers;
readonly AuthZoneNotify _notify;
readonly IReadOnlyCollection<IPAddress> _notifyNameServers;
readonly DateTime _expiry;
readonly IReadOnlyList<DnsResourceRecord> _zoneHistory; //for IXFR support
readonly IReadOnlyDictionary<string, object> _tsigKeyNames;
readonly IReadOnlyCollection<DnssecPrivateKey> _dnssecPrivateKeys;
readonly bool _notifyFailed; //not serialized
readonly bool _syncFailed; //not serialized
#endregion
#region constructor
public AuthZoneInfo(string name, AuthZoneType type, bool disabled)
{
_name = name;
_type = type;
_disabled = disabled;
switch (_type)
{
case AuthZoneType.Primary:
_zoneTransfer = AuthZoneTransfer.AllowOnlyZoneNameServers;
_notify = AuthZoneNotify.ZoneNameServers;
break;
default:
_zoneTransfer = AuthZoneTransfer.Deny;
_notify = AuthZoneNotify.None;
break;
}
}
public AuthZoneInfo(BinaryReader bR)
{
byte version = bR.ReadByte();
switch (version)
{
case 1:
case 2:
case 3:
case 4:
case 5:
_name = bR.ReadShortString();
_type = (AuthZoneType)bR.ReadByte();
_disabled = bR.ReadBoolean();
if (version >= 2)
{
{
_zoneTransfer = (AuthZoneTransfer)bR.ReadByte();
int count = bR.ReadByte();
if (count > 0)
{
IPAddress[] nameServers = new IPAddress[count];
for (int i = 0; i < count; i++)
nameServers[i] = IPAddressExtension.ReadFrom(bR);
_zoneTransferNameServers = nameServers;
}
}
{
_notify = (AuthZoneNotify)bR.ReadByte();
int count = bR.ReadByte();
if (count > 0)
{
IPAddress[] nameServers = new IPAddress[count];
for (int i = 0; i < count; i++)
nameServers[i] = IPAddressExtension.ReadFrom(bR);
_notifyNameServers = nameServers;
}
}
}
else
{
switch (_type)
{
case AuthZoneType.Primary:
_zoneTransfer = AuthZoneTransfer.AllowOnlyZoneNameServers;
_notify = AuthZoneNotify.ZoneNameServers;
break;
default:
_zoneTransfer = AuthZoneTransfer.Deny;
_notify = AuthZoneNotify.None;
break;
}
}
switch (_type)
{
case AuthZoneType.Primary:
if (version >= 3)
{
int count = bR.ReadInt32();
DnsResourceRecord[] zoneHistory = new DnsResourceRecord[count];
for (int i = 0; i < count; i++)
{
zoneHistory[i] = new DnsResourceRecord(bR.BaseStream);
zoneHistory[i].Tag = new DnsResourceRecordInfo(bR, zoneHistory[i].Type == DnsResourceRecordType.SOA);
}
_zoneHistory = zoneHistory;
}
if (version >= 4)
{
int count = bR.ReadByte();
Dictionary<string, object> tsigKeyNames = new Dictionary<string, object>(count);
for (int i = 0; i < count; i++)
tsigKeyNames.Add(bR.ReadShortString(), null);
_tsigKeyNames = tsigKeyNames;
}
if (version >= 5)
{
int count = bR.ReadByte();
if (count > 0)
{
List<DnssecPrivateKey> dnssecPrivateKeys = new List<DnssecPrivateKey>(count);
for (int i = 0; i < count; i++)
dnssecPrivateKeys.Add(DnssecPrivateKey.Parse(bR));
_dnssecPrivateKeys = dnssecPrivateKeys;
}
}
break;
case AuthZoneType.Secondary:
_expiry = bR.ReadDateTime();
if (version >= 4)
{
int count = bR.ReadInt32();
DnsResourceRecord[] zoneHistory = new DnsResourceRecord[count];
for (int i = 0; i < count; i++)
{
zoneHistory[i] = new DnsResourceRecord(bR.BaseStream);
zoneHistory[i].Tag = new DnsResourceRecordInfo(bR, zoneHistory[i].Type == DnsResourceRecordType.SOA);
}
_zoneHistory = zoneHistory;
}
if (version >= 4)
{
int count = bR.ReadByte();
Dictionary<string, object> tsigKeyNames = new Dictionary<string, object>(count);
for (int i = 0; i < count; i++)
tsigKeyNames.Add(bR.ReadShortString(), null);
_tsigKeyNames = tsigKeyNames;
}
break;
case AuthZoneType.Stub:
_expiry = bR.ReadDateTime();
break;
}
break;
default:
throw new InvalidDataException("AuthZoneInfo format version not supported.");
}
}
internal AuthZoneInfo(ApexZone apexZone, bool loadHistory = false)
{
_apexZone = apexZone;
_name = _apexZone.Name;
if (_apexZone is PrimaryZone primaryZone)
{
_type = AuthZoneType.Primary;
if (loadHistory)
_zoneHistory = primaryZone.GetHistory();
_tsigKeyNames = primaryZone.TsigKeyNames;
_dnssecPrivateKeys = primaryZone.DnssecPrivateKeys;
_notifyFailed = primaryZone.NotifyFailed;
}
else if (_apexZone is SecondaryZone secondaryZone)
{
_type = AuthZoneType.Secondary;
if (loadHistory)
_zoneHistory = secondaryZone.GetHistory();
_expiry = secondaryZone.Expiry;
_tsigKeyNames = secondaryZone.TsigKeyNames;
_notifyFailed = secondaryZone.NotifyFailed;
_syncFailed = secondaryZone.SyncFailed;
}
else if (_apexZone is StubZone stubZone)
{
_type = AuthZoneType.Stub;
_expiry = stubZone.Expiry;
_syncFailed = stubZone.SyncFailed;
}
else if (_apexZone is ForwarderZone)
{
_type = AuthZoneType.Forwarder;
}
else
{
_type = AuthZoneType.Unknown;
}
_disabled = _apexZone.Disabled;
_zoneTransfer = _apexZone.ZoneTransfer;
_zoneTransferNameServers = _apexZone.ZoneTransferNameServers;
_notify = _apexZone.Notify;
_notifyNameServers = _apexZone.NotifyNameServers;
}
#endregion
#region public
public IReadOnlyList<DnsResourceRecord> GetRecords(DnsResourceRecordType type)
{
if (_apexZone is null)
throw new InvalidOperationException();
return _apexZone.GetRecords(type);
}
public void TriggerNotify()
{
if (_apexZone is null)
throw new InvalidOperationException();
switch (_type)
{
case AuthZoneType.Primary:
(_apexZone as PrimaryZone).TriggerNotify();
break;
case AuthZoneType.Secondary:
(_apexZone as SecondaryZone).TriggerNotify();
break;
default:
throw new InvalidOperationException();
}
}
public void TriggerRefresh()
{
if (_apexZone is null)
throw new InvalidOperationException();
switch (_type)
{
case AuthZoneType.Secondary:
(_apexZone as SecondaryZone).TriggerRefresh();
break;
case AuthZoneType.Stub:
(_apexZone as StubZone).TriggerRefresh();
break;
default:
throw new InvalidOperationException();
}
}
public void TriggerResync()
{
if (_apexZone is null)
throw new InvalidOperationException();
switch (_type)
{
case AuthZoneType.Secondary:
(_apexZone as SecondaryZone).TriggerResync();
break;
case AuthZoneType.Stub:
(_apexZone as StubZone).TriggerResync();
break;
default:
throw new InvalidOperationException();
}
}
public Task<IReadOnlyList<NameServerAddress>> GetPrimaryNameServerAddressesAsync(DnsServer dnsServer)
{
if (_apexZone is null)
throw new InvalidOperationException();
return _apexZone.GetPrimaryNameServerAddressesAsync(dnsServer);
}
public Task<IReadOnlyList<NameServerAddress>> GetSecondaryNameServerAddressesAsync(DnsServer dnsServer)
{
if (_apexZone is null)
throw new InvalidOperationException();
return _apexZone.GetSecondaryNameServerAddressesAsync(dnsServer);
}
public void WriteTo(BinaryWriter bW)
{
if (_apexZone is null)
throw new InvalidOperationException();
bW.Write((byte)5); //version
bW.WriteShortString(_name);
bW.Write((byte)_type);
bW.Write(_disabled);
bW.Write((byte)_zoneTransfer);
if (_zoneTransferNameServers is null)
{
bW.Write((byte)0);
}
else
{
bW.Write(Convert.ToByte(_zoneTransferNameServers.Count));
foreach (IPAddress nameServer in _zoneTransferNameServers)
nameServer.WriteTo(bW);
}
bW.Write((byte)_notify);
if (_notifyNameServers is null)
{
bW.Write((byte)0);
}
else
{
bW.Write(Convert.ToByte(_notifyNameServers.Count));
foreach (IPAddress nameServer in _notifyNameServers)
nameServer.WriteTo(bW);
}
switch (_type)
{
case AuthZoneType.Primary:
if (_zoneHistory is null)
{
bW.Write(0);
}
else
{
bW.Write(_zoneHistory.Count);
foreach (DnsResourceRecord record in _zoneHistory)
{
record.WriteTo(bW.BaseStream);
if (record.Tag is not DnsResourceRecordInfo rrInfo)
rrInfo = new DnsResourceRecordInfo(); //default info
rrInfo.WriteTo(bW);
}
}
if (_tsigKeyNames is null)
{
bW.Write((byte)0);
}
else
{
bW.Write(Convert.ToByte(_tsigKeyNames.Count));
foreach (KeyValuePair<string, object> tsigKeyName in _tsigKeyNames)
bW.WriteShortString(tsigKeyName.Key);
}
if (_dnssecPrivateKeys is null)
{
bW.Write((byte)0);
}
else
{
bW.Write(Convert.ToByte(_dnssecPrivateKeys.Count));
foreach (DnssecPrivateKey dnssecPrivateKey in _dnssecPrivateKeys)
dnssecPrivateKey.WriteTo(bW);
}
break;
case AuthZoneType.Secondary:
bW.Write(_expiry);
if (_zoneHistory is null)
{
bW.Write(0);
}
else
{
bW.Write(_zoneHistory.Count);
foreach (DnsResourceRecord record in _zoneHistory)
{
record.WriteTo(bW.BaseStream);
if (record.Tag is not DnsResourceRecordInfo rrInfo)
rrInfo = new DnsResourceRecordInfo(); //default info
rrInfo.WriteTo(bW);
}
}
if (_tsigKeyNames is null)
{
bW.Write((byte)0);
}
else
{
bW.Write(Convert.ToByte(_tsigKeyNames.Count));
foreach (KeyValuePair<string, object> tsigKeyName in _tsigKeyNames)
bW.WriteShortString(tsigKeyName.Key);
}
break;
case AuthZoneType.Stub:
bW.Write(_expiry);
break;
}
}
public int CompareTo(AuthZoneInfo other)
{
return _name.CompareTo(other._name);
}
public override string ToString()
{
return _name;
}
#endregion
#region properties
internal ApexZone ApexZone
{ get { return _apexZone; } }
public string Name
{ get { return _name; } }
public AuthZoneType Type
{ get { return _type; } }
public bool Disabled
{
get { return _disabled; }
set
{
if (_apexZone is null)
throw new InvalidOperationException();
_apexZone.Disabled = value;
}
}
public AuthZoneTransfer ZoneTransfer
{
get { return _zoneTransfer; }
set
{
if (_apexZone is null)
throw new InvalidOperationException();
_apexZone.ZoneTransfer = value;
}
}
public IReadOnlyCollection<IPAddress> ZoneTransferNameServers
{
get { return _zoneTransferNameServers; }
set
{
if (_apexZone is null)
throw new InvalidOperationException();
_apexZone.ZoneTransferNameServers = value;
}
}
public AuthZoneNotify Notify
{
get { return _notify; }
set
{
if (_apexZone is null)
throw new InvalidOperationException();
_apexZone.Notify = value;
}
}
public IReadOnlyCollection<IPAddress> NotifyNameServers
{
get { return _notifyNameServers; }
set
{
if (_apexZone is null)
throw new InvalidOperationException();
_apexZone.NotifyNameServers = value;
}
}
public DateTime Expiry
{ get { return _expiry; } }
public bool IsExpired
{
get
{
if (_apexZone is null)
throw new InvalidOperationException();
switch (_type)
{
case AuthZoneType.Secondary:
return (_apexZone as SecondaryZone).IsExpired;
case AuthZoneType.Stub:
return (_apexZone as StubZone).IsExpired;
default:
return false;
}
}
}
public bool Internal
{
get
{
if (_apexZone is null)
throw new InvalidOperationException();
switch (_type)
{
case AuthZoneType.Primary:
return (_apexZone as PrimaryZone).Internal;
default:
return false;
}
}
}
public IReadOnlyList<DnsResourceRecord> ZoneHistory
{ get { return _zoneHistory; } }
public IReadOnlyDictionary<string, object> TsigKeyNames
{
get { return _tsigKeyNames; }
set
{
if (_apexZone is null)
throw new InvalidOperationException();
switch (_type)
{
case AuthZoneType.Primary:
(_apexZone as PrimaryZone).TsigKeyNames = value;
break;
case AuthZoneType.Secondary:
(_apexZone as SecondaryZone).TsigKeyNames = value;
break;
default:
throw new InvalidOperationException();
}
}
}
public AuthZoneDnssecStatus DnssecStatus
{
get
{
if (_apexZone is null)
throw new InvalidOperationException();
return _apexZone.DnssecStatus;
}
}
public uint DnsKeyTtl
{
get
{
if (_apexZone is null)
throw new InvalidOperationException();
switch (_type)
{
case AuthZoneType.Primary:
return (_apexZone as PrimaryZone).GetDnsKeyTtl();
default:
throw new NotSupportedException();
}
}
}
public IReadOnlyCollection<DnssecPrivateKey> DnssecPrivateKeys
{ get { return _dnssecPrivateKeys; } }
public bool NotifyFailed
{ get { return _notifyFailed; } }
public bool SyncFailed
{ get { return _syncFailed; } }
#endregion
}
}