C++ 简单的 Tcp 实现[socket] 服务器端与客户端通信

#include "stdafx.h"
#include "winsock2.h"
#pragma comment(lib, "ws2_32.lib")
#include <iostream>
using namespace std;
int TestServer();
int TestClient();

const int BUF_SIZE = 64;

SOCKET sServer;					//服务器套接字
SOCKET sClient;					//客户端套接字
WSADATA wsd;					//WSADATA变量
SOCKADDR_IN servAddr;			//服务器地址
char bufRecv[BUF_SIZE] = { 0 }; //接收数据缓冲区
char sendBuf[BUF_SIZE];			//发送给客户端得数据
int retVal;						//返回值

int _tmain(int argc, _TCHAR* argv[])
{
//初始化套结字动态库
if (WSAStartup(MAKEWORD(2, 2), &wsd) != 0)
{
cout << " WSAStartup failed " << endl;
return -1;
}

//创建套接字
sServer = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (INVALID_SOCKET == sServer)
{
cout << " socket failed " << endl;
WSACleanup();//释放套接字资源;
return  -1;
}

int x;
cout << "请选择1服务器,2客户端:";
cin >> x;

switch (x)
{
case 1:
TestServer();
break;
case 2:
TestClient();
break;
}
return 0;
}

int TestServer(){

//服务器套接字地址
servAddr.sin_family = AF_INET;
servAddr.sin_port = htons(4999);
servAddr.sin_addr.s_addr = INADDR_ANY;
//绑定套接字
retVal = bind(sServer, (LPSOCKADDR)&servAddr, sizeof(SOCKADDR_IN));

if (SOCKET_ERROR == retVal)
{
cout <<" bind failed"<<endl;
closesocket(sServer); //关闭套接字
WSACleanup(); //释放套接字资源;
return -1;
}
//开始监听
retVal = listen(sServer, 1);
if (SOCKET_ERROR == retVal)
{
cout <<" listen failed "<<endl;
closesocket(sServer); //关闭套接字
WSACleanup(); //释放套接字资源;
return -1;
}
//接受客户端请求
sockaddr_in addrClient;
int addrClientlen = sizeof(addrClient);
sClient = accept(sServer, (SOCKADDR *)&addrClient, &addrClientlen);
if (INVALID_SOCKET == sClient)
{
cout <<" accept failed "<<endl;
closesocket(sServer); //关闭套接字
WSACleanup(); //释放套接字资源;
return -1;
}

while (true){
//接收客户端数据
ZeroMemory(bufRecv, BUF_SIZE);
retVal = recv(sClient, bufRecv, BUF_SIZE, 0);
if (SOCKET_ERROR == retVal)
{
cout <<" recv failed"<<endl;
closesocket(sServer); //关闭套接字
closesocket(sClient); //关闭套接字
WSACleanup(); //释放套接字资源;
return -1;
}
if (bufRecv[0] == '0')
break;
cout << "Peer:" << bufRecv << endl;
cout << "I Speak:";
cin >> sendBuf;
send(sClient, sendBuf, strlen(sendBuf), 0);
}
//退出
closesocket(sServer); //关闭套接字
closesocket(sClient); //关闭套接字
WSACleanup(); //释放套接字资源;

}
int TestClient(){

//设置服务器地址
servAddr.sin_family = AF_INET;
servAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
servAddr.sin_port = htons((short)4999);
int nServAddlen = sizeof(servAddr);

//连接服务器
retVal = connect(sServer, (LPSOCKADDR)&servAddr, sizeof(servAddr));
if (SOCKET_ERROR == retVal)
{
cout << " connect failed " << endl;
closesocket(sServer); //关闭套接字
WSACleanup(); //释放套接字资源
return -1;
}
while (true){
//向服务器发送数据
ZeroMemory(bufRecv, BUF_SIZE);
cout <<"I Speak:";
cin >> sendBuf;
retVal = send(sServer, sendBuf, strlen(sendBuf), 0);
if (SOCKET_ERROR == retVal)
{
cout << " send failed " << endl;
closesocket(sServer); //关闭套接字
WSACleanup(); //释放套接字资源
return -1;
}
//RecvLine(sHost, bufRecv);
recv(sServer, bufRecv, BUF_SIZE, 0); // 接收服务器端的数据， 只接收 BUF_SIZE 个字符
cout <<"Peer:"<< bufRecv << endl;

}
//退出
closesocket(sServer); //关闭套接字
WSACleanup(); //释放套接字资源
return 0;
}