使用Select的TCP客户端_客户端用select-CSDN博客

本文链接：https://blog.csdn.net/ljiijo/article/details/128537354

该代码示例展示了如何用C语言和select系统调用创建一个TCP客户端，使其能同时进行接收和发送数据。程序通过socket创建套接字，connect连接服务器，然后使用select监控标准输入和套接字，实现数据交互。当有输入事件时，从键盘接收数据并发送给服务器；当有服务器交互事件时，接收服务器的数据并打印。

摘要由CSDN通过智能技术生成

需求：

使用select函数完成TCP客户端，实现可随时收发。

代码实现过程：

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

#define ERR_MSG(msg) do{\
	fprintf(stderr,"line : %d\n",__LINE__);\
	perror("msg");\
}while(0)

#define IP "192.168.8.164" //填本機ubantuIP ifconfig查看
#define PORT 6666			//1024~49151

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

	//填充地址信息結構體
	//根據地址族的不同，地址信息結構體不一致，AF_INET:man 7 ip
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET; 		//必須填AF_INET
	sin.sin_port 		= htons(PORT); 	//端口號的網絡字節序,1024~49151
	sin.sin_addr.s_addr = inet_addr(IP); 	//本機IP地址的網絡字節序,ifconfig

	if(-1 == connect(sfd,(struct sockaddr*)&sin,sizeof(sin)) < 0){
		ERR_MSG(connetct);
		return -1;
	}
	printf("与目标服务器链接成功\n");

	//创建集合
	fd_set readfds;
	fd_set tempfds;

	//清空文件描述符集合
	FD_ZERO(&readfds);

	//将需要的文件描述符添加到读集合中
	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);

	int maxfd = sfd;
	char buf[128] = "";
	ssize_t res = 0;
	int s_res = 0;

	while(1){
		tempfds = readfds;
		//让内核检测集合
		s_res = select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(s_res < 0){
			ERR_MSG("poll");
			return -1;
		}
		else if( 0 == s_res){
			printf("time out...\n");
			break;
		}
		for(int i=0;i<=maxfd;i++){
			if(FD_ISSET(i,&tempfds) == 0){
				continue;
			}
			if(0 == i){
				printf("键盘输入事件\n");
				bzero(buf,sizeof(buf));
				fgets(buf,sizeof(buf),stdin);
				buf[strlen(buf)-1]='\0';
				strcat(buf,"---client");
				if(send(sfd,buf,sizeof(buf),0) < 0){
					ERR_MSG(send);
					return -1;
				}
				printf("发送：newfd=%d: %s\n",sfd,buf);
			}

			if(sfd == i){
				printf("服务器交互事件\n");
				bzero(buf,sizeof(buf));
				//接受
				res = recv(sfd,buf,sizeof(buf),0);
				if(res < 0){
					ERR_MSG(recv);
					return -1;
				}
				else if(0 == res){
					fprintf(stderr,"newfd=%d 客戶端下線\n",sfd);
					break;
				}
				printf("接收：newfd=%d: %s\n",sfd,buf);
			}
		}
	}
	close(sfd);

	return 0;
}