Move SynchronizationContext to shared partition

Author: marek-safar

src/System.Private.CoreLib/System.Private.CoreLib.csproj

@@ -268,7 +268,7 @@
     <Compile Include="$(BclSourcesRoot)\System\Threading\Interlocked.cs" />
     <Compile Include="$(BclSourcesRoot)\System\Threading\Monitor.cs" />
     <Compile Include="$(BclSourcesRoot)\System\Threading\Overlapped.cs" />
-    <Compile Include="$(BclSourcesRoot)\System\Threading\SynchronizationContext.cs" />
+    <Compile Include="$(BclSourcesRoot)\System\Threading\SynchronizationContext.CoreCLR.cs" />
     <Compile Include="$(BclSourcesRoot)\System\Threading\Thread.cs" />
     <Compile Include="$(BclSourcesRoot)\System\Threading\ThreadPool.CoreCLR.cs" />
     <Compile Include="$(BclSourcesRoot)\System\Threading\Timer.CoreCLR.cs" />
@@ -367,6 +367,9 @@
     <Compile Include="$(BclSourcesRoot)\System\Globalization\GlobalizationMode.Windows.cs" />
     <Compile Include="$(BclSourcesRoot)\System\Threading\ClrThreadPoolBoundHandle.Windows.cs" />
   </ItemGroup>
+  <ItemGroup Condition="'$(FeatureAppX)' == 'true' or '$(EnableWinRT)' == 'true'">
+    <Compile Include="$(BclSourcesRoot)\System\Threading\SynchronizationContext.Uap.cs" />
+  </ItemGroup>
   <ItemGroup>
     <!--
     These files are also added to CoreLib.Shared.projitems, but they don't show up in

src/System.Private.CoreLib/shared/System.Private.CoreLib.Shared.projitems

@@ -788,6 +788,7 @@
     <Compile Include="$(MSBuildThisFileDirectory)System\Threading\SendOrPostCallback.cs" />
     <Compile Include="$(MSBuildThisFileDirectory)System\Threading\SpinLock.cs" />
     <Compile Include="$(MSBuildThisFileDirectory)System\Threading\SpinWait.cs" />
+    <Compile Include="$(MSBuildThisFileDirectory)System\Threading\SynchronizationContext.cs" />
     <Compile Include="$(MSBuildThisFileDirectory)System\Threading\SynchronizationLockException.cs" />
     <Compile Include="$(MSBuildThisFileDirectory)System\Threading\ThreadLocal.cs" />
     <Compile Include="$(MSBuildThisFileDirectory)System\Threading\Tasks\AsyncCausalityTracerConstants.cs" />

src/System.Private.CoreLib/shared/System/Threading/SynchronizationContext.cs

@@ -0,0 +1,66 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+// See the LICENSE file in the project root for more information.
+
+#if CORERT
+using Thread = Internal.Runtime.Augments.RuntimeThread;
+#endif
+
+namespace System.Threading
+{
+    public partial class SynchronizationContext
+    {
+        private bool _requireWaitNotification;
+
+        public SynchronizationContext()
+        {
+        }
+
+#if !FEATURE_APPX
+        public static SynchronizationContext Current => Thread.CurrentThread.SynchronizationContext;
+#endif
+
+        protected void SetWaitNotificationRequired() => _requireWaitNotification = true;
+
+        public bool IsWaitNotificationRequired() => _requireWaitNotification;
+
+        public virtual void Send(SendOrPostCallback d, object state) => d(state);
+
+        public virtual void Post(SendOrPostCallback d, object state) => ThreadPool.QueueUserWorkItem(s => s.d(s.state), (d, state), preferLocal: false);
+
+        /// <summary>
+        ///     Optional override for subclasses, for responding to notification that operation is starting.
+        /// </summary>
+        public virtual void OperationStarted()
+        {
+        }
+
+        /// <summary>
+        ///     Optional override for subclasses, for responding to notification that operation has completed.
+        /// </summary>
+        public virtual void OperationCompleted()
+        {
+        }
+
+        [CLSCompliant(false)]
+        public virtual int Wait(IntPtr[] waitHandles, bool waitAll, int millisecondsTimeout)
+        {
+            return WaitHelper(waitHandles, waitAll, millisecondsTimeout);
+        }
+
+        [CLSCompliant(false)]
+        protected static int WaitHelper(IntPtr[] waitHandles, bool waitAll, int millisecondsTimeout)
+        {
+            if (waitHandles == null)
+            {
+                throw new ArgumentNullException(nameof(waitHandles));
+            }
+
+            return WaitHelperNative(waitHandles, waitAll, millisecondsTimeout);
+        }
+
+        public static void SetSynchronizationContext(SynchronizationContext syncContext) => Thread.CurrentThread.SynchronizationContext = syncContext;
+
+        public virtual SynchronizationContext CreateCopy() => new SynchronizationContext();
+    }
+}

src/System.Private.CoreLib/src/System/Threading/SynchronizationContext.CoreCLR.cs

@@ -0,0 +1,22 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+// See the LICENSE file in the project root for more information.
+
+using System.Runtime.CompilerServices;
+
+namespace System.Threading
+{
+    public partial class SynchronizationContext
+    {
+        private static int InvokeWaitMethodHelper(SynchronizationContext syncContext, IntPtr[] waitHandles, bool waitAll, int millisecondsTimeout)
+        {
+            return syncContext.Wait(waitHandles, waitAll, millisecondsTimeout);
+        }
+
+        // Static helper to which the above method can delegate to in order to get the default
+        // COM behavior.
+        [MethodImplAttribute(MethodImplOptions.InternalCall)]
+        private static extern int WaitHelperNative(IntPtr[] waitHandles, bool waitAll, int millisecondsTimeout);
+
+    }
+}

src/System.Private.CoreLib/src/System/Threading/SynchronizationContext.Uap.cs

@@ -0,0 +1,77 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+// See the LICENSE file in the project root for more information.
+
+using System.Runtime.InteropServices;
+using System.Runtime.CompilerServices;
+using System.Reflection;
+using System.Diagnostics;
+
+namespace System.Threading
+{
+    public partial class SynchronizationContext
+    {
+        public static SynchronizationContext Current
+        {
+            get
+            {
+                SynchronizationContext context = Thread.CurrentThread.SynchronizationContext;
+
+                if (context == null && ApplicationModel.IsUap)
+                    context = GetWinRTContext();
+
+                return context;
+            }
+        }
+
+        private static SynchronizationContext GetWinRTContext()
+        {
+            Debug.Assert(Environment.IsWinRTSupported);
+            Debug.Assert(ApplicationModel.IsUap);
+
+            //
+            // We call into the VM to get the dispatcher.  This is because:
+            //
+            //  a) We cannot call the WinRT APIs directly from mscorlib, because we don't have the fancy projections here.
+            //  b) We cannot call into System.Runtime.WindowsRuntime here, because we don't want to load that assembly
+            //     into processes that don't need it (for performance reasons).
+            //
+            // So, we check the VM to see if the current thread has a dispatcher; if it does, we pass that along to
+            // System.Runtime.WindowsRuntime to get a corresponding SynchronizationContext.
+            //
+            object dispatcher = GetWinRTDispatcherForCurrentThread();
+            if (dispatcher != null)
+                return GetWinRTSynchronizationContext(dispatcher);
+
+            return null;
+        }
+
+        private static Func<object, SynchronizationContext> s_createSynchronizationContextDelegate;
+
+        private static SynchronizationContext GetWinRTSynchronizationContext(object dispatcher)
+        {
+            //
+            // Since we can't directly reference System.Runtime.WindowsRuntime from mscorlib, we have to get the factory via reflection.
+            // It would be better if we could just implement WinRTSynchronizationContextFactory in mscorlib, but we can't, because
+            // we can do very little with WinRT stuff in mscorlib.
+            //
+            Func<object, SynchronizationContext> createSynchronizationContextDelegate = s_createSynchronizationContextDelegate;
+            if (createSynchronizationContextDelegate == null)
+            {
+                Type factoryType = Type.GetType("System.Threading.WinRTSynchronizationContextFactory, System.Runtime.WindowsRuntime", throwOnError: true);
+                
+                // Create an instance delegate for the Create static method
+                MethodInfo createMethodInfo = factoryType.GetMethod("Create", BindingFlags.Static | BindingFlags.NonPublic | BindingFlags.Public);
+                createSynchronizationContextDelegate = (Func<object, SynchronizationContext>)Delegate.CreateDelegate(typeof(Func<object, SynchronizationContext>), createMethodInfo, /* throwOnBindFailure */ true);
+
+                s_createSynchronizationContextDelegate = createSynchronizationContextDelegate;
+            }
+
+            return s_createSynchronizationContextDelegate(dispatcher);
+        }
+
+        [DllImport(JitHelpers.QCall, CharSet = CharSet.Unicode)]
+        [return: MarshalAs(UnmanagedType.Interface)]
+        private static extern object GetWinRTDispatcherForCurrentThread();
+    }
+}