// 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;

}