一个简单的传输的例子 单线程 阻塞模式.
以下是发送端代码
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;
- }
- //---------------------------------------------------------------------------