cmd/cli: get AD domain using Windows API

cmd/cli/ad_windows.go

@@ -1,8 +1,11 @@
 package cli
 
 import (
-	"fmt"
 	"strings"
+	"syscall"
+	"unsafe"
+
+	"golang.org/x/sys/windows"
 
 	"github.com/Control-D-Inc/ctrld"
 )
@@ -40,10 +43,15 @@ func addExtraSplitDnsRule(cfg *ctrld.Config) bool {
 
 // getActiveDirectoryDomain returns AD domain name of this computer.
 func getActiveDirectoryDomain() (string, error) {
-	cmd := "$obj = Get-WmiObject Win32_ComputerSystem; if ($obj.PartOfDomain) { $obj.Domain }"
+	var domain *uint16
-	output, err := powershell(cmd)
+	var status uint32
+	err := syscall.NetGetJoinInformation(nil, &domain, &status)
 	if err != nil {
-		return "", fmt.Errorf("failed to get domain name: %w, output:\n\n%s", err, string(output))
+		return "", err
 	}
-	return string(output), nil
+	defer syscall.NetApiBufferFree((*byte)(unsafe.Pointer(domain)))
+	if status == syscall.NetSetupDomainName {
+		return windows.UTF16PtrToString(domain), nil
+	}
+	return "", nil
 }

cmd/cli/ad_windows_test.go

@@ -0,0 +1,36 @@
+package cli
+
+import (
+	"fmt"
+	"testing"
+	"time"
+)
+
+func Test_getActiveDirectoryDomain(t *testing.T) {
+	start := time.Now()
+	domain, err := getActiveDirectoryDomain()
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Logf("Using Windows API takes: %d", time.Since(start).Milliseconds())
+
+	start = time.Now()
+	domainPowershell, err := getActiveDirectoryDomainPowershell()
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Logf("Using Powershell takes: %d", time.Since(start).Milliseconds())
+
+	if domain != domainPowershell {
+		t.Fatalf("result mismatch, want: %v, got: %v", domainPowershell, domain)
+	}
+}
+
+func getActiveDirectoryDomainPowershell() (string, error) {
+	cmd := "$obj = Get-WmiObject Win32_ComputerSystem; if ($obj.PartOfDomain) { $obj.Domain }"
+	output, err := powershell(cmd)
+	if err != nil {
+		return "", fmt.Errorf("failed to get domain name: %w, output:\n\n%s", err, string(output))
+	}
+	return string(output), nil
+}