Mongoose TLS 数据接收问题分析与解决方案

问题背景

在使用 Mongoose 网络库的 Mbed TLS 实现时，开发者遇到了一个 TLS 数据接收问题：当服务器快速连续发送多条消息时，只有第一条消息能够被及时处理，后续消息会出现显著延迟。这一问题在使用 Mongoose 内置 TCP 栈和 Mbed TLS 2.28.1 版本的 Raspberry Pi Pico SDK 环境中尤为明显。

问题现象

具体表现为：

1. 当 MQTT 客户端订阅多个主题时
2. 服务器快速连续发布多条消息（如客户端连接数更新和测试消息）
3. 客户端只能及时处理第一条消息
4. 后续消息需要等待较长时间才能被处理

通过 Wireshark 抓包分析，发现多个 TLS 记录被封装在同一个 TCP 数据包中，但 Mongoose 只处理了第一个记录。

技术分析

根本原因

Mongoose 的 TLS 数据处理机制存在以下问题：

1. 单次读取限制：当前实现只对每个 TCP 数据包调用一次 mbedtls_ssl_read，即使缓冲区中仍有未处理的 TLS 记录
2. 轮询机制不足：内置 TCP 栈的轮询机制没有充分考虑 TLS 层可能存在的未处理数据
3. 缓冲区管理：TLS 解密后的数据直接放入接收缓冲区，但没有充分处理多记录情况

影响范围

该问题主要影响：

• 使用 Mongoose 内置 TCP 栈的场景
• 高延迟网络环境（更容易出现多记录打包）
• 需要快速连续处理多条消息的应用（如 MQTT）

解决方案探索

开发者提出了几种解决方案：

方案一：接收循环

在 read_conn 函数中添加循环，持续调用 mg_tls_recv 直到返回 MBEDTLS_ERR_SSL_WANT_READ 或其他错误：

while(true) {
  long n = mg_tls_recv(c, &io->buf[io->len], io->size - io->len);
  if (n == MG_IO_ERR) {
    mg_error(c, "TLS recv error");
    break;
  } else if (n > 0) {
    io->len += (size_t) n;
    mg_call(c, MG_EV_READ, &n);
  } else {
    break;
  }
}

方案二：轮询机制增强

修改 mg_mgr_poll 函数，增加对未处理 TLS 数据的检查：

if (c->is_tls && mg_tls_pending(c) > 0)
  handle_tls_recv((struct mg_iobuf *) &c->rtls);

方案三：TLS 接收函数改进

修改 mg_tls_recv 实现，使其能够处理多个 TLS 记录：

long mg_tls_recv(struct mg_connection *c, void *buf, size_t len) {
  struct mg_tls *tls = (struct mg_tls *) c->tls;
  size_t total_read = 0;
  
  while (total_read < len) {
    long n = mbedtls_ssl_read(&tls->ssl, buf + total_read, len - total_read);
    if(n > 0) {
      total_read += (size_t) n;
    } else if (n == MBEDTLS_ERR_SSL_WANT_READ || n == MBEDTLS_ERR_SSL_WANT_WRITE) {
      return total_read ? (long)total_read : MG_IO_WAIT;
    } else {
      return total_read ? (long)total_read : MG_IO_ERR;
    }
  }
  return (long) total_read;
}

最佳实践建议

1. 网络环境考虑：在高延迟网络环境中应特别注意此问题
2. 缓冲区管理：合理设置接收缓冲区大小以容纳可能的多个 TLS 记录
3. 错误处理：完善错误处理逻辑，特别是部分读取情况的处理
4. 测试验证：在模拟高延迟环境下进行充分测试

结论

Mongoose 在处理多个 TLS 记录时存在数据接收不完全的问题，特别是在使用内置 TCP 栈和高延迟网络环境下。通过改进接收循环、增强轮询机制或修改 TLS 接收函数实现，可以有效解决这一问题。开发者应根据具体应用场景选择最适合的解决方案，并在实际部署前进行充分测试验证。

对于关键业务系统，建议考虑使用经过充分验证的 TLS 实现，并关注 Mongoose 官方后续的修复和更新。