LInux作业

最新推荐文章于 2024-07-25 15:50:18 发布

Onesefl

最新推荐文章于 2024-07-25 15:50:18 发布

阅读量22

点赞数

文章标签： linux 运维服务器网络 c++

本文链接：https://blog.csdn.net/Onesefl/article/details/130817938

版权

2023-5-23作业

搭建聊天室客户端

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

#define N 128
//协议
#define LOGIN 	1
#define CHAT 	2
#define QUIT 	3

#define ERR_LOG(msg) do{printf("%d %s %s \n", __LINE__, __func__, __FILE__);perror(msg);}while(0)

typedef struct
{
	int type;
	char name[20];
	char text[N] ;

}MSG;

typedef void (*sighandler_t)(int);

int do_recv(int sfd);
int do_chat(int sfd, MSG msg, struct sockaddr_in sin);

//回收子进程资源并退出
void handler(int sig)
{while(waitpid(-1,NULL, WNOHANG)>0);	exit(0);}

int main(int argc, const char *argv[])
{
	if(argc < 3){printf("Enter IP and port number!\n");return -1;} 

	//注册信号处理函数
	sighandler_t s = signal(SIGCHLD, handler);
	if(s == SIG_ERR){ERR_LOG("signal");return -1;}

	//创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0){perror("socket");return -1;}

	//填充服务器ip和端口号
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;
	sin.sin_port 		= htons(atoi(argv[2]));
	sin.sin_addr.s_addr = inet_addr(argv[1]);
	socklen_t addrlen = sizeof(sin);

	//登录协议
	MSG msg ;msg.type = htonl(LOGIN);

	printf("Enter name:");
	fgets(msg.name, 20, stdin);
	msg.name[strlen(msg.name)-1] = 0;

	//发送登录信息
	if(sendto(sfd, &msg, sizeof(msg), 0, (void*)&sin, sizeof(sin)) < 0)
	{ERR_LOG("sendto");	return -1;}

	pid_t pid = fork();

	//父进程获取信息
	if(pid > 0)
	{do_recv(sfd);}
	//子进程发送信息
	else if(0 == pid)
	{do_chat(sfd, msg, sin);}

	//关闭套接字
	close(sfd);
	return 0;
}


int do_chat(int sfd, MSG msg, struct sockaddr_in sin)
{
	while(1)
	{
		//从终端获取聊天信息
		bzero(msg.text, N);
		fgets(msg.text, N ,stdin);
		msg.text[strlen(msg.text)-1] = 0;

		//封装CHAT协议和QUIT协议
		if(strncasecmp(msg.text, "quit" , 4) == 0)
		{msg.type = htonl(QUIT);}
		else{msg.type = htonl(CHAT);}

		//发送 
		if(sendto(sfd, &msg, sizeof(msg), 0, (void*)&sin, sizeof(sin)) < 0)
		{ERR_LOG("sendto");return -1;}

		//退出子进程;
		if(msg.type == htonl(QUIT)){exit(0);}
	}
}

int do_recv(int sfd)
{
	MSG rcv_msg ;
	while(1)
	{
		if(recvfrom(sfd, &rcv_msg, sizeof(rcv_msg), 0, NULL, NULL) < 0)
		{perror("recvfrom");return -1;}
		printf("%s\n",rcv_msg.text);
	}
}