网络编程0816作业

完成select的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>

//打印错误新的宏函数
#define ERR_MSG(msg)  do{\
    fprintf(stderr, " __%d__ ", __LINE__);\
    perror(msg);\
}while(0)

//指定要连接的服务器的地址信息
#define PORT 6666               //1024~49151                       
#define IP  "192.168.31.12"    //本机IP,用ifconfig查看


int update_maxfd(int maxfd,fd_set readfds)
{
	int z = maxfd;
	for( ;z > 0;z++)
	{
		if(FD_ISSET(z,&readfds));
		return z;
	}                                                 
	return 0;
}

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


	//绑定客户端的地址信息结构体--》非必须绑定 
	
	//填充要连接的服务器的地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;
	sin.sin_port 		= htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);

	//连接服务器 connect
	if(connect(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("connect");
		return -1;
	}

	char buf[128] = "";
	ssize_t res = 0;


	//创建一个读集合
	fd_set readfds;

	//将集合清空
	FD_ZERO(&readfds);

	//将0号文件描述符添加到集合
	FD_SET(0,&readfds);

	//将sfd文件描述符添加到集合                             
	FD_SET(sfd,&readfds);

	int maxfd = sfd;
    int select_res = 0;
    int i = 0;
	

	while(1)
	{
		fd_set newreadfds;
		newreadfds = readfds;

		select_res = select(maxfd+1,&newreadfds, NULL, NULL, NULL);
		if(select_res < 0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(select_res == 0)
		{
			fprintf(stderr,"select time out");
			break;
		}

		for(i = 0;i <= maxfd;i++)
		{
			if(FD_ISSET(i,&newreadfds) == 0)
			{
				continue;
			}

			if(i == 0)
			{
				bzero(buf, sizeof(buf));
				//发送
				printf("请输入>>>");
				fgets(buf, sizeof(buf), stdin);
				buf[strlen(buf)-1] = 0;

				if(send(sfd, buf, sizeof(buf), 0) < 0)
				{
					ERR_MSG("send");
					return -1;
				}
				printf("send success\n");
			}
			else
			{
				//接收 
				bzero(buf, sizeof(buf));
				res = recv(i, buf, sizeof(buf), 0);
				if(res < 0)
				{
					ERR_MSG("recv");
					return -1;
				}
				else if(0 == res)
				{
					printf("server is off-line\n");
					//关闭文件描述符
					close(i);

					//将文件描述符从集合中踢出
					FD_CLR(i,&readfds);

					//更新maxfd
					maxfd = update_maxfd(maxfd,readfds);                

					break;
				}
				else
				{
					printf("%ld : %s\n", res, buf);
				}

			}
		}

	}

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

	return 0;
}