網絡編程——C++實現socket通信(TCP)

相關鏈接：TCP連接與釋放

相關函數：

服務端：
socket()
bind()
listen()
accept()
read() 或 recv()等
write() 或 send()等
close()

客戶端：
socket()
connect()
write() 或 send()等
read() 或 recv()等
close()

注意：每當服務端連接斷開後，進入TIME_WAIT狀態，等待2msl時間之後才能重新使用IP和端口，否則在bind時就會報錯。要解決這個問題可以在程序開始時調用端口複用函數setsockopt。原型如下：

//int setsockopt(int sockfd, int level, int optname,const void *optval, socklen_t optlen);
    /*　sockfd：標識一個套接口的描述字。
　　    level：選項定義的層次；支持SOL_SOCKET、IPPROTO_TCP、IPPROTO_IP和IPPROTO_IPV6。
　　    optname：需設置的選項。
　　    optval：指針，指向存放選項值的緩衝區
　　    optlen：optval緩衝區長度。
　　    返回值：  成功返回0，失敗返回 -1.  */
　　    

實際調用：
 setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));

廢話不多說，上源碼！

實現的功能：客戶端C向服務端S發送一串字符數據，S端會對字符串做轉大寫操作然後回發給C端

服務端TCP_Server.cpp

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <netinet/in.h>
#include <ctype.h>

#define MAXSIZE 1024
#define IP_ADDR "127.0.0.1"
#define IP_PORT 8888

int main()
{
	int i_listenfd, i_connfd;
	struct sockaddr_in st_sersock;
	char msg[MAXSIZE];
	int nrecvSize = 0;

	if((i_listenfd = socket(AF_INET, SOCK_STREAM, 0) ) < 0)	//建立socket套接字
	{
		printf("socket Error: %s (errno: %d)\n", strerror(errno), errno);
		exit(0);
	}

	memset(&st_sersock, 0, sizeof(st_sersock));
	st_sersock.sin_family = AF_INET;  //IPv4協議
	st_sersock.sin_addr.s_addr = htonl(INADDR_ANY);	//INADDR_ANY轉換過來就是0.0.0.0，泛指本機的意思，也就是表示本機的所有IP，因爲有些機子不止一塊網卡，多網卡的情況下，這個就表示所有網卡ip地址的意思。
	st_sersock.sin_port = htons(IP_PORT);

	if(bind(i_listenfd,(struct sockaddr*)&st_sersock, sizeof(st_sersock)) < 0) //將套接字綁定IP和端口用於監聽
	{
		printf("bind Error: %s (errno: %d)\n", strerror(errno), errno);
		exit(0);
	}

	if(listen(i_listenfd, 20) < 0)	//設定可同時排隊的客戶端最大連接個數
	{
		printf("listen Error: %s (errno: %d)\n", strerror(errno), errno);
		exit(0);
	}

	printf("======waiting for client's request======\n");
	//準備接受客戶端連接
	{
		if((i_connfd = accept(i_listenfd, (struct sockaddr*)NULL, NULL)) < 0)	//阻塞等待客戶端連接
		{
			printf("accept Error: %s (errno: %d)\n", strerror(errno), errno);
		//	continue;
		}	
		else
		{
			printf("Client[%d], welcome!\n", i_connfd);
		}


		while(1)	//循環 接受客戶端發來的消息並作處理(小寫轉大寫)後回寫給客戶端
		{
			memset(msg, 0 ,sizeof(msg));
			if((nrecvSize = read(i_connfd, msg, MAXSIZE)) < 0)
			{
				printf("accept Error: %s (errno: %d)\n", strerror(errno), errno);
				continue;
			}
			else if( nrecvSize == 0)	//read返回0代表對方已close斷開連接。
			{
				printf("client has disconnected!\n");
				close(i_connfd);  //
				break;
			}
			else
			{
				printf("recvMsg:%s", msg);
				for(int i=0; msg[i] != '\0'; i++)
				{
					msg[i] = toupper(msg[i]);
				}
				if(write(i_connfd, msg, strlen(msg)+1) < 0)
				{
					printf("accept Error: %s (errno: %d)\n", strerror(errno), errno);
				}

			}
		}
	}
	close(i_connfd);
	close(i_listenfd);

	return 0;
}

客戶端TCP_Client.cpp

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <netinet/in.h>
#include <signal.h>
#include <arpa/inet.h>

#define MAXSIZE 1024
#define IP_ADDR "127.0.0.1"
#define IP_PORT 8888

int i_sockfd = -1;

void SigCatch(int sigNum)	//信號捕捉函數(捕獲Ctrl+C)
{
	if(i_sockfd != -1)
	{
		close(i_sockfd);
	}
	printf("Bye~! Will Exit...\n");
	exit(0);
}

int main()
{
	struct sockaddr_in st_clnsock;
	char msg[1024];
	int nrecvSize = 0;

	signal(SIGINT, SigCatch);	//註冊信號捕獲函數

	if((i_sockfd = socket(AF_INET, SOCK_STREAM, 0) ) < 0)	//建立套接字
	{
		printf("socket Error: %s (errno: %d)\n", strerror(errno), errno);
		exit(0);
	}

	memset(&st_clnsock, 0, sizeof(st_clnsock));
	st_clnsock.sin_family = AF_INET;  //IPv4協議
	//IP地址轉換(直接可以從物理字節序的點分十進制 轉換成網絡字節序)
	if(inet_pton(AF_INET, IP_ADDR, &st_clnsock.sin_addr) <= 0)
	{
		printf("inet_pton Error: %s (errno: %d)\n", strerror(errno), errno);
		exit(0);
	}
	st_clnsock.sin_port = htons(IP_PORT);	//端口轉換(物理字節序到網絡字節序)

	if(connect(i_sockfd, (struct sockaddr*)&st_clnsock, sizeof(st_clnsock)) < 0)	//主動向設置的IP和端口號的服務端發出連接
	{
		printf("connect Error: %s (errno: %d)\n", strerror(errno), errno);
		exit(0);
	}

	printf("======connect to server, sent data======\n");

	while(1)	//循環輸入，向服務端發送數據並接受服務端返回的數據
	{
		fgets(msg, MAXSIZE, stdin);
		printf("will send: %s", msg);
		if(write(i_sockfd, msg, MAXSIZE) < 0)	//發送數據
		{
			printf("write Error: %s (errno: %d)\n", strerror(errno), errno);
			exit(0);
		}

		memset(msg, 0, sizeof(msg));
		if((nrecvSize = read(i_sockfd, msg, MAXSIZE)) < 0)	//接受數據
		{
			printf("read Error: %s (errno: %d)\n", strerror(errno), errno);
		}
		else if(nrecvSize == 0)
		{
			printf("Service Close!\n");
		}
		else
		{
			printf("Server return: %s\n", msg);
		}

	}
	return 0;
}