mirror of
https://github.com/fergalmoran/DnsServer.git
synced 2025-12-22 09:29:50 +00:00
120 lines
3.1 KiB
C#
120 lines
3.1 KiB
C#
/*
|
|
Technitium DNS Server
|
|
Copyright (C) 2024 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.ApplicationCommon;
|
|
using MaxMind.GeoIP2;
|
|
using System;
|
|
using System.IO;
|
|
|
|
namespace GeoCountry
|
|
{
|
|
class MaxMind : IDisposable
|
|
{
|
|
#region variables
|
|
|
|
static MaxMind _maxMind;
|
|
|
|
readonly DatabaseReader _mmCountryReader;
|
|
readonly DatabaseReader _mmIspReader;
|
|
readonly DatabaseReader _mmAsnReader;
|
|
|
|
#endregion
|
|
|
|
#region constructor
|
|
|
|
private MaxMind(IDnsServer dnsServer)
|
|
{
|
|
string mmCountryFile = Path.Combine(dnsServer.ApplicationFolder, "GeoIP2-Country.mmdb");
|
|
|
|
if (!File.Exists(mmCountryFile))
|
|
mmCountryFile = Path.Combine(dnsServer.ApplicationFolder, "GeoLite2-Country.mmdb");
|
|
|
|
if (!File.Exists(mmCountryFile))
|
|
throw new FileNotFoundException("MaxMind Country file is missing!");
|
|
|
|
_mmCountryReader = new DatabaseReader(mmCountryFile);
|
|
|
|
string mmIspFile = Path.Combine(dnsServer.ApplicationFolder, "GeoIP2-ISP.mmdb");
|
|
if (File.Exists(mmIspFile))
|
|
{
|
|
_mmIspReader = new DatabaseReader(mmIspFile);
|
|
return;
|
|
}
|
|
|
|
string mmAsnFile = Path.Combine(dnsServer.ApplicationFolder, "GeoLite2-ASN.mmdb");
|
|
if (File.Exists(mmAsnFile))
|
|
_mmAsnReader = new DatabaseReader(mmAsnFile);
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region IDisposable
|
|
|
|
bool _disposed;
|
|
|
|
protected virtual void Dispose(bool disposing)
|
|
{
|
|
if (_disposed)
|
|
return;
|
|
|
|
if (disposing)
|
|
{
|
|
_mmCountryReader?.Dispose();
|
|
_mmIspReader?.Dispose();
|
|
_mmAsnReader?.Dispose();
|
|
}
|
|
|
|
_disposed = true;
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
Dispose(true);
|
|
GC.SuppressFinalize(this);
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region public
|
|
|
|
public static MaxMind Create(IDnsServer dnsServer)
|
|
{
|
|
if (_maxMind is null)
|
|
_maxMind = new MaxMind(dnsServer);
|
|
|
|
return _maxMind;
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region properties
|
|
|
|
public DatabaseReader CountryReader
|
|
{ get { return _mmCountryReader; } }
|
|
|
|
public DatabaseReader IspReader
|
|
{ get { return _mmIspReader; } }
|
|
|
|
public DatabaseReader AsnReader
|
|
{ get { return _mmAsnReader; } }
|
|
|
|
#endregion
|
|
}
|
|
}
|