#include <idc.idc>

//
//+++++++++++++++++++++++++++++++++++++++++++++
//
// IDB2EXE.idc 
//
// Written by Spark.   nkspark_at_gmail.com
// 
//          2010.01.06
//+++++++++++++++++++++++++++++++++++++++++++++
//

#define E_MAGIC 							0x5a4d
#define E_LFANEW							0xe0
#define E_LFANEW_OFFSET				0x3c
#define SIGNATURE 						0x00004550
#define MACHINE								0x014c
#define SECTIONNUMBER_OFFSET	0x06
#define IAT_OFFSET						0x80
#define PEHEADERLENGTH				0xf8
#define SECTIONTABLE_OFFSET		E_LFANEW + PEHEADERLENGTH
#define SECTIONENTRYLENGTH		0x28


static GetIDBFilePath()
{
	auto strIDBFilePath;//idb full path name;
	auto strEXEFileName;
	auto iNameEnd;
	auto iPathEnd;
	auto strPath;
	auto pFile;
	
	strPath = GetInputFilePath();
	strEXEFileName = GetInputFile();
	iPathEnd = strstr(strPath,strEXEFileName);
	iNameEnd = strstr(strEXEFileName,".");
	strIDBFilePath = substr(strPath,0,iPathEnd) + substr(strEXEFileName,0,iNameEnd) + ".idb";
	
	//Message("idb file lies in %s\n",strIDBFilePath);
	
	pFile = fopen(strIDBFilePath,"r");
	if(!pFile)
	{
		strIDBFilePath = AskStr(strIDBFilePath,"Please provide the path of IDB file:");
		return strIDBFilePath;
	}
	fclose(pFile);
	return 	strIDBFilePath;
}

//从IDB文件中获取IAT的偏移地址
static GetIATOffset(strIDBFilePath, PEHeaderOffset)
{
		auto pIDB;
		auto IATOffset;
		
		pIDB =fopen(strIDBFilePath,"rb");

		if(!pIDB)
		{
			Message("Can not open idb file!\n");
			return -1;
		}
		
		fseek(pIDB,PEHeaderOffset + IAT_OFFSET,0);
		IATOffset = readlong(pIDB,0);

		fclose(pIDB);
		return IATOffset;
	
}

//从IDB文件中获取PE头的起始地址
static SearchPEHeader(pIDB)
{
		auto PEHeaderOffset;
		auto i;
		auto iFileSize;
		
		
		fseek(pIDB,0x0,2);
		iFileSize = ftell(pIDB);
		
		if( iFileSize == -1)
			return -1;
		
		for(i=0;i<iFileSize;i++)
		{
			fseek(pIDB,i,0);
			//Signature "PE"
			if(readlong(pIDB,0) == SIGNATURE)
				//Machine 0x014c
				if(readshort(pIDB,0) == MACHINE)
				{
					PEHeaderOffset=ftell(pIDB);
					PEHeaderOffset = PEHeaderOffset -0x06;
					return PEHeaderOffset;					
				}
		}		
		return -1;
}

//IDB中的FirstThrunk已经被IDA自动填充，需要清理回原状
static PatchIAT(addrImageBase, addrIAT)
{		
		auto addrOriginalFirstThunk;
		auto addrFirstThunk;
		auto ValueOfIAT;
		
		Message("Start patching IAT at %x\n",addrIAT);
		
		if(Dword(addrIAT) == BADADDR)
		{
			Message("IAT data was not loaded into IDB file!\n");
			return -1;
		}
		
		do
		{
			addrOriginalFirstThunk = addrImageBase + Dword(addrIAT);
			addrFirstThunk = addrImageBase + Dword(addrIAT+0x10);
			
			do
			{
				PatchDword(addrFirstThunk,Dword(addrOriginalFirstThunk));
				addrOriginalFirstThunk = addrOriginalFirstThunk +0x04;
				addrFirstThunk = addrFirstThunk +0x04;
									
			}while(Dword(addrOriginalFirstThunk) != 0);
			
			addrIAT = addrIAT + 0x14;
			
		}while(Dword(addrIAT) != 0);
		
}

//为创建的EXE文件生成一个PE头
static BuildHeader(pFile,strIDBFilePath)
{
		auto i,j;
		auto pIDB;
		auto addrPE;
		auto iSectionNumber;
		auto lByte;

		pIDB =fopen(strIDBFilePath,"rb");
		if(!pIDB)
		{
			Message("Can not open idb file!\n");
			return -1;
		}
		
		fseek(pFile,0x0,0);
		writeshort(pFile,E_MAGIC,0);
		fseek(pFile,E_LFANEW_OFFSET,0);
		writeshort(pFile,E_LFANEW,0);
		
		
		addrPE = SearchPEHeader(pIDB);
		if(addrPE == -1)
		{
			Message("Can not find PE header in IDB file!\n");
			return -1;
		}
		Message("PEHeaderOffset is %x\n",addrPE);
		
		fseek(pIDB,addrPE,0);
		fseek(pFile,E_LFANEW,0);
		
		//Copy PE header
		//IAT should be set to 0 later!!!
		for(i=0;i<PEHEADERLENGTH;i++)
		{
				lByte = fgetc(pIDB);
				fputc(lByte,pFile);
		}

		fseek(pIDB,addrPE+SECTIONNUMBER_OFFSET,0);
		
		iSectionNumber = fgetc(pIDB);
					
		Message("there are %d sections!\n",iSectionNumber);
		
		//Prepare to write section table
		fseek(pFile,SECTIONTABLE_OFFSET,0);
		fseek(pIDB,addrPE+SECTIONTABLE_OFFSET-E_LFANEW,0);
		
		for(j=0;j<iSectionNumber;j++)
		{
			
			//每个表项有6字节偏移；
			fseek(pIDB,6,1);
			//最后一个表项有多余4字节偏移；
			if(j == (iSectionNumber-1))
				fseek(pIDB,4,1);
			
			
			//section table should be corrected later!!!!		
			for(i=0;i<SECTIONENTRYLENGTH;i++)
			{
					lByte = fgetc(pIDB);
					fputc(lByte,pFile);
			}
			Message("write %d section table!\n",j);
		}
		fclose(pIDB);
		
		//返回addrPE以备后用 :)
		return addrPE;
		
}

//内存地址转换为文件地址
static RVA2FA(pFile)
{
		auto iSectionNumber;
		auto iSectionBase;
		auto i;
		
		fseek(pFile,E_LFANEW + SECTIONNUMBER_OFFSET,0);
		iSectionNumber = readshort(pFile,0);

		fseek(pFile,SECTIONTABLE_OFFSET + (iSectionNumber -1) * SECTIONENTRYLENGTH + 0x14,0);
		iSectionBase = readlong(pFile,0);			
		Message("Section base in exe file is %x\n",iSectionBase);

		return iSectionBase;
		
}
//修正PE头信息
static PEHeaderCorrect(pFile)
{
		//关键处理，暂不外传！
		return 0;
}

static main()
{
	auto temp,i;
	auto startAddr, endAddr;
	auto countOR;
	auto pFile;
	auto strOutFileName;
	auto segStart;
	auto segEnd;
	auto iImageBase;
	auto iSectionAlignment;
	auto PEHeaderOffset;
	auto IATOffset;
	auto strIDBFilePath;
	auto iSectionBase;

	startAddr = MinEA();
	endAddr = MaxEA();
	iImageBase = FirstSeg() & 0xffff0000;
	iSectionAlignment = FirstSeg() & 0xffff;
	
	Message("Base:%x ,EP:%x , Min:%x  , Max:%x  \n",iImageBase,BeginEA(),MinEA(),MaxEA());
	
	strIDBFilePath = GetIDBFilePath();
	Message("idb file lies in %s\n",strIDBFilePath);
	
	strOutFileName = strIDBFilePath + "_dump.exe";

	pFile=fopen(strOutFileName,"w+b");
	
	PEHeaderOffset = BuildHeader(pFile,strIDBFilePath);
	if (PEHeaderOffset == -1)
		return -1;
		
	IATOffset = GetIATOffset(strIDBFilePath,PEHeaderOffset);
	if (IATOffset == -1)
		return -1;
		
	iSectionBase = RVA2FA(pFile);

	PatchIAT(iImageBase,iImageBase + IATOffset);
	
	for(i=startAddr;i<endAddr;)
	{
		segStart = GetSegmentAttr(i,SEGATTR_START);
		if(segStart == BADADDR) break;
		segEnd = GetSegmentAttr(i,SEGATTR_END);
		if(segEnd == BADADDR) break;		
		savefile(pFile, iSectionBase + segStart - iImageBase -iSectionAlignment,segStart,segEnd-segStart);
		Message("seg %x : %x : %x \n",GetSegmentAttr(i,SEGATTR_ALIGN),segStart,segEnd);
		i = segEnd + 1;
	}
	
	PEHeaderCorrect(pFile);
	
	fclose(pFile);
	
	return 0;
}

[培训]内核驱动高级班，冲击BAT一流互联网大厂工作，每周日13:00-18:00直播授课