下載app免費領(lǐng)取會員
HOOK(鉤子,掛鉤)是一種實現(xiàn)Windows平臺下類似于中斷的機制。HOOK機制允許應用程序攔截并處理Windows消息或指定事件,當指定的消息發(fā)出后,HOOK程序就可以在消息到達目標窗口之前將其捕獲,從而得到對消息的控制權(quán),進而可以對該消息進行處理或修改,加入我們所需的功能。鉤子按使用范圍分,可分為線程鉤子和系統(tǒng)鉤子,其中,系統(tǒng)鉤子具有相當大的功能,幾乎可以實現(xiàn)對所有Windows消息的攔截、處理和監(jiān)控。這項技術(shù)涉及到兩個重要的API,一個是SetWindowsHookEx,安裝鉤子;另一個是UnHookWindowsHookEx,卸載鉤子。
本文使用的HOOK API技術(shù),是指截獲系統(tǒng)或進程對某個API函數(shù)的調(diào)用,使得API的執(zhí)行流程轉(zhuǎn)向我們指定的代碼段,從而實現(xiàn)我們所需的功能。Windows下的每個進程均擁有自己的地址空間,并且進程只能調(diào)用其地址空間內(nèi)的函數(shù),因此HOOK API尤為關(guān)鍵的一步是,設法將自己的代碼段注入到目標進程中,才能進一步實現(xiàn)對該進程調(diào)用的API進行攔截。然而微軟并沒有提供HOOK API的調(diào)用接口,這就需要開發(fā)者自己編程實現(xiàn)。
一般來說,HOOK API由兩個組成部分,即實現(xiàn)HOOK API的DLL文件,和啟動注入的主調(diào)程序。本文采用HOOK API 技術(shù)對剪切板相關(guān)的API 函數(shù)進行攔截,從而實現(xiàn)對剪切板內(nèi)容的監(jiān)控功能,同樣使用該技術(shù)實現(xiàn)進程防終止功能。其中DLL文件支持HOOK API的實現(xiàn),而主調(diào)客戶端程序?qū)⒃诔跏蓟瘯r把帶有HOOK API功能的DLL隨著鼠標鉤子的加載注入到目標進程中,這里的鼠標鉤子屬于系統(tǒng)鉤子。
下面介紹在Revit中,如何應用Hook對鼠標雙擊元素事件進行攔截。
第一步,先封裝HookBase抽象類,因所有Hook的都具有注冊、卸載邏輯,且注冊、卸載大同小易。如下
public abstract class HookBase : IHook
{
private static Dictionary<int, IHook> m_Hooks;
private IntPtr m_ProcessId;
private int m_ThreadId;
private HookType m_HookType;
private HookProc m_HookProc;
protected internal int m_HookId;
static HookBase(){
m_Hooks = new Dictionary<int, IHook>();
}
private HookBase(HookType hookType){
m_HookType = hookType;
m_HookProc = HookProc;
}
protected HookBase(IntPtr processId, HookType hookType):this(hookType){
m_ProcessId = processId;
if (m_ProcessId == IntPtr.Zero)
{
m_ProcessId = HookHelper.GetCurrentProcessId();
}
}
protected HookBase(int threadId, HookType hookType):this(hookType){
m_ThreadId = threadId;
if (m_ThreadId == 0)
{
m_ThreadId = HookHelper.GetCurrentThreadId();
}
}
public void Install(){
if (m_ThreadId != 0)
{
m_HookId = HookHelper.SetWindowsHookEx(m_HookType, m_HookProc, IntPtr.Zero, m_ThreadId);
}
else
{
if (m_ProcessId == IntPtr.Zero)
{
return;
}
m_HookId = HookHelper.SetWindowsHookEx(m_HookType, m_HookProc, m_ProcessId,0);
}
if (m_HookId == 0)
{
return;
}
if (!m_Hooks.ContainsKey(m_HookId))
{
m_Hooks.Add(m_HookId, this);
}
}
public void Uninstall()
{
if (m_HookId == 0)
{
return;
}
var flag = HookHelper.UnhookWindowsHookEx(m_HookId);
if (flag)
{
if (m_Hooks.Remove(m_HookId))
{
m_HookId = 0;
}
}
}
protected abstract int HookProc(int nCode, IntPtr wParam, IntPtr lParam);
第二步 ,因鼠標Hook分為線程鼠標Hook以及全局鼠標Hook兩種,僅注冊方式有點區(qū)別。為使用方便,將其封裝為事件注冊方式。如下
public abstract class MouseHookBase : HookBase
{
protected MouseHookBase(IntPtr processId)
: base(processId, HookType.WH_MOUSE_LL)
{
}
protected MouseHookBase(int threadId)
: base(threadId, HookType.WH_MOUSE)
{
}
public event HookHandler<MouseEventArgs> MouseDoubleClick;
public event HookHandler<MouseEventArgs> MouseMove;
public event HookHandler<MouseEventArgs> MouseDown;
public event HookHandler<MouseEventArgs> MouseUp;
protected override int HookProc(int nCode, IntPtr wParam, IntPtr lParam)
{
if (nCode < 0)
{
return HookHelper.CallNextHookEx(m_HookId, nCode, wParam, lParam);
}
var mouseMsg = (MouseMessage)wParam.ToInt32();
var mouseHookStruct = lParam.ToStruct<MOUSEHOOKSTRUCT>();
var button = this.GetMouseButtons(mouseMsg);
switch (mouseMsg)
{
case MouseMessage.WM_LBUTTONDOWN:
case MouseMessage.WM_RBUTTONDOWN:
case MouseMessage.WM_MBUTTONDOWN:
return this.OnRaiseMouseDown(button, 1, mouseHookStruct.pt.X, mouseHookStruct.pt.Y, mouseHookStruct.mouseData);
case MouseMessage.WM_LBUTTONUP:
case MouseMessage.WM_MBUTTONUP:
case MouseMessage.WM_RBUTTONUP:
return this.OnRaiseMouseUp(button, 1, mouseHookStruct.pt.X, mouseHookStruct.pt.Y, mouseHookStruct.mouseData);
case MouseMessage.WM_LBUTTONDBLCLK:
case MouseMessage.WM_RBUTTONDBLCLK:
case MouseMessage.WM_MBUTTONDBLCLK:
return this.OnRaiseMouseDoubleClick(button, 2, mouseHookStruct.pt.X, mouseHookStruct.pt.Y, mouseHookStruct.mouseData);
case MouseMessage.WM_MOUSEMOVE:
return this.OnRaiseMouseMove(MouseButtons.None, 0, mouseHookStruct.pt.X, mouseHookStruct.pt.Y, mouseHookStruct.mouseData);
default:
return HookHelper.CallNextHookEx(m_HookId, nCode, wParam, lParam);
}
}
private MouseButtons GetMouseButtons(MouseMessage mouseMsg)
{
MouseButtons result = MouseButtons.None;
switch (mouseMsg)
{
case MouseMessage.WM_LBUTTONDBLCLK:
case MouseMessage.WM_LBUTTONDOWN:
case MouseMessage.WM_LBUTTONUP:
result = MouseButtons.Left;
break;
case MouseMessage.WM_MBUTTONDBLCLK:
case MouseMessage.WM_MBUTTONDOWN:
case MouseMessage.WM_MBUTTONUP:
result = MouseButtons.Middle;
break;
case MouseMessage.WM_RBUTTONDBLCLK:
case MouseMessage.WM_RBUTTONDOWN:
case MouseMessage.WM_RBUTTONUP:
result = MouseButtons.Right;
break;
}
return result;
}
private int OnRaiseMouseDoubleClick(MouseButtons button, int clicks, int x, int y, intdelta)
{
if (this.MouseDoubleClick != null)
{
return this.MouseDoubleClick(this, new MouseEventArgs(button, clicks, x, y, delta));
}
return 0;
}
private int OnRaiseMouseDown(MouseButtons button, int clicks, int x, int y, int delta)
{
if (this.MouseDown != null)
{
return this.MouseDown(this, new MouseEventArgs(button, clicks, x, y, delta));
}
return 0;
}
private int OnRaiseMouseUp(MouseButtons button, int clicks, int x, int y, int delta)
{
if (this.MouseUp != null)
{
return this.MouseUp(this, new MouseEventArgs(button, clicks, x, y, delta));
}
return 0;
}
private int OnRaiseMouseMove(MouseButtons button, int clicks, int x, int y, int delta)
{
if (this.MouseMove != null)
{
return this.MouseMove(this, new MouseEventArgs(button, clicks, x, y, delta));
}
return 0;
}
}
第三步,依次實現(xiàn)線程鼠標Hook以及全局鼠標Hook.
public class MouseHook : MouseHookBase{
public MouseHook(int threadId = 0)
: base(threadId)
{
}
}
public class GlobalMouseHook : MouseHookBase
{
public GlobalMouseHook(IntPtr processId)
: base(processId)
{
}
}
第四步,有了鼠標Hook,我們?nèi)绻赗evit內(nèi)使用并且攔截鼠標雙擊元素事件呢?我們繼續(xù)封裝一個元素監(jiān)控類 ,如下:
public class ElementMonitor
{
private static ElementMonitor m_Instance;
private MouseHook m_MouseHook;
private bool m_IsMonitor;
private UIApplication m_UIApplication;
private ElementMonitor(UIApplication uiApp)
{
m_Instance = this;
m_UIApplication = uiApp;
m_MouseHook = new MouseHook();
m_MouseHook.Install();
m_MouseHook.MouseDoubleClick += OnRaiseMouseDoubleClick;
}
/// <summary>
/// 靜態(tài)實例,可在入口類判斷此實例是否為null,防止重復注冊.
/// </summary>
public static ElementMonitor Instance
{
get
{
return m_Instance;
}
}
/// <summary>
/// 當鼠標雙擊元素時觸發(fā)此事件.
/// </summary>
public event HookHandler<DoubleClickElementEventArgs> DoubleClickElement;
/// <summary>
/// 注冊元素監(jiān)控,并指定是否立即監(jiān)控.
/// </summary>
public static void Register(UIApplication uiApp, bool immediatelyMonitor = true)
{
if (uiApp == null)
{
throw new ArgumentNullException(nameof(uiApp));
}
new ElementMonitor(uiApp)
{
m_IsMonitor = immediatelyMonitor
};
}
/// <summary>
/// 注冊元素監(jiān)控,并指定是否立即監(jiān)控.
/// </summary>
public static void Register(UIControlledApplication uiControllApp, bool immediatelyMonitor = true)
{
if (uiControllApp == null)
{
throw new ArgumentNullException(nameof(uiControllApp));
}
var flag = System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance | System.Reflection.BindingFlags.InvokeMethod;
var uiApp = (UIApplication)uiControllApp.GetType().InvokeMember("getUIApplication", flag, Type.DefaultBinder, uiControllApp, null);
Register(uiApp, immediatelyMonitor);
}
/// <summary>
/// 返回1,則攔截鼠標消息,返回0則傳遞給真正消息接收者.
/// </summary>
private int OnRaiseMouseDoubleClick(object sender, System.Windows.Forms.MouseEventArgs e)
{
if (!m_IsMonitor || e.Button != MouseButtons.Left || e.Clicks != 2)
{
return 0;
}
var uiDoc = m_UIApplication.ActiveUIDocument;
if (uiDoc == null)
{
return 0;
}
var elemIds = uiDoc.Selection.GetElementIds();
if (elemIds.Count == 1)
{
var elem = uiDoc.Document.GetElement(elemIds.First());
if (elem == null)
{
return 0;
}
if (this.DoubleClickElement == null)
{
return 0;
}
return this.DoubleClickElement(this, new DoubleClickElementEventArgs(elem));
}
return 0;
}
}
第五步,調(diào)用測試,如下
[Transaction(TransactionMode.Manual)]
public class MouseHookTest : IExternalCommand
{
Result IExternalCommand.Execute(ExternalCommandData commandData, ref string message, ElementSet elements)
{
if (ElementMonitor.Instance == null)
{
ElementMonitor.Register(commandData.Application);
}
ElementMonitor.Instance.DoubleClickElement += OnRaiseDoubleClickElement;
return Result.Succeeded;
}
private int OnRaiseDoubleClickElement(object sender, DoubleClickElementEventArgs e)
{
if (e.Element == null)
{
return 0;
}
System.Windows.Forms.MessageBox.Show(string.Format("雙擊擊元素Id: {0}", e.Element.Id));
return 1;
}
}
本文版權(quán)歸腿腿教學網(wǎng)及原創(chuàng)作者所有,未經(jīng)授權(quán),謝絕轉(zhuǎn)載。
上一篇:Revit二次開發(fā)教程:Revit族參數(shù)可見性設置
推薦專題