Ejemplo n.º 1
0
// patches file's PE header to make it compatible with WinXP
void patchHeader(LPCSTR filename)
{
	PeLib::PeFile* pef = PeLib::openPeFile(filename);
	pef->readMzHeader();
	pef->readPeHeader();

	PatchPeHeaderVisitor vis;
	pef->visit(vis);

	delete pef;
}
Ejemplo n.º 2
0
void pack_pe(char * name)
{
    std::string filename = name;
    PeLib::PeFile* pef = PeLib::openPeFile(filename);
	
	if (!pef)
	{
		std::cout << "Invalid PE File" << std::endl;
		return ;
	}

	pef->readMzHeader();
	pef->readPeHeader();


	dump(centerOutput("----------------------------------------------"));
	dump(centerOutput("PE Loaded"));
	dump(centerOutput("----------------------------------------------"));
	
	PeRebuilderVisitor  v2;
	pef->visit(v2);
	
	delete pef;     
}
Ejemplo n.º 3
0
/**
* Extracts all icons from a PE file.
* @param filename Name of the PE file.
**/
void extractIcons(const std::string& filename)
{
	std::cout << "Opening file: " << filename << std::endl;
	
	PeLib::PeFile* pef = PeLib::openPeFile(filename);
	
	if (!pef)
	{
		std::cout << "Not a valid PE file" << std::endl;
		return;
	}
	
	try
	{
		pef->readMzHeader();
		pef->readPeHeader();
		pef->readResourceDirectory();
	}
	catch(std::exception& e)
	{
		delete pef;
		std::cout << "Not a valid PE file." << std::endl << std::endl;
		return;
	}
	
	unsigned int icons = pef->resDir().getNumberOfResources(PeLib::PELIB_RT_ICON);
	unsigned int icongroups = pef->resDir().getNumberOfResources(PeLib::PELIB_RT_GROUP_ICON);
	
	if (icongroups == -1 || icons == -1)
	{
		std::cout << "No icons found." << std::endl << std::endl;
		delete pef;
		return;
	}
	
	std::cout << icons << " icons found in " << icongroups << " icon groups." << std::endl;

	std::vector<PeLib::byte> data;
	
	using PeLib::word;
	
	// Icons in ICO fils have a different header than icons in PE files.
	// The first six bytes never change for simple ICO files.
	char iconheader[22] = {0, 0, 1, 0, 1, 0};
	
	unsigned int icongroupindex = pef->resDir().resourceTypeIdToIndex(PeLib::PELIB_RT_GROUP_ICON);
	
	for (unsigned int i=0;i<icongroups;i++)
	{
		std::cout << "Reading icon group " << i + 1 << "." << std::endl;
		
//		std::cout << "OH :" << pef->resDir().getResourceIdByIndex(icongroupindex, i) << std::endl;
		
		pef->resDir().getResourceData(PeLib::PELIB_RT_GROUP_ICON, pef->resDir().getResourceIdByIndex(icongroupindex, i), data);
		
		word entries = *(word*)(&data[4]); // Number of icons in icon group.
		for (int j=0;j<entries;j++)
		{
			std::cout << "Reading icon " << j + 1<< "." << std::endl;
			
			// Parts of the icon header can be copied from the PE icon to the ICO icon.
			std::copy(data.begin() + 6 + j * 14, data.begin() + 6 + j * 14 + 12, iconheader + 6);
			*(word*)(&iconheader[18]) = 0x16;
			
			// Read the icon data.
			std::vector<PeLib::byte> icondata;
			word iconid = *(word*)(&data[6 + 12 + j * 14]);
			pef->resDir().getResourceData(PeLib::PELIB_RT_ICON, iconid, icondata);
			
			std::string iconname = makeIconName(filename, iconid);
			std::ofstream file(iconname.c_str(), std::ios::binary);
			if (file)
			{
				std::cout << "Writing icon " << j + 1 << " to file " << iconname << "." << std::endl;
	
				// Write the icon header followed by the icon data.
				file.write(iconheader, 22);
				file.write(reinterpret_cast<char*>(&icondata[0]), icondata.size());
			}
			else
			{
				std::cout << "Couldn't open file: " << iconname << ". Make sure the icons subdirectory exists." << std::endl;
			}
		}
	}
	delete pef;
	std::cout << std::endl;
}