Method addSection

src/pe.cpp:202–231

}

bool PE::addSection(std::string name, DWORD flags, std::vector<BYTE> bytes)
{
	// get next section's file offset
	DWORD newSectionFO = getNewSectionFileOffset();

	// create IMAGE_SECTION_HEADER for new section
	PIMAGE_SECTION_HEADER pNewSectionHeader = &pSectionHeader[pNtHeader->FileHeader.NumberOfSections];
	pNewSectionHeader->Characteristics = flags;
	pNewSectionHeader->Misc.VirtualSize = align(bytes.size(), pNtHeader->OptionalHeader.SectionAlignment);
	pNewSectionHeader->SizeOfRawData = align(bytes.size(), pNtHeader->OptionalHeader.FileAlignment);
	pNewSectionHeader->VirtualAddress = getNewSectionVirtualAddress();
	pNewSectionHeader->PointerToRawData = newSectionFO;
	CopyMemory(pNewSectionHeader->Name, name.c_str(), min(name.size(), IMAGE_SIZEOF_SHORT_NAME));

	// increase section count in the file header, and update size of image
	pNtHeader->FileHeader.NumberOfSections += 1;
	pNtHeader->OptionalHeader.SizeOfImage += pNewSectionHeader->Misc.VirtualSize;

	// resize our bytes vector to fit our new section bytes in
	this->bytes.resize(newSectionFO + pNewSectionHeader->SizeOfRawData);

	// actually copy the section bytes into our bytes vector
	std::copy(bytes.begin(), bytes.end(), this->bytes.begin() + newSectionFO);

	// parse section headers again (likely not required as we only store pointers to headers)
	if (!parseHeaders())
		return 0;

	return 1;
}

DWORD PE::getNewSectionVirtualAddress()
{