diff --git a/src/__tests__/protobuf.test.ts b/src/__tests__/protobuf.test.ts new file mode 100644 index 0000000..d3ad702 --- /dev/null +++ b/src/__tests__/protobuf.test.ts @@ -0,0 +1,25 @@ +import { describe, expect, it } from "vitest"; +import { encodeMessage } from "../protobuf.js"; + +const validSend = { + fromAddress: "11".repeat(20), + toAddress: "22".repeat(20), + amount: 1, +}; + +describe("encodeMessage", () => { + it("encodes a valid send amount", () => { + const { typeUrl, msgBytes } = encodeMessage("send", validSend); + + expect(typeUrl).toBe("type.googleapis.com/types.MessageSend"); + expect(msgBytes.length).toBeGreaterThan(0); + }); + + it("rejects send amounts that cannot be encoded as uint64 values", () => { + for (const amount of [-1, 1.5, Number.NaN]) { + expect(() => encodeMessage("send", { ...validSend, amount })).toThrow( + "amount must be a non-negative safe integer", + ); + } + }); +}); \ No newline at end of file diff --git a/src/keystore.ts b/src/keystore.ts index 193bde3..120ffc1 100644 --- a/src/keystore.ts +++ b/src/keystore.ts @@ -26,9 +26,12 @@ export function importFromGoKeystore( const curveType = detectPublicKeyCurve(entry.publicKey); const derivedAddress = deriveAddress(entry.publicKey, curveType); + // Throw instead of warn: a mismatch means the entry is tampered or corrupted. + // Silently importing the wrong address would cause funds to be sent to the + // wrong destination or transactions to be rejected by the network. if (derivedAddress.toLowerCase() !== entry.keyAddress.toLowerCase()) { - console.warn( - `Address mismatch: expected ${entry.keyAddress}, derived ${derivedAddress}` + throw new Error( + `Keystore integrity check failed: stored address ${entry.keyAddress} does not match address derived from public key ${derivedAddress}. The keystore entry may be corrupted or tampered.` ); } diff --git a/src/protobuf.ts b/src/protobuf.ts index c53cdee..4b505bb 100644 --- a/src/protobuf.ts +++ b/src/protobuf.ts @@ -1,6 +1,13 @@ import protobuf from "protobufjs"; import { hexToBytes } from "@noble/hashes/utils.js"; +function assertUint64Field(name: string, value: unknown): number { + if (typeof value !== "number" || !Number.isSafeInteger(value) || value < 0) { + throw new Error(`${name} must be a non-negative safe integer`); + } + return value; +} + function shouldOmit(value: any): boolean { if (value === undefined || value === null) return true; if (typeof value === "string" && value === "") return true; @@ -74,7 +81,7 @@ const MESSAGE_REGISTRY: Record< MsgSend.create({ from_address: hexToBytes(msg.fromAddress), to_address: hexToBytes(msg.toAddress), - amount: msg.amount, + amount: assertUint64Field("amount", msg.amount), }) ).finish(), }, diff --git a/src/provably-fair.ts b/src/provably-fair.ts index a60371d..f849c84 100644 --- a/src/provably-fair.ts +++ b/src/provably-fair.ts @@ -36,7 +36,15 @@ export function computeHMAC( /** * Computes a dice roll in the range [0, 9999] from the given seeds and nonce. - * Maps the first 4 bytes of HMAC-SHA256 to a uint32, then takes modulo 10000. + * + * Uses rejection sampling over successive 4-byte HMAC windows to eliminate + * modulo bias. A naive `uint32 % 10000` is biased because 2^32 (4 294 967 296) + * is not evenly divisible by 10000 — values 0–7295 appear once more often than + * values 7296–9999, giving the house a hidden systematic edge on those outcomes. + * + * Rejection threshold: 4 294 960 000 (= floor(2^32 / 10000) * 10000). + * Values at or above the threshold are discarded and the next 4-byte window is + * tried. In practice fewer than 2 iterations are needed on average. */ export function computeDiceRoll( serverSeed: string, @@ -44,9 +52,16 @@ export function computeDiceRoll( nonce: number ): number { const h = computeHMAC(serverSeed, clientSeed, nonce); - const view = new DataView(h.buffer, h.byteOffset, h.byteLength); - const raw = view.getUint32(0, false); // big-endian - return raw % 10000; + const RANGE = 10000; + const THRESHOLD = Math.floor(0x100000000 / RANGE) * RANGE; // 4_294_960_000 + for (let offset = 0; offset + 4 <= h.length; offset += 4) { + const view = new DataView(h.buffer, h.byteOffset + offset, 4); + const raw = view.getUint32(0, false); // big-endian + if (raw < THRESHOLD) return raw % RANGE; + } + // Fallback (astronomically unlikely): use modulo on last window + const view = new DataView(h.buffer, h.byteOffset, 4); + return view.getUint32(0, false) % RANGE; } /** @@ -63,18 +78,30 @@ export function verifyDiceRoll( /** * Computes the crash point for a rocket round. - * Uses HMAC-SHA256(serverSeed, nonce) with 3% house edge. + * Uses HMAC-SHA256(serverSeed, clientSeed:nonce) with 3% house edge. * Result is clamped to [1.01, 100.0]. + * + * @param clientSeed - Player-supplied seed that contributes entropy to the outcome. + * Must not be empty: passing "" removes the client's ability to independently + * influence and verify the result, breaking the provably-fair guarantee. */ export function computeCrashPoint( serverSeed: string, + clientSeed: string, nonce: number ): number { - const h = computeHMAC(serverSeed, "", nonce); + if (!clientSeed) { + throw new Error("clientSeed must not be empty — an empty client seed removes player entropy from the provably-fair computation"); + } + const h = computeHMAC(serverSeed, clientSeed, nonce); const hex8 = bytesToHex(h).slice(0, 8); const result = parseInt(hex8, 16) >>> 0; // unsigned 32-bit - if (result % 33 === 0) return 1.0; + // Instant-crash case: return the advertised floor (1.01), not 1.0. + // The docstring and UI both promise outcomes in [1.01, 100.0]; returning + // 1.0 violates the stated minimum and gives players a worse payout than + // the published house-edge formula implies. + if (result % 33 === 0) return 1.01; const houseEdge = 0.03; const e = 0x100000000; @@ -83,4 +110,4 @@ export function computeCrashPoint( if (crashPoint < 1.01) crashPoint = 1.01; if (crashPoint > 100.0) crashPoint = 100.0; return crashPoint; -} +} \ No newline at end of file