CTF-MISC-多文件处理

import os
import glob

def find_and_remove_hex(input_file, hex_str):
    with open(input_file, 'rb') as f:
        file_content = f.read()
    
    hex_content = file_content.hex()
    start_index = hex_content.find(hex_str)

    if start_index == -1:
        return None
    
    cleaned_hex_content = hex_content[start_index:]
    cleaned_binary_content = bytes.fromhex(cleaned_hex_content)

    return cleaned_binary_content

def main():
    tex_files = glob.glob('*.tex')

    for tex_file in tex_files:
        png_file = os.path.splitext(tex_file)[0] + '.png'
        cleaned_content = find_and_remove_hex(tex_file, '89504e47')

        if cleaned_content is not None:
            with open(png_file, 'wb') as f:
                f.write(cleaned_content)
        else:
            print(f"ERROR")

if __name__ == "__main__":
    main()