fix #1 call addref in queryinterface

Connicpu · Jan 22, 2021 · 9803f31 · 9803f31
1 parent bc6bb53
commit 9803f31
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/derive-com-impl/src/derive.rs b/derive-com-impl/src/derive.rs
@@ -122,6 +122,8 @@ impl<'a> ComImpl<'a> {
                         return winapi::shared::winerror::E_POINTER;
                     }
                     if #( #is_equal_iid )||* {
+                        let that = &*(this as *const Self);
+                        that.#refcount.add_ref();
                         *ppv = this as *mut winapi::ctypes::c_void;
                         winapi::shared::winerror::S_OK
                     } else {