Cool_Tools/Mikado
1
0
Fork 0
mirror of https://github.com/adamaq01/mikado.git synced 2024-11-23 22:20:56 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: use card whitelist for cloudlink scores injection

Browse Source
This commit is contained in:
Adamaq01 2023-09-07 01:20:08 +02:00
parent f750adc485
commit 8418d47080
2 changed files with 31 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/helpers.rs
View File

@ -78,6 +78,20 @@ pub fn get_current_card_id() -> Option<String> {
guard.clone()
}
pub fn is_current_card_id_whitelisted() -> bool {
if let Some(card) = get_current_card_id() {
if !CONFIGURATION.cards.whitelist.is_empty()
&& !CONFIGURATION.cards.whitelist.contains(&card)
{
return false;
}
return true;
}
false
}
pub fn read_node_str(node: *const (), path: *const u8, length: usize) -> Option<String> {
let mut buffer = [0u8; 32];
let result = unsafe {

28
src/mikado.rs
View File

@ -8,6 +8,7 @@ use log::{debug, error, info, warn};
use crate::handlers::save::process_save;
use crate::handlers::scores::process_scores;
use crate::helpers::is_current_card_id_whitelisted;
use crate::sys::{
property_clear_error, property_mem_write, property_node_name, property_node_refer,
property_query_size, property_search, property_set_flag, NodeType,
@ -200,10 +201,11 @@ pub unsafe fn property_mem_read_hook_wrapped(
Ok(response)
})())
} else if load
.then(|| root.pointer(&["game", "code"]))
.flatten()
.is_some()
} else if is_current_card_id_whitelisted()
&& load
.then(|| root.pointer(&["game", "code"]))
.flatten()
.is_some()
{
Some((|| {
let game = root
@ -222,14 +224,18 @@ pub unsafe fn property_mem_read_hook_wrapped(
Ok(response)
})())
} else if let Some(music) = load_m.then(|| root.pointer(&["game", "music"])).flatten() {
Some((|| {
let user = USER.load(Ordering::SeqCst).to_string();
let response = crate::cloudlink::process_pbs(user.as_str(), music)?;
let response = build_response(&original_signature, response, encoding)?;
LOAD_M.store(false, Ordering::Relaxed);
if is_current_card_id_whitelisted() {
Some((|| {
let user = USER.load(Ordering::SeqCst).to_string();
let response = crate::cloudlink::process_pbs(user.as_str(), music)?;
let response = build_response(&original_signature, response, encoding)?;
LOAD_M.store(false, Ordering::Relaxed);
Ok(response)
})())
Ok(response)
})())
} else {
None
}
} else {
None
}