fergalmoran/DnsServer
1
0
Fork 0
mirror of https://github.com/fergalmoran/DnsServer.git synced 2026-01-28 19:46:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

DnsServer: minor refactoring done in PostProcessQuery().

Browse Source
This commit is contained in:
Shreyas Zare
2021-11-06 13:32:35 +05:30
parent c115b70fec
commit 35a22bae12
1 changed files with 19 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
DnsServerCore/Dns/DnsServer.cs
View File

@@ -1044,32 +1044,30 @@ namespace DnsServerCore.Dns
private DnsDatagram PostProcessQuery(DnsDatagram request, DnsDatagram response)
{
if (request.EDNS is not null)
if (request.EDNS is null)
return response;
if (response.Additional.Count == 0)
return response.Clone(null, null, new DnsResourceRecord[] { DnsDatagramEdns.GetOPTFor(_udpPayloadSize, response.RCODE) });
if (response.IsSigned)
return response;
DnsResourceRecord[] newAdditional = new DnsResourceRecord[response.Additional.Count + 1];
for (int i = 0; i < response.Additional.Count; i++)
{
if (response.Additional.Count == 0)
return response.Clone(null, null, new DnsResourceRecord[] { DnsDatagramEdns.GetOPTFor(_udpPayloadSize, response.RCODE) });
DnsResourceRecord record = response.Additional[i];
if (!response.IsSigned)
{
DnsResourceRecord[] newAdditional = new DnsResourceRecord[response.Additional.Count + 1];
if (record.Type == DnsResourceRecordType.OPT)
throw new InvalidOperationException("DnsDatagram already contains an OPT record.");
for (int i = 0; i < response.Additional.Count; i++)
{
DnsResourceRecord record = response.Additional[i];
if (record.Type == DnsResourceRecordType.OPT)
throw new InvalidOperationException("DnsDatagram already contains an OPT record.");
newAdditional[i] = record;
}
newAdditional[response.Additional.Count] = DnsDatagramEdns.GetOPTFor(_udpPayloadSize, response.RCODE);
return response.Clone(null, null, newAdditional);
}
newAdditional[i] = record;
}
return response;
newAdditional[response.Additional.Count] = DnsDatagramEdns.GetOPTFor(_udpPayloadSize, response.RCODE);
return response.Clone(null, null, newAdditional);
}
private IReadOnlyList<DnsResourceRecord> RemoveOPTFromAdditional(IReadOnlyList<DnsResourceRecord> additional)