www.pudn.com > 基于SHA-256的HMAC文件校验器.rar > VeriFileView.cpp
// VeriFileView.cpp : implementation of the CVeriFileView class
// Programmed by Gamsn 2005.10.22
#include "stdafx.h"
#include "VeriFile.h"
#include "VeriFileDoc.h"
#include "VeriFileView.h"
#include "InputKey.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
bool g_bIsStop=0;
bool g_bIsPro=0;
/////////////////////////////////////////////////////////////////////////////
// CVeriFileView
IMPLEMENT_DYNCREATE(CVeriFileView, CFormView)
BEGIN_MESSAGE_MAP(CVeriFileView, CFormView)
//{{AFX_MSG_MAP(CVeriFileView)
ON_COMMAND(ID_M_OPEN, OnMOpen)
ON_WM_TIMER()
ON_BN_CLICKED(IDC_CHECK_READONLY, OnCheckReadonly)
ON_BN_CLICKED(IDC_CHECK_HIDDEN, OnCheckHidden)
ON_BN_CLICKED(IDC_CHECK_SYS, OnCheckSys)
ON_BN_CLICKED(IDC_CHECK_ARCHIVE, OnCheckArchive)
ON_COMMAND(ID_M_VERI, OnMVeri)
ON_WM_DROPFILES()
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CVeriFileView construction/destruction
CVeriFileView::CVeriFileView()
: CFormView(CVeriFileView::IDD)
{
//{{AFX_DATA_INIT(CVeriFileView)
m_strFileName = _T("");
m_nFileSize = 0;
m_strCreatDate = _T("");
m_strLastModiDate = _T("");
m_strLastAccessDate = _T("");
m_strHMac = _T("");
//}}AFX_DATA_INIT
// TODO: add construction code here
}
CVeriFileView::~CVeriFileView()
{
}
void CVeriFileView::DoDataExchange(CDataExchange* pDX)
{
CFormView::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CVeriFileView)
DDX_Control(pDX, IDC_CHECK_ARCHIVE, m_chk_archive);
DDX_Control(pDX, IDC_CHECK_HIDDEN, m_chk_hidden);
DDX_Control(pDX, IDC_CHECK_READONLY, m_chk_readonly);
DDX_Control(pDX, IDC_CHECK_SYS, m_chk_sys);
DDX_Text(pDX, IDC_EDIT_FILENAME, m_strFileName);
DDX_Text(pDX, IDC_EDIT_FILESIZE, m_nFileSize);
DDX_Text(pDX, IDC_EDIT_CREATDATE, m_strCreatDate);
DDX_Text(pDX, IDC_EDIT_LASTMODIDATE, m_strLastModiDate);
DDX_Text(pDX, IDC_EDIT_LASTACCESSDATE, m_strLastAccessDate);
DDX_Text(pDX, IDC_EDIT_HMAC, m_strHMac);
//}}AFX_DATA_MAP
}
BOOL CVeriFileView::PreCreateWindow(CREATESTRUCT& cs)
{
// TODO: Modify the Window class or styles here by modifying
// the CREATESTRUCT cs
return CFormView::PreCreateWindow(cs);
}
void CVeriFileView::OnInitialUpdate()
{
CFormView::OnInitialUpdate();
GetParentFrame()->RecalcLayout();
ResizeParentToFit();
pDoc=GetDocument();
m_Mac.m_nArchive=pDoc->m_nArchive;
m_Mac.m_nFileSize=pDoc->m_nFileSize;
m_Mac.m_nHidden=pDoc->m_nHidden;
m_Mac.m_nReadOnly=pDoc->m_nReadOnly;
m_Mac.m_nSys=pDoc->m_nSys;
m_Mac.m_strFileCreateDate=pDoc->m_strFileCreateDate;
m_Mac.m_strFileLastModiDate=pDoc->m_strFileLastModiDate;
m_Mac.m_strFileLastAccessDate=pDoc->m_strFileLastAccessDate;
m_Mac.m_strFileName=pDoc->m_strFileName;
m_Mac.m_strMac=pDoc->m_strMac;
pDoc->m_pMac=&m_Mac;
ShowMsg();
}
/////////////////////////////////////////////////////////////////////////////
// CVeriFileView diagnostics
#ifdef _DEBUG
void CVeriFileView::AssertValid() const
{
CFormView::AssertValid();
}
void CVeriFileView::Dump(CDumpContext& dc) const
{
CFormView::Dump(dc);
}
CVeriFileDoc* CVeriFileView::GetDocument() // non-debug version is inline
{
ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CVeriFileDoc)));
return (CVeriFileDoc*)m_pDocument;
}
#endif //_DEBUG
/////////////////////////////////////////////////////////////////////////////
// CVeriFileView message handlers
UINT ThreadsProc(LPVOID pParam)
{
CWnd *pMainWnd=AfxGetMainWnd();
if(pParam==NULL)
{
::MessageBox(pMainWnd->m_hWnd,"系统参数传递错误!","AESer",MB_OK|MB_ICONEXCLAMATION);
AfxEndThread(0);
}
CVeriFileView *pDlg=(CVeriFileView *)pParam;
pDlg->m_HMac.PreCalcu(pDlg->m_strKey);
pDlg->m_HMac.GenMac(pDlg->m_Mac.m_strFileName,&(pDlg->m_Mac.m_strMac));
g_bIsStop=1;
g_bIsPro=0;
return 0;
}
void CVeriFileView::OnMOpen()
{
if(g_bIsPro) return;
CFileDialog dlg(1,NULL,NULL,OFN_ENABLESIZING|OFN_EXPLORER,"全部 (*.*)|*.*||",this);
dlg.m_ofn.lpstrTitle="请选择要被校验的文件";
if(dlg.DoModal()==IDOK)
{
CString strSrcPath=dlg.GetPathName();
CreateVeri(strSrcPath);
}
}
void CVeriFileView::ShowMsg()
{
if(m_Mac.m_nSys==1) m_chk_sys.SetCheck(1);
else m_chk_sys.SetCheck(0);
if(m_Mac.m_nReadOnly==1) m_chk_readonly.SetCheck(1);
else m_chk_readonly.SetCheck(0);
if(m_Mac.m_nHidden==1) m_chk_hidden.SetCheck(1);
else m_chk_hidden.SetCheck(0);
if(m_Mac.m_nArchive==1) m_chk_archive.SetCheck(1);
else m_chk_archive.SetCheck(0);
m_nFileSize=m_Mac.m_nFileSize;
m_strLastModiDate=m_Mac.m_strFileLastModiDate;
m_strCreatDate=m_Mac.m_strFileCreateDate;
m_strLastAccessDate=m_Mac.m_strFileLastAccessDate;
m_strFileName=m_Mac.m_strFileName;
m_strHMac=m_Mac.m_strMac;
UpdateData(0);
}
void CVeriFileView::OnTimer(UINT nIDEvent)
{
if(nIDEvent==0)
{
if(g_bIsStop)
{
KillTimer(0);
ShowMsg();
}
}
else if(nIDEvent==1)
{
if(g_bIsStop)
{
KillTimer(1);
UpdateData(1);
if(m_Mac.m_strMac!=m_strHMac)
{
MessageBox("您的文件没有通过校验，\n\n该文件很可能被非法修改，或是您输入的校验密钥不对，\n\n或是您的校验文件被非法修改过。","VERIFILE",MB_OK|MB_ICONINFORMATION);
Restore();
}
else MessageBox("恭喜！您的文件通过校验，它未经非法修改过。","VERIFILE",MB_OK|MB_ICONINFORMATION);
}
}
if(nIDEvent==2)
{
if(!g_bIsPro)
{
m_WaitDlg->DestroyWindow();
KillTimer(2);
}
}
CFormView::OnTimer(nIDEvent);
}
void CVeriFileView::OnCheckReadonly()
{
if(m_chk_readonly.GetCheck()==1) m_chk_readonly.SetCheck(0);
else m_chk_readonly.SetCheck(1);
}
void CVeriFileView::OnCheckHidden()
{
if(m_chk_hidden.GetCheck()==1) m_chk_hidden.SetCheck(0);
else m_chk_hidden.SetCheck(1);
}
void CVeriFileView::OnCheckSys()
{
if(m_chk_sys.GetCheck()==1) m_chk_sys.SetCheck(0);
else m_chk_sys.SetCheck(1);
}
void CVeriFileView::OnCheckArchive()
{
if(m_chk_archive.GetCheck()==1) m_chk_archive.SetCheck(0);
else m_chk_archive.SetCheck(1);
}
void CVeriFileView::OnMVeri()
{
if(g_bIsPro) return;
if(m_Mac.m_strMac.IsEmpty())
{
MessageBox("当前没有有效的校验文件，请先打开校验文件，再选择相应的源文件进行校验",
"VERIFILE",MB_OK|MB_ICONEXCLAMATION);
}
else
{
CFileDialog dlg(1,NULL,NULL,OFN_ENABLESIZING|OFN_EXPLORER,"全部 (*.*)|*.*||",this);
dlg.m_ofn.lpstrTitle="请选择要被校验的文件";
if(dlg.DoModal()==IDOK)
{
CString strSrcPath=dlg.GetPathName();
m_Mac.m_strFileName=strSrcPath;
CInputKey Dlg;
if(Dlg.DoModal()==IDOK)
{
m_strKey=Dlg.m_strKey;
m_Mac.m_strMac.Empty();
g_bIsStop=0;
g_bIsPro=1;
SetTimer(1,1000,NULL);
SetTimer(2,1000,NULL);
CRect rect;
GetWindowRect(rect);
m_WaitDlg=new CWaitDlg;
m_WaitDlg->Create(IDD_DIALOG_WAIT);
m_WaitDlg->SetWindowPos(NULL,(rect.left+rect.right)/2-100,(rect.top+rect.bottom)/2-50,143,70,SWP_SHOWWINDOW);
m_WaitDlg->ShowWindow(SW_SHOW);
AfxBeginThread(&ThreadsProc,this);
}
}
}
}
void CVeriFileView::Restore()
{
UpdateData(1);
m_Mac.m_strFileName=m_strFileName;
m_Mac.m_strMac=m_strHMac;
}
void CVeriFileView::OnDropFiles(HDROP hDropInfo)
{
CString strFileName;
char buffer[255];
int nRes=0,nAll=-1;
nAll=DragQueryFile(hDropInfo,0xFFFFFFFF,buffer,10);
if(nAll>=0)
{
nRes=DragQueryFile(hDropInfo,0,buffer,255);
strFileName.Insert(0,buffer);
if(strFileName.IsEmpty()) return;
if(strFileName.Right(4)==".vri")
{
CFile f;
CFileException e;
f.Open(strFileName,CFile::modeRead|CFile::typeBinary,&e);
if(e.m_cause!=CFileException::none)
{
MessageBox("打开文件失败!","HMac",MB_OK|MB_ICONEXCLAMATION);
return;
}
CArchive ar(&f,CArchive::load);
ar>>m_Mac.m_nArchive;
ar>>m_Mac.m_nHidden;
ar>>m_Mac.m_nReadOnly;
ar>>m_Mac.m_nSys;
ar>>m_Mac.m_nFileSize;
ar>>m_Mac.m_strFileCreateDate;
ar>>m_Mac.m_strFileLastModiDate;
ar>>m_Mac.m_strFileLastAccessDate;
ar>>m_Mac.m_strFileName;
ar>>m_Mac.m_strMac;
ShowMsg();
}
else
{
int nRes=MessageBox("请选择：\n\n是：生成该文件的校验文件；\n\n否：校验该文件；\n\n取消：什么也不做","VERIFILE",MB_YESNOCANCEL);
if(nRes==IDYES) CreateVeri(strFileName);
else if(nRes==IDNO)
{
m_Mac.m_strFileName=strFileName;
CInputKey Dlg;
if(Dlg.DoModal()==IDOK)
{
m_strKey=Dlg.m_strKey;
m_Mac.m_strMac.Empty();
g_bIsStop=0;
g_bIsPro=1;
SetTimer(1,1000,NULL);
SetTimer(2,1000,NULL);
CRect rect;
GetWindowRect(rect);
m_WaitDlg=new CWaitDlg;
m_WaitDlg->Create(IDD_DIALOG_WAIT);
m_WaitDlg->SetWindowPos(NULL,(rect.left+rect.right)/2-100,(rect.top+rect.bottom)/2-50,143,70,SWP_SHOWWINDOW);
m_WaitDlg->ShowWindow(SW_SHOW);
AfxBeginThread(&ThreadsProc,this);
}
}
}
}
CFormView::OnDropFiles(hDropInfo);
}
void CVeriFileView::CreateVeri(CString strSrcPath)
{
CFileStatus rStatus;
if(CFile::GetStatus(strSrcPath,rStatus))
{
m_Mac.m_nFileSize=rStatus.m_size;
m_Mac.m_strFileName=strSrcPath;
m_Mac.m_strFileCreateDate=rStatus.m_ctime.Format("%Y-%m-%d %H:%M:%S");
m_Mac.m_strFileLastModiDate=rStatus.m_mtime.Format("%Y-%m-%d %H:%M:%S");
m_Mac.m_strFileLastAccessDate=rStatus.m_atime.Format("%Y-%m-%d %H:%M:%S");
if((rStatus.m_attribute & 0x01)==0x01) m_Mac.m_nReadOnly=1;
else m_Mac.m_nReadOnly=0;
if((rStatus.m_attribute & 0x02)==0x02) m_Mac.m_nHidden=1;
else m_Mac.m_nHidden=0;
if((rStatus.m_attribute & 0x20)==0x20) m_Mac.m_nArchive=1;
else m_Mac.m_nArchive=0;
if((rStatus.m_attribute & 0x04)==0x04) m_Mac.m_nSys=1;
else m_Mac.m_nSys=0;
}
else
{
MessageBox("无法得到文相关属性，一定要生成校验文件吗？","VERIFILE",MB_YESNO|MB_ICONQUESTION);
}
CInputKey Dlg;
if(Dlg.DoModal()==IDOK)
{
m_strKey=Dlg.m_strKey;
g_bIsStop=0;
g_bIsPro=1;
SetTimer(0,1000,NULL);
SetTimer(2,1000,NULL);
m_Mac.m_strMac.Empty();
pDoc->SetModifiedFlag(1);
CRect rect;
GetWindowRect(rect);
m_WaitDlg=new CWaitDlg;
m_WaitDlg->Create(IDD_DIALOG_WAIT);
m_WaitDlg->SetWindowPos(NULL,(rect.left+rect.right)/2-100,(rect.top+rect.bottom)/2-50,143,70,SWP_SHOWWINDOW);
m_WaitDlg->ShowWindow(SW_SHOW);
AfxBeginThread(&ThreadsProc,this);
}
}