Serg4451D commited on
Commit
2f848a4
·
verified ·
1 Parent(s): aa95668

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +94 -0
app.py ADDED
@@ -0,0 +1,94 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import hashlib
2
+ import requests
3
+ import tkinter as tk
4
+ from tkinter import filedialog, messagebox, ttk
5
+
6
+ # Ссылка на файл с хешами MD5
7
+ MD5_HASHES_URL = "https://virusshare.com/hashfiles/VirusShare_00000.md5"
8
+
9
+ def calculate_file_hash(filepath, algorithm="md5"):
10
+ """Вычисляет хеш файла (по умолчанию MD5)."""
11
+ hasher = hashlib.new(algorithm)
12
+ with open(filepath, 'rb') as file:
13
+ while True:
14
+ chunk = file.read(4096)
15
+ if not chunk:
16
+ break
17
+ hasher.update(chunk)
18
+ return hasher.hexdigest()
19
+
20
+ def load_virus_hashes_from_url(url):
21
+ """Загружает хеши вирусов из файла по URL."""
22
+ try:
23
+ response = requests.get(url)
24
+ response.raise_for_status() # Проверка на ошибки HTTP
25
+ hashes = set()
26
+ for line in response.text.splitlines():
27
+ hashes.add(line.strip().split()[0].lower()) # Берем только MD5 хеш и приводим к нижнему регистру
28
+ return hashes
29
+ except requests.exceptions.RequestException as e:
30
+ messagebox.showerror("Ошибка", f"Ошибка при загрузке хешей: {e}")
31
+ return set()
32
+
33
+ def scan_file(filepath, virus_hashes):
34
+ """Сканирует файл, сравнивая его хеш с хешами в базе данных."""
35
+ file_hash = calculate_file_hash(filepath).lower() # Приводим к нижнему регистру для сравнения
36
+ if file_hash in virus_hashes:
37
+ return f"Угроза обнаружена! Файл: {filepath}\nMD5 хеш совпадает с известным вирусом."
38
+ else:
39
+ return f"Файл чист: {filepath}"
40
+
41
+ def browse_file():
42
+ """Открывает диалоговое окно выбора файла."""
43
+ filepath = filedialog.askopenfilename()
44
+ if filepath:
45
+ file_entry.delete(0, tk.END)
46
+ file_entry.insert(0, filepath)
47
+
48
+ def start_scan():
49
+ """Запускает сканирование файла."""
50
+ filepath = file_entry.get()
51
+ if filepath:
52
+ result = scan_file(filepath, virus_hashes)
53
+ result_text.config(state=tk.NORMAL)
54
+ result_text.delete(1.0, tk.END)
55
+ result_text.insert(tk.END, result)
56
+ result_text.config(state=tk.DISABLED)
57
+
58
+ if "Угроза обнаружена" in result:
59
+ messagebox.showwarning("Внимание!", result)
60
+ else:
61
+ messagebox.showinfo("Результат сканирования", result)
62
+ else:
63
+ messagebox.showerror("Ошибка", "Выберите файл для сканирования.")
64
+
65
+ # Загрузка хешей вирусов
66
+ virus_hashes = load_virus_hashes_from_url(MD5_HASHES_URL)
67
+ if not virus_hashes:
68
+ messagebox.showerror("Ошибка", "Не удалось загрузить хеши вирусов. Программа будет закрыта.")
69
+ root.quit()
70
+ exit()
71
+
72
+ # Создание главного окна
73
+ root = tk.Tk()
74
+ root.title("Мини-антивирус")
75
+
76
+ # Элементы интерфейса
77
+ file_label = tk.Label(root, text="Выберите файл для сканирования:")
78
+ file_label.grid(row=0, column=0, padx=5, pady=5, sticky="w")
79
+
80
+ file_entry = tk.Entry(root, width=50)
81
+ file_entry.grid(row=0, column=1, padx=5, pady=5)
82
+
83
+ browse_button = tk.Button(root, text="Обзор", command=browse_file)
84
+ browse_button.grid(row=0, column=2, padx=5, pady=5)
85
+
86
+ scan_button = tk.Button(root, text="Сканировать", command=start_scan)
87
+ scan_button.grid(row=1, column=0, columnspan=3, pady=10)
88
+
89
+ result_text = tk.Text(root, wrap=tk.WORD, width=60, height=10)
90
+ result_text.grid(row=2, column=0, columnspan=3, padx=5, pady=5)
91
+ result_text.config(state=tk.DISABLED)
92
+
93
+ # Запуск главного цикла
94
+ root.mainloop()