101 lines
3.1 KiB
C++
101 lines
3.1 KiB
C++
#include <iostream>
|
|
#include <string>
|
|
#include <getopt.h>
|
|
|
|
#include <LIEF/LIEF.hpp>
|
|
|
|
#include "accumulation.h"
|
|
#include "symbol_collection.h"
|
|
|
|
/*
|
|
Notes:
|
|
Datapins for same var/symbol are randomized in order of their level, from bottom of nest to top
|
|
The CU uid symbol helps differentiating between e.g. "file.c" and "sub/file.c" (symbtab has no idea)
|
|
Between CUs, types with the same name HAVE TO HAVE the same layout -> randomized together
|
|
To begin with, anonymous types are not allowed for randomization (later solved with hash(type) instead of name)!
|
|
*/
|
|
|
|
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;
|
|
}
|
|
|
|
if (!accumulate(spslr_dir)) {
|
|
std::cerr << "Failed to accumulate data from spslr directory!" << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
std::cout << "Gathered a total of " << targets.size() << " distinct targets from "
|
|
<< units.size() << " compilation units!" << std::endl;
|
|
|
|
std::unique_ptr<LIEF::ELF::Binary> bin = LIEF::ELF::Parser::parse(bin_file);
|
|
if (!bin) {
|
|
std::cerr << "Failed to parse binary \"" << bin_file << "\"!" << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
if (!collect_symbols(bin)) {
|
|
std::cerr << "Failed to collect symbols from \"" << bin_file << "\"!" << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
// TODO
|
|
/*
|
|
1. Find virtual address and file address for all pins
|
|
2. Emit patcher program into final executable and set __spslr_program
|
|
-> make all patcher addresses relative to &__spslr_program
|
|
-> dpatch entries ordered by level (high levels/nest depths first)
|
|
*/
|
|
}
|
|
|