From cdff8c0bae9f420bbb6152733c0ea273c71f3fa1 Mon Sep 17 00:00:00 2001
From: Pavel Gurvich <pavel@daynix.com>
Date: Mon, 5 May 2014 11:11:09 +0300
Subject: [PATCH] UsbDkHelper: Add interface GetRedirectorSystemHandle()

Signed-off-by: Pavel Gurvich <pavel@daynix.com>
Signed-off-by: Dmitry Fleytman <dfleytma@redhat.com>
---
 UsbDkHelper/RedirectorAccess.h | 3 +++
 UsbDkHelper/UsbDkHelper.cpp    | 7 +++++++
 UsbDkHelper/UsbDkHelper.h      | 2 ++
 3 files changed, 12 insertions(+)

diff --git a/UsbDkHelper/RedirectorAccess.h b/UsbDkHelper/RedirectorAccess.h
index 4dd15ce..bf863c7 100644
--- a/UsbDkHelper/RedirectorAccess.h
+++ b/UsbDkHelper/RedirectorAccess.h
@@ -38,5 +38,8 @@ public:
         DWORD BytesWritten;
         Write(&Request, sizeof(Request), &BytesWritten);
     }
+
+    HANDLE GetSystemHandle() const
+    { return m_hDriver; }
 };
 //-----------------------------------------------------------------------------------
\ No newline at end of file
diff --git a/UsbDkHelper/UsbDkHelper.cpp b/UsbDkHelper/UsbDkHelper.cpp
index 9ce0987..cd1d03f 100644
--- a/UsbDkHelper/UsbDkHelper.cpp
+++ b/UsbDkHelper/UsbDkHelper.cpp
@@ -248,3 +248,10 @@ BOOL ReadPipe(HANDLE DeviceHandle, PUSB_DK_TRANSFER_REQUEST Request)
     }
 }
 //-------------------------------------------------------------------------------------------
+
+HANDLE GetRedirectorSystemHandle(HANDLE DeviceHandle)
+{
+    auto deviceHandle = reinterpret_cast<PREDIRECTED_DEVICE_HANDLE>(DeviceHandle);
+    return deviceHandle->RedirectorAccess->GetSystemHandle();
+}
+//-------------------------------------------------------------------------------------------
diff --git a/UsbDkHelper/UsbDkHelper.h b/UsbDkHelper/UsbDkHelper.h
index 06e15e5..1814a56 100644
--- a/UsbDkHelper/UsbDkHelper.h
+++ b/UsbDkHelper/UsbDkHelper.h
@@ -40,6 +40,8 @@ extern "C" {
 
     DLL BOOL             WritePipe(HANDLE DeviceHandle, PUSB_DK_TRANSFER_REQUEST Request);
     DLL BOOL             ReadPipe(HANDLE DeviceHandle, PUSB_DK_TRANSFER_REQUEST Request);
+
+    DLL HANDLE           GetRedirectorSystemHandle(HANDLE DeviceHandle);
 #ifdef __cplusplus
 }
 #endif