Repo-Mirror/Iris-World-Engine
9
0
Fork 0
mirror of https://github.com/VolmitSoftware/Iris.git synced 2025-12-19 15:09:18 +00:00
Code Issues Releases Activity

fix potential issues in the CountingDataInputStream

Browse Source
This commit is contained in:
Julian Krings
2025-08-24 17:20:01 +02:00
parent e8bfce469d
commit 29390c5e0a
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
core/src/main/java/com/volmit/iris/util/io/CountingDataInputStream.java
View File

@@ -44,14 +44,14 @@ public class CountingDataInputStream extends DataInputStream {
} }
@Override @Override
public int read(@NotNull byte[] b, int off, int len) throws IOException { public int read(byte @NotNull [] b, int off, int len) throws IOException {
int i = in.read(b, off, len); int i = in.read(b, off, len);
count(i); if (i != -1) count(i);
return i; return i;
} }
private void count(int i) { private void count(int i) {
count += i; count = Math.addExact(count, i);
if (mark == -1) if (mark == -1)
return; return;
@@ -69,6 +69,12 @@ public class CountingDataInputStream extends DataInputStream {
public synchronized void mark(int readlimit) { public synchronized void mark(int readlimit) {
if (!in.markSupported()) return; if (!in.markSupported()) return;
in.mark(readlimit); in.mark(readlimit);
if (readlimit <= 0) {
mark = -1;
markLimit = 0;
return;
}
mark = count; mark = count;
markLimit = readlimit; markLimit = readlimit;
} }