在 C++ 中解析基于 TCP 的 TLV 协议时出现问题；

Question

我正在尝试通过 TCP 实现 (T)LV 协议，从 python 客户端发送协议缓冲区并使用 C++ 服务器接收。

我的代码看起来或多或少像这样：

char* buffer[RCVBUFSIZE];
int recvsize= 0;
// Filling my buffer with as much as possible.
while(true) {
  if(recvsize == RCVBUFSIZE) {
    break;
  } else if(recvsize+= recv(sock, buffer+recvsize, sizeof(buffer)-recvsize, 0) < 1) {
    break;
  }
}
//Parsing LV protocol
while(true) {
  unsigned short protosize= 0;
  //Copy first two bytes into protosize
  memcpy((char *) &protosize, buffer, sizeof(unsigned short));
  if(protosize == 0) { break; } // Protocol indicates EOM be setting length to "0"
  void* protomsg[protosize];
  memcpy(protomsg, buffer+sizeof(unsigned short), protosize);
  int msglength= sizeof(unsigned short)+protosize;
  //Now I'll move the whole buffer to the left so that I don't have to keep track of where I'm at.
  memmove(buffer, buffer+msglength, RCVBUFSIZE-msglength);
  protoMsg pm;
  if(!pm.ParseFromArray(protomsg, protosize)) { break; } // Parsing failed.
  // Do stuff with parsed message.
}

现在我有几个问题：

• 接收消息的 while 循环永远不会终止。我怀疑当不再有任何数据时，recv 调用会阻塞，而我预计它会返回错误。我找到了 select 函数来检查是否有东西可读。我会尝试一下。 但是，当我仅调用 receive 一次来跳过此问题时（收到的消息大约有 10 个字节，因此我希望在一次调用中收集所有消息。）我遇到另一个问题：
• memcpy 和 memmove 似乎无法正常工作预期的。在第一个循环中，短路按预期进行处理（我收到的值与在另一端发送的值相同），但随后协议缓冲区解析的所有内容都失败了。我是否误解了什么？

编辑：关于 ntohs 的评论——我目前正在以小端方式传输短消息，忘了提及这一点。 （顺便说一句，我仍然会更改这一点。）

第三次编辑：代码现在可以工作，但我必须更改以下内容：

char* buffer[RCVBUFSIZE];
int recvsize= 0;
// Filling my buffer with as much as possible.
while(true) {
  if(recvsize == RCVBUFSIZE) {
    break;
  } else if((recvsize+= recv(sock, buffer+recvsize, sizeof(buffer)-recvsize, 0)) < 1) {
    break;
  } else if(recvsize > 1) {
    unsigned short msglength= 0;
    memcpy((char *) &msglength, buffer+recvsize-sizeof(unsigned short), sizeof(unsigned short));
    if(msglength == 0) { break; } // Received a full transmission.
  }
}

所以首先我需要在 recvsize+= recv() 周围添加括号 语句和第二个非阻塞方法由于某种原因不起作用，我现在正在检查在读取无符号短整型时传输的最后两个字节是否转换为 0。如果我偶然读取到不是长度字段的 0，这可能会导致问题。我可能会就此开始另一个问题。

我还将 protomsg 更改为 char[]，但我认为这并没有真正改变任何东西。 （我已经解析了一个空数组..）

Answer 1

如果您收到的消息始终在 10 个字节左右，并且 RCVBUFSIZE 大于此值，那么您将永远不会终止，直到读取数据时出现错误。另外，代码中的 buffer 变量是一个 RCVBUFSIZE 指针数组，可能不是您想要的。

修改您的代码如下：

#define MINIMALMESSAGESIZE 10  // Or what the size may be
char buffer[RCVBUFSIZE];
int totalrecvsize= 0;
int recvsize= 0;
while(true) {
  if(totalrecvsize >= MINIMALMESSAGESIZE) {
    break;
  } else if(recvsize= recv(sock, buffer+totalrecvsize, sizeof(buffer)-totalrecvsize, 0) < 1) {
    break;
  } else {
    totalrecvsize += recvsize;
  }
}