sync price if stock % 2 == 1

Author: Trolldemorted

Cargo.toml

@@ -15,6 +15,7 @@ members = [
     "mod-shipyard-details",
     "mod-tavern-show-all-sailors",
     "mod-town-hall-details",
+    "mod-trading-office-prices-synchronization",
     #"p3-agent",
     "p3-agent-loader",
     "p3-aim",
@@ -30,7 +31,7 @@ exclude = [
 version = "0.1.9"
 
 [workspace.dependencies]
-hooklet = "^0.1.6"
+hooklet = "^0.1.7"
 clap = { version = "4.3.0", features = ["derive"] }
 simple_logger = "4.1"
 log = "0.4"

mod-high-res/Cargo.toml

@@ -12,10 +12,3 @@ log = { workspace = true }
 win_dbg_logger = { workspace = true }
 p3-api = { path = "../p3-api" }
 hooklet = { workspace = true }
-
-[dependencies.windows]
-version = "0.48"
-features = [
-    "Win32_Foundation",
-    "Win32_System_Memory",
-]

mod-high-res/src/ffi.rs

@@ -1,18 +1,17 @@
 use std::{
     arch::global_asm,
     ffi::{c_void, CStr},
-    mem, ptr,
+    mem,
     sync::atomic::{AtomicPtr, Ordering},
 };
 
-use hooklet::windows::x86::{deploy_rel32_raw, hook_call_rel32, CallRel32Hook, X86Rel32Type};
+use hooklet::windows::x86::{deploy_rel32_raw, hook_call_rel32, hook_call_rel32_with_module, CallRel32Hook, X86Rel32Type};
 use log::debug;
 use p3_api::{
     data::{class27::Class27Ptr, ddraw_fill_solid_rect, ddraw_set_constant_color, ddraw_set_render_dest, get_resolution_height, get_resolution_width},
     latin1_ptr_to_string,
     ui::class73::Class73Ptr,
 };
-use windows::core::PCSTR;
 
 const FULLHD_STRING: &CStr = c"1920 x 1080";
 
@@ -103,7 +102,7 @@ pub unsafe extern "C" fn start() -> u32 {
 
     // Fix empty bottom right corner
     debug!("Deploying render_all_objects_hook");
-    match hook_call_rel32(PCSTR::from_raw(ptr::null()), 0x28649, maybe_render_all_objects_hook as usize as u32) {
+    match hook_call_rel32(0x28649, maybe_render_all_objects_hook as usize as u32) {
         Ok(hook) => {
             HOOK_PTR.store(Box::into_raw(Box::new(hook)), Ordering::SeqCst);
         }
@@ -112,11 +111,7 @@ pub unsafe extern "C" fn start() -> u32 {
 
     // Fix acceleration map
     debug!("Deploying dddraw_dll.dll decode_supported_files hook to replace the background image");
-    match hook_call_rel32(
-        PCSTR::from_raw(c"aim.dll".as_ptr() as _),
-        0x2984,
-        ddraw_dll_decode_supported_files_hook as usize as u32,
-    ) {
+    match hook_call_rel32_with_module("aim.dll", 0x2984, ddraw_dll_decode_supported_files_hook as usize as u32) {
         Ok(hook) => {
             debug!("Hook {hook:?} set");
             DECODE_SUPPORTED_FILES_HOOK_PTR.store(Box::into_raw(Box::new(hook)), Ordering::SeqCst);
@@ -126,7 +121,7 @@ pub unsafe extern "C" fn start() -> u32 {
 
     // Fix acceleration map
     debug!("Deploying load screen settings from accelMap.ini");
-    match hook_call_rel32(PCSTR::from_raw(ptr::null()), 0x12C5, class73_place_ui_element_hook as usize as u32) {
+    match hook_call_rel32(0x12C5, class73_place_ui_element_hook as usize as u32) {
         Ok(hook) => {
             debug!("Hook {hook:?} set");
             LOAD_ACCEL_MAP_INI_HOOK_PTR.store(Box::into_raw(Box::new(hook)), Ordering::SeqCst);

mod-town-hall-details/src/ffi.rs

@@ -5,8 +5,8 @@ use std::{
     sync::atomic::{AtomicPtr, Ordering},
 };
 
-use hooklet::windows::x86::hook_function_pointer;
-use hooklet::windows::x86::{deploy_rel32_raw, hook_call_rel32, CallRel32Hook, FunctionPointerHook, X86Rel32Type};
+use hooklet::windows::x86::{deploy_rel32_raw, CallRel32Hook, FunctionPointerHook, X86Rel32Type};
+use hooklet::windows::x86::{hook_call_rel32, hook_function_pointer_width_module};
 use log::debug;
 use p3_api::ui::ui_town_hall_window::UITownHallWindowPtr;
 use windows::core::PCSTR;
@@ -26,7 +26,7 @@ pub unsafe extern "C" fn start() -> u32 {
     log::set_max_level(log::LevelFilter::Trace);
 
     debug!("Hooking town hall window's open function through vtable");
-    match hook_function_pointer(
+    match hook_function_pointer_width_module(
         PCSTR::from_raw(ptr::null()),
         TOWN_HALL_WINDOW_OPEN_POINTER_OFFSET,
         town_hall_window_open_hook as usize as u32,
@@ -42,7 +42,6 @@ pub unsafe extern "C" fn start() -> u32 {
 
     debug!("Hooking sidepanel's call to ui_town_hall_window_set_selected_page");
     match hook_call_rel32(
-        PCSTR::from_raw(ptr::null()),
         TOWN_HALL_SIDEPANEL_SET_SELECTED_PAGE_PATCH_OFFSET,
         town_hall_window_set_selected_page_hook as usize as u32,
     ) {

mod-trading-office-prices-synchronization/Cargo.toml

@@ -0,0 +1,16 @@
+[package]
+name = "mod-trading-office-prices-synchronization"
+edition = "2021"
+version.workspace = true
+
+[lib]
+crate-type = ["cdylib"]
+name="trading_office_prices_synchronization"
+
+[dependencies]
+log = { workspace = true }
+win_dbg_logger = { workspace = true }
+p3-api = { path = "../p3-api" }
+hooklet = { workspace = true }
+num-traits = { workspace = true }
+num-derive = { workspace = true }

mod-trading-office-prices-synchronization/src/ffi.rs

@@ -0,0 +1,39 @@
+use std::{
+    mem::{self},
+    panic::{self},
+    sync::atomic::{AtomicPtr, Ordering},
+};
+
+use hooklet::windows::x86::{hook_function_pointer, FunctionPointerHook};
+use log::error;
+use p3_api::ui::ui_trading_office_window::UITradingOfficeWindowPtr;
+
+const TOWN_TRADING_OFFICE_CLOSE_POINTER_OFFSET: u32 = UITradingOfficeWindowPtr::VTABLE_OFFSET + 0x118;
+static WINDOW_CLOSE_HOOK_PTR: AtomicPtr<FunctionPointerHook> = AtomicPtr::new(std::ptr::null_mut());
+
+#[no_mangle]
+unsafe extern "C" fn start() -> u32 {
+    let _ = log::set_logger(&win_dbg_logger::DEBUGGER_LOGGER);
+    log::set_max_level(log::LevelFilter::Trace);
+
+    panic::set_hook(Box::new(|p| {
+        error!("{p}");
+    }));
+
+    match hook_function_pointer(TOWN_TRADING_OFFICE_CLOSE_POINTER_OFFSET, window_close_hook as usize as u32) {
+        Ok(hook) => {
+            WINDOW_CLOSE_HOOK_PTR.store(Box::into_raw(Box::new(hook)), Ordering::SeqCst);
+        }
+        Err(_) => return 1,
+    }
+
+    0
+}
+
+#[no_mangle]
+unsafe extern "thiscall" fn window_close_hook(window_address: u32) {
+    crate::synchronize_autotrade_settings();
+    let orig_address = (*WINDOW_CLOSE_HOOK_PTR.load(Ordering::SeqCst)).old_absolute;
+    let orig: extern "thiscall" fn(window_address: u32) = unsafe { mem::transmute(orig_address) };
+    orig(window_address)
+}

mod-trading-office-prices-synchronization/src/lib.rs

@@ -0,0 +1,56 @@
+use log::{debug, warn};
+use num_traits::cast::FromPrimitive;
+use p3_api::{
+    data::enums::WareId, game_world::GAME_WORLD_PTR, operation::Operation, operations::OPERATIONS_PTR, town::get_town_name,
+    ui::ui_trading_office_window::UITradingOfficeWindowPtr,
+};
+
+mod ffi;
+
+pub(crate) unsafe fn synchronize_autotrade_settings() {
+    // Pull merchant index from operatons.field_924?
+    let merchant_index = OPERATIONS_PTR.get_player_merchant_index();
+    let window = UITradingOfficeWindowPtr::default();
+    let town_index = window.get_town_index();
+    let town_name = get_town_name(town_index as _).unwrap();
+    debug!("synchronize_autotrade_settings from {town_name}");
+    let setting_office = GAME_WORLD_PTR.get_office_in_of(town_index as _, merchant_index as _);
+    let setting_office = match setting_office {
+        Some(office) => office,
+        None => {
+            warn!("Could not find office in {town_name} of merchant {merchant_index:#x}");
+            return;
+        }
+    };
+
+    let setting_prices = setting_office.get_administrator_trade_prices();
+    let setting_stock = setting_office.get_administrator_trade_stock();
+    let merchant = GAME_WORLD_PTR.get_merchant(merchant_index as _);
+    let mut office_index = merchant.get_first_office_index();
+    while office_index < GAME_WORLD_PTR.get_offices_count() {
+        let office = GAME_WORLD_PTR.get_office(office_index);
+        if office.address == setting_office.address {
+            office_index = office.get_next_office_of_merchant_index();
+            continue;
+        }
+        let stock = office.get_administrator_trade_stock();
+
+        for i in 0..20 {
+            let ware_id = WareId::from_usize(i).unwrap();
+            if setting_stock[i] / ware_id.get_scaling() % 2 == 0 {
+                continue;
+            }
+            let stock = stock[i];
+            let price = setting_prices[i];
+            
+            OPERATIONS_PTR.enqueue_operation(Operation::OfficeAutotradeSettingChange {
+                stock,
+                price,
+                office_index: office_index as _,
+                ware_id,
+            });
+        }
+
+        office_index = office.get_next_office_of_merchant_index();
+    }
+}

p3-api/src/data/office.rs

@@ -28,13 +28,25 @@ impl OfficePtr {
         unsafe { self.get(0x2ca) }
     }
 
+    pub unsafe fn get_administrator_trade_prices(&self) -> [i32; 20] {
+        self.get(0x2f4)
+    }
+
     pub unsafe fn set_administrator_trade_prices(&self, prices: [i32; 20]) {
         self.set(0x2f4, &prices)
     }
 
     pub unsafe fn set_administrator_trade_actions(&self, actions: [i32; 20]) {
         self.set(0x2f4, &actions)
     }
+
+    pub unsafe fn get_administrator_trade_stock(&self) -> [i32; 20] {
+        self.get(0x354)
+    }
+
+    pub unsafe fn get_administrator_trade_lock_bitmap(&self) -> u32 {
+        self.get(0x3b4)
+    }
 }
 
 impl P3Pointer for OfficePtr {

p3-api/src/operation.rs

@@ -40,6 +40,12 @@ pub enum Operation {
     RepairConvoy {
         convoy_index: u32,
     },
+    OfficeAutotradeSettingChange {
+        stock: i32,
+        price: i32,
+        office_index: u32,
+        ware_id: WareId,
+    },
 }
 
 impl Operation {
@@ -126,6 +132,20 @@ impl Operation {
                 op[0..4].copy_from_slice(&opcode.to_le_bytes());
                 op[0x04..0x08].copy_from_slice(&convoy_id.to_le_bytes());
             }
+            Operation::OfficeAutotradeSettingChange {
+                stock,
+                price,
+                office_index,
+                ware_id,
+            } => {
+                let opcode: u32 = 0x5b;
+                let ware_id: u32 = *ware_id as _;
+                op[0..4].copy_from_slice(&opcode.to_le_bytes());
+                op[4..8].copy_from_slice(&stock.to_le_bytes());
+                op[8..0x0c].copy_from_slice(&price.to_le_bytes());
+                op[0x0c..0x10].copy_from_slice(&office_index.to_le_bytes());
+                op[0x10..0x14].copy_from_slice(&ware_id.to_le_bytes());
+            }
         }
         op
     }

p3-api/src/operations.rs

@@ -1,12 +1,14 @@
-use std::mem::transmute;
+use std::{ffi::c_void, mem::transmute};
 
 use log::debug;
 
 use crate::{data::p3_ptr::P3Pointer, operation::Operation};
 
-const OPERATIONS_ADDRESS: u32 = 0x006DF2F0;
 pub const OPERATIONS_PTR: OperationsPtr = OperationsPtr::new();
-const EXECUTE_OPERATION_ADDRESS: u32 = 0x00535760;
+const OPERATIONS_ADDRESS: u32 = 0x006DF2F0;
+static EXECUTE_OPERATION_ADDRESS: u32 = 0x00535760;
+static ENQUEUE_OPERATION_ADDRESS: u32 = 0x0054AA70;
+static ENQUEUE_OPERATION: &extern "thiscall" fn(*mut c_void, *const c_void) = unsafe { transmute(&ENQUEUE_OPERATION_ADDRESS) };
 
 #[derive(Clone, Debug, Copy)]
 pub struct OperationsPtr {
@@ -27,6 +29,12 @@ impl OperationsPtr {
     pub unsafe fn get_player_merchant_index(&self) -> i32 {
         self.get(0x0924)
     }
+
+    pub unsafe fn enqueue_operation(&self, op: Operation) {
+        debug!("Enqueuing operation {op:?}");
+        let op_bytes = op.to_raw();
+        ENQUEUE_OPERATION(OPERATIONS_ADDRESS as _, op_bytes.as_ptr() as _)
+    }
 }
 
 impl P3Pointer for OperationsPtr {