Grafting Compiled Code: Unlimited Code Reuse

// PEChecksum.h : header file
//

#pragma once


// CPEChecksum dialog
class CPEChecksum: public CDialog
{
    // Construction
public:
    CPEChecksum(CWnd* pParent = NULL); // standard constructor

    // Dialog Data
    enum{ IDD = IDD_PECHECKSUM_DIALOG };

protected:
    virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support


    // Implementation
protected:
    HICON m_hIcon;

    // Generated message map functions
    virtual BOOL OnInitDialog();
    afx_msg void OnPaint();
    afx_msg HCURSOR OnQueryDragIcon();
    DECLARE_MESSAGE_MAP()
public:
    afx_msg void OnBnClickedRetrieveChecksum();
    afx_msg void OnBnClickedBrowse();

protected:
    BOOL SelectFile( CString& );
    BOOL RetrieveFilename( CString& );
    BOOL RetrieveAndCalculateChecksum( const CString&, DWORD32&, DWORD32& );
    BOOL RetrieveNewChecksum( DWORD32& );
    BOOL CalculateChecksum( const CString&, DWORD32&, DWORD32& );
    BOOL WriteChecksum( DWORD32& );

    BOOL MapFile( const CString& szFilename );
    BOOL UnmapFile( );

    VOID DisplayError( DWORD dwError );
    VOID DisplayWarning( DWORD dwError );

protected:
    HANDLE hFile;
    HANDLE hFileMapping;
    PVOID pBaseAddress;

public:
    afx_msg void OnBnClickedBtnWriteChecksum();
};