Linux作业

2023-5-21作业补

搭建TCP服务器端

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

#define ERR_MSG(msg) do{printf("line:%d\n",__LINE__);perror(msg);}while(0)     //封装错误宏
#define PORT 6666    //端口号
#define IP "192.168.8.254"    //主机IP

struct climsg
{
	int newfd;
	struct sockaddr_in cin;
};

void*deal_cli_msg(void*arg);

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 __%d__\n",sfd,__LINE__);
        
	int reuse=1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
        //绑定服务器IP和端口号
	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 __%d__\n",__LINE__);
        //设定监听组发送处理信号
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success __%d__\n",__LINE__);

	fd_set readfds,tempfds;

	FD_ZERO(&tempfds);
	FD_ZERO(&readfds);

	FD_SET(sfd,&readfds);
	FD_SET(0,&readfds);

	int maxfd=sfd;

	int s_res=0;
	struct sockaddr_in cin;
	socklen_t len=sizeof(cin);
	int newfd=-1;
	char buf[128]="";
	ssize_t res=0;
	while(1)
	{
                //创建处理与客户机的会话连接组
		tempfds=readfds;
		s_res=select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(0>s_res)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(0==s_res)
		{
			printf("Overrun!\n");
			break;
		}
                //键盘触发事件
		for(int i=0;i<=maxfd;i++)
		{
			if(!FD_ISSET(i,&tempfds))continue;
			if(0==i)
			{
				printf("Trigger keboard input event!\n");

				fgets(buf,sizeof(buf),stdin);
				buf[strlen(buf)-1]=0;
				printf("buf=%s\n",buf);
			}
                        //客户机连接事件
			if(i==sfd)
			{
				printf("Trigger client link event!\n");
				newfd=accept(sfd,(struct sockaddr*)&cin,&len);
				if(newfd<0)
				{
					ERR_MSG("accept");
					return -1;	
				}
				printf("[%s:%d] The client link is successful! newfd=%d __%d__\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,__LINE__);
				FD_SET(newfd,&readfds);
				maxfd=maxfd<newfd?newfd:maxfd;
			}
                        //客户机交互事件
			else
			{
				printf("Trigger client interaction events! __%d__\n",__LINE__);
				bzero(buf,sizeof(buf));
				res=recv(i,buf,sizeof(buf),0);
				if(res<0)
				{
					ERR_MSG("recv");
					return -1;
				}
                                //客户机下线事件
				else if(0==res)
				{
					printf("[%s:%d] The client is offline! newfd=%d __%d__\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),i,__LINE__);
					close(i);
					FD_CLR(i,&readfds);
					while(!FD_ISSET(maxfd,&readfds)&&maxfd-->=0);
					continue;
				}
				printf("[%s:%d] newfd=%d:%s,__%d__\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),i,buf,__LINE__);
				strcat(buf,"*_*");
				if(send(i,buf,sizeof(buf),0)<0)
				{
					ERR_MSG("send");
					return -1;
				}
				printf("Transmit success!\n");
			}
		}

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