lhq/FT546-WCS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: reject malformed history snapshot text

Browse Source
This commit is contained in:
zgw
2026-07-31 10:03:54 +08:00
parent 19807a5bf0
commit 969332c115
5 changed files with 26 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
javaTest/src/main/java/com/ft546/wcs/shadow/history/HistoricalQueueLoader.java
View File

@@ -8,6 +8,10 @@ import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.StandardCharsets;
import java.nio.ByteBuffer;
import java.nio.charset.CharacterCodingException;
import java.nio.charset.CharsetDecoder;
import java.nio.charset.CodingErrorAction;
import java.nio.file.Path;
import java.security.MessageDigest;
import java.net.URLClassLoader;
@@ -82,7 +86,16 @@ public final class HistoricalQueueLoader {
}
public static void verifySha256(String expected, byte[] bytes) {
byte[] canonicalBytes = new String(bytes, StandardCharsets.UTF_8)
CharsetDecoder decoder = StandardCharsets.UTF_8.newDecoder()
.onMalformedInput(CodingErrorAction.REPORT)
.onUnmappableCharacter(CodingErrorAction.REPORT);
String text;
try {
text = decoder.decode(ByteBuffer.wrap(bytes)).toString();
} catch (CharacterCodingException ex) {
throw new IllegalArgumentException("history snapshot must be valid UTF-8", ex);
}
byte[] canonicalBytes = text
.replace("\r\n", "\n")
.replace('\r', '\n')
.getBytes(StandardCharsets.UTF_8);

2
javaTest/src/main/java/com/ft546/wcs/shadow/history/HistoricalReplayReportWriter.java
View File

@@ -10,7 +10,6 @@ import java.nio.file.Path;
import java.security.MessageDigest;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.EnumMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
@@ -50,6 +49,7 @@ public final class HistoricalReplayReportWriter {
}
summary.put("vehicles", vehicles);
summary.put("taskTypes", taskTypes);
summary.put("dimensionScope", "REPRESENTATIVE_SAMPLE");
summary.put("replayableRows", paths.stream().filter(HistoricalPath::isReplayable).count());
summary.put("qualityIssueRows", paths.stream().filter(p -> !p.getIssues().isEmpty()).count());
Map<String, Integer> issueCounts = new LinkedHashMap<String, Integer>();