如何用 C 语言将任意大的文件读取到 TCP 套接字？

Question

我刚刚学习套接字编程，我正在尝试编写一个回显客户端，该客户端从标准输入读取并写入套接字，然后从套接字读取服务器响应到标准输出。问题是我不知道标准输入有多长，也不知道服务器的响应有多长。我尝试使用的代码如下（创建套接字并连接到服务器被省略）：

length = BUF_SIZE;
while (length == BUF_SIZE) {  // length will equal BUF_SIZE if buf is full, when length < BUF_SIZE we have reached an EOF                                 
   // Reads from STDIN to buf                                                                                                                              
   if ((length = read(STDIN_FILENO, buf, BUF_SIZE)) < 0){
     fprintf(stderr, "Error in reading from STDIN");
     return 4;
   }
   // Writes from buf to the socket                                                                                                                        
   if ((write(sock, buf, BUF_SIZE)) < 0){
     fprintf(stderr, "Error writing to socket");
     return 5;
   }
 }

if ((status = shutdown(sock, 1)) < 0){ // Shuts down socket from doing more receives                                                                      
   fprintf(stderr, "Error shutting down socket for writing");
   return 6;
}

length = BUF_SIZE; 
while (length == BUF_SIZE){
   // Read from socket to buf                                                                                                                              
   if ((length = read(sock, buf, BUF_SIZE)) < 0){
     fprintf(stderr, "Error reading from socket");
     return 7;
   }
   // Write from buf to STDOUT                                                                                                                             
   if ((write(STDOUT_FILENO, buf, BUF_SIZE)) < 0){
     fprintf(stderr, "Error writing to STDOUT");
     return 8;
   }
}

close(sock);
exit(0);

BUF_SIZE 定义为 100。当我运行我的程序时，程序通常会连接到服务器并发送正确的消息，但是什么它写入标准输出要么什么都没有，要么是乱码。

我做错了什么？

Answer 1

你的 while 循环只会在第一次运行。 read()/write() 只会返回它们实际读/写的数量，这很可能不等于 BUF_SIZE。假设您从套接字读取了 10 个字节，然后向 stdout 写入了 100 个字节 - 最后 90 个字节将是垃圾。

沿着这些思路的一些事情会让你更接近你想要的。

while (1)
{
    if ((length = read(STDIN_FILENO, buf, BUF_SIZE)) < 0)
    {
        fprintf(stderr, "Error in reading from STDIN");
        return 4;
    }

    if ((write(sock, buf, length)) < 0)
    {
        fprintf(stderr, "Error writing to socket");
        return 5;
    }

    if ((length = read(sock, buf, BUF_SIZE)) < 0)
    {
        fprintf(stderr, "Error reading from socket");
        return 7;
    }

    if ((write(STDOUT_FILENO, buf, length)) < 0)
    {
        fprintf(stderr, "Error writing to STDOUT");
        return 8;
    }
}