SectionInfo.cpp

#include "stdafx.h"
#include "SectionInfo.h"
#include "afxdialogex.h"
#include "puPEinfoData.h"

const TCHAR SectionInfo::strName[6][20] = { L"Name", L"V.Address", L"V.Size", L"Offset", L"R.Size", L"Flag" };

// SectionInfo ¶Ô»°¿ò

IMPLEMENT_DYNAMIC(SectionInfo, CDialogEx)

SectionInfo::SectionInfo(CWnd* pParent /*=NULL*/)
	: CDialogEx(SectionInfo::IDD, pParent)
{
}

SectionInfo::~SectionInfo()
{
}

void SectionInfo::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_LIST1, m_SectionList);
}

BEGIN_MESSAGE_MAP(SectionInfo, CDialogEx)
END_MESSAGE_MAP()

BOOL SectionInfo::OnInitDialog()
{
	CDialogEx::OnInitDialog();
	InitSectionData();
	return TRUE;
}

void SectionInfo::InitSectionData()
{
	CRect rc;
	DWORD dwSize = m_SectionList.GetExtendedStyle();
	m_SectionList.SetExtendedStyle(dwSize | LVS_EX_FULLROWSELECT | LVS_EX_GRIDLINES);
	m_SectionList.GetClientRect(rc);
	int nWidth = rc.Width();
	for (int i = 0; i < 6; ++i) { m_SectionList.InsertColumn(i, strName[i], LVCFMT_CENTER, nWidth / 6); }

	pSectionHeadre = { 0 };	pNtHeadre = { 0 };
	if(!m_csFilePath.IsEmpty())
		SinglePuPEInfo::instance()->puOpenFileLoadEx(m_csFilePath);
	else
	{
		if (!SinglePuPEInfo::instance()->puGetLoadFileSuccess()) {
			if (!m_csFilePath.IsEmpty())
				return;
		}
		SinglePuPEInfo::instance()->puOpenFileLoadEx(m_csFilePath);
	}
	pSectionHeadre = (PIMAGE_SECTION_HEADER)SinglePuPEInfo::instance()->puGetSection();
	pNtHeadre = (PIMAGE_NT_HEADERS)SinglePuPEInfo::instance()->puGetNtHeadre();
	SectionCount = pNtHeadre->FileHeader.NumberOfSections;
	if (!pSectionHeadre || (!pNtHeadre) || (SectionCount <= 0))
		return;
	ShowSectionInfo();
	// Clear
	SinglePuPEInfo::instance()->puClearPeData();
}

void SectionInfo::ShowSectionInfo()
{ 
	if (!m_SectionList || (SectionCount <= 0))
		return;
	PIMAGE_SECTION_HEADER tempSectionHeadre = pSectionHeadre;
	if (!tempSectionHeadre)
		return;

	DWORD dwTemp = 0;	CString str;
	try
	{
		for (DWORD i = 0; i < SectionCount; ++i)
		{
			m_SectionList.InsertItem(i, NULL);
			str = tempSectionHeadre->Name;
			m_SectionList.SetItemText(i, 0, str);

			dwTemp = tempSectionHeadre->VirtualAddress;
			str.Format(L"%08X", dwTemp);
			m_SectionList.SetItemText(i, 1, str);

			dwTemp = tempSectionHeadre->SizeOfRawData;
			str.Format(L"%08X", dwTemp);
			m_SectionList.SetItemText(i, 2, str);

			dwTemp = tempSectionHeadre->PointerToRawData;
			str.Format(L"%08X", dwTemp);
			m_SectionList.SetItemText(i, 3, str);

			dwTemp = tempSectionHeadre->Misc.VirtualSize;
			str.Format(L"%08X", dwTemp);
			m_SectionList.SetItemText(i, 4, str);

			dwTemp = tempSectionHeadre->Characteristics;
			str.Format(L"%08X", dwTemp);
			m_SectionList.SetItemText(i, 5, str);

			++tempSectionHeadre;
		}
	}
	catch (const std::exception&)
	{
	}
}