读取请求正文而不删除它

Question

所以我有一个MyHandler,它必须知道请求体内的内容:

class MyHandler
  include HTTP::Handler

  def call(context)
    p "MyHandler got body: " + context.request.body.not_nil!.gets_to_end
    call_next(context)
  end
end

server = HTTP::Server.new(42, [MyHandler.new]) do |context|
  p "Server got body: " + context.request.body.not_nil!.gets_to_end
end

正如预期的那样,在MyHandler读完之后,服务器收到一个空体.如何在不修改原始上下文的情况下复制正文？

Answer 1

Crystal支持流请求主体,这意味着一旦您在请求中流,IO是EOF,第二个处理程序无法读取任何数据.

解决此问题的一种简单方法是使用检索整个内容body_string = context.request.body.try(&.gets_to_end),然后使用将请求主体设置为返回的字符串context.request.body = body_string.这会将整个主体缓冲到内存中,然后将主体设置为存储在内存中的缓冲区.这种方法的缺点是攻击者可以发送一个无限大小的请求体并占用服务器上的所有内存,从而导致DOS攻击.另一个缺点是,如果您正在使用二进制数据,则需要使用#to_slice它来将字符串转换为切片.

解决DOS攻击问题的一种方法 - 如果你有一个最大的体型大小 - 是在身体太大的情况下使请求失败:

if body = context.request.body
  body_io = IO::Memory.new
  bytes_read = IO.copy(body, body_io, limit: 1_048_576) # 1GiB limit
  body_io.rewind
  if bytes_read == 1_048_576
    # Fail request
  end

  # use body_io

  body_io.rewind # Reset body_io to start
  context.request.body = body_io
end

如果你需要接受一个无限大小的主体,而不是将它缓冲到内存中,你应该创建一个自定义IO实现,它包装现有的主体IO并在里面运行所需的转换IO#read(Bytes).这种方法非常复杂,以前的方法几乎涵盖了所有情况,因此我不会为此选项提供代码示例.