C++ socket通信:阻塞IO(入門級)

我所記錄的是Windows下的Sosket阻塞通信方式

阻塞IO注意兩個點:

1、#include <winsock2.h>一定要在#include <windows.h>之前，否則會報錯。

2、客戶端和服務器緩存區大小需要保持一致才能夠進行數據傳輸

開發環境

vs2015

服務器端

//MiNetService.h
#pragma once
#include <winsock2.h>
#include<iostream>
#pragma comment(lib, "ws2_32.lib")  //加載 ws2_32.dll

using namespace std;
/*---------------------------------------------------------------------------
** 類名 : MiNetService
**---------------------------------------------------------------------------
** 功能 : 服務器網絡部分
**---------------------------------------------------------------------------
** Date			Name
** 2020.01.03	Mark
**---------------------------------------------------------------------------*/
class MiNetService
{
public:

	// 構造函數
	MiNetService();
	
	// 析構函數
	~MiNetService();

public:

	// 初始化網絡
	int InitNet();

	// 處理請求
	void AcceptRequest(int iFd);

	// 測試功能
	void TestNet();

};

//MiNetService.cpp
#include "MiNetService.h"


/*--------------------------------------------------------------------
** 名稱 : MiNetService
**--------------------------------------------------------------------
** 功能 : 初始化網絡服務
**--------------------------------------------------------------------
** 參數 : NULL
** 返值 : NULL
**--------------------------------------------------------------------
** Date:		Name
** 20.01.03		Mark
**-------------------------------------------------------------------*/
MiNetService::MiNetService()
{

}

/*--------------------------------------------------------------------
** 名稱 : ~MiNetService
**--------------------------------------------------------------------
** 功能 : 初始化網絡服務
**--------------------------------------------------------------------
** 參數 : NULL
** 返值 : NULL
**--------------------------------------------------------------------
** Date:		Name
** 20.01.03		Mark
**-------------------------------------------------------------------*/
MiNetService::~MiNetService()
{
}
/*--------------------------------------------------------------------
** 名稱 : InitNet
**--------------------------------------------------------------------
** 功能 : 初始化網絡服務
**--------------------------------------------------------------------
** 參數 : NULL
** 返值 : NULL
**--------------------------------------------------------------------
** Date:		Name
** 20.01.03		Mark
**-------------------------------------------------------------------*/
int MiNetService::InitNet()
{
	// 初始化化DLL
	WSADATA wsaData;
	int iResult = WSAStartup(MAKEWORD(2, 2), &wsaData);
	if(NO_ERROR != iResult)
	{
		return 0;
	}
	
	//創建套接字
	SOCKET iListenSocket = INVALID_SOCKET;
	iListenSocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
	if(INVALID_SOCKET == iListenSocket)
	{
		std::cout << "To Client: Create Socket Error!" << std::endl;
		WSACleanup();

		return 0;
	}
	
	sockaddr_in service;
	service.sin_family = AF_INET;
	service.sin_addr.s_addr = inet_addr("127.0.0.1");
	service.sin_port = htons(6666);

	iResult = bind(iListenSocket, (SOCKADDR *)&service, sizeof(service));
	if(INVALID_SOCKET == iResult)
	{
		std::cout << "To Client: Bind Socket Error!" << std::endl;
		WSACleanup();

		return 0;
	}

	int iRet = listen(iListenSocket, SOMAXCONN);
	if(0 != iRet)
	{
		return 0;
	}
	
	return iListenSocket;

}
/*--------------------------------------------------------------------
** 名稱 : AcceptRequest
**--------------------------------------------------------------------
** 功能 : 處理請求
**--------------------------------------------------------------------
** 參數 : NULL
** 返值 : NULL
**--------------------------------------------------------------------
** Date:		Name
** 20.01.03		Mark
**-------------------------------------------------------------------*/
void MiNetService::AcceptRequest(int iSeverSocketfd)
{
	if(iSeverSocketfd <0)
	{
		return;
	}

	while(true)
	{
		int sock_client = accept(iSeverSocketfd, NULL, NULL);
		int iRet = 0;
		char buffer[256];
		memset(buffer, '\0', sizeof(buffer));
		if(sock_client > 0)
		{
			iRet = recv(sock_client, buffer, 256, 0);
		}
		if(iRet > 0)
		{
			cout << "Recv:    " << buffer << endl;
		}
	}
}

/*--------------------------------------------------------------------
** 名稱 : TestNet
**--------------------------------------------------------------------
** 功能 : 網絡測試
**--------------------------------------------------------------------
** 參數 : NULL
** 返值 : NULL
**--------------------------------------------------------------------
** Date:		Name
** 20.01.03		Mark
**-------------------------------------------------------------------*/
void MiNetService::TestNet()
{
	int iServerFd = InitNet();
	if(iServerFd <0)
	{
		return;
	}

	AcceptRequest(iServerFd); 
}
// main.cpp
#include <iostream>
#include "MiNetService.h"
#include <windows.h>
using namespace std;

int main()
{
	MiNetService m_MiNetService;
	m_MiNetService.TestNet();

	system("pause");
	return 0;
}

客戶端

//MiClient.h
#pragma once
#include <winsock2.h>
#include<iostream>
#pragma comment(lib, "ws2_32.lib")  //加載 ws2_32.dll

using namespace std;

/*---------------------------------------------------------------------------
** 類名 : MiClient
**---------------------------------------------------------------------------
** 功能 : 客戶端網絡部分
**---------------------------------------------------------------------------
** Date			Name
** 2020.01.05	Mark
**---------------------------------------------------------------------------*/
class MiClient
{
public:

	// 構造函數
	MiClient();

	// 析構函數
	~MiClient();

	// 初始化網絡
	int CreateAndConnect();

	// 發送數據
	bool SendData(int Fd, const char* pszData);

public:
	// 測試函數
	void testFunc();

};


//MiClient.cpp
#include "MiClient.h"


/*--------------------------------------------------------------------
** 名稱 : MiClient
**--------------------------------------------------------------------
** 功能 : 構造函數
**--------------------------------------------------------------------
** 參數 : NULL
** 返值 : NULL
**--------------------------------------------------------------------
** Date:		Name
** 20.01.05		Mark
**-------------------------------------------------------------------*/
MiClient::MiClient()
{
}

/*--------------------------------------------------------------------
** 名稱 : ~MiClient
**--------------------------------------------------------------------
** 功能 : 析構函數
**--------------------------------------------------------------------
** 參數 : NULL
** 返值 : NULL
**--------------------------------------------------------------------
** Date:		Name
** 20.01.05		Mark
**-------------------------------------------------------------------*/
MiClient::~MiClient()
{
}
/*--------------------------------------------------------------------
** 名稱 : CreateAndConnect
**--------------------------------------------------------------------
** 功能 : 創建並連接套接字
**--------------------------------------------------------------------
** 參數 : NULL
** 返值 : NULL
**--------------------------------------------------------------------
** Date:		Name
** 20.01.05		Mark
**-------------------------------------------------------------------*/
int MiClient::CreateAndConnect()
{
	// 初始化化DLL
	WSADATA wsaData;
	int iResult = WSAStartup(MAKEWORD(2, 2), &wsaData);
	if(NO_ERROR != iResult)
	{
		return 0;
	}

	//創建套接字
	SOCKET iSocket = INVALID_SOCKET;
	iSocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
	if(INVALID_SOCKET == iSocket)
	{
		std::cout << "To Client: Create Socket Error!" << std::endl;
		WSACleanup();

		return 0;
	}

	sockaddr_in clientService;
	clientService.sin_family = AF_INET;
	clientService.sin_addr.s_addr = inet_addr("127.0.0.1");
	clientService.sin_port = htons(6666);

	// Connect to server.
	iResult = connect(iSocket, (SOCKADDR *)& clientService, sizeof(clientService));
	if(iResult == SOCKET_ERROR) 
	{
		wprintf(L"connect function failed with error: %ld\n", WSAGetLastError());
		iResult = closesocket(iSocket);
		if (iResult == SOCKET_ERROR)
			wprintf(L"closesocket function failed with error: %ld\n", WSAGetLastError());
		WSACleanup();
		return 1;
	}

	return iSocket;
}

/*--------------------------------------------------------------------
** 名稱 : SendData
**--------------------------------------------------------------------
** 功能 : 發送數據
**--------------------------------------------------------------------
** 參數 : NULL
** 返值 : NULL
**--------------------------------------------------------------------
** Date:		Name
** 20.01.05		Mark
**-------------------------------------------------------------------*/
bool MiClient::SendData(int Fd, const char* pszData)
{
	int iResult = send(Fd, pszData, (int)strlen(pszData), 0);

	if(iResult == SOCKET_ERROR) 
	{
		wprintf(L"send failed with error: %d\n", WSAGetLastError());
		closesocket(Fd);
		WSACleanup();
		return 1;
	}

	return true;
}
/*--------------------------------------------------------------------
** 名稱 : testFunc
**--------------------------------------------------------------------
** 功能 : 測試函數
**--------------------------------------------------------------------
** 參數 : NULL
** 返值 : NULL
**--------------------------------------------------------------------
** Date:		Name
** 20.01.05		Mark
**-------------------------------------------------------------------*/
void MiClient::testFunc()
{
	int iSocket = CreateAndConnect(); // 初始化網絡
	if(iSocket <=0)
	{
		return;
	}

	char temp[512] = "Hello";
	SendData(iSocket, temp);

	WSACleanup();
}

/**
 * If no error occurs, connect returns zero. Otherwise, it returns SOCKET_ERROR, 
 */
// main.cpp
#include <iostream>
#include "MiClient.h"
#include <windows.h>
using namespace std;

int main()
{
	MiClient m_client;

	m_client.testFunc();

	
	system("pause");
	return 0;
}