Повернення

Інструкція повернення завершує наш цикл флеш-кредиту, забезпечуючи повернення позичених коштів з відповідними комісіями. Ця інструкція виконує два важливих кроки:

1. Отримання суми позики: Використовуємо інтроспекцію інструкцій для отримання оригінального amount_borrowed з даних інструкції запозичення

2. Повернення коштів: Обчислюємо fee і переказуємо позичену суму плюс комісію назад до протоколу

Інтроспекція інструкцій

Спочатку нам потрібно перевірити першу інструкцію в транзакції, щоб отримати оригінальну суму позики:

let ixs = ctx.accounts.instructions.to_account_info();

let mut amount_borrowed: u64;

if let Ok(borrow_ix) = load_instruction_at_checked(0, &ixs) {
    // Check the amount borrowed:
    let mut borrowed_data: [u8;8] = [0u8;8];
    borrowed_data.copy_from_slice(&borrow_ix.data[8..16]);
    amount_borrowed = u64::from_le_bytes(borrowed_data)

} else {
    return Err(ProtocolError::MissingBorrowIx.into());
}

Ми не перевіряємо, чи це справді borrow_ix за допомогою ідентифікатора програми та дискримінатора, оскільки не має значення, чи вони насправді створюють "фальшиву" інструкцію; це безпечно для протоколу, оскільки він просто отримує оплату. Водночас, якщо ми позичили гроші, ми знаємо, що це буде перша інструкція і що amount_borrowed буде там.

Переказ коштів

Далі ми обчислюємо комісію протоколу та переказуємо загальну суму назад:

// Add the fee to the amount borrowed (In our case we hardcoded it to 500 basis point)
let fee = (amount_borrowed as u128).checked_mul(500).unwrap().checked_div(10_000).ok_or(ProtocolError::Overflow)? as u64;
amount_borrowed = amount_borrowed.checked_add(fee).ok_or(ProtocolError::Overflow)?;

// Transfer the funds from the protocol to the borrower
transfer(
    CpiContext::new(ctx.accounts.token_program.to_account_info(), Transfer {
        from: ctx.accounts.borrower_ata.to_account_info(),
        to: ctx.accounts.protocol_ata.to_account_info(),
        authority: ctx.accounts.borrower.to_account_info(),
    }), 
    amount_borrowed
)?;

Наша комісія жорстко закодована на рівні 500 базисних пунктів, і ми виконуємо "перевірену" математику, щоб переконатися, що сума не переповнюється, що могло б бути використано для експлуатації з дуже великими числами. Крім того, ми конвертуємо суму в u128 для множення, щоб запобігти проміжному переповненню, а потім безпечно перетворюємо назад у u64