tests aes_gcm

Author: seathiel-12

src/detecteur_crypto.py

@@ -10,7 +10,7 @@
 from .analyzers.blowfish_analyzer import Blowfish_Analyzer
 from .analyzers.aes_gcm_analyzer import Aes_Gcm_Analyzer
 from .rapport_mission import generer_rapport_mission
-
+from .analyzers.fernet_analyzer import FernetAnalyzer
 # Import des modules utilitaries
 from utils import est_dechiffre
 
@@ -42,7 +42,8 @@ def __init__(self):
             "AES-CBC": Aes_Cbc_Analyzer(),
             "ChaCha20": ChaCha20_Analyzer(),
             "Blowfish": Blowfish_Analyzer(),
-            "AES-GCM": Aes_Gcm_Analyzer()
+            "AES-GCM": Aes_Gcm_Analyzer(),
+            "Fernet": FernetAnalyzer(),
         }
         self.missions_completees: list[dict[str, Union[str, list[ResultatAnalyse], float]]]  = []
         self.statistiques_globales: dict[str, Union[int, float]] = {

src/utils.py

@@ -1,4 +1,5 @@
 import math
+import re
 import string
 from pathlib import Path
 from typing import Any, Dict, List, TypedDict
@@ -58,7 +59,7 @@ def est_dechiffre(texte:str) -> bool:
     if stats['ponctuation_valide'] > 50 :
         pourcent += 20
 
-    return True if pourcent > 70 else False
+    return True if pourcent > 80 else False
 
 
 
@@ -119,7 +120,7 @@ def verifier_texte_dechiffre(texte: str) -> Dict[str, Any]:
                 try:
                     with open(chemin, 'r', encoding='latin-1') as f: 
                         for ligne in f:
-                            if ligne.strip() == mot:
+                            if  re.match(ligne.strip().removesuffix('\n'), mot, re.I):
                                 mots_valides += 1
                                 trouve=True
                                 break

tests/fichiers_pour_tests/aes_gcm_invalide.enc

@@ -0,0 +1 @@
+_����̎��~�lY{�攳F���y�x��FN��fY��k]8]0���ވ�M��[DW�&��W�Z�3�A���	p~�uV�

tests/test_analyzers.py

@@ -3,12 +3,14 @@
 import sys
 import hashlib
 from cryptography.hazmat.primitives.ciphers.aead import ChaCha20Poly1305
-
+from cryptography.hazmat.primitives.ciphers.aead import AESGCM
+from pathlib import Path
 sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
 sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..'))
 
 from src.analyzers.aes_cbc_analyzer import Aes_Cbc_Analyzer
 from src.analyzers.chacha20_analyzer import ChaCha20_Analyzer
+from src.analyzers.aes_gcm_analyzer import Aes_Gcm_Analyzer
 
 
 class AesCbcAnalyzerTester(TestCase):
@@ -104,6 +106,42 @@ def test_chacha20_dechiffrer_fichier_non_existant(self):
         with self.assertRaises(FileNotFoundError):
             self.analyser_chacha.dechiffrer("chemin_invalide.enc", cle_valide)
 
-            
+class AesGcmTester(TestCase) :
+    _wordlist = "keys/wordlist.txt"
+    _analyzer=Aes_Gcm_Analyzer()
+    _fichier="data/mission3.enc"
+    _fichier_test = Path('tests/fichiers_pour_tests') / 'aes_gcm_invalide.enc'
+    _texte_test = b"Test effectue pour AesGcm, encore. Nous en sommes a la.fin"
+    
+    
+    def setUp(self): 
+        """
+        Crée un fchier de test crypté en AESGCM pour les tests unitaires
+        """
+        key = AESGCM.generate_key(128)
+        nonce = os.urandom(12)
+        aad = os.urandom(16)
+        texte_chiffre = AESGCM(key).encrypt(nonce, self._texte_test, aad)
+        with open(self._fichier_test, '+wb') as f:
+            f.write(nonce)
+            f.write(texte_chiffre)
+        f.close()
+        
+    def test_aesgcm_generer_cles_candidates(self):
+        #Vérifie que les clés candidates générés par cet algorithme sont une liste de bytes
+        with self.assertRaises(ValueError):
+            self.assertIsInstance(self._analyzer.generer_cles_candidates(self._wordlist), list[bytes])
+    
+    def test_aes_gcm_identifier_algo(self):
+        #Vérifie que la probabilité retournée pour le fichier mission3.enc est un float et élevée
+        with self.assertRaises(ValueError):
+            self.assertIsInstance(self._analyzer.identifier_algo(self._fichier_test), float)
+        self.assertAlmostEqual(self._analyzer.identifier_algo(self._fichier_test, 0))
+    
+    def test_aes_gcm_dechiffrer(self):
+        self.assertIsInstance(self._analyzer.dechiffrer(self._fichier_test), bytes)
+        
+      
+
 if __name__ == '__main__':
     main()