mirror of https://github.com/pozm/gdke.git
works
This commit is contained in:
parent
967352460b
commit
f30700d15a
|
@ -0,0 +1,101 @@
|
|||
{
|
||||
// Use IntelliSense to learn about possible attributes.
|
||||
// Hover to view descriptions of existing attributes.
|
||||
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
|
||||
"version": "0.2.0",
|
||||
"configurations": [
|
||||
{
|
||||
"type": "lldb",
|
||||
"request": "launch",
|
||||
"name": "Debug unit tests in library 'gdke-gui'",
|
||||
"cargo": {
|
||||
"args": [
|
||||
"test",
|
||||
"--no-run",
|
||||
"--lib",
|
||||
"--package=gdke-gui"
|
||||
],
|
||||
"filter": {
|
||||
"name": "gdke-gui",
|
||||
"kind": "lib"
|
||||
}
|
||||
},
|
||||
"args": [],
|
||||
"cwd": "${workspaceFolder}"
|
||||
},
|
||||
{
|
||||
"type": "lldb",
|
||||
"request": "launch",
|
||||
"name": "Debug executable 'gdke-gui'",
|
||||
"cargo": {
|
||||
"args": [
|
||||
"build",
|
||||
"--bin=gdke-gui",
|
||||
"--package=gdke-gui"
|
||||
],
|
||||
"filter": {
|
||||
"name": "gdke-gui",
|
||||
"kind": "bin"
|
||||
}
|
||||
},
|
||||
"args": [],
|
||||
"cwd": "${workspaceFolder}"
|
||||
},
|
||||
{
|
||||
"type": "lldb",
|
||||
"request": "launch",
|
||||
"name": "Debug unit tests in executable 'gdke-gui'",
|
||||
"cargo": {
|
||||
"args": [
|
||||
"test",
|
||||
"--no-run",
|
||||
"--bin=gdke-gui",
|
||||
"--package=gdke-gui"
|
||||
],
|
||||
"filter": {
|
||||
"name": "gdke-gui",
|
||||
"kind": "bin"
|
||||
}
|
||||
},
|
||||
"args": [],
|
||||
"cwd": "${workspaceFolder}"
|
||||
},
|
||||
{
|
||||
"type": "lldb",
|
||||
"request": "launch",
|
||||
"name": "Debug executable 'gdke'",
|
||||
"cargo": {
|
||||
"args": [
|
||||
"build",
|
||||
"--bin=gdke",
|
||||
"--package=gdke"
|
||||
],
|
||||
"filter": {
|
||||
"name": "gdke",
|
||||
"kind": "bin"
|
||||
}
|
||||
},
|
||||
"args": [],
|
||||
"cwd": "${workspaceFolder}"
|
||||
},
|
||||
{
|
||||
"type": "lldb",
|
||||
"request": "launch",
|
||||
"name": "Debug unit tests in executable 'gdke'",
|
||||
"cargo": {
|
||||
"args": [
|
||||
"test",
|
||||
"--no-run",
|
||||
"--bin=gdke",
|
||||
"--package=gdke"
|
||||
],
|
||||
"filter": {
|
||||
"name": "gdke",
|
||||
"kind": "bin"
|
||||
}
|
||||
},
|
||||
"args": [],
|
||||
"cwd": "${workspaceFolder}"
|
||||
}
|
||||
]
|
||||
}
|
File diff suppressed because it is too large
Load Diff
|
@ -6,4 +6,9 @@ edition = "2021"
|
|||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||
|
||||
[dependencies]
|
||||
poggers = {version = "0.1.1-dev3", path = "../poggers"}
|
||||
poggers = {version = "0.2.*", path = "../poggers"}
|
||||
|
||||
[workspace]
|
||||
members = [
|
||||
"gdke-gui",
|
||||
]
|
|
@ -0,0 +1,12 @@
|
|||
[package]
|
||||
name = "gdke-gui"
|
||||
version = "0.1.0"
|
||||
edition = "2021"
|
||||
|
||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||
|
||||
[dependencies]
|
||||
egui = "0.20.1"
|
||||
eframe = { version = "0.20.1", features = ["persistence"] }
|
||||
serde = { version = "1", features = ["derive"] } # You only need this if you want app persistence
|
||||
poggers = {version = "0.2.*", path = "../../poggers"}
|
|
@ -0,0 +1,130 @@
|
|||
use std::{borrow::BorrowMut, ops::Deref, sync::mpsc::{Receiver, Sender}};
|
||||
|
||||
use eframe::CreationContext;
|
||||
use egui::{TextStyle, TextEdit};
|
||||
use poggers::external::process::{ExPartialProcess, ExProcess};
|
||||
|
||||
use crate::Data;
|
||||
|
||||
#[derive(serde::Deserialize, serde::Serialize,Debug)]
|
||||
pub struct gdkeApp {
|
||||
#[serde(skip)]
|
||||
procs : Vec<ExPartialProcess>,
|
||||
#[serde(skip)]
|
||||
selected: Option<ExPartialProcess>,
|
||||
#[serde(skip)]
|
||||
awaiting: bool,
|
||||
#[serde(skip)]
|
||||
last_key: String,
|
||||
#[serde(skip)]
|
||||
process: Option<ExProcess>,
|
||||
search_query: String,
|
||||
#[serde(skip)]
|
||||
rx: Option<std::sync::mpsc::Receiver<Data>>,
|
||||
#[serde(skip)]
|
||||
tx: Option<std::sync::mpsc::Sender<Data>>
|
||||
}
|
||||
impl Default for gdkeApp {
|
||||
fn default() -> Self {
|
||||
let procs = if let Ok(procs) = ExProcess::get_processes() {
|
||||
procs
|
||||
} else {
|
||||
Vec::new()
|
||||
};
|
||||
Self {
|
||||
procs,
|
||||
selected: None,
|
||||
process: None,
|
||||
search_query: String::new(),
|
||||
rx: None,
|
||||
awaiting: false,
|
||||
last_key: String::new(),
|
||||
tx: None
|
||||
}
|
||||
}
|
||||
}
|
||||
impl gdkeApp {
|
||||
pub fn new(cc: &CreationContext<'_>, rx: Receiver<Data>,tx: Sender<Data>) -> gdkeApp {
|
||||
if let Some(stor) = cc.storage {
|
||||
if let Some(data) = eframe::get_value::<Self>(stor, "d") {
|
||||
println!("Loaded data: {:?}", data);
|
||||
return Self {
|
||||
tx: Some(tx),
|
||||
rx: Some(rx),
|
||||
..Default::default()
|
||||
}
|
||||
} else {
|
||||
Self {
|
||||
tx: Some(tx),
|
||||
rx: Some(rx),
|
||||
..Default::default()
|
||||
}
|
||||
}
|
||||
} else {
|
||||
Self {
|
||||
tx: Some(tx),
|
||||
rx: Some(rx),
|
||||
..Default::default()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
impl eframe::App for gdkeApp {
|
||||
fn update(&mut self, ctx: &egui::Context, frame: &mut eframe::Frame) {
|
||||
let Self {last_key, awaiting, rx,tx, procs, selected, process, search_query } = self;
|
||||
egui::CentralPanel::default().show(ctx, |ui| {
|
||||
ui.heading("GDKE");
|
||||
ui.separator();
|
||||
egui::Window::new("Key")..open(awaiting).show(ctx, |ui| {
|
||||
ui.label("Getting key, please wait...");
|
||||
if !last_key.is_empty() {
|
||||
let mut keyda = last_key.clone();
|
||||
TextEdit::singleline(&mut keyda).show(ui);
|
||||
ui.label("Close this window when done.");
|
||||
}
|
||||
else if let Ok(data) = rx.as_ref().unwrap().try_recv() {
|
||||
match data {
|
||||
Data::Key(key) => {
|
||||
println!("Got key: {}", key);
|
||||
*last_key = key;
|
||||
}
|
||||
_ =>{ }
|
||||
}
|
||||
};
|
||||
});
|
||||
if !*awaiting {
|
||||
|
||||
|
||||
ui.label("Select a Godot process to find the encryption key for.");
|
||||
egui::TextEdit::singleline(&mut self.search_query).hint_text("Search...").show(ui);
|
||||
let text_style = TextStyle::Body;
|
||||
let row_height = ui.text_style_height(&text_style);
|
||||
let filtered_procs = if self.search_query.is_empty() {self.procs.iter().collect::<Vec::<&ExPartialProcess>>()} else {self.procs.iter()
|
||||
.filter(|p| p.name.contains(&self.search_query) || p.pid.to_string().contains(&self.search_query)).collect()
|
||||
};
|
||||
let selval = selected.clone();
|
||||
ui.separator();
|
||||
egui::ScrollArea::vertical().max_height(260f32).auto_shrink([false;2])
|
||||
.show_rows(ui, row_height, filtered_procs.len(), move |ui,row_range| {
|
||||
for row in row_range {
|
||||
if let Some(proc) = (&filtered_procs).get(row) {
|
||||
let owner_proc = proc.deref();
|
||||
ui.selectable_value(selected, Some(owner_proc.clone()) , proc.name.clone());
|
||||
}
|
||||
}
|
||||
});
|
||||
ui.separator();
|
||||
if let Some(selected) = selval {
|
||||
if ui.button(format!("get key for {}",selected.name)).clicked() {
|
||||
tx.as_ref().unwrap().send(Data::Pid(selected.pid)).unwrap();
|
||||
*awaiting = true;
|
||||
last_key.clear();
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
fn save(&mut self, storage: &mut dyn eframe::Storage) {
|
||||
eframe::set_value(storage, "d", &self)
|
||||
}
|
||||
}
|
|
@ -0,0 +1,8 @@
|
|||
pub mod app;
|
||||
|
||||
|
||||
pub enum Data {
|
||||
Pid(u32),
|
||||
Key(String),
|
||||
|
||||
}
|
|
@ -0,0 +1,61 @@
|
|||
use gdke_gui::{app::gdkeApp, Data};
|
||||
use poggers::{external::process::ExProcess, traits::Mem};
|
||||
|
||||
const SIGS: [&str; 3] = ["48 8D 05 ? ? ? ? 41 8A 04 04","48 8D 05 ? ? ? ? 0F B6 ? 03","4C 8D 05 ? ? ? ? 0F 1F 40 00"];
|
||||
|
||||
fn main() {
|
||||
|
||||
let (stx, srx) = std::sync::mpsc::channel::<Data>();
|
||||
let (ctx, crx) = std::sync::mpsc::channel::<Data>();
|
||||
|
||||
|
||||
let jh = std::thread::spawn(move|| {
|
||||
loop {
|
||||
if let Ok(x) = crx.try_recv() {
|
||||
match x {
|
||||
Data::Pid(pid) => {
|
||||
println!("Got pid: {}", pid);
|
||||
|
||||
let proc = ExProcess::new_from_pid(pid).unwrap();
|
||||
let bm = proc.get_base_module().unwrap();
|
||||
for sig in SIGS.iter() {
|
||||
let res = unsafe {bm.scan_virtual(sig)};
|
||||
if let Some(x) = res {
|
||||
let data = unsafe {bm.resolve_relative_ptr(x+3, 4)};
|
||||
if let Ok(x) = data {
|
||||
println!("found key @ {:X}", x);
|
||||
let key_data = unsafe {bm.read_sized(x, 32)};
|
||||
if let Ok(x) = key_data {
|
||||
// print!("Key: ");
|
||||
let mut data_string = String::new();
|
||||
for i in x {
|
||||
data_string.push_str(&format!("{:02X}", i));
|
||||
}
|
||||
println!("sending data {}", data_string);
|
||||
stx.send(Data::Key(data_string)).unwrap();
|
||||
break;
|
||||
}
|
||||
} else {
|
||||
println!("Unable to resolve lea relative ptr");
|
||||
}
|
||||
// println!("Found sig: {:X}", x);
|
||||
} else {
|
||||
println!("Failed to find with sig: {}", sig);
|
||||
}
|
||||
}
|
||||
},
|
||||
_ => {}
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
let native_options = eframe::NativeOptions::default();
|
||||
eframe::run_native(
|
||||
"gdke",
|
||||
native_options,
|
||||
Box::new(move |cc| Box::new(gdkeApp::new(cc, srx, ctx))),
|
||||
);
|
||||
|
||||
jh.join();
|
||||
}
|
|
@ -0,0 +1,11 @@
|
|||
{
|
||||
"folders": [
|
||||
{
|
||||
"path": "."
|
||||
},
|
||||
{
|
||||
"path": "../poggers"
|
||||
}
|
||||
],
|
||||
"settings": {}
|
||||
}
|
Loading…
Reference in New Issue