channel: store virtual packet

polymerdao · Apr 30, 2024 · f204504 · f204504
1 parent a485548
commit f204504
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/modules/core/04-channel/keeper/keeper.go b/modules/core/04-channel/keeper/keeper.go
@@ -242,9 +242,20 @@ func (k Keeper) deletePacketCommitment(ctx sdk.Context, portID, channelID string
 	store.Delete(host.PacketCommitmentKey(portID, channelID, sequence))
 }
 
-func (k Keeper) SetVirtualPacket(ctx sdk.Context, portID, channelID string, sequence uint64, packet types.Packet) {
-	store := ctx.KVStore(k.storeKey)
-	bz := k.cdc.MustMarshal(&packet)
+func (k Keeper) SetVirtualPacket(ctx sdk.Context, portID, channelID string, sequence uint64, packet exported.PacketI) {
+	timeoutHeight := packet.GetTimeoutHeight()
+	p := types.NewPacket(
+		packet.GetData(),
+		packet.GetSequence(),
+		packet.GetSourcePort(),
+		packet.GetSourceChannel(),
+		packet.GetDestPort(),
+		packet.GetDestChannel(),
+		clienttypes.NewHeight(timeoutHeight.GetRevisionNumber(), timeoutHeight.GetRevisionHeight()),
+		packet.GetTimeoutTimestamp(),
+	)
+	store := ctx.KVStore(k.storeKey)
+	bz := k.cdc.MustMarshal(&p)
 	store.Set(host.VirtualPacketKey(portID, channelID, sequence), bz)
 }
 

diff --git a/modules/core/04-channel/keeper/packet.go b/modules/core/04-channel/keeper/packet.go
@@ -283,6 +283,7 @@ func (k Keeper) RecvPacket(
 
 	// emit an event that the relayer can query for
 	EmitRecvPacketEvent(ctx, packet, channel)
+	k.SetVirtualPacket(ctx, packet.GetDestPort(), packet.GetDestChannel(), packet.GetSequence(), packet)
 
 	return nil
 }