90 lines
2.8 KiB
Python
Executable File
90 lines
2.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse, sys, platform, os
|
|
|
|
pref = "\033["
|
|
reset = f"{pref}0m"
|
|
|
|
class colors:
|
|
black = "30m"
|
|
red = "31m"
|
|
green = "32m"
|
|
yellow = "33m"
|
|
blue = "34m"
|
|
magenta = "35m"
|
|
cyan = "36m"
|
|
white = "37m"
|
|
|
|
def puts(text, *args, color=colors.white, is_bold=False, **kwargs):
|
|
print(f'{pref}{1 if is_bold else 0};{color}' + text + reset, *args, **kwargs)
|
|
|
|
def sep(): puts("-----------------------------------", is_bold=True)
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--port', "-p", type=int, required=False, help='Port where open the web service of the firewall', default=4444)
|
|
parser.add_argument('--threads', "-t", type=int, required=False, help='Number of threads started for each service/utility', default=1)
|
|
parser.add_argument('--no-autostart', "-n", required=False, action="store_true", help='Auto-execute "docker-compose up -d --build"', default=False)
|
|
parser.add_argument('--build', "-b", required=False, action="store_true", help='Build the container locally', default=False)
|
|
|
|
args = parser.parse_args()
|
|
sep()
|
|
puts(f"Firegex", color=colors.yellow, end="")
|
|
puts(" will start on port ", end="")
|
|
puts(f"{args.port}", color=colors.cyan)
|
|
|
|
if args.threads < 1:
|
|
puts("Insert a valid number of threads", color=colors.red)
|
|
exit()
|
|
|
|
os.chdir(os.path.dirname(os.path.realpath(__file__)))
|
|
|
|
with open("docker-compose.yml","wt") as compose:
|
|
|
|
if "linux" in sys.platform and not 'microsoft-standard' in platform.uname().release: #Check if not is a wsl also
|
|
compose.write(f"""
|
|
version: '3.9'
|
|
|
|
services:
|
|
firewall:
|
|
restart: unless-stopped
|
|
{"build: ." if args.build else "image: ghcr.io/pwnzer0tt1/firegex:latest"}
|
|
network_mode: "host"
|
|
environment:
|
|
- PORT={args.port}
|
|
- NTHREADS={args.threads}
|
|
volumes:
|
|
- /execute/db
|
|
cap_add:
|
|
- NET_ADMIN
|
|
""")
|
|
|
|
else:
|
|
sep()
|
|
puts("--- WARNING ---", color=colors.yellow)
|
|
puts("You are not in a linux machine, due to docker limitation on other platform, the firewall will not work in this machine. You will only see the interface of firegex.", color=colors.red)
|
|
compose.write(f"""
|
|
version: '3.9'
|
|
|
|
services:
|
|
firewall:
|
|
restart: unless-stopped
|
|
{"build: ." if args.build else "image: ghcr.io/pwnzer0tt1/firegex:latest"}
|
|
ports:
|
|
- {args.port}:{args.port}
|
|
environment:
|
|
- PORT={args.port}
|
|
- NTHREADS={args.threads}
|
|
volumes:
|
|
- /execute/db
|
|
cap_add:
|
|
- NET_ADMIN
|
|
""")
|
|
|
|
sep()
|
|
if not args.no_autostart:
|
|
puts("Running 'docker-compose up -d --build'\n", color=colors.green)
|
|
os.system("docker-compose up -d --build")
|
|
else:
|
|
puts("Done! You can start firegex with docker-compose up -d --build", color=colors.yellow)
|
|
sep()
|
|
|