import pefile

def extract_text_section(pe_path, output_path):
    pe = pefile.PE(pe_path)

    # Parcourir les sections et trouver celle qui contient le code (.text)
    text_section = None
    for section in pe.sections:
        if section.Name.decode('utf-8').strip('x00') == '.text':
            text_section = section
            break

    if text_section is None:
        print("Aucune section .text n'a été trouvée.")
        return

    # Extraire les données brutes de la section .text
    raw_data = bytearray(text_section.get_data())

    # Enregistrer les données brutes dans un fichier binaire
    with open(output_path, 'wb') as output_file:
        output_file.write(raw_data)

if __name__ == "__main__":
    # Spécifier le chemin vers le fichier PE
    pe_path = "path/to/your/file.exe"

    # Spécifier le chemin de sortie pour les données brutes de la section .text
    output_path = "output/text_section.raw"

    # Appeler la fonction d'extraction
    extract_text_section(pe_path, output_path)