Code:
/ 4.0 / 4.0 / DEVDIV_TFS / Dev10 / Releases / RTMRel / wpf / src / Shared / MS / Win32 / ManagedWndProcTracker.cs / 1305600 / ManagedWndProcTracker.cs
//#define LOGGING
using System;
using System.Collections;
using System.Threading;
using System.Runtime.InteropServices;
using MS.Internal;
using MS.Internal.Interop;
using System.Security;
using System.Security.Permissions;
// The SecurityHelper class differs between assemblies and could not actually be
// shared, so it is duplicated across namespaces to prevent name collision.
#if WINDOWS_BASE
using MS.Internal.WindowsBase;
#elif PRESENTATION_CORE
using MS.Internal.PresentationCore;
#elif PRESENTATIONFRAMEWORK
using MS.Internal.PresentationFramework;
#elif DRT
using MS.Internal.Drt;
#else
#error Attempt to use a class (duplicated across multiple namespaces) from an unknown assembly.
#endif
namespace MS.Win32
{
internal static class ManagedWndProcTracker
{
///
/// Critical: This code calls into Link demanded methods to attach handlers
/// TreatAsSafe: This code does not take any parameter or return state.
/// It simply attaches private call back.
///
[SecurityCritical,SecurityTreatAsSafe]
static ManagedWndProcTracker()
{
// Listen for ProcessExit so we can detach ourselves when the CLR shuts down
// and avoid unmanaged code from calling back in to managed code during shutdown.
ManagedWndProcTrackerShutDownListener listener = new ManagedWndProcTrackerShutDownListener();
}
///
/// Critical: Uses critical member _hwndList
///
[SecurityCritical]
internal static void TrackHwndSubclass(HwndSubclass subclass, IntPtr hwnd)
{
lock (_hwndList)
{
// We use HwndSubclass as the key and the hwnd ptr as the value.
// This supports the case where two (or more) HwndSubclasses
// get attached to the same Hwnd. At AppDomain shutdown, we may
// end up sending the Detach message to the Hwnd more than once,
// but that won't cause any harm.
_hwndList[subclass] = hwnd;
}
#if LOGGING
LogStartHWND(hwnd, "Core HwndWrapper..ctor");
#endif
}
///
/// Critical: Uses critical member _hwndList
///
[SecurityCritical]
internal static void UnhookHwndSubclass(HwndSubclass subclass)
{
// if exiting the AppDomain, ignore this call. This avoids changing
// the list during the loop in OnAppDomainProcessExit
if (_exiting)
return;
lock (_hwndList)
{
_hwndList.Remove(subclass);
}
}
///
/// Critical performs an elevation to call HookUpDefWindowProc.
/// TreatAsSafe - net effect of this is to remove our already registered WndProc's on domain shutdown.
/// safe - as you had to elevate to add these already. Removing them is considered safe.
///
[SecurityCritical, SecurityTreatAsSafe]
private static void OnAppDomainProcessExit()
{
// AppDomain is exiting -- if anyone tries to call back into managed code
// after this point, bad things will happen. We must remove all unmanaged
// code references to our WndProc delegates. USER will explode if we set the
// WndProc to null, so the next most reasonable thing we can do is hook up
// the DefaultWindowProc.
//DbgUserBreakPoint();
_exiting = true;
lock (_hwndList)
{
new SecurityPermission(SecurityPermissionFlag.UnmanagedCode).Assert(); // BlessedAssert:
try
{
foreach (DictionaryEntry entry in _hwndList)
{
IntPtr hwnd = (IntPtr)entry.Value;
int windowStyle = UnsafeNativeMethods.GetWindowLong(new HandleRef(null,hwnd), NativeMethods.GWL_STYLE);
if((windowStyle & NativeMethods.WS_CHILD) != 0)
{
// Tell all the HwndSubclass WndProcs for WS_CHILD windows
// to detach themselves. This is particularly important when
// the parent hwnd belongs to a separate AppDomain in a
// cross AppDomain hosting scenario. In this scenario it is
// possible that the host has subclassed the WS_CHILD window
// and hence it is important to notify the host before we set the
// WndProc to DefWndProc. Also note that we do not want to make a
// blocking SendMessage call to all the subclassed Hwnds in the
// AppDomain because this can lead to slow shutdown speed.
// Eg. Consider a MessageOnlyHwnd created and subclassed on a
// worker thread which is no longer responsive. The SendMessage
// call in this case will block. To avoid this we limit the conversation
// only to WS_CHILD windows. We understand that this solution is
// not foolproof but it is the best outside of re-designing the cleanup
// of Hwnd subclasses.
UnsafeNativeMethods.SendMessage(hwnd, HwndSubclass.DetachMessage,
IntPtr.Zero /* wildcard */,
(IntPtr) 2 /* force and forward */);
}
// the last WndProc on the chain might be managed as well
// (see HwndSubclass.SubclassWndProc for explanation).
// Just in case, restore the DefaultWindowProc.
HookUpDefWindowProc(hwnd);
}
}
finally
{
CodeAccessPermission.RevertAssert();
}
}
}
///
/// TreatAsSafe: Demands for unmanaged code
/// Critical: Elevates by calling an unverifieds UnsafeNativeMethod call
///
[SecurityTreatAsSafe, SecurityCritical]
private static void HookUpDefWindowProc(IntPtr hwnd)
{
SecurityHelper.DemandUnmanagedCode();
#if LOGGING
LogFinishHWND(hwnd, "Core HookUpDWP");
#endif
IntPtr result = IntPtr.Zero ;
// We've already cleaned up, return immediately.
if (hwnd == IntPtr.Zero)
{
return;
}
IntPtr defWindowProc = GetDefWindowProcAddress(hwnd);
if (defWindowProc != IntPtr.Zero)
{
try
{
result = UnsafeNativeMethods.SetWindowLong(new HandleRef(null,hwnd), NativeMethods.GWL_WNDPROC, defWindowProc);
}
catch(System.ComponentModel.Win32Exception e)
{
// We failed to change the window proc. Now what?
if (e.NativeErrorCode != 1400) // ERROR_INVALID_WINDOW_HANDLE
{
// For debugging purposes, throw an exception so we can debug
// this and know if it's possible to call SetWindowLong on
// the wrong thread.
throw;
}
}
if (result != IntPtr.Zero )
{
UnsafeNativeMethods.PostMessage(new HandleRef(null,hwnd), WindowMessage.WM_CLOSE, IntPtr.Zero, IntPtr.Zero);
}
}
}
// Get the DWP for the given HWND -- returns DefWindowProcA or DefWindowProcW
// depending on IsWindowUnicode(hwnd).
private static IntPtr GetDefWindowProcAddress(IntPtr hwnd)
{
// We need to swap back in the DefWindowProc, but which one we use depends on
// what the Unicode-ness of the window.
if (SafeNativeMethods.IsWindowUnicode(new HandleRef(null,hwnd)))
{
if (_cachedDefWindowProcW == IntPtr.Zero)
{
_cachedDefWindowProcW = GetUser32ProcAddress("DefWindowProcW");
}
return _cachedDefWindowProcW;
}
else
{
if (_cachedDefWindowProcA == IntPtr.Zero)
{
_cachedDefWindowProcA = GetUser32ProcAddress("DefWindowProcA") ;
}
return _cachedDefWindowProcA;
}
}
///
/// SecurityCritical: elevates via a call to unsafe native methods
/// SecurityTreatAsSafe: Demands unmgd code permission via SecurityHelper
///
[SecurityCritical, SecurityTreatAsSafe]
private static IntPtr GetUser32ProcAddress(string export)
{
SecurityHelper.DemandUnmanagedCode();
IntPtr hModule = UnsafeNativeMethods.GetModuleHandle(ExternDll.User32);
if (hModule != IntPtr.Zero)
{
return UnsafeNativeMethods.GetProcAddress(new HandleRef(null, hModule), export);
}
return IntPtr.Zero;
}
private sealed class ManagedWndProcTrackerShutDownListener : ShutDownListener
{
///
/// Critical: accesses AppDomain.DomainUnload event
/// TreatAsSafe: This code does not take any parameter or return state.
/// It simply attaches private callbacks.
///
[SecurityCritical,SecurityTreatAsSafe]
public ManagedWndProcTrackerShutDownListener()
: base(null, ShutDownEvents.AppDomain)
{
}
internal override void OnShutDown(object target, object sender, EventArgs e)
{
ManagedWndProcTracker.OnAppDomainProcessExit();
}
}
#if LOGGING
[DllImport("ntdll.dll")]
private static extern void DbgUserBreakPoint();
[DllImport("ntdll.dll")]
private static extern void DbgPrint(string msg);
internal static void LogStartHWND(IntPtr hwnd, string fromWhere)
{
string msg = String.Format("BEGIN: {0:X} -- Setting DWP, process = {1} ({2}) {3}",
hwnd,
System.Diagnostics.Process.GetCurrentProcess().ProcessName,
fromWhere,
System.Environment.NewLine);
Log(msg);
}
internal static void LogFinishHWND(IntPtr hwnd, string fromWhere)
{
string msg = String.Format("END: {0:X} -- Setting DWP, process = {1} ({2}) {3}",
hwnd,
System.Diagnostics.Process.GetCurrentProcess().ProcessName,
fromWhere,
System.Environment.NewLine);
Log(msg);
}
private static void Log(string msg)
{
//DbgUserBreakPoint();
/*
byte[] msgBytes = System.Text.Encoding.ASCII.GetBytes(msg);
System.IO.FileStream fs = System.IO.File.Open("c:\\dwplog.txt", System.IO.FileMode.Append, System.IO.FileAccess.Write, System.IO.FileShare.ReadWrite);
fs.Write(msgBytes, 0, msgBytes.Length);
fs.Flush();
fs.Close();
*/
}
#endif
private static IntPtr _cachedDefWindowProcA = IntPtr.Zero;
private static IntPtr _cachedDefWindowProcW = IntPtr.Zero;
///
/// Critical - used as input to unsafe calls
///
[SecurityCritical]
private static Hashtable _hwndList = new Hashtable(10);
private static bool _exiting = false;
}
}
// File provided for Reference Use Only by Microsoft Corporation (c) 2007.
// Copyright (c) Microsoft Corporation. All rights reserved.
//#define LOGGING
using System;
using System.Collections;
using System.Threading;
using System.Runtime.InteropServices;
using MS.Internal;
using MS.Internal.Interop;
using System.Security;
using System.Security.Permissions;
// The SecurityHelper class differs between assemblies and could not actually be
// shared, so it is duplicated across namespaces to prevent name collision.
#if WINDOWS_BASE
using MS.Internal.WindowsBase;
#elif PRESENTATION_CORE
using MS.Internal.PresentationCore;
#elif PRESENTATIONFRAMEWORK
using MS.Internal.PresentationFramework;
#elif DRT
using MS.Internal.Drt;
#else
#error Attempt to use a class (duplicated across multiple namespaces) from an unknown assembly.
#endif
namespace MS.Win32
{
internal static class ManagedWndProcTracker
{
///
/// Critical: This code calls into Link demanded methods to attach handlers
/// TreatAsSafe: This code does not take any parameter or return state.
/// It simply attaches private call back.
///
[SecurityCritical,SecurityTreatAsSafe]
static ManagedWndProcTracker()
{
// Listen for ProcessExit so we can detach ourselves when the CLR shuts down
// and avoid unmanaged code from calling back in to managed code during shutdown.
ManagedWndProcTrackerShutDownListener listener = new ManagedWndProcTrackerShutDownListener();
}
///
/// Critical: Uses critical member _hwndList
///
[SecurityCritical]
internal static void TrackHwndSubclass(HwndSubclass subclass, IntPtr hwnd)
{
lock (_hwndList)
{
// We use HwndSubclass as the key and the hwnd ptr as the value.
// This supports the case where two (or more) HwndSubclasses
// get attached to the same Hwnd. At AppDomain shutdown, we may
// end up sending the Detach message to the Hwnd more than once,
// but that won't cause any harm.
_hwndList[subclass] = hwnd;
}
#if LOGGING
LogStartHWND(hwnd, "Core HwndWrapper..ctor");
#endif
}
///
/// Critical: Uses critical member _hwndList
///
[SecurityCritical]
internal static void UnhookHwndSubclass(HwndSubclass subclass)
{
// if exiting the AppDomain, ignore this call. This avoids changing
// the list during the loop in OnAppDomainProcessExit
if (_exiting)
return;
lock (_hwndList)
{
_hwndList.Remove(subclass);
}
}
///
/// Critical performs an elevation to call HookUpDefWindowProc.
/// TreatAsSafe - net effect of this is to remove our already registered WndProc's on domain shutdown.
/// safe - as you had to elevate to add these already. Removing them is considered safe.
///
[SecurityCritical, SecurityTreatAsSafe]
private static void OnAppDomainProcessExit()
{
// AppDomain is exiting -- if anyone tries to call back into managed code
// after this point, bad things will happen. We must remove all unmanaged
// code references to our WndProc delegates. USER will explode if we set the
// WndProc to null, so the next most reasonable thing we can do is hook up
// the DefaultWindowProc.
//DbgUserBreakPoint();
_exiting = true;
lock (_hwndList)
{
new SecurityPermission(SecurityPermissionFlag.UnmanagedCode).Assert(); // BlessedAssert:
try
{
foreach (DictionaryEntry entry in _hwndList)
{
IntPtr hwnd = (IntPtr)entry.Value;
int windowStyle = UnsafeNativeMethods.GetWindowLong(new HandleRef(null,hwnd), NativeMethods.GWL_STYLE);
if((windowStyle & NativeMethods.WS_CHILD) != 0)
{
// Tell all the HwndSubclass WndProcs for WS_CHILD windows
// to detach themselves. This is particularly important when
// the parent hwnd belongs to a separate AppDomain in a
// cross AppDomain hosting scenario. In this scenario it is
// possible that the host has subclassed the WS_CHILD window
// and hence it is important to notify the host before we set the
// WndProc to DefWndProc. Also note that we do not want to make a
// blocking SendMessage call to all the subclassed Hwnds in the
// AppDomain because this can lead to slow shutdown speed.
// Eg. Consider a MessageOnlyHwnd created and subclassed on a
// worker thread which is no longer responsive. The SendMessage
// call in this case will block. To avoid this we limit the conversation
// only to WS_CHILD windows. We understand that this solution is
// not foolproof but it is the best outside of re-designing the cleanup
// of Hwnd subclasses.
UnsafeNativeMethods.SendMessage(hwnd, HwndSubclass.DetachMessage,
IntPtr.Zero /* wildcard */,
(IntPtr) 2 /* force and forward */);
}
// the last WndProc on the chain might be managed as well
// (see HwndSubclass.SubclassWndProc for explanation).
// Just in case, restore the DefaultWindowProc.
HookUpDefWindowProc(hwnd);
}
}
finally
{
CodeAccessPermission.RevertAssert();
}
}
}
///
/// TreatAsSafe: Demands for unmanaged code
/// Critical: Elevates by calling an unverifieds UnsafeNativeMethod call
///
[SecurityTreatAsSafe, SecurityCritical]
private static void HookUpDefWindowProc(IntPtr hwnd)
{
SecurityHelper.DemandUnmanagedCode();
#if LOGGING
LogFinishHWND(hwnd, "Core HookUpDWP");
#endif
IntPtr result = IntPtr.Zero ;
// We've already cleaned up, return immediately.
if (hwnd == IntPtr.Zero)
{
return;
}
IntPtr defWindowProc = GetDefWindowProcAddress(hwnd);
if (defWindowProc != IntPtr.Zero)
{
try
{
result = UnsafeNativeMethods.SetWindowLong(new HandleRef(null,hwnd), NativeMethods.GWL_WNDPROC, defWindowProc);
}
catch(System.ComponentModel.Win32Exception e)
{
// We failed to change the window proc. Now what?
if (e.NativeErrorCode != 1400) // ERROR_INVALID_WINDOW_HANDLE
{
// For debugging purposes, throw an exception so we can debug
// this and know if it's possible to call SetWindowLong on
// the wrong thread.
throw;
}
}
if (result != IntPtr.Zero )
{
UnsafeNativeMethods.PostMessage(new HandleRef(null,hwnd), WindowMessage.WM_CLOSE, IntPtr.Zero, IntPtr.Zero);
}
}
}
// Get the DWP for the given HWND -- returns DefWindowProcA or DefWindowProcW
// depending on IsWindowUnicode(hwnd).
private static IntPtr GetDefWindowProcAddress(IntPtr hwnd)
{
// We need to swap back in the DefWindowProc, but which one we use depends on
// what the Unicode-ness of the window.
if (SafeNativeMethods.IsWindowUnicode(new HandleRef(null,hwnd)))
{
if (_cachedDefWindowProcW == IntPtr.Zero)
{
_cachedDefWindowProcW = GetUser32ProcAddress("DefWindowProcW");
}
return _cachedDefWindowProcW;
}
else
{
if (_cachedDefWindowProcA == IntPtr.Zero)
{
_cachedDefWindowProcA = GetUser32ProcAddress("DefWindowProcA") ;
}
return _cachedDefWindowProcA;
}
}
///
/// SecurityCritical: elevates via a call to unsafe native methods
/// SecurityTreatAsSafe: Demands unmgd code permission via SecurityHelper
///
[SecurityCritical, SecurityTreatAsSafe]
private static IntPtr GetUser32ProcAddress(string export)
{
SecurityHelper.DemandUnmanagedCode();
IntPtr hModule = UnsafeNativeMethods.GetModuleHandle(ExternDll.User32);
if (hModule != IntPtr.Zero)
{
return UnsafeNativeMethods.GetProcAddress(new HandleRef(null, hModule), export);
}
return IntPtr.Zero;
}
private sealed class ManagedWndProcTrackerShutDownListener : ShutDownListener
{
///
/// Critical: accesses AppDomain.DomainUnload event
/// TreatAsSafe: This code does not take any parameter or return state.
/// It simply attaches private callbacks.
///
[SecurityCritical,SecurityTreatAsSafe]
public ManagedWndProcTrackerShutDownListener()
: base(null, ShutDownEvents.AppDomain)
{
}
internal override void OnShutDown(object target, object sender, EventArgs e)
{
ManagedWndProcTracker.OnAppDomainProcessExit();
}
}
#if LOGGING
[DllImport("ntdll.dll")]
private static extern void DbgUserBreakPoint();
[DllImport("ntdll.dll")]
private static extern void DbgPrint(string msg);
internal static void LogStartHWND(IntPtr hwnd, string fromWhere)
{
string msg = String.Format("BEGIN: {0:X} -- Setting DWP, process = {1} ({2}) {3}",
hwnd,
System.Diagnostics.Process.GetCurrentProcess().ProcessName,
fromWhere,
System.Environment.NewLine);
Log(msg);
}
internal static void LogFinishHWND(IntPtr hwnd, string fromWhere)
{
string msg = String.Format("END: {0:X} -- Setting DWP, process = {1} ({2}) {3}",
hwnd,
System.Diagnostics.Process.GetCurrentProcess().ProcessName,
fromWhere,
System.Environment.NewLine);
Log(msg);
}
private static void Log(string msg)
{
//DbgUserBreakPoint();
/*
byte[] msgBytes = System.Text.Encoding.ASCII.GetBytes(msg);
System.IO.FileStream fs = System.IO.File.Open("c:\\dwplog.txt", System.IO.FileMode.Append, System.IO.FileAccess.Write, System.IO.FileShare.ReadWrite);
fs.Write(msgBytes, 0, msgBytes.Length);
fs.Flush();
fs.Close();
*/
}
#endif
private static IntPtr _cachedDefWindowProcA = IntPtr.Zero;
private static IntPtr _cachedDefWindowProcW = IntPtr.Zero;
///
/// Critical - used as input to unsafe calls
///
[SecurityCritical]
private static Hashtable _hwndList = new Hashtable(10);
private static bool _exiting = false;
}
}
// File provided for Reference Use Only by Microsoft Corporation (c) 2007.
// Copyright (c) Microsoft Corporation. All rights reserved.
Link Menu
This book is available now!
Buy at Amazon US or
Buy at Amazon UK
- DtrList.cs
- PageAsyncTask.cs
- TreeNodeStyle.cs
- ComponentChangedEvent.cs
- MappingMetadataHelper.cs
- MappingException.cs
- ResourceExpressionEditor.cs
- RectangleGeometry.cs
- SqlAggregateChecker.cs
- ComplusTypeValidator.cs
- MimeTypePropertyAttribute.cs
- SimpleType.cs
- ServiceModelEnhancedConfigurationElementCollection.cs
- SqlDataSourceSelectingEventArgs.cs
- Pointer.cs
- querybuilder.cs
- SQLUtility.cs
- CancellableEnumerable.cs
- Visual3D.cs
- CodeTypeMember.cs
- ControllableStoryboardAction.cs
- Directory.cs
- TraceSection.cs
- CustomActivityDesigner.cs
- DataShape.cs
- TransformGroup.cs
- PeerCollaboration.cs
- QualificationDataItem.cs
- DocumentViewerConstants.cs
- TrackingProfileSerializer.cs
- GacUtil.cs
- safelink.cs
- TextRangeEditLists.cs
- ClientRuntimeConfig.cs
- RenderDataDrawingContext.cs
- DataPager.cs
- CompoundFileStorageReference.cs
- BamlRecordWriter.cs
- XmlElement.cs
- PanelStyle.cs
- XmlWriterDelegator.cs
- UTF7Encoding.cs
- SaveFileDialog.cs
- ButtonPopupAdapter.cs
- SerializationSectionGroup.cs
- Serializer.cs
- WebPartConnectionsConnectVerb.cs
- EditCommandColumn.cs
- ScaleTransform.cs
- TransactionManager.cs
- SmiMetaDataProperty.cs
- FixedPage.cs
- figurelengthconverter.cs
- NTAccount.cs
- StyleCollection.cs
- ToolStripContentPanel.cs
- VolatileEnlistmentState.cs
- ValidatedMobileControlConverter.cs
- Geometry3D.cs
- TypeSemantics.cs
- HttpProtocolReflector.cs
- MailAddress.cs
- TargetInvocationException.cs
- ReadOnlyHierarchicalDataSource.cs
- Utility.cs
- PreloadedPackages.cs
- DataServiceQuery.cs
- odbcmetadatafactory.cs
- VectorConverter.cs
- Point4D.cs
- CustomWebEventKey.cs
- RuntimeTrackingProfile.cs
- SqlNamer.cs
- ListViewItemEventArgs.cs
- SecurityHeader.cs
- BuildProvider.cs
- AsyncOperation.cs
- CustomPeerResolverService.cs
- SecurityTokenValidationException.cs
- ParenthesizePropertyNameAttribute.cs
- LineVisual.cs
- ISCIIEncoding.cs
- BitmapEffectDrawingContextWalker.cs
- CorePropertiesFilter.cs
- ContentOperations.cs
- XmlSchemaFacet.cs
- odbcmetadatacolumnnames.cs
- MobileContainerDesigner.cs
- Propagator.JoinPropagator.cs
- TileModeValidation.cs
- ModelProperty.cs
- Point3DAnimation.cs
- OracleConnectionString.cs
- CodeChecksumPragma.cs
- Rights.cs
- RuntimeConfigLKG.cs
- ReflectPropertyDescriptor.cs
- CompiledIdentityConstraint.cs
- XNodeValidator.cs
- EditBehavior.cs