encryptfile.cpp

采用DES加密算法对文件内容进行加密

// encryptfile.cpp : implementation file
//
#include <fstream.h>
#include "stdafx.h"
#include "encrypt.h"
#include "encryptfile.h"
#include "des.h"
#include<windows.h>

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
/////////////////////////////////////////////////////////////////////////////
// Cencryptfile dialog

Cencryptfile::Cencryptfile(CWnd* pParent /*=NULL*/)
	: CDialog(Cencryptfile::IDD, pParent)
{
	//{{AFX_DATA_INIT(Cencryptfile)
	m_filename = _T("");
	m_password = _T("");
	//}}AFX_DATA_INIT
}


void Cencryptfile::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(Cencryptfile)
	DDX_Control(pDX, IDC_PROGRESS1, m_progress);
	DDX_Text(pDX, IDC_EDIT_FILENAME, m_filename);
	DDX_Text(pDX, IDC_EDIT_PASSWORD, m_password);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(Cencryptfile, CDialog)
	//{{AFX_MSG_MAP(Cencryptfile)
	ON_BN_CLICKED(IDC_BUTTON_BEGIN, OnButtonBegin)
	ON_BN_CLICKED(IDC_BUTTON_SCAN, OnButtonScan)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// Cencryptfile message handlers
void Cencryptfile::OnButtonBegin() 
{

}

void Cencryptfile::OnOK() 
{
	GetDlgItem(IDCANCEL)->EnableWindow(FALSE);
	GetDlgItem(IDOK)->EnableWindow(FALSE);
	GetDlgItem(IDC_BUTTON_SCAN)->EnableWindow(FALSE);

UpdateData(TRUE);//将对文本框的内容映射到所关联的变量上
if(m_filename=="")
		{	MessageBox("请选择你要加密的文件");
			GetDlgItem(IDCANCEL)->EnableWindow(true);
			GetDlgItem(IDOK)->EnableWindow(true);
			GetDlgItem(IDC_BUTTON_SCAN)->EnableWindow(true);
					return;
			}
if(m_password=="")
		{	MessageBox("请输入您的密码");
			GetDlgItem(IDCANCEL)->EnableWindow(true);
			GetDlgItem(IDOK)->EnableWindow(true);
			GetDlgItem(IDC_BUTTON_SCAN)->EnableWindow(true);
					return;
		}	
if(m_password.GetLength()>8||m_password.GetLength()<8)
		{
			MessageBox("密码字符必须为8位之间，请重新输入");
			m_password="";
			UpdateData(FALSE);
			UpdateData(TRUE);
			GetDlgItem(IDCANCEL)->EnableWindow(true);
			GetDlgItem(IDOK)->EnableWindow(true);
			GetDlgItem(IDC_BUTTON_SCAN)->EnableWindow(true);
			return;
		}	
	if(SearchFile("C:\\mylockedfile",strfilename))
	{
		m_password="";
		m_filename="";
		UpdateData(FALSE);
		UpdateData(TRUE);
		MessageBox("有重名文件存在，请更换文件名以后再加密");
		GetDlgItem(IDCANCEL)->EnableWindow(true);
		GetDlgItem(IDOK)->EnableWindow(true);
		GetDlgItem(IDC_BUTTON_SCAN)->EnableWindow(true);
		return;
	}
	CFile fin;//建立文件指针
	if(!fin.Open(m_filename,CFile::modeRead))//判断打开文件是否正常
	{
	MessageBox("文件打开失败，请重新检查你的文件是否存在或文件名是否正确");
	m_filename="";
	m_password="";
	UpdateData(FALSE);
	UpdateData(TRUE);
	GetDlgItem(IDCANCEL)->EnableWindow(true);
	GetDlgItem(IDOK)->EnableWindow(true);
	GetDlgItem(IDC_BUTTON_SCAN)->EnableWindow(true);
	return;
	}
	//如果正常
	CString savename;
	savename=m_filename+".jib";
	CFile fout(savename,CFile::modeCreate|CFile::modeWrite);//新建一个文件	
	long lFileLen=fin.GetLength();//求出文件的字节数
	long c=lFileLen/8;//计算文件可以分多少块进行加密
	int d=lFileLen%8;//计算最后个块的字节数
	fin.SeekToBegin();
	fout.SeekToBegin();


	char inbuf[8],key[8];//定义一个字符数组，用来存储每次读入的信息
	memcpy(key,(LPCTSTR)m_password,8);

	float m;
	int t=0;
	m=c/20;
	m_progress.SetRange(0,20);
	m_progress.SetStep(1);
	m_progress.SetPos(0);


	//把密码加密后加入到新的文件前8位
    for(long i=0;i<8;i++)
		inbuf[i]=key[i];
	Des_SetKey(key);
	Des_Run(inbuf, inbuf, ENCRYPT);
	fout.Write(inbuf,8);
	//把特殊的标志位加入到文件中的第二个8位
	for(i=0;i<8;i++)
		inbuf[i]=48+d;
	Des_SetKey(key);
	Des_Run(inbuf, inbuf, ENCRYPT);
	fout.Write(inbuf,8);



//开始加密数据
	for( i=0;i<c;i++)//加密成块数据
	{
		m_progress.SetPos((int)((i+1)/m));
		if(t!=(int)((i+1)/m))
		{t=(int)((i+1)/m);
		m_progress.StepIt();
		}
		fin.Read(inbuf,8);
		Des_SetKey(key);
		Des_Run(inbuf, inbuf, ENCRYPT);
		fout.Write(inbuf,8);

	}
	if(d>0)//加密尾数
	{
		fin.Read(inbuf,d);
		for(int k=d;k<8;k++)
		{
			inbuf[k]=48+(int)d;
		}
		Des_SetKey(key);
		Des_Run(inbuf, inbuf, ENCRYPT);
		fout.Write(inbuf,8);
	}
	fin.Close();
	fout.Close();
	CString savetofilename("C:\\mylockedfile\\");
	savetofilename=savetofilename+strfilename;
	MoveFile(savename,savetofilename);
	DeleteFile(m_filename);
	CDialog::OnOK();
}

void Cencryptfile::OnButtonScan() 
{
	CString filepath="";
	CString filename="";
	CFileDialog dlg(TRUE,"txt","*.txt",OFN_HIDEREADONLY,"文本文件(*.txt)|*.txt|Word文件(*.doc)|*.doc|全部(*.*)|*.*||");
	if(dlg.DoModal()==IDOK)
	{
		filepath=dlg.GetPathName();
		filename=dlg.GetFileName();
	}
	strfilename=filename;
	m_filename=filepath;
	UpdateData(FALSE);
}

BOOL Cencryptfile::SearchFile(CString strDir, CString strFile)
{
CFileFind ff;
CString szDir=strDir;
BOOL flag=false;
if(szDir.Right(1)!="\\")
	szDir+="\\";

szDir+="*.*";
BOOL res=ff.FindFile(szDir);
while(res)
{
	res=ff.FindNextFile();
	if(ff.GetFileName()==strFile)
	{
		flag=true;
	}
}
return flag;

}