Removed requirement of -x64 arg for 64bit binaries.

Fixed wrong byte order

README.md

@@ -19,7 +19,6 @@ 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() \
 {\
-	__nop();\
-	__nop();\
 	__debugbreak();\
 	__debugbreak();\
+	__nop();\
+	__nop();\
 	_disable();\
 }
 
 #define PERSES_MUTATION_END() \
 {\
 	_enable();\
-	__debugbreak();\
-	__debugbreak();\
 	__nop();\
 	__nop();\
+	__debugbreak();\
+	__debugbreak();\
 }

src/perses.cpp

@@ -1,6 +1,7 @@
 #include "perses.hpp"
 #include <argparse/argparse.hpp>
 
+
 template<int BitSize>
 void createApplication(perses::X86BinaryApplication<BitSize>* app, argparse::ArgumentParser& args)
 {
@@ -101,10 +102,6 @@ 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();
@@ -145,10 +142,19 @@ int main(int argc, char* argv[])
 	if (!std::filesystem::exists(filepath))
 	{
 		logger()->critical("Unable to find file: {}.", filepath);
-		return 0;
+		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;
 	}
 
-	if (args.get<bool>("-x64"))
+	if (type == SCS_64BIT_BINARY)
 	{
 		createApplication(new perses::X86BinaryApplication<PERSES_64BIT>(filepath), args);
 	}

src/x86application.cpp

@@ -37,11 +37,14 @@ 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());
 }
@@ -714,7 +717,10 @@ 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)
 	{
@@ -856,6 +862,8 @@ 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;
@@ -888,6 +896,7 @@ 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();