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