使用套接字发送和接收文件

时间:2012-08-29 11:57:20

标签: c sockets

这部分代码在接收文件时由客户端使用:

void do_retr_cmd(int f_sockd){
  int fd;
  ssize_t nread = 0;
  uint32_t fsize, fsize_tmp, total_bytes_read, size_to_receive;
  char *filename = NULL, *conferma = NULL, *filebuffer = NULL;
  char buf[256], dirp[256], t_buf[256];

  memset(dirp, 0, sizeof(dirp));
  memset(buf, 0, sizeof(buf));
  memset(t_buf, 0, sizeof(t_buf));
  printf("Write the name of file to download: ");
  fgets(dirp, BUFFGETS, stdin)
  filename = NULL;
  filename = strtok(dirp, "\n");
  sprintf(buf, "RETR %s", dirp);
  if(send(f_sockd, buf, strlen(buf), 0) < 0){
    perror("Errore durante l'invio del nome del file");
    onexit(f_sockd, 0, 0, 1);
  }
  fsize = 0;
  recv(f_sockd, t_buf, sizeof(t_buf), 0)
  fsize = atoi(t_buf);
  fd = open(filename, O_CREAT | O_WRONLY, 0644);
  fsize_tmp = fsize;
  filebuffer = (char *)malloc(fsize);
  total_bytes_read = 0;
  nread = 0;
  for(size_to_receive = fsize; size_to_receive > 0;){
    nread = read(f_sockd, filebuffer, size_to_receive);
    if(nread < 0){
      perror("read error on retr");
      onexit(f_sockd, 0, 0, 1);
    }
    if(write(fd, filebuffer, nread) != nread){
      perror("write error on retr");
      onexit(f_sockd, 0, 0, 1);
    }
    size_to_receive -= nread;
  }
  close(fd);
  fflush(stdout);
  fflush(stdin);
  memset(buf, 0, sizeof(buf));
  recv(f_sockd, buf, 21, 0)
  printf("%s", buf);
  memset(buf, 0, sizeof(buf));
  memset(t_buf, 0, sizeof(t_buf));
  memset(dirp, 0, sizeof(dirp));
  free(filebuffer);
}

这部分代码在发送文件时由服务器使用:

void do_server_retr_cmd(f_sockd, m_sockd){
  int fd, rc;
  uint32_t fsize, size_to_send;
  char *filename = NULL, *other = NULL;
  char buf[512], t_buf[256];
  off_t offset;
  struct stat fileStat;

  memset(buf, 0, sizeof(buf));
  memset(t_buf, 0, sizeof(t_buf));
  recv(f_sockd, buf, sizeof(buf), 0)
  other = NULL;
  filename = NULL;
  other = strtok(buf, " ");
  filename = strtok(NULL, "\n");

  if(strcmp(other, "RETR") == 0){
    printf("Ricevuta richiesta RETR\n");
  } else /* do something */

  fd = open(filename, O_RDONLY);

  memset(&fileStat, 0, sizeof(fileStat));
  fileStat.st_size = 0;
  fstat(fd, &fileStat)
  fsize = fileStat.st_size;
  snprintf(t_buf, 255, "%" PRIu32, fsize);
  send(f_sockd, t_buf, sizeof(t_buf), 0)
  offset = 0;
  for (size_to_send = fsize; size_to_send > 0; ){
    rc = sendfile(f_sockd, fd, &offset, size_to_send);
    if (rc <= 0){
      perror("sendfile");
      onexit(f_sockd, m_sockd, fd, 3);
    }
    size_to_send -= rc;
  }
  close(fd);
  fflush(stdout);
  fflush(stdin);
  memset(buf, 0, sizeof(buf));
  strcpy(buf, "226 File transfered\n");
  send(f_sockd, buf, strlen(buf), 0)
  memset(buf, 0, sizeof(buf));
  memset(t_buf, 0, sizeof(t_buf));
}

- &GT;错误检查已被省略&lt; -
这两段代码我遇到了很大的问题。当我启动主程序时,我必须写:
1. retr 然后我按下
2. Write the filename to download: 我写了文件名然后按回车
问题是,有时文件正确下载,有时它下载,但它的一部分显示在stdout(在终端上)。
我不明白为什么我得到了这个奇怪的行为。左:PS:我知道我的代码很丑,但我是C-newbie!

我正在开发Ubuntu amd64并使用GCC-4.6.3(C语言)。

3 个答案:

答案 0 :(得分:3)

TCP连接为您提供了可靠的双向字节流,但是您的应用程序消息的界限是&#34;未保留,这意味着可以在另一侧的多个send()中接收一个recv(),反之,可以将多个send()折叠为一个recv() (你可以收到你发送的最后一块的部分)。好处是你按照发送的顺序接收你发送的字节。

服务器代码中的行recv(f_sockd, buf, sizeof(buf), 0);假设您在此处读取文件名,而实际上您可以获得客户端发送的256字节。

您需要在裸TCP上施加某种应用程序级协议。一个非常简单的方法是在文件内容中以文本形式发送文本标题:

file-size file-name\n

因此,您的服务器可以查找第一个换行符,在第一个空格中分割行并具有预期的字节数,以及用于保存这些字节的文件名。在该换行符之后不要忽略接收缓冲区的其余部分,将其保存到文件中。这也使您可以重新使用该连接进行多个文件传输。

希望这有帮助。

答案 1 :(得分:1)

recv(f_sockd, buf, 21, 0)
printf("%s", buf);

printf打印了大量的随机垃圾,因为没有实际的协议来控制它接收的内容和打印的内容。例如,printf如何知道要打印多少字节?

答案 2 :(得分:1)

我的答案的先前版本不太正确,但这就是为什么你看到这种奇怪的行为。您将文件大小发送为

snprintf(t_buf, 255, "%" PRIu32, fsize);

然后您将收到

recv(f_sockd, t_buf, sizeof(t_buf), 0)

但不能保证实际读取sizeof(t_buf)个字节。然后atoi有时会返回不正确的大小,文件的其余部分将被视为状态消息,该消息将在末尾打印(直到第一个空字符)。

由于recv可能无法同时返回您要求的所有数据,因此您必须检查其返回值并可能重复调用recv

size_t to_recv = sizeof(t_buf);
size_t rcvd = 0;
while (to_recv > 0) {
    ssize_t r = recv(f_sockd, t_buf + rcvd, sizeof(t_buf) - rcvd, 0);
    if (r < 0) {
        //error
    }
    else {
        to_recv -= r;
        rcvd += r;
    }
}

显然,你要么必须知道预期的数据量,要么提出其他答案中建议的更好的协议(例如,寻找终结器以确定何时读取了大小)。