cr feedback, fixed storageAppend

dot/core/service_test.go

@@ -289,6 +289,7 @@ func TestHandleChainReorg_WithReorg_Transactions(t *testing.T) {
 	// create extrinsic
 	enc, err := scale.Marshal([]byte("nootwashere"))
 	require.NoError(t, err)
+	// we prefix with []byte{2} here since that's the enum index for the old IncludeDataExt extrinsic
 	tx := append([]byte{2}, enc...)
 
 	bhash := s.blockState.BestBlockHash()

lib/runtime/life/exports.go

@@ -37,12 +37,13 @@ func (in *Instance) Version() (runtime.Version, error) {
 	version := &runtime.VersionData{}
 	err = version.Decode(res)
 	// error comes from scale now, so do a string check
-	if strings.Contains(fmt.Sprintf("%s", err), "EOF") {
-		// TODO: kusama seems to use the legacy version format
-		lversion := &runtime.LegacyVersionData{}
-		err = lversion.Decode(res)
-		return lversion, err
-	} else if err != nil {
+	if err != nil {
+		if strings.Contains(err.Error(), "EOF") {
+			// TODO: kusama seems to use the legacy version format
+			lversion := &runtime.LegacyVersionData{}
+			err = lversion.Decode(res)
+			return lversion, err
+		}
 		return nil, err
 	}
 

lib/runtime/life/resolver.go

@@ -431,8 +431,13 @@ func storageAppend(storage runtime.Storage, key, valueToAppend []byte) error {
 			return nil
 		}
 
-		// append new item, pop off first byte now, since we're not using old scale.Decoder
-		valueRes = append(valueCurr[1:], valueToAppend...)
+		lengthBytes, err := scale.Marshal(currLength)
+		if err != nil {
+			return err
+		}
+		// append new item, pop off number of bytes required for length encoding,
+		// since we're not using old scale.Decoder
+		valueRes = append(valueCurr[len(lengthBytes):], valueToAppend...)
 
 		// increase length by 1
 		nextLength = big.NewInt(0).Add(currLength, big.NewInt(1))

lib/runtime/wasmer/exports.go

@@ -58,12 +58,13 @@ func (in *Instance) Version() (runtime.Version, error) {
 	version := &runtime.VersionData{}
 	err = version.Decode(res)
 	// error comes from scale now, so do a string check
-	if strings.Contains(fmt.Sprintf("%s", err), "EOF") {
-		// TODO: kusama seems to use the legacy version format
-		lversion := &runtime.LegacyVersionData{}
-		err = lversion.Decode(res)
-		return lversion, err
-	} else if err != nil {
+	if err != nil {
+		if strings.Contains(err.Error(), "EOF") {
+			// TODO: kusama seems to use the legacy version format
+			lversion := &runtime.LegacyVersionData{}
+			err = lversion.Decode(res)
+			return lversion, err
+		}
 		return nil, err
 	}
 

lib/runtime/wasmer/imports.go

@@ -1571,8 +1571,13 @@ func storageAppend(storage runtime.Storage, key, valueToAppend []byte) error {
 			return nil
 		}
 
-		// append new item, pop off first byte now, since we're not using old scale.Decoder
-		valueRes = append(valueCurr[1:], valueToAppend...)
+		lengthBytes, err := scale.Marshal(currLength)
+		if err != nil {
+			return err
+		}
+		// append new item, pop off number of bytes required for length encoding,
+		// since we're not using old scale.Decoder
+		valueRes = append(valueCurr[len(lengthBytes):], valueToAppend...)
 
 		// increase length by 1
 		nextLength = big.NewInt(0).Add(currLength, big.NewInt(1))