www.pudn.com > TrueCrypt.rar > MainFrm.cpp
// MainFrm.cpp : implementation of the CMainFrame class
//
#include "stdafx.h"
#include "CMyExplorer.h"
#include "MainFrm.h"
#include "ExplorerListView.h"
#include "ExplorerTreeView.h"
#include "ShellContextMenu.h"
#include "ShellFileOp.h"
#include "windows.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
extern CExplorerListView * g_ListView;
extern CExplorerTreeView * g_TreeView;
/////////////////////////////////////////////////////////////////////////////
// CMainFrame
IMPLEMENT_DYNAMIC(CMainFrame, CFrameWnd)
BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
//{{AFX_MSG_MAP(CMainFrame)
ON_WM_CREATE()
ON_WM_SETFOCUS()
ON_COMMAND(ID_VIEW_SMALLICON, OnViewSmallicon)
ON_COMMAND(ID_VIEW_LIST, OnViewList)
ON_COMMAND(ID_VIEW_LARGEICON, OnViewLargeicon)
ON_COMMAND(ID_VIEW_REPORT, OnViewReport)
ON_COMMAND(ID_EDIT_COPY, OnEditCopy)
ON_COMMAND(ID_EDIT_CUT, OnEditCut)
ON_COMMAND(ID_EDIT_PASTE, OnEditPaste)
ON_COMMAND(ID_ENCRYPT, Onfile_encrypt)
ON_COMMAND(ID_DECRYPT, Onfile_decrypt)
//}}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
}
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.dwExStyle &= ~WS_EX_CLIENTEDGE;
cs.lpszClass = AfxRegisterWndClass(0);
SetTitle( "透明加密器TrueCrypt" );
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::OnSetFocus(CWnd* pOldWnd)
{
// forward focus to the view window
// m_wndView.SetFocus();
}
BOOL CMainFrame::OnCmdMsg(UINT nID, int nCode, void* pExtra, AFX_CMDHANDLERINFO* pHandlerInfo)
{
// let the view have first crack at the command
// if (m_wndView.OnCmdMsg(nID, nCode, pExtra, pHandlerInfo))
// return TRUE;
// otherwise, do default handling
return CFrameWnd::OnCmdMsg(nID, nCode, pExtra, pHandlerInfo);
}
BOOL CMainFrame::OnCreateClient(LPCREATESTRUCT lpcs, CCreateContext* pContext)
{
// TODO: Add your specialized code here and/or call the base class
CRect rect;
GetClientRect(&rect);
if ( !m_wndSplitter.CreateStatic(this,1,2) )
{
TRACE0("Failed to CreateStaticSplitter\n");
return FALSE;
}
// add the first splitter pane - the default view in column 0
if (!m_wndSplitter.CreateView(0, 0, RUNTIME_CLASS(CExplorerTreeView),
CSize(240,rect.bottom), pContext))
{
TRACE0("Failed to create first pane\n");
return FALSE;
}
// add the second splitter pane - an input view in column 1
if (!m_wndSplitter.CreateView(0, 1, RUNTIME_CLASS(CExplorerListView),
CSize(rect.right,rect.bottom-350), pContext))
{
TRACE0("Failed to create second pane\n");
return FALSE;
}
m_pTreeView = (CExplorerTreeView*)m_wndSplitter.GetPane(0,0);
m_pListView = (CExplorerListView*)m_wndSplitter.GetPane(0,1);
return CFrameWnd::OnCreateClient(lpcs, pContext);
}
///////////////////////////////////////////
void CMainFrame::OnViewSmallicon()
{
// TODO: Add your command handler code here
g_ListView->SetListViewType(LVS_SMALLICON);
}
void CMainFrame::OnViewList()
{
// TODO: Add your command handler code here
g_ListView->SetListViewType(LVS_LIST);
}
void CMainFrame::OnViewLargeicon()
{
// TODO: Add your command handler code here
g_ListView->SetListViewType(LVS_LIST);
}
void CMainFrame::OnViewReport()
{
// TODO: Add your command handler code here
g_ListView->SetListViewType(LVS_REPORT);
}
void CMainFrame::OnEditCopy()
{
// TODO: Add your command handler code here
UINT uDropEffect = RegisterClipboardFormat("Preferred DropEffect");
HGLOBAL hGblEffect = GlobalAlloc(GMEM_ZEROINIT|GMEM_MOVEABLE|GMEM_DDESHARE,sizeof(DWORD));
DWORD * dwDropEffect =(DWORD*)GlobalLock(hGblEffect);
*dwDropEffect = DROPEFFECT_COPY;
GlobalUnlock(hGblEffect);
POSITION pos = m_pListView->m_pListCtrl->GetFirstSelectedItemPosition();
if ( pos==NULL )
return;
char * szSrc = new char[MAX_PATH];
CStringArray sDragArray;
while(pos)
{
int nItem = m_pListView->m_pListCtrl->GetNextSelectedItem(pos);
LPLISTPARAM pliItem = (LPLISTPARAM)m_pListView->m_pListCtrl->GetItemData(nItem);
ZeroMemory(szSrc,MAX_PATH);
SHGetPathFromIDList(pliItem->lpidl,szSrc);
sDragArray.Add(szSrc);
}
delete szSrc;
//创建拖拽数据源
int hgblSize = sizeof(DROPFILES);
for ( int i=0 ; ipFiles = sizeof(DROPFILES);
pDropFiles->fWide = FALSE;
LPSTR pGlobalString = (LPSTR)(pDropFiles) + sizeof(DROPFILES);
for ( i=0 ; im_pListCtrl->GetFirstSelectedItemPosition();
if ( pos==NULL )
return;
char * szSrc = new char[MAX_PATH];
CStringArray sDragArray;
while(pos)
{
int nItem = m_pListView->m_pListCtrl->GetNextSelectedItem(pos);
LPLISTPARAM pliItem = (LPLISTPARAM)m_pListView->m_pListCtrl->GetItemData(nItem);
ZeroMemory(szSrc,MAX_PATH);
SHGetPathFromIDList(pliItem->lpidl,szSrc);
sDragArray.Add(szSrc);
}
delete szSrc;
//创建拖拽数据源
int hgblSize = sizeof(DROPFILES);
for ( int i=0 ; ipFiles = sizeof(DROPFILES);
pDropFiles->fWide = FALSE;
LPSTR pGlobalString = (LPSTR)(pDropFiles) + sizeof(DROPFILES);
for ( i=0 ; im_hMoveSrc = g_TreeView->m_hSelItem;
}
}
void CMainFrame::OnEditPaste()
{
// TODO: Add your command handler code here
if ( !IsClipboardFormatAvailable(CF_HDROP) )
return;
UINT uDropEffect = RegisterClipboardFormat("Preferred DropEffect");
if ( !IsClipboardFormatAvailable(uDropEffect) )
return;
if ( !::OpenClipboard(NULL) )
return;
DWORD dwEffect = *(DWORD*)::GetClipboardData(uDropEffect);
HGLOBAL hglobal = ::GetClipboardData(CF_HDROP);
::CloseClipboard();
if( hglobal != NULL )
{
CShellFileOp op;
if ( dwEffect & DROPEFFECT_MOVE )
op.SetOperationFlags(FO_MOVE, this, FOF_ALLOWUNDO|FOF_NOCONFIRMMKDIR);
else
op.SetOperationFlags(FO_COPY, this, FOF_ALLOWUNDO|FOF_NOCONFIRMMKDIR);
char * szFileName;
szFileName = new char[MAX_PATH];
HDROP hDrop = (HDROP)GlobalLock(hglobal);
int FileNum = DragQueryFile(hDrop,-1,NULL,0);
for ( int i =0 ; i < FileNum ; i++ )
{
ZeroMemory(szFileName,MAX_PATH);
DragQueryFile(hDrop,i,szFileName,255);
op.AddSourceFile(szFileName);
}
delete szFileName;
//设置粘贴目标
char * szDest = new char[MAX_PATH];
ZeroMemory(szDest,MAX_PATH);
LPTREEPARAM ptrItem = (LPTREEPARAM)g_TreeView->GetSelTreeData();
SHGetPathFromIDList(ptrItem->lpidl,szDest);
op.AddDestFile(szDest);
BOOL bSuccess, bAPICalled = FALSE, bAborted = FALSE;
int nAPIReturn = 0;
bSuccess = op.Go (&bAPICalled, &nAPIReturn, &bAborted);
GlobalFree(hglobal);
delete szDest;
//refresh
if ( bSuccess && !bAborted )
{
g_TreeView->SendMessage(WM_REFRESHFILES,0,0);
g_TreeView->SendMessage(WM_REFRESHFILES,(WPARAM)g_TreeView->m_hMoveSrc,0);
g_ListView->SendMessage(WM_REFRESHFILES,0,0);
}
}
}
//加密文件函数
void CMainFrame::Onfile_encrypt()
{
int count=0;
POSITION pos = m_pListView->m_pListCtrl->GetFirstSelectedItemPosition();
if ( pos==NULL )
return;
while(pos)
{
char * szPath = new char[MAX_PATH];
//获取文件路径
int nItem = m_pListView->m_pListCtrl->GetNextSelectedItem(pos);
LPLISTPARAM pliItem = (LPLISTPARAM)m_pListView->m_pListCtrl->GetItemData(nItem);
ZeroMemory(szPath,MAX_PATH);
SHGetPathFromIDList(pliItem->lpidl,szPath);
//加密过程，仅添加标志位
CFile File;
File.Open(szPath,CFile::modeRead,NULL);
unsigned char *inBuff;
unsigned char *outBuff;
CString flag="####";
LPTSTR eflag=flag.GetBuffer(5);
flag.ReleaseBuffer();
long lFileLen=File.GetLength();
inBuff=new unsigned char[lFileLen];
outBuff=new unsigned char[lFileLen+4];
File.Read(inBuff,lFileLen);
for(int i=0;i<4;i++)
outBuff[i]=eflag[i];
for(i=0;im_pListCtrl->SetItemText(nItem,2,"加密");
delete szPath;
count++;
}
CString s;
s.Format("%d",count);
s+="个文件";
AfxMessageBox("成功加密"+s);
}
void CMainFrame::Onfile_decrypt()
{
int count=0;
POSITION pos = m_pListView->m_pListCtrl->GetFirstSelectedItemPosition();
if ( pos==NULL )
return;
while(pos)
{
char * szPath = new char[MAX_PATH];
//获取文件路径
int nItem = m_pListView->m_pListCtrl->GetNextSelectedItem(pos);
LPLISTPARAM pliItem = (LPLISTPARAM)m_pListView->m_pListCtrl->GetItemData(nItem);
ZeroMemory(szPath,MAX_PATH);
SHGetPathFromIDList(pliItem->lpidl,szPath);
//文件解密过程
CFile File;
File.Open(szPath,CFile::modeRead,NULL);
char temp[4];
long lFileLen=File.GetLength();
File.Read(temp,4);
if(temp[0]=='#'&&temp[1]=='#'&&temp[2]=='#'&&temp[3]=='#')
{
unsigned char *inBuff;
inBuff=new unsigned char[lFileLen-4];
File.Read(inBuff,lFileLen-4);
File.Close();
CFile newfile;
newfile.Open(szPath,CFile::modeCreate|CFile::modeWrite);
for(int i=0;im_pListCtrl->SetItemText(nItem,2,"正常");
count++;
}
else
{
AfxMessageBox("文件没有加密！");
return;
}
delete szPath;
}
CString s;
s.Format("%d",count);
s+="个文件";
AfxMessageBox("成功解密"+s);
}