9.22作业

#include <myhead.h>
int get_data(int cfd)
{
	char buf[20]="";
	fgets(buf,sizeof(buf),stdin);
	buf[strlen(buf)-1]=0;
	if(send(cfd,buf,sizeof(buf),0)<0)//发送
		{
			ERR_MSG("send");
			return -1;
		}
		printf("send success\n");

	return 0;
}
int recv_data(int cfd,struct sockaddr_in sin)
{
	char  buf[20]="";
	ssize_t res=-1;
	res=recv(cfd,buf,sizeof(buf),0);//接收
	if(res<0)
	{
		ERR_MSG("recv");
		return -1;
	}
	else if(0==res)
	{
		close(cfd);
		return 0;
	}else{
		printf("[%s:%d] buf=%s\n",inet_ntoa(sin.sin_addr),ntohs(sin.sin_port),\
				buf);
		}
}
int main(int argc, const char *argv[])
{
	int cfd=socket(AF_INET,SOCK_STREAM,0);//创建流式套接字
	if(cfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port=htons(8888);
	sin.sin_addr.s_addr=inet_addr("192.168.125.61");
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("connect");
		return -1;
	}
	fd_set readfds,tempfds;
	FD_ZERO(&readfds);
	FD_SET(0,&readfds);
	FD_SET(cfd,&readfds);

	int newfd=-1;	
	while(1)
	{
		tempfds=readfds;
		if(select(cfd+1,&tempfds,NULL,NULL,NULL)<0)
		{
			ERR_MSG("select");
			return -1;
		}
		if(FD_ISSET(0,&tempfds))
		{
			get_data(cfd);
		}
		if(FD_ISSET(cfd,&tempfds)){
			recv_data(cfd,sin);
		}
	}
	close(cfd);
	return 0;
}