【转】VC++Csocket类的文件收发函数

转自http://blog.sina.com.cn/s/blog_5cec38f30100b8ef.html 

       本文的代码使用CSocket类与Cfile类完成了传输文件的功能。有两段程序，一个是服务器端的，一个是客户端的。可以发现，这个类的引入，让Socket编程方便不少。
      请不要让本程序传送大文件，不然死机没商量，因为程序中发送接收方分配的堆是与文件一样大的！
      编译之前请各位自己修改服务器的IP，发送与接收文件的名称与路径。
      本程序不需要ws2_32.lib库。

几个用到的函数：
-------------------------------------------
(1)初始化Socket
BOOL AfxSocketInit( WSADATA* lpwsaData = NULL );
出错返回0, 否则返回非零值

(2)CSocket类的成员函数Create
BOOL Create( UINT nSocketPort = 0, int nSocketType = SOCK_STREAM, LPCTSTR lpszSocketAddress = NULL );
第一个参数的端口号如果为默认值0，表示由MFC为你任意选择一个端口。
第二个参数默认为字节流方式

(2) CSocket类的基类CAsyncSocket的成员函数Listen
BOOL Listen( int nConnectionBacklog = 5 );
这个侦听函数使用时取默认值，即不带任何参数

(3) CSocket类的基类CAsyncSocket的成员函数Accept
virtual BOOL Accept( CAsyncSocket& rConnectedSocket, SOCKADDR* lpSockAddr = NULL, int* lpSockAddrLen = NULL );
这个函数只用传入一个CSocket对象即可

(4) CSocket类的基类CAsyncSocket的成员函数Send
virtual int Send( const void* lpBuf, int nBufLen, int nFlags = 0 );
参数1为缓冲区地址，2为缓冲区大小,3取默认值

(5)Cfile类的成员函数Open
virtual BOOL Open( LPCTSTR lpszFileName, UINT nOpenFlags, CFileException* pError = NULL );

使用时只用前两个参数，lpszFileName传入文件名，nOpenFlags决定读写方式，取值可以是：
CFile::modeRead //只读
CFile::modeWrite //写入
CFile::typeBinary //二进制方式
CFile::typeText //文本方式

(6) Cfile类的成员函数GetLength
GetLength() 返回文件大小

(7)Cfile::Read
virtual UINT Read( void* lpBuf, UINT nCount );
参数一是缓冲区地址，参数二是一次读多少字节

(8) CSocket类的基类CAsyncSocket的成员函数Receive
virtual int Receive( void* lpBuf, int nBufLen, int nFlags = 0 );
接收堆中的数据

(9)全局函数，初始化mfc，对于控制台应用程序必须调用它初始化mfc
BOOL AFXAPI AfxWinInit( HINSTANCE hInstance, HINSTANCE hPrevInstance, LPTSTR lpCmdLine, int nCmdShow )
参数一是应用程序实例句柄，参数二是父窗口句柄，参数三是指向命令行串的字符指针

(10) 全局函数，得到命令行串
GetCommandLine()


服务端代码：
-------------------------------------------
#include "stdafx.h"
#include "asdfsd.h"

CWinApp theApp; //创建一个应用程序对象

using namespace std;

#define PORT 34000 /// 端口号，可以自由选择>1024的端口

//传送文件函数
void SendFile(LPCTSTR strFilename)
{
AfxSocketInit(NULL); //初始化Socket
CSocket sockSrvr; //CSocket提供的是非阻塞访问方式
sockSrvr.Create(PORT); // 创建服务端的socket，默认为字节流
sockSrvr.Listen(); // 开始在端口侦听客户端的连接,它是基类CAsyncSocket的成员函数
CSocket sockRecv; //这个socket用于传送文件用
sockSrvr.Accept(sockRecv); //Use another CSocket to accept the connection

CFile myFile;
//文件访问方式为只读的二进制方式
//CFile的成员函数Open出错返回0
if(!myFile.Open(strFilename, CFile::modeRead | CFile::typeBinary))
return;

int myFileLength = myFile.GetLength(); // 发送正确的文件大小

sockRecv.Send(&myFileLength, 4); //先发个文件大小给客户端，好让它分配好堆大小

byte* data = new byte[myFileLength]; //分配byte堆，大小就是文件大小

myFile.Read(data, myFileLength); //把文件内容读入堆data中

sockRecv.Send(data, myFileLength); //把堆data中的内容发送出去

myFile.Close();
delete data; //释放堆

sockRecv.Close();
}

int _tmain(int argc, TCHAR* argv[], TCHAR* envp[])
{
int nRetCode = 0;

//初始化MFC并且当出错时输出错误串

if (!AfxWinInit(::GetModuleHandle(NULL), NULL, ::GetCommandLine(), 0))
{
// 错误代码可以改成你需要的
cerr << _T("Fatal Error: MFC initialization failed") << endl;
//_T是为了为字符编码与UNICDOE兼容，上面的_tmain也是一样的目的哦
nRetCode = 1;
}
else
SendFile("C:\\111.rar");
return nRetCode;

}

客户端代码：
-------------------------------------------

#include "stdafx.h"
#include "asdfasdfas.h"

CWinApp theApp;
using namespace std;
#define PORT 34000 //服务端口号

void GetFile(LPCTSTR strFilename)
{
AfxSocketInit(NULL);
CSocket sockClient;
sockClient.Create(); //本地端口由MFC任意指定
sockClient.Connect("127.0.0.1", PORT); // 服务器IP和它的服务端口号

int dataLength;
sockClient.Receive(&dataLength, 4); //这里的dataLength是服务端发过来的文件大小

byte* data = new byte[dataLength]; //于是乎可以分配和文件一样大小的堆
sockClient.Receive(data, dataLength); //接收全部数据

//构造CFile对象destFile,这样写连open方法都省了
CFile destFile(strFilename, CFile::modeCreate | CFile::modeWrite | CFile::typeBinary);

destFile.Write(data, dataLength); // 写入磁盘文件

destFile.Close();

delete data;
sockClient.Close();
}


int _tmain(int argc, TCHAR* argv[], TCHAR* envp[])
{
int nRetCode = 0;

// initialize MFC and print and error on failure
if (!AfxWinInit(::GetModuleHandle(NULL), NULL, ::GetCommandLine(), 0))
{
// TODO: change error code to suit your needs
cerr << _T("Fatal Error: MFC initialization failed") << endl;
nRetCode = 1;
}
else
GetFile("C:\\temp\\autoexec.bat"); //名字位置可以任意指定
return nRetCode;
}