forked from ObolSquad/trigger-patch
82 lines
2.5 KiB
Rust
82 lines
2.5 KiB
Rust
use std::ffi::CString;
|
|
|
|
use ilhook::x64::Registers;
|
|
|
|
use crate::util::{self, import, read_csharp_string};
|
|
|
|
use super::{ModuleInitError, NapModule, NapModuleContext};
|
|
|
|
const MAKE_INITIAL_URL: usize = 0x7CE75F0;
|
|
const WEB_REQUEST_CREATE: usize = 0x798F080;
|
|
|
|
pub struct Network;
|
|
|
|
impl NapModule for NapModuleContext<Network> {
|
|
unsafe fn init(&mut self) -> Result<(), ModuleInitError> {
|
|
self.interceptor.attach(
|
|
self.base.wrapping_add(MAKE_INITIAL_URL),
|
|
Network::on_make_initial_url,
|
|
)?;
|
|
|
|
self.interceptor.attach(
|
|
self.base.wrapping_add(WEB_REQUEST_CREATE),
|
|
on_web_request_create,
|
|
)?;
|
|
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
unsafe extern "win64" fn on_web_request_create(reg: *mut Registers, _: usize) {
|
|
let s = read_csharp_string((*reg).rcx as usize);
|
|
if s.contains("StandaloneWindows64/cn/") {
|
|
let s = s.replace("StandaloneWindows64/cn/", "StandaloneWindows64/oversea/");
|
|
println!("replaced: {s}");
|
|
(*reg).rcx =
|
|
il2cpp_string_new(CString::new(s).unwrap().to_bytes_with_nul().as_ptr()) as u64;
|
|
}
|
|
}
|
|
|
|
import!(il2cpp_string_new(cstr: *const u8) -> usize = 0x3538F0);
|
|
|
|
impl Network {
|
|
const SDK_URL: &str = "http://127.0.0.1:20100";
|
|
const DISPATCH_URL: &str = "http://127.0.0.1:10100";
|
|
const REDIRECT_SDK: bool = true;
|
|
const REDIRECT_DISPATCH: bool = true;
|
|
|
|
unsafe extern "win64" fn on_make_initial_url(reg: *mut Registers, _: usize) {
|
|
let url = util::read_csharp_string((*reg).rcx as usize);
|
|
|
|
let mut new_url = match url.as_str() {
|
|
s if (s.contains("mihoyo.com") || s.contains("hoyoverse.com"))
|
|
&& Self::REDIRECT_SDK =>
|
|
{
|
|
Self::SDK_URL.to_string()
|
|
}
|
|
s if (s.contains("globaldp-prod-os01.zenlesszonezero.com")
|
|
|| s.contains("globaldp-prod-cn01.juequling.com"))
|
|
&& Self::REDIRECT_DISPATCH =>
|
|
{
|
|
Self::DISPATCH_URL.to_string()
|
|
}
|
|
s => {
|
|
println!("Leaving request as-is: {s}");
|
|
return;
|
|
}
|
|
};
|
|
|
|
url.split('/').skip(3).for_each(|s| {
|
|
new_url.push_str("/");
|
|
new_url.push_str(s);
|
|
});
|
|
|
|
println!("UnityWebRequest: \"{url}\", replacing with \"{new_url}\"");
|
|
(*reg).rcx = il2cpp_string_new(
|
|
CString::new(new_url.as_str())
|
|
.unwrap()
|
|
.to_bytes_with_nul()
|
|
.as_ptr(),
|
|
) as u64;
|
|
}
|
|
}
|