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

chg: [style] Reformat with black.

Browse source
This commit is contained in:
Cédric Bonhomme 2025-06-22 10:36:57 +02:00
parent 6fd80429a4
commit e8ed79ae9e
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
1 changed files with 14 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

21
stegano/wav/wav.py
View file

@ -25,13 +25,15 @@ __license__ = "GPLv3"
import wave import wave
from typing import IO, Union from typing import IO, Union
from stegano import tools from stegano import tools
def hide( def hide(
input_file: Union[str, IO[bytes]], input_file: Union[str, IO[bytes]],
message: str, message: str,
output_file: Union[str, IO[bytes]], output_file: Union[str, IO[bytes]],
encoding: str = "UTF-8" encoding: str = "UTF-8",
): ):
""" """
Hide a message (string) in a .wav audio file. Hide a message (string) in a .wav audio file.
@ -49,9 +51,11 @@ def hide(
nchannels, sampwidth, framerate, nframes, comptype, _ = input.getparams() nchannels, sampwidth, framerate, nframes, comptype, _ = input.getparams()
assert comptype == "NONE", "only uncompressed files are supported" assert comptype == "NONE", "only uncompressed files are supported"
nsamples = nframes * nchannels nsamples = nframes * nchannels
message_bits = f"{message_length:08b}" + "".join(tools.a2bits_list(message, encoding)) message_bits = f"{message_length:08b}" + "".join(
tools.a2bits_list(message, encoding)
)
assert len(message_bits) <= nsamples, "message is too long" assert len(message_bits) <= nsamples, "message is too long"
# copy over .wav params to output # copy over .wav params to output
@ -100,6 +104,9 @@ def reveal(input_file: Union[str, IO[bytes]], encoding: str = "UTF-8"):
message_bits += str(frames[i] & 1) message_bits += str(frames[i] & 1)
# Convert bits to string # Convert bits to string
chars = [chr(int(message_bits[i:i+encoding_len], 2)) for i in range(0, len(message_bits), encoding_len)] chars = [
chr(int(message_bits[i : i + encoding_len], 2))
for i in range(0, len(message_bits), encoding_len)
]
message = "".join(chars) message = "".join(chars)
return message return message