鍍金池/ 問答/Java  C  Linux/ ftp client 發(fā)送了 STOR 命令后,命令端口就阻塞了,我關(guān)閉了數(shù)據(jù)端

ftp client 發(fā)送了 STOR 命令后,命令端口就阻塞了,我關(guān)閉了數(shù)據(jù)端口了?

問題

代碼

開了一個(gè)子進(jìn)程來發(fā)送文件,發(fā)送完畢之后就 close 了 client_data_socket (數(shù)據(jù) socket )

// 非阻塞
set_flag(client_data_socket, O_NONBLOCK);
pid_t pid;
if ((pid = fork()) < 0) {
    printf("fork error");
    continue;
} else if (pid == 0) {
    FILE *fp;
    if ((fp = fopen(filename, "rb")) == NULL)
    {
        close(client_data_socket);
        printf("open file failed\n");
        exit(1);
    }
    size_t char_size = sizeof(char);
    char data_buffer[FILE_READ_BUFFER_SIZE];
    int numread;
    for (;;)
    {
        bzero(data_buffer, FILE_READ_BUFFER_SIZE);
        numread = fread(data_buffer, char_size, FILE_READ_BUFFER_SIZE, fp);
        if (numread < 0)
        {
            printf("read file failed\n");
            break;
        } 
        else if (numread > 0)
        {
            int length = send(client_data_socket, data_buffer, numread, 0);
            if (length == 0)
            {
                break;
            }
            else if (length < 0)
            {
                if (errno == EAGAIN || errno == EWOULDBLOCK || errno == EINTR)
                {
                    continue;
                }
                printf("[PUT] command send data failed\n");
                exit(1);
            }
        }
        if (numread == FILE_READ_BUFFER_SIZE) continue;
        else {
            break;
        }
    }
    close(client_data_socket);
    fclose(fp);
    exit(0);
} else {
    int status = 0;
    waitpid(pid, &status, 0);
    if (status == 0)
        printf("send file %s complete.\n", filename);
    else 
        printf("send file %s failed.\n", filename);
}

然后,發(fā)送別的命令在 recv 的時(shí)候,命令端 socket 就阻塞了
Github

回答
編輯回答
拼未來

找到了原因:
shutdown 可以選擇關(guān)閉某個(gè)方向或者同時(shí)關(guān)閉兩個(gè)方向,shutdown how = 0 or how = 1 or how = 2 (SHUT_RD or SHUT_WR or SHUT_RDWR),后兩者可以保證對(duì)等方接收到一個(gè)EOF字符(即發(fā)送了一個(gè)FIN段),而不管其他進(jìn)程是否已經(jīng)打開了這個(gè)套接字。而close不能保證,只有當(dāng)某個(gè)sockfd的引用計(jì)數(shù)為0,close 才會(huì)發(fā)送FIN段,否則只是將引用計(jì)數(shù)減1而已。也就是說只有當(dāng)所有進(jìn)程(可能fork多個(gè)子進(jìn)程都打開了這個(gè)套接字)都關(guān)閉了這個(gè)套接字,close 才會(huì)發(fā)送FIN 段。

因?yàn)槲业腸lient_data_socket是在父進(jìn)程產(chǎn)生的,子進(jìn)程關(guān)閉了并沒有發(fā)送FIN段,所以我需要用shutdown來發(fā)送FIN

2017年10月4日 22:06