mirror of
https://github.com/fergalmoran/DnsServer.git
synced 2025-12-22 09:29:50 +00:00
WebService: implemented backup and restore api calls.
This commit is contained in:
Shreyas Zare
@@ -261,6 +261,14 @@ namespace DnsServerCore
|
|||||||
ForceUpdateBlockLists(request);
|
ForceUpdateBlockLists(request);
|
||||||
break;
|
break;
|
||||||
|
|
||||||
|
case "/api/backupSettings":
|
||||||
|
await BackupSettingsAsync(request, response);
|
||||||
|
return;
|
||||||
|
|
||||||
|
case "/api/restoreSettings":
|
||||||
|
await RestoreSettingsAsync(request);
|
||||||
|
break;
|
||||||
|
|
||||||
case "/api/getStats":
|
case "/api/getStats":
|
||||||
await GetStats(request, jsonWriter);
|
await GetStats(request, jsonWriter);
|
||||||
break;
|
break;
|
||||||
@@ -1345,6 +1353,447 @@ namespace DnsServerCore
|
|||||||
GetDnsSettings(jsonWriter);
|
GetDnsSettings(jsonWriter);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private async Task BackupSettingsAsync(HttpListenerRequest request, HttpListenerResponse response)
|
||||||
|
{
|
||||||
|
bool blockLists = false;
|
||||||
|
bool logs = false;
|
||||||
|
bool scopes = false;
|
||||||
|
bool stats = false;
|
||||||
|
bool zones = false;
|
||||||
|
bool allowedZones = false;
|
||||||
|
bool blockedZones = false;
|
||||||
|
bool dnsSettings = false;
|
||||||
|
bool logSettings = false;
|
||||||
|
|
||||||
|
string strBlockLists = request.QueryString["blockLists"];
|
||||||
|
if (!string.IsNullOrEmpty(strBlockLists))
|
||||||
|
blockLists = bool.Parse(strBlockLists);
|
||||||
|
|
||||||
|
string strLogs = request.QueryString["logs"];
|
||||||
|
if (!string.IsNullOrEmpty(strLogs))
|
||||||
|
logs = bool.Parse(strLogs);
|
||||||
|
|
||||||
|
string strScopes = request.QueryString["scopes"];
|
||||||
|
if (!string.IsNullOrEmpty(strScopes))
|
||||||
|
scopes = bool.Parse(strScopes);
|
||||||
|
|
||||||
|
string strStats = request.QueryString["stats"];
|
||||||
|
if (!string.IsNullOrEmpty(strStats))
|
||||||
|
stats = bool.Parse(strStats);
|
||||||
|
|
||||||
|
string strZones = request.QueryString["zones"];
|
||||||
|
if (!string.IsNullOrEmpty(strZones))
|
||||||
|
zones = bool.Parse(strZones);
|
||||||
|
|
||||||
|
string strAllowedZones = request.QueryString["allowedZones"];
|
||||||
|
if (!string.IsNullOrEmpty(strAllowedZones))
|
||||||
|
allowedZones = bool.Parse(strAllowedZones);
|
||||||
|
|
||||||
|
string strBlockedZones = request.QueryString["blockedZones"];
|
||||||
|
if (!string.IsNullOrEmpty(strBlockedZones))
|
||||||
|
blockedZones = bool.Parse(strBlockedZones);
|
||||||
|
|
||||||
|
string strDnsSettings = request.QueryString["dnsSettings"];
|
||||||
|
if (!string.IsNullOrEmpty(strDnsSettings))
|
||||||
|
dnsSettings = bool.Parse(strDnsSettings);
|
||||||
|
|
||||||
|
string strLogSettings = request.QueryString["logSettings"];
|
||||||
|
if (!string.IsNullOrEmpty(strLogSettings))
|
||||||
|
logSettings = bool.Parse(strLogSettings);
|
||||||
|
|
||||||
|
string tmpFile = Path.GetTempFileName();
|
||||||
|
try
|
||||||
|
{
|
||||||
|
using (FileStream backupZipStream = new FileStream(tmpFile, FileMode.Create, FileAccess.ReadWrite))
|
||||||
|
{
|
||||||
|
//create backup zip
|
||||||
|
using (ZipArchive backupZip = new ZipArchive(backupZipStream, ZipArchiveMode.Create, true, Encoding.UTF8))
|
||||||
|
{
|
||||||
|
if (blockLists)
|
||||||
|
{
|
||||||
|
string[] blockListFiles = Directory.GetFiles(Path.Combine(_configFolder, "blocklists"), "*", SearchOption.TopDirectoryOnly);
|
||||||
|
foreach (string blockListFile in blockListFiles)
|
||||||
|
{
|
||||||
|
string entryName = "blocklists/" + Path.GetFileName(blockListFile);
|
||||||
|
backupZip.CreateEntryFromFile(blockListFile, entryName);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (logs)
|
||||||
|
{
|
||||||
|
string[] logFiles = Directory.GetFiles(_log.LogFolderAbsolutePath, "*.log", SearchOption.TopDirectoryOnly);
|
||||||
|
foreach (string logFile in logFiles)
|
||||||
|
{
|
||||||
|
string entryName = "logs/" + Path.GetFileName(logFile);
|
||||||
|
|
||||||
|
if (logFile.Equals(_log.CurrentLogFile, StringComparison.OrdinalIgnoreCase))
|
||||||
|
{
|
||||||
|
using (FileStream fS = new FileStream(logFile, FileMode.Open, FileAccess.Read, FileShare.ReadWrite))
|
||||||
|
{
|
||||||
|
ZipArchiveEntry entry = backupZip.CreateEntry(entryName);
|
||||||
|
|
||||||
|
using (Stream s = entry.Open())
|
||||||
|
{
|
||||||
|
await fS.CopyToAsync(s);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
backupZip.CreateEntryFromFile(logFile, entryName);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (scopes)
|
||||||
|
{
|
||||||
|
string[] scopFiles = Directory.GetFiles(Path.Combine(_configFolder, "scopes"), "*.scope", SearchOption.TopDirectoryOnly);
|
||||||
|
foreach (string scopFile in scopFiles)
|
||||||
|
{
|
||||||
|
string entryName = "scopes/" + Path.GetFileName(scopFile);
|
||||||
|
backupZip.CreateEntryFromFile(scopFile, entryName);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (stats)
|
||||||
|
{
|
||||||
|
string[] hourlyStatsFiles = Directory.GetFiles(Path.Combine(_configFolder, "stats"), "*.stat", SearchOption.TopDirectoryOnly);
|
||||||
|
foreach (string hourlyStatsFile in hourlyStatsFiles)
|
||||||
|
{
|
||||||
|
string entryName = "stats/" + Path.GetFileName(hourlyStatsFile);
|
||||||
|
backupZip.CreateEntryFromFile(hourlyStatsFile, entryName);
|
||||||
|
}
|
||||||
|
|
||||||
|
string[] dailyStatsFiles = Directory.GetFiles(Path.Combine(_configFolder, "stats"), "*.dstat", SearchOption.TopDirectoryOnly);
|
||||||
|
foreach (string dailyStatsFile in dailyStatsFiles)
|
||||||
|
{
|
||||||
|
string entryName = "stats/" + Path.GetFileName(dailyStatsFile);
|
||||||
|
backupZip.CreateEntryFromFile(dailyStatsFile, entryName);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (zones)
|
||||||
|
{
|
||||||
|
string[] zoneFiles = Directory.GetFiles(Path.Combine(_configFolder, "zones"), "*.zone", SearchOption.TopDirectoryOnly);
|
||||||
|
foreach (string zoneFile in zoneFiles)
|
||||||
|
{
|
||||||
|
string entryName = "zones/" + Path.GetFileName(zoneFile);
|
||||||
|
backupZip.CreateEntryFromFile(zoneFile, entryName);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (allowedZones)
|
||||||
|
{
|
||||||
|
string allowedZonesFile = Path.Combine(_configFolder, "allowed.config");
|
||||||
|
|
||||||
|
if (File.Exists(allowedZonesFile))
|
||||||
|
backupZip.CreateEntryFromFile(allowedZonesFile, "allowed.config");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (blockedZones)
|
||||||
|
{
|
||||||
|
string blockedZonesFile = Path.Combine(_configFolder, "blocked.config");
|
||||||
|
|
||||||
|
if (File.Exists(blockedZonesFile))
|
||||||
|
backupZip.CreateEntryFromFile(blockedZonesFile, "blocked.config");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dnsSettings)
|
||||||
|
{
|
||||||
|
string dnsSettingsFile = Path.Combine(_configFolder, "dns.config");
|
||||||
|
|
||||||
|
if (File.Exists(dnsSettingsFile))
|
||||||
|
backupZip.CreateEntryFromFile(dnsSettingsFile, "dns.config");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (logSettings)
|
||||||
|
{
|
||||||
|
string logSettingsFile = Path.Combine(_configFolder, "log.config");
|
||||||
|
|
||||||
|
if (File.Exists(logSettingsFile))
|
||||||
|
backupZip.CreateEntryFromFile(logSettingsFile, "log.config");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//send zip file
|
||||||
|
backupZipStream.Position = 0;
|
||||||
|
|
||||||
|
response.ContentType = "application/zip";
|
||||||
|
response.ContentLength64 = backupZipStream.Length;
|
||||||
|
response.AddHeader("Content-Disposition", "attachment;filename=DnsServerBackup.zip");
|
||||||
|
|
||||||
|
using (Stream output = response.OutputStream)
|
||||||
|
{
|
||||||
|
await backupZipStream.CopyToAsync(output);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
finally
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
File.Delete(tmpFile);
|
||||||
|
}
|
||||||
|
catch
|
||||||
|
{ }
|
||||||
|
}
|
||||||
|
|
||||||
|
_log.Write(GetRequestRemoteEndPoint(request), "[" + GetSession(request).Username + "] Settings backup zip file was exported.");
|
||||||
|
}
|
||||||
|
|
||||||
|
private async Task RestoreSettingsAsync(HttpListenerRequest request)
|
||||||
|
{
|
||||||
|
bool blockLists = false;
|
||||||
|
bool logs = false;
|
||||||
|
bool scopes = false;
|
||||||
|
bool stats = false;
|
||||||
|
bool zones = false;
|
||||||
|
bool allowedZones = false;
|
||||||
|
bool blockedZones = false;
|
||||||
|
bool dnsSettings = false;
|
||||||
|
bool logSettings = false;
|
||||||
|
|
||||||
|
string strBlockLists = request.QueryString["blockLists"];
|
||||||
|
if (!string.IsNullOrEmpty(strBlockLists))
|
||||||
|
blockLists = bool.Parse(strBlockLists);
|
||||||
|
|
||||||
|
string strLogs = request.QueryString["logs"];
|
||||||
|
if (!string.IsNullOrEmpty(strLogs))
|
||||||
|
logs = bool.Parse(strLogs);
|
||||||
|
|
||||||
|
string strScopes = request.QueryString["scopes"];
|
||||||
|
if (!string.IsNullOrEmpty(strScopes))
|
||||||
|
scopes = bool.Parse(strScopes);
|
||||||
|
|
||||||
|
string strStats = request.QueryString["stats"];
|
||||||
|
if (!string.IsNullOrEmpty(strStats))
|
||||||
|
stats = bool.Parse(strStats);
|
||||||
|
|
||||||
|
string strZones = request.QueryString["zones"];
|
||||||
|
if (!string.IsNullOrEmpty(strZones))
|
||||||
|
zones = bool.Parse(strZones);
|
||||||
|
|
||||||
|
string strAllowedZones = request.QueryString["allowedZones"];
|
||||||
|
if (!string.IsNullOrEmpty(strAllowedZones))
|
||||||
|
allowedZones = bool.Parse(strAllowedZones);
|
||||||
|
|
||||||
|
string strBlockedZones = request.QueryString["blockedZones"];
|
||||||
|
if (!string.IsNullOrEmpty(strBlockedZones))
|
||||||
|
blockedZones = bool.Parse(strBlockedZones);
|
||||||
|
|
||||||
|
string strDnsSettings = request.QueryString["dnsSettings"];
|
||||||
|
if (!string.IsNullOrEmpty(strDnsSettings))
|
||||||
|
dnsSettings = bool.Parse(strDnsSettings);
|
||||||
|
|
||||||
|
string strLogSettings = request.QueryString["logSettings"];
|
||||||
|
if (!string.IsNullOrEmpty(strLogSettings))
|
||||||
|
logSettings = bool.Parse(strLogSettings);
|
||||||
|
|
||||||
|
#region skip to content
|
||||||
|
|
||||||
|
int crlfCount = 0;
|
||||||
|
int byteRead;
|
||||||
|
|
||||||
|
while (crlfCount != 4)
|
||||||
|
{
|
||||||
|
byteRead = request.InputStream.ReadByte();
|
||||||
|
switch (byteRead)
|
||||||
|
{
|
||||||
|
case -1:
|
||||||
|
throw new EndOfStreamException();
|
||||||
|
|
||||||
|
case 13: //CR
|
||||||
|
case 10: //LF
|
||||||
|
crlfCount++;
|
||||||
|
break;
|
||||||
|
|
||||||
|
default:
|
||||||
|
crlfCount = 0;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#endregion
|
||||||
|
|
||||||
|
//write to temp file
|
||||||
|
string tmpFile = Path.GetTempFileName();
|
||||||
|
try
|
||||||
|
{
|
||||||
|
using (FileStream fS = new FileStream(tmpFile, FileMode.Create, FileAccess.ReadWrite))
|
||||||
|
{
|
||||||
|
await request.InputStream.CopyToAsync(fS);
|
||||||
|
|
||||||
|
fS.Position = 0;
|
||||||
|
using (ZipArchive backupZip = new ZipArchive(fS, ZipArchiveMode.Read, false, Encoding.UTF8))
|
||||||
|
{
|
||||||
|
if (logSettings || logs)
|
||||||
|
{
|
||||||
|
//stop logging
|
||||||
|
_log.StopLogging();
|
||||||
|
}
|
||||||
|
|
||||||
|
try
|
||||||
|
{
|
||||||
|
if (logSettings)
|
||||||
|
{
|
||||||
|
ZipArchiveEntry entry = backupZip.GetEntry("log.config");
|
||||||
|
if (entry != null)
|
||||||
|
entry.ExtractToFile(Path.Combine(_configFolder, entry.Name), true);
|
||||||
|
|
||||||
|
//reload config
|
||||||
|
_log.LoadConfig();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (logs)
|
||||||
|
{
|
||||||
|
//extract log files from backup
|
||||||
|
foreach (ZipArchiveEntry entry in backupZip.Entries)
|
||||||
|
{
|
||||||
|
if (entry.FullName.StartsWith("logs/"))
|
||||||
|
entry.ExtractToFile(Path.Combine(_log.LogFolderAbsolutePath, entry.Name), true);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
finally
|
||||||
|
{
|
||||||
|
if (logSettings || logs)
|
||||||
|
{
|
||||||
|
//start logging
|
||||||
|
if (_log.EnableLogging)
|
||||||
|
_log.StartLogging();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (blockLists)
|
||||||
|
{
|
||||||
|
//extract block list files from backup
|
||||||
|
foreach (ZipArchiveEntry entry in backupZip.Entries)
|
||||||
|
{
|
||||||
|
if (entry.FullName.StartsWith("blocklists/"))
|
||||||
|
entry.ExtractToFile(Path.Combine(_configFolder, "blocklists", entry.Name), true);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (scopes)
|
||||||
|
{
|
||||||
|
//stop dhcp server
|
||||||
|
_dhcpServer.Stop();
|
||||||
|
|
||||||
|
try
|
||||||
|
{
|
||||||
|
//delete existing scope files
|
||||||
|
string[] scopeFiles = Directory.GetFiles(Path.Combine(_configFolder, "scopes"), "*.scope", SearchOption.TopDirectoryOnly);
|
||||||
|
foreach (string scopeFile in scopeFiles)
|
||||||
|
{
|
||||||
|
File.Delete(scopeFile);
|
||||||
|
}
|
||||||
|
|
||||||
|
//extract scope files from backup
|
||||||
|
foreach (ZipArchiveEntry entry in backupZip.Entries)
|
||||||
|
{
|
||||||
|
if (entry.FullName.StartsWith("scopes/"))
|
||||||
|
entry.ExtractToFile(Path.Combine(_configFolder, "scopes", entry.Name), true);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
finally
|
||||||
|
{
|
||||||
|
//start dhcp server
|
||||||
|
_dhcpServer.Start();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (stats)
|
||||||
|
{
|
||||||
|
//delete existing stats files
|
||||||
|
string[] hourlyStatsFiles = Directory.GetFiles(Path.Combine(_configFolder, "stats"), "*.stat", SearchOption.TopDirectoryOnly);
|
||||||
|
foreach (string hourlyStatsFile in hourlyStatsFiles)
|
||||||
|
{
|
||||||
|
File.Delete(hourlyStatsFile);
|
||||||
|
}
|
||||||
|
|
||||||
|
string[] dailyStatsFiles = Directory.GetFiles(Path.Combine(_configFolder, "stats"), "*.dstat", SearchOption.TopDirectoryOnly);
|
||||||
|
foreach (string dailyStatsFile in dailyStatsFiles)
|
||||||
|
{
|
||||||
|
File.Delete(dailyStatsFile);
|
||||||
|
}
|
||||||
|
|
||||||
|
//extract stats files from backup
|
||||||
|
foreach (ZipArchiveEntry entry in backupZip.Entries)
|
||||||
|
{
|
||||||
|
if (entry.FullName.StartsWith("stats/"))
|
||||||
|
entry.ExtractToFile(Path.Combine(_configFolder, "stats", entry.Name), true);
|
||||||
|
}
|
||||||
|
|
||||||
|
//reload stats
|
||||||
|
_dnsServer.StatsManager.ReloadStats();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (zones)
|
||||||
|
{
|
||||||
|
//delete existing zone files
|
||||||
|
string[] zoneFiles = Directory.GetFiles(Path.Combine(_configFolder, "zones"), "*.zone", SearchOption.TopDirectoryOnly);
|
||||||
|
foreach (string zoneFile in zoneFiles)
|
||||||
|
{
|
||||||
|
File.Delete(zoneFile);
|
||||||
|
}
|
||||||
|
|
||||||
|
//extract zone files from backup
|
||||||
|
foreach (ZipArchiveEntry entry in backupZip.Entries)
|
||||||
|
{
|
||||||
|
if (entry.FullName.StartsWith("zones/"))
|
||||||
|
entry.ExtractToFile(Path.Combine(_configFolder, "zones", entry.Name), true);
|
||||||
|
}
|
||||||
|
|
||||||
|
//reload zones
|
||||||
|
_dnsServer.AuthZoneManager.LoadAllZoneFiles();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (allowedZones)
|
||||||
|
{
|
||||||
|
ZipArchiveEntry entry = backupZip.GetEntry("allowed.config");
|
||||||
|
if (entry != null)
|
||||||
|
entry.ExtractToFile(Path.Combine(_configFolder, entry.Name), true);
|
||||||
|
|
||||||
|
//reload
|
||||||
|
_dnsServer.AllowedZoneManager.LoadAllowedZoneFile();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (blockedZones)
|
||||||
|
{
|
||||||
|
ZipArchiveEntry entry = backupZip.GetEntry("blocked.config");
|
||||||
|
if (entry != null)
|
||||||
|
entry.ExtractToFile(Path.Combine(_configFolder, entry.Name), true);
|
||||||
|
|
||||||
|
//reload
|
||||||
|
_dnsServer.BlockedZoneManager.LoadBlockedZoneFile();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dnsSettings)
|
||||||
|
{
|
||||||
|
ZipArchiveEntry entry = backupZip.GetEntry("dns.config");
|
||||||
|
if (entry != null)
|
||||||
|
entry.ExtractToFile(Path.Combine(_configFolder, entry.Name), true);
|
||||||
|
|
||||||
|
//reload settings and block list zone
|
||||||
|
LoadConfigFile();
|
||||||
|
_dnsServer.BlockListZoneManager.LoadBlockLists();
|
||||||
|
}
|
||||||
|
|
||||||
|
_log.Write(GetRequestRemoteEndPoint(request), "[" + GetSession(request).Username + "] Settings backup zip file was restored.");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
finally
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
File.Delete(tmpFile);
|
||||||
|
}
|
||||||
|
catch
|
||||||
|
{ }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
private void ForceUpdateBlockLists(HttpListenerRequest request)
|
private void ForceUpdateBlockLists(HttpListenerRequest request)
|
||||||
{
|
{
|
||||||
ForceUpdateBlockLists();
|
ForceUpdateBlockLists();
|
||||||
@@ -2212,6 +2661,9 @@ namespace DnsServerCore
|
|||||||
throw new NotSupportedException("Zone type not supported.");
|
throw new NotSupportedException("Zone type not supported.");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
//delete cache for this zone to allow rebuilding cache data as needed by stub or forwarder zones
|
||||||
|
_dnsServer.CacheZoneManager.DeleteZone(domain);
|
||||||
|
|
||||||
jsonWriter.WritePropertyName("domain");
|
jsonWriter.WritePropertyName("domain");
|
||||||
jsonWriter.WriteValue(string.IsNullOrEmpty(domain) ? "." : domain);
|
jsonWriter.WriteValue(string.IsNullOrEmpty(domain) ? "." : domain);
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user