def test_a2bits_list_UTF8(self): list_of_bits = tools.a2bits_list("Hello World!") self.assertEqual(list_of_bits, [ '01001000', '01100101', '01101100', '01101100', '01101111', '00100000', '01010111', '01101111', '01110010', '01101100', '01100100', '00100001' ])
def hide(input_image_file, message, generator, encoding='UTF-8', auto_convert_rgb=False): """Hide a message (string) in an image with the LSB (Least Significant Bit) technique. """ message_length = len(message) assert message_length != 0, "message length is zero" img = Image.open(input_image_file) if img.mode not in ['RGB', 'RGBA']: if not auto_convert_rgb: print('The mode of the image is not RGB. Mode is {}'.\ format(img.mode)) answer = input('Convert the image to RGB ? [Y / n]\n') or 'Y' if answer.lower() == 'n': raise Exception('Not a RGB image.') img = img.convert('RGB') img_list = list(img.getdata()) width, height = img.size index = 0 message = str(message_length) + ":" + str(message) message_bits = "".join(tools.a2bits_list(message, encoding)) message_bits += '0' * ((3 - (len(message_bits) % 3)) % 3) npixels = width * height len_message_bits = len(message_bits) if len_message_bits > npixels * 3: raise Exception("The message you want to hide is too long: {}". \ format(message_length)) while index + 3 <= len_message_bits: generated_number = next(generator) r, g, b, *a = img_list[generated_number] # Change the Least Significant Bit of each colour component. r = tools.setlsb(r, message_bits[index]) g = tools.setlsb(g, message_bits[index + 1]) b = tools.setlsb(b, message_bits[index + 2]) # Save the new pixel if img.mode == 'RGBA': img_list[generated_number] = (r, g, b, a[0]) else: img_list[generated_number] = (r, g, b) index += 3 # create empty new image of appropriate format encoded = Image.new('RGB', (img.size)) # insert saved data into the image encoded.putdata(img_list) return encoded
def hide(input_image: Union[str, IO[bytes]], message: str, generator: Iterator[int], encoding: str = 'UTF-8', auto_convert_rgb: bool = False): """Hide a message (string) in an image with the LSB (Least Significant Bit) technique. """ message_length = len(message) assert message_length != 0, "message length is zero" img = Image.open(input_image) if img.mode not in ['RGB', 'RGBA']: if not auto_convert_rgb: print('The mode of the image is not RGB. Mode is {}'.\ format(img.mode)) answer = input('Convert the image to RGB ? [Y / n]\n') or 'Y' if answer.lower() == 'n': raise Exception('Not a RGB image.') img = img.convert('RGB') img_list = list(img.getdata()) width, height = img.size index = 0 message = str(message_length) + ":" + str(message) message_bits = "".join(tools.a2bits_list(message, encoding)) message_bits += '0' * ((3 - (len(message_bits) % 3)) % 3) npixels = width * height len_message_bits = len(message_bits) if len_message_bits > npixels * 3: raise Exception("The message you want to hide is too long: {}". \ format(message_length)) while index + 3 <= len_message_bits : generated_number = next(generator) r, g, b, *a = img_list[generated_number] # Change the Least Significant Bit of each colour component. r = tools.setlsb(r, message_bits[index]) g = tools.setlsb(g, message_bits[index+1]) b = tools.setlsb(b, message_bits[index+2]) # Save the new pixel if img.mode == 'RGBA': img_list[generated_number] = (r, g , b, a[0]) else: img_list[generated_number] = (r, g , b) index += 3 # create empty new image of appropriate format encoded = Image.new('RGB', (img.size)) # insert saved data into the image encoded.putdata(img_list) return encoded
def hide(input_image: Union[str, IO[bytes]], message: str, encoding: str = 'UTF-8', auto_convert_rgb: bool = False): """Hide a message (string) in an image with the LSB (Least Significant Bit) technique. """ message_length = len(message) assert message_length != 0, "message length is zero" img = Image.open(input_image) if img.mode not in ['RGB', 'RGBA']: if not auto_convert_rgb: print('The mode of the image is not RGB. Mode is {}'.\ format(img.mode)) answer = input('Convert the image to RGB ? [Y / n]\n') or 'Y' if answer.lower() == 'n': raise Exception('Not a RGB image.') img = img.convert('RGB') encoded = img.copy() width, height = img.size index = 0 message = str(message_length) + ":" + str(message) message_bits = "".join(tools.a2bits_list(message, encoding)) message_bits += '0' * ((3 - (len(message_bits) % 3)) % 3) npixels = width * height len_message_bits = len(message_bits) if len_message_bits > npixels * 3: raise Exception("The message you want to hide is too long: {}". \ format(message_length)) for row in range(height): for col in range(width): if index + 3 <= len_message_bits: # Get the colour component. pixel = img.getpixel((col, row)) r = pixel[0] g = pixel[1] b = pixel[2] # Change the Least Significant Bit of each colour component. r = tools.setlsb(r, message_bits[index]) g = tools.setlsb(g, message_bits[index + 1]) b = tools.setlsb(b, message_bits[index + 2]) # Save the new pixel if img.mode == 'RGBA': encoded.putpixel((col, row), (r, g, b, pixel[3])) else: encoded.putpixel((col, row), (r, g, b)) index += 3 else: img.close() return encoded
def hide(input_image: Union[str, IO[bytes]], message: str, encoding: str = 'UTF-8', auto_convert_rgb: bool = False): """Hide a message (string) in an image with the LSB (Least Significant Bit) technique. """ message_length = len(message) assert message_length != 0, "message length is zero" img = Image.open(input_image) if img.mode not in ['RGB', 'RGBA']: if not auto_convert_rgb: print('The mode of the image is not RGB. Mode is {}'.\ format(img.mode)) answer = input('Convert the image to RGB ? [Y / n]\n') or 'Y' if answer.lower() == 'n': raise Exception('Not a RGB image.') img = img.convert('RGB') encoded = img.copy() width, height = img.size index = 0 message = str(message_length) + ":" + str(message) message_bits = "".join(tools.a2bits_list(message, encoding)) message_bits += '0' * ((3 - (len(message_bits) % 3)) % 3) npixels = width * height len_message_bits = len(message_bits) if len_message_bits > npixels * 3: raise Exception("The message you want to hide is too long: {}". \ format(message_length)) for row in range(height): for col in range(width): if index + 3 <= len_message_bits : # Get the colour component. pixel = img.getpixel((col, row)) r = pixel[0] g = pixel[1] b = pixel[2] # Change the Least Significant Bit of each colour component. r = tools.setlsb(r, message_bits[index]) g = tools.setlsb(g, message_bits[index+1]) b = tools.setlsb(b, message_bits[index+2]) # Save the new pixel if img.mode == 'RGBA': encoded.putpixel((col, row), (r, g, b, pixel[3])) else: encoded.putpixel((col, row), (r, g, b)) index += 3 else: img.close() return encoded
def hide(input_image_file, message, generator, auto_convert_rgb=False): """ Hide a message (string) in an image with the LSB (Least Significant Bit) technique. """ message_length = len(message) assert message_length != 0, "message length is zero" img = Image.open(input_image_file) if img.mode != "RGB": print("The mode of the image is not RGB. Mode is {}".format(img.mode)) if not auto_convert_rgb: answer = input("Convert the image to RGB ? [Y / n]\n") or "Y" if answer.lower() == "n": raise Exception("Not a RGB image.") img = img.convert("RGB") img_list = list(img.getdata()) width, height = img.size index = 0 message = str(len(message)) + ":" + str(message) # message_bits = tools.a2bits(message) message_bits = "".join(tools.a2bits_list(message)) message_bits += "0" * ((3 - (len(message_bits) % 3)) % 3) npixels = width * height len_message_bits = len(message_bits) if len_message_bits > npixels * 3: raise Exception("The message you want to hide is too long: {}".format(message_length)) while index + 3 <= len_message_bits: generated_number = next(generator) (r, g, b) = img_list[generated_number] # Change the Least Significant Bit of each colour component. r = tools.setlsb(r, message_bits[index]) g = tools.setlsb(g, message_bits[index + 1]) b = tools.setlsb(b, message_bits[index + 2]) # Save the new pixel img_list[generated_number] = (r, g, b) index += 3 # create empty new image of appropriate format encoded = Image.new("RGB", (img.size)) # insert saved data into the image encoded.putdata(img_list) return encoded
def test_a2bits_list_UTF32LE(self): list_of_bits = tools.a2bits_list("Hello World!", 'UTF-32LE') self.assertEqual(list_of_bits, ['00000000000000000000000001001000', '00000000000000000000000001100101', '00000000000000000000000001101100', '00000000000000000000000001101100', '00000000000000000000000001101111', '00000000000000000000000000100000', '00000000000000000000000001010111', '00000000000000000000000001101111', '00000000000000000000000001110010', '00000000000000000000000001101100', '00000000000000000000000001100100', '00000000000000000000000000100001'])
def get_max_message(input_image, message, encoding: str = 'UTF-8'): message = message + ' ' img = Image.open(input_image) width, height = img.size n_pixels = width * height message_length = len(message) calc_message = str(message_length) + ":" + str(message) message_bits = "".join(tools.a2bits_list(calc_message, encoding)) message_bits += '0' * ((3 - (len(message_bits) % 3)) % 3) len_message_bits = len(message_bits) max_len = n_pixels * 3 return message * (max_len // len_message_bits)
def test_a2bits_list_UTF32LE(self): list_of_bits = tools.a2bits_list("Hello World!", "UTF-32LE") self.assertEqual( list_of_bits, [ "00000000000000000000000001001000", "00000000000000000000000001100101", "00000000000000000000000001101100", "00000000000000000000000001101100", "00000000000000000000000001101111", "00000000000000000000000000100000", "00000000000000000000000001010111", "00000000000000000000000001101111", "00000000000000000000000001110010", "00000000000000000000000001101100", "00000000000000000000000001100100", "00000000000000000000000000100001", ], )
def hide( input_image: Union[str, IO[bytes]], message: str, generator: Iterator[int], shift: int = 0, encoding: str = "UTF-8", auto_convert_rgb: bool = False, ): """Hide a message (string) in an image with the LSB (Least Significant Bit) technique. """ message_length = len(message) assert message_length != 0, "message length is zero" img = tools.open_image(input_image) if img.mode not in ["RGB", "RGBA"]: if not auto_convert_rgb: print("The mode of the image is not RGB. Mode is {}".format( img.mode)) answer = input("Convert the image to RGB ? [Y / n]\n") or "Y" if answer.lower() == "n": raise Exception("Not a RGB image.") img = img.convert("RGB") img_list = list(img.getdata()) width, height = img.size index = 0 message = str(message_length) + ":" + str(message) message_bits = "".join(tools.a2bits_list(message, encoding)) message_bits += "0" * ((3 - (len(message_bits) % 3)) % 3) npixels = width * height len_message_bits = len(message_bits) if len_message_bits > npixels * 3: raise Exception("The message you want to hide is too long: {}".format( message_length)) while shift != 0: next(generator) shift -= 1 while index + 3 <= len_message_bits: generated_number = next(generator) r, g, b, *a = img_list[generated_number] # Change the Least Significant Bit of each colour component. r = tools.setlsb(r, message_bits[index]) g = tools.setlsb(g, message_bits[index + 1]) b = tools.setlsb(b, message_bits[index + 2]) # Save the new pixel if img.mode == "RGBA": img_list[generated_number] = (r, g, b, *a) else: img_list[generated_number] = (r, g, b) index += 3 # create empty new image of appropriate format encoded = Image.new(img.mode, (img.size)) # insert saved data into the image encoded.putdata(img_list) return encoded