C 使用 Berkeley 套接字发送文件

Question

我正在尝试在用 c 编写的客户端-服务器应用程序上发送文件。

客户端应该上传文件，服务器应该接收文件。 客户端：

if(fstat(fd,&file_stat) < 0){
    perror("Fstat error");
    return -1;
}


FILE *filepointer = fopen(filename,"rb");

if(filepointer == NULL){
    perror("Opening file error");
    return -1;
}

strcat(sendbuf,"8"); // option number
strcat(sendbuf,client_account.user); //user who is sending a file
strcat(sendbuf,"p1.txt"); // file name
printf("sendbuf : %s\n",sendbuf );
if(write(sock_fd,sendbuf,strlen(sendbuf)) < 0 ){
    perror("Writing error");
    return -1;
}

/* Check server's answer */
if((nread = read(sock_fd,buffer,sizeof(buffer))) < 0){
    perror("Reading error");
    return -1;
}
buffer[nread] = '\0';
printf("Buffer : %s : %d",buffer,atoi(buffer));
if(atoi(buffer) != GENERIC_OK){
    printf("Error occurred\n");
    return -1;
}

sprintf(file_size,"%lld",file_stat.st_size);

/* Writing file size */
if((nwritten = write(sock_fd,file_size,sizeof(file_size))) < 0){
    perror("Writing error");
    return -1;
}

memset(buffer,0,sizeof(buffer));
/* Check second server's answer */
if((nread = read(sock_fd,buffer,sizeof(buffer))) < 0){
    perror("Reading error");
    return -1;
}
buffer[nread] = '\0';
printf("Buffer : %s : %d",buffer,atoi(buffer));
if(atoi(buffer) != GENERIC_OK){
    printf("Error occurred\n");
    return -1;
}


while(1){

    nbytes = fread(sendbuf,1,sizeof(sendbuf),filepointer);
    /* There are bytes to send */
    printf("Sendbuf : %s \n" , sendbuf);
    if(nbytes > 0){
        write(sock_fd,sendbuf,nbytes);
    }

    if(nbytes < 256){
        if(feof(filepointer) || ferror(filepointer))
            break;
    }
}

服务器端：

... first buffer is received well ...`
/* WRITE TO CLIENT TO CONTINUE */
if(write(sock_fd,"500",strlen("500")) < 0){ /*GENERIC OK*/
    perror("Writing error");
    return -1;
}

memset(recvBuf,0,sizeof(recvBuf));
/*RECEIVING FILE SIZE */
if((nread = read(sock_fd,buffer,sizeof(buffer)))< 0){
    perror("Reading error");
    return -1;
}
buffer[nread] = '\0';

file_size = atoi(buffer);
printf("file size : %d\n",file_size);



if((fd = open(filename, O_CREAT | O_RDWR | O_TRUNC,  S_IRWXU)) < 0){
    perror("File opening error"); /* file already exists */

}

recv_file = fopen(filename,"wb");
if(recv_file == NULL){
    perror("Opening error");
    return -1;
}

remaining_data = file_size;/*i think processes are blocking on while loops*/
while(((nread = read(sock_fd,recvBuf,sizeof(recvBuf))) > 0) && remaining_data > 0){ 
    recvBuf[nread] = '\0';
    printf("%d bytes received : %s \n",nread,recvBuf);
    fwrite(recvBuf,1,nread,recv_file);
    remaining_data -= nread;
}

if(nread < 0){
    perror("Reading error");
}

我尝试使用 sendfile 函数，但我使用的是 Mac OS 并且不受支持。你对我有什么建议吗？

应该是这样的：

1) 客户端向服务器发送一个缓冲区，宣布它将发送一个带有其名称的文件 - OK

2) 服务器接收到这个缓冲区并向客户端发送一个通用的 ok 代码 - OK

3) 客户端向服务器发送文件的大小 - OK

4) 服务器接收到这个缓冲区并向客户端发送一个通用的 ok 代码 - OK

5) 客户端从文件中读取内容并将其发送到服务器 - 不行

6) 服务器从客户端接收内容并将内容写入文件 - 不正常

Answer 1

这一行，在服务器中，

while(((nread = read(sock_fd,recvBuf,sizeof(recvBuf))) > 0) && remaining_data > 0){

正在尝试先阅读，但没有检查 remaining_data >0。结果是传输完所有文件后的读取操作。

建议使用while(remaining_data>0) { select()/read() }，其中select()有合理的超时参数，超时时会导致循环退出。

Answer 2

这段代码：

if((fd = open(filename, O_CREAT | O_RDWR | O_TRUNC,  S_IRWXU)) < 0){
    perror("File opening error"); /* file already exists */

}

recv_file = fopen(filename,"wb");
if(recv_file == NULL){
    perror("Opening error");
    return -1;
}

在服务器中打开输出文件两次，没有干预close()。这可能不是我们所需要的。

Answer 3

我认为您遗漏了一些相关代码，但很可能关键问题在于服务器的while 条件：

while(((nread = read(sock_fd,recvBuf,sizeof(recvBuf))) > 0) && remaining_data > 0){

请注意，它会总是尝试读取数据，即使它已经读取了预期的字节数。如果客户端保持连接打开（可能它正在等待来自服务器的响应），则不会检测到 EOF 条件，并且您会遇到死锁。

交换条件的顺序应该可以解决这个问题，至少对于表现良好的客户而言。但是请注意，即使进行了更改，如果客户端发送的字节数少于其承诺的字节数，但仍保持连接打开，您仍然可能会遇到死锁。 恶意客户端可能会故意表现出这种行为来对您的服务器执行 DoS。

Answer 4

此外，如果完整的 'sizeof(buffer)' 字节数为收到了。

'file_size = atoi(buffer)' - 不保证能正常工作，因为您不知道是否已收到整个缓冲区。

同样的字节序问题。