反病毒本地引擎特惠

import os
import hashlib

def calculate_md5(file_path):
    hash_md5 = hashlib.md5()
    with open(file_path, "rb") as f:
        for chunk in iter(lambda: f.read(4096), b""):
            hash_md5.update(chunk)
    return hash_md5.hexdigest()

def scan_file(file_path, virus_signatures):
    file_md5 = calculate_md5(file_path)
    if file_md5 in virus_signatures:
        print(f"Virus detected: {file_path}")
    else:
        print(f"File is clean: {file_path}")

# 示例病毒签名数据库
virus_signatures = {
    "d41d8cd98f00b204e9800998ecf8427e": "Trojan.Example",
    "827ccb0eea8a706c4c34a16891f84e7b": "Virus.Example"
}

# 扫描指定目录下的所有文件
scan_directory = "/path/to/scan"
for root, dirs, files in os.walk(scan_directory):
    for file in files:
        file_path = os.path.join(root, file)
        scan_file(file_path, virus_signatures)