diff --git a/UsbDk/WdfRequest.h b/UsbDk/WdfRequest.h index 8827f53..214efc4 100644 --- a/UsbDk/WdfRequest.h +++ b/UsbDk/WdfRequest.h @@ -23,6 +23,11 @@ #pragma once +using WDF_REQUEST_CONTEXT = struct {}; +using PWDF_REQUEST_CONTEXT = WDF_REQUEST_CONTEXT*; + +WDF_DECLARE_CONTEXT_TYPE_WITH_NAME(WDF_REQUEST_CONTEXT, WdfRequestGetContext); + class CWdfRequest { public: @@ -109,6 +114,9 @@ public: return Request; } + PWDF_REQUEST_CONTEXT Context() const + { return WdfRequestGetContext(m_Request); } + operator WDFREQUEST() const { return m_Request; }