mainfrm.cpp

基于Triple DES的文件加密,可以对任意文件进行加密

// MainFrm.cpp : implementation of the CMainFrame class
//

#include "stdafx.h"
#include "FES.h"

#include "MainFrm.h"
#include "FileEncrypt.h"
#include "TripleDes.h"      
#include "SetDlg.h"     
   
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
CProgressCtrl m_progress;//进度条全局变量
CString fileInfo[4];
char key[16];

/////////////////////////////////////////////////////////////////////////////
// CMainFrame

IMPLEMENT_DYNCREATE(CMainFrame, CFrameWnd)

BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
	//{{AFX_MSG_MAP(CMainFrame)
	ON_WM_CREATE()
	ON_COMMAND(ID_TOOL_ENCRYPT, OnToolEncrypt)
	ON_COMMAND(ID_TOOL_DECRYPT, OnToolDecrypt)
	ON_WM_PAINT()
	ON_COMMAND(ID_FILE_OPEN, OnFileOpen)
	ON_UPDATE_COMMAND_UI(ID_TOOL_ENCRYPT, OnUpdateToolEncrypt)
	ON_UPDATE_COMMAND_UI(ID_TOOL_DECRYPT, OnUpdateToolDecrypt)
	//}}AFX_MSG_MAP
	
END_MESSAGE_MAP()

static UINT indicators[] =
{
	ID_SEPARATOR, // status line indicator
	ID_INDICATOR_PROGRESS,
	ID_INDICATOR_CAPS,
	ID_INDICATOR_NUM,
	ID_INDICATOR_SCRL,
};

/////////////////////////////////////////////////////////////////////////////
// CMainFrame construction/destruction

CMainFrame::CMainFrame()
{
	// TODO: add member initialization code here
	fileIn=NULL;
}

CMainFrame::~CMainFrame()
{
}

int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
	if (CFrameWnd::OnCreate(lpCreateStruct) == -1)
		return -1;
	
	if (!m_wndToolBar.CreateEx(this, TBSTYLE_FLAT, WS_CHILD | WS_VISIBLE | CBRS_TOP
		| CBRS_GRIPPER | CBRS_TOOLTIPS | CBRS_FLYBY | CBRS_SIZE_DYNAMIC) ||
		!m_wndToolBar.LoadToolBar(IDR_MAINFRAME))
	{
		TRACE0("Failed to create toolbar\n");
		return -1;      // fail to create
	}

	if (!m_wndStatusBar.Create(this) ||
		!m_wndStatusBar.SetIndicators(indicators,
		  sizeof(indicators)/sizeof(UINT)))
	{
		TRACE0("Failed to create status bar\n");
		return -1;      // fail to create
	}

	// TODO: Delete these three lines if you don't want the toolbar to
	//  be dockable
	m_wndToolBar.EnableDocking(CBRS_ALIGN_ANY);
	EnableDocking(CBRS_ALIGN_ANY);
	DockControlBar(&m_wndToolBar);
    
   



	return 0;
}

BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
{
	if( !CFrameWnd::PreCreateWindow(cs) )
		return FALSE;
	// TODO: Modify the Window class or styles here by modifying
	//  the CREATESTRUCT cs
   cs.style&=~FWS_ADDTOTITLE;
   cs.lpszName="FES v1.0";
   cs.cx=600;
   cs.cy=300;
   cs.x=220;
   cs.y=220;
   
   return TRUE;
}

/////////////////////////////////////////////////////////////////////////////
// CMainFrame diagnostics

#ifdef _DEBUG
void CMainFrame::AssertValid() const
{
	CFrameWnd::AssertValid();
}

void CMainFrame::Dump(CDumpContext& dc) const
{
	CFrameWnd::Dump(dc);
}

#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CMainFrame message handlers

void CMainFrame::OnToolEncrypt() 
{
	// TODO: Add your command handler code here
	
    if(fileIn)
	{
		if(key[0]==0){
          CSetDlg dlg;
		  dlg.DoModal();
		}
		if(key[0]!=0){
		FileEncrypt::HandleFile(fileIn,key,ENCRYPT);
         fileIn=NULL;//将文件指针设置为NULL
	      MessageBox("加密成功!","完成",MB_OK);
		}     
	}
	
}

void CMainFrame::OnToolDecrypt() 
{
	// TODO: Add your command handler code here
 
    if(fileIn)
    {  
		if(key[0]==0){
		  CSetDlg dlg;
		  dlg.DoModal();
		}
        if(key[0]!=0){  
	      FileEncrypt::HandleFile(fileIn,key,DECRYPT);
	      fileIn=NULL;//将文件指针设置为NULL
          MessageBox("解密成功!","完成",MB_OK);
		}
	}
}

void CMainFrame::OnUpdateToolEncrypt(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
	
      if(!fileIn  || fileInfo[3]=="des")//文件为空或者后缀名为des时使加密按钮失效
          pCmdUI->Enable(false);  
      else 
          pCmdUI->Enable(true);
}

void CMainFrame::OnUpdateToolDecrypt(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
    
       if(fileIn && fileInfo[3]=="des")//后缀名为des时使解密按钮有效
          pCmdUI->Enable(true); 
	   else
          pCmdUI->Enable(false);  
}

void CMainFrame::OnPaint() 
{
	CPaintDC dc(this); // device context for painting
	
	// TODO: Add your message handler code here
	
	// Do not call CFrameWnd::OnPaint() for painting messages
    
	CRect rect;
	m_wndStatusBar.SetPaneInfo(1,NULL,NULL,600);//设置进度栏的宽度
	m_wndStatusBar.GetItemRect(1,&rect);//获得进度栏矩形大小
	
	if(!m_progress.m_hWnd)
      m_progress.Create(WS_CHILD | WS_VISIBLE | PBS_SMOOTH,
	                    rect, &m_wndStatusBar, ID_INDICATOR_PROGRESS);
    else
	  m_progress.MoveWindow(rect);
 
}



void CMainFrame::OnFileOpen() 
{
	// TODO: Add your command handler code here

    char sFilter[]="加密文件 (*.des)|*.des|文本 (*.txt)|*.txt|图片 (*.jpg;*.gif)|*.jpg;*.gif|压缩文件 (*.rar;*.zip)|*.rar; *.zip|所有文件 (*.*)|*.*||";


   CFileDialog fileDlg(TRUE,NULL,NULL,OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,sFilter,this);
	if(fileDlg.DoModal()==IDOK)
	{
		 
	 CString  fileName=fileDlg.GetPathName();

	 try{
	  if(fileIn==NULL)
       fileIn=new CFile(fileName,CFile::modeRead );
	          
	          fileInfo[0]=fileDlg.GetFileName();
			  fileInfo[1].Format("%d字节",fileIn->GetLength());
              fileInfo[2]=fileName;
	          fileInfo[3]=fileDlg.GetFileExt();

			  Invalidate(false); //让窗口重绘  
              UpdateWindow();   
	 }
	 catch(CFileException ex){
	  ex.ReportError();
	 }
	 
	 }  
}