C 使用伯克利套接字发送文件

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] = '[=10=]';
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] = '[=10=]';
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] = '[=11=]';

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] = '[=11=]';
    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 条件上：

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

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

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

Answer 2

另外，'read(sock_fd,buffer,sizeof(buffer)'，后面是'buffer[nread] = '\0';'如果接收到完整的 'sizeof(buffer)' 个字节，缓冲区将溢出 1。

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

还有相同的字节顺序问题。

Answer 3

此代码：

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 4

这一行，在服务器中，

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

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

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

C 使用伯克利套接字发送文件

C Sending files with Berkeley socket

c

sockets

networking