Java 中 BufferedReader.readLine() 的最大行长？

Question

我使用 BufferedReader 的 readLine() 方法从套接字读取文本行。

没有明显的方法来限制读取的行长度。

我担心数据源会（恶意或错误地）写入大量数据而没有任何换行符，这将导致 BufferedReader 分配无限量的内存。

有没有办法避免这种情况？还是我必须自己实现readLine() 的有界版本？

Answer 1

最简单的方法是实现您自己的有界线阅读器。

或者更简单，重用来自this BoundedBufferedReader class的代码。

实际上，编写与标准方法相同的readLine() 并非易事。正确处理 3 种行终止符需要一些非常仔细的编码。将上述链接的不同方法与 BufferedReader 的 Sun version 和 Apache Harmony version 进行比较是很有趣的。

注意：我不完全相信有界版本或 Apache 版本是 100% 正确的。有界版本假设底层流支持标记和重置，这当然不总是正确的。如果 Apache 版本将 CR 视为缓冲区中的最后一个字符，则它似乎会预读一个字符。读取用户输入的输入时，这会在 MacOS 上中断。 Sun 版本通过设置一个标志来处理这个问题，以导致在下一个read... 操作时跳过 CR 之后可能的 LF；即没有虚假的预读。

Answer 2

另一个选项是 Apache Commons 的BoundedInputStream：

InputStream bounded = new BoundedInputStream(is, MAX_BYTE_COUNT);
BufferedReader reader = new BufferedReader(new InputStreamReader(bounded));
String line = reader.readLine();

Answer 3

字符串的限制是 20 亿个字符。如果希望限制更小，则需要自己读取数据。您可以从缓冲流中一次读取一个字符，直到达到限制或换行字符。

Answer 4

也许最简单的解决方案是采用稍微不同的方法。与其试图通过限制一次特定的读取来阻止 DoS，不如限制读取的原始数据的总量。通过这种方式，您无需担心每次读取和循环都使用特殊代码，只要分配的内存与传入数据成比例。

您可以测量Reader，或者可能更恰当地测量未解码的Stream 或等效值。

Answer 5

有几种方法可以解决这个问题：

• 如果总体数据量非常小，则将数据从套接字加载到缓冲区（字节数组、字节缓冲区，取决于您的喜好），然后将 BufferedReader 包裹在内存中的数据周围（通过 ByteArrayInputStream 等） ;
• 如果发生 OutOfMemoryError，只需捕获它；捕获这个错误一般是不可靠的，但是在捕获数组分配失败的特定情况下，它基本上是安全的（但不能解决一个线程从堆中分配大量可能对其他线程产生的任何连锁反应的问题例如，在您的应用程序中运行）；
• 实现一个只读取这么多字节的包装器 InputStream，然后将其插入到套接字和 BufferedReader 之间；
• 抛弃 BufferedReader 并通过正则表达式框架拆分行（实现一个 CharSequence，其字符从流中提取，然后定义一个限制行长的正则表达式）；原则上，CharSequence 应该是随机访问，但对于简单的“行拆分”正则表达式，实际上您可能会发现始终请求连续的字符，这样您就可以在实现中“作弊”。

Answer 6

在BufferedReader中，不要使用String readLine()，而是使用int read(char[] cbuf, int off, int len)；然后，您可以使用boolean ready() 来查看您是否得到了所有内容，并使用构造函数String(byte[] bytes, int offset, int length) 将其转换为字符串。

如果您不关心空格，而只想每行有最大字符数，那么 Stephen 提出的建议非常简单，

import java.io.BufferedReader;
import java.io.IOException;

public class BoundedReader extends BufferedReader {

    private final int  bufferSize;
    private       char buffer[];

    BoundedReader(final BufferedReader in, final int bufferSize) {
        super(in);
        this.bufferSize = bufferSize;
        this.buffer     = new char[bufferSize];
    }

    @Override
    public String readLine() throws IOException {
        int no;

        /* read up to bufferSize */
        if((no = this.read(buffer, 0, bufferSize)) == -1) return null;
        String input = new String(buffer, 0, no).trim();

        /* skip the rest */
        while(no >= bufferSize && ready()) {
            if((no = read(buffer, 0, bufferSize)) == -1) break;
        }

        return input;
    }

}

编辑：这是为了从用户终端读取行。它一直阻塞到下一行，并返回一个bufferSize-bounded String；该行的任何进一步输入都将被丢弃。