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());
-			});
-	}
+	_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);
-	}
+	_peFile.getRelocDir().extend(((_routines.size() + 0x3) & ~0x3) * 0x100);
 
 	for (auto& rtn : _routines)
 	{
@@ -862,39 +856,36 @@ X86BA_TEMPLATE void X86BinaryApplication<BitSize>::compile()
 	
 	//
 	// Append all relocations
-	if (_peFile.getRelocDir().isPresent()) 
+	for (auto it = _relocBlocks.begin(); it != _relocBlocks.end(); ++it)
 	{
-		for (auto it = _relocBlocks.begin(); it != _relocBlocks.end(); ++it)
-		{
-			u32 rva = it->first;
-			std::vector<RelocationEntry> const& entries = it->second;
+		u32 rva = it->first;
+		std::vector<RelocationEntry> const &entries = it->second;
 
-			if (entries.empty())
-				continue;
+		if (entries.empty())
+			continue;
 
-			// Pad the amount of relocs to ensure 32bit alignment
-			size_t relocEntrySize = (entries.size() + 0x3) & ~0x3;
-			size_t numHandled = 0ull;
+		// Pad the amount of relocs to ensure 32bit alignment
+		size_t relocEntrySize = (entries.size() + 0x3) & ~0x3;
+		size_t numHandled = 0ull;
 
-			// Create the BlockStream and add all relocatables.
-			pe::BlockStream bs = _peFile.getRelocDir().createBlock(rva, relocEntrySize);
+		// Create the BlockStream and add all relocatables.
+		pe::BlockStream bs = _peFile.getRelocDir().createBlock(rva, relocEntrySize);
 
-			if (!bs.valid())
-				continue;
+		if (!bs.valid())
+			continue;
 
-			// printf("relocEntrySize: 0x%llx entries\n", relocEntrySize);
+		// printf("relocEntrySize: 0x%llx entries\n", relocEntrySize);
 
-			for (auto& entry : entries)
-			{
-				bs.append(entry.type, entry.offset);
-				++numHandled;
-			}
+		for (auto& entry : entries)
+		{
+			bs.append(entry.type, entry.offset);
+			++numHandled;
+		}
 
-			// Pad.
-			for (size_t handled = numHandled; handled < relocEntrySize; ++handled)
-			{
-				bs.append(pe::RelocationType::REL_BASED_ABSOLUTE, 0);
-			}
+		// Pad.
+		for (size_t handled = numHandled; handled < relocEntrySize; ++handled)
+		{
+			bs.append(pe::RelocationType::REL_BASED_ABSOLUTE, 0);
 		}
 	}