TCP和UDP服务器和客户端简单实现

TCP服务器

#include<myhead.h>

int main(int argc, const char *argv[])
{
	//創建流式套接字socket
	int sfd = socket(AF_INET, SOCK_STREAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket()");
		exit(1);
	}
	printf("socket create success sfd = %d\n", sfd);

	//允许端口快速的被复用
	int reuse = 1;
	if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	printf("允许端口快速的被复用成功\n");

	//綁定服務器的地址信息
	struct sockaddr_in sin = {\
		AF_INET, \
			htons(1314), \
			.sin_addr.s_addr = inet_addr("192.168.115.140")\
	};

	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("bind()");
		exit(1);
	}

	//將套接字設置爲被動監聽狀態listen
	if(listen(sfd, 128) < 0)
	{
		ERR_MSG("listen()");
		exit(1);
	}
	printf("Listen success\n");

	//獲取一個已經完成的客戶端信息，生成一個新的文件描述符
	int newfd = -1;
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);
	newfd = accept(sfd, NULL, NULL);
	if(newfd < 0)
	{
		ERR_MSG("newfd");
		exit(1);
	}
	printf("[%s:%d]客戶端連接成功 newfd=%d\n",\
			inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd);

	char buf[128] = "";
	ssize_t res = 0;
	while(1)
	{
		//清空字符串
		bzero(buf, sizeof(buf));
		res = recv(newfd, buf, sizeof(buf), 0);
		
		//接收
		if(res < 0)
		{
			ERR_MSG("recv()");
			exit(1);
		}
		else if(0 == res)
		{
			printf("[%s:%d]客戶端下线 newfd=%d\n",\
				inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd);
			break;
		}
		
		printf("[%s:%d]客戶端連接成功 newfd=%d：%s\n",\
			inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, buf);
		
		//發送
		strcat(buf, "*_*");
		if(send(newfd, buf, sizeof(buf), 0) < 0)
		{
			ERR_MSG("send");
			exit(1);
		}
		printf("send success\n");
	}
	
	//關閉文件描述符
	close(newfd);
	if(close(sfd) < 0)
	{
		ERR_MSG("close()");
		exit(1);
	}
	return 0;
}

TCP客户端

#include<myhead.h>

int main(int argc, const char *argv[])
{
	//創建流式套接字socket
	int cfd = socket(AF_INET, SOCK_STREAM, 0);
	if(cfd < 0)
	{
		ERR_MSG("socket()");
		exit(1);
	}
	printf("socket create success cfd = %d\n", cfd);

	
	//填充服务器的地址信息
	struct sockaddr_in sin = {\
		AF_INET, \
			htons(1314), \
			.sin_addr.s_addr = inet_addr("192.168.115.140")\
	};

	//connect连接
	if(connect(cfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("connect()");
		exit(1);
	}
	printf("Connect success\n");
	
	char buf[128] = "";
	ssize_t res = 0;
	while(1)
	{
		//清空字符串
		bzero(buf, sizeof(buf));
		
		printf("请输入：");
		fgets(buf, sizeof(buf), stdin);
		buf[strlen(buf)-1] = 0;

		//發送
		strcat(buf, "*_*");
		if(send(cfd, buf, sizeof(buf), 0) < 0)
		{
			ERR_MSG("send");
			exit(1);
		}
		printf("send success\n");	
		
		bzero(buf, sizeof(buf));
		res = recv(cfd, buf, sizeof(buf), 0);
		//接收
		if(res < 0)
		{
			ERR_MSG("recv()");
			exit(1);
		}
		else if(0 == res)
		{
			printf("客户端下线 cfd=%d\n", cfd);
			break;
		}
		
		printf("cfd=%d：%s\n", cfd, buf);
	}
	
	//關閉文件描述符
	if(close(cfd) < 0)
	{
		ERR_MSG("close()");
		exit(1);
	}
	return 0;
}

UDP服务器

#include<myhead.h>

#define PORT 1314
#define IP "192.168.115.140"

int main(int argc, const char *argv[])
{
	//创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket()");
		exit(1);
	}
	printf("Socket create success sfd=%d\n", sfd);

	struct sockaddr_in sin = {AF_INET, htons(PORT), inet_addr(IP)};

	//绑定地址信息结构体
	if(bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("bind()");
		exit(1);
	}
	printf("Bind success\n");

	char buf[128] = "";
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);

	while(1)
	{
		bzero(buf, sizeof(buf));
		//接收数据
		if(recvfrom(sfd, buf, sizeof(buf), 0, (struct sockaddr*)&cin, &addrlen) < 0)
		{
			ERR_MSG("recvfron()");
			exit(1);
		}
		printf("[%s:%d]：%s\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), buf);

		//发送数据
		strcat(buf, "*_*");
		if(sendto(sfd, buf, sizeof(buf), 0, (struct sockaddr*)&cin, sizeof(cin)) < 0)
		{
			ERR_MSG("sendto()");
			exit(1);
		}
		printf("Sendto success\n");
	}

	//关闭套接字
	close(sfd);

	return 0;
}

UDP客户端

#include<myhead.h>

#define PORT 1314
#define IP "192.168.115.140"

int main(int argc, const char *argv[])
{
	//创建报式套接字
	int cfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(cfd < 0)
	{
		ERR_MSG("socket()");
		exit(1);
	}
	printf("Socket create success cfd=%d\n", cfd);
	
	//服务器地址
	struct sockaddr_in sin = {AF_INET, htons(PORT), inet_addr(IP)};
	socklen_t addrlen = sizeof(sin);

	char buf[128] = "";
	ssize_t ret = 0;
	while(1)
	{
		bzero(buf, sizeof(buf));
		//发送数据
		printf("请输入：");
		fgets(buf, sizeof(buf), stdin);
		buf[strlen(buf)-1] = 0;


		if(sendto(cfd, buf, sizeof(buf), 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
		{
			ERR_MSG("sendto()");
			exit(1);
		}
		printf("Sendto success\n");

		bzero(buf, sizeof(buf));
		//接收数据
		if(recvfrom(cfd, buf, sizeof(buf), 0, (struct sockaddr*)&sin, &addrlen) < 0)
		{
			ERR_MSG("recvfron()");
			exit(1);
		}
		printf("[%s:%d]：%s\n", inet_ntoa(sin.sin_addr), ntohs(sin.sin_port), buf);

		}

	//关闭套接字
	close(cfd);

	return 0;
}