// MemSpyyDlg.cpp : implementation file
//
#include "stdafx.h"
#include "MemSpyy.h"
#include "MemSpyyDlg.h"
#include ".\MemSpyydlg.h"
#include "psapi.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
// CAboutDlg dialog used for App About
class CAboutDlg : public CDialog
{
public:
CAboutDlg();
// Dialog Data
enum { IDD = IDD_ABOUTBOX };
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support
// Implementation
protected:
DECLARE_MESSAGE_MAP()
};
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}
void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()
// CMemSpyyDlg dialog
CMemSpyyDlg::CMemSpyyDlg(CWnd* pParent /*=NULL*/)
: CDialog(CMemSpyyDlg::IDD, pParent)
, m_PID(0)
, m_hLibModule(0)
{
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
void CMemSpyyDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
DDX_Control(pDX, IDC_LARGEST, m_LargestCtl);
DDX_Text(pDX, IDC_PID, m_PID);
DDX_Control(pDX, IDC_TOTALFREE, m_TotalFreeCtl);
DDX_Control(pDX, IDC_OGLCUSTOM, m_OpenGLControl);
DDX_Control(pDX, IDC_TOTALDLL, m_TotalDllCtl);
DDX_Control(pDX, IDC_PID, m_PIDCtl);
DDX_Control(pDX, IDC_PIDNAME, m_PIDNameCtl);
DDX_Control(pDX, IDC_STATUS, m_StatusCtl);
}
BEGIN_MESSAGE_MAP(CMemSpyyDlg, CDialog)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
//}}AFX_MSG_MAP
ON_WM_TIMER()
ON_EN_UPDATE(IDC_PID, OnEnUpdatePid)
END_MESSAGE_MAP()
// CMemSpyyDlg message handlers
BOOL CMemSpyyDlg::OnInitDialog()
{
CDialog::OnInitDialog();
// Add "About..." menu item to system menu.
// IDM_ABOUTBOX must be in the system command range.
ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
ASSERT(IDM_ABOUTBOX < 0xF000);
CMenu* pSysMenu = GetSystemMenu(FALSE);
if (pSysMenu != NULL)
{
CString strAboutMenu;
strAboutMenu.LoadString(IDS_ABOUTBOX);
if (!strAboutMenu.IsEmpty())
{
pSysMenu->AppendMenu(MF_SEPARATOR);
pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
}
}
// Set the icon for this dialog. The framework does this automatically
// when the application's main window is not a dialog
SetIcon(m_hIcon, TRUE); // Set big icon
SetIcon(m_hIcon, FALSE); // Set small icon
m_nTimer = SetTimer(1, 1000, 0);
return TRUE; // return TRUE unless you set the focus to a control
}
void CMemSpyyDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
if ((nID & 0xFFF0) == IDM_ABOUTBOX)
{
CAboutDlg dlgAbout;
dlgAbout.DoModal();
}
else
{
CDialog::OnSysCommand(nID, lParam);
}
}
// If you add a minimize button to your dialog, you will need the code below
// to draw the icon. For MFC applications using the document/view model,
// this is automatically done for you by the framework.
void CMemSpyyDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // device context for painting
SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
// Center icon in client rectangle
int cxIcon = GetSystemMetrics(SM_CXICON);
int cyIcon = GetSystemMetrics(SM_CYICON);
CRect rect;
GetClientRect(&rect);
int x = (rect.Width() - cxIcon + 1) / 2;
int y = (rect.Height() - cyIcon + 1) / 2;
// Draw the icon
dc.DrawIcon(x, y, m_hIcon);
}
else
{
CDialog::OnPaint();
}
}
// The system calls this function to obtain the cursor to display while the user drags
// the minimized window.
HCURSOR CMemSpyyDlg::OnQueryDragIcon()
{
return static_cast<HCURSOR>(m_hIcon);
}
void CMemSpyyDlg::SetLargestFree( unsigned long inLargestFree )
{
char msg[255];
sprintf( msg, "%4.2fMB", (double) ( (double) inLargestFree / 1024.0 / 1024.0) );
m_LargestCtl.SetWindowText( msg );
}
void CMemSpyyDlg::SetTotalFree( unsigned long inTotalFree )
{
char msg[255];
sprintf( msg, "%4.2fMB", (double) ((double)inTotalFree / 1024.0 / 1024.0 ) );
m_TotalFreeCtl.SetWindowText( msg );
}
void CMemSpyyDlg::SetTotalDLL( unsigned long inTotalDLL )
{
char msg[255];
sprintf( msg, "%4.2fMB", (double) ( (double)inTotalDLL / 1024.0 / 1024.0 ) );
m_TotalDllCtl.SetWindowText( msg );
}
void CMemSpyyDlg::OnTimer(UINT nIDEvent)
{
m_OpenGLControl.ForceDraw();
CDialog::OnTimer(nIDEvent);
}
void CMemSpyyDlg::OnEnUpdatePid()
{
// TODO: If this is a RICHEDIT control, the control will not
// send this notification unless you override the CDialog::OnInitDialog()
// function to send the EM_SETEVENTMASK message to the control
// with the ENM_UPDATE flag ORed into the lParam mask.
UpdateData();
m_OpenGLControl.SetPID( m_PID );
m_PIDNameCtl.SetWindowText( GetEXENameByPID( m_PID ) );
}
CString CMemSpyyDlg::GetEXENameByPID( DWORD inPID )
{
char pathName[255];
HANDLE process = OpenProcess( PROCESS_QUERY_INFORMATION|PROCESS_VM_READ,false, inPID );
GetProcessImageFileName( process, pathName, 255);
CString str = CString(pathName);
CString strExeName = str.Right( str.GetLength() - str.ReverseFind( '\\') - 1 );
return strExeName;
}
void CMemSpyyDlg::UpdateStatus( PVOID inBaseAddress )
{
// Get a handle to the process.
HANDLE process = OpenProcess( PROCESS_QUERY_INFORMATION|PROCESS_VM_READ,false, m_PID );
SIZE_T retVal = 0;
MEMORY_BASIC_INFORMATION memBlock;
retVal = VirtualQueryEx(process, inBaseAddress,&memBlock,sizeof(memBlock));
if (retVal)
{
if (memBlock.State & MEM_FREE)
{
char tmpBuffer[255];
sprintf( tmpBuffer, "Base: 0x%p Size: %4.2fMB Description: Free", inBaseAddress, (double)memBlock.RegionSize/1024/1024);
m_StatusCtl.SetWindowText( tmpBuffer );
}
else
{
char mappedFile[255];
char tmpBuffer[255];
if ( GetMappedFileName( process, inBaseAddress, mappedFile, 255) )
{
sprintf( tmpBuffer, "Base: 0x%p Size: %4.2fMB Description: %s", inBaseAddress, (double) memBlock.RegionSize/1024/1024, strrchr(mappedFile, '\\') + 1 );
m_StatusCtl.SetWindowText( tmpBuffer );
}
else
{
sprintf( tmpBuffer, "Base: 0x%p Size: %4.2fMB Description: committed", inBaseAddress, (double) memBlock.RegionSize/1024/1024 );
m_StatusCtl.SetWindowText( tmpBuffer );
}
}
}
}