[VC++] socket网络传输
不研究TCP/IP协议,没有三次握手,不说 物数网传会表应(只要大学学过计算机基础的都背过吧!)这七层。
封装好socket。
server:
class ServerSock { public: ServerSock(); ~ServerSock(); SOCKET getClient() {return sClient;}; private: static const int PORT = 10086; int retVal; //返回值 SOCKET sServer; //服务器套接字 SOCKET sClient; //客户端套接字 SOCKADDR_IN addrServ;; //服务器地址 }; ServerSock::ServerSock() { WSADATA wsd; //WSADATA变量 //初始化套结字动态库 if (WSAStartup(MAKEWORD(2,2), &wsd) != 0) { cout << "WSAStartup failed!" << endl; } //创建套接字 sServer = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); if(INVALID_SOCKET == sServer) { cout << "socket failed!" << endl; WSACleanup();//释放套接字资源; } //服务器套接字地址 addrServ.sin_family = AF_INET; addrServ.sin_port = htons(PORT); addrServ.sin_addr.s_addr = INADDR_ANY; //绑定套接字 retVal = bind(sServer, (LPSOCKADDR)&addrServ, sizeof(SOCKADDR_IN)); if(SOCKET_ERROR == retVal) { cout << "bind failed!" << endl; closesocket(sServer); //关闭套接字 WSACleanup(); //释放套接字资源; } //开始监听 cout<<"start listen!"<<endl; retVal = listen(sServer, 1); if(SOCKET_ERROR == retVal) { cout << "listen failed!" << endl; closesocket(sServer); //关闭套接字 WSACleanup(); //释放套接字资源; } //接受客户端请求 sockaddr_in addrClient; int addrClientlen = sizeof(addrClient); sClient = accept(sServer,(sockaddr FAR*)&addrClient, &addrClientlen); cout<<"accept!"<<endl; if(INVALID_SOCKET == sClient) { cout << "accept failed!" << endl; closesocket(sServer); //关闭套接字 WSACleanup(); //释放套接字资源; } } ServerSock::~ServerSock() { closesocket(sServer); //关闭套接字 //closesocket(sClient); //关闭套接字 WSACleanup(); //释放套接字资源; }
client:
class ClientSock { public: ClientSock(); ~ClientSock(); SOCKET getClient() {return cClient;}; private: static const int PORT = 10086; int retVal; //返回值 SOCKET cClient; //客户端套接字 SOCKADDR_IN addrServ;; //服务器地址 }; ClientSock::ClientSock() { WSADATA wsd; //WSADATA变量 //初始化套结字动态库 if (WSAStartup(MAKEWORD(2,2), &wsd) != 0) { cout << "WSAStartup failed!" << endl; } //创建套接字 cClient = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); if(INVALID_SOCKET == cClient) { cout << "socket failed!" << endl; WSACleanup();//释放套接字资源; } //服务器套接字地址 addrServ.sin_family = AF_INET; addrServ.sin_port = htons(PORT); addrServ.sin_addr.s_addr = inet_addr(IP_ADDR); //绑定套接字 retVal = connect(cClient, (LPSOCKADDR)&addrServ, sizeof(SOCKADDR_IN)); if(SOCKET_ERROR == retVal) { cout << "connect failed!" << endl; closesocket(cClient); //关闭套接字 WSACleanup(); //释放套接字资源; } } ClientSock::~ClientSock() { //closesocket(cClient); //关闭套接字 WSACleanup(); //释放套接字资源; }
关于数据部分,两个是通用的
class SockData { public: SockData(SOCKET _client) { client = _client;}; ~SockData() { closesocket(client); }; bool sendData(const char sData); bool sendData(int sData); bool sendData(char* sData,int Len = BUFSIZE); bool sendData(const string sfileRoad); int recvData(); bool recvData(char* buf, int Len = BUFSIZE); bool recvData(const string rfileRoad); private: static const int BUFSIZE = 1024; //默认大小 int retVal; //返回值 SOCKET client; //客户端套接字 }; bool SockData::sendData(const char sData) { retVal = send(client,&sData,sizeof(char),0); if(retVal==SOCKET_ERROR) { cout<<"Send Into error: "<<GetLastError()<<endl; cout<<"Fail char: "<<sData<<endl; return false; } return true; } bool SockData::sendData(int sData) { stringstream strTemp;strTemp.clear();strTemp.str(""); strTemp << sData; const string sTemp = strTemp.str(); const char* cTemp = sTemp.c_str(); if(! sendData(char('0'+strlen(cTemp)))) { cout<<"In trouble"<<endl; } retVal = send(client,cTemp,sizeof(char) * strlen(cTemp),0); if(retVal==SOCKET_ERROR) { cout<<"Send Into error: "<<GetLastError()<<endl; cout<<"Fail int: "<<sData<<endl; return false; } return true; } bool SockData::sendData(char* sData,int Len) { retVal=send(client,sData,sizeof(char) * Len,0); if(retVal==SOCKET_ERROR) { cout<<"Send Into error: "<<GetLastError()<<endl; return false; } return true; } bool SockData::sendData(const string sfileRoad) { ifstream in(sfileRoad,fstream::in|fstream::binary); if(!in.is_open()) { cout<<"fail open file."<<endl; return false; } in.seekg (0, ios::end); int length = in.tellg(); in.seekg (0, ios::beg); sendData(length); char* buf = new char[BUFSIZE]; //接收数据缓冲区 ZeroMemory(buf, BUFSIZE); while(!in.eof()) { in.read(buf,BUFSIZE); retVal=send(client,buf,sizeof(char) * BUFSIZE,0); if(retVal==SOCKET_ERROR) { cout<<"Send Into error: "<<GetLastError()<<endl; return false; } } in.close(); delete[] buf; return true; } int SockData::recvData() { char intBit; retVal = recv(client,&intBit,sizeof(char),0); if(retVal == SOCKET_ERROR) { cout<<"Recv Into error: "<<GetLastError()<<endl; cout<<"Fail intBit!"<<endl; return false; } int Bit = atoi(&intBit); char* cInt = new char[Bit]; retVal = recv(client,cInt,sizeof(char) * Bit,0); if(retVal == SOCKET_ERROR) { cout<<"Recv Into error: "<<GetLastError()<<endl; cout<<"Fail intBit!"<<endl; return false; } int rData = atoi(cInt); delete[] cInt; return rData; } bool SockData::recvData(char* buf, int Len) { ZeroMemory(buf,Len); if(Len <= BUFSIZE) { retVal = recv(client,buf,sizeof(char) * Len,0); if(retVal == SOCKET_ERROR) { cout<<"Recv Into error: "<<GetLastError()<<endl; cout<<"Fail char*!"<<endl; return false; } } else { int rbufReal = 0, rbufNeed = Len; while (rbufReal < Len) { retVal = recv(client,buf,sizeof(char) * BUFSIZE,0); if(retVal == SOCKET_ERROR) { cout<<"Recv Into error: "<<GetLastError()<<endl; cout<<"Fail char*!"<<endl; return false; } rbufReal += retVal; rbufNeed = Len - rbufReal; } } return true; } bool SockData::recvData(const string rfileRoad) { int length = recvData(); char* buf = new char[BUFSIZE]; ZeroMemory(buf, BUFSIZE); ofstream out(rfileRoad,fstream::out|fstream::binary); if(!out.is_open()) { cout<<"don't create file."<<endl; return false; } int rbufReal = 0, rbufNeed = length; while (rbufReal < length) { if(recvData(buf)) { rbufReal += retVal; rbufNeed = length - rbufReal; out.write(buf,sizeof(char)*retVal); } } out.close(); delete[] buf; return true; }
代码托管:https://github.com/zhanxiage1994/VCplusSocket.git