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

DnsServer: made changes to handle gracefull shutdown and restarting the DNS server.

Browse Source
This commit is contained in:
Shreyas Zare
2020-12-25 18:27:04 +05:30
parent f0dd31ae6d
commit d50fb087fb
1 changed files with 43 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
DnsServerCore/Dns/DnsServer.cs
View File

@@ -265,6 +265,10 @@ namespace DnsServerCore.Dns
}
}
}
catch (ObjectDisposedException)
{
//server stopping
}
catch (Exception ex)
{
if ((_state == ServiceState.Stopping) || (_state == ServiceState.Stopped))
@@ -273,8 +277,6 @@ namespace DnsServerCore.Dns
LogManager log = _log;
if (log != null)
log.Write(ex);
throw;
}
}
@@ -363,6 +365,15 @@ namespace DnsServerCore.Dns
_ = ProcessConnectionAsync(socket, protocol, usingHttps);
}
}
catch (SocketException ex)
{
if (ex.SocketErrorCode == SocketError.OperationAborted)
return; //server stopping
LogManager log = _log;
if (log != null)
log.Write(localEP, protocol, ex);
}
catch (Exception ex)
{
if ((_state == ServiceState.Stopping) || (_state == ServiceState.Stopped))
@@ -371,8 +382,6 @@ namespace DnsServerCore.Dns
LogManager log = _log;
if (log != null)
log.Write(localEP, protocol, ex);
throw;
}
}
@@ -2234,31 +2243,56 @@ namespace DnsServerCore.Dns
foreach (Socket udpListener in _udpListeners)
{
for (int i = 0; i < listenerTaskCount; i++)
_ = ReadUdpRequestAsync(udpListener);
{
_ = Task.Factory.StartNew(delegate ()
{
return ReadUdpRequestAsync(udpListener);
}, CancellationToken.None, TaskCreationOptions.DenyChildAttach, TaskScheduler.Current);
}
}
foreach (Socket tcpListener in _tcpListeners)
{
for (int i = 0; i < listenerTaskCount; i++)
_ = AcceptConnectionAsync(tcpListener, DnsTransportProtocol.Tcp, false);
{
_ = Task.Factory.StartNew(delegate ()
{
return AcceptConnectionAsync(tcpListener, DnsTransportProtocol.Tcp, false);
}, CancellationToken.None, TaskCreationOptions.DenyChildAttach, TaskScheduler.Current);
}
}
foreach (Socket httpListener in _httpListeners)
{
for (int i = 0; i < listenerTaskCount; i++)
_ = AcceptConnectionAsync(httpListener, DnsTransportProtocol.Https, false);
{
_ = Task.Factory.StartNew(delegate ()
{
return AcceptConnectionAsync(httpListener, DnsTransportProtocol.Https, false);
}, CancellationToken.None, TaskCreationOptions.DenyChildAttach, TaskScheduler.Current);
}
}
foreach (Socket tlsListener in _tlsListeners)
{
for (int i = 0; i < listenerTaskCount; i++)
_ = AcceptConnectionAsync(tlsListener, DnsTransportProtocol.Tls, false);
{
_ = Task.Factory.StartNew(delegate ()
{
return AcceptConnectionAsync(tlsListener, DnsTransportProtocol.Tls, false);
}, CancellationToken.None, TaskCreationOptions.DenyChildAttach, TaskScheduler.Current);
}
}
foreach (Socket httpsListener in _httpsListeners)
{
for (int i = 0; i < listenerTaskCount; i++)
_ = AcceptConnectionAsync(httpsListener, DnsTransportProtocol.Https, true);
{
_ = Task.Factory.StartNew(delegate ()
{
return AcceptConnectionAsync(httpsListener, DnsTransportProtocol.Https, true);
}, CancellationToken.None, TaskCreationOptions.DenyChildAttach, TaskScheduler.Current);
}
}
_cachePrefetchSamplingTimer = new Timer(CachePrefetchSamplingTimerCallback, null, Timeout.Infinite, Timeout.Infinite);