From 570531fe32c8e6c2947bdfe963885e4be7b7f32d Mon Sep 17 00:00:00 2001 From: Zoe <62722391+juls0730@users.noreply.github.com> Date: Fri, 21 Nov 2025 16:20:07 +0000 Subject: [PATCH] Implement kCTF strategy This implementation is pretty scuffed, but its more exploratory than anything else. --- example-app/app/pages/index.vue | 424 ++----------------- example-app/app/pages/index.vue.old | 411 ++++++++++++++++++ example-app/app/pages/widget.vue | 38 -- example-app/config.toml | 5 +- example-app/package-lock.json | 7 +- example-app/package.json | 4 +- example-app/server/api/pow/challenge.get.ts | 33 +- example-app/server/api/pow/challenge.post.ts | 27 +- example-app/server/utils/config.ts | 33 +- justfile | 10 +- packages/lib/package-lock.json | 3 - packages/lib/src/index.ts | 40 +- packages/lib/src/solver.ts | 76 ++-- packages/lib/src/validator.ts | 208 +++------ packages/widget/package-lock.json | 4 - packages/widget/src/pow-captcha.ts | 172 ++++---- packages/widget/src/solver-worker.ts | 34 +- packages/widget/src/types/worker.ts | 45 +- solver/build.zig | 28 +- solver/src/kctf.zig | 208 +++++++++ solver/src/solver.zig | 181 +++----- solver/src/validator.zig | 106 +++-- 22 files changed, 1090 insertions(+), 1007 deletions(-) create mode 100644 example-app/app/pages/index.vue.old delete mode 100644 example-app/app/pages/widget.vue diff --git a/example-app/app/pages/index.vue b/example-app/app/pages/index.vue index e1609f6..b4072ee 100644 --- a/example-app/app/pages/index.vue +++ b/example-app/app/pages/index.vue @@ -1,411 +1,37 @@ - - \ No newline at end of file + \ No newline at end of file diff --git a/example-app/app/pages/index.vue.old b/example-app/app/pages/index.vue.old new file mode 100644 index 0000000..e1609f6 --- /dev/null +++ b/example-app/app/pages/index.vue.old @@ -0,0 +1,411 @@ + + + + + \ No newline at end of file diff --git a/example-app/app/pages/widget.vue b/example-app/app/pages/widget.vue deleted file mode 100644 index af5a7e7..0000000 --- a/example-app/app/pages/widget.vue +++ /dev/null @@ -1,38 +0,0 @@ - - - \ No newline at end of file diff --git a/example-app/config.toml b/example-app/config.toml index 175d496..e3e4da3 100644 --- a/example-app/config.toml +++ b/example-app/config.toml @@ -1,7 +1,10 @@ -strategy = "target_number" +strategy = "kctf" [leading_zeroes] difficulty = 4 [target_number] max_number = 10000 + +[kctf] +difficulty = 100 diff --git a/example-app/package-lock.json b/example-app/package-lock.json index 133031d..c7e8194 100644 --- a/example-app/package-lock.json +++ b/example-app/package-lock.json @@ -6,8 +6,8 @@ "": { "name": "hello-nuxt", "dependencies": { - "@impost/lib": "file:../packages/lib", - "@impost/widget": "file:../packages/widget", + "@impost/lib": "^0.1.0", + "@impost/widget": "^0.1.0", "@lit/reactive-element": "^2.1.1", "js-toml": "^1.0.2", "lit-element": "^4.2.1", @@ -41,6 +41,9 @@ "name": "@impost/lib", "version": "0.1.0", "license": "BSL-1.0", + "dependencies": { + "uuidv7": "^1.0.2" + }, "devDependencies": { "oxc-minify": "^0.97.0", "tslib": "^2.6.2", diff --git a/example-app/package.json b/example-app/package.json index 2d301c0..89ed353 100644 --- a/example-app/package.json +++ b/example-app/package.json @@ -16,8 +16,8 @@ "nuxt": "latest", "nuxt-ssr-lit": "1.6.32", "zod": "^4.1.12", - "@impost/lib": "file:../packages/lib", - "@impost/widget": "file:../packages/widget" + "@impost/lib": "^0.1.0", + "@impost/widget": "^0.1.0" }, "devDependencies": { "@types/node": "^24.10.0", diff --git a/example-app/server/api/pow/challenge.get.ts b/example-app/server/api/pow/challenge.get.ts index 607cc42..cb0f522 100644 --- a/example-app/server/api/pow/challenge.get.ts +++ b/example-app/server/api/pow/challenge.get.ts @@ -6,19 +6,28 @@ import { CHALLENGE_TIMEOUT_MS, outstandingChallenges } from '~~/server/utils/pow export default defineEventHandler(async () => { let challenge_config; + // switch (config.strategy) { + // case ChallengeStrategy.LeadingZeroes: + // challenge_config = { + // parameters: { expires_at: CHALLENGE_TIMEOUT_MS }, + // strategy: config.strategy, + // difficulty: config.leading_zeroes?.difficulty!, + // }; + // break; + // case ChallengeStrategy.TargetNumber: + // challenge_config = { + // parameters: { expires_at: CHALLENGE_TIMEOUT_MS }, + // strategy: config.strategy, + // max_number: config.target_number.max_number, + // }; + // break; + // } switch (config.strategy) { - case ChallengeStrategy.LeadingZeroes: + case ChallengeStrategy.kCTF: challenge_config = { parameters: { expires_at: CHALLENGE_TIMEOUT_MS }, strategy: config.strategy, - difficulty: config.leading_zeroes?.difficulty!, - }; - break; - case ChallengeStrategy.TargetNumber: - challenge_config = { - parameters: { expires_at: CHALLENGE_TIMEOUT_MS }, - strategy: config.strategy, - max_number: config.target_number.max_number, + difficulty: config.kctf.difficulty, }; break; } @@ -31,10 +40,10 @@ export default defineEventHandler(async () => { }); } - outstandingChallenges.set(challenge.salt, { + outstandingChallenges.set(challenge.challenge, { challenge, timeout: setTimeout(() => { - console.log("Challenge timed out:", challenge.salt); - outstandingChallenges.delete(challenge.salt); + console.log("Challenge timed out:", challenge.challenge); + outstandingChallenges.delete(challenge.challenge); }, CHALLENGE_TIMEOUT_MS) }); diff --git a/example-app/server/api/pow/challenge.post.ts b/example-app/server/api/pow/challenge.post.ts index e140f92..58d39ae 100644 --- a/example-app/server/api/pow/challenge.post.ts +++ b/example-app/server/api/pow/challenge.post.ts @@ -4,12 +4,13 @@ import * as z from 'zod'; import { outstandingChallenges } from '~~/server/utils/pow'; const challengeSchema = z.object({ - challenge: z.string(), - nonce: z.string() + challenge: z.string().startsWith("s."), + solution: z.string().startsWith("s.") }) // post handler that takes in the challenge, and the nonce export default defineEventHandler(async (event) => { + console.log(await readBody(event)); const body = await readValidatedBody(event, challengeSchema.safeParse); if (!body.success) { @@ -19,19 +20,25 @@ export default defineEventHandler(async (event) => { }) } - let target = body.data.challenge; - let nonce = body.data.nonce; + let { challenge, solution } = body.data; + + const outstanding_challenge = outstandingChallenges.get(challenge); + if (outstanding_challenge === undefined) { + throw createError({ + statusCode: 400, + statusMessage: 'Challenge not found' + }) + } // check if the challenge is valid - let challenge_valid = await validate_challenge(outstandingChallenges.get(target)!.challenge, { - challenge: target, - nonce: nonce - }); + const challenge_valid = await validate_challenge(outstanding_challenge.challenge, solution); + + console.log("CHALLENGE VALID", challenge_valid); if (challenge_valid) { // clear the challenge - clearTimeout(outstandingChallenges.get(target)!.timeout); - outstandingChallenges.delete(target); + clearTimeout(outstandingChallenges.get(challenge)!.timeout); + outstandingChallenges.delete(challenge); return { message: 'Challenge solved' diff --git a/example-app/server/utils/config.ts b/example-app/server/utils/config.ts index 8bab744..e2ad38a 100644 --- a/example-app/server/utils/config.ts +++ b/example-app/server/utils/config.ts @@ -3,26 +3,31 @@ import { load } from 'js-toml'; import z from 'zod'; import { ChallengeStrategy } from "@impost/lib"; -const LeadingZeroesSchema = z.object({ - strategy: z.literal(ChallengeStrategy.LeadingZeroes), - leading_zeroes: z.object({ - difficulty: z.number().int().min(1).max(64), +// const LeadingZeroesSchema = z.object({ +// strategy: z.literal(ChallengeStrategy.LeadingZeroes), +// leading_zeroes: z.object({ +// difficulty: z.number().int().min(1).max(64), +// }), +// }); + +// const TargetNumberSchema = z.object({ +// strategy: z.literal(ChallengeStrategy.TargetNumber), +// target_number: z.object({ +// max_number: z.number().int().min(1).max(100_000), +// }), +// }); + +const kCTFSchema = z.object({ + strategy: z.literal(ChallengeStrategy.kCTF), + kctf: z.object({ + difficulty: z.number().int().min(1), }), }); -const TargetNumberSchema = z.object({ - strategy: z.literal(ChallengeStrategy.TargetNumber), - target_number: z.object({ - max_number: z.number().int().min(1).max(100_000), - }), -}); - - export type Config = z.infer; export const Config = z.discriminatedUnion('strategy', [ - LeadingZeroesSchema, - TargetNumberSchema, + kCTFSchema, ]); export let config: Config; diff --git a/justfile b/justfile index 3d10f4c..c47db72 100644 --- a/justfile +++ b/justfile @@ -1,10 +1,14 @@ -build: build-widget - wasm-opt-args := "--strip-debug --strip-dwarf --enable-tail-call --enable-bulk-memory -Oz" -zig-build-args := "--release=fast" +zig-build-args := "--release=fast -Dtarget=wasm32-freestanding -Dcpu=generic+bulk_memory+bulk_memory_opt+simd128+tail_call" npm-runner := "npm" +[working-directory: "example-app"] +playground: build + {{npm-runner}} run dev + +build: build-widget + [working-directory: "solver"] build-wasm: zig build {{zig-build-args}} diff --git a/packages/lib/package-lock.json b/packages/lib/package-lock.json index d643549..e66168c 100644 --- a/packages/lib/package-lock.json +++ b/packages/lib/package-lock.json @@ -2058,7 +2058,6 @@ "integrity": "sha512-5gTmgEY/sqK6gFXLIsQNH19lWb4ebPDLA4SdLP7dsWkIXHWlG66oPuVvXSGFPppYZz8ZDZq0dYYrbHfBCVUb1Q==", "dev": true, "license": "MIT", - "peer": true, "engines": { "node": ">=12" }, @@ -2332,7 +2331,6 @@ "integrity": "sha512-jl1vZzPDinLr9eUt3J/t7V6FgNEw9QjvBPdysz9KfQDD41fQrC2Y4vKQdiaUpFT4bXlb1RHhLpp8wtm6M5TgSw==", "dev": true, "license": "Apache-2.0", - "peer": true, "bin": { "tsc": "bin/tsc", "tsserver": "bin/tsserver" @@ -2383,7 +2381,6 @@ "integrity": "sha512-BxAKBWmIbrDgrokdGZH1IgkIk/5mMHDreLDmCJ0qpyJaAteP8NvMhkwr/ZCQNqNH97bw/dANTE9PDzqwJghfMQ==", "dev": true, "license": "MIT", - "peer": true, "dependencies": { "esbuild": "^0.25.0", "fdir": "^6.5.0", diff --git a/packages/lib/src/index.ts b/packages/lib/src/index.ts index 01081a7..9cfa23b 100644 --- a/packages/lib/src/index.ts +++ b/packages/lib/src/index.ts @@ -1,28 +1,44 @@ +import { UUID } from "uuidv7"; + export enum ChallengeAlgorithm { Argon2id = "argon2id", } export enum ChallengeStrategy { + kCTF = "kctf", LeadingZeroes = "leading_zeroes", TargetNumber = "target_number", } // In this case, the client will repeatedly hash a number with has until it // finds a hash thaat starts with *difficulty* leading zeroes -export interface ChallengeLeadingZeroes { - algorithm: ChallengeAlgorithm; - strategy: ChallengeStrategy.LeadingZeroes; - salt: string; // random string +// export interface ChallengeLeadingZeroes { +// algorithm: ChallengeAlgorithm; +// strategy: ChallengeStrategy.LeadingZeroes; +// salt: string; // random string +// difficulty: number; +// } + +// // In this case, the server generates a random number, and the client will hash +// // the salt (a random string) + a random number until it finds a hash that is equal to challenge +// export interface ChallengeTargetNumber { +// algorithm: ChallengeAlgorithm; +// strategy: ChallengeStrategy.TargetNumber; +// salt: string; // random string +// target: string; // hash of salt + random number +// } + +export interface InnerChallengekCTF { + strategy: ChallengeStrategy.kCTF; + salt: UUID; // UUIDv7 difficulty: number; } -// In this case, the server generates a random number, and the client will hash -// the salt (a random string) + a random number until it finds a hash that is equal to challenge -export interface ChallengeTargetNumber { - algorithm: ChallengeAlgorithm; - strategy: ChallengeStrategy.TargetNumber; - salt: string; // random string - target: string; // hash of salt + random number +export interface ChallengekCTF { + strategy: ChallengeStrategy.kCTF; + challenge: string; } -export type Challenge = ChallengeLeadingZeroes | ChallengeTargetNumber; \ No newline at end of file +export type InnerChallenge = InnerChallengekCTF; + +export type Challenge = ChallengekCTF; \ No newline at end of file diff --git a/packages/lib/src/solver.ts b/packages/lib/src/solver.ts index 7161566..0ba4144 100644 --- a/packages/lib/src/solver.ts +++ b/packages/lib/src/solver.ts @@ -3,8 +3,10 @@ import WASMSolverUrl from '../../../solver/zig-out/bin/solver.wasm?url&inline'; type WasmExports = Record & { "malloc": (byte_count: number) => number | null; "free": (ptr: number | null, byte_count: number) => void; - "solve_leaading_zeroes_challenge": (challenge_ptr: number, challenge_len: number, difficulty: number) => number; - "solve_target_number_challenge": (challenge_ptr: number, challenge_len: number, target_ptr: number, target_len: number) => number; + // "solve_leaading_zeroes_challenge": (challenge_ptr: number, challenge_len: number, difficulty: number) => number; + // "solve_target_number_challenge": (challenge_ptr: number, challenge_len: number, target_ptr: number, target_len: + // number) => number; + "solve": (value_ptr: number, value_len: number) => number, "memory": WebAssembly.Memory; } @@ -17,6 +19,7 @@ export type SolverEnv = { __set_solution: (value: number) => void; __cmpxchg_solution: (expected: number, replacement: number) => number; __fetch_add_nonce: (value: number) => number; + __log: (str_ptr: number, str_len: number) => void; }; export async function get_wasm_module(): Promise { @@ -29,64 +32,35 @@ export async function init_solver(env: SolverEnv, module: WebAssembly.Module): P }) as unknown as SolverModule; } -export function solve_leaading_zeroes_challenge(solver: SolverModule, challenge: { salt: string, difficulty: number }): number { - const { salt, difficulty } = challenge; +export function solve(solver: SolverModule, challenge: string): string { + console.log(challenge); + const encoder = new TextEncoder(); + const challenge_buf = encoder.encode(challenge); - const salt_bytes = encoder.encode(salt); - - const salt_ptr = solver.exports.malloc(salt_bytes.length); - if (salt_ptr === 0 || salt_ptr === null) { + const challenge_ptr = solver.exports.malloc(challenge_buf.length); + if (challenge_ptr === 0 || challenge_ptr === null) { throw new Error("Failed to allocate memory for challenge string"); } const memory = new Uint8Array(solver.exports.memory.buffer); - memory.set(salt_bytes, salt_ptr); + memory.set(challenge_buf, challenge_ptr); - const ret = solver.exports.solve_leaading_zeroes_challenge( - salt_ptr, - salt_bytes.length, - difficulty, - ); + const ret = solver.exports.solve(challenge_ptr, challenge_buf.length); - if (ret < 0) { + console.log("RET", ret); + + if (ret <= 0) { throw new Error("Failed to solve challenge"); } - return ret; + const length = new DataView(solver.exports.memory.buffer, ret, 2).getUint16(0, true); + const solution = new TextDecoder().decode(solver.exports.memory.buffer.slice(ret + 2, ret + 2 + length)); + + console.log("SOLUTION", solution); + console.log("LENGTH", length); + + solver.exports.free(ret, 2 + length); + + return solution; } - -export function solve_target_number_challenge(solver: SolverModule, challenge: { salt: string, target: string }): number { - const { salt, target } = challenge; - const encoder = new TextEncoder(); - - const salt_bytes = encoder.encode(salt); - const target_bytes = encoder.encode(target); - - const salt_ptr = solver.exports.malloc(salt_bytes.length); - if (salt_ptr === 0 || salt_ptr === null) { - throw new Error("Failed to allocate memory for salt string"); - } - - const target_ptr = solver.exports.malloc(target_bytes.length); - if (target_ptr === 0 || target_ptr === null) { - throw new Error("Failed to allocate memory for target string"); - } - - const memory = new Uint8Array(solver.exports.memory.buffer); - memory.set(salt_bytes, salt_ptr); - memory.set(target_bytes, target_ptr); - - const ret = solver.exports.solve_target_number_challenge( - target_ptr, - target_bytes.length, - salt_ptr, - salt_bytes.length, - ); - - if (ret < 0) { - throw new Error("Failed to solve challenge"); - } - - return ret; -} \ No newline at end of file diff --git a/packages/lib/src/validator.ts b/packages/lib/src/validator.ts index 3c5185c..3498735 100644 --- a/packages/lib/src/validator.ts +++ b/packages/lib/src/validator.ts @@ -1,12 +1,11 @@ -import { ChallengeAlgorithm, ChallengeStrategy, type Challenge } from '.'; +import { ChallengeStrategy, type Challenge, type InnerChallenge } from '.'; import WASMValidatorUrl from '../../../solver/zig-out/bin/validator.wasm?url&inline'; +import { uuidv7obj } from 'uuidv7'; type WasmExports = Record & { "malloc": (byte_count: number) => number | null; "free": (ptr: number | null, byte_count: number) => void; - "validate_leading_zeroes_challenge": (challenge_ptr: number, challenge_len: number, nonce_ptr: number, nonce_len: number, difficulty: number) => number; - "validate_target_number_challenge": (target_ptr: number, target_len: number, nonce_ptr: number, nonce_len: number, salt_ptr: number, salt_len: number) => number; - "hash": (challenge_ptr: number, challenge_len: number, nonce_ptr: number, nonce_len: number) => bigint; + "validate": (challenge_ptr: number, challenge_len: number, solution_ptr: number, solution_len: number) => boolean; "memory": WebAssembly.Memory; } @@ -14,159 +13,92 @@ export interface ValidatorModule extends WebAssembly.Instance { exports: WasmExports; } -function array_to_base64(buffer: ArrayBuffer): string { - return btoa(String.fromCharCode(...new Uint8Array(buffer))); -} - -async function generate_leading_zeroes_challenge(parameters: Object, difficulty: number): Promise { - let parameters_str = Object.entries(parameters).map(([key, value]) => `${key}=${value}`).join("&"); - let salt = `${array_to_base64(crypto.getRandomValues(new Uint8Array(32)).buffer)}?${parameters_str}`; - - let challenge: Challenge = { - algorithm: ChallengeAlgorithm.Argon2id, - strategy: ChallengeStrategy.LeadingZeroes, - salt, - difficulty, - }; - - return challenge; -} - -async function generate_target_number_challenge(parameters: Object, max_number: number): Promise { - // in target number config, since we need to generate a target hash, we - // need to hash the salt + nonce, so the client knows what the target is - const validator = (await WebAssembly.instantiateStreaming(fetch(WASMValidatorUrl))).instance as unknown as ValidatorModule; - - let parameters_str = Object.entries(parameters).map(([key, value]) => `${key}=${value}`).join("&"); - let salt = `${array_to_base64(crypto.getRandomValues(new Uint8Array(32)).buffer)}?${parameters_str}`; - let random_number = new DataView(crypto.getRandomValues(new Uint8Array(4)).buffer).getUint32(0, true) % max_number; - - const encoder = new TextEncoder(); - const salt_bytes = encoder.encode(salt); - const random_number_bytes = encoder.encode(random_number.toString()); - - const salt_ptr = validator.exports.malloc(salt_bytes.length); - const random_number_ptr = validator.exports.malloc(random_number_bytes.length); - - if (salt_ptr === 0 || salt_ptr === null || random_number_ptr === 0 || random_number_ptr === null) { - console.error("Failed to allocate memory for challenge string"); - return null; +function array_to_base64(buffer: ArrayBufferLike): string { + let binary = ''; + const bytes = new Uint8Array(buffer); + for (var i = 0; i < bytes.byteLength; i++) { + binary += String.fromCharCode(bytes[i]); } - - const memory = new Uint8Array(validator.exports.memory.buffer); - memory.set(salt_bytes, salt_ptr); - memory.set(random_number_bytes, random_number_ptr); - - let target_blob: bigint = validator.exports.hash(salt_ptr, salt_bytes.length, random_number_ptr, random_number_bytes.length); - let target_ptr = Number(target_blob & BigInt(0xFFFFFFFF)); - let target_len = Number(target_blob >> BigInt(32)); - - validator.exports.free(salt_ptr, salt_bytes.length); - validator.exports.free(random_number_ptr, random_number_bytes.length); - - // do NOT use `memory` here, by this time it has almost definitely been resized and will cause errors to touch - let target_slice = new Uint8Array(validator.exports.memory.buffer.slice(target_ptr, target_ptr + target_len)); - const target = new TextDecoder().decode(target_slice); - - let challenge: Challenge = { - algorithm: ChallengeAlgorithm.Argon2id, - strategy: ChallengeStrategy.TargetNumber, - salt, - target - }; - - return challenge; + return btoa(binary); } -export interface LeadingZeroesChallengeConfig { +export interface kCTFChallengeConfig { parameters: Object; - strategy: ChallengeStrategy.LeadingZeroes; + strategy: ChallengeStrategy.kCTF; difficulty: number; } -export interface TargetNumberChallengeConfig { - parameters: Object; - strategy: ChallengeStrategy.TargetNumber; - max_number: number; -} +export type ChallengeConfig = kCTFChallengeConfig; -export type ChallengeConfig = LeadingZeroesChallengeConfig | TargetNumberChallengeConfig; +const VERSION = "s"; + +async function encode_challenge(challenge: InnerChallenge, parameters: Object = {}): Promise { + if (challenge.strategy !== ChallengeStrategy.kCTF) { + throw new Error("Unsupported challenge strategy"); + } + + if (challenge.difficulty < 1) { + throw new Error("Difficulty must be at least 1"); + } + + const difficulty_buf = new Uint8Array(4); + const view = new DataView(difficulty_buf.buffer); + view.setUint32(0, challenge.difficulty, false); + + const difficulty_str = array_to_base64(difficulty_buf.buffer); + const salt_str = array_to_base64(challenge.salt.bytes.buffer); + + // the parameters str is expected to be sliced out of the challenge via the widget before it sends it to the wasm solver. + let parameters_str = Object.entries(parameters).map(([key, value]) => `${key}=${value}`).join("&"); + if (parameters_str.length > 0) { + parameters_str = "?" + parameters_str; + } + + return `${VERSION}.${difficulty_str}.${salt_str}${parameters_str}`; +} export async function generate_challenge(config: ChallengeConfig): Promise { - let challenge: Challenge | null = null; - switch (config.strategy) { - case ChallengeStrategy.LeadingZeroes: - challenge = await generate_leading_zeroes_challenge(config.parameters, config.difficulty); - break; - case ChallengeStrategy.TargetNumber: - challenge = await generate_target_number_challenge(config.parameters, config.max_number); - break; + if (config.difficulty < 1) { + throw new Error("Difficulty must be at least 1"); } - if (challenge === null) { - return null; + const challenge: InnerChallenge = { + strategy: ChallengeStrategy.kCTF, + salt: uuidv7obj(), + difficulty: config.difficulty, } - return challenge; + return { + strategy: ChallengeStrategy.kCTF, + challenge: await encode_challenge(challenge), + }; } -export async function validate_challenge(challenge: Challenge, challenge_solution: { challenge: string, nonce: string }): Promise { - const validator = (await WebAssembly.instantiateStreaming(fetch(WASMValidatorUrl))).instance as unknown as ValidatorModule - +export async function validate_challenge(challenge: Challenge, challenge_solution: string): Promise { + const validator = (await WebAssembly.instantiateStreaming(fetch(WASMValidatorUrl), { + env: { + __log: (str_ptr: number, str_len: number) => console.log(new TextDecoder().decode(new Uint8Array(validator.exports.memory.buffer, str_ptr, str_len))), + } + })).instance as unknown as ValidatorModule const encoder = new TextEncoder(); + const challenge_buf = encoder.encode(challenge.challenge); + const solution_buf = encoder.encode(challenge_solution); - let err; - let memory; - let nonce_bytes, nonce_ptr; - let target_bytes, target_ptr; - switch (challenge.strategy) { - case ChallengeStrategy.LeadingZeroes: - target_bytes = encoder.encode(challenge_solution.challenge); - nonce_bytes = encoder.encode(challenge_solution.nonce); + const challenge_ptr = validator.exports.malloc(challenge_buf.length); + const solution_ptr = validator.exports.malloc(solution_buf.length); - target_ptr = validator.exports.malloc(challenge_solution.challenge.length); - nonce_ptr = validator.exports.malloc(challenge_solution.nonce.length); - - if (target_ptr === 0 || target_ptr === null || nonce_ptr === 0 || nonce_ptr === null) { - console.error("Failed to allocate memory for challenge string"); - return false; - } - - memory = new Uint8Array(validator.exports.memory.buffer); - memory.set(target_bytes, target_ptr); - memory.set(nonce_bytes, nonce_ptr); - - err = validator.exports.validate_leading_zeroes_challenge(target_ptr, target_bytes.length, nonce_ptr, nonce_bytes.length, challenge.difficulty); - - validator.exports.free(target_ptr, target_bytes.length); - validator.exports.free(nonce_ptr, nonce_bytes.length); - break; - case ChallengeStrategy.TargetNumber: - target_bytes = encoder.encode(challenge.target); - const salt_bytes = encoder.encode(challenge.salt); - nonce_bytes = encoder.encode(challenge_solution.nonce); - - const salt_ptr = validator.exports.malloc(salt_bytes.length); - target_ptr = validator.exports.malloc(target_bytes.length); - nonce_ptr = validator.exports.malloc(nonce_bytes.length); - - if (salt_ptr === 0 || salt_ptr === null || target_ptr === 0 || target_ptr === null || nonce_ptr === 0 || nonce_ptr === null) { - console.error("Failed to allocate memory for challenge string"); - return false; - } - - memory = new Uint8Array(validator.exports.memory.buffer); - memory.set(salt_bytes, salt_ptr); - memory.set(target_bytes, target_ptr); - memory.set(nonce_bytes, nonce_ptr); - - err = validator.exports.validate_target_number_challenge(target_ptr, target_bytes.length, nonce_ptr, nonce_bytes.length, salt_ptr, salt_bytes.length); - - validator.exports.free(salt_ptr, salt_bytes.length); - validator.exports.free(target_ptr, target_bytes.length); - validator.exports.free(nonce_ptr, nonce_bytes.length); - break; + if (challenge_ptr === 0 || challenge_ptr === null || solution_ptr === 0 || solution_ptr === null) { + console.error("Failed to allocate memory for challenge string"); + return false; } - return err === 0; + const memory = new Uint8Array(validator.exports.memory.buffer); + memory.set(challenge_buf, challenge_ptr); + memory.set(solution_buf, solution_ptr); + + const is_valid = validator.exports.validate(challenge_ptr, challenge.challenge.length, solution_ptr, challenge_solution.length); + validator.exports.free(challenge_ptr, challenge.challenge.length); + validator.exports.free(solution_ptr, challenge_solution.length); + + return is_valid; } \ No newline at end of file diff --git a/packages/widget/package-lock.json b/packages/widget/package-lock.json index ff2b26c..76bc129 100644 --- a/packages/widget/package-lock.json +++ b/packages/widget/package-lock.json @@ -1433,7 +1433,6 @@ "integrity": "sha512-ZsJzA5thDQMSQO788d7IocwwQbI8B5OPzmqNvpf3NY/+MHDAS759Wo0gd2WQeXYt5AAAQjzcrTVC6SKCuYgoCQ==", "dev": true, "license": "MIT", - "peer": true, "dependencies": { "undici-types": "~6.21.0" } @@ -2148,7 +2147,6 @@ "integrity": "sha512-5gTmgEY/sqK6gFXLIsQNH19lWb4ebPDLA4SdLP7dsWkIXHWlG66oPuVvXSGFPppYZz8ZDZq0dYYrbHfBCVUb1Q==", "dev": true, "license": "MIT", - "peer": true, "engines": { "node": ">=12" }, @@ -2422,7 +2420,6 @@ "integrity": "sha512-jl1vZzPDinLr9eUt3J/t7V6FgNEw9QjvBPdysz9KfQDD41fQrC2Y4vKQdiaUpFT4bXlb1RHhLpp8wtm6M5TgSw==", "dev": true, "license": "Apache-2.0", - "peer": true, "bin": { "tsc": "bin/tsc", "tsserver": "bin/tsserver" @@ -2461,7 +2458,6 @@ "integrity": "sha512-BxAKBWmIbrDgrokdGZH1IgkIk/5mMHDreLDmCJ0qpyJaAteP8NvMhkwr/ZCQNqNH97bw/dANTE9PDzqwJghfMQ==", "dev": true, "license": "MIT", - "peer": true, "dependencies": { "esbuild": "^0.25.0", "fdir": "^6.5.0", diff --git a/packages/widget/src/pow-captcha.ts b/packages/widget/src/pow-captcha.ts index d441202..6f9c5d9 100644 --- a/packages/widget/src/pow-captcha.ts +++ b/packages/widget/src/pow-captcha.ts @@ -21,6 +21,9 @@ export class PowCaptcha extends LitElement { background-color: var(--impost-widget-background-color, #070408); border-radius: var(--impost-widget-border-radius, 8px); } + .impost-error-icon { + color: var(--impost-widget-error-icon-color, #FF8117); + } .impost-main { display: flex; align-items: center; @@ -86,32 +89,20 @@ export class PowCaptcha extends LitElement { @state() private solution: string = ''; - @state() - private errorMessage: string = ''; - @state() private challengeData: Challenge | null = null; @state() - private solved: boolean = false; - - @state() - private isSolving: boolean = false; + private status: 'unsolved' | 'solving' | 'solved' | 'error' = 'unsolved'; @state() private disabled: boolean = true; - @state() - private hashRate: number = 0; - // stores the nonce and solution atomics private sab: SharedArrayBuffer = new SharedArrayBuffer(8); private solverWorkers: Worker[] | null = null; - private solveStartTime: number | null = null; - private hashRateInterval: number | null = null; - override connectedCallback() { super.connectedCallback(); this.fetchChallenge(); @@ -141,10 +132,6 @@ export class PowCaptcha extends LitElement { override disconnectedCallback() { super.disconnectedCallback(); - if (this.hashRateInterval !== null) { - clearInterval(this.hashRateInterval); - this.hashRateInterval = null; - } for (const worker of this.solverWorkers || []) { worker.terminate(); @@ -157,7 +144,6 @@ export class PowCaptcha extends LitElement { } async fetchChallenge() { - this.errorMessage = ''; if (this.challengeData !== null) { return; } @@ -179,14 +165,14 @@ export class PowCaptcha extends LitElement { }) .catch(error => { console.error('Error fetching challenge:', error); - this.errorMessage = 'Failed to fetch challenge. Please try again.'; + console.error('Failed to fetch challenge'); }); } async initWorkers() { this.solverWorkers = []; - const num_workers = navigator.hardwareConcurrency; + const num_workers = 1; for (let i = 0; i < num_workers; i++) { this.solverWorkers.push(new ChallengeWorker()); } @@ -237,7 +223,8 @@ export class PowCaptcha extends LitElement { let timeout: number; const timeoutPromise = new Promise((_, reject) => { timeout = setTimeout(() => { - this.errorMessage = 'Failed to initialize workers in time. Please refresh the page.'; + console.error('Failed to initialize workers in time'); + this.status = 'error'; reject(new Error(`Function timed out after ${timeoutMs}ms`)); }, timeoutMs); }); @@ -272,19 +259,27 @@ export class PowCaptcha extends LitElement { worker.addEventListener('message', message_handler); worker.addEventListener('error', error_handler); + // switch (request.strategy) { + // case ChallengeStrategy.LeadingZeroes: + // worker.postMessage({ + // strategy: ChallengeStrategy.LeadingZeroes, + // salt: request.salt, + // difficulty: request.difficulty, + // } as WorkerRequest); + // break; + // case ChallengeStrategy.TargetNumber: + // worker.postMessage({ + // strategy: ChallengeStrategy.TargetNumber, + // target: request.target, + // salt: request.salt, + // } as WorkerRequest); + // break; + // } switch (request.strategy) { - case ChallengeStrategy.LeadingZeroes: + case ChallengeStrategy.kCTF: worker.postMessage({ - strategy: ChallengeStrategy.LeadingZeroes, - salt: request.salt, - difficulty: request.difficulty, - } as WorkerRequest); - break; - case ChallengeStrategy.TargetNumber: - worker.postMessage({ - strategy: ChallengeStrategy.TargetNumber, - target: request.target, - salt: request.salt, + strategy: ChallengeStrategy.kCTF, + challenge: request.challenge, } as WorkerRequest); break; } @@ -293,7 +288,8 @@ export class PowCaptcha extends LitElement { async solveChallenge() { if (!this.challengeData || this.solverWorkers === null) { - this.errorMessage = 'Captcha is not ready. Please wait or refresh.'; + console.error('solveChallenge called before challenge is ready'); + this.status = 'error'; return; } @@ -301,25 +297,14 @@ export class PowCaptcha extends LitElement { return; } - this.solveStartTime = performance.now(); - this.hashRateInterval = setInterval(async () => { - const nonce = this.getCurrentWorkingNonce(); - - this.hashRate = (nonce / ((performance.now() - this.solveStartTime!) / 1000)); - }, 250); - this.dispatchEvent(new CustomEvent('impost:solve', { - detail: { - solution: this.solution, - }, bubbles: true, composed: true, })) console.log(this.challengeData); - this.isSolving = true; - this.errorMessage = ''; + this.status = 'solving'; this.solution = ''; const atomics_view = new Int32Array(this.sab); @@ -328,19 +313,27 @@ export class PowCaptcha extends LitElement { let request: ChallengeSolveRequest; + // switch (this.challengeData.strategy) { + // case ChallengeStrategy.LeadingZeroes: + // request = { + // strategy: ChallengeStrategy.LeadingZeroes, + // salt: this.challengeData.salt, + // difficulty: this.challengeData.difficulty, + // }; + // break; + // case ChallengeStrategy.TargetNumber: + // request = { + // strategy: ChallengeStrategy.TargetNumber, + // target: this.challengeData.target, + // salt: this.challengeData.salt, + // }; + // break; + // } switch (this.challengeData.strategy) { - case ChallengeStrategy.LeadingZeroes: + case ChallengeStrategy.kCTF: request = { - strategy: ChallengeStrategy.LeadingZeroes, - salt: this.challengeData.salt, - difficulty: this.challengeData.difficulty, - }; - break; - case ChallengeStrategy.TargetNumber: - request = { - strategy: ChallengeStrategy.TargetNumber, - target: this.challengeData.target, - salt: this.challengeData.salt, + strategy: ChallengeStrategy.kCTF, + challenge: this.challengeData.challenge, }; break; } @@ -350,7 +343,7 @@ export class PowCaptcha extends LitElement { // blocking, some workers may block on the read, and as soon as they // unblock, they return 0 since the challenge is already solved. // - // We need to do a better job of tracking solvers, so if one worker + // TODO: We need to do a better job of tracking solvers, so if one worker // errors out, we only error out if all workers have errored out. let worker_promises: Promise[] = []; for (let worker of this.solverWorkers) { @@ -361,31 +354,44 @@ export class PowCaptcha extends LitElement { let solution = await Promise.race(worker_promises); if (solution.type === WorkerResponseType.Error) { - this.errorMessage = solution.error; + console.error("Worker error:", solution.error); + this.status = 'error'; return; } if (solution.type !== WorkerResponseType.Solution) { - this.errorMessage = "Something went wrong, please try again later."; + console.error("Worker sent spurious message"); + this.status = 'error'; return; } - this.solution = Atomics.load(atomics_view, 1).toString(); - this.isSolving = false; - this.solved = true; + // TODO: configure if we should fetch or not + try { + await fetch(`${this.challengeUrl}/challenge`, { + method: 'POST', + body: JSON.stringify({ + challenge: this.challengeData.challenge, + solution: solution.solution, + }), + headers: { + 'Content-Type': 'application/json' + } + }) - if (this.hashRateInterval !== null) { - clearInterval(this.hashRateInterval); - this.hashRateInterval = null; + this.status = 'solved'; + + this.dispatchEvent(new CustomEvent('impost:solved', { + detail: { + challenge: this.challengeData.challenge, + solution: solution.solution, + }, + bubbles: true, + composed: true, + })) + } catch { + console.error('Failed to submit solution'); + this.status = 'error'; } - - this.dispatchEvent(new CustomEvent('impost:solved', { - detail: { - solution: this.solution, - }, - bubbles: true, - composed: true, - })) } solvePreventDefault(event: Event) { @@ -400,12 +406,6 @@ export class PowCaptcha extends LitElement { this.challengejson = ''; } - if (this.errorMessage) { - return html` -
${this.errorMessage}
- `; - } - if (this.challengeData === null) { return html`
Loading captcha challenge...
@@ -416,9 +416,9 @@ export class PowCaptcha extends LitElement {
- ${!this.isSolving ? html` - - ` : html` + ${this.status !== 'solving' ? html`${this.status === 'error' ? html`` : html` + + `}` : html` @@ -431,16 +431,10 @@ export class PowCaptcha extends LitElement { `}
- +
- ${this.isSolving && this.hashRate > 0 ? html` -
- H/s: - ${this.hashRate.toFixed(2)} -
- ` : ``} diff --git a/packages/widget/src/solver-worker.ts b/packages/widget/src/solver-worker.ts index 1c495ee..45d16ff 100644 --- a/packages/widget/src/solver-worker.ts +++ b/packages/widget/src/solver-worker.ts @@ -8,9 +8,7 @@ import { WorkerResponseType, } from "./types/worker"; -import { ChallengeStrategy } from "@impost/lib"; - -import { type SolverModule, init_solver, solve_leaading_zeroes_challenge, solve_target_number_challenge } from '@impost/lib/solver'; +import { type SolverModule, init_solver, solve } from '@impost/lib/solver'; let solver: SolverModule | null = null; @@ -29,6 +27,7 @@ onmessage = async (event: MessageEvent) => { __set_solution: (value: number) => Atomics.store(atomic_solution!, 0, value), __cmpxchg_solution: (expected: number, replacement: number) => Atomics.compareExchange(atomic_solution!, 0, expected, replacement), __fetch_add_nonce: (value: number) => Atomics.add(atomic_nonce!, 0, value), + __log: (str_ptr: number, str_len: number) => console.log(new TextDecoder().decode(new Uint8Array(solver!.exports.memory.buffer, str_ptr, str_len))), }, module); } catch (error: any) { postMessage({ @@ -60,32 +59,19 @@ onmessage = async (event: MessageEvent) => { return; } - if (atomic_nonce === null || atomic_solution === null) { - throw new Error("Atomics not initialized"); - } - - const { strategy } = event.data; - - let solution: number; - switch (strategy) { - case ChallengeStrategy.LeadingZeroes: - solution = solve_leaading_zeroes_challenge(solver, event.data) - break; - case ChallengeStrategy.TargetNumber: - solution = solve_target_number_challenge(solver, event.data) - break; - } - - // we are just assuming that if its less than -1, its the min i32 - if (solution < 0) { - return postMessage({ + let solution: string; + try { + solution = solve(solver, event.data.challenge); + } catch (error: any) { + postMessage({ type: WorkerResponseType.Error, - error: "failed to solve challenge", + error: `Failed to solve challenge: ${error.message}`, } as SolutionMessage); + return; } postMessage({ type: WorkerResponseType.Solution, - nonce: solution === -1 ? null : solution.toString() + solution, } as SolutionMessage); }; diff --git a/packages/widget/src/types/worker.ts b/packages/widget/src/types/worker.ts index dcfb1db..464ca19 100644 --- a/packages/widget/src/types/worker.ts +++ b/packages/widget/src/types/worker.ts @@ -12,28 +12,37 @@ interface WorkerInitRequest { sab: SharedArrayBuffer; } -interface ChallengeLeadingZeroesSolveRequest { - strategy: ChallengeStrategy.LeadingZeroes; - salt: string; - difficulty: number; +// interface ChallengeLeadingZeroesSolveRequest { +// strategy: ChallengeStrategy.LeadingZeroes; +// salt: string; +// difficulty: number; +// } + +// interface WorkerChallengeLeadingZeroesSolveRequest extends ChallengeLeadingZeroesSolveRequest { +// type: WorkerMessageType.Challenge; +// } + +// interface ChallengeTargetNumberSolveRequest { +// strategy: ChallengeStrategy.TargetNumber; +// target: string; +// salt: string; +// } + +// interface WorkerChallengeTargetNumberSolveRequest extends ChallengeTargetNumberSolveRequest { +// type: WorkerMessageType.Challenge; +// } + +interface ChallengekCTFSolveRequest { + strategy: ChallengeStrategy.kCTF; + challenge: string; } -interface WorkerChallengeLeadingZeroesSolveRequest extends ChallengeLeadingZeroesSolveRequest { +interface WorkerChallengekCTFSolveRequest extends ChallengekCTFSolveRequest { type: WorkerMessageType.Challenge; } -interface ChallengeTargetNumberSolveRequest { - strategy: ChallengeStrategy.TargetNumber; - target: string; - salt: string; -} - -interface WorkerChallengeTargetNumberSolveRequest extends ChallengeTargetNumberSolveRequest { - type: WorkerMessageType.Challenge; -} - -export type ChallengeSolveRequest = ChallengeLeadingZeroesSolveRequest | ChallengeTargetNumberSolveRequest; -type WorkerChallengeSolveRequest = WorkerChallengeLeadingZeroesSolveRequest | WorkerChallengeTargetNumberSolveRequest; +export type ChallengeSolveRequest = ChallengekCTFSolveRequest; +type WorkerChallengeSolveRequest = WorkerChallengekCTFSolveRequest; export type WorkerRequest = WorkerInitRequest | WorkerChallengeSolveRequest; @@ -50,7 +59,7 @@ interface ErrorMessageResponse { interface SolutionMessageResponse { type: WorkerResponseType.Solution; - nonce: string; + solution: string; } interface InitOkMessageResponse { diff --git a/solver/build.zig b/solver/build.zig index 848d1d3..6a8d6bb 100644 --- a/solver/build.zig +++ b/solver/build.zig @@ -2,17 +2,13 @@ const std = @import("std"); pub fn build(b: *std.Build) void { const optimize = b.standardOptimizeOption(.{}); - const wasm_target = b.resolveTargetQuery(.{ - .cpu_arch = .wasm32, - .os_tag = .freestanding, - .cpu_features_add = std.Target.wasm.featureSet(&.{ .bulk_memory, .bulk_memory_opt, .simd128, .tail_call }), - }); + const target = b.standardTargetOptions(.{}); // solver const solver_mod = b.addModule("solver", .{ .root_source_file = b.path("src/solver.zig"), .optimize = optimize, - .target = wasm_target, + .target = target, }); const solver_exe = b.addExecutable(.{ @@ -20,10 +16,12 @@ pub fn build(b: *std.Build) void { .root_module = solver_mod, }); - solver_exe.entry = .disabled; - solver_exe.rdynamic = true; - solver_exe.lto = .full; - solver_exe.link_gc_sections = true; + if (target.result.cpu.arch == .wasm32) { + solver_exe.entry = .disabled; + solver_exe.rdynamic = true; + solver_exe.link_gc_sections = true; + solver_exe.lto = .full; + } b.installArtifact(solver_exe); @@ -31,16 +29,18 @@ pub fn build(b: *std.Build) void { const validator_mod = b.addModule("validator", .{ .root_source_file = b.path("src/validator.zig"), .optimize = optimize, - .target = wasm_target, + .target = target, }); - const validator_exe = b.addLibrary(.{ + const validator_exe = b.addExecutable(.{ .name = "validator", .root_module = validator_mod, }); - validator_exe.entry = .disabled; - validator_exe.rdynamic = true; + if (target.result.cpu.arch == .wasm32) { + validator_exe.entry = .disabled; + validator_exe.rdynamic = true; + } b.installArtifact(validator_exe); } diff --git a/solver/src/kctf.zig b/solver/src/kctf.zig index f0159c7..957d6e9 100644 --- a/solver/src/kctf.zig +++ b/solver/src/kctf.zig @@ -5,3 +5,211 @@ const std = @import("std"); const Allocator = std.mem.Allocator; const assert = std.debug.assert; const math = std.math; +const Int = math.big.int.Managed; + +var managed_one: ?Int = null; + +fn get_bit(n: *Int, idx: usize) !bool { + if (n.len() < idx / @typeInfo(usize).int.bits) { + return false; + } + + var foo = try n.clone(); + defer foo.deinit(); + + try foo.shiftRight(n, idx); + try foo.bitAnd(&foo, &managed_one.?); + return foo.eql(managed_one.?); +} + +pub fn square_mod(n: *Int) !void { + const allocator = n.allocator; + try n.sqr(n); + + var high = try Int.init(allocator); + defer high.deinit(); + try high.shiftRight(n, 1279); // high = n >> 1279 + + var mask = try Int.init(allocator); + defer mask.deinit(); + + if (managed_one == null) { + managed_one = try Int.init(allocator); + try managed_one.?.set(1); + } + + try mask.set(1); + try mask.shiftLeft(&mask, 1279); + try mask.sub(&mask, &managed_one.?); + + try n.bitAnd(n, &mask); + + try n.add(n, &high); + + if (try get_bit(n, 1279)) { + // clear bit 1279 + var power_of_2 = try Int.init(allocator); + defer power_of_2.deinit(); + try power_of_2.set(1); + try power_of_2.shiftLeft(&power_of_2, 1279); + try n.sub(n, &power_of_2); + + // *n += 1; + try n.add(n, &managed_one.?); + } +} + +pub const Version = "s"; + +pub const Challenge = struct { + difficulty: ?u32, + salt: std.math.big.int.Managed, + + const Self = @This(); + + pub fn destroy(self: *Self, allocator: Allocator) void { + self.salt.deinit(); + allocator.destroy(self); + } + + pub fn encode(self: *Self, allocator: Allocator) ![]u8 { + const solution_base64_len = std.base64.standard.Encoder.calcSize(self.salt.len() * @sizeOf(usize)); + const dest = try allocator.alloc(u8, solution_base64_len); + defer allocator.free(dest); + @memset(dest, 0); + + const limbs_u8_buffer: []u8 = std.mem.sliceAsBytes(self.salt.limbs[0..self.salt.len()]); + const base64_str = std.base64.standard.Encoder.encode(dest, limbs_u8_buffer); + + return try std.fmt.allocPrint(allocator, "{s}", .{base64_str}); + } +}; + +pub fn decode(allocator: Allocator, challenge: []const u8) !*Challenge { + var parts = std.mem.splitAny(u8, challenge, "."); + if (parts.next()) |part| { + if (!std.mem.eql(u8, part, Version)) { + return error.InvalidChallenge; + } + } else { + return error.InvalidChallenge; + } + + var difficulty: ?u32 = null; + var next_part = parts.next() orelse return error.InvalidChallenge; + if (parts.peek()) |_| { + // must be .. + const difficulty_bytes = try allocator.alloc(u8, try std.base64.standard.Decoder.calcSizeForSlice(next_part)); + defer allocator.free(difficulty_bytes); + + try std.base64.standard.Decoder.decode(difficulty_bytes, next_part); + + std.log.info("Decoded difficulty bytes: {any}\n", .{difficulty_bytes}); + + var difficulty_array: [4]u8 = .{0} ** 4; + if (difficulty_bytes.len > 4) { + const split_idx = difficulty_bytes.len - 4; + for (difficulty_bytes[0..split_idx]) |byte| { + if (byte != 0) return error.DifficultyTooLarge; + } + + @memcpy(&difficulty_array, difficulty_bytes[split_idx..]); + difficulty = std.mem.readInt(u32, &difficulty_array, .big); + } else { + const start_idx = 4 - difficulty_bytes.len; + @memcpy(&difficulty_array, difficulty_bytes[start_idx..]); + difficulty = std.mem.readInt(u32, &difficulty_array, .big); + } + + next_part = parts.next() orelse return error.InvalidChallenge; + } + + var salt = try std.math.big.int.Managed.init(allocator); + errdefer salt.deinit(); + + const salt_str = next_part; + const salt_bytes_len = try std.base64.standard.Decoder.calcSizeForSlice(salt_str); + + std.log.info("salt_bytes_len: {d}\n", .{salt_bytes_len}); + + const salt_bytes = try allocator.alloc(u8, salt_bytes_len); + defer allocator.free(salt_bytes); + + try std.base64.standard.Decoder.decode(salt_bytes, salt_str); + + std.log.info("decoded salt: {any}\n", .{salt_bytes}); + + const usize_salt_bytes: []align(1) usize = std.mem.bytesAsSlice(usize, salt_bytes); + // TODO: the bytes are being read in as little endian, but need to be read in as big endian + std.log.info("usize_salt_bytes: {any}\n", .{usize_salt_bytes}); + try salt.ensureCapacity(usize_salt_bytes.len); + @memcpy(salt.limbs[0..usize_salt_bytes.len], usize_salt_bytes); + salt.setLen(usize_salt_bytes.len); + + const challenge_ptr = try allocator.create(Challenge); + errdefer challenge_ptr.destroy(allocator); + + challenge_ptr.* = Challenge{ + .difficulty = difficulty, + .salt = salt, + }; + + return challenge_ptr; +} + +pub fn solve(allocator: Allocator, challenge: *Challenge) ![]u8 { + if (challenge.difficulty == null) { + return error.InvalidChallenge; + } + + for (0..challenge.difficulty.?) |_| { + std.log.info("Solving challenge with difficulty {d}\n", .{challenge.difficulty.?}); + for (0..1277) |_| { + try square_mod(&challenge.salt); + } + try challenge.salt.bitXor(&challenge.salt, &managed_one.?); + } + + std.log.info("solved challenge: {any}\n", .{challenge}); + + return try challenge.encode(allocator); +} + +pub fn check(allocator: Allocator, challenge: *Challenge, solution: *Challenge) !bool { + std.log.info("{d}", .{challenge.difficulty.?}); + std.log.info("{any} vs {any}\n", .{ challenge, solution }); + + if (challenge.difficulty == null) { + return error.InvalidChallenge; + } + + if (managed_one == null) { + managed_one = try Int.init(allocator); + try managed_one.?.set(1); + } + + for (0..challenge.difficulty.?) |_| { + try solution.salt.bitXor(&solution.salt, &managed_one.?); + + try square_mod(&solution.salt); + } + + std.log.info("{any} vs {any}\n", .{ challenge, solution }); + + if (challenge.salt.eql(solution.salt)) { + return true; + } + + var foo = try std.math.big.int.Managed.initSet(allocator, 2); + defer foo.deinit(); + try foo.pow(&foo, 1279); + try foo.sub(&foo, &managed_one.?); + try foo.sub(&foo, &challenge.salt); + + if (foo.eql(solution.salt)) { + std.log.info("challenge solved!\n", .{}); + return true; + } + + return false; +} diff --git a/solver/src/solver.zig b/solver/src/solver.zig index 744dc13..6a40ca5 100644 --- a/solver/src/solver.zig +++ b/solver/src/solver.zig @@ -1,23 +1,26 @@ const std = @import("std"); +const builtin = @import("builtin"); const Allocator = std.mem.Allocator; -const hasher = @import("hasher.zig"); +const kCTF = @import("kctf.zig"); -// var gpa: std.heap.GeneralPurposeAllocator(.{}) = .init; -// var allocator = gpa.allocator(); +var gpa: std.heap.GeneralPurposeAllocator(.{}) = .init; +var allocator = gpa.allocator(); -var allocator = std.heap.wasm_allocator; +extern fn __log(str_ptr: usize, str_len: usize) void; -extern fn __get_solution() i32; -extern fn __set_solution(value: i32) void; -extern fn __cmpxchg_solution(old: i32, new: i32) i32; -extern fn __fetch_add_nonce(value: i32) i32; +fn log(comptime level: std.log.Level, comptime scope: @TypeOf(.EnumLiteral), comptime fmt: []const u8, args: anytype) void { + if (comptime builtin.target.cpu.arch != .wasm32) { + std.log.defaultLog(level, scope, fmt, args); + return; + } -// fn log(comptime fmt: []const u8, args: anytype) void { -// const formatted = std.fmt.allocPrint(allocator, fmt, args) catch return; -// __log(@intFromPtr(formatted.ptr), formatted.len); -// allocator.free(formatted); -// } + const formatted = std.fmt.allocPrint(allocator, fmt, args) catch return; + __log(@intFromPtr(formatted.ptr), formatted.len); + allocator.free(formatted); +} + +pub const std_options: std.Options = .{ .logFn = log }; export fn malloc(byte_count: usize) ?*u8 { const ptr = allocator.alloc(u8, byte_count) catch return null; @@ -31,129 +34,55 @@ export fn free(ptr: ?*anyopaque, byte_count: usize) void { } } -const SolveError = enum(u32) { - InvalidDifficulty = 1, - InvalidNonce = 2, - NoSolution = 3, - OutOfMemory = 4, -}; +// value_ptr is a string to the entire encoded challenge string (e.g. "s.xxxxxxxxx.xxxxxxx") +export fn solve(value_ptr: [*]u8, value_len: usize) usize { + const challenge_slice = value_ptr[0..value_len]; -var solve_error: ?SolveError = null; -export fn get_solve_error() u32 { - if (solve_error) |err| { - return @intFromEnum(err); - } - return 0; -} + std.log.info("Solve called with challenge {s}\n", .{challenge_slice}); -// returns nonce on success, -1 on failure -// to get the error, call get_solve_error -export fn solve_leaading_zeroes_challenge(challenge_ptr: [*]u8, challenge_len: usize, difficulty: u32) i32 { - solve_error = null; - - const challenge_slice = challenge_ptr[0..challenge_len]; - - if (difficulty < 1 or difficulty > 64) { - solve_error = SolveError.InvalidDifficulty; - return -1; - } - - var target_prefix_buffer: [64]u8 = @splat('0'); - const target_prefix = target_prefix_buffer[0..difficulty]; - - const max_nonce_iterations: u64 = 1_000_000_000; - - // 64 + 9 digits for nonce since the max nonce is 999_999_999 (not 1 billion since nonce < max_nonce_iterations) - var input_buffer: []u8 = allocator.alloc(u8, challenge_len + 9) catch { - // log("Failed to allocate memory for challenge\n", .{}); - solve_error = SolveError.OutOfMemory; - return -1; + const challenge = kCTF.decode(allocator, challenge_slice) catch |err| { + std.log.info("Error decoding challenge: {s}\n", .{@errorName(err)}); + return 0; }; - // dont leak memory :pepega: - defer allocator.free(input_buffer); + defer challenge.destroy(allocator); - @memcpy(input_buffer[0..challenge_len], challenge_slice); - - var nonce = __fetch_add_nonce(1); - - while (nonce < max_nonce_iterations) : (nonce = __fetch_add_nonce(1)) { - if (__get_solution() != -1) { - // solution has already been found, no point in continuing - return 0; - } - - const nonce_str = std.fmt.bufPrint(input_buffer[challenge_len..], "{d}", .{nonce}) catch { - solve_error = SolveError.InvalidNonce; - return -1; - }; - - const hash_hex_slice = hasher.hash(allocator, input_buffer[0..challenge_len], input_buffer[challenge_len .. challenge_len + nonce_str.len]) catch { - solve_error = SolveError.OutOfMemory; - return -1; - }; - - if (std.mem.startsWith(u8, hash_hex_slice, target_prefix)) { - // Found a solution! - if (__cmpxchg_solution(-1, nonce) == -1) { - // we found a solution, and we are the first to do so - return nonce; - } else { - // we found a solution, but we are not the first to do so - return 0; - } - } - } - - solve_error = SolveError.NoSolution; - return -1; -} - -export fn solve_target_number_challenge(target_ptr: [*]u8, target_len: usize, salt_ptr: [*]u8, salt_len: usize) i32 { - solve_error = null; - - const target_slice = target_ptr[0..target_len]; - const salt_slice = salt_ptr[0..salt_len]; - - // TODO: take in max number - const max_nonce_iterations: usize = 1_000_000_000; - const max_digits = std.math.log10(max_nonce_iterations); - - var input_buffer: []u8 = allocator.alloc(u8, salt_len + max_digits) catch { - solve_error = SolveError.OutOfMemory; - return -1; + std.log.info("decoded challenge {any}\n", .{challenge}); + const solution = kCTF.solve(allocator, challenge) catch |err| { + std.log.info("Error solving challenge: {s}\n", .{@errorName(err)}); + return 0; }; - defer allocator.free(input_buffer); - @memcpy(input_buffer[0..salt_len], salt_slice); + std.log.info("Solution: {s}\n", .{solution}); - var nonce = __fetch_add_nonce(1); - while (nonce < max_nonce_iterations) : (nonce = __fetch_add_nonce(1)) { - if (__get_solution() != -1) { - return 0; - } + const output_ptr = allocator.alloc(u8, solution.len + 4) catch return 0; - const nonce_str = std.fmt.bufPrint(input_buffer[salt_len..], "{d}", .{nonce}) catch { - solve_error = SolveError.InvalidNonce; - return -1; - }; + var output_slice = output_ptr[0 .. solution.len + 4]; + if (output_slice.len - 2 > std.math.maxInt(u16)) { + return 0; + } + const output_len: u16 = @intCast(output_slice.len - 2); + // convert to little endian + output_slice[0] = @intCast(output_len & 0xFF); // LSB + output_slice[1] = @intCast(output_len >> 8); // MSB - const hash_hex_slice = hasher.hash(allocator, input_buffer[0..salt_len], input_buffer[salt_len .. salt_len + nonce_str.len]) catch { - solve_error = SolveError.OutOfMemory; - return -1; - }; + @memcpy(output_slice[2..4], "s."); + @memcpy(output_slice[4 .. 4 + solution.len], solution); + allocator.free(solution); - if (std.mem.eql(u8, target_slice, hash_hex_slice)) { - // Found a solution! - if (__cmpxchg_solution(-1, nonce) == -1) { - // we found a solution, and we are the first to do so - return nonce; - } else { - // we found a solution, but we are not the first to do so - return 0; - } - } + return @intFromPtr(output_ptr.ptr); +} + +pub fn main() anyerror!void { + if (comptime builtin.cpu.arch == .wasm32) return; + + var args = try std.process.argsAlloc(allocator); + if (args.len < 2) { + std.log.err("Usage: zig run src/kctf.zig ", .{}); + return; } - solve_error = SolveError.NoSolution; - return -1; + const challenge = try kCTF.decode(allocator, args[1]); + const solution = try kCTF.solve(allocator, challenge); + + std.log.info("Solution: {s}", .{solution}); } diff --git a/solver/src/validator.zig b/solver/src/validator.zig index f1ca850..d3bee4f 100644 --- a/solver/src/validator.zig +++ b/solver/src/validator.zig @@ -1,9 +1,26 @@ const std = @import("std"); +const builtin = @import("builtin"); const Allocator = std.mem.Allocator; -const hasher = @import("hasher.zig"); +const kCTF = @import("kctf.zig"); -var allocator = std.heap.wasm_allocator; +var gpa: std.heap.GeneralPurposeAllocator(.{}) = .init; +var allocator = gpa.allocator(); + +extern fn __log(str_ptr: usize, str_len: usize) void; + +fn log(comptime level: std.log.Level, comptime scope: @TypeOf(.EnumLiteral), comptime fmt: []const u8, args: anytype) void { + if (comptime builtin.target.cpu.arch != .wasm32) { + std.log.defaultLog(level, scope, fmt, args); + return; + } + + const formatted = std.fmt.allocPrint(allocator, fmt, args) catch return; + __log(@intFromPtr(formatted.ptr), formatted.len); + allocator.free(formatted); +} + +pub const std_options: std.Options = .{ .logFn = log }; export fn malloc(byte_count: usize) ?*u8 { const ptr = allocator.alloc(u8, byte_count) catch return null; @@ -26,56 +43,51 @@ fn bytesToHex(bytes: []const u8, buf: []u8) void { } } -export fn validate_leading_zeroes_challenge(challenge_ptr: [*]u8, challenge_len: usize, nonce_ptr: [*]u8, nonce_len: usize, difficulty: u32) i32 { - const challenge_slice = challenge_ptr[0..challenge_len]; - const nonce_slice = nonce_ptr[0..nonce_len]; +// challenge_ptr should look like s.. +// solution_ptr should look like s. +export fn validate(challenge_ptr: [*]u8, challenge_len: usize, solution_ptr: [*]u8, solution_len: usize) bool { + const challenge_buf = challenge_ptr[0..challenge_len]; + const solution_buf = solution_ptr[0..solution_len]; - if (difficulty < 1 or difficulty > 64) { - return -1; + std.log.info("Validate called with challenge {s} and solution {s}\n", .{ challenge_buf, solution_buf }); + + const challenge = kCTF.decode(allocator, challenge_buf) catch return false; + std.log.info("decoded challenge {any}\n", .{challenge}); + const solution = kCTF.decode(allocator, solution_buf) catch return false; + defer { + challenge.destroy(allocator); + solution.destroy(allocator); } - var target_prefix_buffer: [64]u8 = @splat('0'); - const target_prefix = target_prefix_buffer[0..difficulty]; + std.log.info("decoded challenge and solution\n", .{}); - const hash_hex_slice = hasher.hash(allocator, challenge_slice, nonce_slice) catch return -2; + const is_valid = kCTF.check(allocator, challenge, solution) catch return false; - if (!std.mem.startsWith(u8, hash_hex_slice, target_prefix)) { - return -3; + return is_valid; +} + +pub fn main() anyerror!void { + if (comptime builtin.cpu.arch == .wasm32) return; + + const args = try std.process.argsAlloc(allocator); + if (args.len < 3) { + std.log.err("Usage: zig run src/validator.zig ", .{}); + return; } - return 0; + const challenge = try kCTF.decode(allocator, args[1]); + defer challenge.destroy(allocator); + + const solution = try kCTF.decode(allocator, args[2]); + defer solution.destroy(allocator); + + std.log.info("Challenge: {any}\n", .{challenge}); + std.log.info("Solution: {any}\n", .{solution}); + + const is_valid = kCTF.check(allocator, challenge, solution) catch |err| { + std.log.info("Error checking challenge: {s}\n", .{@errorName(err)}); + return; + }; + + std.log.info("Is valid: {}\n", .{is_valid}); } - -export fn validate_target_number_challenge(target_ptr: [*]u8, target_len: usize, nonce_ptr: [*]u8, nonce_len: usize, salt_ptr: [*]u8, salt_len: usize) i32 { - const target_slice = target_ptr[0..target_len]; - const salt_slice = salt_ptr[0..salt_len]; - const nonce_slice = nonce_ptr[0..nonce_len]; - - const hash_hex_slice = hasher.hash(allocator, salt_slice, nonce_slice) catch return -2; - - if (!std.mem.eql(u8, target_slice, hash_hex_slice)) { - return -3; - } - - return 0; -} - -export fn hash(challenge_ptr: [*]u8, challenge_len: usize, nonce_ptr: [*]u8, nonce_len: usize) u64 { - const challenge = challenge_ptr[0..challenge_len]; - const nonce = nonce_ptr[0..nonce_len]; - const hash_slice = hasher.hash(allocator, challenge, nonce) catch return 0; - - // bs to get the compiler to not whine about hash_slice.len being a u5 annd thus cannot be shifted by 32 - var ret: u64 = hash_slice.len; - ret <<= 32; - ret |= @intFromPtr(hash_slice.ptr); - return ret; -} - -// pub fn main() void { -// const challenge = "4d7220e22a1ea588fea60000ab8874194e4c6ffd71077adbae915826c73dbf48"; -// const nonce = "4302"; -// const difficulty = 3; - -// std.log.info("{d}", .{validate_challenge(@constCast(challenge[0..].ptr), challenge.len, @constCast(nonce[0..].ptr), nonce.len, difficulty)}); -// }