C#开发SFTP服务器：消息实际长度与头部长度不符的处理咨询

问题背景 #

我用C#开发了一款包含SSH层的SFTP服务器，SFTP消息存储在SSH层的"ssh data"字段中。传递到SFTP层的消息有固定头部：

• Message Length（消息长度）
• Message Type（消息类型）
• Request ID（请求ID）

现在遇到两个核心问题：

1. 接收消息时经常出现实际长度小于头部标注的预期长度，这种情况该选择聚合消息与下一个数据包、断开客户端连接还是其他处理方式？
2. 如果实际长度大于预期长度，又该如何处理？

当前SFTP层代码示例 #

public void OnInput(byte[] data)
{
    DataWorker reader = new DataWorker(data);             

    if (reader.DataAvailable >= 5)
    {
        try
        {
            var msglength = reader.ReadUInt32();
 
            if (reader.DataAvailable != msglength)
            {
                //DoSome
            }

            var msgtype = (SftpMessageType)(int)reader.ReadByte();

            switch (msgtype)
            {
                case SftpMessageType.SSH_FXP_INIT:
                    _logger.Log(LogLevel.Information, "FXP_INIT");
                    HandleInit(reader);
                    break;

                //etc..
            }
        }
        catch (Exception e)
        {
            _channel?.SendClose(1);
            return;
        }
    }
    else
    {
        _logger.Log(LogLevel.Error, $"no data avialble. data length is: {reader.DataAvailable}.");
    }
}

解决方案 #

1. 实际长度小于预期长度的处理 #

这是TCP流式传输的正常分包场景，聚合消息是标准且正确的处理方式。SFTP基于SSH协议，而SSH运行在TCP之上——TCP本身是无边界的流式协议，不会保证一次发送完整的应用层消息，因此必须由上层应用自行维护缓冲区拼接数据。

具体实现逻辑：

• 维护一个类级别的缓冲区（比如MemoryStream或List<byte>），每次收到新数据时追加到缓冲区中
• 循环检查缓冲区是否包含完整消息：
  1. 先读取头部4字节的Message Length，计算完整消息的总长度（4字节长度字段 + 消息长度值）
  2. 如果缓冲区当前长度 >= 完整消息总长度，解析这条消息，然后从缓冲区中移除已解析的部分，继续处理剩余内容
  3. 如果缓冲区长度不足，等待下一次数据到来后重复检查
• 除非出现超时、缓冲区溢出等异常情况，否则不要直接断开连接。

2. 实际长度大于预期长度的处理 #

这种情况属于协议违规，处理分两种场景：

• 严格合规模式：直接断开客户端连接。该情况说明客户端发送了不符合SFTP协议规范的数据，可能是恶意攻击或客户端实现存在严重bug，继续连接可能带来安全风险或后续解析混乱。
• 兼容容错模式：如果需要兼容某些有问题的客户端，可以尝试：
  1. 先解析头部标注长度对应的消息内容并处理
  2. 将剩余字节作为下一条消息的起始部分，放入缓冲区等待后续数据拼接
     但这种方式需谨慎，仅在明确客户端行为且必须兼容时使用，否则可能引发连锁解析错误。

代码优化方向 #

当前代码未维护缓冲区，仅处理单块数据，这是导致分包问题的核心原因。修改示例思路：

private MemoryStream _messageBuffer = new MemoryStream();

public void OnInput(byte[] data)
{
    // 将新数据追加到缓冲区
    _messageBuffer.Write(data, 0, data.Length);
    _messageBuffer.Position = 0;
    DataWorker reader = new DataWorker(_messageBuffer);

    while (reader.DataAvailable >= 4) // 至少能读取长度字段
    {
        try
        {
            var originalPosition = reader.Position;
            var msgLength = reader.ReadUInt32();
            var totalMsgLength = 4 + msgLength; // 包含长度字段本身的总长度

            // 剩余数据不足以构成完整消息，回退指针并退出循环
            if (_messageBuffer.Length < totalMsgLength)
            {
                reader.Position = originalPosition;
                break;
            }

            // 解析并处理消息
            var msgType = (SftpMessageType)(int)reader.ReadByte();
            switch (msgType)
            {
                case SftpMessageType.SSH_FXP_INIT:
                    _logger.Log(LogLevel.Information, "FXP_INIT");
                    HandleInit(reader);
                    break;
                // 其他消息类型处理
            }

            // 移除已解析的内容，保留剩余数据
            var remainingData = new byte[_messageBuffer.Length - totalMsgLength];
            _messageBuffer.Position = totalMsgLength;
            _messageBuffer.Read(remainingData, 0, remainingData.Length);
            _messageBuffer.SetLength(0);
            _messageBuffer.Write(remainingData, 0, remainingData.Length);
            _messageBuffer.Position = 0;
            reader = new DataWorker(_messageBuffer);
        }
        catch (Exception e)
        {
            _logger.LogError(e, "消息解析异常");
            _channel?.SendClose(1);
            _messageBuffer.SetLength(0);
            return;
        }
    }
}

内容的提问来源于stack exchange，提问作者Ebay Eliav