valentin/steganographie-python
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add insert and extract commands

Browse Source 
In order to be able to put and read binary files in an image
This commit is contained in:
Valentin Moguérou 2022-03-19 20:45:17 +01:00
parent 11bd28de6d
commit ba7c493208
2 changed files with 40 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
libstegano.py
View File

@ -63,6 +63,7 @@ def decode_string(bytelist: bytearray) -> str:
def write_band(band: bytearray, byte_list: bytearray, starting_pos=0) -> bool:
"Writes a byte sequence in the given band"
index = starting_pos
for index in range(starting_pos, len(band), 8):
band[index:index+8] = (write_lsb_in_bin(band[index+i], bit) for i, bit in enumerate(split_byte_into_bits(byte_list[index//8])))
if byte_list[index//8] == 0:

40
steganocli.py
View File

@ -50,6 +50,33 @@ def write(args):
write_file(from_file, to_file, byte_list, verbose=args.verbose)
def extract(args):
if args.verbose:
print("Extract mode enabled.")
with open(args.from_file, 'rb') as from_file, open(args.to_file, 'wb') as to_file:
byte_list = read_file(from_file, verbose=args.verbose)
if args.verbose:
print(f"Read {(len(byte_list)+1)*8} bits, {len(byte_list)+1} bytes.")
print("================== BYTE STREAM ==================")
hex_print(byte_list, margin=1)
print("================== BYTE STREAM ==================")
to_file.write(byte_list)
def insert(args):
with open(args.from_file, 'rb') as from_file, open(args.content_file, 'rb') as content_file, open(args.to_file, 'wb') as to_file:
byte_list = content_file.read()+b'\0'
if args.verbose:
print("Insert mode enabled.")
print("================== BYTE STREAM ==================")
hex_print(byte_list, margin=1)
print("================== BYTE STREAM ==================")
write_file(from_file, to_file, byte_list, verbose=args.verbose)
def main():
parser = ArgParser()
parser.add_argument('-v', '--verbose', action='store_true')
@ -60,12 +87,23 @@ def main():
parser_read.add_argument('from_file', type=Path)
parser_read.set_defaults(func=read)
parser_write = subparsers.add_parser('write', help='Dissimulate data in an image')
parser_write = subparsers.add_parser('write', help='Dissimulate a string in an image')
parser_write.add_argument('string', type=str)
parser_write.add_argument('to_file', type=Path)
parser_write.add_argument('--from', type=Path, dest='from_file', default=Path('default_image.jpg'))
parser_write.set_defaults(func=write)
parser_read = subparsers.add_parser('extract', help='Retrieve data from an image')
parser_read.add_argument('from_file', type=Path)
parser_read.add_argument('to_file', type=Path)
parser_read.set_defaults(func=extract)
parser_write = subparsers.add_parser('insert', help='Dissimulate data in an image')
parser_write.add_argument('content_file', type=Path)
parser_write.add_argument('to_file', type=Path)
parser_write.add_argument('--from', type=Path, dest='from_file', default=Path('default_image.jpg'))
parser_write.set_defaults(func=insert)
args = parser.parse_args()
args.func(args)