From 6283519edb2267f1be7b2a9493498e876db53ae1 Mon Sep 17 00:00:00 2001 From: Laurent Date: Fri, 24 Oct 2025 15:20:16 +0200 Subject: [PATCH] Fix message sending --- src/main/java/step1/Main.java | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/src/main/java/step1/Main.java b/src/main/java/step1/Main.java index b0e9b36..af2b86b 100644 --- a/src/main/java/step1/Main.java +++ b/src/main/java/step1/Main.java @@ -49,9 +49,9 @@ public class Main { System.out.println("Received message: " + text); //Encrypting the message - String encryptedMessage = encrypt(key, text); - String base64Message = Base64.getEncoder().encodeToString(encryptedMessage.getBytes()); - System.out.println("Encrypted message to send : " + encryptedMessage); + byte[] encryptedMessage = encrypt(key, text); + String base64Message = Base64.getEncoder().encodeToString(encryptedMessage); + System.out.println("Encrypted message to send : " + base64Message); //Sending encrypted message send(socket, base64Message); @@ -83,13 +83,12 @@ public class Main { return keyGen.generateKey(); } - public static String encrypt(SecretKey key, String message) throws Exception { + public static byte[] encrypt(SecretKey key, String message) throws Exception { Cipher cipher = Cipher.getInstance(TRANSFORMATION); cipher.init(Cipher.ENCRYPT_MODE, key); - byte[] ciphertext = cipher.doFinal(message.getBytes()); + return cipher.doFinal(message.getBytes()); - return Base64.getEncoder().encodeToString(ciphertext); } public static String decrypt(byte[] keyBytes, String base64IvAndCiphertext) throws Exception {