margin_trade: forbid direct self-calls

2022-06-07 13:25:30 +02:00 · 2022-06-07 13:25:30 +02:00 · a14f37362d
parent 685fd3122f
commit a14f37362d
1 changed files with 2 additions and 0 deletions
--- a/programs/mango-v4/src/instructions/margin_trade.rs
+++ b/programs/mango-v4/src/instructions/margin_trade.rs
@ -110,6 +110,8 @@ pub fn margin_trade<'key, 'accounts, 'remaining, 'info>(
    msg!("pre_cpi_health {:?}", pre_cpi_health);

    let cpi_program_id = *ctx.remaining_accounts[num_health_accounts].key;
+    require_keys_neq!(cpi_program_id, crate::id(), MangoError::SomeError);
+
    let cpi_ais = &ctx.remaining_accounts[num_health_accounts + 1..];
    let mut cpi_ams = cpi_ais
        .iter()