Skip to content

Instantly share code, notes, and snippets.

@raandree
Created March 11, 2023 16:41
Show Gist options
  • Save raandree/aa8f9244e5a2568ef32042f3da8f58c8 to your computer and use it in GitHub Desktop.
Save raandree/aa8f9244e5a2568ef32042f3da8f58c8 to your computer and use it in GitHub Desktop.
This script shows how to use a Win32 function (GetDcName) from PowerShell.
$code = @'
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Linq;
using System.Runtime.InteropServices;
using System.Text;
using System.Threading.Tasks;
namespace Test
{
public class AD
{
[StructLayout(LayoutKind.Sequential, CharSet = CharSet.Unicode)]
public struct DOMAIN_CONTROLLER_INFO
{
[MarshalAs(UnmanagedType.LPTStr)]
public string DomainControllerName;
[MarshalAs(UnmanagedType.LPTStr)]
public string DomainControllerAddress;
public uint DomainControllerAddressType;
public Guid DomainGuid;
[MarshalAs(UnmanagedType.LPTStr)]
public string DomainName;
[MarshalAs(UnmanagedType.LPTStr)]
public string DnsForestName;
public uint Flags;
[MarshalAs(UnmanagedType.LPTStr)]
public string DcSiteName;
[MarshalAs(UnmanagedType.LPTStr)]
public string ClientSiteName;
}
[DllImport("Netapi32.dll", CharSet = CharSet.Auto, SetLastError = true)]
public static extern int DsGetDcName
(
[MarshalAs(UnmanagedType.LPTStr)]
string ComputerName,
[MarshalAs(UnmanagedType.LPTStr)]
string DomainName,
[In] int DomainGuid,
[MarshalAs(UnmanagedType.LPTStr)]
string SiteName,
[MarshalAs(UnmanagedType.U4)]
DSGETDCNAME_FLAGS flags,
out IntPtr pDOMAIN_CONTROLLER_INFO
);
[DllImport("Netapi32.dll", SetLastError = true)]
public static extern int NetApiBufferFree(IntPtr Buffer);
[Flags]
public enum DSGETDCNAME_FLAGS : uint
{
DS_FORCE_REDISCOVERY = 0x00000001,
DS_DIRECTORY_SERVICE_REQUIRED = 0x00000010,
DS_DIRECTORY_SERVICE_PREFERRED = 0x00000020,
DS_GC_SERVER_REQUIRED = 0x00000040,
DS_PDC_REQUIRED = 0x00000080,
DS_BACKGROUND_ONLY = 0x00000100,
DS_IP_REQUIRED = 0x00000200,
DS_KDC_REQUIRED = 0x00000400,
DS_TIMESERV_REQUIRED = 0x00000800,
DS_WRITABLE_REQUIRED = 0x00001000,
DS_GOOD_TIMESERV_PREFERRED = 0x00002000,
DS_AVOID_SELF = 0x00004000,
DS_ONLY_LDAP_NEEDED = 0x00008000,
DS_IS_FLAT_NAME = 0x00010000,
DS_IS_DNS_NAME = 0x00020000,
DS_RETURN_DNS_NAME = 0x40000000,
DS_RETURN_FLAT_NAME = 0x80000000
}
public DOMAIN_CONTROLLER_INFO GetDomainInfo()
{
DOMAIN_CONTROLLER_INFO domainInfo;
const int ERROR_SUCCESS = 0;
IntPtr pDCI = IntPtr.Zero;
try
{
int val = DsGetDcName("", "", 0, "",
DSGETDCNAME_FLAGS.DS_DIRECTORY_SERVICE_REQUIRED |
DSGETDCNAME_FLAGS.DS_RETURN_DNS_NAME |
DSGETDCNAME_FLAGS.DS_IP_REQUIRED, out pDCI);
//check return value for error
if (ERROR_SUCCESS == val)
{
domainInfo = (DOMAIN_CONTROLLER_INFO)Marshal.PtrToStructure(pDCI, typeof(DOMAIN_CONTROLLER_INFO));
string msg = "Forest : " + domainInfo.DnsForestName + "\r\n";
msg += "DC-Site: " + domainInfo.DomainControllerName + "\r\n";
msg += " Client: " + domainInfo.ClientSiteName + "\r\n";
Console.WriteLine(msg);
}
else
{
throw new Win32Exception(val);
}
}
finally
{
NetApiBufferFree(pDCI);
}
return domainInfo;
}
}
}
'@
Add-Type -TypeDefinition $code
$ERROR_SUCCESS = 0;
[IntPtr]$pDCI = [IntPtr]::Zero
try {
$flags = [Test.AD+DSGETDCNAME_FLAGS]::DS_DIRECTORY_SERVICE_REQUIRED -bor [Test.AD+DSGETDCNAME_FLAGS]::DS_RETURN_DNS_NAME -bor [Test.AD+DSGETDCNAME_FLAGS]::DS_IP_REQUIRED
$val = [Test.AD]::DsGetDcName('', '', 0, '', $flags, [ref]$pDCI);
if ($ERROR_SUCCESS -eq $val)
{
$domainInfo = [Test.AD+DOMAIN_CONTROLLER_INFO][System.Runtime.InteropServices.Marshal]::PtrToStructure($pDCI, [type][Test.AD+DOMAIN_CONTROLLER_INFO])
Write-Host "Forest: $($domainInfo.DnsForestName)"
Write-Host "DC-Site: $($domainInfo.DomainControllerName)"
Write-Host "Client: $($domainInfo.ClientSiteName)"
}
else
{
$ex = [System.ComponentModel.Win32Exception]::new($val)
Write-Error -Exception $ex
}
}
finally
{
[void][Test.AD]::NetApiBufferFree($pDCI)
}
$domainInfo
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment