我正在使用recvmsg和sendmsg通过异步STREAM套接字发送数据。要传输的数据量很大,介于15 MB和30 MB之间。

我不明白为什么数据到达损坏。 errno EINTR和EAGAIN中断了发送消息和接收功能。

我正在使用的功能是:

int receive_result_with_extra(int fd, struct syscall_result * result, int extra_size, char * buf) {
    struct iovec io[2];
    struct msghdr msg;
    int transfered=0, temp=0;
    const int total =  SIZE_RESULT + extra_size;

    CLEAN_MSG(&msg);
    memset(io, 0, sizeof(io));

    // result header
    io[1].iov_len=SIZE_RESULT;
    io[1].iov_base=result;
    // result buffer
    io[0].iov_len = extra_size;
    io[0].iov_base = buf;
    // iov struct
    msg.msg_iov=io;
    msg.msg_iovlen=2;

    do {
      temp = recvmsg(fd,&msg, 0);
      if ( temp < 0 && (errno==EAGAIN || errno == EINTR || errno == EWOULDBLOCK))
        continue;
      else if ( temp < 0 )
          die("Error receiveing data recvmsg (receive_result_with_extra)");
      transfered += temp;
    } while(transfered < total);

    if ( transfered < 0)
        die("recvmsg (receive_result_with_extra)");

    assert(transfered == total);
    return transfered;
}
int send_result_with_extra(int fd, struct syscall_result * result, int extra_size, char * buf) {

    struct iovec io[2];
    struct msghdr msg;
    int transfered=0, temp=0;
    const int total =  SIZE_RESULT + extra_size;

    CLEAN_MSG(&msg);
    memset(io, 0, sizeof(io));

    // result header
    io[1].iov_len=SIZE_RESULT;
    io[1].iov_base=result;
    // result buffer
    io[0].iov_len = extra_size;
    io[0].iov_base = buf;
    // iov struct
    msg.msg_iov=io;
    msg.msg_iovlen=2;

    do {
      temp=sendmsg(fd,&msg, 0);
      if ( temp < 0 && (errno == EINTR || errno == EAGAIN || errno == EWOULDBLOCK))
          continue;
      else if ( temp < 0)
          die("Failed sending data ( send_result_with_extra)");
      transfered += temp;
    } while( transfered < total);

    if ( transfered < 0)
        die("recvmsg (fstat handler)");

    assert(transfered == total);

    return transfered;
}

最佳答案

您无需在send/recv循环中更新iovecs,因此,如果数据太大而无法在单个网络数据包中传输,则将重复地将相同的前缀/recv发送到数据缓冲区的开头。

我将对发送和接收使用通用函数来更新iovec:

/**
 * Shrinks the iovec associated with a msg by a given number of bytes.
 *
 * @msg    The message whose iovec needs updating.
 * @bytes  Number of bytes to remove from the beginning of the iovec.
 *
 * @return Returns 0 iff the updated iovec is empty.
 */
int update_buffer(struct msghdr* msg, size_t bytes) {
  while (msg->msg_iovlen > 0) {
    if (bytes < msg->msg_iov[0].iov_len) {
      msg->msg_iov[0].iov_len -= bytes;
      msg->msg_iov[0].iov_base += bytes;
      return 1;
    }
    bytes -= msg->msg_iov[0].iov_len;
    ++msg->msg_iov;
    --msg->msg_iovlen;
  }
  return 0;
}

然后,可以在send/recv循环的情况下使用此更新函数的返回值。例如。:
do {
  do {
    temp = sendmsg(fd, &msg, 0);
  } while (temp < 0 && (errno == EINTR || errno == EAGAIN || errno == EWOULDBLOCK));
  if (temp < 0)
    die("Failed sending data (send_result_with_extra)");
} while (update_buffer(&msg, temp));

09-30 22:40
查看更多