runtime/parachain: add ValidationCodeByHash and copy tests to wazero (

#3427) Co-authored-by: Kishan Sagathiya <[email protected]>
ChainSafe · Jun 20, 2024 · 9697a6e · 9697a6e
1 parent c62759c
commit 9697a6e
Show file tree

Hide file tree

Showing 13 changed files with 545 additions and 0 deletions.
diff --git a/dot/core/mock_runtime_instance_test.go b/dot/core/mock_runtime_instance_test.go
diff --git a/dot/state/mocks_runtime_test.go b/dot/state/mocks_runtime_test.go
diff --git a/dot/sync/mock_runtime_test.go b/dot/sync/mock_runtime_test.go
diff --git a/lib/babe/mocks/runtime.go b/lib/babe/mocks/runtime.go
diff --git a/lib/blocktree/mocks_test.go b/lib/blocktree/mocks_test.go
diff --git a/lib/grandpa/mocks_runtime_test.go b/lib/grandpa/mocks_runtime_test.go
diff --git a/lib/runtime/constants.go b/lib/runtime/constants.go
@@ -78,6 +78,8 @@ const (
 	ParachainHostPersistedValidationData = "ParachainHost_persisted_validation_data"
 	// ParachainHostValidationCode returns parachain host's validation code
 	ParachainHostValidationCode = "ParachainHost_validation_code"
+	// ParachainHostValidationCodeByHash returns parachain host's validation code by hash
+	ParachainHostValidationCodeByHash = "ParachainHost_validation_code_by_hash"
 	// ParachainHostValidators is the runtime API call ParachainHost_validators
 	ParachainHostValidators = "ParachainHost_validators"
 	// ParachainHostValidatorGroups is the runtime API call ParachainHost_validator_groups

diff --git a/lib/runtime/interface.go b/lib/runtime/interface.go
@@ -55,6 +55,7 @@ type Instance interface {
 	) (*parachaintypes.PersistedValidationData, error)
 	ParachainHostValidationCode(parachaidID uint32, assumption parachaintypes.OccupiedCoreAssumption,
 	) (*parachaintypes.ValidationCode, error)
+	ParachainHostValidationCodeByHash(validationCodeHash common.Hash) (*parachaintypes.ValidationCode, error)
 	ParachainHostValidators() ([]parachaintypes.ValidatorID, error)
 	ParachainHostValidatorGroups() (*parachaintypes.ValidatorGroups, error)
 	ParachainHostAvailabilityCores() ([]parachaintypes.CoreState, error)

diff --git a/lib/runtime/mocks/mocks.go b/lib/runtime/mocks/mocks.go
diff --git a/lib/runtime/wazero/instance.go b/lib/runtime/wazero/instance.go
@@ -1336,6 +1336,30 @@ func (in *Instance) ParachainHostSessionInfo(sessionIndex parachaintypes.Session
 	return sessionInfo, nil
 }
 
+// ParachainHostValidationCodeByHash returns validation code for the given hash.
+func (in *Instance) ParachainHostValidationCodeByHash(validationCodeHash common.Hash) (
+	*parachaintypes.ValidationCode, error) {
+	buffer := bytes.NewBuffer(nil)
+	encoder := scale.NewEncoder(buffer)
+	err := encoder.Encode(validationCodeHash)
+	if err != nil {
+		return nil, fmt.Errorf("encoding validation code hash: %w", err)
+	}
+
+	encodedValidationCodeHash, err := in.Exec(runtime.ParachainHostValidationCodeByHash, buffer.Bytes())
+	if err != nil {
+		return nil, err
+	}
+
+	var validationCode *parachaintypes.ValidationCode
+	err = scale.Unmarshal(encodedValidationCodeHash, &validationCode)
+	if err != nil {
+		return nil, fmt.Errorf("unmarshalling validation code: %w", err)
+	}
+
+	return validationCode, nil
+}
+
 func (*Instance) RandomSeed() {
 	panic("unimplemented")
 }