From 31b4ffccfc1cdb622f9601e00e8248af59b23a6b Mon Sep 17 00:00:00 2001
From: realrasengan <andrew@joseon.com>
Date: Thu, 2 Apr 2026 23:40:59 -0700
Subject: [PATCH] cap state map to 16 entries to prevent OOM in WASM

---
 cmd/mosh-wasm/state.go | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/cmd/mosh-wasm/state.go b/cmd/mosh-wasm/state.go
index dc907b3..2e5eaf9 100644
--- a/cmd/mosh-wasm/state.go
+++ b/cmd/mosh-wasm/state.go
@@ -107,6 +107,21 @@ func (st *stateTracker) applyDiff(diff []byte, oldNum, newNum, throwawayNum uint
 		}
 	}
 
+	// Hard cap: keep at most 16 states to prevent unbounded memory growth.
+	if len(st.states) > 16 {
+		minKeep := st.latestState
+		if minKeep > 16 {
+			minKeep -= 16
+		} else {
+			minKeep = 0
+		}
+		for n := range st.states {
+			if n < minKeep {
+				delete(st.states, n)
+			}
+		}
+	}
+
 	// Confirm predictions against server state.
 	latest := st.states[st.latestState]
 	if latest != nil {