Linux网络编程——TCP&UDP编程流程（二）

TCP编程实例

本例包括两部分代码，分别为客户端和服务器端，主要实现了客服端向服务器端发送数据，服务器端接收并打印的简单功能，以此来展示TCP编程的一个流程

服务器端代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include <unistd.h>

#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>


int main()
{
	// 1
	int sockfd = socket(AF_INET, SOCK_STREAM, 0);
	assert(sockfd != -1);

	// 手动输入地址结构
	struct sockaddr_in ser_addr;
	memset(&ser_addr, 0, sizeof(ser_addr));
	ser_addr.sin_family = AF_INET;
	ser_addr.sin_port = htons(6000);  //必须得使用网络字节序 htons
	ser_addr.sin_addr.s_addr = inet_addr("127.0.0.1");

	// 2
	int res = bind(sockfd, (struct sockaddr*)&ser_addr, sizeof(ser_addr));
	assert(res != -1);

	// 3 
	res = listen(sockfd, 5);
	assert(res != -1);

	// 4
	while(1)  //服务器循环的处理客户端的链接
	{
		struct sockaddr_in cli_addr;
		memset(&cli_addr, 0, sizeof(cli_addr));
		socklen_t addrlen = sizeof(cli_addr);

		//第一次阻塞 等待客户端连接
		int c = accept(sockfd, (struct sockaddr*)&cli_addr, &addrlen);
		if(c == -1)
		{
			printf("accept error\n");
			close(c);
			continue;
		}

		printf("%d link success\n", c);

		//第二次阻塞 等待客户端发送数据
		char buff[128] = { 0 };
		int n = recv(c, buff, 127, 0);
		if(n == 0 || n == -1)
		{
			printf("recv error\n");
			close(c);
			continue;
		}
		printf("n = %d, buff =  %s\n", n, buff);

		n = send(c, "OK", 2, 0);
		if(n == 0 || n == -1)
		{
			printf("send error\n");
			close(c);
			continue;
		}

		close(c); // 一定要关闭，防止浪费
	}

	close(sockfd);
	exit(0);
}

客户端代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include <unistd.h>

#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>


int main()
{
	int sockfd = socket(AF_INET, SOCK_STREAM, 0);
	assert(sockfd != -1);

	struct sockaddr_in ser_addr;
	memset(&ser_addr, 0, sizeof(ser_addr));
	ser_addr.sin_family = AF_INET;
	ser_addr.sin_port = htons(6000);
	ser_addr.sin_addr.s_addr = inet_addr("127.0.0.1");

	int res = connect(sockfd, (struct sockaddr*)&ser_addr, sizeof(ser_addr));
	assert(res != -1);

	printf("please input:");
	char buff[128] = { 0 };
	fgets(buff, 128, stdin);  //将最后的回车符也写入到buff中

	int n = send(sockfd, buff, strlen(buff)-1, 0);
	if(n == -1 || n == 0)
	{
		printf("send error\n");
		close(sockfd);
		exit(0);
	}

	memset(buff, 0, 128);
	n = recv(sockfd, buff, 127, 0);
	if(n == -1 || n == 0)
	{
		printf("recv error\n");
		close(sockfd);
		exit(0);
	}

	printf("n = %d, buff = %s\n", n, buff);

	close(sockfd);
	exit(0);
}