修复Test中System.in.read()导致的内存溢出

xiangyuecn · Feb 2, 2023 · bbcc9c1 · bbcc9c1
1 parent 829bc4e
commit bbcc9c1
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Test.java b/Test.java
@@ -736,6 +736,11 @@ static public String ReadIn() throws Exception {
 			if(byt=='\n') {
 				break;
 			}
+			if(in.size()>=2048) {//防止内存溢出，某些环境下可能会有无限的输入
+				byte[] bytes=in.toByteArray();
+				in=new ByteArrayOutputStream();
+				in.write(bytes, bytes.length-1024, 1024);
+			}
 			in.write(byt);
 		}
 		return in.toString();