www.pudn.com > aes_encrypt.rar > MainFrm.cpp
// MainFrm.cpp : implementation of the CMainFrame class
//
//////////////////////////
#include "stdafx.h"
#include "AES.h"
#include "iostream.h"
#include "string.h"
#include "MainFrm.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
/////////////////////////////////////////////////////////////////////////////
// CMainFrame
IMPLEMENT_DYNCREATE(CMainFrame, CFrameWnd)
BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
//{{AFX_MSG_MAP(CMainFrame)
ON_WM_CREATE()
ON_COMMAND(ID_AES, OnAes)
ON_UPDATE_COMMAND_UI(ID_AES, OnUpdateAes)
ON_COMMAND(ID_UNAES, OnUnaes)
ON_UPDATE_COMMAND_UI(ID_UNAES, OnUpdateUnaes)
ON_COMMAND(ID_CLEAR, OnClear)
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
static UINT indicators[] =
{
ID_SEPARATOR, // status line indicator
ID_INDICATOR_CAPS,
ID_INDICATOR_NUM,
ID_INDICATOR_SCRL,
};
/////////////////////////////////////////////////////////////////////////////
// CMainFrame construction/destruction
CMainFrame::CMainFrame()
{
// TODO: add member initialization code here
m_Aes=false;
m_Unaes=false;
/////////////////////////
for(int i=0;i<10;i++)
table[i]=i+48; //0~9
for(i=10;i<36;i++)
table[i]=i+55; //A~Z
for(i=36;i<62;i++)
table[i]=i+61; // a~z
table[62]='-'; //-
table[63]='/'; ///
}
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
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::OnAes()
{
if(Dlg.DoModal()==IDOK)
{
if(Dlg.m_str1!=Dlg.m_str2)
{ MessageBox(" 两次密码输入不一致，请重新输入！ ","密码错误");
Dlg.m_str1=Dlg.m_str2="";
}
else if(Dlg.m_str2=="")
MessageBox(" 密码不能为空，请重新输入 ","密码错误");
else if(!(Dlg.m_str2.GetLength()>7&&Dlg.m_str2.GetLength()<17))
{
MessageBox(" 密码必须在8—16位之间，请重新输入 ","提示");
Dlg.m_str1=Dlg.m_str2="";
}
else
{
((CEditView*)this->GetActiveView())->GetWindowText(szContent);
AES(szContent);
((CEditView*)this->GetActiveView())->SetWindowText(szContent2);
m_Aes=true;
m_Unaes=false;
m_str=Dlg.m_str1;
Dlg.m_str1=Dlg.m_str2="";
}
}
else
Dlg.m_str1=Dlg.m_str2="";
}
//****************************************//
void CMainFrame::OnUpdateAes(CCmdUI* pCmdUI)
{
// TODO: Add your command update UI handler code here
if (m_Aes)
pCmdUI->Enable(false);
else
pCmdUI->Enable(true);
}
//*************************************//
void CMainFrame::OnUnaes()
{
// TODO: Add your command handler code here
UpdateData(false);
if(Dlg1.DoModal()==IDOK)
{
if(Dlg1.m_str3==m_str)
{
((CEditView*)this->GetActiveView())->GetWindowText(szContent);
UNAES(szContent);
((CEditView*)this->GetActiveView())->SetWindowText(szContent2);
Dlg1.m_str3="";
m_Aes=false;
m_Unaes=true;
}
else
MessageBox(" 密码错误，请重新输入 ","密码错误");
Dlg1.m_str3="";
}
Dlg1.m_str3="";
}
//******************************************//
void CMainFrame::AES(CString)
{
char *key,*text_p,*text_x;
int i,j;
int x,y,z;
key=new char[17];
strncpy(key,(LPCTSTR)m_str,m_str.GetLength());
for(j=0;j>2;
x=x&63;//00111111;
*(text_x+i)=table[x]; //first chracter
i++;
y=x=*(text_p+j);
j++;
y=y>>4;
y=y&15;//00001111;
z=z<<4;
z=z&63;
y=y|z;
*(text_x+i)=table[y]; // second character
i++;
z=y=*(text_p+j);
j++;
z=z>>6;
z=z&3;//00000011;
x=x<<2;
x=x&63;//00111111;
z=x|z;
*(text_x+i)=table[z]; //third character
i++;
y=y&63;//001111111;
*(text_x+i)=table[y]; // fourth character
i++;
}
szContent=text_x;
szContent2=szContent.Left(after_file_length);
delete text_x;
delete text_p;
}
//******************************************//
void CMainFrame::UNAES(CString)
{
// TODO: Add your command handler code here
char *text_p,*text_x;
int i,j;
int x,y,z,z1;
text_p=new char[2*file_length];
text_x=new char[2*after_file_length+3];
strncpy(text_x,(LPCTSTR)szContent,after_file_length);
for(i=0,j=0;i>4;
y=y|z;
*(text_p+j)=y; //save first byte
j++;
x=*(text_x+i); //read 3th byte
i++;
y=z=find(x);
z1=z1<<4;
y=y>>2;
y=y|z1;
*(text_p+j)=y; //save second byte
j++;
x=*(text_x+i);
i++;
y=find(x);
z=z<<6;
y=y|z;
*(text_p+j)=y; //save third byte
j++;
}
for(i=0;iEnable(false);
else
pCmdUI->Enable(true);
}
void CMainFrame::OnClear()
{
// TODO: Add your command handler code here
((CEditView*)this->GetActiveView())->GetWindowText(szContent);
szContent.Empty();
((CEditView*)this->GetActiveView())->SetWindowText(szContent);
}
// search table()
int CMainFrame::find(int p)
{
int k=-1;
do{
k++;
} while(table[k]!=p);
return k;
}