From 9b18e9ed3c97170b5627c77075262bb1164d5ec9 Mon Sep 17 00:00:00 2001 From: Hatter Jiang Date: Mon, 1 Jan 2024 14:48:11 +0800 Subject: [PATCH] feat: fix buffer --- src/main/java/me/hatter/tools/secureeditor/Main.java | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/main/java/me/hatter/tools/secureeditor/Main.java b/src/main/java/me/hatter/tools/secureeditor/Main.java index 8140cc1..1fedb94 100644 --- a/src/main/java/me/hatter/tools/secureeditor/Main.java +++ b/src/main/java/me/hatter/tools/secureeditor/Main.java @@ -58,6 +58,7 @@ public class Main { final byte[] encrypted = gcmCrypt(keyBytes, nonceBytes, true, resultBytes); writeFile(file, encrypted); } + System.exit(0); } @@ -139,11 +140,10 @@ public class Main { final ByteArrayOutputStream baos = new ByteArrayOutputStream(); final byte[] buffer = new byte[1024 * 8]; try (FileInputStream fis = new FileInputStream(file)) { - final int len = fis.read(buffer); - if (len == -1) { - return baos.toByteArray(); + int len; + while ((len = fis.read(buffer)) != -1) { + baos.write(buffer, 0, len); } - baos.write(buffer, 0, len); } catch (IOException e) { throw new RuntimeException("Read file: " + file + " failed: " + e.getMessage(), e); }