Compare commits

...

10 commits

33 changed files with 817 additions and 43 deletions

36
.cargo/config.toml Normal file
View file

@ -0,0 +1,36 @@
[build]
target = "x86_64-pc-windows-msvc"
[target.x86_64-pc-windows-msvc]
linker = "lld-link"
rustflags = [
"-C",
"link-args=/libpath:/home/user/.xwin/xwin/crt/lib/x86_64",
"-C",
"link-args=/libpath:/home/user/.xwin/xwin/sdk/lib/um/x86_64",
"-C",
"link-args=/libpath:/home/user/.xwin/xwin/sdk/lib/ucrt/x86_64"
]
[env]
CC_x86_64_pc_windows_msvc="clang-cl"
CXX_x86_64_pc_windows_msvc="clang-cl"
AR_x86_64_pc_windows_msvc="llvm-lib"
LD_x86_64_pc_windows_msvc="lld-link"
# wine can be quite spammy with log messages and they're generally uninteresting
WINEDEBUG="-all"
# Use wine to run test executables
CARGO_TARGET_X86_64_PC_WINDOWS_MSVC_RUNNER="wine"
# Note that we only disable unused-command-line-argument here since clang-cl
# doesn't implement all of the options supported by cl, but the ones it doesn't
# are _generally_ not interesting.
CL_FLAGS="-Wno-unused-command-line-argument -fuse-ld=lld-link /imsvc/home/user/.xwin/xwin/crt/include /imsvc/home/user/.xwin/xwin/sdk/include/ucrt /imsvc/home/user/.xwin/xwin/sdk/include/um /imsvc/home/user/.xwin/xwin/sdk/include/shared"
CFLAGS_x86_64_pc_windows_msvc="-Wno-unused-command-line-argument -fuse-ld=lld-link /imsvc/home/user/.xwin/xwin/crt/include /imsvc/home/user/.xwin/xwin/sdk/include/ucrt /imsvc/home/user/.xwin/xwin/sdk/include/um /imsvc/home/user/.xwin/xwin/sdk/include/shared"
CXXFLAGS_x86_64_pc_windows_msvc="-Wno-unused-command-line-argument -fuse-ld=lld-link /imsvc/home/user/.xwin/xwin/crt/include /imsvc/home/user/.xwin/xwin/sdk/include/ucrt /imsvc/home/user/.xwin/xwin/sdk/include/um /imsvc/home/user/.xwin/xwin/sdk/include/shared"
# Let cargo know what linker to invoke if you haven't already specified it
# in a .cargo/config.toml file
CARGO_TARGET_X86_64_PC_WINDOWS_MSVC_LINKER="lld-link"
CARGO_TARGET_X86_64_PC_WINDOWS_MSVC_RUSTFLAGS="-Lnative=/home/user/.xwin/xwin/crt/lib/x86_64 -Lnative=/home/user/.xwin/xwin/sdk/lib/um/x86_64 -Lnative=/home/user/.xwin/xwin/sdk/lib/ucrt/x86_64"
LDFLAGS_x86_64_pc_windows_msvc="/libpath:/home/user/.xwin/xwin/crt/lib/x86_64 libpath:/home/user/.xwin/xwin/sdk/lib/um/x86_64 /libpath:/home/user/.xwin/xwin/sdk/lib/ucrt/x86_64"

View file

@ -1,20 +1,3 @@
[package]
name = "unreal-sdk"
version = "0.1.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
log = "0.4.0"
thiserror = "1.0.0"
itertools = "0.10.0"
rayon = "1.0.0"
bitflags = "1.0.0"
anyhow = "1.0"
widestring = "1.0"
lazy_static = "1.4.0"
once_cell = "1.17.1"
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0"
[workspace]
members = ["sdk-serializer", "unreal-sdk", "sdk-generator", "pdb-helper"]
resolver = "2"

21
pdb-helper/Cargo.toml Normal file
View file

@ -0,0 +1,21 @@
[package]
name = "pdb-helper"
version = "0.1.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
log = "0.4.0"
anyhow = "1.0"
once_cell = "1.17.1"
pdb = "0.8.0"
[dependencies.windows]
version = "0.44"
features = [
"Win32_Foundation",
"Win32_System_Threading",
"Win32_System_LibraryLoader",
"Win32_UI_WindowsAndMessaging",
]

189
pdb-helper/src/lib.rs Normal file
View file

@ -0,0 +1,189 @@
use std::{ptr::NonNull, sync::Mutex};
use anyhow::Context;
use once_cell::sync::OnceCell;
use pdb::{DataSymbol, FallibleIterator, ProcedureReferenceSymbol, ProcedureSymbol, SymbolData};
use windows::{core::PCWSTR, Win32::System::LibraryLoader::GetModuleHandleW};
#[derive(Debug, Clone, Copy)]
pub struct Globals {
pub names: NonNull<()>,
pub objects: NonNull<()>,
pub process_event: usize,
}
static PDB: OnceCell<Mutex<PdbCtx>> = OnceCell::new();
pub fn get_pdb() -> std::sync::MutexGuard<'static, PdbCtx> {
PDB.get_or_init(|| Mutex::new(PdbCtx::new().expect("failed to create PDB helper.")))
.lock()
.unwrap()
}
#[derive(Debug)]
pub struct PdbCtx {
base: usize,
pdb: pdb::PDB<'static, std::fs::File>,
}
// i dont care i need this to be safe and its stored as a dyn so any Send trait is removed
unsafe impl Send for PdbCtx {}
impl PdbCtx {
pub fn new() -> anyhow::Result<Self> {
let base = unsafe {
GetModuleHandleW(PCWSTR::null())
.context("could not get process base")?
.0 as usize
};
let pdb_path = std::env::current_exe()
.context("current exe")?
.with_file_name("ShooterGame.pdb");
let pdb_file = std::fs::File::open(&pdb_path).context("pdb file open")?;
let pdb = pdb::PDB::open(pdb_file).context("parse pdb file")?;
Ok(Self { base, pdb })
}
/// returns the address of the function
pub fn find_function(&mut self, function_name: &str) -> anyhow::Result<usize> {
let globals = self.pdb.global_symbols().context("global symbols")?;
let mut iter = globals.iter();
while let Some(symbol) = iter.next()? {
match symbol.parse() {
Ok(pdb::SymbolData::ProcedureReference(data))
if data.name.map(|name| name.to_string() == function_name) == Some(true) =>
{
log::trace!("{data:?}");
let dbg_info = self.pdb.debug_information()?;
let mods = dbg_info
.modules()?
.nth(data.module.context("module idx none")?)
.context("module not found")?
.context("module found but none")?;
let a = self
.pdb
.module_info(&mods)
.context("module info not found")?
.context("module info found but none")?;
let sym = a
.symbols_at(data.symbol_index)
.context("symbol not found")?
.next()
.context("symbol failed to read")?
.context("symbol read but none")?
.parse()
.context("symbol failed to parse")?;
log::debug!("actual symbol: {sym:?}");
if let SymbolData::Procedure(ProcedureSymbol { offset, .. }) = sym {
// asdf
let get_names = offset
.to_rva(&self.pdb.address_map()?)
.context("rva none")?;
return Ok(self.base + get_names.0 as usize);
} else {
return Err(anyhow::anyhow!("no proceduresymbol found"));
}
}
_ => {}
}
}
return Err(anyhow::anyhow!("no procedureref found"));
}
pub fn parse_names(&mut self, data: ProcedureReferenceSymbol) -> anyhow::Result<usize> {
log::trace!("{data:?}");
let dbg_info = self.pdb.debug_information()?;
let mods = dbg_info
.modules()?
.nth(data.module.context("module idx none")?)
.context("module not found")?
.context("module found but none")?;
let a = self
.pdb
.module_info(&mods)
.context("module info not found")?
.context("module info found but none")?;
let sym = a
.symbols_at(data.symbol_index)
.context("symbol not found")?
.next()
.context("symbol failed to read")?
.context("symbol read but none")?
.parse()
.context("symbol failed to parse")?;
log::debug!("actual symbol: {sym:?}");
if let SymbolData::Procedure(ProcedureSymbol { offset, .. }) = sym {
// asdf
let get_names = offset
.to_rva(&self.pdb.address_map()?)
.context("rva none")?;
let get_names = unsafe {
core::mem::transmute::<_, unsafe extern "win64" fn() -> *mut ()>(
self.base + get_names.0 as usize,
)
};
let names = unsafe { get_names() };
log::debug!("names: {:?}", names);
Ok(names as usize)
} else {
Err(anyhow::anyhow!("no proceduresymbol found"))
}
}
pub fn parse_objects(&mut self, data: DataSymbol) -> anyhow::Result<usize> {
let get_names = data
.offset
.to_rva(&self.pdb.address_map()?)
.context("rva none")?;
let objects = self.base + get_names.0 as usize;
Ok(objects)
}
pub fn find_ue_globals(&mut self) -> anyhow::Result<Globals> {
let globals = self.pdb.global_symbols().context("global symbols")?;
let mut iter = globals.iter();
let mut names = None;
let mut objects = None;
while let Some(symbol) = iter.next()? {
match symbol.parse() {
Ok(pdb::SymbolData::Data(data)) if data.name.to_string() == "GUObjectArray" => {
objects = Some(self.parse_objects(data)?);
}
Ok(pdb::SymbolData::ProcedureReference(data))
if data.name.map(|name| name.to_string() == "FName::GetNames")
== Some(true) =>
{
names = Some(self.parse_names(data)?);
}
_ => {}
}
}
let process_event = self
.find_function("UObject::ProcessEvent")
.context("could not find UObject::ProcessEvent!")?;
let names = names.context("could not find names!")?;
let objects = objects.context("could not find objects!")?;
Ok(Globals {
names: NonNull::new(names as _).context("GNames was null!")?,
objects: NonNull::new(objects as _).context("GObjects was null!")?,
process_event,
})
}
}

2
sdk-generator/.gitignore vendored Normal file
View file

@ -0,0 +1,2 @@
/target
/Cargo.lock

28
sdk-generator/Cargo.toml Normal file
View file

@ -0,0 +1,28 @@
[package]
name = "sdk-generator"
version = "0.1.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[lib]
name = "sdk_generator_version"
path = "src/lib.rs"
crate-type = ["cdylib"]
[dependencies]
log = "0.4.0"
anyhow = "1.0"
pdb-helper = {path = "../pdb-helper"}
simple_logger = "*"
unreal-sdk = {path = "../unreal-sdk"}
[dependencies.windows]
version = "0.44"
features = [
"Win32_Foundation",
"Win32_System_Threading",
]

11
sdk-generator/build.rs Normal file
View file

@ -0,0 +1,11 @@
use std::path::PathBuf;
fn main() {
println!("cargo:rerun-if-changed=build.rs");
let root = PathBuf::from(std::env::var("CARGO_MANIFEST_DIR").unwrap());
let def = root.join("proxy.def");
println!("cargo:rustc-link-lib=dylib=d3d11");
println!("cargo:rustc-link-arg=/def:{}", def.display());
}

20
sdk-generator/proxy.def Normal file
View file

@ -0,0 +1,20 @@
LIBRARY VERSION.dll
EXPORTS
GetFileVersionInfoA=c:\windows\system32\version.dll.GetFileVersionInfoA
GetFileVersionInfoByHandle=c:\windows\system32\version.dll.GetFileVersionInfoByHandle
GetFileVersionInfoExA=c:\windows\system32\version.dll.GetFileVersionInfoExA
GetFileVersionInfoExW=c:\windows\system32\version.dll.GetFileVersionInfoExW
GetFileVersionInfoSizeA=c:\windows\system32\version.dll.GetFileVersionInfoSizeA
GetFileVersionInfoSizeExA=c:\windows\system32\version.dll.GetFileVersionInfoSizeExA
GetFileVersionInfoSizeExW=c:\windows\system32\version.dll.GetFileVersionInfoSizeExW
GetFileVersionInfoSizeW=c:\windows\system32\version.dll.GetFileVersionInfoSizeW
GetFileVersionInfoW=c:\windows\system32\version.dll.GetFileVersionInfoW
VerFindFileA=c:\windows\system32\version.dll.VerFindFileA
VerFindFileW=c:\windows\system32\version.dll.VerFindFileW
VerInstallFileA=c:\windows\system32\version.dll.VerInstallFileA
VerInstallFileW=c:\windows\system32\version.dll.VerInstallFileW
VerLanguageNameA=c:\windows\system32\version.dll.VerLanguageNameA
VerLanguageNameW=c:\windows\system32\version.dll.VerLanguageNameW
VerQueryValueA=c:\windows\system32\version.dll.VerQueryValueA
VerQueryValueW=c:\windows\system32\version.dll.VerQueryValueW

88
sdk-generator/src/lib.rs Normal file
View file

@ -0,0 +1,88 @@
use std::{collections::HashSet, time::Duration};
use pdb_helper as pdb;
use simple_logger::SimpleLogger;
use unreal_sdk::sdk::output::rust::generate_partial_sdk_to_tmp;
use windows::s;
#[no_mangle]
pub extern "win64" fn DllMain(_module: usize, reason: u32, _: usize) -> bool {
match reason {
1 => {
if std::env::current_exe()
.expect("current exe")
.file_name()
.expect("file name")
.to_string_lossy()
== "ShooterGame.exe"
{
SimpleLogger::new().init().unwrap();
log::info!("hello");
std::thread::spawn(move || {
let _wnd = loop {
let window = unsafe {
windows::Win32::UI::WindowsAndMessaging::FindWindowA(
None,
s!("ARK: Survival Evolved"),
)
};
if window.0 != 0 {
log::debug!("found window: {window:?}");
break window;
}
log::trace!("window was 0");
std::thread::sleep(std::time::Duration::from_millis(150));
};
let globals = pdb::get_pdb().find_ue_globals().expect("globals");
unreal_sdk::global_tables::names::GNAMES
.write()
.unwrap()
.set_names(globals.names.cast());
unreal_sdk::global_tables::objects::GOBJECTS
.write()
.unwrap()
.set_objects(globals.objects.cast());
std::thread::sleep(Duration::from_secs(30));
let sdk = unreal_sdk::sdk::Sdk::new().patch();
_ = generate_partial_sdk_to_tmp(
&sdk,
HashSet::from_iter([
"Package_ShooterGame",
"Package_DeathItemCache",
"Package_WeapC4",
"Package_ExplorerChest_Base",
"Package_PointOfInterestBP_MissionStart",
"Package_BP_DedicatedStorage",
"Package_ElectricGenerator",
"Package_ExplorerChest_Base",
"Package_FeedingTroughBaseBP",
"Package_MilkGlider_Character_BP",
"Package_PointOfInterestBP_MissionStart",
"Package_SleepingBag",
"Package_StorageBox_Huge",
"Package_StorageBox_TekGenerator",
"Package_StructureAmmoContainer",
"Package_StructureTurretBaseBP_Heavy",
"Package_StructureTurretBaseBP",
"Package_StructureTurretTek",
"Package_Structure_DinoLeash",
"Package_Structure_TekAlarm",
"Package_BP_DedicatedStorage",
"Package_Buff_Companion_HLNA",
"Package_Buff_TekArmor_Pants",
"Package_Buff_TekArmor_Gloves",
"Package_Buff_TekArmor_Sword",
"Package_Buff_TekArmor_Shirt_Rework",
]),
);
});
}
}
_ => {}
}
true
}

2
sdk-serializer/.gitignore vendored Normal file
View file

@ -0,0 +1,2 @@
/target
/Cargo.lock

26
sdk-serializer/Cargo.toml Normal file
View file

@ -0,0 +1,26 @@
[package]
name = "sdk-serializer"
version = "0.1.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[lib]
name = "sdk_serializer_version"
path = "src/lib.rs"
crate-type = ["cdylib"]
[dependencies]
log = "0.4.0"
anyhow = "1.0"
pdb-helper = {path = "../pdb-helper"}
simple_logger = "*"
unreal-sdk = {path = "../unreal-sdk"}
[dependencies.windows]
version = "0.44"
features = [
"Win32_Foundation",
"Win32_System_Threading",
]

11
sdk-serializer/build.rs Normal file
View file

@ -0,0 +1,11 @@
use std::path::PathBuf;
fn main() {
println!("cargo:rerun-if-changed=build.rs");
let root = PathBuf::from(std::env::var("CARGO_MANIFEST_DIR").unwrap());
let def = root.join("proxy.def");
println!("cargo:rustc-link-lib=dylib=d3d11");
println!("cargo:rustc-link-arg=/def:{}", def.display());
}

20
sdk-serializer/proxy.def Normal file
View file

@ -0,0 +1,20 @@
LIBRARY VERSION.dll
EXPORTS
GetFileVersionInfoA=c:\windows\system32\version.dll.GetFileVersionInfoA
GetFileVersionInfoByHandle=c:\windows\system32\version.dll.GetFileVersionInfoByHandle
GetFileVersionInfoExA=c:\windows\system32\version.dll.GetFileVersionInfoExA
GetFileVersionInfoExW=c:\windows\system32\version.dll.GetFileVersionInfoExW
GetFileVersionInfoSizeA=c:\windows\system32\version.dll.GetFileVersionInfoSizeA
GetFileVersionInfoSizeExA=c:\windows\system32\version.dll.GetFileVersionInfoSizeExA
GetFileVersionInfoSizeExW=c:\windows\system32\version.dll.GetFileVersionInfoSizeExW
GetFileVersionInfoSizeW=c:\windows\system32\version.dll.GetFileVersionInfoSizeW
GetFileVersionInfoW=c:\windows\system32\version.dll.GetFileVersionInfoW
VerFindFileA=c:\windows\system32\version.dll.VerFindFileA
VerFindFileW=c:\windows\system32\version.dll.VerFindFileW
VerInstallFileA=c:\windows\system32\version.dll.VerInstallFileA
VerInstallFileW=c:\windows\system32\version.dll.VerInstallFileW
VerLanguageNameA=c:\windows\system32\version.dll.VerLanguageNameA
VerLanguageNameW=c:\windows\system32\version.dll.VerLanguageNameW
VerQueryValueA=c:\windows\system32\version.dll.VerQueryValueA
VerQueryValueW=c:\windows\system32\version.dll.VerQueryValueW

59
sdk-serializer/src/lib.rs Normal file
View file

@ -0,0 +1,59 @@
use pdb_helper as pdb;
use simple_logger::SimpleLogger;
use std::time::Duration;
use windows::s;
#[no_mangle]
pub extern "win64" fn DllMain(_module: usize, reason: u32, _: usize) -> bool {
match reason {
1 => {
if std::env::current_exe()
.expect("current exe")
.file_name()
.expect("file name")
.to_string_lossy()
== "ShooterGame.exe"
{
SimpleLogger::new().init().unwrap();
log::info!("hello");
std::thread::spawn(move || {
let _wnd = loop {
let window = unsafe {
windows::Win32::UI::WindowsAndMessaging::FindWindowA(
None,
s!("ARK: Survival Evolved"),
)
};
if window.0 != 0 {
log::debug!("found window: {window:?}");
break window;
}
log::trace!("window was 0");
std::thread::sleep(std::time::Duration::from_millis(150));
};
let globals = pdb::get_pdb().find_ue_globals().expect("globals");
unreal_sdk::global_tables::names::GNAMES
.write()
.unwrap()
.set_names(globals.names.cast());
unreal_sdk::global_tables::objects::GOBJECTS
.write()
.unwrap()
.set_objects(globals.objects.cast());
std::thread::sleep(Duration::from_secs(30));
let sdk = unreal_sdk::sdk::repr::Sdk::build().expect("sdk");
sdk.save_to_path("z:/tmp/sdk.ron").expect("writing sdk");
log::info!("done.");
});
}
}
_ => {}
}
true
}

2
unreal-sdk/.gitignore vendored Normal file
View file

@ -0,0 +1,2 @@
/target
/Cargo.lock

24
unreal-sdk/Cargo.toml Normal file
View file

@ -0,0 +1,24 @@
[package]
name = "unreal-sdk"
version = "0.1.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
log = "0.4.0"
thiserror = "1.0.0"
itertools = "0.10.0"
rayon = "1.0.0"
bitflags = "1.0.0"
anyhow = "1.0"
widestring = "1.0"
lazy_static = "1.4.0"
once_cell = "1.17.1"
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0"
serde_json_any_key = "2.0.0"
ron = "0.8.0"

View file

@ -0,0 +1 @@
nightly

View file

@ -1,9 +1,12 @@
use std::collections::{btree_map::Entry, BTreeMap};
use std::path::Path;
use anyhow::Context;
use itertools::Itertools;
use rayon::prelude::{IntoParallelRefIterator, ParallelIterator};
use rayon::prelude::{IntoParallelIterator, IntoParallelRefIterator, ParallelIterator};
use ron::ser::PrettyConfig;
use crate::global_tables::objects::GOBJECTS;
use crate::sdk::repr::*;
use crate::v2_types::{
any_type::{self, AnyField, AnyObject, AnyProperty, AnyStruct},
@ -11,12 +14,79 @@ use crate::v2_types::{
AsUObject, UArrayPropertyTrait, UBoolPropertyTrait, UEnumPropertyTrait, UEnumTrait,
UObjectNonConst, UObjectTrait, UPropertyTrait, UStructPropertyTrait, *,
},
UClass, UEnum, UFunction, UProperty, UScriptStruct,
UClass, UEnum, UFunction, UProperty,
};
use crate::v2_types::{UScriptStruct, UStruct};
use super::repr::{
Class, ClassField, ClassMethod, Enum, Package, ProcessedPackage, Type, UnrealType,
};
use super::FoldIntoPackages;
impl Sdk {
pub fn build() -> anyhow::Result<Self> {
let packages = Self::get_packages()?;
let packages = packages
.into_par_iter()
.map(|pkg| {
let package_ref = pkg
.package_object
.as_package_ref()
.context("object was not a package object.");
let output =
package_ref.and_then(|package_ref| pkg.process().map(|pkg| (package_ref, pkg)));
output
})
// remove empty packages
.filter(|pkg| {
pkg.as_ref()
.map(|(_, pkg)| !pkg.types.is_empty())
.unwrap_or(true)
})
.collect::<anyhow::Result<BTreeMap<_, _>>>()?;
Ok(Self { packages })
}
pub fn to_json(&self) -> Result<String, serde_json::Error> {
serde_json::to_string(self)
}
pub fn to_ron(&self) -> Result<String, ron::Error> {
ron::ser::to_string_pretty(self, PrettyConfig::new().compact_arrays(true))
}
pub fn save_to_path<P: AsRef<Path>>(&self, path: P) -> Result<(), anyhow::Error> {
self.to_ron()
.context("failed to serialize")
.and_then(|text| std::fs::write(path, text).context("failed to write to file."))
}
fn get_packages() -> anyhow::Result<Vec<Package>> {
let objects = GOBJECTS
.read()
.map_err(|_| anyhow::anyhow!("couldn't read gobject table"))?;
let objects = objects.as_objects().context("no object array")?;
let sorted_objects = objects
.iter()
.filter_map(|item| item.object())
.fold_into_packages();
let packages = sorted_objects
.into_iter()
.map(|(package_object, children)| Package {
package_object,
children,
})
.collect::<Vec<_>>();
Ok(packages)
}
}
fn default_or_anon(name: &str) -> bool {
name.contains("Default__")
@ -61,6 +131,7 @@ impl Enum {
});
Ok(UnrealType::Enum(Enum {
obj_ref: value.as_uobject().object_ref(),
name: value.get_name().context("could not get name")?,
full_name: value.get_full_name().context("could not get full name")?,
values,
@ -69,11 +140,112 @@ impl Enum {
}
impl Class {
pub fn from_uclass(value: UClass) -> anyhow::Result<UnrealType> {
todo!()
pub fn from_uclass(class: UClass) -> anyhow::Result<UnrealType> {
Ok(UnrealType::Class(Self {
kind: StructKind::Object,
..Self::from_struct_inner(unsafe { class.cast() })?
}))
}
pub fn from_uscriptstruct(value: UScriptStruct) -> anyhow::Result<UnrealType> {
todo!()
pub fn from_actor(class: UClass) -> anyhow::Result<UnrealType> {
Ok(UnrealType::Actor(Self {
kind: StructKind::Actor,
..Self::from_struct_inner(unsafe { class.cast() })?
}))
}
pub fn from_struct(strct: UScriptStruct) -> anyhow::Result<UnrealType> {
Ok(UnrealType::Struct(Self {
kind: StructKind::Struct,
..Self::from_struct_inner(unsafe { strct.cast() })?
}))
}
fn from_struct_inner(strct: UStruct) -> anyhow::Result<Self> {
let name = strct.get_name().context("could not get function name")?;
let full_name = strct
.get_full_name()
.context("could not get full function name")?;
let super_class = strct.super_field().and_then(|supr| {
if supr != strct {
Some(supr.as_uobject().object_ref())
} else {
None
}
});
let (fields, methods) = Self::process_children(&strct);
Ok(Self {
obj_ref: strct.as_uobject().object_ref(),
kind: StructKind::Struct,
size: *strct.property_size() as u32,
name,
full_name,
super_class,
properties_size: *strct.property_size() as u32,
min_alignment: *strct.min_alignment() as u32,
fields,
methods,
})
}
/// returns tuple of (fields, methods)
fn process_children(strct: &UStruct) -> (Vec<ClassField>, Vec<ClassMethod>) {
enum ClassChild {
Field(ClassField),
Method(ClassMethod),
}
let (fields, methods) = strct
.iter_children()
.map(|field| -> anyhow::Result<Option<ClassChild>> {
let child = match any_type::AnyField::from_field(field) {
AnyField::Property(prop) => Some(ClassChild::Field(ClassField::from_uprop(prop)?)),
AnyField::Struct(strct)
if let AnyStruct::Function(func) = AnyStruct::from_struct(strct) => {
{
Some(ClassChild::Method(ClassMethod::from_ufunction(func)?))
}
}
_ => None,
};
Ok(child)
})
.filter_map(|prop| match prop {
Ok(field) => field,
Err(err) => {
log::warn!("skipping field because: {err}");
None
}
})
.fold(
(Vec::new(), Vec::new()),
|(mut fields, mut methods), child| {
match child {
ClassChild::Field(field) => fields.push(field),
ClassChild::Method(method) => methods.push(method),
}
(fields, methods)
},
);
(fields, methods)
}
pub fn get_dependent_types(&self) -> Vec<ObjectRef> {
self.fields
.iter()
.filter_map(|field| field.ty.dependent_type())
.chain(
self.methods
.iter()
.flat_map(|method| method.get_dependent_types()),
)
.chain(self.super_class)
.collect::<Vec<_>>()
}
}
@ -90,6 +262,10 @@ impl ClassField {
ty: resolve_type(prop).context("failed to get field type")?,
})
}
pub fn get_dependent_types(&self) -> Vec<ObjectRef> {
self.ty.dependent_type().into_iter().collect()
}
}
impl ClassMethod {
@ -115,29 +291,69 @@ impl ClassMethod {
parameters: params,
})
}
pub fn get_dependent_types(&self) -> Vec<ObjectRef> {
self.parameters
.iter()
.filter_map(|field| field.ty.dependent_type())
.collect()
}
}
impl Package {
pub fn process(self) -> anyhow::Result<ProcessedPackage> {
self.children
let types = self
.children
.par_iter()
.filter_map(|&object| -> Option<UnrealType> {
match AnyObject::from_object(object) {
let ty = match AnyObject::from_object(object) {
AnyObject::Field(field) => match AnyField::from_field(field) {
AnyField::Enum(my_enum) => {}
AnyField::Enum(my_enum) => Enum::from_uenum(my_enum),
AnyField::Struct(my_struct) => match AnyStruct::from_struct(my_struct) {
my_struct @ AnyStruct::Class(_)
| my_struct @ AnyStruct::ScriptStruct(_) => {}
_ => {}
AnyStruct::Class(class) => Class::from_uclass(class),
AnyStruct::ScriptStruct(strct) => Class::from_struct(strct),
AnyStruct::Actor(actor) => Class::from_actor(actor),
_ => {
return None;
}
},
_ => {}
_ => {
return None;
}
},
_ => {}
}
_ => {
return None;
}
};
todo!()
});
todo!()
match ty {
Ok(ty) => {
return Some(ty);
}
Err(err) => {
log::error!("failed to serialize object {object:?}: {err}");
return None;
}
}
})
.map(|ty| (ty.obj_ref(), ty))
.collect::<BTreeMap<_, _>>();
let dependencies = types
.iter()
.flat_map(|(_, ty)| ty.get_dependent_types().into_iter().map(|obj| obj.package))
.sorted()
.dedup()
.collect::<Vec<_>>();
Ok(ProcessedPackage {
package_object: self
.package_object
.as_package_ref()
.context("not actually a package object")?,
types,
dependencies,
})
}
}

View file

@ -35,7 +35,7 @@ pub struct ObjectRef {
#[derive(Debug, Serialize, Deserialize)]
pub struct Sdk {
packages: BTreeMap<PackageRef, ProcessedPackage>,
pub packages: BTreeMap<PackageRef, ProcessedPackage>,
}
/// A package represents some group of packages that are related to another,
@ -58,7 +58,7 @@ pub struct ProcessedPackage {
/// all types extracted from this package referenced by their `ObjectRef`.
pub types: BTreeMap<ObjectRef, UnrealType>,
/// All other packages that types in this package depend on directly.
pub dependencies: Vec<ObjectRef>,
pub dependencies: Vec<PackageRef>,
}
#[derive(Debug, Serialize, Deserialize)]
@ -69,6 +69,26 @@ pub enum UnrealType {
Enum(Enum),
}
impl UnrealType {
pub fn obj_ref(&self) -> ObjectRef {
match self {
UnrealType::Class(obj) => obj.obj_ref,
UnrealType::Struct(obj) => obj.obj_ref,
UnrealType::Actor(obj) => obj.obj_ref,
UnrealType::Enum(obj) => obj.obj_ref,
}
}
pub fn get_dependent_types(&self) -> Vec<ObjectRef> {
match self {
UnrealType::Class(obj) | UnrealType::Struct(obj) | UnrealType::Actor(obj) => {
obj.get_dependent_types()
}
UnrealType::Enum(_) => vec![],
}
}
}
#[derive(Debug, PartialEq, Eq, Serialize, Deserialize)]
pub enum StructKind {
Object,
@ -78,6 +98,7 @@ pub enum StructKind {
#[derive(Debug, Serialize, Deserialize)]
pub struct Enum {
pub obj_ref: ObjectRef,
pub name: String,
pub full_name: String,
pub values: BTreeMap<u32, String>,
@ -85,6 +106,7 @@ pub struct Enum {
#[derive(Debug, Serialize, Deserialize)]
pub struct Class {
pub obj_ref: ObjectRef,
pub kind: StructKind,
pub size: u32,
pub name: String,
@ -94,9 +116,6 @@ pub struct Class {
pub min_alignment: u32,
pub fields: Vec<ClassField>,
pub methods: Vec<ClassMethod>,
/// types this class depends on; includes super types, types of fields and
/// types of function parameters.
pub dependencies: Vec<ObjectRef>,
}
#[derive(Debug, Serialize, Deserialize)]
@ -209,3 +228,19 @@ pub enum Type {
Class(ObjectRef),
Struct(ObjectRef),
}
impl Type {
pub fn dependent_type(&self) -> Option<ObjectRef> {
match self {
Type::Ptr(ty) | Type::Ref(ty) | Type::Array(ty) | Type::RawArray { ty, .. } => {
ty.dependent_type()
}
Type::WeakPtr(ty) | Type::SoftPtr(ty) | Type::LazyPtr(ty) | Type::AssetPtr(ty) => {
Some(*ty)
}
Type::Enum { underlying, .. } => underlying.dependent_type(),
Type::Class(ty) | Type::Struct(ty) => Some(*ty),
_ => None,
}
}
}