zydronium/websitepanel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Initial project's source code check-in.

Browse source
This commit is contained in:
ptsurbeleu 2011-07-13 16:07:32 -07:00
commit b03b0b373f
4573 changed files with 981205 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

966
WebsitePanel/Sources/WebsitePanel.Providers.DNS.MsDNS/MsDNS.cs Normal file
View file

@ -0,0 +1,966 @@
// Copyright (c) 2011, Outercurve Foundation.
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without modification,
// are permitted provided that the following conditions are met:
//
// - Redistributions of source code must retain the above copyright notice, this
// list of conditions and the following disclaimer.
//
// - Redistributions in binary form must reproduce the above copyright notice,
// this list of conditions and the following disclaimer in the documentation
// and/or other materials provided with the distribution.
//
// - Neither the name of the Outercurve Foundation nor the names of its
// contributors may be used to endorse or promote products derived from this
// software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
// WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
// DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
// ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
// (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
// LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
// ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
using System;
using System.Management;
using System.Collections.Generic;
using System.Text;
using Microsoft.Win32;
using WebsitePanel.Server.Utils;
using WebsitePanel.Providers.Utils;
namespace WebsitePanel.Providers.DNS
{
public class MsDNS : HostingServiceProviderBase, IDnsServer
{
#region Properties
protected int ExpireLimit
{
get { return ProviderSettings.GetInt("ExpireLimit"); }
}
protected int MinimumTTL
{
get { return ProviderSettings.GetInt("MinimumTTL"); }
}
protected int RefreshInterval
{
get { return ProviderSettings.GetInt("RefreshInterval"); }
}
protected int RetryDelay
{
get { return ProviderSettings.GetInt("RetryDelay"); }
}
protected bool AdMode
{
get { return ProviderSettings.GetBool("AdMode"); }
}
#endregion
private WmiHelper wmi = null;
private bool bulkRecords;
public MsDNS()
{
if (IsDNSInstalled())
{
// creare WMI helper
wmi = new WmiHelper("root\\MicrosoftDNS");
}
}
#region Zones
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <returns></returns>
public virtual string[] GetZones()
{
List<string> zones = new List<string>();
using (ManagementObjectCollection objZones = wmi.GetClass("MicrosoftDNS_Zone").GetInstances())
{
foreach (ManagementObject objZone in objZones) using(objZone)
{
if ((uint)objZone.Properties["ZoneType"].Value == 1)
zones.Add((string)objZone.Properties["Name"].Value);
}
}
return zones.ToArray();
}
/// <summary>
///
/// </summary>
/// <param name="zoneName"></param>
/// <returns></returns>
/// <remarks>Supports managed resources disposal</remarks>
public virtual bool ZoneExists(string zoneName)
{
using (RegistryKey root = Registry.LocalMachine)
{
using (RegistryKey rk = root.OpenSubKey("Software\\Microsoft\\Windows NT\\CurrentVersion\\DNS Server\\Zones\\" + zoneName))
{
return (rk != null);
}
}
}
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <param name="zoneName"></param>
/// <returns></returns>
public virtual DnsRecord[] GetZoneRecords(string zoneName)
{
//using (ManagementObjectCollection rrs = wmi.ExecuteQuery(
// String.Format("SELECT * FROM MicrosoftDNS_ResourceRecord WHERE DomainName='{0}'", zoneName)))
//ManagementObjectCollection rrs = GetWmiObjects("MicrosoftDNS_ResourceRecord", "DomainName='{0}'",zoneName);
ManagementObjectCollection rrsA = wmi.GetWmiObjects("MicrosoftDNS_AType", "DomainName='{0}'", zoneName);
ManagementObjectCollection rrsCNAME = wmi.GetWmiObjects("MicrosoftDNS_CNAMEType", "DomainName='{0}'", zoneName);
ManagementObjectCollection rrsMX = wmi.GetWmiObjects("MicrosoftDNS_MXType", "DomainName='{0}'", zoneName);
ManagementObjectCollection rrsNS = wmi.GetWmiObjects("MicrosoftDNS_NSType", "DomainName='{0}'", zoneName);
ManagementObjectCollection rrsTXT = wmi.GetWmiObjects("MicrosoftDNS_TXTType", "DomainName='{0}'", zoneName);
List<DnsRecord> records = new List<DnsRecord>();
DnsRecord record = new DnsRecord();
foreach (ManagementObject rr in rrsA)
{
record = new DnsRecord();
record.RecordType = DnsRecordType.A;
record.RecordName = CorrectHost(zoneName, (string)rr.Properties["OwnerName"].Value);
record.RecordData = (string)rr.Properties["RecordData"].Value;
records.Add(record);
}
foreach (ManagementObject rr in rrsCNAME)
{
record = new DnsRecord();
record.RecordType = DnsRecordType.CNAME;
record.RecordName = CorrectHost(zoneName, (string)rr.Properties["OwnerName"].Value);
record.RecordData = RemoveTrailingDot((string)rr.Properties["RecordData"].Value);
records.Add(record);
}
foreach (ManagementObject rr in rrsMX)
{
record = new DnsRecord();
record.RecordType = DnsRecordType.MX;
record.RecordName = CorrectHost(zoneName, (string)rr.Properties["OwnerName"].Value);
record.RecordData = RemoveTrailingDot((string)rr.Properties["MailExchange"].Value);
record.MxPriority = Convert.ToInt32(rr.Properties["Preference"].Value);
records.Add(record);
}
foreach (ManagementObject rr in rrsNS)
{
record = new DnsRecord();
record.RecordType = DnsRecordType.NS;
record.RecordName = CorrectHost(zoneName, (string)rr.Properties["OwnerName"].Value);
record.RecordData = RemoveTrailingDot((string)rr.Properties["NSHost"].Value);
records.Add(record);
}
foreach (ManagementObject rr in rrsTXT)
{
record = new DnsRecord();
record.RecordType = DnsRecordType.TXT;
record.RecordName = CorrectHost(zoneName, (string)rr.Properties["OwnerName"].Value);
string text = (string)rr.Properties["RecordData"].Value;
record.RecordData = text.Substring(1, text.Length - 2);
records.Add(record);
}
return records.ToArray();
}
private string RemoveTrailingDot(string str)
{
return (str.EndsWith(".")) ? str.Substring(0, str.Length - 1) : str;
}
private string CorrectHost(string zoneName, string host)
{
if (host.ToLower() == zoneName.ToLower())
return "";
else
return host.Substring(0, (host.Length - zoneName.Length - 1));
}
private ManagementObject GetZone(string zoneName)
{
ManagementObject objZone = null;
try
{
objZone = wmi.GetObject(String.Format(
"MicrosoftDNS_Zone.ContainerName='{0}',DnsServerName='{1}',Name='{2}'",
zoneName, System.Net.Dns.GetHostEntry("LocalHost").HostName, zoneName));
objZone.Get();
/*
objZone = wmi.GetWmiObject("MicrosoftDNS_Zone", "ContainerName = '{0}' AND DnsServerName = '{1}' AND Name = '{2}'",
new object[] { zoneName, System.Net.Dns.GetHostEntry("LocalHost").HostName, zoneName });
*/
}
catch(Exception ex)
{
objZone = null;
Log.WriteError("Could not get DNS Zone", ex);
}
return objZone;
}
/// <summary>
///
/// </summary>
/// <param name="zoneName"></param>
/// <param name="secondaryServers"></param>
/// <remarks>Supports managed resources disposal</remarks>
public virtual void AddPrimaryZone(string zoneName, string[] secondaryServers)
{
// check if zone exists
if (ZoneExists(zoneName))
return;
// create a zone
using (ManagementClass clsZone = wmi.GetClass("MicrosoftDNS_Zone"))
{
using (ManagementBaseObject inParams = clsZone.GetMethodParameters("CreateZone"))
{
inParams["ZoneName"] = zoneName;
inParams["ZoneType"] = 0; // primary zone
// create zones in AD if required
if (AdMode)
inParams["DsIntegrated"] = true;
using (ManagementBaseObject outParams = clsZone.InvokeMethod("CreateZone", inParams, null))
{
// update created zone
using (ManagementObject objZone = wmi.GetObject(String.Format(
"MicrosoftDNS_Zone.ContainerName='{0}',DnsServerName='{1}',Name='{2}'",
zoneName, System.Net.Dns.GetHostEntry("LocalHost").HostName, zoneName)))
{
try
{
// invoke ResetSecondaries method
using (ManagementBaseObject inParams2 = objZone.GetMethodParameters("ResetSecondaries"))
{
inParams2["SecondaryServers"] = new string[] { };
inParams2["NotifyServers"] = new string[] { };
if (secondaryServers == null || secondaryServers.Length == 0)
{
// transfers are not allowed
inParams2["SecureSecondaries"] = 3;
inParams2["Notify"] = 0;
}
else if (secondaryServers.Length == 1 &&
secondaryServers[0] == "*")
{
// allowed transfer from all servers
inParams2["SecureSecondaries"] = 0;
inParams2["Notify"] = 1;
}
else
{
// allowed transfer from specified servers
inParams2["SecureSecondaries"] = 2;
inParams2["SecondaryServers"] = secondaryServers;
inParams2["NotifyServers"] = secondaryServers;
inParams2["Notify"] = 2;
}
objZone.InvokeMethod("ResetSecondaries", inParams2, null);
}
}
catch
{
Log.WriteWarning("Error resetting/notifying secondary name servers");
}
}
}
}
}
// delete orphan NS records
DeleteOrphanNsRecords(zoneName);
}
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <param name="zoneName"></param>
/// <param name="masterServers"></param>
public virtual void AddSecondaryZone(string zoneName, string[] masterServers)
{
// check if zone exists
using (ManagementObject objSecondary = GetZone(zoneName))
{
if (objSecondary != null)
return;
}
// create a zone
using (ManagementClass clsZone = wmi.GetClass("MicrosoftDNS_Zone"))
{
using (ManagementBaseObject inParams = clsZone.GetMethodParameters("CreateZone"))
{
inParams["ZoneName"] = zoneName;
inParams["ZoneType"] = 1; // secondary zone
inParams["IpAddr"] = masterServers;
inParams["DataFileName"] = zoneName + ".dns";
// create zones in AD if required
inParams["DsIntegrated"] = AdMode;
using (ManagementBaseObject outParams = clsZone.InvokeMethod("CreateZone", inParams, null))
{
try
{
// update created zone
/*ManagementObject objZone = wmi.GetObject(String.Format(
"MicrosoftDNS_Zone.ContainerName='{0}',DnsServerName='{1}',Name='{2}'",
zoneName, System.Net.Dns.GetHostEntry("LocalHost").HostName, zoneName));
objZone.InvokeMethod("ForceRefresh", null);*/
using (ManagementObject objZone = (ManagementObject)outParams["RR"])
{
objZone.InvokeMethod("ReloadZone", null);
}
}
catch (Exception ex)
{
Log.WriteWarning("Error ReloadZone for secondary zone '{0}': {1}", zoneName, ex.Message);
}
}
}
}
// delete orphan NS records
DeleteOrphanNsRecords(zoneName);
}
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <param name="zoneName"></param>
public virtual void DeleteZone(string zoneName)
{
try
{
using (ManagementObject objZone = GetZone(zoneName))
{
if (objZone != null)
objZone.Delete();
}
}
catch (Exception ex)
{
Log.WriteError(ex);
}
}
public virtual void AddZoneRecord(string zoneName, DnsRecord record)
{
try
{
if (record.RecordType == DnsRecordType.A)
AddARecord(zoneName, record.RecordName, record.RecordData);
else if (record.RecordType == DnsRecordType.CNAME)
AddCNameRecord(zoneName, record.RecordName, record.RecordData);
else if (record.RecordType == DnsRecordType.MX)
AddMXRecord(zoneName, record.RecordName, record.RecordData, record.MxPriority);
else if (record.RecordType == DnsRecordType.NS)
AddNsRecord(zoneName, record.RecordName, record.RecordData);
else if (record.RecordType == DnsRecordType.TXT)
AddTxtRecord(zoneName, record.RecordName, record.RecordData);
}
catch (Exception ex)
{
// log exception
Log.WriteError(ex);
}
}
public virtual void AddZoneRecords(string zoneName, DnsRecord[] records)
{
bulkRecords = true;
foreach (DnsRecord record in records)
AddZoneRecord(zoneName, record);
UpdateSoaRecord(zoneName);
}
public virtual void DeleteZoneRecord(string zoneName, DnsRecord record)
{
try
{
if (record.RecordType == DnsRecordType.A)
DeleteARecord(zoneName, record.RecordName, record.RecordData);
else if (record.RecordType == DnsRecordType.CNAME)
DeleteCNameRecord(zoneName, record.RecordName, record.RecordData);
else if (record.RecordType == DnsRecordType.MX)
DeleteMXRecord(zoneName, record.RecordName, record.RecordData);
else if (record.RecordType == DnsRecordType.NS)
DeleteNsRecord(zoneName, record.RecordName, record.RecordData);
else if (record.RecordType == DnsRecordType.TXT)
DeleteTxtRecord(zoneName, record.RecordName, record.RecordData);
}
catch (Exception ex)
{
// log exception
Log.WriteError(ex);
}
}
public virtual void DeleteZoneRecords(string zoneName, DnsRecord[] records)
{
foreach (DnsRecord record in records)
DeleteZoneRecord(zoneName, record);
}
public void AddZoneRecord(string zoneName, string recordText)
{
try
{
Log.WriteStart(string.Format("Adding MS DNS Server zone '{0}' record '{1}'", zoneName, recordText));
AddDnsRecord(zoneName, recordText);
Log.WriteEnd("Added MS DNS Server zone record");
}
catch (Exception ex)
{
Log.WriteError(ex);
throw;
}
}
#endregion
#region SOA Record
public virtual void UpdateSoaRecord(string zoneName, string host, string primaryNsServer,
string primaryPerson)
{
host = CorrectHostName(zoneName, host);
// delete record if exists
DeleteSoaRecord(zoneName);
// format record data
string recordText = GetSoaRecordText(host, primaryNsServer, primaryPerson);
// add record
AddDnsRecord(zoneName, recordText);
// update SOA record
UpdateSoaRecord(zoneName);
}
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <param name="zoneName"></param>
private void DeleteSoaRecord(string zoneName)
{
string query = String.Format("SELECT * FROM MicrosoftDNS_SOAType " +
"WHERE OwnerName = '{0}'",
zoneName);
using (ManagementObjectCollection objRRs = wmi.ExecuteQuery(query))
{
foreach (ManagementObject objRR in objRRs) using (objRR)
objRR.Delete();
}
}
private string GetSoaRecordText(string host, string primaryNsServer,
string primaryPerson)
{
return String.Format("{0} IN SOA {1} {2} 1 900 600 86400 3600", host,
primaryNsServer, primaryPerson);
}
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <param name="zoneName"></param>
private void UpdateSoaRecord(string zoneName)
{
// get existing SOA record in order to read serial number
try
{
//ManagementObject obj = GetWmiObject("MicrosoftDNS_Zone", "ContainerName = '{0}'", zoneName);
//ManagementObject objSoa = GetRelatedWmiObject(obj, "MicrosoftDNS_SOAType");
ManagementObject objSoa = wmi.GetWmiObject("MicrosoftDNS_SOAType", "ContainerName = '{0}'", RemoveTrailingDot(zoneName));
if (objSoa != null)
{
if (objSoa.Properties["OwnerName"].Value.Equals(zoneName))
{
string primaryServer = (string) objSoa.Properties["PrimaryServer"].Value;
string responsibleParty = (string) objSoa.Properties["ResponsibleParty"].Value;
UInt32 serialNumber = (UInt32) objSoa.Properties["SerialNumber"].Value;
// update record's serial number
string sn = serialNumber.ToString();
string todayDate = DateTime.Now.ToString("yyyyMMdd");
if (sn.Length < 10 || !sn.StartsWith(todayDate))
{
// build a new serial number
sn = todayDate + "01";
serialNumber = UInt32.Parse(sn);
}
else
{
// just increment serial number
serialNumber += 1;
}
// update SOA record
using (ManagementBaseObject methodParams = objSoa.GetMethodParameters("Modify"))
{
methodParams["ResponsibleParty"] = responsibleParty;
methodParams["PrimaryServer"] = primaryServer;
methodParams["SerialNumber"] = serialNumber;
methodParams["ExpireLimit"] = ExpireLimit;
methodParams["MinimumTTL"] = MinimumTTL;
methodParams["TTL"] = MinimumTTL;
methodParams["RefreshInterval"] = RefreshInterval;
methodParams["RetryDelay"] = RetryDelay;
ManagementBaseObject outParams = objSoa.InvokeMethod("Modify", methodParams, null);
}
//
objSoa.Dispose();
}
}
}
catch (Exception ex)
{
Log.WriteError(ex);
}
}
#endregion
#region A Record
/// <summary>
///
/// </summary>
/// <param name="zoneName"></param>
/// <param name="host"></param>
/// <param name="ip"></param>
/// <remarks>Supports managed resources disposal</remarks>
private void AddARecord(string zoneName, string host, string ip)
{
// add record
using (ManagementClass clsRR = wmi.GetClass("MicrosoftDNS_AType"))
{
clsRR.InvokeMethod("CreateInstanceFromPropertyData", new object[] {
GetDnsServerName(),
zoneName,
CorrectHostName(zoneName, host),
1,
MinimumTTL,
ip
});
}
// update SOA record
if (bulkRecords) return;
UpdateSoaRecord(zoneName);
}
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <param name="zoneName"></param>
/// <param name="host"></param>
/// <param name="ip"></param>
private void DeleteARecord(string zoneName, string host, string ip)
{
string query = String.Format("SELECT * FROM MicrosoftDNS_AType " +
"WHERE ContainerName = '{0}' AND OwnerName = '{1}'",
zoneName, CorrectHostName(zoneName, host));
if (ip != null)
query += String.Format(" AND RecordData = '{0}'", ip);
using (ManagementObjectCollection objRRs = wmi.ExecuteQuery(query))
{
foreach (ManagementObject objRR in objRRs) using(objRR)
objRR.Delete();
}
// update SOA record
UpdateSoaRecord(zoneName);
}
#endregion
#region CNAME Record
/// <summary>
///
/// </summary>
/// <param name="zoneName"></param>
/// <param name="alias"></param>
/// <param name="targetHost"></param>
/// <remarks>Supports managed resources disposal</remarks>
private void AddCNameRecord(string zoneName, string alias, string targetHost)
{
// add record
using (ManagementClass clsRR = wmi.GetClass("MicrosoftDNS_CNAMEType"))
{
clsRR.InvokeMethod("CreateInstanceFromPropertyData", new object[] {
GetDnsServerName(),
zoneName,
CorrectHostName(zoneName, alias),
1,
MinimumTTL,
targetHost
});
}
// update SOA record
if (bulkRecords) return;
UpdateSoaRecord(zoneName);
}
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <param name="zoneName"></param>
/// <param name="alias"></param>
/// <param name="targetHost"></param>
private void DeleteCNameRecord(string zoneName, string alias, string targetHost)
{
string query = String.Format("SELECT * FROM MicrosoftDNS_CNAMEType " +
"WHERE ContainerName = '{0}' AND OwnerName = '{1}'",
zoneName, CorrectHostName(zoneName, alias));
if (targetHost != null)
query += String.Format(" AND RecordData='{0}.'", targetHost);
using (ManagementObjectCollection objRRs = wmi.ExecuteQuery(query))
{
foreach (ManagementObject objRR in objRRs) using(objRR)
objRR.Delete();
}
// update SOA record
UpdateSoaRecord(zoneName);
}
#endregion
#region MX Record
/// <summary>
///
/// </summary>
/// <param name="zoneName"></param>
/// <param name="host"></param>
/// <param name="mailServer"></param>
/// <param name="mailServerPriority"></param>
/// <remarks>Supports managed resources disposal</remarks>
private void AddMXRecord(string zoneName, string host, string mailServer, int mailServerPriority)
{
// add record
using (ManagementClass clsRR = wmi.GetClass("MicrosoftDNS_MXType"))
{
clsRR.InvokeMethod("CreateInstanceFromPropertyData", new object[] {
GetDnsServerName(),
zoneName,
CorrectHostName(zoneName, host),
1,
MinimumTTL,
mailServerPriority,
mailServer
});
}
// update SOA record
if (bulkRecords) return;
UpdateSoaRecord(zoneName);
}
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <param name="zoneName"></param>
/// <param name="host"></param>
/// <param name="mailServer"></param>
private void DeleteMXRecord(string zoneName, string host, string mailServer)
{
string query = String.Format("SELECT * FROM MicrosoftDNS_MXType " +
"WHERE ContainerName = '{0}' AND OwnerName ='{1}'",
zoneName, CorrectHostName(zoneName, host));
if (mailServer != null)
query += String.Format(" AND MailExchange = '{0}.'", CorrectHostName(zoneName, mailServer));
using (ManagementObjectCollection objRRs = wmi.ExecuteQuery(query))
{
foreach (ManagementObject objRR in objRRs) using(objRR)
objRR.Delete();
}
// update SOA record
UpdateSoaRecord(zoneName);
}
#endregion
#region NS Record
/// <summary>
///
/// </summary>
/// <param name="zoneName"></param>
/// <param name="host"></param>
/// <param name="nameServer"></param>
/// <remarks>Supports managed resources disposal</remarks>
private void AddNsRecord(string zoneName, string host, string nameServer)
{
// add record
using (ManagementClass clsRR = wmi.GetClass("MicrosoftDNS_NSType"))
{
clsRR.InvokeMethod("CreateInstanceFromPropertyData", new object[] {
GetDnsServerName(),
zoneName,
CorrectHostName(zoneName, host),
1,
MinimumTTL,
nameServer
});
}
// update SOA record
if (bulkRecords) return;
UpdateSoaRecord(zoneName);
}
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <param name="zoneName"></param>
/// <param name="host"></param>
/// <param name="nameServer"></param>
private void DeleteNsRecord(string zoneName, string host, string nameServer)
{
string query = String.Format("SELECT * FROM MicrosoftDNS_NSType " +
"WHERE ContainerName = '{0}' AND OwnerName = '{1}'",
zoneName, CorrectHostName(zoneName, host));
if (nameServer != null)
query += String.Format(" AND NSHost = '{0}.'", nameServer);
using (ManagementObjectCollection objRRs = wmi.ExecuteQuery(query))
{
foreach (ManagementObject objRR in objRRs) using(objRR)
objRR.Delete();
}
// update SOA record
UpdateSoaRecord(zoneName);
}
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <param name="zoneName"></param>
private void DeleteOrphanNsRecords(string zoneName)
{
string machineName = System.Net.Dns.GetHostEntry("LocalHost").HostName.ToLower();
string computerName = Environment.MachineName.ToLower();
using (ManagementObjectCollection objRRs = wmi.ExecuteQuery(
String.Format("SELECT * FROM MicrosoftDNS_NSType WHERE DomainName = '{0}'", zoneName)))
{
foreach (ManagementObject objRR in objRRs)
using(objRR)
{
string ns = ((string)objRR.Properties["NSHost"].Value).ToLower();
if (ns.StartsWith(machineName) || ns.StartsWith(computerName))
objRR.Delete();
}
}
}
#endregion
#region TXT Record
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <param name="zoneName"></param>
/// <param name="host"></param>
/// <param name="text"></param>
private void AddTxtRecord(string zoneName, string host, string text)
{
using (ManagementClass clsRR = wmi.GetClass("MicrosoftDNS_TXTType"))
{
clsRR.InvokeMethod("CreateInstanceFromPropertyData", new object[] {
GetDnsServerName(),
zoneName,
CorrectHostName(zoneName, host),
1,
MinimumTTL,
System.Text.RegularExpressions.Regex.Replace(text, @"""|\\", "")
});
}
// update SOA record
if (bulkRecords) return;
UpdateSoaRecord(zoneName);
}
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <param name="zoneName"></param>
/// <param name="host"></param>
/// <param name="text"></param>
private void DeleteTxtRecord(string zoneName, string host, string text)
{
string query = String.Format("SELECT * FROM MicrosoftDNS_TXTType " +
"WHERE ContainerName = '{0}' AND OwnerName = '{1}'",
zoneName, CorrectHostName(zoneName, host));
if (text != null)
query += String.Format(" AND RecordData = '\"{0}\"'", System.Text.RegularExpressions.Regex.Replace(text, @"""|\\", ""));
using (ManagementObjectCollection objRRs = wmi.ExecuteQuery(query))
{
foreach (ManagementObject objRR in objRRs) using(objRR)
{
objRR.Delete();
}
}
// update SOA record
UpdateSoaRecord(zoneName);
}
#endregion
#region private helper methods
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <returns></returns>
private string GetDnsServerName()
{
using (ManagementObject objServer = wmi.GetObject("MicrosoftDNS_Server.Name=\".\""))
{
return (string)objServer.Properties["Name"].Value;
}
}
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <param name="zoneName"></param>
/// <param name="recordText"></param>
/// <returns></returns>
private string AddDnsRecord(string zoneName, string recordText)
{
// get the name of the server
string serverName = GetDnsServerName();
// add record
using (ManagementClass clsRR = wmi.GetClass("MicrosoftDNS_ResourceRecord"))
{
object[] prms = new object[] { serverName, zoneName, recordText, null };
clsRR.InvokeMethod("CreateInstanceFromTextRepresentation", prms);
return (string)prms[3];
}
}
private string CorrectHostName(string zoneName, string host)
{
// if host is empty or null
if (host == null || host == "")
return zoneName;
// if there are not dot at all
else if (host.IndexOf(".") == -1)
return host + "." + zoneName;
// if only one dot at the end
else if (host[host.Length - 1] == '.' && host.IndexOf(".") == (host.Length - 1))
return host + zoneName;
// other cases
else
return host;
}
#endregion
#region IHostingServiceProvier methods
public override void DeleteServiceItems(ServiceProviderItem[] items)
{
foreach (ServiceProviderItem item in items)
{
if (item is DnsZone)
{
try
{
// delete DNS zone
DeleteZone(item.Name);
}
catch (Exception ex)
{
Log.WriteError(String.Format("Error deleting '{0}' MS DNS zone", item.Name), ex);
}
}
}
}
/// <summary>
/// Supports managed resources disposal
/// </summary>
/// <returns></returns>
protected bool IsDNSInstalled()
{
using (RegistryKey root = Registry.LocalMachine)
{
using (RegistryKey key = root.OpenSubKey("SYSTEM\\CurrentControlSet\\Services\\DNS"))
{
bool res = key != null;
if (key != null)
key.Close();
return res;
}
}
}
public override bool IsInstalled()
{
return IsDNSInstalled();
}
#endregion
}
}

21
WebsitePanel/Sources/WebsitePanel.Providers.DNS.MsDNS/Properties/AssemblyInfo.cs Normal file
View file

@ -0,0 +1,21 @@
using System.Reflection;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;
// General Information about an assembly is controlled through the following
// set of attributes. Change these attribute values to modify the information
// associated with an assembly.
[assembly: AssemblyTitle("WebsitePanel.Providers.DNS.MsDNS")]
[assembly: AssemblyDescription("")]
[assembly: AssemblyConfiguration("")]
[assembly: AssemblyProduct("WebsitePanel.Providers.DNS.MsDNS")]
[assembly: AssemblyTrademark("")]
[assembly: AssemblyCulture("")]
// Setting ComVisible to false makes the types in this assembly not visible
// to COM components. If you need to access a type in this assembly from
// COM, set the ComVisible attribute to true on that type.
[assembly: ComVisible(false)]
// The following GUID is for the ID of the typelib if this project is exposed to COM
[assembly: Guid("ab3d314f-e3b2-43c3-8b43-5ecfa0a41cc3")]

104
WebsitePanel/Sources/WebsitePanel.Providers.DNS.MsDNS/WebsitePanel.Providers.DNS.MsDNS.csproj Normal file
View file

@ -0,0 +1,104 @@
<?xml version="1.0" encoding="utf-8"?>
<Project DefaultTargets="Build" xmlns="http://schemas.microsoft.com/developer/msbuild/2003" ToolsVersion="4.0">
<PropertyGroup>
<Configuration Condition=" '$(Configuration)' == '' ">Debug</Configuration>
<Platform Condition=" '$(Platform)' == '' ">AnyCPU</Platform>
<ProductVersion>9.0.30729</ProductVersion>
<SchemaVersion>2.0</SchemaVersion>
<ProjectGuid>{54D5BDBC-42FC-41C3-902A-3583B9B496F4}</ProjectGuid>
<OutputType>Library</OutputType>
<AppDesignerFolder>Properties</AppDesignerFolder>
<RootNamespace>WebsitePanel.Providers.DNS</RootNamespace>
<AssemblyName>WebsitePanel.Providers.DNS.MsDNS</AssemblyName>
<FileUpgradeFlags>
</FileUpgradeFlags>
<OldToolsVersion>3.5</OldToolsVersion>
<UpgradeBackupLocation>
</UpgradeBackupLocation>
<TargetFrameworkVersion>v4.0</TargetFrameworkVersion>
<PublishUrl>publish\</PublishUrl>
<Install>true</Install>
<InstallFrom>Disk</InstallFrom>
<UpdateEnabled>false</UpdateEnabled>
<UpdateMode>Foreground</UpdateMode>
<UpdateInterval>7</UpdateInterval>
<UpdateIntervalUnits>Days</UpdateIntervalUnits>
<UpdatePeriodically>false</UpdatePeriodically>
<UpdateRequired>false</UpdateRequired>
<MapFileExtensions>true</MapFileExtensions>
<ApplicationRevision>0</ApplicationRevision>
<ApplicationVersion>1.0.0.%2a</ApplicationVersion>
<IsWebBootstrapper>false</IsWebBootstrapper>
<UseApplicationTrust>false</UseApplicationTrust>
<BootstrapperEnabled>true</BootstrapperEnabled>
<TargetFrameworkProfile />
</PropertyGroup>
<PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Debug|AnyCPU' ">
<DebugSymbols>true</DebugSymbols>
<DebugType>full</DebugType>
<Optimize>false</Optimize>
<OutputPath>..\WebsitePanel.Server\bin\</OutputPath>
<DefineConstants>DEBUG;TRACE</DefineConstants>
<ErrorReport>prompt</ErrorReport>
<WarningLevel>4</WarningLevel>
<CodeAnalysisRuleSet>AllRules.ruleset</CodeAnalysisRuleSet>
<WarningsAsErrors>618</WarningsAsErrors>
</PropertyGroup>
<PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Release|AnyCPU' ">
<DebugType>none</DebugType>
<Optimize>true</Optimize>
<OutputPath>..\WebsitePanel.Server\bin\</OutputPath>
<DefineConstants>TRACE</DefineConstants>
<ErrorReport>prompt</ErrorReport>
<WarningLevel>4</WarningLevel>
<CodeAnalysisRuleSet>AllRules.ruleset</CodeAnalysisRuleSet>
</PropertyGroup>
<ItemGroup>
<Reference Include="System" />
<Reference Include="System.Data" />
<Reference Include="System.Management" />
<Reference Include="System.Xml" />
</ItemGroup>
<ItemGroup>
<Compile Include="..\VersionInfo.cs">
<Link>VersionInfo.cs</Link>
</Compile>
<Compile Include="MsDNS.cs" />
<Compile Include="Properties\AssemblyInfo.cs" />
</ItemGroup>
<ItemGroup>
<ProjectReference Include="..\WebsitePanel.Providers.Base\WebsitePanel.Providers.Base.csproj">
<Project>{684C932A-6C75-46AC-A327-F3689D89EB42}</Project>
<Name>WebsitePanel.Providers.Base</Name>
</ProjectReference>
<ProjectReference Include="..\WebsitePanel.Server.Utils\WebsitePanel.Server.Utils.csproj">
<Project>{E91E52F3-9555-4D00-B577-2B1DBDD87CA7}</Project>
<Name>WebsitePanel.Server.Utils</Name>
</ProjectReference>
</ItemGroup>
<ItemGroup>
<BootstrapperPackage Include="Microsoft.Net.Client.3.5">
<Visible>False</Visible>
<ProductName>.NET Framework 3.5 SP1 Client Profile</ProductName>
<Install>false</Install>
</BootstrapperPackage>
<BootstrapperPackage Include="Microsoft.Net.Framework.3.5.SP1">
<Visible>False</Visible>
<ProductName>.NET Framework 3.5 SP1</ProductName>
<Install>true</Install>
</BootstrapperPackage>
<BootstrapperPackage Include="Microsoft.Windows.Installer.3.1">
<Visible>False</Visible>
<ProductName>Windows Installer 3.1</ProductName>
<Install>true</Install>
</BootstrapperPackage>
</ItemGroup>
<Import Project="$(MSBuildBinPath)\Microsoft.CSharp.targets" />
<!-- To modify your build process, add your task inside one of the targets below and uncomment it.
Other similar extension points exist, see Microsoft.Common.targets.
<Target Name="BeforeBuild">
</Target>
<Target Name="AfterBuild">
</Target>
-->
</Project>