From cf899ee70754ceea15aa7a9ef3ab95707d5e66f7 Mon Sep 17 00:00:00 2001
From: Dmitry Fleytman <dfleytma@redhat.com>
Date: Tue, 15 Apr 2014 18:16:59 +0300
Subject: [PATCH] UsbDk: Extend request interface with buffers lengthes

Signed-off-by: Dmitry Fleytman <dfleytma@redhat.com>
---
 UsbDk/WdfRequest.h | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/UsbDk/WdfRequest.h b/UsbDk/WdfRequest.h
index 3752e6d..c2638fd 100644
--- a/UsbDk/WdfRequest.h
+++ b/UsbDk/WdfRequest.h
@@ -15,15 +15,15 @@ public:
     }
 
     template <typename TObject>
-    NTSTATUS FetchInputObject(TObject &objectPtr)
+    NTSTATUS FetchInputObject(TObject &objectPtr, size_t *Length = nullptr)
     {
-        return WdfRequestRetrieveInputBuffer(m_Request, sizeof(*objectPtr), (PVOID *)&objectPtr, nullptr);
+        return WdfRequestRetrieveInputBuffer(m_Request, sizeof(*objectPtr), (PVOID *)&objectPtr, Length);
     }
 
     template <typename TObject>
-    NTSTATUS FetchOutputObject(TObject &objectPtr)
+    NTSTATUS FetchOutputObject(TObject &objectPtr, size_t *Length = nullptr)
     {
-        return WdfRequestRetrieveOutputBuffer(m_Request, sizeof(*objectPtr), (PVOID *)&objectPtr, nullptr);
+        return WdfRequestRetrieveOutputBuffer(m_Request, sizeof(*objectPtr), (PVOID *)&objectPtr, Length);
     }
 
     template <typename TObject>