CryByteArrayGenerationWindow.cpp

#include "CryByteArrayGenerationWindow.h"
#include "ImlProvider.h"
#include "UIUtilities.h"
#include "BackendGlobalDef.h"

// This window needs access to the visible disassembly lines.
extern Vector<LONG_PTR> DisasmVisibleLines;

// The CryByteArrayGenerationWindow default constructor, accepting a number of selected rows.
CryByteArrayGenerationWindow::CryByteArrayGenerationWindow(const Vector<int>& rows) : CryDialogTemplate(CrySearchIml::GenerateByteArrayButton())
{
	this->Title("Generate Byte-array").SetRect(0, 0, 300, 140);
	
	this->mClose <<= THISBACK(CloseWindow);
	
	*this
		<< this->mCPPStyleSection.SetLabel("C++").HSizePos(5, 5).TopPos(5, 45)
		<< this->mCPPStyle.SetLabel("Byte-array:").LeftPos(10, 80).TopPos(20, 25)
		<< this->mCPPStyleSig.SetEditable(false).HSizePos(90, 10).TopPos(20, 25)
		<< this->mCSharpStyleSection.SetLabel("C#").HSizePos(5, 5).TopPos(55, 45)
		<< this->mCSharpStyle.SetLabel("Byte-array:").LeftPos(10, 80).TopPos(70, 25)
		<< this->mCSharpStyleSig.SetEditable(false).HSizePos(90, 10).TopPos(70, 25)
		<< this->mClose.SetLabel("Close").RightPos(5, 70).BottomPos(5, 25)
	;
	
	// Retrieve byte sets that are selected.
	Vector<Byte> byteSets;
	const int count = rows.GetCount();
	for (int i = 0; i < count; ++i)
	{
		ArrayOfBytes sequence;
#ifdef _WIN64
		DisasmGetLine(DisasmVisibleLines[rows[i]], mMemoryScanner->IsX86Process() ? CS_MODE_32 : CS_MODE_64, &sequence);
#else
		DisasmGetLine(DisasmVisibleLines[rows[i]], CS_MODE_32, &sequence);
#endif
		for (int y = 0; y < sequence.Size; ++y)
		{
			byteSets << sequence.Data[y];
		}
	}
	
	// Generate signatures in all supported styles.
	this->GenerateCPPStyle(byteSets);
	this->GenerateCSharpStyle(byteSets);
}

// The CryByteArrayGenerationWindow default destructor.
CryByteArrayGenerationWindow::~CryByteArrayGenerationWindow()
{
	
}

// Generates a C++-style byte array.
void CryByteArrayGenerationWindow::GenerateCPPStyle(const Vector<Byte>& aobs)
{
	// Set the string in the user interface controls.
	this->mCPPStyleSig.SetText(GenerateByteArray(aobs, ARRAYTYPE_CPP));
}

// Generates a C#-style byte array.
void CryByteArrayGenerationWindow::GenerateCSharpStyle(const Vector<Byte>& aobs)
{
	// Set generated signature inside top textbox.
	this->mCSharpStyleSig.SetText(GenerateByteArray(aobs, ARRAYTYPE_CSHARP));
}

// Executed when the dialog is closed.
void CryByteArrayGenerationWindow::CloseWindow()
{
	this->Close();
}