transfiledlg.cpp

这是网络信息安全课程写的一个给文件加密传输的源代码

// TransFileDlg.cpp : implementation file
//

#include "stdafx.h"
#include "TransFile.h"
#include "TransFileDlg.h"
#include "ServerSocket.h"
#include "ClientSocket.h"
#include "DesEncrypt.h"

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


/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About
const int SERVERPORT = 1004;
const int CLIENTPORT = 1521;
const int MAXSENDNUM = 8192; //缓冲区发送和接收的最大字节数

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	//{{AFX_DATA_INIT(CAboutDlg)
	//}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CAboutDlg)
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CTransFileDlg dialog

CTransFileDlg::CTransFileDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CTransFileDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CTransFileDlg)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDI_FILE);
}

void CTransFileDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CTransFileDlg)
	DDX_Control(pDX, IDC_LIST, m_List);
	DDX_Control(pDX, IDC_HOST, m_Host);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CTransFileDlg, CDialog)
	//{{AFX_MSG_MAP(CTransFileDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_BUTTONFILE, OnButtonfile)
	ON_BN_CLICKED(IDC_BUTTONTRANS, OnButtontrans)
	ON_WM_DROPFILES()
	ON_WM_TIMER()
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CTransFileDlg message handlers

BOOL CTransFileDlg::OnInitDialog()
{
	CDialog::OnInitDialog();
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	SetIcon(m_hIcon, TRUE);	
	SetIcon(m_hIcon, FALSE);
	
	m_ImageList.Create(32,32,ILC_COLOR24|ILC_MASK,0,0);
	m_ImageList.Add(LoadIcon(AfxGetInstanceHandle(), MAKEINTRESOURCE(IDI_FILE)));
	m_ImageList.Add(LoadIcon(AfxGetInstanceHandle(), MAKEINTRESOURCE(IDI_FILE1)));

	m_List.SetImageList(&m_ImageList,LVSIL_NORMAL);

	m_pServerSock = new CServerSocket(this);
	m_ServerSock  = new CServerSocket(this);
	m_pSendSock   = new CClientSocket(this);

	m_IsReveived = TRUE;

	char host[128];
	gethostname(host,128);

	hostent* tent;
	tent=  (hostent*)gethostbyname(host);
	m_IP =  inet_ntoa(*(in_addr*)tent->h_addr_list[0]) ;

	m_pServerSock->Create(SERVERPORT);
	m_pSendSock->Create(CLIENTPORT);

	if (!m_pServerSock->Listen()) //开始监听
		MessageBox("监听失败");
	CString str ;
	str.Format("%i",999999);
	m_pServerSock->SetSockOpt(SO_RCVBUF,str.GetBuffer(0),str.GetLength());
	m_pSendSock->SetSockOpt(SO_SNDBUF,str.GetBuffer(0),str.GetLength());	
	m_ServerSock->SetSockOpt(SO_RCVBUF,str.GetBuffer(0),str.GetLength());

	m_Send            = FALSE;
	m_SendNext        = FALSE;
	m_Received        = FALSE;
	m_SendFinished    = FALSE;
	m_Count = 0;
	m_FileSize = 0;
	return TRUE;  
}

void CTransFileDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CTransFileDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CTransFileDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

void CTransFileDlg::OnOK() 
{

}

BOOL CTransFileDlg::PreTranslateMessage(MSG* pMsg) 
{
	if ((pMsg->hwnd == m_List.m_hWnd) && (pMsg->message==WM_DROPFILES))
	{
		OnDropFiles((HDROP)pMsg->wParam);
		return TRUE;
	}
	//防止按Esc键关闭对话框
	if (pMsg->message==WM_KEYDOWN && (int)pMsg->wParam==VK_ESCAPE)
			return TRUE;
	return CDialog::PreTranslateMessage(pMsg);
}

void CTransFileDlg::OnButtonfile() 
{
	CFileDialog fDlg (TRUE,NULL,NULL,OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,NULL,this);
	if (fDlg.DoModal()==IDOK)
	{
		m_List.DeleteAllItems();
		m_FileName = fDlg.GetPathName();
		m_List.InsertItem(0,m_FileName,0);
	}
}

void CTransFileDlg::OnButtontrans() 
{

	if (m_Send==FALSE)
	{
		if (m_List.GetItemCount()<1)
		{
			MessageBox("请选择发送的文件");
			return;
		}

		CString host;
		m_Host.GetWindowText(host);
		if (host.IsEmpty())
		{
			MessageBox("请设置主机名称或IP");
			return;
		}
		m_SendFile.Abort();
		m_SendFile.Open(m_FileName,CFile::modeRead);
		m_SendFileLen = m_SendFile.GetLength();

		char* fileinfo = (char*)LocalAlloc(LMEM_FIXED,MAX_PATH+4);
		memset(fileinfo,0,MAX_PATH+4);
		memcpy(fileinfo,m_FileName.GetBuffer(0),m_FileName.GetLength());

		CString filesize;
		filesize.Format("%i",m_SendFileLen);
		memcpy(fileinfo+m_FileName.GetLength()+1,filesize.GetBuffer(0),filesize.GetLength());

		m_pSendSock->Close();
		m_pSendSock->Create(CLIENTPORT);
		if (!m_pSendSock->Connect(host,SERVERPORT))
		{
			int aa = m_pSendSock->GetLastError();
			m_SendFile.Close();
			m_pSendSock->Close();
			m_pSendSock->Create(CLIENTPORT);

			MessageBox("连接失败");
			m_Send = FALSE;
			return;
		}
		SetTimer(1,2000,NULL);
		//发送文件信息
		m_SendNum = 0;
		m_pSendSock->Send(fileinfo,MAX_PATH+4);

		m_SendLen = 0;
		LocalFree(fileinfo);
	}
	else
	{
		MessageBox("当前发送任务没有完成");
	}

}

void CTransFileDlg::AcceptConnection()
{
	m_ServerSock->Close();
	m_pServerSock->Accept(*m_ServerSock);
}

void CTransFileDlg::OnCancel()
{
	m_pServerSock->Close();
	if (m_pSendSock != NULL)
	{
		m_pSendSock->Close();
		delete m_pSendSock;
		m_pSendSock = NULL;
	}
	delete m_pServerSock;
	m_ServerSock->Close();
	delete m_ServerSock;
	CDialog::OnCancel();
}

//接收数据
void CTransFileDlg::ReceiveData()
{

	void* buffer = (void*)LocalAlloc(LMEM_FIXED,MAXSENDNUM+1);
	if (m_IsReveived) //第一次接收数据,获取接收的文件信息
	{
		char* fileinfo = (char*)LocalAlloc(LMEM_FIXED,MAX_PATH+4);
		memset(fileinfo,0,MAX_PATH+4);
		int len = m_ServerSock->Receive(fileinfo,MAX_PATH+4);
		m_ReceiveFile.Format("%s",fileinfo); //记录接收的文件名

		char* temp = fileinfo;
		while (temp[0] !=0)
		{
			temp +=1;
		}
		temp+=1;
		m_FileSize= atoi(temp); //记录接收的文件大小
		LocalFree(fileinfo);
		m_FileLen = 0;
		CString name = GetFileName(m_ReceiveFile);
		if (MessageBox("是否接收数据?","提示",MB_YESNO)==IDYES)
		{
			BROWSEINFO bInfo;
			bInfo.hwndOwner = m_hWnd;
			bInfo.pidlRoot  = NULL;
			char filename[MAX_PATH];
			char dir[MAX_PATH];
			memset(filename,0,MAX_PATH);
			memset(dir,0,MAX_PATH);
			
			bInfo.pszDisplayName = filename;
			bInfo.lpszTitle ="请选择目的路径";
			bInfo.ulFlags = BIF_BROWSEINCLUDEFILES|BIF_RETURNFSANCESTORS|BIF_RETURNONLYFSDIRS|BIF_BROWSEFORCOMPUTER;
			bInfo.lParam  =NULL;
			bInfo.lpfn  =NULL;
			bInfo.iImage = 0;

			LPITEMIDLIST lpList =  SHBrowseForFolder(&bInfo);
			
			CString filedir;
			if (SHGetPathFromIDList(lpList,dir))
			{
				m_File.Abort();
				filedir = dir;
				filedir += "\\";
				m_File.Open(filedir+name,CFile::modeCreate|CFile::modeWrite);
				m_IsReveived = FALSE;  //文件信息接收结束
				m_ServerSock->Send("1111",4);

				LocalFree(buffer);
				return;
			}
			else  //拒绝接收文件
			{
				m_File.Close();
				LocalFree(buffer);
				m_ServerSock->Send("取消",4);
				m_ServerSock->Close();
				m_IsReveived = TRUE; 
				return;					
			}
		}
		else
		{	
			LocalFree(buffer);
			m_ServerSock->Send("取消",4);
			m_ServerSock->Close();
			m_IsReveived = TRUE; 
			return;
		}
	}
	else //开始接收数据
	{

		int filelen =  m_ServerSock->Receive(buffer,MAXSENDNUM+1);
		if (filelen==SOCKET_ERROR)
		{
			m_IsReveived = TRUE;
			m_File.Close();
			m_ServerSock->Close();
			LocalFree(buffer);
			MessageBox("数据接收错误");

			return;
		}
		m_FileLen += filelen;
		//Des st
		if (IsDlgButtonChecked(IDC_DES) == 1)
		{
			char key[]={1,2,3,4,5,6,7,8};
			CDesEncrypt Des;
			Des.DesGo((char*)buffer, (char*)buffer, filelen, key, sizeof(key), FALSE);
		}

		//Des end
		m_File.WriteHuge(buffer,filelen);
		LocalFree(buffer);
		if (m_FileLen>=m_FileSize)  //接收完成
		{
			m_File.Close();
			m_ServerSock->Send("结束",4);
			m_IsReveived = TRUE;
			m_ServerSock->Close();
			MessageBox("接收完成");
		}
		else
			m_ServerSock->Send("1111",4);
	}
}

CString CTransFileDlg::GetFileName(CString FileName)
{
	int pos = FileName.ReverseFind('\\');
	CString name = FileName.Mid(pos+1,FileName.GetLength());
	return name;
}


void CTransFileDlg::SendFileData()
{
	m_Send=TRUE;
	m_SendNext = TRUE;
	m_Received = TRUE;

	char data[100];
	memset(data,0,100);
	m_SendNum +=1;
	m_pSendSock->Receive(data,100);  //获得对方的应答信息

	CString rdata = data;

	if (m_SendLen>=m_SendFileLen)    //发送文件完成
	{
		m_Send = FALSE;
		m_SendNext = FALSE;
		m_SendFile.Close();
		m_pSendSock->Close();
		m_pSendSock->Create(CLIENTPORT);
		MessageBox("发送完成");
		return;
	}

	if ((m_SendLen==0)&&(rdata=="取消"))
	{
		m_Send = FALSE;
		m_SendNext = FALSE;
		m_SendFile.Close();
		m_pSendSock->Close();
		m_pSendSock->Create(CLIENTPORT);

		MessageBox("对方取消了文件接收");
		return;
	}

	char* membuf= (char*)LocalAlloc(LMEM_FIXED,MAXSENDNUM);
	memset(membuf,0,MAXSENDNUM);
	int result;

	if ((m_SendLen <m_SendFileLen)&&((m_SendLen+MAXSENDNUM)>m_SendFileLen))
	{
		m_SendFile.ReadHuge(membuf,m_SendFileLen-m_SendLen);
		//Des st
		if (IsDlgButtonChecked(IDC_DES) == 1)
		{
			char key[]={1,2,3,4,5,6,7,8};
			CDesEncrypt Des;
			Des.DesGo((char*)membuf, (char*)membuf, m_SendFileLen-m_SendLen, key, sizeof(key), TRUE);
		}
		//Des end
		result = m_pSendSock->Send(membuf, m_SendFileLen-m_SendLen);
		if (result ==SOCKET_ERROR)
		{
			m_Send = FALSE;
			m_SendNext = FALSE;
			m_SendFile.Close();
			m_pSendSock->Close();
			m_pSendSock->Create(CLIENTPORT);
			LocalFree(membuf);
			KillTimer(1);
			MessageBox("发送失败");	
			return ;
		}
		m_SendLen =m_SendFileLen;
	}
	else
	{	
		m_SendFile.ReadHuge(membuf,MAXSENDNUM);
		//Des st
		if (IsDlgButtonChecked(IDC_DES) == 1)
		{
		char key[]={1,2,3,4,5,6,7,8};
		CDesEncrypt Des;
		Des.DesGo((char*)membuf, (char*)membuf, MAXSENDNUM, key, sizeof(key), TRUE);
		}
		//Des end
		result = m_pSendSock->Send(membuf, MAXSENDNUM);
		if (result ==SOCKET_ERROR)
		{
			m_Send = FALSE;
			m_SendNext = FALSE;
			m_SendFile.Close();
			m_pSendSock->Close();
			m_pSendSock->Create(CLIENTPORT);

			LocalFree(membuf);
			KillTimer(1);
			MessageBox("发送失败");	
			return;
		}
		LocalFree(membuf);
		m_SendLen +=MAXSENDNUM;
	}
}

void CTransFileDlg::OnDropFiles(HDROP hDropInfo) 
{
	char filename[MAX_PATH];
	DragQueryFile(hDropInfo,0,filename,MAX_PATH);
	m_List.DeleteAllItems();
	m_FileName = filename;
	m_List.InsertItem(0,m_FileName,0);
	//CDialog::OnDropFiles(hDropInfo);
}

void CTransFileDlg::OnTimer(UINT nIDEvent) 
{
	if (m_Send==TRUE)  //文件传输中
	{
		if (m_Received == TRUE)
		{
			m_Count = 0;
			m_Received =FALSE;
		}
		else
		{
			m_Count +=1;

			if (m_Count >5)
			{
				m_Count  = 0;
				KillTimer(1);
				if (MessageBox("对方长时间没有回应信息,是否断开连接?","提示",MB_YESNO)==IDYES)
				{
					m_Send = FALSE;
					m_SendNext = FALSE;
					m_SendFile.Abort();
					m_ServerSock->Close();
					m_pSendSock->Close();
					m_pSendSock->Create(CLIENTPORT);
					BOOL value = TRUE;
					m_pSendSock->SetSockOpt(SO_LINGER,(void*)&value,1);
					m_IsReveived = TRUE;
				}
				else
				{
					SetTimer(1,2000,NULL);
					m_Received = FALSE;
				}
			}
		}
	}
	CDialog::OnTimer(nIDEvent);
}