fix(ledger): reject posting sets that overflow the zero-sum check

An adversarial posting set of two MaxInt64 legs plus one of 2 wraps to
zero in int64 arithmetic, so the balance check passed and the ledger
minted 18 quintillion millisatoshis from nothing. The sum is now
accumulated in big.Int, per-account balance arithmetic is checked for
wraparound, and the audit totals parse through big.Int so a corrupt
ledger reports a clear error rather than failing to scan.

Adds room package tests (0% -> covered) and ledger edge cases.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
drjones
2026-08-05 15:43:21 +00:00
parent 2a2a1db8de
commit f2c02e2bde
4 changed files with 929 additions and 24 deletions

232
pkg/ledger/edge_test.go Normal file
View File

@@ -0,0 +1,232 @@
package ledger_test
import (
"context"
"math"
"sync"
"testing"
"github.com/drjones/quantum-arcade/pkg/ledger"
)
// These tests attack the ledger with extreme values. Money code fails at the
// boundaries, so the boundaries are where it should be hit hardest.
func TestHugeBalanceIsExact(t *testing.T) {
l := ledger.New(testPool(t))
ctx := context.Background()
p, err := l.EnsurePlayer(ctx, uniqueKey(t, "whale"))
if err != nil {
t.Fatal(err)
}
// 21 million BTC in millisatoshis is the largest amount that can ever
// exist: 2.1e15. It must round-trip exactly, with no float contamination.
const allTheBitcoin int64 = 21_000_000 * 100_000_000 * 1000
if _, err := l.Deposit(ctx, p, allTheBitcoin); err != nil {
t.Fatalf("depositing the entire supply: %v", err)
}
bal, err := l.Balance(ctx, p)
if err != nil {
t.Fatal(err)
}
if bal != allTheBitcoin {
t.Fatalf("balance = %d, want %d (off by %d)", bal, allTheBitcoin, bal-allTheBitcoin)
}
}
// A deposit that would overflow int64 must be refused, not wrap around.
func TestOverflowingDepositIsRejected(t *testing.T) {
l := ledger.New(testPool(t))
ctx := context.Background()
p, _ := l.EnsurePlayer(ctx, uniqueKey(t, "overflow"))
// A MaxInt64 deposit must be refused: it would underflow the bridge, whose
// balance is already negative by everything owed to players.
if _, err := l.Deposit(ctx, p, math.MaxInt64); err == nil {
t.Fatal("a MaxInt64 deposit was accepted")
}
if bal, _ := l.Balance(ctx, p); bal != 0 {
t.Fatalf("rejected deposit still moved the balance to %d", bal)
}
}
// Credit overflow is structurally unreachable, and that is a stronger
// guarantee than the runtime guard.
//
// Every millisatoshi inside the system was issued by debiting the bridge, and
// the bridge cannot pass MinInt64. So the sum of all non-bridge balances is
// bounded by MaxInt64, and no individual account can be pushed past it by any
// sequence of balanced transactions. The guard in Post remains as defence in
// depth against a future issuance path that does not go through the bridge.
func TestIssuanceIsBoundedByTheBridge(t *testing.T) {
l := ledger.New(testPool(t))
ctx := context.Background()
p, _ := l.EnsurePlayer(ctx, uniqueKey(t, "bounded"))
// Attempting to issue more than the bridge can back must fail.
if _, err := l.Deposit(ctx, p, math.MaxInt64); err == nil {
t.Fatal("issued more than the bridge can back")
}
// And whatever has been issued must still fit in an int64, which is what
// makes every downstream balance arithmetic safe.
issued, err := l.TotalIssued(ctx)
if err != nil {
t.Fatalf("total issuance is no longer representable: %v", err)
}
if issued < 0 {
t.Fatalf("total issued is negative: %d", issued)
}
}
// Postings that individually fit but collectively overflow the zero-sum check.
func TestOverflowingPostingSetIsRejected(t *testing.T) {
l := ledger.New(testPool(t))
ctx := context.Background()
a, _ := l.EnsurePlayer(ctx, uniqueKey(t, "a"))
b, _ := l.EnsurePlayer(ctx, uniqueKey(t, "b"))
c, _ := l.EnsurePlayer(ctx, uniqueKey(t, "c"))
// These sum to zero only if you ignore wraparound.
_, err := l.Post(ctx, "attack", nil, []ledger.Posting{
{AccountID: a, AmountMsat: math.MaxInt64},
{AccountID: b, AmountMsat: math.MaxInt64},
{AccountID: c, AmountMsat: 2},
})
if err == nil {
t.Fatal("a posting set that overflows int64 was accepted")
}
}
func TestSmallestPossibleAmount(t *testing.T) {
l := ledger.New(testPool(t))
ctx := context.Background()
p, _ := l.EnsurePlayer(ctx, uniqueKey(t, "dust"))
if _, err := l.Deposit(ctx, p, 1); err != nil {
t.Fatalf("one millisatoshi rejected: %v", err)
}
if bal, _ := l.Balance(ctx, p); bal != 1 {
t.Fatalf("balance = %d, want 1", bal)
}
// Spending exactly the balance must leave zero, not fail.
q, _ := l.EnsurePlayer(ctx, uniqueKey(t, "dust2"))
if _, err := l.Transfer(ctx, p, q, 1); err != nil {
t.Fatalf("spending the exact balance failed: %v", err)
}
if bal, _ := l.Balance(ctx, p); bal != 0 {
t.Fatalf("balance = %d after spending everything, want 0", bal)
}
}
// Spending one millisatoshi more than you hold must fail, at every scale.
func TestOffByOneOverdraftAtEveryScale(t *testing.T) {
l := ledger.New(testPool(t))
ctx := context.Background()
for _, amount := range []int64{1, 1000, 1_000_000, 100_000_000_000} {
p, _ := l.EnsurePlayer(ctx, uniqueKey(t, "scale"+itoa(amount)))
q, _ := l.EnsurePlayer(ctx, uniqueKey(t, "scaledst"+itoa(amount)))
if _, err := l.Deposit(ctx, p, amount); err != nil {
t.Fatal(err)
}
if _, err := l.Transfer(ctx, p, q, amount+1); err == nil {
t.Fatalf("overdraft by 1 accepted at scale %d", amount)
}
if _, err := l.Transfer(ctx, p, q, amount); err != nil {
t.Fatalf("exact-balance transfer rejected at scale %d: %v", amount, err)
}
}
}
// Hammer one account from many goroutines and confirm not a single
// millisatoshi is created or lost.
func TestHighContentionConservesExactly(t *testing.T) {
l := ledger.New(testPool(t))
ctx := context.Background()
hub, _ := l.EnsurePlayer(ctx, uniqueKey(t, "hub"))
if _, err := l.Deposit(ctx, hub, 1_000_000); err != nil {
t.Fatal(err)
}
const workers = 16
spokes := make([]int64, workers)
for i := range spokes {
spokes[i], _ = l.EnsurePlayer(ctx, uniqueKey(t, "spoke"+itoa(int64(i))))
}
var wg sync.WaitGroup
for i := 0; i < workers; i++ {
wg.Add(1)
go func(i int) {
defer wg.Done()
for j := 0; j < 20; j++ {
// Push out and pull back; net zero if nothing is lost.
if _, err := l.Transfer(ctx, hub, spokes[i], 137); err == nil {
_, _ = l.Transfer(ctx, spokes[i], hub, 137)
}
}
}(i)
}
wg.Wait()
total := int64(0)
for _, id := range spokes {
bal, err := l.Balance(ctx, id)
if err != nil {
t.Fatal(err)
}
total += bal
}
hubBal, err := l.Balance(ctx, hub)
if err != nil {
t.Fatal(err)
}
if total+hubBal != 1_000_000 {
t.Fatalf("value changed under contention: %d, want 1000000", total+hubBal)
}
}
// Self-transfers must not mint money through double-counting the same account.
func TestSelfTransferDoesNotMint(t *testing.T) {
l := ledger.New(testPool(t))
ctx := context.Background()
p, _ := l.EnsurePlayer(ctx, uniqueKey(t, "self"))
if _, err := l.Deposit(ctx, p, 10_000); err != nil {
t.Fatal(err)
}
_, _ = l.Transfer(ctx, p, p, 5_000)
bal, err := l.Balance(ctx, p)
if err != nil {
t.Fatal(err)
}
if bal != 10_000 {
t.Fatalf("self-transfer changed balance to %d, want 10000", bal)
}
}
func itoa(v int64) string {
if v == 0 {
return "0"
}
neg := v < 0
if neg {
v = -v
}
var buf [24]byte
i := len(buf)
for v > 0 {
i--
buf[i] = byte('0' + v%10)
v /= 10
}
if neg {
i--
buf[i] = '-'
}
return string(buf[i:])
}