文件从客户端传输到服务器

时间:2016-08-20 16:44:32

标签: c sockets client-server file-transfer

我正在尝试从客户端向服务器发送文件。它有时是工作,有时候不是。当第一次尝试发送文件时,它发送损坏的文件,并且在服务器端形成的文件侧是其大小的两倍。但是在第一次尝试后为所有尝试成功传输。任何人都可以帮我这个吗?

Client.c

     message msg;
        msg.type = SEND_FILE;
        char *username, *filename;

        username = strtok(input+6, " ");
        filename = strtok(NULL, "");

        //Get Picture Size
        printf("Getting Picture Size\n");
        FILE *picture;
        picture = fopen(filename, "r");
        int size;
        fseek(picture, 0, SEEK_END);
        size = ftell(picture);
        fseek(picture, 0, SEEK_SET);

        //Send Picture Size
        printf("Getting Picture Size\n");
        sprintf(msg.data, "%d", size);
        strncpy(msg.username, username, 20);

        if(send(connection->socket, &msg, sizeof(message), 0) < 0)
        {
            perror("Send failed");
            exit(1);
        }
        //Send Picture as Byte Array
        printf("Sending Picture as Byte Array\n");
        char send_buffer[size+1];
        fread(send_buffer, 1, sizeof(send_buffer), picture);
        write(connection->socket, send_buffer, sizeof(send_buffer));
        bzero(send_buffer, sizeof(send_buffer));

server.c

  //Read Picture Size
printf("Reading Picture Size\n");
int size = atoi(message_text);

//Read Picture Byte Array
printf("Reading Picture Byte Array\n");
char p_array[size];
printf("Converting Byte Array to Picture %d\n", size);
FILE *image;
image = fopen("c4.png", "w");
int readSize = 0;
while (readSize < size) {
  readSize = readSize + read(clients[sender].socket, p_array, size);
  fwrite(p_array, 1, sizeof(p_array), image);
}

fclose(image);

1 个答案:

答案 0 :(得分:1)

首先:

您不希望在每个fwrite()

上存储相同数量的字节
fwrite(p_array, 1, sizeof(p_array), image);

但只是实际读取的字节数。

sizeof(p_array)返回p_array的大小,这是size大小的char倍。后者被定义为1

除此之外,对read() 的完全的调用缺少错误检查,以及测试对方是否关闭连接。这远远不够。

要解决这个问题,你可以这样做:

#include <errno.h> /* for errno */

...

  size_t size = atoi(message_text);
  printf("Reading Picture Byte Array\n");
  char p_array[size];
  size_t readSize = 0;
  while (readSize < size) {
    ssize_t result = read(clients[sender].socket, p_array, size); /* Mind the ssize_t, it's *not* size_t! */
    if (0 >= result)
    {
      if (0 == result)
      {
        fprintf(stderr, "The other end gracefully shut down the connection.\n");

        break;
      }
      else
      {
        if (EINTR == errno)  /* got interrupted, start over */
        {
           continue;
        }

        if (EAGAIN == errno)  /* in case reading from a non-blocking socket: no data available, start over */
        {
           continue;
        }

        /* Something went wrong unrecoverable. */
        perror("read() failed");

        break;
      }
    }
    else
    {
       fwrite(p_array, 1, result, image); 
       /* Also add error checking here! */
       readSize += result;
    }
 }