Constante de tamaño de cabecera

Author: rodgallex

pitea/constantes.py

@@ -35,6 +35,8 @@ def __init__(self):
             "Robot36": (Robot36, (320, 240)),
             "ScottieS1": (ScottieS1, (320, 256)),
         }
+
+        self._TAMAÑO_CABECERA_LSB = 32
 
         self._SCRIPT_PATH = "script_ejecucion.py"
         self._RESET = "\033[0m"

pitea/pitea/audio/OcultadorAudioLSB.py

@@ -1,4 +1,5 @@
 from pitea.audio.OcultadorAudio import OcultadorAudio
+from constantes import constantes
 
 
 class OcultadorAudioLSB(OcultadorAudio):
@@ -46,7 +47,7 @@ def _ocultar(self, datos_imagen):
         binarios_imagen = "".join(format(byte, "08b") for byte in datos_imagen)
 
         # Añadir una cabecera con el tamaño de los datos (32 bits)
-        tamano_datos = format(len(binarios_imagen), "032b")
+        tamano_datos = format(len(binarios_imagen), "0"+str(constantes._TAMAÑO_CABECERA_LSB)+"b")
         binarios_imagen = tamano_datos + binarios_imagen  # Cabecera + datos
 
         assert self._audio.getsampwidth() == 2, "El archivo de audio debe ser de 16 bits"
@@ -85,9 +86,9 @@ def _desocultar(self):
         datos_binarios = ""
         for i in range(len(frames)):
             datos_binarios += str(frames[i] & 1)  # Extraer el bit menos significativo
-            if len(datos_binarios) >= 32:
-                tamano_datos = int(datos_binarios[:32], 2)  # Leer el tamaño de los datos
-                datos_binarios = datos_binarios[32:]  # Eliminar la cabecera
+            if len(datos_binarios) >= constantes._TAMAÑO_CABECERA_LSB:
+                tamano_datos = int(datos_binarios[:constantes._TAMAÑO_CABECERA_LSB], 2)  # Leer el tamaño de los datos
+                datos_binarios = datos_binarios[constantes._TAMAÑO_CABECERA_LSB:]  # Eliminar la cabecera
                 break
 
         # Verificar que la cabecera tenga el tamaño correcto
@@ -96,10 +97,10 @@ def _desocultar(self):
 
         for i in range(len(frames)):
             datos_binarios += str(frames[i] & 1)  # Extraer el bit menos significativo
-            if len(datos_binarios) >= tamano_datos + 32:
+            if len(datos_binarios) >= tamano_datos + constantes._TAMAÑO_CABECERA_LSB:
                 break
 
-        datos_binarios = datos_binarios[32:]
+        datos_binarios = datos_binarios[constantes._TAMAÑO_CABECERA_LSB:]
 
         datos_extraidos = int(datos_binarios, 2).to_bytes(tamano_datos // 8, byteorder="big")
 

pitea/pitea/configuracion.toml

@@ -1,20 +1,14 @@
-# Define parámetros de cache, SSTV y ocultación de texto en imagen.
-
 [persistente]
 contador_cache = 0
-ult_fecha = "15-04-2025_16:31"
+ult_fecha = "02-05-2025_15:21"
 
 [Ajustes_sstv]
-# Modo SSTV por defecto: opciones -> "MartinM1", "Robot36", "ScottieS1"
 modo_sstv = "MartinM1"
-# Frecuencia de muestreo de audio en Hz para SSTV
 samples_per_sec = 48000
-# Profundidad de bits para el audio SSTV
 bits = 16
 
 [Ajustes_ocultador_imagen_text]
 tamanio_fuente = 17
 anchura_maxima = 800
 ruta_fuente = "../fuentes/DejaVuSans-Bold.ttf"
-# Indica si se debe usar GPU para el renderizado ("True" o "False")
 gpu = "True"

pitea/pitea/imagen/OcultadorImagenLSB.py

@@ -1,5 +1,5 @@
 from pitea.imagen.OcultadorImagen import OcultadorImagen
-
+from constantes import constantes
 
 
 class OcultadorImagenLSB(OcultadorImagen):
@@ -64,7 +64,7 @@ def _ocultar(self, datos, altura_imagen=None, anchura_imagen=None):
 
 
         # Añadir una cabecera con el tamaño de los datos (32 bits)
-        tamano_datos_binarios = format(len(datos_binarios), "032b")
+        tamano_datos_binarios = format(len(datos_binarios),  "0"+str(constantes._TAMAÑO_CABECERA_LSB)+"b")
 
         # Cabecera + Datos
         mensaje = tamano_datos_binarios + datos_binarios
@@ -124,11 +124,11 @@ def _desocultar(self):
                     datos_binarios += str(
                         pixel[canal] & 1
                     )  # Extraer el bit menos significativo
-                    if len(datos_binarios) >= 32:
+                    if len(datos_binarios) >= constantes._TAMAÑO_CABECERA_LSB:
                         tamano_datos = int(
-                            datos_binarios[:32], 2
+                            datos_binarios[:constantes._TAMAÑO_CABECERA_LSB], 2
                         )  # Leer el tamaño de los datos
-                        datos_binarios = datos_binarios[32:]  # Eliminar la cabecera
+                        datos_binarios = datos_binarios[constantes._TAMAÑO_CABECERA_LSB:]  # Eliminar la cabecera
 
                         # Si el tamaño de los datos es mayor que 0, seguimos extrayendo los datos
                         if tamano_datos > 0:
@@ -141,21 +141,21 @@ def _desocultar(self):
         if tamano_datos == 0:
             raise ValueError("No se pudo extraer el tamaño de los datos ocultos.")
 
-        while len(datos_binarios) < tamano_datos + 32:
+        while len(datos_binarios) < tamano_datos + constantes._TAMAÑO_CABECERA_LSB:
             for y in range(alto):
                 for x in range(ancho):
                     pixel = list(pixeles[x, y])
                     for canal in range(3):
                         datos_binarios += str(pixel[canal] & 1)
-                        if len(datos_binarios) >= tamano_datos + 32:
+                        if len(datos_binarios) >= tamano_datos + constantes._TAMAÑO_CABECERA_LSB:
                             break
-                    if len(datos_binarios) >= tamano_datos + 32:
+                    if len(datos_binarios) >= tamano_datos + constantes._TAMAÑO_CABECERA_LSB:
                         break
-                if len(datos_binarios) >= tamano_datos + 32:
+                if len(datos_binarios) >= tamano_datos + constantes._TAMAÑO_CABECERA_LSB:
                     break
 
         # Convertir los datos binarios en bytes
-        datos_binarios = datos_binarios[32:]  # Eliminar la cabecera
+        datos_binarios = datos_binarios[constantes._TAMAÑO_CABECERA_LSB:]  # Eliminar la cabecera
         datos_extraidos = int(datos_binarios, 2).to_bytes(
             tamano_datos // 8, byteorder="big"
         )

pitea/tests.py

@@ -319,9 +319,9 @@ def run_tests():
         test_ocultar_desocultar_aes_lsb_lsb()
         print("🧪 Prueba de ocultar y desocultar con cifrado None y LSB")
         test_ocultar_desocultar_none_lsb_lsb()
-        print("🧪 Prueba transformada de imagenes")
-        test_ocultar_desocultar_aes_lsb_lsb_imagen_impar()
         print("🧪 Prueba de ocultar y desocultar con cifrado AES y LSB, imagenes con dimesiones impares")
+        test_ocultar_desocultar_aes_lsb_lsb_imagen_impar()
+        print("🧪 Prueba transformada de imagenes")
         test_transformada()