2023.05.16 网络编程day2

1.完成TCP服务器

2.完成TCP客户端

1.TCP服务器

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

#define ERR_MSG(msg) do{\
	printf("line:%d ", __LINE__);\
	perror(msg);\
} while(0)
#define IP "192.168.9.83"
#define PORT 6666

int main(int argc, const char *argv[])
{
	//创建套接字
	int sfd = socket(AF_INET, SOCK_STREAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -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;
	}

	//绑定
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);
	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");

	//监听
	if(listen(sfd, 128) < 0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");

	//获取新套接字
	struct sockaddr_in cin;
	int len = sizeof(cin);
	int newfd = accept(sfd, (struct sockaddr*)&cin, &len);
	if(newfd < 0)
	{
		ERR_MSG("accept");
		return -1;
	}
	printf("[%s:%d] 客户端连接成功 newfd=%d\n", inet_ntoa(cin.sin_addr), cin.sin_port, newfd);

	char buf[128] = "";
	ssize_t res;
	while(1)
	{
		//接收数据
        bzero(buf, sizeof(buf));
		res = recv(newfd, buf, sizeof(buf), 0);
		if(res == -1)
		{
			ERR_MSG("recv");
			return -1;
		}
		else if(res == 0)
		{
			printf("[%s:%d] 客户端下线 newfd=%d\n", inet_ntoa(cin.sin_addr), cin.sin_port, newfd);
			break;
		}
		printf("[%s:%d] newfd=%d: %s\n", inet_ntoa(cin.sin_addr), cin.sin_port, newfd, buf);

		//发送数据
		strcat(buf, "*_*");
		if(send(newfd, buf, sizeof(buf), 0) < 0)
		{
			ERR_MSG("send");
			return -1;
		}
		printf("发送成功\n");
	}


	//关闭
	if(close(sfd) < 0)
	{
		ERR_MSG("close");
		return -1;
	}
	if(close(newfd) < 0)
	{
		ERR_MSG("close");
		return -1;
	}
	return 0;
}

2.TCP客户端

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

#define ERR_MSG(msg) do{\
	printf("line:%d ", __LINE__);\
	perror(msg);\
} while(0)
#define IP "192.168.9.83"
#define PORT 6666

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

	//连接服务器
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);
	if(connect(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success\n");

	char buf[128] = "";
	ssize_t res = 0;
	while(1)
	{
		//发送数据
		printf("请输入要发送的数据:");
		fflush(stdout);
		bzero(buf, sizeof(buf));
		read(0, buf, sizeof(buf));
		buf[strlen(buf)-1] = 0;
		if(send(sfd, buf, sizeof(buf), 0) < 0)
		{
			ERR_MSG("send");
			return -1;
		}
		printf("发送成功\n");

		//接收数据
		bzero(buf, sizeof(buf));
		res = recv(sfd, buf, sizeof(buf), 0);
		if(res == -1)
		{
			ERR_MSG("recv");
			return -1;
		}
		else if(res == 0)
		{
			printf("服务器关闭\n");
			break;
		}
		printf("%s\n", buf);
	}

	//关闭
	if(close(sfd) < 0)
	{
		ERR_MSG("close");
		return -1;
	}
	return 0;
}

运行结果