使用select()系统调用侦听stdin和服务器

时间:2021-05-03 03:41:12

I want to use select system call to multiplex STDIN and SOCKFD (connected to a server) such that I must be listening to both STDIN and SOCKFD and depending upon where the data is available for read I must proceed further.

我想使用select系统调用来复用STDIN和SOCKFD(连接到服务器),这样我必须同时监听STDIN和SOCKFD,并且根据数据可用于读取的位置,我必须继续进行。

Note: #define STDIN 0

注意:#define STDIN 0

This is what happens when I do the following.

当我执行以下操作时会发生这种情况。

  • I connect to the server the usual way [ socket() then connect() ]
  • 我以通常的方式连接到服务器[socket()然后connect()]

  • I add STDIN and connection socket descriptor 'SOCKFD' to the fd_set (named 'readset') meant for reading.
  • 我将STDIN和连接套接字描述符'SOCKFD'添加到用于读取的fd_set(名为'readset')。

  • Then I call select().
  • 然后我调用select()。

  • Use FD_ISSET to determine which fd is ready for read.
  • 使用FD_ISSET确定哪个fd已准备好读取。

The problem with this set up is that FD_ISSET is always true for both the fd. Once I get connected to the server I see FD_ISSET(sockfd, &readset) is always true and so is FD_ISSET(STDIN, &readset) whether, the server is sending data or not or if I am entering data from keyboard or not.

这个设置的问题是FD_ISSET对于fd都是正确的。一旦我连接到服务器,我看到FD_ISSET(sockfd,&readset)始终为true,因此FD_ISSET(STDIN,&readset)是否,服务器是否正在发送数据,或者我是否从键盘输入数据。

What could I be doing wrong?

我能做错什么?

/* connection successful */

/ *连接成功* /

FD_ZERO(&connset);

while(1) {

    FD_SET(sockfd,&connset); /* add sockfd to connset */
    FD_SET(STDIN,&connset); /* add STDIN to connset */
    fdmax=sockfd;

    if(select(fdmax+1,&readset,NULL,NULL,NULL)<0){
        fprintf(stdout, "select() error\n");
        exit(0);
    }       

    /* select returned
    * check which socket is set
    */

    if(FD_ISSET(sockfd,&connset)) {

        /*
         * Server sends msglen 
         * client reads msg of length msglen
         * client prints it to stdout
         * client waits for next activity of its listen sockets
         */
        size=4;
        ptr=(char *)&msglen;
        while(1) {
            if((nread=recv(sockfd,ptr,size,0)),MSG_DONTWAIT) {
                close(sockfd);
                exit(0);
            }
            size-=nread;
            ptr+=nread;
        }
        bytesToRead = ntohl(msglen);
        readbuf = (char *)malloc(sizeof(char)*(bytesToRead+1));
        ptr=readbuf;
        while(1) {
            nread=recv(sockfd,ptr,bytesToRead,MSG_DONTWAIT);
            if(nread<=0) {
                close(sockfd);
                exit(0);
            }
            bytesToRead-=nread;
            ptr+=nread;
        }
        /* msg read successfully */
        *ptr='\0';
        fprintf(stdout, "in: %s\n",readbuf);
        free(readbuf);
    }

    if(FD_ISSET(STDIN,&connset)) {
        /* data at STDIN */
        fgets(buf,2,stdin); /* read the first newline char (< enter >) */
        fprintf(stdout,"Enter msg: ");
        fgets(buf,MAXLEN,stdin); /* read the msg */
        buf[strlen(buf)-1]='\0';
        msglen = htonl((uint32_t)(strlen(buf)-1));
        ptr = (char *)&msglen;
        bytesToSend = sizeof(uint32_t);
        cnt = bytesToSend;
        while(cnt>0) {
            if((nsent=send(sockfd,ptr,cnt,0))>0) {
                cnt-=nsent;
                ptr+=nsent;
            }
            else {
                fprintf(stdout,"send error\n");
                exit(0);
            }
        }
        ptr=buf;
        bytesToSend=(uint32_t)(strlen(buf)-1);
        cnt=bytesToSend;
        while(cnt>0) {
            if((nsent=send(sockfd,ptr,cnt,0))>0) {
                cnt-=nsent;
                ptr+=nsent;
            }
            else {
                fprintf(stdout,"send error\n");
                exit(0);
            }
        }       
    }       
}   
return 0;

}

1 个解决方案

#1


4  

What is STDIN? There is stdin which is declared in <stdio.h>, is a FILE* and not a file descriptor so can't be used in an fdset and STDIN_FILENO, which is defined in <unistd.h> and is a file descriptor and thus can be used with fdsets.

什么是STDIN?在 中声明了stdin,它是一个FILE *而不是文件描述符,所以不能在fdset和STDIN_FILENO中使用,它在 中定义并且是文件描述符,因此可以与fdsets一起使用。

BTW, in <stdio.h> there is also a function fileno() which returns the file descriptor of a FILE* (depending on your compilation flags, you may need to define some feature macros to get the declaration).

顺便说一句,在 中还有一个函数fileno(),它返回FILE *的文件描述符(根据你的编译标志,你可能需要定义一些特征宏来获取声明)。

#1


4  

What is STDIN? There is stdin which is declared in <stdio.h>, is a FILE* and not a file descriptor so can't be used in an fdset and STDIN_FILENO, which is defined in <unistd.h> and is a file descriptor and thus can be used with fdsets.

什么是STDIN?在 中声明了stdin,它是一个FILE *而不是文件描述符,所以不能在fdset和STDIN_FILENO中使用,它在 中定义并且是文件描述符,因此可以与fdsets一起使用。

BTW, in <stdio.h> there is also a function fileno() which returns the file descriptor of a FILE* (depending on your compilation flags, you may need to define some feature macros to get the declaration).

顺便说一句,在 中还有一个函数fileno(),它返回FILE *的文件描述符(根据你的编译标志,你可能需要定义一些特征宏来获取声明)。