Contract Update Validator

runtime/contract_update_test.go

@@ -716,6 +716,9 @@ func TestRuntimeLegacyContractUpdate(t *testing.T) {
 		OnGetAccountContractCode: func(location common.AddressLocation) (code []byte, err error) {
 			return accountCodes[location], nil
 		},
+		OnGetAccountContractNames: func(_ Address) ([]string, error) {
+			return []string{"Foo"}, nil
+		},
 		OnUpdateAccountContractCode: func(location common.AddressLocation, code []byte) error {
 			accountCodes[location] = code
 			return nil

runtime/contract_update_validation_test.go

@@ -2561,7 +2561,6 @@ func TestRuntimeContractUpdateConformanceChanges(t *testing.T) {
 		RequireError(t, err)
 
 		cause := getSingleContractUpdateErrorCause(t, err, "Test")
-
 		assertConformanceMismatchError(t, cause, "Foo")
 	})
 

runtime/interpreter/interpreter.go

@@ -4577,6 +4577,24 @@
 	return subInterpreter.Program.Elaboration
 }
 
+func (interpreter *Interpreter) AllElaborations() (elaborations map[common.Location]*sema.Elaboration) {
+
+	elaborations = map[common.Location]*sema.Elaboration{}
+
+	// Ensure the program for this location is loaded,
+	// so its checker is available
+
+	for location := range interpreter.SharedState.allInterpreters { //nolint:maprange
+		subInterpreter := interpreter.EnsureLoaded(location)
+		if subInterpreter == nil || subInterpreter.Program == nil {
+			return nil
+		}
+		elaborations[location] = subInterpreter.Program.Elaboration
+	}
+
+	return
+}
+
 // GetContractComposite gets the composite value of the contract at the address location.
 func (interpreter *Interpreter) GetContractComposite(contractLocation common.AddressLocation) (*CompositeValue, error) {
 	contractGlobal := interpreter.Globals.Get(contractLocation.Name)

runtime/stdlib/account.go

@@ -1586,6 +1586,8 @@ func changeAccountContracts(
 
 	// Validate the contract update
 
+	inter := invocation.Interpreter
+
 	if isUpdate {
 		oldCode, err := handler.GetAccountContractCode(location)
 		handleContractUpdateError(err)
@@ -1618,11 +1620,13 @@ func changeAccountContracts(
 
 		var validator UpdateValidator
 		if legacyContractUpgrade {
-			validator = NewLegacyContractUpdateValidator(
+			validator = NewCadenceV042ToV1ContractUpdateValidator(
 				location,
 				contractName,
+				handler,
 				oldProgram,
 				program.Program,
+				inter.AllElaborations(),
 			)
 		} else {
 			validator = NewContractUpdateValidator(
@@ -1637,8 +1641,6 @@ func changeAccountContracts(
 		handleContractUpdateError(err)
 	}
 
-	inter := invocation.Interpreter
-
 	err = updateAccountContractCode(
 		handler,
 		location,