Rim/iw5-mod
Rim/iw5-mod
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

cleanup the cleanup 😹

Browse Source
This commit is contained in:
Diavolo 2023-01-17 20:29:16 +01:00
parent 49789c15e8
commit 4459132c05
No known key found for this signature in database
GPG Key ID: FA77F074E98D98A5
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/launcher/html/html_frame.cpp
View File

@ -1,14 +1,14 @@
#include <std_include.hpp> #include <std_include.hpp>
#include "html_frame.hpp" #include "html_frame.hpp"
#include "utils/nt.hpp" #include <utils/nt.hpp>
#include "utils/hook.hpp" #include <utils/hook.hpp>
std::atomic<int> html_frame::frame_count_ = 0; std::atomic<int> html_frame::frame_count_ = 0;
namespace namespace
{ {
void* original_func{}; void* original_func{};
GUID browser_emulation_guid{0xac969931, 0x3566, 0x4b50, {0xae, 0x48, 0x71, 0xb9, 0x6a, 0x75, 0xc8, 0x79}}; GUID browser_emulation_guid{0xAC969931, 0x3566, 0x4B50, {0xAE, 0x48, 0x71, 0xB9, 0x6A, 0x75, 0xC8, 0x79}};
int WINAPI co_internet_feature_value_internal_stub(const GUID* guid, uint32_t* result) int WINAPI co_internet_feature_value_internal_stub(const GUID* guid, uint32_t* result)
{ {
@ -28,7 +28,7 @@ namespace
static const auto _ = [] static const auto _ = []
{ {
const auto urlmon = utils::nt::library::load("urlmon.dll"s); const auto urlmon = utils::nt::library::load("urlmon.dll"s);
const auto target = urlmon.get_iat_entry("iertutil.dll", MAKEINTRESOURCEA(700)); auto** target = urlmon.get_iat_entry("iertutil.dll"s, MAKEINTRESOURCEA(700));
original_func = *target; original_func = *target;
utils::hook::set(target, co_internet_feature_value_internal_stub); utils::hook::set(target, co_internet_feature_value_internal_stub);
@ -118,9 +118,10 @@ std::shared_ptr<IWebBrowser2> html_frame::get_web_browser() const
if (!this->browser_object_) return {}; if (!this->browser_object_) return {};
IWebBrowser2* web_browser = nullptr; IWebBrowser2* web_browser = nullptr;
if (FAILED(this->browser_object_->QueryInterface(IID_IWebBrowser2, reinterpret_cast<void**>(&web_browser))) || ! if (FAILED(this->browser_object_->QueryInterface(IID_IWebBrowser2, reinterpret_cast<void**>(&web_browser))) || !web_browser)
web_browser) {
return {}; return {};
}
return std::shared_ptr<IWebBrowser2>(web_browser, object_deleter); return std::shared_ptr<IWebBrowser2>(web_browser, object_deleter);
} }
@ -143,8 +144,9 @@ std::shared_ptr<IHTMLDocument2> html_frame::get_document() const
IHTMLDocument2* document = nullptr; IHTMLDocument2* document = nullptr;
if (FAILED(dispatch->QueryInterface(IID_IHTMLDocument2, reinterpret_cast<void**>(&document))) || !document) if (FAILED(dispatch->QueryInterface(IID_IHTMLDocument2, reinterpret_cast<void**>(&document))) || !document)
return {
{}; return {};
}
return std::shared_ptr<IHTMLDocument2>(document, object_deleter); return std::shared_ptr<IHTMLDocument2>(document, object_deleter);
} }
@ -161,9 +163,7 @@ void html_frame::initialize(const HWND window)
void html_frame::create_browser() void html_frame::create_browser()
{ {
LPCLASSFACTORY class_factory = nullptr; LPCLASSFACTORY class_factory = nullptr;
if (FAILED( if (FAILED(CoGetClassObject(CLSID_WebBrowser, CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER, nullptr, IID_IClassFactory, reinterpret_cast<void **>(&class_factory))) || !class_factory)
CoGetClassObject(CLSID_WebBrowser, CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER, nullptr, IID_IClassFactory,
reinterpret_cast<void **>(&class_factory))) || !class_factory)
{ {
throw std::runtime_error("Unable to get the class factory"); throw std::runtime_error("Unable to get the class factory");
} }