Flash in browser workaround

2 files changed, 6 insertions, 2 deletions

diff --git a/overlay/d3d9.cpp b/overlay/d3d9.cpp
index 3e9768206..85249b582 100644
--- a/overlay/d3d9.cpp
+++ b/overlay/d3d9.cpp
@@ -42,7 +42,7 @@ struct D3DTLVERTEX {
 };
 static const DWORD D3DFVF_TLVERTEX = D3DFVF_XYZRHW | D3DFVF_TEX1;
 
-class DevState : protected Pipe {
+class DevState : public Pipe {
 	public:
 		IDirect3DDevice9 *dev;
 		IDirect3DStateBlock9 *pSB;
@@ -429,6 +429,9 @@ static ULONG __stdcall myRelease(IDirect3DDevice9 *idd) {
 			ds->refCount--;
 		}
 
+		if (ds->refCount <= 1)
+			ds->disconnect();
+
 		if (ds->refCount >= 0)
 			return ds->refCount + ds->initRefCount;
 
diff --git a/overlay/lib.h b/overlay/lib.h
index f06b32957..43427f04a 100644
--- a/overlay/lib.h
+++ b/overlay/lib.h
@@ -96,7 +96,6 @@ class Pipe {
 		HANDLE hMemory;
 
 		void release();
-		void disconnect();
 	protected:
 		unsigned int uiWidth, uiHeight;
 		unsigned int uiLeft, uiTop, uiRight, uiBottom;
@@ -111,6 +110,8 @@ class Pipe {
 		virtual void newTexture(unsigned int w, unsigned int h) = 0;
 		Pipe();
 		~Pipe();
+	public:
+		void disconnect();
 };
 
 extern void checkDXGIHook(bool preonly = false);