add /solver endpoint for PoW solver script download

This commit is contained in:
Username
2025-12-20 20:32:39 +01:00
parent 4f0b33fd7b
commit 3fe3f6f160

View File

@@ -469,6 +469,7 @@ class IndexView(MethodView):
f"GET {prefixed_url('/')}": "API information",
f"GET {prefixed_url('/health')}": "Health check",
f"GET {prefixed_url('/client')}": "Download CLI client",
f"GET {prefixed_url('/solver')}": "Download PoW solver script",
f"GET {prefixed_url('/challenge')}": "Get PoW challenge",
f"POST {prefixed_url('/')}": "Create paste",
f"GET {prefixed_url('/pastes')}": "List your pastes (auth required)",
@@ -756,6 +757,91 @@ class ClientView(MethodView):
return error_response("Client not available", 404)
class SolverView(MethodView):
"""PoW solver script download endpoint."""
def get(self) -> Response:
"""Serve a Python script for solving PoW challenges."""
server_url = base_url()
script = f'''#!/usr/bin/env python3
"""FlaskPaste PoW solver - solves challenges for curl-based uploads."""
import hashlib
import json
import sys
import urllib.request
SERVER = "{server_url}"
def get_challenge():
"""Fetch a new challenge from the server."""
with urllib.request.urlopen(f"{{SERVER}}/challenge") as r:
return json.load(r)
def solve(nonce: str, difficulty: int) -> int:
"""Find solution where sha256(nonce:solution) has sufficient leading zero bits."""
i = 0
while True:
h = hashlib.sha256(f"{{nonce}}:{{i}}".encode()).digest()
zero_bits = 0
for byte in h:
if byte == 0:
zero_bits += 8
else:
zero_bits += 8 - byte.bit_length()
break
if zero_bits >= difficulty:
return i
i += 1
def main():
if len(sys.argv) < 2:
print("Usage: fpaste-solver <file>", file=sys.stderr)
print(" cat data | fpaste-solver -", file=sys.stderr)
sys.exit(1)
# Read content
if sys.argv[1] == "-":
content = sys.stdin.buffer.read()
else:
with open(sys.argv[1], "rb") as f:
content = f.read()
# Get and solve challenge
ch = get_challenge()
if not ch.get("enabled", True):
token, solution = "", ""
else:
print(f"Solving PoW (difficulty={{ch['difficulty']}})...", file=sys.stderr)
solution = solve(ch["nonce"], ch["difficulty"])
token = ch["token"]
print(f"Solved: {{solution}}", file=sys.stderr)
# Upload
req = urllib.request.Request(
f"{{SERVER}}/",
data=content,
headers={{
"Content-Type": "application/octet-stream",
"X-PoW-Token": token,
"X-PoW-Solution": str(solution),
}},
)
with urllib.request.urlopen(req) as r:
result = json.load(r)
print(result.get("raw", result.get("url", json.dumps(result))))
if __name__ == "__main__":
main()
'''
response = Response(script, mimetype="text/x-python")
response.headers["Content-Disposition"] = "attachment; filename=fpaste-solver"
return response
class PasteView(MethodView):
"""Paste metadata operations."""
@@ -1399,6 +1485,7 @@ bp.add_url_rule("/", view_func=IndexView.as_view("index"))
bp.add_url_rule("/health", view_func=HealthView.as_view("health"))
bp.add_url_rule("/challenge", view_func=ChallengeView.as_view("challenge"))
bp.add_url_rule("/client", view_func=ClientView.as_view("client"))
bp.add_url_rule("/solver", view_func=SolverView.as_view("solver"))
# Paste operations
bp.add_url_rule("/pastes", view_func=PastesListView.as_view("pastes_list"))