lee/OptiScaler
1
0
Fork 0
mirror of https://github.com/optiscaler/OptiScaler.git synced 2026-05-04 00:31:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Used mutex for AddRef / Release

Browse Source
This commit is contained in:
cdozdil
2025-07-15 15:17:51 +03:00
parent 7eff8ec5d2
commit 1339937149
2 changed files with 36 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
OptiScaler/hooks/wrapped_swapchain.cpp
+34 -24
View File
@@ -20,12 +20,29 @@ WrappedIDXGISwapChain4::WrappedIDXGISwapChain4(IDXGISwapChain* real, IUnknown* p
ReleaseTrig(releaseTrig), m_iRefcount(1), UWP(isUWP)
{
id = ++scCount;
LOG_INFO("{0} created, real: {1:X}", id, (UINT64) real);
m_pReal->QueryInterface(IID_PPV_ARGS(&m_pReal1));
if (m_pReal1 != nullptr)
m_pReal->Release();
m_pReal->QueryInterface(IID_PPV_ARGS(&m_pReal2));
if (m_pReal2 != nullptr)
m_pReal->Release();
m_pReal->QueryInterface(IID_PPV_ARGS(&m_pReal3));
if (m_pReal3 != nullptr)
m_pReal->Release();
m_pReal->QueryInterface(IID_PPV_ARGS(&m_pReal4));
if (m_pReal4 != nullptr)
m_pReal->Release();
m_pReal->AddRef();
auto refCount = m_pReal->Release();
Device2 = Device;
LOG_INFO("{} created, real: {:X}, refCount: {}", id, (UINT64) real, refCount);
}
WrappedIDXGISwapChain4::~WrappedIDXGISwapChain4() {}
@@ -33,6 +50,8 @@ WrappedIDXGISwapChain4::~WrappedIDXGISwapChain4() {}
//
HRESULT STDMETHODCALLTYPE WrappedIDXGISwapChain4::QueryInterface(REFIID riid, void** ppvObject)
{
LOG_TRACE("Caller: {}", Util::WhoIsTheCaller(_ReturnAddress()));
if (riid == __uuidof(IDXGISwapChain))
{
AddRef();
@@ -121,42 +140,33 @@ HRESULT STDMETHODCALLTYPE WrappedIDXGISwapChain4::QueryInterface(REFIID riid, vo
ULONG STDMETHODCALLTYPE WrappedIDXGISwapChain4::AddRef()
{
InterlockedIncrement(&m_iRefcount);
LOG_DEBUG_ONLY("Count: {}", m_iRefcount);
LOG_TRACE("Waiting mutex, owner: {}", _refMutex.getOwner());
OwnedLockGuard lock(_refMutex, 1);
m_iRefcount++;
LOG_TRACE("Count: {}, caller: {}", m_iRefcount, Util::WhoIsTheCaller(_ReturnAddress()));
return m_iRefcount;
}
ULONG STDMETHODCALLTYPE WrappedIDXGISwapChain4::Release()
{
auto ret = InterlockedDecrement(&m_iRefcount);
LOG_DEBUG_ONLY("Count: {}", ret);
ULONG relCount = 10;
// LOG_TRACE("Waiting mutex, owner: {}", _refMutex.getOwner());
OwnedLockGuard lock(_refMutex, 2);
auto ret = --m_iRefcount;
// LOG_TRACE("Count: {}, caller: {}", ret, Util::WhoIsTheCaller(_ReturnAddress()));
if (ret == 0)
{
if (ReleaseTrig != nullptr)
ReleaseTrig(Handle);
#ifdef USE_LOCAL_MUTEX
OwnedLockGuard lock(_localMutex, 999);
#endif
if (ClearTrig != nullptr)
ClearTrig(true, Handle);
if (m_pReal4 != nullptr && relCount > 0)
relCount = m_pReal4->Release();
if (ReleaseTrig != nullptr)
ReleaseTrig(Handle);
if (m_pReal3 != nullptr && relCount > 0)
relCount = m_pReal3->Release();
if (m_pReal2 != nullptr && relCount > 0)
relCount = m_pReal2->Release();
if (m_pReal1 != nullptr && relCount > 0)
relCount = m_pReal1->Release();
if (m_pReal != nullptr && relCount > 0)
relCount = m_pReal->Release();
LOG_INFO("{} released", id);
auto refCount = m_pReal->Release();
delete this;
}
OptiScaler/hooks/wrapped_swapchain.h
+2
View File
@@ -97,6 +97,8 @@ struct DECLSPEC_UUID("3af622a3-82d0-49cd-994f-cce05122c222") WrappedIDXGISwapCha
PFN_SC_Release ReleaseTrig = nullptr;
HWND Handle = nullptr;
OwnedMutex _refMutex;
#ifdef USE_LOCAL_MUTEX
OwnedMutex _localMutex;
#endif