Initial community commit

1 files changed, 549 insertions, 0 deletions

diff --git a/Src/omBrowser/component.cpp b/Src/omBrowser/component.cpp
new file mode 100644
index 00000000..4ec89e02
--- /dev/null
+++ b/Src/omBrowser/component.cpp
@@ -0,0 +1,549 @@
+#include "./main.h"
+#include "./component.h"
+#include "./browserFactory.h"
+#include "./serviceFactory.h"
+#include "./utilityFactory.h"
+#include "./wasabiHelper.h"
+#include "./winampHook.h"
+#include "./skinHelper.h"
+#include "./browserClass.h"
+#include "./internetFeatures.h"
+#include "./ieVersion.h"
+
+#include <wininet.h>
+#include <strsafe.h>
+
+static OmBrowserFactory browserFactory;
+static OmServiceFactory serviceFactory;
+static OmUtilityFactory utilityFactory;
+
+OmBrowserComponent::OmBrowserComponent()
+	: wasabiHelper(NULL), winampHook(NULL), skinHelper(NULL), hookCookie(0), internetFeatures(NULL)
+{
+	InitializeCriticalSection(&lock);
+}
+
+OmBrowserComponent::~OmBrowserComponent()
+{
+	ReleaseServices();
+
+	if (NULL != internetFeatures)
+	{
+		delete(internetFeatures);
+		internetFeatures = NULL;
+
+		InternetSetOption(NULL, INTERNET_OPTION_END_BROWSER_SESSION, NULL, 0);
+	}
+
+	DeleteCriticalSection(&lock);
+}
+
+size_t OmBrowserComponent::AddRef()
+{
+	return 1;
+}
+
+size_t OmBrowserComponent::Release()
+{
+	return 1;
+}
+
+int OmBrowserComponent::QueryInterface(GUID interface_guid, void **object)
+{
+	if (NULL == object) return E_POINTER;
+
+	if (IsEqualIID(interface_guid, IFC_WinampHook))
+		*object = static_cast<ifc_winamphook*>(this);
+	else
+	{
+		*object = NULL;
+		return E_NOINTERFACE;
+	}
+
+	if (NULL == *object)
+		return E_UNEXPECTED;
+
+	AddRef();
+	return S_OK;
+}
+
+void OmBrowserComponent::RegisterServices(api_service *service)
+{
+	EnterCriticalSection(&lock);
+
+	if (NULL == wasabiHelper)
+		WasabiHelper::CreateInstance(service, &wasabiHelper);
+
+	aTRACE_LINE("omBrowser Registered");
+	browserFactory.Register(service);
+	serviceFactory.Register(service);
+	utilityFactory.Register(service);
+
+	LeaveCriticalSection(&lock);
+}
+
+void OmBrowserComponent::ReleaseServices()
+{
+	EnterCriticalSection(&lock);
+
+	if (NULL != wasabiHelper)
+	{
+		wasabiHelper->Release();
+		wasabiHelper = NULL;
+	}
+
+	if (NULL != winampHook)
+	{
+		winampHook->Release();
+		winampHook = NULL;
+	}
+
+	if (0 != hookCookie) 
+	{
+		UnregisterWinampHook(hookCookie);
+		hookCookie = 0;
+	}
+
+	if (NULL != skinHelper)
+	{
+		skinHelper->Release();
+		skinHelper = NULL;
+	}
+
+	LeaveCriticalSection(&lock);
+}
+
+int OmBrowserComponent::RegisterServicesSafeModeOk()
+{
+	return 1;
+}
+
+void OmBrowserComponent::DeregisterServices(api_service *service)
+{
+	browserFactory.Unregister(service);
+	serviceFactory.Unregister(service);
+	utilityFactory.Unregister(service);
+
+	ReleaseServices();
+
+	size_t index = unloadCallbacks.size();
+	while(index--)
+	{
+		PLUGINUNLOADCALLBACK callback = unloadCallbacks[index];
+		if (NULL != callback) callback();
+	}
+	unloadCallbacks.clear();
+}
+
+HRESULT OmBrowserComponent::InitializeComponent(HWND hwndWinamp)
+{
+	HRESULT hr(S_FALSE);
+
+	EnterCriticalSection(&lock);
+
+	if(NULL == winampHook)
+	{
+		HRESULT hookResult = WinampHook::CreateInstance(hwndWinamp, &winampHook);
+		if (FAILED(hookResult) && SUCCEEDED(hr)) hr = hookResult;
+	}
+
+	if (SUCCEEDED(hr) && 0 == hookCookie)
+	{
+		hr = winampHook->RegisterCallback(this, &hookCookie);
+	}
+
+	if (NULL == skinHelper)
+	{
+		HRESULT skinResult = SkinHelper::CreateInstance(hwndWinamp, &skinHelper);
+		if (FAILED(skinResult) && SUCCEEDED(hr)) hr = skinResult;
+	}
+
+	LeaveCriticalSection(&lock);
+
+	return hr;
+}
+
+HRESULT OmBrowserComponent::GetWasabiHelper( ifc_wasabihelper **wasabiOut )
+{
+	if ( NULL == wasabiOut ) return E_POINTER;
+
+	EnterCriticalSection( &lock );
+
+	*wasabiOut = wasabiHelper;
+	if ( NULL != wasabiHelper )
+		wasabiHelper->AddRef();
+
+	LeaveCriticalSection( &lock );
+
+	return ( NULL != *wasabiOut ) ? S_OK : E_NOINTERFACE;
+}
+
+HRESULT OmBrowserComponent::GetSkinHelper(ifc_skinhelper **skinOut)
+{
+	if (NULL == skinOut) return E_POINTER;
+
+	EnterCriticalSection(&lock);
+
+	*skinOut = skinHelper;
+	if (NULL != skinHelper)
+		skinHelper->AddRef();
+
+	LeaveCriticalSection(&lock);
+
+	return (NULL != *skinOut) ? S_OK : E_NOINTERFACE;
+}
+
+HRESULT OmBrowserComponent::RegisterWinampHook(ifc_winamphook *hook, UINT *cookieOut)
+{
+	if (NULL == cookieOut) return E_POINTER;
+	*cookieOut = NULL;
+
+	EnterCriticalSection(&lock);
+	HRESULT hr = (NULL != winampHook) ? winampHook->RegisterCallback(hook, cookieOut) : E_FAIL;
+	LeaveCriticalSection(&lock);
+
+    return hr;
+}
+
+HRESULT OmBrowserComponent::UnregisterWinampHook(UINT cookie)
+{
+	EnterCriticalSection(&lock);
+	HRESULT hr = (NULL != winampHook) ? winampHook->UnregisterCallback(cookie) : E_FAIL;
+	LeaveCriticalSection(&lock);
+
+	return hr;
+}
+
+HRESULT OmBrowserComponent::GetWinampWnd(HWND *hwndWinamp)
+{
+	if (NULL == hwndWinamp) return E_POINTER;
+
+	EnterCriticalSection(&lock);
+	*hwndWinamp = (NULL != winampHook) ? winampHook->GetWinamp() : NULL;
+	LeaveCriticalSection(&lock);
+
+	HRESULT hr;
+	if (NULL == *hwndWinamp) 
+		hr = E_FAIL;
+	else
+		hr = S_OK;
+
+	return hr;
+}
+
+HRESULT OmBrowserComponent::ResetFont(void)
+{
+	if (NULL != skinHelper)
+		skinHelper->ResetFontCache();
+	return S_OK;
+}
+
+HRESULT OmBrowserComponent::SkinChanged(const wchar_t *skinName)
+{
+	UpdateColors();
+	return S_OK;
+}
+
+HRESULT OmBrowserComponent::SkinColorChange(const wchar_t *colorTheme)
+{
+	UpdateColors();
+	return S_OK;
+}
+
+HRESULT OmBrowserComponent::RegisterUnloadCallback(PLUGINUNLOADCALLBACK callback)
+{
+	if(NULL == callback) return E_INVALIDARG;
+	unloadCallbacks.push_back(callback);
+	return S_OK;
+}
+
+HRESULT OmBrowserComponent::GetBrowserClass(LPCWSTR pszName, ifc_ombrowserclass **instance)
+{
+	if (NULL == instance) return E_POINTER;
+
+	HRESULT hr(S_OK);
+	ifc_ombrowserclass *browserClass = NULL;
+
+	EnterCriticalSection(&lock);
+
+	size_t index = browserClasses.size();
+	while(index--)
+	{
+		browserClass = browserClasses[index];
+		if (S_OK == browserClass->IsEqual(pszName))
+		{
+			browserClass->AddRef();
+			break;
+		}
+	}
+
+	if (((size_t)-1) == index)
+	{
+		hr = OmBrowserClass::CreateInstance(pszName, (OmBrowserClass**)&browserClass);
+		if (SUCCEEDED(hr) && browserClass != NULL)
+		{
+			if (0 == browserClasses.size())
+			{
+				SetUserAgent();
+				SetInternetFeautures();
+			}
+
+			browserClasses.push_back(browserClass);
+		}
+	}
+
+	*instance = (SUCCEEDED(hr) && browserClass != NULL) ? browserClass : NULL;	
+
+	LeaveCriticalSection(&lock);
+
+	return hr;
+}
+
+HRESULT OmBrowserComponent::UnregisterBrowserClass(LPCWSTR pszName)
+{
+	HRESULT hr(S_FALSE);
+
+	EnterCriticalSection(&lock);
+
+	size_t index = browserClasses.size();
+	while(index--)
+	{
+		ifc_ombrowserclass *browserClass = browserClasses[index];
+		if (S_OK == browserClass->IsEqual(pszName))
+		{
+			browserClasses.erase(browserClasses.begin() + index);
+			hr = S_OK;
+			break;
+		}
+	}
+
+	LeaveCriticalSection(&lock);
+	return hr;
+}
+
+void OmBrowserComponent::UpdateColors()
+{
+	if (NULL != skinHelper)
+		skinHelper->ResetColorCache();
+
+	EnterCriticalSection(&lock);
+
+	size_t index = browserClasses.size();
+	while(index--)
+	{
+		browserClasses[index]->UpdateRegColors();
+	}
+
+	LeaveCriticalSection(&lock);
+
+	InternetSetOption(NULL, INTERNET_OPTION_SETTINGS_CHANGED, NULL, 0 );
+}
+
+typedef struct __INTERNETFEATUREREC
+{
+	INTERNETFEATURELIST entry;
+	DWORD				flags;
+	BOOL				enabled;
+} INTERNETFEATUREREC;
+
+void OmBrowserComponent::SetInternetFeautures()
+{
+	const INTERNETFEATUREREC szFeatures[] = 
+	{  
+		{ FEATURE_DISABLE_NAVIGATION_SOUNDS, SET_FEATURE_ON_PROCESS, TRUE },
+		{ FEATURE_TABBED_BROWSING, SET_FEATURE_ON_PROCESS, TRUE },
+		{ FEATURE_WINDOW_RESTRICTIONS, SET_FEATURE_ON_PROCESS, FALSE },
+		{ FEATURE_SSLUX, SET_FEATURE_ON_PROCESS, TRUE },
+		{ FEATURE_FORCE_ADDR_AND_STATUS, SET_FEATURE_ON_PROCESS, FALSE },
+		{ FEATURE_BLOCK_INPUT_PROMPTS, SET_FEATURE_ON_PROCESS, FALSE },
+		{ FEATURE_MIME_HANDLING, SET_FEATURE_ON_PROCESS, TRUE },
+		{ FEATURE_LOCALMACHINE_LOCKDOWN, SET_FEATURE_ON_PROCESS, TRUE },
+	};
+
+	EnterCriticalSection(&lock);
+
+	if (NULL == internetFeatures)
+	{
+		internetFeatures = new InternetFeatures();
+		if (NULL == internetFeatures) 
+		{
+			LeaveCriticalSection(&lock);
+			return;
+		}
+	}
+
+	UINT modified = 0;
+	HRESULT hr;
+
+	for (INT i = 0; i < ARRAYSIZE(szFeatures); i++)
+	{
+		const INTERNETFEATUREREC *rec = &szFeatures[i];
+		hr = internetFeatures->IsEnabled(rec->entry, rec->flags);
+		if ((S_OK == hr && FALSE == rec->enabled) || (S_FALSE == hr && FALSE != rec->enabled))
+		{
+			if (SUCCEEDED(internetFeatures->SetEnabled(rec->entry, rec->flags, rec->enabled)))
+				modified++;
+		}
+	}
+
+	int majorVersion = 0;
+	if (SUCCEEDED(MSIE_GetVersion(&majorVersion, NULL, NULL, NULL)))
+	{
+		unsigned long browserEmulation = 0;
+		if (8 == majorVersion)
+			browserEmulation = 8000;
+		else if (9 == majorVersion)
+			browserEmulation = 9000;
+		else if (10 <= majorVersion)
+			browserEmulation = 10000;
+		else
+			browserEmulation = 0;
+
+		unsigned long valueAlreadySet = 0;
+		hr = internetFeatures->GetDWORDFeature(L"FEATURE_BROWSER_EMULATION", TRUE, &valueAlreadySet);
+		if (FAILED(hr) || valueAlreadySet != browserEmulation)
+		{
+			if (0 == browserEmulation)
+			{
+				if (0x80070002 /*ERROR_FILE_NOT_FOUND */ != hr)
+					internetFeatures->DeleteFeature(L"FEATURE_BROWSER_EMULATION", TRUE);
+			}
+			else
+			{
+				if (SUCCEEDED(internetFeatures->SetDWORDFeature(L"FEATURE_BROWSER_EMULATION", TRUE, browserEmulation)))
+					modified++;
+			}
+		}
+	}
+
+	LeaveCriticalSection(&lock);
+
+	if (0 != modified)
+		InternetSetOption(NULL, INTERNET_OPTION_SETTINGS_CHANGED, NULL, 0 );
+}
+
+static HRESULT
+Component_PrintWinampUA(char *buffer, size_t bufferMax)
+{
+	char *cursor = buffer;
+	size_t remaining = bufferMax;
+
+	HRESULT hr = StringCchPrintfExA(cursor, remaining, 
+						&cursor, &remaining, 
+						STRSAFE_NULL_ON_FAILURE, 
+						"%S/%d.%d", 
+						OMBROWSER_NAME, 
+						OMBROWSER_VERSION_MAJOR, 
+						OMBROWSER_VERSION_MINOR);
+
+	if (SUCCEEDED(hr))
+	{
+		ifc_wasabihelper *wasabi = NULL;
+		if (SUCCEEDED(Plugin_GetWasabiHelper(&wasabi)) && wasabi != NULL)
+		{
+			api_application *app = NULL;
+			if (SUCCEEDED(wasabi->GetApplicationApi(&app)) && app != NULL)
+			{
+				char *rollback = cursor;
+				hr = StringCchPrintfExA(cursor, remaining, 
+						&cursor, &remaining, 
+						STRSAFE_NULL_ON_FAILURE, 
+						" (%S)", 
+						app->main_getVersionString());
+				
+				if (FAILED(hr))
+					*rollback = '\0';
+
+				app->Release();
+			}
+			wasabi->Release();
+		}
+	}
+
+	if (FAILED(hr))
+		buffer[0] = '\0';
+
+	return hr;
+}
+
+BOOL OmBrowserComponent::SetUserAgent(void)
+{
+	unsigned long bufferSize = 0;
+
+	HRESULT hr = UrlMkGetSessionOption(URLMON_OPTION_USERAGENT, 
+								NULL, 
+								0,
+								&bufferSize,
+								0);
+	if(E_OUTOFMEMORY == hr)
+		hr = S_OK;
+
+	if (FAILED(hr))
+		return FALSE;
+
+	bufferSize += 512;
+
+	char *buffer = (char *)calloc(bufferSize, sizeof(char));
+	if (NULL == buffer)
+		return FALSE;
+
+	unsigned long bufferLength = 0;
+	hr = UrlMkGetSessionOption(URLMON_OPTION_USERAGENT, 
+								buffer, 
+								bufferSize,
+								&bufferLength,
+								0);
+	if (E_OUTOFMEMORY == hr
+		&& bufferLength <= bufferSize)
+	{
+		hr = S_OK;
+	}
+
+	if (SUCCEEDED(hr))
+	{
+		if (bufferLength > 0 
+			&& bufferLength < bufferSize)
+		{
+			buffer[bufferLength - 1] = ' ';
+		}
+		hr = Component_PrintWinampUA(buffer + bufferLength, 
+									 bufferSize - bufferLength);
+		if (FAILED(hr))
+		{
+			if (bufferLength > 0)
+				buffer[bufferLength - 1] = '\0';
+		}
+		else
+		{
+			bufferLength = lstrlenA(buffer);
+
+			hr = UrlMkSetSessionOption(URLMON_OPTION_USERAGENT, 
+										buffer, 
+										bufferLength,
+										0);
+		}
+	}
+
+	free(buffer);
+	return SUCCEEDED(hr);
+}
+
+#define CBCLASS OmBrowserComponent
+START_MULTIPATCH;
+ START_PATCH(MPIID_WA5COMPONENT)
+  M_CB(MPIID_WA5COMPONENT, ifc_wa5component, ADDREF, AddRef);
+  M_CB(MPIID_WA5COMPONENT, ifc_wa5component, RELEASE, Release);
+  M_CB(MPIID_WA5COMPONENT, ifc_wa5component, QUERYINTERFACE, QueryInterface);
+  M_VCB(MPIID_WA5COMPONENT, ifc_wa5component, API_WA5COMPONENT_REGISTERSERVICES, RegisterServices);
+  M_CB(MPIID_WA5COMPONENT, ifc_wa5component, 15, RegisterServicesSafeModeOk)
+  M_VCB(MPIID_WA5COMPONENT, ifc_wa5component, API_WA5COMPONENT_DEREEGISTERSERVICES, DeregisterServices);
+ NEXT_PATCH(MPIID_WINAMPHOOK)
+  M_CB(MPIID_WINAMPHOOK, ifc_winamphook, ADDREF, AddRef);
+  M_CB(MPIID_WINAMPHOOK, ifc_winamphook, RELEASE, Release);
+  M_CB(MPIID_WINAMPHOOK, ifc_winamphook, QUERYINTERFACE, QueryInterface);
+  M_CB(MPIID_WINAMPHOOK, ifc_winamphook, API_RESETFONT, ResetFont);
+  M_CB(MPIID_WINAMPHOOK, ifc_winamphook, API_SKINCHANGED, SkinChanged);
+  M_CB(MPIID_WINAMPHOOK, ifc_winamphook, API_SKINCOLORCHANGE, SkinColorChange);
+ END_PATCH
+END_MULTIPATCH;
+#undef CBCLASS
\ No newline at end of file