Handle UnauthorizedAccessException in ETWHelper (#3228)

* catch UnauthorizedAccessException * stop before showing error
microsoft · Jun 14, 2024 · 51447fe · 51447fe
1 parent d30b18e
commit 51447fe
Showing 1 changed file with 18 additions and 9 deletions.
diff --git a/tools/PI/DevHome.PI/Helpers/ETWHelper.cs b/tools/PI/DevHome.PI/Helpers/ETWHelper.cs
@@ -55,18 +55,27 @@ public void Start()
                 session.Dispose();
             }
 
-            using (session = new TraceEventSession(sessionName))
+            try
             {
-                // Filter the provider events based on processId
-                var providerOptions = new TraceEventProviderOptions { ProcessIDFilter = [targetProcess.Id] };
-                foreach (var provider in ProviderList)
+                using (session = new TraceEventSession(sessionName))
                 {
-                    session.EnableProvider(provider, TraceEventLevel.Always, options: providerOptions);
+                    // Filter the provider events based on processId
+                    var providerOptions = new TraceEventProviderOptions { ProcessIDFilter = [targetProcess.Id] };
+                    foreach (var provider in ProviderList)
+                    {
+                        session.EnableProvider(provider, TraceEventLevel.Always, options: providerOptions);
+                    }
+
+                    session.Source.Dynamic.All += EventsHandler;
+                    session.Source.UnhandledEvents += UnHandledEventsHandler;
+                    session.Source.Process();
                 }
-
-                session.Source.Dynamic.All += EventsHandler;
-                session.Source.UnhandledEvents += UnHandledEventsHandler;
-                session.Source.Process();
+            }
+            catch (UnauthorizedAccessException ex)
+            {
+                Stop();
+                WinLogsEntry entry = new(DateTime.Now, WinLogCategory.Error, ex.Message, WinLogsHelper.EtwLogsName);
+                output.Add(entry);
             }
         }
     }