TCP 协议文件传输示例

一个简单的传输的例子 单线程 阻塞模式.  

 以下是发送端代码

send.cpp

//---------------------------------------------------------------------------

#include <iostream>

#include "winsock2.h"

using namespace std;



#pragma comment(lib , "ws2_32.lib")

//---------------------------------------------------------------------------



WSADATA wsaData;

UINT uTimerID;

DWORD MAX_SIZE = 1024 * 4;

DWORD FinishSize = 0;

DWORD SendSize = MAX_SIZE;

LPSTR SFileName;

DWORD FSize;

DWORD DWSpeed=0;

DWORD OldFinish=0;



MMRESULT (WINAPI *ptimeSetEvent)(UINT uDelay, UINT uResolution, LPTIMECALLBACK fptc, DWORD dwUser, UINT fuEvent);

MMRESULT (WINAPI *ptimeKillEvent)(UINT uTimerID);

HINSTANCE DLLInst;



bool init()

{

    DLLInst=LoadLibrary("winmm.dll");

    if(DLLInst)

    {

        ptimeSetEvent = (MMRESULT (WINAPI *)(UINT uDelay, UINT uResolution, LPTIMECALLBACK fptc, DWORD dwUser, UINT fuEvent))GetProcAddress(DLLInst, "timeSetEvent");

        if(!ptimeSetEvent) return false;

        ptimeKillEvent = (MMRESULT (WINAPI *)(UINT uTimerID))GetProcAddress(DLLInst, "timeKillEvent");

        if(!ptimeKillEvent) return false;

        return true;

    }

    return false;

}



void Byte2Human(const DWORD & byte)

{

    if(byte < 1024){

        cout << ((float)((int)(byte*100+0.5))/100.0) << "B";

        return;

    }

    float kb = byte/1024;

    if (kb<1024){

        cout << ((float)((int)(kb*100+0.5))/100.0) << "KB";

        return ;

    }

    float mb = kb/1024;

    if (mb<1024){

        cout << ((float)((int)(mb*100+0.5))/100.0) << "MB";

        return;

    }

    float gb = mb/1024;

    cout << ((float)((int)(gb*100+0.5))/100.0) << "GB";

}



void WINAPI TimeProc(UINT uID, UINT uMsg, DWORD dwUser, DWORD dw1, DWORD dw2)

{

    DWSpeed = FinishSize-OldFinish;

    OldFinish = FinishSize;

    cout << '/r';

    cout << "                                                                              ";

    cout << '/r';

    cout << "文件名:" << SFileName << " " << FinishSize << "/" << FSize << " ";

    Byte2Human(DWSpeed);

    cout << "/S";

}



int main(int argc, char* argv[])

{

    if(!init())

    {

        cout<<"初始化失败"<<endl;

        return 0;

    }

    int ret;

    if(argc < 4)

    {

        cout << "App.exe Host Port FileName" << endl;

        return 0;

    }



    if((ret=WSAStartup(0x0202, &wsaData))!=0)

    {

        cout << "WSAStartup Failed!! Code:" << ret << endl;

        return 0;

    }

    SOCKET sock;

    struct sockaddr_in server;

    sock=socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

    if(sock==INVALID_SOCKET)

    {

        cout << "创建socket出错" << endl;

        return 0;

    }

    server.sin_family=AF_INET;

    server.sin_port=htons(atoi(argv[2]));

    server.sin_addr.s_addr=inet_addr(argv[1]);

    cout << "正在连接 " << argv[1] << ":" << argv[2] << endl;

    if(connect(sock, (struct sockaddr *)&server, sizeof(server))==SOCKET_ERROR)

    {

        cout << "连接失败" << endl;

        return 0;

    }

    cout << "连接成功" << endl;



    //打开文件准备发送

    HANDLE hFile;

    OFSTRUCT ofs;

    HANDLE hMap;

    void *pVoid;

    SFileName = argv[3];

    hFile=(HANDLE)OpenFile(argv[3], &ofs, OF_READWRITE | OF_SHARE_DENY_NONE);

    if((HFILE)hFile==HFILE_ERROR)

    {

        cout << "打开文件失败 ErrorCode=" << GetLastError() << endl;

        return 0;

    }

    FSize=GetFileSize((HANDLE)hFile, NULL);

    hMap=CreateFileMapping(hFile, NULL, PAGE_READWRITE, 0, 0, NULL);

    if(hMap==NULL)

    {

        cout << "打开映射失败 ErrorCode=" << GetLastError() << endl;

        return 0;

    }

    pVoid=MapViewOfFile(hMap, FILE_MAP_ALL_ACCESS, 0, 0, 0);

    if(pVoid==NULL)

    {

        cout << "映射数据失败 ErrorCode=" << GetLastError() << endl;

        return 0;

    }

    char *p = (char *)pVoid;

    cout << "正在发送文件 " << argv[3] << endl;

    uTimerID=ptimeSetEvent(1000, 0, (LPTIMECALLBACK)TimeProc, NULL, TIME_PERIODIC|TIME_CALLBACK_FUNCTION);



    while(1)

    {

        if(FSize - FinishSize < MAX_SIZE)

        {

            SendSize = FSize - FinishSize;

        }

        int ret = send(sock, p+FinishSize, SendSize, 0);

        if(ret == SOCKET_ERROR )

        {

            cout << endl << "Ret=" << ret << " ErrorCode=" << WSAGetLastError() << endl;

            return 0;

        }

        FinishSize += ret;

        if(FinishSize == FSize) break;

    }

    closesocket(sock);

    ptimeKillEvent(uTimerID);

    CloseHandle(hMap);

    CloseHandle(hFile);

    cout << endl;

    cout << "完成 发送数据量:" << FinishSize << endl;

    return 0;

}

//---------------------------------------------------------------------------

//---------------------------------------------------------------------------

#include <iostream>

#include "winsock2.h"

using namespace std;



#pragma comment(lib , "ws2_32.lib")

//---------------------------------------------------------------------------



WSADATA wsaData;

UINT uTimerID;

DWORD MAX_SIZE = 1024 * 4;

DWORD FinishSize = 0;

DWORD SendSize = MAX_SIZE;

LPSTR SFileName;

DWORD FSize;

DWORD DWSpeed=0;

DWORD OldFinish=0;



MMRESULT (WINAPI *ptimeSetEvent)(UINT uDelay, UINT uResolution, LPTIMECALLBACK fptc, DWORD dwUser, UINT fuEvent);

MMRESULT (WINAPI *ptimeKillEvent)(UINT uTimerID);

HINSTANCE DLLInst;



bool init()

{

	DLLInst=LoadLibrary("winmm.dll");

	if(DLLInst)

    {

		ptimeSetEvent = (MMRESULT (WINAPI *)(UINT uDelay, UINT uResolution, LPTIMECALLBACK fptc, DWORD dwUser, UINT fuEvent))GetProcAddress(DLLInst, "timeSetEvent");

		if(!ptimeSetEvent) return false;

		ptimeKillEvent = (MMRESULT (WINAPI *)(UINT uTimerID))GetProcAddress(DLLInst, "timeKillEvent");

		if(!ptimeKillEvent) return false;

		return true;

	}

	return false;

}



void Byte2Human(const DWORD & byte)

{

	if(byte < 1024){

		cout << ((float)((int)(byte*100+0.5))/100.0) << "B";

		return;

	}

	float kb = byte/1024;

	if (kb<1024){

		cout << ((float)((int)(kb*100+0.5))/100.0) << "KB";

		return ;

	}

	float mb = kb/1024;

	if (mb<1024){

		cout << ((float)((int)(mb*100+0.5))/100.0) << "MB";

		return;

	}

	float gb = mb/1024;

	cout << ((float)((int)(gb*100+0.5))/100.0) << "GB";

}



void WINAPI TimeProc(UINT uID, UINT uMsg, DWORD dwUser, DWORD dw1, DWORD dw2)

{

	DWSpeed = FinishSize-OldFinish;

	OldFinish = FinishSize;

	cout << '/r';

 	cout << "                                                                              ";

	cout << '/r';

	cout << "文件名:" << SFileName << " " << FinishSize << "/" << FSize << " ";

	Byte2Human(DWSpeed);

	cout << "/S";

}



int main(int argc, char* argv[])

{

	if(!init())

	{

		cout<<"初始化失败"<<endl;

		return 0;

	}

	int ret;

	if(argc < 4)

	{

		cout << "App.exe Host Port FileName" << endl;

		return 0;

	}



	if((ret=WSAStartup(0x0202, &wsaData))!=0)

	{

		cout << "WSAStartup Failed!! Code:" << ret << endl;

		return 0;

	}

	SOCKET sock;

	struct sockaddr_in server;

	sock=socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

	if(sock==INVALID_SOCKET)

	{

		cout << "创建socket出错" << endl;

		return 0;

	}

	server.sin_family=AF_INET;

	server.sin_port=htons(atoi(argv[2]));

	server.sin_addr.s_addr=inet_addr(argv[1]);

	cout << "正在连接 " << argv[1] << ":" << argv[2] << endl;

	if(connect(sock, (struct sockaddr *)&server, sizeof(server))==SOCKET_ERROR)

	{

		cout << "连接失败" << endl;

		return 0;

	}

	cout << "连接成功" << endl;



	//打开文件准备发送

	HANDLE hFile;

	OFSTRUCT ofs;

	HANDLE hMap;

	void *pVoid;

	SFileName = argv[3];

	hFile=(HANDLE)OpenFile(argv[3], &ofs, OF_READWRITE | OF_SHARE_DENY_NONE);

	if((HFILE)hFile==HFILE_ERROR)

	{

		cout << "打开文件失败 ErrorCode=" << GetLastError() << endl;

		return 0;

	}

	FSize=GetFileSize((HANDLE)hFile, NULL);

	hMap=CreateFileMapping(hFile, NULL, PAGE_READWRITE, 0, 0, NULL);

	if(hMap==NULL)

	{

		cout << "打开映射失败 ErrorCode=" << GetLastError() << endl;

		return 0;

	}

	pVoid=MapViewOfFile(hMap, FILE_MAP_ALL_ACCESS, 0, 0, 0);

	if(pVoid==NULL)

	{

		cout << "映射数据失败 ErrorCode=" << GetLastError() << endl;

		return 0;

	}

	char *p = (char *)pVoid;

	cout << "正在发送文件 " << argv[3] << endl;

	uTimerID=ptimeSetEvent(1000, 0, (LPTIMECALLBACK)TimeProc, NULL, TIME_PERIODIC|TIME_CALLBACK_FUNCTION);



	while(1)

	{

		if(FSize - FinishSize < MAX_SIZE)

		{

			SendSize = FSize - FinishSize;

		}

		int ret = send(sock, p+FinishSize, SendSize, 0);

		if(ret == SOCKET_ERROR )

		{

			cout << endl << "Ret=" << ret << " ErrorCode=" << WSAGetLastError() << endl;

			return 0;

		}

		FinishSize += ret;

		if(FinishSize == FSize) break;

	}

	closesocket(sock);

	ptimeKillEvent(uTimerID);

	CloseHandle(hMap);

	CloseHandle(hFile);

	cout << endl;

	cout << "完成 发送数据量:" << FinishSize << endl;

	return 0;

}

//---------------------------------------------------------------------------

以下是接收端代码

以下是接收端代码

recv.cpp<PRE class=csharp name="code">//---------------------------------------------------------------------------

#include <iostream>
#include "winsock2.h"
using namespace std;</PRE>
<PRE class=csharp name="code">
#pragma comment(lib , "ws2_32.lib")

//---------------------------------------------------------------------------</PRE>
<PRE class=csharp name="code">

WSADATA wsaData;

DWORD FinishSize = 0;

DWORD OldFinish=0;

DWORD DWSpeed=0;

UINT uTimerID;



MMRESULT (WINAPI *ptimeSetEvent)(UINT uDelay, UINT uResolution, LPTIMECALLBACK fptc, DWORD dwUser, UINT fuEvent);

MMRESULT (WINAPI *ptimeKillEvent)(UINT uTimerID);

HINSTANCE DLLInst;



bool init()

{

    DLLInst=LoadLibrary("winmm.dll");

    if(DLLInst)

    {

        ptimeSetEvent = (MMRESULT (WINAPI *)(UINT uDelay, UINT uResolution, LPTIMECALLBACK fptc, DWORD dwUser, UINT fuEvent))GetProcAddress(DLLInst, "timeSetEvent");

        if(!ptimeSetEvent) return false;

        ptimeKillEvent = (MMRESULT (WINAPI *)(UINT uTimerID))GetProcAddress(DLLInst, "timeKillEvent");

        if(!ptimeKillEvent) return false;

        return true;

    }

    return false;

}



void Byte2Human(const DWORD & byte)

{

    if(byte < 1024){

        cout << ((float)((int)(byte*100+0.5))/100.0) << "B";

        return;

    }

    float kb = byte/1024;

    if (kb<1024){

        cout << ((float)((int)(kb*100+0.5))/100.0) << "KB";

        return ;

    }

    float mb = kb/1024;

    if (mb<1024){

        cout << ((float)((int)(mb*100+0.5))/100.0) << "MB";

        return;

    }

    float gb = mb/1024;

    cout << ((float)((int)(gb*100+0.5))/100.0) << "GB";

}



void CALLBACK TimeProc(UINT uID, UINT uMsg, DWORD dwUser, DWORD dw1, DWORD dw2)

{

    DWSpeed = FinishSize-OldFinish;

    OldFinish = FinishSize;

    cout << '/r';

    cout << "                                                                              ";

    cout << '/r';

    cout << "已接收 " << FinishSize << " ";

    Byte2Human(DWSpeed);

    cout << "/S";

}



int main(int argc, char* argv[])

{

    if(!init())

    {

        cout<<"初始化失败"<<endl;

        return 0;

    }



    int ret;

    if(argc < 3)

    {

        cout << "App.exe Port NewFileName" << endl;

        return 0;

    }



    HANDLE hFile;

    OFSTRUCT ofs;

    HANDLE hMap;

    void *pVoid;

    hFile=(HANDLE)OpenFile(argv[2], &ofs, OF_CREATE | OF_READ);

    if((HFILE)hFile==HFILE_ERROR)

    {

        cout << "打开文件失败";

        ret = GetLastError();

        cout << " ErrorCode=" << ret << endl;

        return 0;

    }



    if((ret=WSAStartup(0x0202, &wsaData))!=0)

    {

        cout << "WSAStartup Failed!! Code:" << ret << endl;

        return 0;

    }



    SOCKET sock, ssock;

    struct sockaddr_in fsin, sin;

    int alen;



    memset(&sin, 0, sizeof(sin));

    sin.sin_family=AF_INET;

    sin.sin_addr.s_addr=inet_addr("");

    sin.sin_port=htons(atoi(argv[1]));

    sock=socket(PF_INET, SOCK_STREAM, 0);

    if(sock==INVALID_SOCKET)

    {

        cout << "创建socket出错" << endl;

        return 0;

    }

    if(bind(sock, (struct sockaddr*)&sin, sizeof(sin))==SOCKET_ERROR)

    {

        cout << "帮定Socket失败" << endl;

        return 0;

    }

    if(listen(sock, SOMAXCONN)==SOCKET_ERROR)

    {

        cout << "监听失败" << endl;

        return 0;

    }

    cout << "正在监听... 端口" << argv[1] << endl;

    alen=sizeof(struct sockaddr);

    ssock=accept(sock, (struct sockaddr *)&fsin, &alen);

    if(ssock==INVALID_SOCKET)

    {

        cout << "接受请求失败" << endl;

        return 0;

    }

    closesocket(sock);

    char buf[4096];

    DWORD dwWrite;

    cout << "连接已建立" << endl;

    uTimerID=ptimeSetEvent(1000, 0, (LPTIMECALLBACK)TimeProc, NULL, TIME_PERIODIC|TIME_CALLBACK_FUNCTION);



    while(1)

    {

        memset(buf, 0x00, 4096);

        ret = recv(ssock, buf, 4096, 0);

        if(ret == SOCKET_ERROR )

        {

            cout << endl << "接收数据量:" << FinishSize << endl;

            cout << "Ret=" << ret;

            ret = WSAGetLastError();

            cout << " ErrorCode=" << ret << endl;

            return 0;

        }

        if(ret == 0)

        {

            cout << endl << "连接已断开" << endl;

            break;

        }

        FinishSize += ret;

        if(!WriteFile(hFile, buf, ret, &dwWrite, NULL))

        {

            cout << "写文件失败" << endl;

            return 0;

        }



    }

    ptimeKillEvent(uTimerID);

    closesocket(sock);

    CloseHandle(hFile);

    cout << endl;

    cout << "完成 接收数据量:" << FinishSize << endl;

    return 0;

}

//---------------------------------------------------------------------------