YJN/selfpatch-slr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

spslr_finalize option parsing

Browse Source
This commit is contained in:
York Jasper Niebuhr 2025-10-27 16:10:49 +01:00
parent 54002646b0
commit 1585426dae
1 changed files with 68 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
finalize/finalize.cpp
View File

@ -1,14 +1,21 @@
#include <iostream> #include <iostream>
#include <string>
#include <getopt.h>
/* /*
TODO TODO
1. getopt -> --spslr=<dir>, --bin=<file> 1. Recursively gather all spslr CU files
2. Recursively gather all spslr CU files cu uid -> data {
3. Loop over all symbols of the binary target uid -> fields
ipin label -> data
dpin symbol -> data (list of sub dpins)
}
2. Merge types between CUs (per CU mapping to global target UIDs)
3. Find dpatch application order based on levels (high level (very nested) to low level (root))
4. Loop over all symbols of the binary
-> associate blocks via CU uid symbol -> associate blocks via CU uid symbol
-> find __spslr_program symbol (spslr vaddr pivot) -> find __spslr_program symbol (spslr vaddr pivot)
4. Find virtual address and file address for all pins 5. Find virtual address and file address for all pins
5. Calculate target randomization order
6. Emit patcher program into final executable and set __spslr_program 6. Emit patcher program into final executable and set __spslr_program
*/ */
@ -40,5 +47,61 @@ Note -> field alignment should probably be gathered by pinpoint plugin!
*/ */
int main(int argc, char** argv) { int main(int argc, char** argv) {
static option long_options[] = {
{ "help", no_argument, 0, 0 },
{ "spslr", required_argument, 0, 0 },
{ "bin", required_argument, 0, 0 },
{ "out", required_argument, 0, 0 },
{ "strip", no_argument, 0, 0 },
{ 0, 0, 0, 0 }
};
int option_index = 0;
int c;
std::string spslr_dir, bin_file, out_file;
while ((c = getopt_long(argc, argv, "", long_options, &option_index)) == 0) {
const option& opt = long_options[option_index];
std::string optname { opt.name };
if (optname == "help") {
std::cout << "To use spslr_finalize, supply these 3 arguments:" << std::endl;
std::cout << " --spslr=<dir> (the directory of .spslr files produced by spslr_pinpoint)" << std::endl;
std::cout << " --bin=<file> (the binary compiled with spslr_pinpoint to be finalized)" << std::endl;
std::cout << " --out=<file> (the finalized binary file to be written)" << std::endl;
return 0;
} else if (optname == "spslr") {
spslr_dir = std::string{ optarg };
} else if (optname == "bin") {
bin_file = std::string{ optarg };
} else if (optname == "out") {
out_file = std::string{ optarg };
} else if (optname == "strip") {
std::cerr << "Symbol stripping (--strip) is not yet implemented!" << std::endl;
return 1;
} else {
std::cerr << "Invalid option, try \"--help\"!" << std::endl;
return 1;
}
}
if (spslr_dir.empty()) {
std::cerr << "Missing spslr directory, supply it via --spslr=<dir>!" << std::endl;
return 1;
}
if (bin_file.empty()) {
std::cerr << "Missing input file path, supply it via --bin=<file>!" << std::endl;
return 1;
}
if (out_file.empty()) {
std::cerr << "Missing output file path, supply it via --out=<file>!" << std::endl;
return 1;
}
// TODO
std::cout << "Hello World!" << std::endl; std::cout << "Hello World!" << std::endl;
} }