fergalmoran/DnsServer
1
0
Fork 0
mirror of https://github.com/fergalmoran/DnsServer.git synced 2026-02-15 04:14:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

DnsServer: implemented DirectQuery() method to allow internal dns resolution.

Browse Source
This commit is contained in:
Shreyas Zare
2019-12-29 14:55:06 +05:30
parent 5fa63a5927
commit 2e571fba3a
1 changed files with 32 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
DnsServerCore/Dns/DnsServer.cs
View File

@@ -247,7 +247,7 @@ namespace DnsServerCore.Dns
try
{
DnsDatagram response = ProcessQuery(request, remoteEP, DnsTransportProtocol.Udp);
DnsDatagram response = ProcessQuery(request, remoteEP, IsRecursionAllowed(remoteEP), DnsTransportProtocol.Udp);
//send response
if (response != null)
@@ -450,7 +450,7 @@ namespace DnsServerCore.Dns
try
{
DnsDatagram response = ProcessQuery(request, remoteEP, protocol);
DnsDatagram response = ProcessQuery(request, remoteEP, IsRecursionAllowed(remoteEP), protocol);
//send response
if (response != null)
@@ -673,7 +673,7 @@ namespace DnsServerCore.Dns
throw new NotSupportedException("DoH request type not supported.");
}
DnsDatagram dnsResponse = ProcessQuery(dnsRequest, remoteEP, protocol);
DnsDatagram dnsResponse = ProcessQuery(dnsRequest, remoteEP, IsRecursionAllowed(remoteEP), protocol);
if (dnsResponse != null)
{
using (MemoryStream mS = new MemoryStream())
@@ -709,7 +709,7 @@ namespace DnsServerCore.Dns
dnsRequest = new DnsDatagram(new DnsHeader(0, false, DnsOpcode.StandardQuery, false, false, true, false, false, false, DnsResponseCode.NoError, 1, 0, 0, 0), new DnsQuestionRecord[] { new DnsQuestionRecord(strName, (DnsResourceRecordType)int.Parse(strType), DnsClass.IN) }, null, null, null);
DnsDatagram dnsResponse = ProcessQuery(dnsRequest, remoteEP, protocol);
DnsDatagram dnsResponse = ProcessQuery(dnsRequest, remoteEP, IsRecursionAllowed(remoteEP), protocol);
if (dnsResponse != null)
{
using (MemoryStream mS = new MemoryStream())
@@ -856,13 +856,11 @@ namespace DnsServerCore.Dns
return true;
}
internal DnsDatagram ProcessQuery(DnsDatagram request, EndPoint remoteEP, DnsTransportProtocol protocol)
internal DnsDatagram ProcessQuery(DnsDatagram request, EndPoint remoteEP, bool isRecursionAllowed, DnsTransportProtocol protocol)
{
if (request.Header.IsResponse)
return null;
bool isRecursionAllowed = IsRecursionAllowed(remoteEP);
switch (request.Header.OPCODE)
{
case DnsOpcode.StandardQuery:
@@ -1850,6 +1848,33 @@ namespace DnsServerCore.Dns
_state = ServiceState.Stopped;
}
public DnsDatagram DirectQuery(DnsQuestionRecord question, int timeout = 2000)
{
object timeoutLock = new object();
DnsDatagram response = null;
lock (timeoutLock)
{
ThreadPool.QueueUserWorkItem(delegate (object state)
{
try
{
response = ProcessQuery(new DnsDatagram(new DnsHeader(0, false, DnsOpcode.StandardQuery, false, false, true, false, false, false, DnsResponseCode.NoError, 1, 0, 0, 0), new DnsQuestionRecord[] { question }, null, null, null), new IPEndPoint(IPAddress.Any, 0), true, DnsTransportProtocol.Tcp);
lock (timeoutLock)
{
Monitor.Pulse(timeoutLock);
}
}
catch
{ }
});
Monitor.Wait(timeoutLock, timeout);
return response;
}
}
#endregion
#region properties