Add Ability to specify port for Gui command (#693)

microsoft · May 3, 2023 · b0059c4 · b0059c4
1 parent 8f3ee69
commit b0059c4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/Cli/AttackSurfaceAnalyzerClient.cs b/Cli/AttackSurfaceAnalyzerClient.cs
@@ -469,19 +469,21 @@ private static ASA_ERROR RunGuiCommand(GuiCommandOptions opts)
                     }
                 }
             }
+            var url = $"http://localhost:{opts.Port}";
 
             var host = server.ConfigureWebHostDefaults(webBuilder =>
                 {
                     webBuilder.UseWebRoot(webRoot);
                     webBuilder.UseStartup<Startup>();
+                    webBuilder.UseUrls(url);
                 })
                 .Build();
 
             if (!opts.NoLaunch)
             {
                 ((Action)(async () =>
                 {
-                    await Task.Run(() => SleepAndOpenBrowser(1500)).ConfigureAwait(false);
+                    await Task.Run(() => SleepAndOpenBrowser(1500, url)).ConfigureAwait(false);
                 }))();
             }
 
@@ -490,14 +492,12 @@ private static ASA_ERROR RunGuiCommand(GuiCommandOptions opts)
             return ASA_ERROR.NONE;
         }
 
-        private static void SleepAndOpenBrowser(int sleep)
+        private static void SleepAndOpenBrowser(int sleep, string url)
         {
             Thread.Sleep(sleep);
-            AsaHelpers.OpenBrowser(new System.Uri("http://localhost:5000")); /*DevSkim: ignore DS137138*/
+            AsaHelpers.OpenBrowser(new System.Uri(url));
         }
 
-        [System.Diagnostics.CodeAnalysis.SuppressMessage("Usage", "CA2241:Provide correct arguments to formatting methods", Justification = "<Pending>")]
-        [System.Diagnostics.CodeAnalysis.SuppressMessage("Globalization", "CA1305:Specify IFormatProvider", Justification = "<Pending>")]
         private static ASA_ERROR RunConfigCommand(ConfigCommandOptions opts)
         {
             if (opts.ResetDatabase)

diff --git a/Lib/Objects/CommandOptions.cs b/Lib/Objects/CommandOptions.cs
@@ -270,6 +270,9 @@ public class GuiCommandOptions : CommandOptions
     {
         [Option(HelpText = "Disable launching a browser after gui starts.")]
         public bool NoLaunch { get; set; }
+
+        [Option(HelpText = "Port to listen on.", Default = 5000)]
+        public int Port { get; set; } = 5000;
     }
 
     [Verb("guide", HelpText = "Gather and Analyze metrics using a combination of Collectors and Monitors.")]