DemonMemory/源代码/HandleMemoryDlg.cpp

// HandleMemoryDlg.cpp : ʵ<><CAB5><EFBFBD>ļ<EFBFBD>
//

#include "stdafx.h"
#include "HandleMemory.h"
#include "HandleMemoryDlg.h"
#include "afxdialogex.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// <20><><EFBFBD><EFBFBD>Ӧ<EFBFBD>ó<EFBFBD><C3B3>򡰹<EFBFBD><F2A1B0B9>ڡ<EFBFBD><DAA1>˵<EFBFBD><CBB5><EFBFBD><EFBFBD><EFBFBD> CAboutDlg <20>Ի<EFBFBD><D4BB><EFBFBD>

class CAboutDlg : public CDialogEx
{
public:
	CAboutDlg();

// <20>Ի<EFBFBD><D4BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
	enum { IDD = IDD_ABOUTBOX };

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV ֧<><D6A7>

// ʵ<><CAB5>
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialogEx(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()


// CHandleMemoryDlg <20>Ի<EFBFBD><D4BB><EFBFBD>



CHandleMemoryDlg::CHandleMemoryDlg(CWnd* pParent /*=NULL*/)
	: CDialogEx(CHandleMemoryDlg::IDD, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CHandleMemoryDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_TAB_MAIN, m_TabMain);
}

BEGIN_MESSAGE_MAP(CHandleMemoryDlg, CDialogEx)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_NOTIFY(TCN_SELCHANGE, IDC_TAB_MAIN, &CHandleMemoryDlg::OnSelchangeTabMain)
	ON_COMMAND(ID_SYSTEM_VIEW, &CHandleMemoryDlg::OnSystemView)
	ON_COMMAND(ID_SYSTEM_INJECT, &CHandleMemoryDlg::OnSystemInject)
	ON_COMMAND(ID_HELP_ABOUT, &CHandleMemoryDlg::OnHelpAbout)
	ON_COMMAND(ID_SYSTEM_EXIT, &CHandleMemoryDlg::OnSystemExit)
END_MESSAGE_MAP()


// CHandleMemoryDlg <20><>Ϣ<EFBFBD><CFA2><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>

BOOL EnablePriv()
{  
	HANDLE hToken;
    if(::OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken))
	{
		TOKEN_PRIVILEGES tkp;
		::LookupPrivilegeValue(NULL, SE_DEBUG_NAME, &tkp.Privileges[0].Luid);//<2F>޸Ľ<DEB8><C4BD><EFBFBD>Ȩ<EFBFBD><C8A8>
		tkp.PrivilegeCount = 1;
		tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
		::AdjustTokenPrivileges(hToken, FALSE,&tkp, sizeof tkp, NULL, NULL);//֪ͨϵͳ<CFB5>޸Ľ<DEB8><C4BD><EFBFBD>Ȩ<EFBFBD><C8A8>
		return ((::GetLastError() == ERROR_SUCCESS));
	} 
	return FALSE;
}

BOOL CHandleMemoryDlg::OnInitDialog()
{
	CDialogEx::OnInitDialog();

	// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>...<2E><><EFBFBD>˵<EFBFBD><CBB5><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ӵ<EFBFBD>ϵͳ<CFB5>˵<EFBFBD><CBB5>С<EFBFBD>

	// IDM_ABOUTBOX <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ϵͳ<CFB5><CDB3><EFBFBD>Χ<EEB7B6>ڡ<EFBFBD>
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		BOOL bNameValid;
		CString strAboutMenu;
		bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
		ASSERT(bNameValid);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// <20><><EFBFBD>ô˶Ի<CBB6><D4BB><EFBFBD><EFBFBD><EFBFBD>ͼ<EFBFBD>ꡣ<EFBFBD><EAA1A3>Ӧ<EFBFBD>ó<EFBFBD><C3B3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ڲ<EFBFBD><DAB2>ǶԻ<C7B6><D4BB><EFBFBD>ʱ<EFBFBD><CAB1><EFBFBD><EFBFBD><EFBFBD>ܽ<EFBFBD><DCBD>Զ<EFBFBD>
	//  ִ<>д˲<D0B4><CBB2><EFBFBD>
	SetIcon(m_hIcon, TRUE);			// <20><><EFBFBD>ô<EFBFBD>ͼ<EFBFBD><CDBC>
	SetIcon(m_hIcon, FALSE);		// <20><><EFBFBD><EFBFBD>Сͼ<D0A1><CDBC>

	// TODO: <20>ڴ<EFBFBD><DAB4><EFBFBD><EFBFBD>Ӷ<EFBFBD><D3B6><EFBFBD><EFBFBD>ij<EFBFBD>ʼ<EFBFBD><CABC><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
	// <20><>Ȩ -- <20>ز<EFBFBD><D8B2><EFBFBD><EFBFBD>١<EFBFBD><D9A1><EFBFBD><EFBFBD>ɻ<EFBFBD>ȱ
	if(!EnablePriv())
	{
		::MessageBox(m_hWnd, "Please Enable Privilege!", "Warn", MB_OK | MB_ICONWARNING);
	}
	InitTabControl();               // <20><>ʼ<EFBFBD><CABC>TAB<41>ؼ<EFBFBD>

	return TRUE;  // <20><><EFBFBD>ǽ<EFBFBD><C7BD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>õ<EFBFBD><C3B5>ؼ<EFBFBD><D8BC><EFBFBD><EFBFBD><EFBFBD><EFBFBD>򷵻<EFBFBD> TRUE
}

void CHandleMemoryDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialogEx::OnSysCommand(nID, lParam);
	}
}

// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ի<EFBFBD><D4BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>С<EFBFBD><D0A1><EFBFBD><EFBFBD>ť<EFBFBD><C5A5><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ҫ<EFBFBD><D2AA><EFBFBD><EFBFBD><EFBFBD>Ĵ<EFBFBD><C4B4><EFBFBD>
//  <20><><EFBFBD><EFBFBD><EFBFBD>Ƹ<EFBFBD>ͼ<EFBFBD>ꡣ<EFBFBD><EAA1A3><EFBFBD><EFBFBD>ʹ<EFBFBD><CAB9><EFBFBD>ĵ<EFBFBD>/<2F><>ͼģ<CDBC>͵<EFBFBD> MFC Ӧ<>ó<EFBFBD><C3B3><EFBFBD><EFBFBD><EFBFBD>
//  <20>⽫<EFBFBD>ɿ<EFBFBD><C9BF><EFBFBD><EFBFBD>Զ<EFBFBD><D4B6><EFBFBD><EFBFBD>ɡ<EFBFBD>

void CHandleMemoryDlg::OnPaint()
{
	if (IsIconic())
	{
		CPaintDC dc(this); // <20><><EFBFBD>ڻ<EFBFBD><DABB>Ƶ<EFBFBD><C6B5>豸<EFBFBD><E8B1B8><EFBFBD><EFBFBD><EFBFBD><EFBFBD>

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// ʹͼ<CAB9><CDBC><EFBFBD>ڹ<EFBFBD><DAB9><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>о<EFBFBD><D0BE><EFBFBD>
		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;

		// <20><><EFBFBD><EFBFBD>ͼ<EFBFBD><CDBC>
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialogEx::OnPaint();
	}
}

//<2F><><EFBFBD>û<EFBFBD><C3BB>϶<EFBFBD><CFB6><EFBFBD>С<EFBFBD><D0A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʱϵͳ<CFB5><CDB3><EFBFBD>ô˺<C3B4><CBBA><EFBFBD>ȡ<EFBFBD>ù<EFBFBD><C3B9><EFBFBD>
//<2F><>ʾ<EFBFBD><CABE>
HCURSOR CHandleMemoryDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}


void CHandleMemoryDlg::OnSystemView()
{
	// TODO: <20>ڴ<EFBFBD><DAB4><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EEB4A6><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
	NMHDR nmhdr;
	nmhdr.code = TCN_SELCHANGE;
	nmhdr.hwndFrom = ::GetDlgItem(m_hWnd, IDC_TAB_MAIN);
	nmhdr.idFrom = IDC_TAB_MAIN;
	m_TabMain.SetCurSel(0);
	::SendMessage(m_hWnd, WM_NOTIFY, MAKELONG(TCN_SELCHANGE, 0), (LPARAM)(&nmhdr));
}


void CHandleMemoryDlg::OnSystemInject()
{
	// TODO: <20>ڴ<EFBFBD><DAB4><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EEB4A6><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
	NMHDR nmhdr;
	nmhdr.code = TCN_SELCHANGE;
	nmhdr.hwndFrom = ::GetDlgItem(m_hWnd, IDC_TAB_MAIN);
	nmhdr.idFrom = IDC_TAB_MAIN;
	m_TabMain.SetCurSel(1);
	::SendMessage(m_hWnd, WM_NOTIFY, MAKELONG(TCN_SELCHANGE, 0), (LPARAM)(&nmhdr));
}


void CHandleMemoryDlg::OnHelpAbout()
{
	// TODO: <20>ڴ<EFBFBD><DAB4><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EEB4A6><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
	CAboutDlg dlg;
	dlg.DoModal();
}


void CHandleMemoryDlg::OnSystemExit()
{
	// TODO: <20>ڴ<EFBFBD><DAB4><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EEB4A6><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
	if(IDYES == ::MessageBox(m_hWnd, "Are you sure exit ?", "Confirm", MB_YESNO | MB_ICONWARNING))
	{
		exit(0);
	}
}


BOOL CHandleMemoryDlg::InitTabControl()
{
	// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ǩ
	m_TabMain.InsertItem(0, "   View     ");
	m_TabMain.InsertItem(1, "   Inject     ");
	// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ի<EFBFBD><D4BB><EFBFBD>
	m_ViewDlg.Create(IDD_DIALOG_VIEW, GetDlgItem(IDC_TAB_MAIN));
	m_InjectDlg.Create(IDD_DIALOG_INJECT, GetDlgItem(IDC_TAB_MAIN));
	// <20><>ȡIDC_TAB<41>ͻ<EFBFBD><CDBB><EFBFBD><EFBFBD>Ĵ<EFBFBD>С
	CRect rc;
	m_TabMain.GetClientRect(&rc);
	// <20><><EFBFBD><EFBFBD><EFBFBD>ӶԻ<D3B6><D4BB><EFBFBD><EFBFBD>ڸ<EFBFBD><DAB8>Ի<EFBFBD><D4BB><EFBFBD><EFBFBD>е<EFBFBD>λ<EFBFBD><CEBB>
	rc.top+=20;
	rc.bottom-=1;
	rc.left+=1;
	rc.right-=1;
	// <20>ƶ<EFBFBD><C6B6>ӶԻ<D3B6><D4BB><EFBFBD><EFBFBD><EFBFBD>ָ<EFBFBD><D6B8>λ<EFBFBD><CEBB>
	m_ViewDlg.MoveWindow(&rc);
	m_InjectDlg.MoveWindow(&rc);
	// <20>ֱ<EFBFBD><D6B1><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>غ<EFBFBD><D8BA><EFBFBD>ʾ
	m_ViewDlg.ShowWindow(SW_SHOW);
	m_InjectDlg.ShowWindow(SW_HIDE);
	// <20><><EFBFBD><EFBFBD>Ĭ<EFBFBD>ϵ<EFBFBD>ѡ<EFBFBD>
	m_TabMain.SetCurSel(0);

	return TRUE;
}

void CHandleMemoryDlg::OnSelchangeTabMain(NMHDR *pNMHDR, LRESULT *pResult)
{
	// TODO: <20>ڴ<EFBFBD><DAB4><EFBFBD><EFBFBD>ӿؼ<D3BF>֪ͨ<CDA8><D6AA><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
	int iCurSel = m_TabMain.GetCurSel();
	switch(iCurSel)
	{
	case 0:
		{
			m_ViewDlg.ShowWindow(SW_SHOW);
			m_InjectDlg.ShowWindow(SW_HIDE);
			break;
		}
	case 1:
		{
			m_ViewDlg.ShowWindow(SW_HIDE);
			m_InjectDlg.ShowWindow(SW_SHOW);
			break;
		}
	default:
		break;
	}

	*pResult = 0;
}