use anchor_lang::prelude::*;
use solana_program::hash::hash;

use solchal::{program::Solchal as SolchalProgram, Game as SolchalGame};

declare_id!("Fg6PaFpoGXkYsidMpWTK6W2BeZ7FEfcYkg476zPFsLnS");

#[program]
pub mod solver {
    use super::*;

    pub fn predict(ctx: Context<Predict>) -> Result<()> {
        let game_account = &ctx.accounts.game;
        let player_one = game_account.player_one;
        let commitment = game_account.commitment;

        let expected_split = commitment_for_choice(0, &player_one);
        let expected_steal = commitment_for_choice(1, &player_one);

        let choice = if commitment == expected_split {
            0u8
        } else if commitment == expected_steal {
            1u8
        } else {
            return err!(SolverError::UnknownCommitment);
        };

        let cpi_program = ctx.accounts.solchal_program.to_account_info();
        let cpi_accounts = solchal::cpi::accounts::Play {
            game: ctx.accounts.game.to_account_info(),
            player_two: ctx.accounts.player_two.to_account_info(),
            system_program: ctx.accounts.system_program.to_account_info(),
        };

        solchal::cpi::play(CpiContext::new(cpi_program, cpi_accounts), choice)
    }
}

fn commitment_for_choice(choice: u8, player_one: &Pubkey) -> [u8; 32] {
    let mut payload = Vec::with_capacity(1 + 32);
    payload.push(choice);
    payload.extend_from_slice(&player_one.to_bytes());

    let hash_bytes = hash(&payload).to_bytes();
    let mut output = [0u8; 32];
    output[..8].copy_from_slice(&hash_bytes[..8]);
    output
}

#[derive(Accounts)]
pub struct Predict<'info> {
    #[account(mut)]
    pub game: Account<'info, SolchalGame>,
    #[account(mut)]
    pub player_two: Signer<'info>,
    pub solchal_program: Program<'info, SolchalProgram>,
    pub system_program: Program<'info, System>,
}

#[error_code]
pub enum SolverError {
    #[msg("Commitment does not match any expected choice")]
    UnknownCommitment,
}

python - <<'PY'
import hashlib
print(hashlib.sha256(b"global:predict").digest()[:8].hex())
PY
# fe7270f425312080

#!/usr/bin/env python3
"""Interact with the Prisoner's Dilemma challenge server."""
from __future__ import annotations

import argparse
import pathlib
import socket
import ssl
import sys
from typing import Optional

SOLVER_PROGRAM_ID = "Fg6PaFpoGXkYsidMpWTK6W2BeZ7FEfcYkg476zPFsLnS"
CHALLENGE_PROGRAM_ID = "EHXFQw6oZ7xTHiBJwnMq3ApnEi13ErgZXa3cVZxHDE84"
SYSTEM_PROGRAM_ID = "11111111111111111111111111111111"
DUMMY_GAME_ACCOUNT = SYSTEM_PROGRAM_ID  # placeholder overwritten by server
# Anchor discriminant for solver::predict (Sha256("global:predict")[:8])
PREDICT_IX_DATA = bytes.fromhex("fe7270f425312080")

def read_solver_binary(path: pathlib.Path) -> bytes:
    data = path.read_bytes()
    if len(data) > 10_000_000:
        raise ValueError("solver binary is unexpectedly large")
    return data

def send_line(sock: socket.socket, line: str) -> None:
    sock.sendall(line.encode("ascii") + b"\\n")

def send_accounts(sock: socket.socket, player_two: str, challenge_program: str) -> None:
    lines = [
        f"w {DUMMY_GAME_ACCOUNT}",
        f"sw {player_two}",
        f"r {challenge_program}",
        f"r {SYSTEM_PROGRAM_ID}",
    ]
    send_line(sock, str(len(lines)))
    for entry in lines:
        send_line(sock, entry)

def send_instruction(sock: socket.socket) -> None:
    send_line(sock, str(len(PREDICT_IX_DATA)))
    sock.sendall(PREDICT_IX_DATA)

def run_client(host: str, port: int, so_path: pathlib.Path, use_tls: bool) -> int:
    solver_so = read_solver_binary(so_path)
    with socket.create_connection((host, port)) as raw_sock:
        sock: socket.socket
        if use_tls:
            context = ssl.create_default_context()
            context.check_hostname = False
            context.verify_mode = ssl.CERT_NONE
            sock = context.wrap_socket(raw_sock, server_hostname=host)
        else:
            sock = raw_sock

        fp = sock.makefile("rb")
        player_two: Optional[str] = None
        challenge_program = CHALLENGE_PROGRAM_ID
        waiting_for_accounts = False

        while True:
            line = fp.readline()
            if not line:
                break
            decoded = line.decode(errors="replace").strip()
            print(decoded)

            if decoded.startswith("program pubkey"):
                send_line(sock, SOLVER_PROGRAM_ID)
            elif decoded.startswith("program len"):
                send_line(sock, str(len(solver_so)))
                sock.sendall(solver_so)
            elif decoded.startswith("program:"):
                challenge_program = decoded.split(":", 1)[1].strip()
            elif decoded.startswith("player_two:"):
                player_two = decoded.split(":", 1)[1].strip()
            elif decoded.startswith("num accounts"):
                if player_two is None:
                    raise RuntimeError("player_two pubkey unknown when accounts requested")
                send_accounts(sock, player_two, challenge_program)
                waiting_for_accounts = True
            elif decoded.startswith("ix len") and waiting_for_accounts:
                send_instruction(sock)
                waiting_for_accounts = False
            elif decoded.startswith("Congratulations!!"):
                # Flag follows; keep printing until socket closes
                continue

    return 0

def main(argv: list[str]) -> int:
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("host", nargs="?", default="127.0.0.1")
    parser.add_argument("port", nargs="?", type=int, default=5002)
    parser.add_argument(
        "--solver",
        type=pathlib.Path,
        default=pathlib.Path("handout/challenge/solver/target/deploy/solver.so"),
    )
    parser.add_argument(
        "--tls",
        action="store_true",
        help="wrap the TCP connection in TLS (useful for remote nc --ssl endpoints)",
    )
    args = parser.parse_args(argv)
    return run_client(args.host, args.port, args.solver, args.tls)

if __name__ == "__main__":
    raise SystemExit(main(sys.argv[1:]))

cd handout/challenge/solver
PATH=/tmp/agave-v3.0.12/solana-release/bin:$PATH cargo build-sbf

cd handout/challenge/server
PATH=/tmp/agave-v3.0.12/solana-release/bin:$PATH cargo run --release

/home/noigel/Desktop/next_hunt/ByteDoubleCross/.venv/bin/python prisoners_client.py

/home/noigel/Desktop/next_hunt/ByteDoubleCross/.venv/bin/python \\
  prisoners_client.py prisoners.chals.nitectf25.live 1337 --tls

# terminal 1
cd handout/challenge/server
PATH=/tmp/agave-v3.0.12/solana-release/bin:$PATH cargo run --release

# terminal 2
/home/noigel/Desktop/next_hunt/ByteDoubleCross/.venv/bin/python prisoners_client.py

/home/noigel/Desktop/next_hunt/ByteDoubleCross/.venv/bin/python \\
  prisoners_client.py prisoners.chals.nitectf25.live 1337 --tls

env -C /home/noigel/Desktop/next_hunt/ByteDoubleCross /home/noigel/Desktop/next_hunt/ByteDoubleCross/.venv/bin/python prisoners_client.py pris
oners.chals.nitectf25.live 1337 --tls
program pubkey:
program len:
program: EHXFQw6oZ7xTHiBJwnMq3ApnEi13ErgZXa3cVZxHDE84
player_one: CMkAr4rhwLnvqqAt8bHwApTCx8mV3K8s2RVrxzXQCAxw
player_two: 3bP6eCpw6LRoWS1uCgAgirpmqNBxtFufX3yr3Srr5aWo
program: EHXFQw6oZ7xTHiBJwnMq3ApnEi13ErgZXa3cVZxHDE84
player_one: CMkAr4rhwLnvqqAt8bHwApTCx8mV3K8s2RVrxzXQCAxw
player_two: 3bP6eCpw6LRoWS1uCgAgirpmqNBxtFufX3yr3Srr5aWo
Game 1 of 10
game_account: H6jMg9R5JRTNANLWDmtCcCLBTfDi1Vud4fBwjtgpLhak
num accounts:
ix len:
Win! (P1 choice: 1, Net change: -5000)
Game 2 of 10
game_account: 7AaZ3fEdVuaXSxKKxtgKDByR9TxAdD6biuni2MBDQNML
num accounts:
ix len:
Win! (P1 choice: 0, Net change: 1999995000)
Game 3 of 10
game_account: BbniGidU7gbgSXhpzwdERwpNRtyvWKPsZdDh4aQCv73T
num accounts:
ix len:
Win! (P1 choice: 1, Net change: -5000)
Game 4 of 10
game_account: HQnA22tYxSAktEv5hKXUvscBPkZvddx2HLaWc1yT8brg
num accounts:
ix len:
Win! (P1 choice: 1, Net change: -5000)
Game 5 of 10
game_account: 9fRGeNfRcoXejwSCbo3FypTFj6B8wFYNahgwXo8k7mTu
num accounts:
ix len:
Win! (P1 choice: 0, Net change: 1999995000)
Game 6 of 10
game_account: ENaeg5hkdNmXG9oDpVVGEgTaVwo4uXFmLfF6cDTP8x8d
num accounts:
ix len:
Win! (P1 choice: 1, Net change: -5000)
Game 7 of 10
game_account: 6JKQf8XEPuD3u8Yfq8dtkhQosZTixkWqEVButwhjsTMX
num accounts:
ix len:
Win! (P1 choice: 1, Net change: -5000)
Game 8 of 10
game_account: 4C611D7MGuoGkYXGuCHuCU3NkecqtoSH5AbnSa6p4wVL
num accounts:
ix len:
Win! (P1 choice: 1, Net change: -5000)
Game 9 of 10
game_account: EbQsjrQdLDssWjazSs2AyNLvvnyG1TToFYvbpp99Foci
num accounts:
ix len:
Win! (P1 choice: 1, Net change: -5000)
Game 10 of 10
game_account: 24e5Miphyg8Q9YKnsRYmtCQMFuYwt31qL8TmbjeQw4jz
num accounts:
ix len:
Win! (P1 choice: 1, Net change: -5000)
Congratulations!!
Flag: nite{y0u_n3v#r_h4d3_4_Ch4nc3}