fz0x1/Stegano
1
0
Fork 0
mirror of https://github.com/cedricbonhomme/Stegano.git synced 2025-05-12 17:18:30 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refacto lsbset.reveal to use getpixel

Browse source
This commit is contained in:
Flavien 2022-11-08 21:44:43 +01:00
parent 257d2c2f68
commit 58dbb94c5e
1 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
stegano/lsbset/lsbset.py
View file

@ -25,11 +25,8 @@ __date__ = "$Date: 2016/03/13 $"
__revision__ = "$Date: 2019/05/31 $" __revision__ = "$Date: 2019/05/31 $"
__license__ = "GPLv3" __license__ = "GPLv3"
import sys
from typing import IO, Iterator, Union from typing import IO, Iterator, Union
from PIL import Image
from stegano import tools from stegano import tools
@ -120,8 +117,16 @@ def reveal(
while True: while True:
generated_number = next(generator) generated_number = next(generator)
# color = [r, g, b]
for color in img_list[generated_number][:3]: # ignore the alpha col = generated_number % width
row = int(generated_number / width)
# pixel = [r, g, b] or [r,g,b,a]
pixel = img.getpixel((col, row))
if img.mode == "RGBA":
pixel = pixel[:3] # ignore the alpha
for color in pixel:
buff += (color & 1) << (tools.ENCODINGS[encoding] - 1 - count) buff += (color & 1) << (tools.ENCODINGS[encoding] - 1 - count)
count += 1 count += 1
if count == tools.ENCODINGS[encoding]: if count == tools.ENCODINGS[encoding]:
@ -132,5 +137,6 @@ def reveal(
limit = int("".join(bitab[:-1])) limit = int("".join(bitab[:-1]))
else: else:
raise IndexError("Impossible to detect message.") raise IndexError("Impossible to detect message.")
if len(bitab) - len(str(limit)) - 1 == limit: if len(bitab) - len(str(limit)) - 1 == limit:
return "".join(bitab)[len(str(limit)) + 1 :] return "".join(bitab)[len(str(limit)) + 1 :]