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 - 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 - 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