blhackbox/Makefile at main · valITino/blhackbox · GitHub

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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
.PHONY: help setup up up-full up-gateway down logs test test-local lint format clean nuke \
       pull status health portainer gateway-logs \
       claude-code \
       neo4j-browser logs-kali \
       logs-wireshark logs-screenshot \
       restart-kali \
       restart-wireshark restart-screenshot \
       push-all wordlists recon report \
       inject-verification

COMPOSE := docker compose

setup: ## Interactive setup wizard (prerequisites, .env, pull, start, health check)
	@bash setup.sh

help: ## Show this help
	@grep -E '^[a-zA-Z_-]+:.*?## .*$$' $(MAKEFILE_LIST) | \
		awk 'BEGIN {FS = ":.*?## "}; {printf "\033[36m%-26s\033[0m %s\n", $$1, $$2}'

# ── Core ───────────────────────────────────────────────────────
pull: ## Pull all pre-built images from Docker Hub
	$(COMPOSE) pull

up: ## Start core stack (4 containers)
	$(COMPOSE) up -d

down: ## Stop all services (all profiles)
	$(COMPOSE) --profile gateway --profile neo4j --profile claude-code down

logs: ## Tail logs from all services
	$(COMPOSE) logs -f

# ── Stack variations ─────────────────────────────────────────────
up-full: ## Start full stack: core + Neo4j (5 containers)
	$(COMPOSE) --profile neo4j up -d

up-gateway: ## Start core + MCP Gateway for Claude Desktop / ChatGPT (5 containers)
	$(COMPOSE) --profile gateway up -d

# ── Testing & Code Quality ─────────────────────────────────────
test: ## Run tests locally
	pytest tests/ -v --tb=short

test-local: ## Run tests locally (alias)
	pytest tests/ -v --tb=short

lint: ## Run linter
	ruff check blhackbox/ tests/

format: ## Auto-format code
	ruff format blhackbox/ tests/

clean: ## Remove containers, volumes, networks, and build artifacts (keeps images)
	$(COMPOSE) --profile gateway --profile neo4j --profile claude-code down -v --remove-orphans
	find . -type d -name __pycache__ -exec rm -rf {} + 2>/dev/null || true
	rm -rf dist/ build/ *.egg-info

nuke: ## Full cleanup: containers + volumes + ALL images (frees max disk space)
	@echo "\033[1;33m  WARNING: This will remove ALL blhackbox containers, volumes, AND images.\033[0m"
	@echo "\033[2m  You will need to 'docker compose pull' or 'docker compose build' again.\033[0m"
	@echo ""
	$(COMPOSE) --profile gateway --profile neo4j --profile claude-code down -v --remove-orphans --rmi all
	@echo ""
	@echo "\033[2m  Pruning dangling images and build cache...\033[0m"
	docker image prune -f
	docker builder prune -f
	@echo ""
	docker volume rm blhackbox_portainer_data 2>/dev/null || true
	docker volume rm blhackbox_neo4j_data 2>/dev/null || true
	docker volume rm blhackbox_neo4j_logs 2>/dev/null || true
	docker volume rm blhackbox_wordlists 2>/dev/null || true
	docker volume rm blhackbox_shared-output 2>/dev/null || true
	find . -type d -name __pycache__ -exec rm -rf {} + 2>/dev/null || true
	rm -rf dist/ build/ *.egg-info
	@echo ""
	@echo "\033[32m  Done. Run 'docker system df' to verify disk usage.\033[0m"

# ── Claude Code (Docker) ────────────────────────────────────────
claude-code: ## Build and launch Claude Code in a Docker container
	$(COMPOSE) --profile claude-code pull claude-code || $(COMPOSE) --profile claude-code build claude-code
	@echo ""
	@echo "\033[1m  Pre-flight Container Status\033[0m"
	@echo "\033[2m  ──────────────────────────────────────\033[0m"
	@$(COMPOSE) ps --format "table {{.Name}}\t{{.Status}}" 2>/dev/null || $(COMPOSE) ps
	@echo ""
	@echo "\033[2m  Waiting for all dependencies to become healthy...\033[0m"
	@echo ""
	$(COMPOSE) --profile claude-code run --rm claude-code

# ── Health & Status ──────────────────────────────────────────────
status: ## Health status of all containers
	@echo ""
	@echo "\033[1m  blhackbox Container Status\033[0m"
	@echo "\033[2m  ──────────────────────────────────────\033[0m"
	@$(COMPOSE) --profile gateway --profile neo4j --profile claude-code ps --format "table {{.Name}}\t{{.Status}}\t{{.Ports}}" 2>/dev/null || $(COMPOSE) ps
	@echo ""

health: ## Quick health check of all MCP servers
	@echo ""
	@echo "\033[1m  MCP Server Health Check\033[0m"
	@echo "\033[2m  ──────────────────────────────────────\033[0m"
	@printf "  %-22s " "Kali MCP (9001)"; \
		docker exec blhackbox-kali-mcp python3 -c "import urllib.request; urllib.request.urlopen('http://localhost:9001/sse')" > /dev/null 2>&1 \
		&& echo "\033[32m[OK]\033[0m" || echo "\033[31m[FAIL]\033[0m"
	@printf "  %-22s " "WireMCP (9003)"; \
		docker exec blhackbox-wire-mcp python3 -c "import urllib.request; urllib.request.urlopen('http://localhost:9003/sse')" > /dev/null 2>&1 \
		&& echo "\033[32m[OK]\033[0m" || echo "\033[31m[FAIL]\033[0m"
	@printf "  %-22s " "Screenshot MCP (9004)"; \
		docker exec blhackbox-screenshot-mcp python3 -c "import urllib.request; urllib.request.urlopen('http://localhost:9004/health')" > /dev/null 2>&1 \
		&& echo "\033[32m[OK]\033[0m" || echo "\033[31m[FAIL]\033[0m"
	@printf "  %-22s " "MCP Gateway (8080)"; \
		docker inspect --format='{{.State.Running}}' blhackbox-mcp-gateway 2>/dev/null | grep -q "true" \
		&& echo "\033[32m[OK]\033[0m" || echo "\033[33m[OFF]\033[0m  (optional — enable with: make up-gateway)"
	@printf "  %-22s " "Portainer (9443)"; \
		curl -skf --max-time 3 https://localhost:9443 > /dev/null 2>&1 \
		&& echo "\033[32m[OK]\033[0m  https://localhost:9443" || echo "\033[31m[FAIL]\033[0m"
	@echo ""

# ── Monitoring ──────────────────────────────────────────────────
portainer: ## Open Portainer dashboard (first run: create admin account)
	@echo ""
	@echo "\033[1m  Portainer Web UI\033[0m"
	@echo "\033[2m  ──────────────────────────────────────\033[0m"
	@echo "  URL:  \033[36mhttps://localhost:9443\033[0m"
	@echo ""
	@echo "  First run: Create an admin account within 5 minutes of startup."
	@echo "  If the window expired, restart: docker compose restart portainer"
	@echo ""
	@open https://localhost:9443 2>/dev/null || xdg-open https://localhost:9443 2>/dev/null || echo "  Open the URL above in your browser."

gateway-logs: ## Live MCP tool call log (requires --profile gateway)
	$(COMPOSE) logs -f mcp-gateway

logs-kali: ## Tail Kali MCP server logs
	$(COMPOSE) logs -f kali-mcp

logs-wireshark: ## Tail WireMCP server logs
	$(COMPOSE) logs -f wire-mcp

logs-screenshot: ## Tail Screenshot MCP server logs
	$(COMPOSE) logs -f screenshot-mcp

neo4j-browser: ## Open Neo4j Browser
	@open http://localhost:7474 2>/dev/null || xdg-open http://localhost:7474

# ── Per-service restart ──────────────────────────────────────────
restart-kali: ## Restart Kali MCP server
	$(COMPOSE) restart kali-mcp

restart-wireshark: ## Restart WireMCP server
	$(COMPOSE) restart wire-mcp

restart-screenshot: ## Restart Screenshot MCP server
	$(COMPOSE) restart screenshot-mcp

# ── Recon & Reporting ──────────────────────────────────────────
wordlists: ## Download common wordlists
	mkdir -p wordlists
	@echo "Downloading common wordlists..."
	curl -sL -o wordlists/common.txt \
		https://raw.githubusercontent.com/danielmiessler/SecLists/master/Discovery/Web-Content/common.txt
	curl -sL -o wordlists/subdomains-top1million-5000.txt \
		https://raw.githubusercontent.com/danielmiessler/SecLists/master/Discovery/DNS/subdomains-top1million-5000.txt
	curl -sL -o wordlists/raft-medium-directories.txt \
		https://raw.githubusercontent.com/danielmiessler/SecLists/master/Discovery/Web-Content/raft-medium-directories.txt
	@echo "Done. Wordlists saved to ./wordlists/"

recon: ## Quick recon example (requires TARGET env var)
	blhackbox recon --target $(TARGET)

report: ## Generate report for a session (requires SESSION env var)
	blhackbox report --session $(SESSION) --format pdf

# ── Verification ─────────────────────────────────────────────────
inject-verification: ## Render verification.env into active authorization document
	python -m blhackbox.prompts.inject_verification

# ── Build and push (Docker Hub: crhacky/blhackbox) ──────────────
push-all: ## Build and push all custom images to Docker Hub
	docker build -f docker/kali-mcp.Dockerfile -t crhacky/blhackbox:kali-mcp .
	docker build -f docker/wire-mcp.Dockerfile -t crhacky/blhackbox:wire-mcp .
	docker build -f docker/screenshot-mcp.Dockerfile -t crhacky/blhackbox:screenshot-mcp .
	docker build -f docker/claude-code.Dockerfile -t crhacky/blhackbox:claude-code .
	docker push crhacky/blhackbox:kali-mcp
	docker push crhacky/blhackbox:wire-mcp
	docker push crhacky/blhackbox:screenshot-mcp
	docker push crhacky/blhackbox:claude-code