From 82f0350af4254aea67174058bab1285fd7e42660 Mon Sep 17 00:00:00 2001 From: devoalda Date: Tue, 23 May 2023 11:21:45 +0800 Subject: [PATCH] Fixed Delimeter error for image steg --- Image_Steg/img_steg.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/Image_Steg/img_steg.py b/Image_Steg/img_steg.py index 6b3907b..4aaf542 100644 --- a/Image_Steg/img_steg.py +++ b/Image_Steg/img_steg.py @@ -64,12 +64,13 @@ class img_steg: # Split by 8 bits all_bytes = [binary_data[i: i + 8] for i in range(0, len(binary_data), 8)] # Convert from bits to characters - decoded_data = self.from_bin(''.join(all_bytes)) - if decoded_data.endswith(self.delimiter): - decoded_data = decoded_data[:-4] + decoded_data = "" + for byte in all_bytes: + decoded_data += chr(int(byte, 2)) + if decoded_data[-len(self.delimiter):] == self.delimiter: + break - decoded_data = decoded_data.split(self.delimiter)[0] - return decoded_data + return decoded_data[:-len(self.delimiter)] def encode(self, secret_data: str = "Hello World") -> np.ndarray: """