wrap submitBlockSignatures call

Author: cboh4

go-cosmwasm/api/lib.go

@@ -48,6 +48,31 @@ func HealthCheck() ([]byte, error) {
 }
 
 func SubmitBlockSignatures(header []byte, commit []byte, txs []byte, encRandom []byte /* valSet []byte, nextValSet []byte */) ([]byte, []byte, error) {
+	recorder := GetRecorder()
+
+	// Create combined input for recording/replay
+	input := make([]byte, 0, len(header)+len(commit)+len(txs)+len(encRandom))
+	input = append(input, header...)
+	input = append(input, commit...)
+	input = append(input, txs...)
+	input = append(input, encRandom...)
+
+	// In replay mode, return recorded data
+	if recorder.IsReplayMode() {
+		if output, err, found := recorder.Replay("SubmitBlockSignatures", input); found {
+			if err != nil {
+				return nil, nil, err
+			}
+			// Output is 64 bytes: [32 bytes random][32 bytes validator_set_evidence]
+			if len(output) == 64 {
+				return output[:32], output[32:], nil
+			}
+			return nil, nil, fmt.Errorf("SubmitBlockSignatures: invalid recorded data (expected 64 bytes, got %d)", len(output))
+		}
+		return nil, nil, fmt.Errorf("SubmitBlockSignatures: no recorded data found (replay mode)")
+	}
+
+	// SGX mode: call the actual enclave
 	errmsg := C.Buffer{}
 	spidSlice := sendSlice(header)
 	defer freeAfterSend(spidSlice)
@@ -62,7 +87,19 @@ func SubmitBlockSignatures(header []byte, commit []byte, txs []byte, encRandom [
 	if err != nil {
 		return nil, nil, errorWithMessage(err, errmsg)
 	}
-	return receiveVector(res.buf1), receiveVector(res.buf2), nil
+
+	buf1 := receiveVector(res.buf1)
+	buf2 := receiveVector(res.buf2)
+
+	// Record the result - 64 bytes: [32 bytes buf1][32 bytes buf2]
+	output := make([]byte, 64)
+	copy(output[:32], buf1)
+	copy(output[32:], buf2)
+	if recordErr := recorder.Record("SubmitBlockSignatures", input, output, nil); recordErr != nil {
+		fmt.Printf("[SubmitBlockSignatures] Warning: failed to record: %v\n", recordErr)
+	}
+
+	return buf1, buf2, nil
 }
 
 func SubmitValidatorSetEvidence(evidence []byte) error {

go.mod

@@ -8,7 +8,7 @@ replace (
 	cosmossdk.io/api => github.com/scrtlabs/cosmos-sdk-api v0.7.6-secret.0
 	cosmossdk.io/store => github.com/scrtlabs/cosmos-sdk-store v1.1.1-secret.1
 	cosmossdk.io/x/tx => github.com/scrtlabs/cosmos-sdk-x-tx v0.13.7-secret.0
-	github.com/cometbft/cometbft => github.com/scrtlabs/tendermint v0.38.19-secret.1-non-sgx.1
+	github.com/cometbft/cometbft => github.com/scrtlabs/tendermint v0.38.19-secret.1-non-sgx.2
 	github.com/cosmos/cosmos-sdk => github.com/scrtlabs/cosmos-sdk v0.50.14-secret.7
 	github.com/cosmos/iavl => github.com/scrtlabs/iavl v1.2.2-secret.0
 	github.com/syndtr/goleveldb => github.com/syndtr/goleveldb v1.0.1-0.20210819022825-2ae1ddf74ef7

go.sum

@@ -1619,8 +1619,8 @@ github.com/scrtlabs/cosmos-sdk-x-tx v0.13.7-secret.0 h1:i3k5706sDHKhaCvzokB+n33/
 github.com/scrtlabs/cosmos-sdk-x-tx v0.13.7-secret.0/go.mod h1:V6DImnwJMTq5qFjeGWpXNiT/fjgE4HtmclRmTqRVM3w=
 github.com/scrtlabs/iavl v1.2.2-secret.0 h1:P96PL1Lf8OBSW9pMrlaRxhceZ4z9Hc7jk12g9ShWeHw=
 github.com/scrtlabs/iavl v1.2.2-secret.0/go.mod h1:GiM43q0pB+uG53mLxLDzimxM9l/5N9UuSY3/D0huuVw=
-github.com/scrtlabs/tendermint v0.38.19-secret.1-non-sgx.1 h1:l4kcsJdGq20NctF4VMYq2mQkV8k31Vcy7/U4Ud2TuqI=
-github.com/scrtlabs/tendermint v0.38.19-secret.1-non-sgx.1/go.mod h1:I8kyAQPjMco6jjCshUSMYFoY6TybsbADGBmGcjWrgKM=
+github.com/scrtlabs/tendermint v0.38.19-secret.1-non-sgx.2 h1:StYLcnDnqE9GB0LxHyh1UIMcy0XuJiFUNV2W5MG/cRQ=
+github.com/scrtlabs/tendermint v0.38.19-secret.1-non-sgx.2/go.mod h1:I8kyAQPjMco6jjCshUSMYFoY6TybsbADGBmGcjWrgKM=
 github.com/scrtlabs/tm-secret-enclave v1.13.1 h1:0mXcBdoWyqEGhQEdbXMjSuTi9LKKMld2BqEj0eNpoxU=
 github.com/scrtlabs/tm-secret-enclave v1.13.1/go.mod h1:nxZQtzzAqBNBLOEXSv4cKlUnVA4vRmHOn6ujr3kxVME=
 github.com/sean-/seed v0.0.0-20170313163322-e2103e2c3529/go.mod h1:DxrIzT+xaE7yg65j358z/aeFdxmN0P9QXhEzd20vsDc=

x/compute/module.go

@@ -157,10 +157,6 @@ func (am AppModule) ExportGenesis(ctx sdk.Context, cdc codec.JSONCodec) json.Raw
 
 // BeginBlock returns the begin blocker for the compute module.
 func (am AppModule) BeginBlock(c context.Context) error {
-	if api.GetRecorder().IsReplayMode() {
-		return nil
-	}
-
 	// Note: as of tendermint v0.38.0 block begin request info is no longer available
 	ctx := c.(sdk.Context)
 	block_header := ctx.BlockHeader()