initial code commit

Author: dariowouters

.gitmodules

@@ -0,0 +1,9 @@
+[submodule "vendor/fmt"]
+	path = vendor/fmt
+	url = https://github.com/fmtlib/fmt.git
+[submodule "vendor/imgui"]
+	path = vendor/imgui
+	url = https://github.com/ocornut/imgui.git
+[submodule "vendor/minhook"]
+	path = vendor/minhook
+	url = https://github.com/TsudaKageyu/minhook.git

CHANGELOG.md

@@ -0,0 +1,2 @@
+1.0.0
+    - Initial Version

CMakeLists.txt

@@ -0,0 +1,40 @@
+cmake_minimum_required(VERSION 3.15)
+project(ts-extra-utilities VERSION 1.0.0)
+
+set_property(GLOBAL PROPERTY USE_FOLDERS ON)
+
+file(GLOB_RECURSE ${CMAKE_PROJECT_NAME}-src src/*.hpp src/*.cpp)
+
+set(PLUGIN_VERSION_MAJOR 1)
+set(PLUGIN_VERSION_MINOR 0)
+set(PLUGIN_VERSION_PATCH 0)
+
+configure_file(
+    ${CMAKE_CURRENT_SOURCE_DIR}/version.hpp.in
+    ${CMAKE_CURRENT_SOURCE_DIR}/src/version.hpp
+)
+
+configure_file(
+    ${CMAKE_CURRENT_SOURCE_DIR}/version.rc.in
+    ${CMAKE_CURRENT_BINARY_DIR}/version.rc
+)
+
+foreach(_source IN ITEMS ${${CMAKE_PROJECT_NAME}-src})
+    get_filename_component(_source_path "${_source}" PATH)
+    string(REPLACE "${CMAKE_SOURCE_DIR}" "" _group_path "${_source_path}")
+    string(REPLACE "/" "\\" _group_path "${_group_path}")
+    source_group("${_group_path}" FILES "${_source}")
+endforeach()
+
+add_library(${CMAKE_PROJECT_NAME} SHARED ${${CMAKE_PROJECT_NAME}-src} ${CMAKE_CURRENT_BINARY_DIR}/version.rc)
+
+target_compile_definitions(${CMAKE_PROJECT_NAME} PRIVATE TS_EXTRA_UTILITIES_DLL_EXPORTS UNICODE _UNICODE)
+
+target_compile_features(${CMAKE_PROJECT_NAME} PUBLIC cxx_std_17)
+set_target_properties(${CMAKE_PROJECT_NAME} PROPERTIES MSVC_RUNTIME_LIBRARY "MultiThreaded$<$<CONFIG:Debug>:Debug>")
+
+add_subdirectory(vendor)
+
+target_include_directories(${CMAKE_PROJECT_NAME} PRIVATE src vendor/scs_sdk/include vendor/imgui vendor/minhook/include)
+
+target_link_libraries(${CMAKE_PROJECT_NAME} PRIVATE fmt::fmt-header-only imgui minhook)

CMakeSettings.json

@@ -0,0 +1,22 @@
+{
+    "configurations": [
+        {
+            "name": "x64-Debug",
+            "generator": "Visual Studio 17 2022 Win64",
+            "configurationType": "Debug",
+            "inheritEnvironments": [
+                "msvc_x64_x64"
+            ],
+            "buildRoot": "${projectDir}\\build\\${name}"
+        },
+        {
+            "name": "x64-Release",
+            "generator": "Visual Studio 17 2022 Win64",
+            "configurationType": "Release",
+            "inheritEnvironments": [
+                "msvc_x64_x64"
+            ],
+            "buildRoot": "${projectDir}\\build\\${name}"
+        }
+    ]
+}

README.md

@@ -0,0 +1,38 @@
+# ts-extra-utilities
+
+> [!WARNING]
+> This is more of a PoC(Proof of Concept) and very much WIP(Work In Progress), it is not extensively tested or in any way a stable finished product.
+> If you decide to use this, expect possible issues / crashes (especially after game updates)
+
+A plugin for ATS/ETS2 to add some experimental extra functionality.
+
+## Current features
+
+ - Manually steerable trailer wheels
+    - Abilty to take control of the steerable wheels on a trailer
+
+   [Preview video](https://youtu.be/0kRavShaXy0)
+
+ - Individually detachable trailers
+    - Ability to disconnect and connect trailers individually and out of order
+    - Still has multiple issues when trailers are disconnected, trailer cables, 3rd person camera, ...
+
+   [Preview video](https://youtu.be/Nu6YvKKSL2g)
+
+ - Lockable trailer joints (PhysX only, `-physx` as a game launch parameter in Steam)
+    - Ability to lock the joints between vehicle and trailers so you can stop them from pivoting
+
+   [Preview video](https://youtu.be/zXtlzMVNEXM)
+
+## How to use
+
+Currently only works with DirectX11.
+
+Made for singleplayer, **NOT** recommended in multiplayer.
+
+
+Download the [latest release](https://github.com/dariowouters/ts-extra-utilities/releases/latest), copy the `ts-extra-utilities.dll` to `<game_install_location>/bin/win_x64/plugins`  
+(if the plugins folder does not exists, you can create one)
+
+Then in-game you can toggle the UI with `delete` and `insert` to toggle the cursor.  
+(Keybinds are currently hardcoded, will make it changable at some point.)

src/consts.hpp

@@ -0,0 +1,7 @@
+#pragma once
+
+#include "version.hpp"
+
+namespace ts_extra_utilities
+{
+}

src/core.cpp

@@ -0,0 +1,188 @@
+#include "core.hpp"
+
+#include <MinHook.h>
+
+#include "consts.hpp"
+
+#include "backends/imgui_impl_dx11.h"
+#include "backends/imgui_impl_win32.h"
+#include "memory/memory_utils.hpp"
+
+#include "managers/window_manager.hpp"
+#include "windows/trailer_manipulation.hpp"
+
+extern IMGUI_IMPL_API LRESULT ImGui_ImplWin32_WndProcHandler( HWND, UINT, WPARAM, LPARAM );
+
+namespace ts_extra_utilities
+{
+    CCore* CCore::g_instance = nullptr;
+
+    CCore::CCore( const scs_telemetry_init_params_v101_t* init_params ) : init_params_( init_params )
+    {
+        this->hooks_manager_ = new CHooksManager();
+        this->window_manager_ = new CWindowManager();
+        scs_log_ = init_params->common.log;
+        g_instance = this;
+    }
+
+    CCore::~CCore()
+    {
+        this->destroy();
+        MH_RemoveHook( nullptr );
+    }
+
+    bool CCore::init()
+    {
+        MH_Initialize();
+        this->info( "Initializing {}", VERSION );
+        truckersmp_ = GetModuleHandle( L"core_ets2mp.dll" ) != nullptr || GetModuleHandle( L"core_atsmp.dll" ) != nullptr;
+
+        this->dx11_hook = new CDirectX11Hook();
+        if ( !this->dx11_hook->hook_present() )
+        {
+            return false;
+        }
+        this->di8_hook = new CDirectInput8Hook();
+        if ( !this->di8_hook->hook() )
+        {
+            return false;
+        }
+
+        const auto trailer_manipulation = this->window_manager_->register_window( std::make_shared< CTrailerManipulation >() );
+
+        if ( !trailer_manipulation->init() )
+        {
+            g_instance->error( "Could not initialize the trailer manipulation module" );
+        }
+
+        return true;
+    }
+
+    void CCore::destroy()
+    {
+        delete this->dx11_hook;
+        delete this->di8_hook;
+        delete this->hooks_manager_;
+        delete this->window_manager_;
+    }
+
+    bool CCore::on_mouse_input( LPDIDEVICEOBJECTDATA rgdod )
+    {
+        if ( !this->disable_in_game_mouse ) return false;
+
+        auto& io = ImGui::GetIO();
+        if ( rgdod->dwOfs == DIMOFS_X )
+        {
+            io.MousePos.x += static_cast< float >( static_cast< int >( rgdod->dwData ) );
+        }
+        else if ( rgdod->dwOfs == DIMOFS_Y )
+        {
+            io.MousePos.y += static_cast< float >( static_cast< int >( rgdod->dwData ) );
+        }
+
+        return true;
+    }
+
+    bool CCore::render()
+    {
+        ImGui_ImplDX11_NewFrame();
+        ImGui_ImplWin32_NewFrame();
+        ImGui::NewFrame();
+
+        if ( this->render_ui )
+        {
+#ifdef _DEBUG
+            bool s = true;
+            ImGui::ShowDemoWindow( &s );
+#endif
+            this->window_manager_->render();
+        }
+        ImGui::EndFrame();
+        ImGui::Render();
+        ImGui_ImplDX11_RenderDrawData( ImGui::GetDrawData() );
+        return true;
+    }
+
+    void CCore::toggle_input_hook()
+    {
+        this->disable_in_game_mouse = !this->disable_in_game_mouse;
+        auto& io = ImGui::GetIO();
+        if ( this->disable_in_game_mouse )
+        {
+            io.MousePos.x = last_mouse_pos_x_;
+            io.MousePos.y = last_mouse_pos_y_;
+        }
+        else
+        {
+            last_mouse_pos_x_ = io.MousePos.x;
+            last_mouse_pos_y_ = io.MousePos.y;
+        }
+
+        io.MouseDrawCursor = this->disable_in_game_mouse;
+
+        this->debug( "Mouse hook is now {}", this->disable_in_game_mouse ? "active" : "disabled" );
+    }
+
+    void CCore::toggle_ui()
+    {
+        this->render_ui = !this->render_ui;
+    }
+
+    // TODO: add keybind settings
+    bool CCore::on_wnd_proc( HWND hwnd, UINT umsg, WPARAM wparam, LPARAM lparam )
+    {
+        if ( umsg == WM_KEYDOWN )
+        {
+            if ( wparam == VK_INSERT )
+            {
+                toggle_input_hook();
+                return true;
+            }
+            if ( wparam == VK_DELETE )
+            {
+                toggle_ui();
+                return true;
+            }
+        }
+        else if ( umsg == WM_MOUSEMOVE )
+        {
+            if ( this->disable_in_game_mouse )
+            {
+                return true;
+            }
+        }
+
+        if ( ImGui_ImplWin32_WndProcHandler( hwnd, umsg, wparam, lparam ) )
+        {
+            return true;
+        }
+        return false;
+    }
+
+    prism::base_ctrl_u* CCore::get_base_ctrl_instance()
+    {
+        if ( this->base_ctrl_instance_ptr_address != 0 ) return *reinterpret_cast< prism::base_ctrl_u** >( this->base_ctrl_instance_ptr_address );
+
+        const auto addr = memory::get_address_for_pattern( "48 8b 05 ? ? ? ? 48 8b 4b ? 48 8b 80 ? ? ? ? 48 8b b9" );
+
+        if ( addr == 0 ) return nullptr;
+        this->base_ctrl_instance_ptr_address = addr + *reinterpret_cast< int32_t* >( addr + 3 ) + 7;
+        this->game_actor_offset_in_base_ctrl = *reinterpret_cast< int32_t* >( addr + 14 );
+
+        this->info( "Found base_ctrl @ +{:x}", memory::as_offset( this->base_ctrl_instance_ptr_address ) );
+
+        return *reinterpret_cast< prism::base_ctrl_u** >( this->base_ctrl_instance_ptr_address );
+    }
+
+    prism::game_actor_u* CCore::get_game_actor()
+    {
+        auto* base_ctrl = this->get_base_ctrl_instance();
+
+        if ( base_ctrl == nullptr || this->game_actor_offset_in_base_ctrl == 0 )
+        {
+            return nullptr;
+        }
+
+        return *reinterpret_cast< prism::game_actor_u** >( reinterpret_cast< uint64_t >( base_ctrl ) + this->game_actor_offset_in_base_ctrl );
+    }
+}