在程序终止之前,C Socket Send()不会发送

时间:2011-08-13 15:43:51

标签: c sockets client-server fork nonblocking

这是我的第一个问题,所以你的恩典会受到赞赏。我正在尝试学习Unix的C套接字。作为我学习的一部分,我尝试编写一个简单的类似telnet的客户端,它连接到指定端口上的主机,打印从服务器接收的任何字符,并将用户写入的任何内容发送到控制台。收到罚款,但是当我尝试发送字符串时,没有任何反应。然后,当我中断程序时,我尝试发送的所有字符串都被发送。提前致谢。我可能只是愚蠢。

#include <stdio.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netdb.h>
#include <netinet/in.h>
#include <errno.h>
#include <stdlib.h>
#include <signal.h>
#include <fcntl.h>

int SetupSock(char *host, char *port) {
    int s, status;  
    struct addrinfo hints, *res;
    memset(&hints, 0, sizeof hints);
    hints.ai_family = AF_UNSPEC;
    hints.ai_socktype = SOCK_STREAM;
    getaddrinfo(host, port, &hints, &res);  
    if((s = socket(res->ai_family, res->ai_socktype, res->ai_protocol)) == -1) {
        printf("Could not create sock...root?\n");
        exit(1);
    }
    if((status = connect(s, res->ai_addr, res->ai_addrlen)) == -1) {
        printf("Connect Failed: %s\n", strerror(errno));
        printf("%s", strerror( errno ));
        printf("\n");       
        exit(1);
    }
    return s;
}

int main (void) {
    char *host = malloc(100), *port = malloc(20), buf[2], *msg = malloc(1000);
    struct timeval waitid;
    fd_set read_flags, write_flags;
    signal(SIGPIPE, SIG_IGN);
    int sock, flags;
    //Input Host and Port
    printf("Host: ");
    gets(host);
    printf("Port: ");
    gets(port);
    sock = SetupSock(host, port);
    flags = fcntl(sock, F_GETFL, 0);
    fcntl(sock, F_SETFL, flags | O_NONBLOCK);
    FD_ZERO(&read_flags);
    FD_ZERO(&write_flags);
    FD_SET(sock, &read_flags);
    FD_SET(fileno(stdin), &read_flags);
    fflush(0);
    int pid = fork();
    if (pid == 0) {
        int status;
        close(stdout);
        while(1) {
        select(sock + 1, &read_flags, &write_flags, (fd_set*) 0, &waitid);
        if (FD_ISSET(fileno(stdin), &read_flags)) {
            gets(msg);
            if((status = send(sock, msg, strlen(msg), MSG_NOSIGNAL)) == -1) {
                printf("Send Failed: %s\n", strerror(errno));
            }
        }
        }
    }
    else {
        close(stdin);
        while(1) {
            select(sock + 1, &read_flags, &write_flags, (fd_set*) 0, &waitid);
            if (FD_ISSET(sock, &read_flags)) {
                if(recv(sock, &buf, 1, 0) > 0){     
                    printf("%s", buf);
                }
            }
        }
    }
    close(sock);
    return 0;
}

2 个答案:

答案 0 :(得分:1)

您的select仅在此执行一次。它应该在循环内。 select只是等待在提供的文件描述符集上发生事件。 应在每个事件处理完毕后再次调用它。

答案 1 :(得分:0)

要覆盖某个区域,错过了其他答案:select和stdio 不要混用。由于缓冲是否stdin是否可读,您无法获得fileno(stdin)是否可读的准确结果。