Skip to content

Commit

Permalink
Start using IBC ports to determine message formats.
Browse files Browse the repository at this point in the history
  • Loading branch information
murisi committed Jun 19, 2024
1 parent 8de2ec0 commit 439e5a9
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 15 deletions.
21 changes: 10 additions & 11 deletions crates/namada/src/ledger/native_vp/masp.rs
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,6 @@ use token::storage_key::{
use token::Amount;

use crate::address::{InternalAddress, IBC, MASP};
use crate::ibc::{MsgRecvPacket, MsgTransfer};
use crate::ledger::ibc::storage;
use crate::ledger::ibc::storage::{
ibc_trace_key, ibc_trace_key_prefix, is_ibc_commitment_key,
Expand All @@ -50,7 +49,7 @@ use crate::ledger::ibc::storage::{
use crate::ledger::native_vp;
use crate::ledger::native_vp::ibc::context::VpValidationContext;
use crate::ledger::native_vp::{Ctx, NativeVp};
use crate::sdk::ibc::apps::transfer::types::ack_success_b64;
use crate::sdk::ibc::apps::transfer::types::{ack_success_b64, PORT_ID_STR};
use crate::sdk::ibc::core::channel::types::acknowledgement::AcknowledgementStatus;
use crate::sdk::ibc::core::channel::types::commitment::{
compute_ack_commitment, AcknowledgementCommitment, PacketCommitment,
Expand Down Expand Up @@ -385,7 +384,7 @@ where
// is done in the IBC VP, the test is repeated here to avoid making
// assumptions about how the IBC VP interprets the given message.
if self
.find_ibc_transfer_sequence(&msg, keys_changed)?
.find_ibc_transfer_sequence(msg, keys_changed)?
.is_none()
{
return Ok(acc);
Expand Down Expand Up @@ -518,13 +517,13 @@ where
self.apply_transfer_msg(acc, &msg.message, keys_changed)?;
}
// This event is emitted on the receiver
IbcMessage::RecvPacket(msg) => {
// Check if this is a Transfer packet
let Ok(packet_data) = serde_json::from_slice::<PacketData>(
IbcMessage::RecvPacket(msg)
if msg.message.packet.port_id_on_b.as_str() == PORT_ID_STR =>
{
let packet_data = serde_json::from_slice::<PacketData>(
&msg.message.packet.data,
) else {
return Ok(acc);
};
)
.map_err(native_vp::Error::new)?;
let receiver = packet_data.receiver.to_string();
let addr = TAddrData::Ibc(receiver.clone());
acc.decoder.insert(ibc_taddr(receiver), addr);
Expand Down Expand Up @@ -633,7 +632,7 @@ where
Error::NativeVpError(native_vp::Error::new_const(msg))
})?;
let conversion_state = self.ctx.state.in_mem().get_conversion_state();
let ibc_msgs = self.ctx.get_ibc_message(tx_data)?;
let ibc_msgs = self.ctx.get_ibc_message(tx_data).ok();

// Get the Transaction object from the actions
let masp_section_ref = namada_tx::action::get_masp_section_ref(
Expand Down Expand Up @@ -669,7 +668,7 @@ where
// Check the validity of the keys and get the transfer data
let mut changed_balances = self.validate_state_and_get_transfer_data(
keys_changed,
&ibc_msgs.as_slice(),
ibc_msgs.as_slice(),
)?;

let masp_address_hash = addr_taddr(MASP);
Expand Down
8 changes: 4 additions & 4 deletions crates/vp_env/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ use namada_core::hash::Hash;
use namada_core::storage::{BlockHeight, Epoch, Epochs, Header, Key, TxIndex};
use namada_events::{Event, EventType};
use namada_ibc::{decode_message, IbcMessage};
use namada_storage::{OptionExt, StorageRead};
use namada_storage::{OptionExt, ResultExt, StorageRead};
use namada_tx::BatchedTxRef;

/// Validity predicate's environment is available for native VPs and WASM VPs
Expand Down Expand Up @@ -120,16 +120,16 @@ where
/// Get a tx hash
fn get_tx_code_hash(&self) -> Result<Option<Hash>, namada_storage::Error>;

/// Get the masp tx part of the shielded action
/// Get the IBC message from the data section
fn get_ibc_message(
&self,
batched_tx: &BatchedTxRef<'_>,
) -> Result<Option<IbcMessage>, namada_storage::Error> {
) -> Result<IbcMessage, namada_storage::Error> {
let data = batched_tx
.tx
.data(batched_tx.cmt)
.ok_or_err_msg("No transaction data")?;
Ok(decode_message(&data).ok())
decode_message(&data).into_storage_result()
}

/// Charge the provided gas for the current vp
Expand Down

0 comments on commit 439e5a9

Please sign in to comment.