fix: invalidate WASM cache on upload + wire cache invalidator to RPC

- Always invalidate WASM cache on any ChallengeUpdate (including wasm_upload)
- Add wasm_cache_invalidator callback to RpcHandler
- Invalidate cache immediately in RPC upload handler (self-broadcast not received)
- Fixes stale WASM module after sudo upload

Author: factorydroid

bins/validator-node/src/main.rs

@@ -862,6 +862,15 @@ async fn main() -> Result<()> {
             rpc_server.rpc_handler().set_route_handler(handler);
         }
 
+        // Wire WASM cache invalidation callback for uploads
+        if let Some(ref executor) = wasm_executor {
+            let exec = Arc::clone(executor);
+            let invalidator: Arc<dyn Fn(&str) + Send + Sync> = Arc::new(move |challenge_id: &str| {
+                exec.invalidate_cache(challenge_id);
+            });
+            *rpc_server.rpc_handler().wasm_cache_invalidator.write() = Some(invalidator);
+        }
+
         // Wire real-time weight computation for subnet_getWeights RPC
         {
             let uid_map = Arc::clone(&shared_uid_map);
@@ -3064,11 +3073,9 @@ async fn handle_network_event(
                         }
                     }
 
-                    // Invalidate WASM cache (skip for wasm_upload - module was just compiled fresh)
-                    if update.update_type != "wasm_upload" {
-                        if let Some(ref executor) = wasm_executor_ref {
-                            executor.invalidate_cache(&challenge_id_str);
-                        }
+                    // Always invalidate WASM cache on any update (including wasm_upload)
+                    if let Some(ref executor) = wasm_executor_ref {
+                        executor.invalidate_cache(&challenge_id_str);
                     }
                 } else {
                     warn!(

crates/rpc-server/src/jsonrpc.rs

@@ -194,6 +194,8 @@ pub struct RpcHandler {
     pub keypair: Arc<RwLock<Option<platform_core::Keypair>>>,
     /// Real-time weight computation handler
     pub get_weights_handler: Arc<RwLock<Option<GetWeightsHandler>>>,
+    /// Callback to invalidate WASM cache for a challenge (called on wasm_upload)
+    pub wasm_cache_invalidator: Arc<RwLock<Option<Arc<dyn Fn(&str) + Send + Sync>>>>,
 }
 
 impl RpcHandler {
@@ -209,6 +211,7 @@ impl RpcHandler {
             broadcast_tx: Arc::new(RwLock::new(None)),
             keypair: Arc::new(RwLock::new(None)),
             get_weights_handler: Arc::new(RwLock::new(None)),
+            wasm_cache_invalidator: Arc::new(RwLock::new(None)),
         }
     }
 

crates/rpc-server/src/server.rs

@@ -1238,6 +1238,14 @@ async fn sudo_challenge_handler(
             "Challenge update broadcast initiated"
         );
 
+        // Invalidate local WASM cache so new module is loaded on next request
+        if request.action == "wasm_upload" {
+            if let Some(ref invalidator) = *handler.wasm_cache_invalidator.read() {
+                invalidator(&request.challenge_id);
+                info!(challenge_id = %request.challenge_id, "WASM cache invalidated after upload");
+            }
+        }
+
         (
             StatusCode::OK,
             Json(serde_json::json!({