75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
# This scoreboard is a community contribution by Robin Jadoul. Thank you!
|
||
|
|
|
||
|
|
import collections, pickle
|
||
|
|
import rich.console, rich.table
|
||
|
|
from pwn import *
|
||
|
|
from web3 import Web3, HTTPProvider
|
||
|
|
from web3.middleware import geth_poa_middleware
|
||
|
|
import eth_account, eth_abi
|
||
|
|
import pprint, time, json, shutil, os
|
||
|
|
import requests
|
||
|
|
|
||
|
|
web3 = Web3(HTTPProvider("https://polygon-rpc.com/"))
|
||
|
|
web3.middleware_onion.inject(geth_poa_middleware, layer=0)
|
||
|
|
|
||
|
|
GAME_CONTRACT = "0x36A1424DA63a50627863d8f65C0669da7347814A"
|
||
|
|
GAME = web3.eth.contract(GAME_CONTRACT, abi=json.load(open("game_abi.json")))
|
||
|
|
|
||
|
|
known_addresses = {
|
||
|
|
#"address": "username", # Fill it if you want to see the names
|
||
|
|
}
|
||
|
|
|
||
|
|
#############################################################
|
||
|
|
|
||
|
|
BLOCK_START = 32265959
|
||
|
|
BLOCK_END = web3.eth.block_number
|
||
|
|
BLOCK_RANGE = 10000
|
||
|
|
if os.path.exists("scoreboard.pkl"):
|
||
|
|
with open("scoreboard.pkl", "rb") as f:
|
||
|
|
seenEvents = pickle.load(f)
|
||
|
|
else:
|
||
|
|
seenEvents = {}
|
||
|
|
|
||
|
|
def iterEvents(ev):
|
||
|
|
name = ev.event_name
|
||
|
|
known, start = seenEvents.get(name, ([], BLOCK_START))
|
||
|
|
|
||
|
|
yield from known
|
||
|
|
for i in range(start, BLOCK_END + 1, BLOCK_RANGE):
|
||
|
|
res = ev.getLogs(fromBlock=hex(i), toBlock=hex(i + BLOCK_RANGE))
|
||
|
|
known += list(res)
|
||
|
|
yield from res
|
||
|
|
|
||
|
|
seenEvents[name] = (known, BLOCK_END)
|
||
|
|
with open("scoreboard.pkl", "wb") as f:
|
||
|
|
pickle.dump(seenEvents, f)
|
||
|
|
|
||
|
|
def main():
|
||
|
|
con = rich.console.Console()
|
||
|
|
|
||
|
|
allchals = []
|
||
|
|
solves = collections.defaultdict(lambda: collections.defaultdict(lambda: False))
|
||
|
|
|
||
|
|
for chal in iterEvents(GAME.events.FlagAdded):
|
||
|
|
chal = (chal["args"]["flagId"], GAME.caller.flags(chal["args"]["flagId"])[3])
|
||
|
|
allchals.append(chal)
|
||
|
|
|
||
|
|
for solve in iterEvents(GAME.events.FlagSolved):
|
||
|
|
solver, chal = solve["args"]["solver"], solve["args"]["flagId"]
|
||
|
|
solves[solver][chal] = True
|
||
|
|
|
||
|
|
board = [(solver if solver not in known_addresses else f"[bold cyan]{known_addresses[solver]}", str(GAME.caller.getPlayerPoints(solver))) + tuple(["[bold red]:x:", "[bold green]:heavy_check_mark:"][solved[c[0]]] for c in allchals) for solver, solved in solves.items()]
|
||
|
|
|
||
|
|
allchals.sort()
|
||
|
|
board.sort(key = lambda e: (int(e[1]), e[0]), reverse=True)
|
||
|
|
|
||
|
|
tab = rich.table.Table("Rank", "User", "Score", *[c[1] for c in allchals], title="Scoreboard")
|
||
|
|
for i, b in enumerate(board, 1):
|
||
|
|
tab.add_row(str(i), *b)
|
||
|
|
con.print(tab)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|