update smoldot

.gitmodules

@@ -1,3 +1,3 @@
 [submodule "vendor/smoldot"]
 	path = vendor/smoldot
-	url = https://github.com/ermalkaleci/smoldot.git
+	url = https://github.com/smol-dot/smoldot.git

executor/src/task.rs

@@ -67,6 +67,7 @@ pub struct TaskCall {
     mock_signature_host: bool,
     allow_unresolved_imports: bool,
     runtime_log_level: u32,
+    storage_proof_size: u64,
 }
 
 #[derive(Serialize, Deserialize, Debug)]
@@ -142,6 +143,8 @@ pub async fn run_task(task: TaskCall, js: crate::JsCallback) -> Result<TaskRespo
             storage_main_trie_changes,
             max_log_level: task.runtime_log_level,
             calculate_trie_changes: false,
+            storage_proof_size_behavior:
+                runtime_call::StorageProofSizeBehavior::ConstantReturnValue(task.storage_proof_size),
         });
 
         let mut vm = match vm {

packages/chopsticks/src/context.ts

@@ -47,6 +47,7 @@ export const setupContext = async (argv: Config, overrideParent = false) => {
     offchainWorker: argv['offchain-worker'],
     maxMemoryBlockCount: argv['max-memory-block-count'],
     processQueuedMessages: argv['process-queued-messages'],
+    storageProofSize: argv['storage-proof-size'],
   })
 
   // load block from db

packages/chopsticks/src/schema/index.ts

@@ -35,6 +35,7 @@ export const configSchema = z.object({
       description: 'Mock signature host so any signature starts with 0xdeadbeef and filled by 0xcd is considered valid',
     })
     .optional(),
+  'storage-proof-size': z.number().optional(),
   'max-memory-block-count': z.number().optional(),
   db: z.string({ description: 'Path to database' }).optional(),
   'wasm-override': z.string({ description: 'Path to wasm override' }).optional(),

packages/chopsticks/src/schema/options.test.ts

@@ -98,6 +98,12 @@ it('get yargs options from zod schema', () => {
         "description": "Runtime maximum log level [off = 0; error = 1; warn = 2; info = 3; debug = 4; trace = 5]",
         "type": "number",
       },
+      "storage-proof-size": {
+        "choices": undefined,
+        "demandOption": false,
+        "description": undefined,
+        "type": "number",
+      },
       "timestamp": {
         "choices": undefined,
         "demandOption": false,

packages/core/src/blockchain/block.ts

@@ -292,6 +292,7 @@ export class Block {
         mockSignatureHost: this.#chain.mockSignatureHost,
         allowUnresolvedImports: this.#chain.allowUnresolvedImports,
         runtimeLogLevel: this.#chain.runtimeLogLevel,
+        storageProofSize: this.#chain.storageProofSize,
       },
       taskHandler(this),
     )

packages/core/src/blockchain/index.ts

@@ -49,6 +49,11 @@ export interface Options {
   maxMemoryBlockCount?: number
   /** Whether to process queued messages */
   processQueuedMessages?: boolean
+  /**
+   * Storage Proof Size passed to runtime.
+   * See <https://github.com/smol-dot/smoldot/blob/26dedea00eafa7073ca1d4678277272d807b3d89/lib/src/executor/host.rs#L248>
+   */
+  storageProofSize?: number
 }
 
 /**
@@ -86,6 +91,8 @@ export class Blockchain {
   #runtimeLogLevel: number
   /** Polkadot.js custom types registration. */
   readonly registeredTypes: RegisteredTypes
+  /** Storage Proof Size passed to runtime. */
+  readonly storageProofSize?: number
 
   readonly #txpool: TxPool
   readonly #inherentProviders: InherentProvider[]
@@ -134,6 +141,7 @@ export class Blockchain {
     header,
     mockSignatureHost = false,
     allowUnresolvedImports = false,
+    storageProofSize = 0,
     runtimeLogLevel = 0,
     registeredTypes = {},
     offchainWorker = false,
@@ -144,6 +152,7 @@ export class Blockchain {
     this.db = db
     this.mockSignatureHost = mockSignatureHost
     this.allowUnresolvedImports = allowUnresolvedImports
+    this.storageProofSize = storageProofSize
     this.#runtimeLogLevel = runtimeLogLevel
     this.registeredTypes = registeredTypes
 

packages/core/src/setup.ts

@@ -25,6 +25,7 @@ export type SetupOptions = {
   offchainWorker?: boolean
   maxMemoryBlockCount?: number
   processQueuedMessages?: boolean
+  storageProofSize?: number
 }
 
 export const processOptions = async (options: SetupOptions) => {
@@ -92,6 +93,7 @@ export const setup = async (options: SetupOptions) => {
     offchainWorker: opts.offchainWorker,
     maxMemoryBlockCount: opts.maxMemoryBlockCount,
     processQueuedMessages: opts.processQueuedMessages,
+    storageProofSize: opts.storageProofSize,
   })
 
   if (opts.genesis) {

packages/core/src/wasm-executor/index.ts

@@ -22,6 +22,15 @@ export type RuntimeVersion = {
   stateVersion: number
 }
 
+export type TaskCall = {
+  wasm: HexString
+  calls: [string, HexString[]][]
+  mockSignatureHost: boolean
+  allowUnresolvedImports: boolean
+  runtimeLogLevel: number
+  storageProofSize?: number
+}
+
 export type RuntimeLog = {
   message: string
   level?: number
@@ -113,16 +122,10 @@ export const createProof = async (nodes: HexString[], updates: [HexString, HexSt
   return { trieRootHash, nodes: newNodes }
 }
 
-export const runTask = async (
-  task: {
-    wasm: HexString
-    calls: [string, HexString[]][]
-    mockSignatureHost: boolean
-    allowUnresolvedImports: boolean
-    runtimeLogLevel: number
-  },
-  callback: JsCallback = emptyTaskHandler,
-) => {
+export const runTask = async (task: TaskCall, callback: JsCallback = emptyTaskHandler) => {
+  if (!task.storageProofSize) {
+    task.storageProofSize = 0
+  }
   const worker = await getWorker()
   logger.trace(truncate(task), 'taskRun')
   const response = await worker.remote.runTask(task, Comlink.proxy(callback))

packages/utils/src/index.ts

@@ -32,6 +32,7 @@ export type SetupOption = {
   runtimeLogLevel?: number
   allowUnresolvedImports?: boolean
   processQueuedMessages?: boolean
+  storageProofSize?: number
 }
 
 export type SetupConfig = Config & {
@@ -51,6 +52,7 @@ export const createConfig = ({
   runtimeLogLevel,
   allowUnresolvedImports,
   processQueuedMessages,
+  storageProofSize,
 }: SetupOption): SetupConfig => {
   // random port if not specified
   port = port ?? Math.floor(Math.random() * 10000) + 10000
@@ -68,6 +70,7 @@ export const createConfig = ({
     resume: resume ?? false,
     'allow-unresolved-imports': allowUnresolvedImports,
     'process-queued-messages': processQueuedMessages,
+    'storage-proof-size': storageProofSize,
   }
   return config
 }