c - 使用select()函数发送和接收数据

我正在编写与服务器通信，发送和接收数据的客户端代码。
我正在使用select函数来监视两个文件描述符-0作为标准输入fd，4(socket_dect)作为套接字fd。
主要思想是查找当前fd的 Activity 状态，如果STDING Activity ，则表示错误读取了该数据并将其发送到服务器，如果socket fd Activity ，则错误地接收了数据并打印了该数据。
我不知道为什么，但是在运行此客户端时一无所有。
这是我的代码:

//聊天客户端
//由omri撰写

#include <stdio.h>
#include <sys/socket.h>
#include <sys/select.h>
#include <arpa/inet.h> //inet_addr
#include <netinet/in.h>
#include <string.h>
#include <signal.h>
#include <sys/time.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <errno.h>
#include <fcntl.h>


#define PORT 9034 // defined port like the server
#define STDIN 0

void main()
{
    struct timeval tv; // time values for select function
    int activity = 0 ;
    int ready; // contains the result of the select function
    char buf[256];
    int len = strlen(buf);
    int byte_sent,i;
    int socket_dect; // creating socket descriptor
    struct sockaddr_in ServerInfo;
    fd_set readfds;

    // creating a new socket, its a number represent a file descriptor
    // socket args : 1)ip protocol ipv4,second tcp/udp, third is number of protocol used
    socket_dect = socket(AF_INET,SOCK_STREAM,0);

    if(socket_dect == -1){
        perror("error creating socket");
    }
    // fill the values of the server
    ServerInfo.sin_family = AF_INET; // ipv4
    ServerInfo.sin_port = htons(PORT); // port number
    //ServerInfo.sin_addr = 127.0.0.1;
    inet_pton(AF_INET, "127.0.0.1", &ServerInfo.sin_addr);//insert the ip to the sin addr

    //fill the values of the time for the select function
    tv.tv_sec = 5; // 5 sec
    tv.tv_usec = 0; // 0 mili sec
    FD_ZERO(&readfds); // clear the socket set
    for(i = 0; i<= socket_dect; i++){
        FD_SET(i,&readfds);// set the sockets
    }

    // making the connection to the server
    //ServerInfo.sin_addr.s_addr = inet_addr("127.0.0.1"); // another way to put ip addr
    connect(socket_dect,(struct sockaddr *)&ServerInfo,sizeof(ServerInfo)); // connected to the server
    printf("after connect");
    while(1){
        ready = select(socket_dect,&readfds,NULL,NULL,&tv);
        printf("%d is ready",ready);
        if(ready < 0){
            perror("select error");
            printf("select error");
        }
        // reading from STDIN to buff
        else if(ready == STDIN_FILENO){
            if(fread(buf,256,1,&readfds) < 0){
                printf("read error");
                perror("read error");
            }
            if(send(socket_dect,buf,len,NULL) < 0){
                perror("send connection");
                printf("send error");
            }//end if send
        }//end elseif STDIN
        else if(ready == socket_dect){
            if(recv(socket_dect,buf,len,0) < 0 ){
                printf("recv error");
            }
            printf("data recived is :%s",buf);
        }//end elseif
        else{
            printf("none is ready");
        }
    }//end while
}//end main

最佳答案

对于初学者来说，还有两个问题。

这个

ready = select(socket_dect,&readfds,NULL,NULL,&tv);

应该

ready = select(socket_dect + 1, &readfds, NULL, NULL, &tv);

从 man select :



代替

if(ready == STDIN_FILENO)

利用

if (FD_ISET(STDIN_FILENO, &readfds)

select()修改了传递的参数，因此您每次需要在调用循环内的select()之前重新初始化它们。

另外，您必须测试所有系统调用的结果。这包括connect()。

同样，^ 2您可能不会假定recv()返回的字节数与告知的字节数相同(此处为len)。它可能会收到很少的返回。为了确保您获得了所需的一切，请围绕recv()计数循环。