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;
}