c - 从套接字读取/写入意外行为

Question

我有一个任务，我必须从服务器下载一个文件..但我有这样的情况，有时数据被发送，有时却没有!

服务器:

printf("Reading and sending the File.\n");
memset(buff, 0, sizeof(buff));
while((n=read(fp, buff, 1024)) > 0 )
{   
    printf ("%s \n", buff); //no problem, coz it reads the file correctly  
    write(connfd, buff, n); 
    memset(buff, 0, sizeof(buff));
}

客户:

printf("Start downloading the file.\n");
bzero(&data,sizeof(data));
n = 0;
while (1)
{
    n = read(sockfd , data, 1024);
    if (n > 0)
    {
         data[n] = 0;
         r = write (fp, data, n);
         if (r < 0)
         {
             printf("Write to file !!!!!!\n");
         exit(1);
         }
    }
    printf ("\n%s\n", data);
    bzero (&data, sizeof (data));
    if (n < 1024)
        break;
}
printf("Done downloading the file\n");
close (fp);

Answer 1

由于实现原因(套接字、TCP 堆栈、以太网堆栈等)，即使发送了这么多字节，您也可能无法在每次读取中始终获得 1024 字节。

您正在使用if (n < 1024)检查您是否完成。这不应该用套接字来完成。

首先您需要更改read至recv它是专门为与套接字配合使用而设计的。 recv将返回0如果服务器已有序关闭连接并且 -1如果发生错误。任何其他返回值都意味着:继续阅读!

printf("Start downloading the file.\n");
ssize_t n;
while (1)
{
    n = recv(sockfd, data, sizeof(data), 0);
    if (n > 0)
    {
         r = write (fp, data, n);
         if (r < 0)
         {
            perror("write");
            exit(1);
         }

        printf ("\n%*s\n", n, data);
    }
    else if (!n)
        break; // orderly shutdown
    else {
        perror("recv");
        exit(1);
    }
}
printf("Done downloading the file\n");
close (fp);