DomainName as command line parameter

microsoft · Nov 18, 2024 · 5bb2241 · 5bb2241
1 parent 91abe69
commit 5bb2241
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 6 deletions.
diff --git a/.github/workflows/dotnet-test.yml b/.github/workflows/dotnet-test.yml
@@ -95,4 +95,4 @@ jobs:
         dotnet-version: 6.0.x
     - name: Run Lang Interop
       shell: pwsh
-      run: scripts/DotNetTest.ps1 -Config Debug -Arch ${{ matrix.vec.arch }} -Tls ${{ matrix.vec.tls }}
+      run: scripts/DotNetTest.ps1 -Config Debug -Arch ${{ matrix.vec.arch }} -Tls ${{ matrix.vec.tls }} -DomainName "google.com"
diff --git a/scripts/DotNetTest.ps1 b/scripts/DotNetTest.ps1
@@ -19,7 +19,10 @@ param (
     [string]$Tls = "",
 
     [Parameter(Mandatory = $false)]
-    [string]$ExtraArtifactDir = ""
+    [string]$ExtraArtifactDir = "",
+
+    [Parameter(Mandatory = $true)]
+    [string]$DomainName = ""
 )
 
 Set-StrictMode -Version 'Latest'
@@ -43,4 +46,4 @@ if ($IsWindows) {
 $RootDir = Split-Path $PSScriptRoot -Parent
 
 dotnet build (Join-Path $RootDir src cs)
-dotnet run --project (Join-Path $RootDir src cs tool) -- (Join-Path $RootArtifactDir $LibName)
+dotnet run --project (Join-Path $RootDir src cs tool) -- $DomainName (Join-Path $RootArtifactDir $LibName)
diff --git a/src/cs/tool/Program.cs b/src/cs/tool/Program.cs
@@ -18,12 +18,32 @@ static unsafe void Main(string[] args)
         {
             // This code lets us pass in an argument of where to search for the library at.
             // Very helpful for testing
-            if (args.Length > 0)
+            if (args.Length == 0)
+            {
+                Console.WriteLine("Usage: MsQuicTool <DomainName> [PathToMsQuic]");
+                return;
+            }
+
+            string DomainName = args[0];
+
+            if (string.IsNullOrWhiteSpace(DomainName))
+            {
+                Console.WriteLine("DomainName cannot be empty.");
+                return;
+            }
+
+            if (DomainName.Length > 253)
+            {
+                Console.WriteLine("DomainName is too long.");
+                return;
+            }
+
+            if (args.Length > 1)
             {
                 NativeLibrary.SetDllImportResolver(typeof(MsQuic).Assembly, (libraryName, assembly, searchPath) =>
                 {
                     if (libraryName != "msquic") return IntPtr.Zero;
-                    if (NativeLibrary.TryLoad(args[0], out var ptr))
+                    if (NativeLibrary.TryLoad(args[1], out var ptr))
                     {
                         return ptr;
                     }
@@ -32,7 +52,6 @@ static unsafe void Main(string[] args)
             }
 
             var ApiTable = MsQuic.Open();
-            const string DomainName = "google.com";
             QUIC_HANDLE* registration = null;
             QUIC_HANDLE* configuration = null;
             QUIC_HANDLE* connection = null;