UsbDkHelper: Add persistent hide rules interface

Signed-off-by: Kirill Moizik <[email protected]> Signed-off-by: Dmitry Fleytman <[email protected]>
xiayibing · Mar 25, 2015 · eb8c876 · eb8c876
1 parent 408a787
commit eb8c876
Show file tree

Hide file tree

Showing 2 changed files with 69 additions and 0 deletions.
diff --git a/UsbDkHelper/UsbDkHelper.cpp b/UsbDkHelper/UsbDkHelper.cpp
@@ -30,6 +30,7 @@
 #include "Installer.h"
 #include "DriverAccess.h"
 #include "RedirectorAccess.h"
+#include "RuleManager.h"
 
 //-------------------------------------------------------------------------------------------
 
@@ -315,3 +316,30 @@ void UsbDk_CloseHiderHandle(HANDLE HiderHandle)
 {
     delete reinterpret_cast<UsbDkHiderAccess *>(HiderHandle);
 }
+
+static inline
+BOOL ModifyPersistentHideRules(const USB_DK_HIDE_RULE &Rule,
+                               void(CRulesManager::*Modifier)(const USB_DK_HIDE_RULE&))
+{
+    try
+    {
+        CRulesManager Manager;
+        (Manager.*Modifier)(Rule);
+        return TRUE;
+    }
+    catch (const exception &e)
+    {
+        printExceptionString(e.what());
+        return FALSE;
+    }
+}
+
+DLL BOOL UsbDk_AddPersistentHideRule(PUSB_DK_HIDE_RULE Rule)
+{
+    return ModifyPersistentHideRules(*Rule, &CRulesManager::AddRule);
+}
+
+DLL BOOL UsbDk_DeletePersistentHideRule(PUSB_DK_HIDE_RULE Rule)
+{
+    return ModifyPersistentHideRules(*Rule, &CRulesManager::DeleteRule);
+}
diff --git a/UsbDkHelper/UsbDkHelperHider.h b/UsbDkHelper/UsbDkHelperHider.h
@@ -104,6 +104,47 @@ extern "C" {
     *
     */
     DLL void             UsbDk_CloseHiderHandle(HANDLE HiderHandle);
+
+    /* Add rule for detaching USB devices from OS stack persistently.
+    *  The rule consists of:
+    *
+    * class, vendor, product, version, allow
+    *
+    * Use -1 for @class/@vendor/@product/@version to accept any value.
+    *
+    * @params
+    *    IN  - Rule - pointer to hide rule
+    *    OUT - None
+    *
+    * @return
+    *  TRUE if function succeeds
+    *
+    * @note
+    * 1. Persistent rule stays until explicitly deleted by
+    *    UsbDk_DeletePersistentHideRule()
+    * 2. This API requires administrative privileges
+    * 3. For already attached devices the rule will be applied after
+    *    device re-plug or system reboot.
+    *
+    */
+    DLL BOOL             UsbDk_AddPersistentHideRule(PUSB_DK_HIDE_RULE Rule);
+
+    /* Delete specific persistent hide rule
+    *
+    * @params
+    *    IN  - Rule - pointer to hide rule
+    *    OUT - None
+    *
+    * @return
+    *  TRUE if function succeeds
+    *
+    * @note
+    * 1. This API requires administrative privileges
+    * 2. For already attached devices the rule will be applied after
+    *    device re-plug or system reboot.
+    *
+    */
+    DLL BOOL             UsbDk_DeletePersistentHideRule(PUSB_DK_HIDE_RULE Rule);
 #ifdef __cplusplus
 }
 #endif