README.md

@@ -19,6 +19,7 @@ PERSES by default works off a command line. Listed below are the arguments requr
 | `-s` or `--symbol` 	| Symbol or list of symbols to be mutated. This requires a linked `.map` file.                                                                                              	| :x:                	|
 | `--map`            	| Map file to be linked. IDA Pro `.map` files must have their extension replaced with `.ida`.                                                                               	| :x:                	|
 | `--list`           	| List of functions to be mutated. Each entry must envelop one line and be formatted as `0x1000:0x2000` where `0x1000` is the start and `0x2000` is the end of the routine. 	| :x:                	|
+| `--x64`            	| Used to indicate that the file is of 64bit architecture (AMD64).                                                                                                          	| :heavy_check_mark: 	|
 | `--rets`           	| Allow PERSES to build a `RET` gadget pool used to create `JMP`s to random locations.                                                                                      	| :x:                	|
 | `--scan`           	| Force PERSES to scan for code protection markers.                                                                                                                         	| :x:                	|
 

sdk/PersesSDK.h

@@ -5,18 +5,18 @@
 
 #define PERSES_MUTATION_START() \
 {\
-	__debugbreak();\
-	__debugbreak();\
 	__nop();\
 	__nop();\
+	__debugbreak();\
+	__debugbreak();\
 	_disable();\
 }
 
 #define PERSES_MUTATION_END() \
 {\
 	_enable();\
-	__nop();\
-	__nop();\
 	__debugbreak();\
 	__debugbreak();\
+	__nop();\
+	__nop();\
 }

src/perses.cpp

@@ -1,7 +1,6 @@
 #include "perses.hpp"
 #include <argparse/argparse.hpp>
 
-
 template<int BitSize>
 void createApplication(perses::X86BinaryApplication<BitSize>* app, argparse::ArgumentParser& args)
 {
@@ -102,6 +101,10 @@ int main(int argc, char* argv[])
 	args.add_argument("-f", "--file")
 		.help("Input file path.")
 		.required();
+	args.add_argument("-x64")
+		.help("Required for X64 PE files.")
+		.default_value(false)
+		.implicit_value(true);
 	args.add_argument("-a", "--address")
 		.help("Address(es) to mutate")
 		.remaining();
@@ -142,19 +145,10 @@ int main(int argc, char* argv[])
 	if (!std::filesystem::exists(filepath))
 	{
 		logger()->critical("Unable to find file: {}.", filepath);
-		return 1;
-	}
-
-	// Determine arch. type automatically.
-	DWORD type = 0ul;
-
-	if (!GetBinaryTypeA(filepath.c_str(), &type) && GetLastError() != ERROR_BAD_EXE_FORMAT)
-	{
-		logger()->critical("Are you sure this is a executable file?");
-		return 1;
+		return 0;
 	}
 
-	if (type == SCS_64BIT_BINARY)
+	if (args.get<bool>("-x64"))
 	{
 		createApplication(new perses::X86BinaryApplication<PERSES_64BIT>(filepath), args);
 	}

src/x86application.cpp

@@ -37,14 +37,11 @@ X86BA_TEMPLATE void X86BinaryApplication<BitSize>::loadFromFile(std::string_view
 
 	_originalRelocs.clear();
 
-	if (_peFile.getRelocDir().isPresent())
-	{
 	_peFile.getRelocDir().forEachEntry(
 		[this](pe::BlockEntry const& entry)
 		{
 			_originalRelocs.insert(entry.getRva());
 		});
-	}
 
 	// printf("* - Number of relocations: 0x%x\n", _originalRelocs.size());
 }
@@ -717,10 +714,7 @@ X86BA_TEMPLATE bool X86BinaryApplication<BitSize>::transformRoutines()
 	}
 
 	// Guesstimate.. Fix this..
-	if (_peFile.getRelocDir().isPresent()) 
-	{
 	_peFile.getRelocDir().extend(((_routines.size() + 0x3) & ~0x3) * 0x100);
-	}
 
 	for (auto& rtn : _routines)
 	{
@@ -862,8 +856,6 @@ X86BA_TEMPLATE void X86BinaryApplication<BitSize>::compile()
 	
 	//
 	// Append all relocations
-	if (_peFile.getRelocDir().isPresent()) 
-	{
 	for (auto it = _relocBlocks.begin(); it != _relocBlocks.end(); ++it)
 	{
 		u32 rva = it->first;
@@ -896,7 +888,6 @@ X86BA_TEMPLATE void X86BinaryApplication<BitSize>::compile()
 			bs.append(pe::RelocationType::REL_BASED_ABSOLUTE, 0);
 		}
 	}
-	}
 
 	pe::SectionHeader& reloc = _peFile.getSectionHdr(".reloc");
 	std::uint32_t fileAlignment = _peFile.getPEHdr().getOptionalHdr().getFileAlignment();