1 files changed, 114 insertions, 0 deletions
diff --git a/tests/src/tracing/common/TraceControl.cs b/tests/src/tracing/common/TraceControl.cs
new file mode 100644
index 0000000000..80d00c072c
--- /dev/null
+++ b/tests/src/tracing/common/TraceControl.cs
@@ -0,0 +1,114 @@
+using System;
+using System.Reflection;
+
+namespace Tracing.Tests.Common
+{
+    public static class TraceControl
+    {
+        private static MethodInfo m_enableMethod;
+        private static MethodInfo m_disableMethod;
+
+        public static void EnableDefault()
+        {
+            EnableDefault(TimeSpan.FromMilliseconds(1));
+        }
+
+        public static void EnableDefault(string outputFile)
+        {
+            EnableDefault(TimeSpan.FromMilliseconds(1), outputFile);
+        }
+
+        public static void EnableDefault(TimeSpan profSampleDelay, string outputFile="default.netperf")
+        {
+            // Setup the configuration values.
+            uint circularBufferMB = 1024; // 1 GB
+            uint level = 5; // Verbose
+
+            // Create a new instance of EventPipeConfiguration.
+            TraceConfiguration config = new TraceConfiguration(outputFile, circularBufferMB);
+            // Setup the provider values.
+            // Public provider.
+            string providerName = "Microsoft-Windows-DotNETRuntime";
+            UInt64 keywords = 0x4c14fccbd;
+
+            // Enable the provider.
+            config.EnableProvider(providerName, keywords, level);
+
+            // Private provider.
+            providerName = "Microsoft-Windows-DotNETRuntimePrivate";
+            keywords = 0x4002000b;
+
+            // Enable the provider.
+            config.EnableProvider(providerName, keywords, level);
+
+            // Sample profiler.
+            providerName = "Microsoft-DotNETCore-SampleProfiler";
+            keywords = 0x0;
+
+            // Enable the provider.
+            config.EnableProvider(providerName, keywords, level);
+
+            // Set the sampling rate.
+            config.SetSamplingRate(profSampleDelay);
+
+            // Enable tracing.
+            Enable(config);
+        }
+
+        public static void Enable(TraceConfiguration traceConfig)
+        {
+            m_enableMethod.Invoke(
+                null,
+                new object[]
+                {
+                    traceConfig.ConfigurationObject
+                });
+        }
+
+        public static void Disable()
+        {
+            m_disableMethod.Invoke(
+                null,
+                null);
+        }
+
+        static TraceControl()
+        {
+            if(!Initialize())
+            {
+                throw new InvalidOperationException("Reflection failed.");
+            }
+        }
+
+        private static bool Initialize()
+        {
+           Assembly SPC = typeof(System.Diagnostics.Tracing.EventSource).Assembly;
+           if(SPC == null)
+           {
+               Console.WriteLine("System.Private.CoreLib assembly == null");
+               return false;
+           }
+           Type eventPipeType = SPC.GetType("System.Diagnostics.Tracing.EventPipe");
+           if(eventPipeType == null)
+           {
+               Console.WriteLine("System.Diagnostics.Tracing.EventPipe type == null");
+               return false;
+           }
+           m_enableMethod = eventPipeType.GetMethod("Enable", BindingFlags.NonPublic | BindingFlags.Static);
+           if(m_enableMethod == null)
+           {
+               Console.WriteLine("EventPipe.Enable method == null");
+               return false;
+           }
+           m_disableMethod = eventPipeType.GetMethod("Disable", BindingFlags.NonPublic | BindingFlags.Static);
+           if(m_disableMethod == null)
+           {
+               Console.WriteLine("EventPipe.Disable method == null");
+               return false;
+           }
+
+           return true;
+        }
+
+    }
+}