Java输入流限制：防范DoS攻击

Question

我正在编写一个工具，给定任何URL，该工具将定期获取其输出。 问题在于，输出可能不是一个简单且轻量级的HTML页面（在大多数情况下会如此），而是一些繁重的数据流（即直接来自/ dev / urandom，可能是DoS攻击）。

我正在使用java.net.URL + java.net.URLConnection ，将连接和读取超时设置为30秒。 当前输入正在由java.io.BufferedReader使用readLine()读取。

可能的解决方案：

1. 逐字节使用java.io.BufferedReader.read() ，对它们进行计数并在达到限制后关闭连接。 问题在于，攻击者可能每29秒发送一个字节，因此几乎不会发生读取/连接超时（204800B * 29sec = 68天）
2. 将线程执行时间限制为1-5分钟，并使用java.io.BufferedReader.readLine() 。 这里有什么问题吗？

我觉得自己想重新发明轮子，解决方案非常简单，只是我不知道。

提前致谢。

Answer 1

您可以通过编写自己的FilterInputStream来封装您的bhhis，该FilterInputStream强制执行您要强制执行的任何操作，并将其放置在堆栈的底部，围绕连接输出流

但是，此建议和建议的补救措施仅在输出以分块传输模式到达时才起作用。 否则，HttpURLConnection可以在读取任何响应之前缓冲整个响应。 通常的解决方案是在防火墙中使用过滤器。

Answer 2

这里似乎有许多拒绝服务的途径。

一条巨大的大线吞噬了记忆。 可能最简单的方法是在遇到字符解码之前使用MeteredInputStream 。 在任何情况下，逐char读取char都会非常慢。 您可以一次读取一个很长的char[] ，但这可能会使代码复杂化。

与对手（或错误）打交道，一次保持许多连接。 您可能希望非阻塞I / O读取整个消息，然后正常进行。