feat: project 12

Author: jacobcreech

project-12-attack-the-bank/bank_one/.gitignore

@@ -0,0 +1,7 @@
+.anchor
+.DS_Store
+target
+**/*.rs.bk
+node_modules
+test-ledger
+.yarn

project-12-attack-the-bank/bank_one/.prettierignore

@@ -0,0 +1,7 @@
+.anchor
+.DS_Store
+target
+node_modules
+dist
+build
+test-ledger

project-12-attack-the-bank/bank_one/Anchor.toml

@@ -0,0 +1,18 @@
+[toolchain]
+
+[features]
+resolution = true
+skip-lint = false
+
+[programs.localnet]
+bank_one = "HAAzYF1LMBi8R2avDaa4s2VZyMyGzA1RGQNFPXPjctZo"
+
+[registry]
+url = "https://api.apr.dev"
+
+[provider]
+cluster = "Localnet"
+wallet = "~/.config/solana/id.json"
+
+[scripts]
+test = "yarn run ts-mocha -p ./tsconfig.json -t 1000000 tests/**/*.ts"

project-12-attack-the-bank/bank_one/Cargo.toml

@@ -0,0 +1,14 @@
+[workspace]
+members = [
+    "programs/*"
+]
+resolver = "2"
+
+[profile.release]
+overflow-checks = true
+lto = "fat"
+codegen-units = 1
+[profile.release.build-override]
+opt-level = 3
+incremental = false
+codegen-units = 1

project-12-attack-the-bank/bank_one/migrations/deploy.ts

@@ -0,0 +1,12 @@
+// Migrations are an early feature. Currently, they're nothing more than this
+// single deploy script that's invoked from the CLI, injecting a provider
+// configured from the workspace's Anchor.toml.
+
+const anchor = require("@coral-xyz/anchor");
+
+module.exports = async function (provider) {
+  // Configure client to use the provider.
+  anchor.setProvider(provider);
+
+  // Add your deploy script here.
+};

project-12-attack-the-bank/bank_one/package.json

@@ -0,0 +1,20 @@
+{
+  "license": "ISC",  
+  "scripts": {
+    "lint:fix": "prettier */*.js \"*/**/*{.js,.ts}\" -w",
+    "lint": "prettier */*.js \"*/**/*{.js,.ts}\" --check"
+  },
+  "dependencies": {
+    "@coral-xyz/anchor": "^0.30.1"
+  },
+  "devDependencies": {
+    "chai": "^4.3.4",
+    "mocha": "^9.0.3",
+    "ts-mocha": "^10.0.0",
+    "@types/bn.js": "^5.1.0",
+    "@types/chai": "^4.3.0",
+    "@types/mocha": "^9.0.0",
+    "typescript": "^4.3.5",
+    "prettier": "^2.6.2"
+  }
+}

project-12-attack-the-bank/bank_one/programs/bank_one/Cargo.toml

@@ -0,0 +1,20 @@
+[package]
+name = "bank_one"
+version = "0.1.0"
+description = "Created with Anchor"
+edition = "2021"
+
+[lib]
+crate-type = ["cdylib", "lib"]
+name = "bank_one"
+
+[features]
+default = []
+cpi = ["no-entrypoint"]
+no-entrypoint = []
+no-idl = []
+no-log-ix-name = []
+idl-build = ["anchor-lang/idl-build"]
+
+[dependencies]
+anchor-lang = { version = "0.30.1", features = ["init-if-needed"] }

project-12-attack-the-bank/bank_one/programs/bank_one/Xargo.toml

@@ -0,0 +1,2 @@
+[target.bpfel-unknown-unknown.dependencies.std]
+features = []

project-12-attack-the-bank/bank_one/programs/bank_one/src/lib.rs

@@ -0,0 +1,74 @@
+use anchor_lang::prelude::*;
+use anchor_lang::system_program::{transfer, Transfer};
+
+declare_id!("HAAzYF1LMBi8R2avDaa4s2VZyMyGzA1RGQNFPXPjctZo");
+
+#[program]
+pub mod bank_one {
+    use super::*;
+
+    pub fn deposit(ctx: Context<Deposit>, amount: u64) -> Result<()> {
+        *ctx.accounts.bank = Bank {
+            authority: ctx.accounts.authority.key(),
+            bank_balance: ctx.accounts.bank.bank_balance + amount,
+            bump: ctx.bumps.bank,
+        };
+        msg!("{:#?}", ctx.accounts.bank);
+
+        transfer(
+            CpiContext::new(
+                ctx.accounts.system_program.to_account_info(),
+                Transfer {
+                    from: ctx.accounts.authority.to_account_info(),
+                    to: ctx.accounts.bank.to_account_info(),
+                },
+            ),
+            amount,
+        )?;
+        Ok(())
+    }
+
+    pub fn withdraw(ctx: Context<Withdraw>, amount: u64) -> Result<()> {
+        ctx.accounts.bank.bank_balance -= amount;
+        ctx.accounts.bank.sub_lamports(amount)?;
+        ctx.accounts.authority.add_lamports(amount)?;
+        msg!("{:#?}", ctx.accounts.bank);
+        Ok(())
+    }
+}
+
+#[derive(Accounts)]
+pub struct Deposit<'info> {
+    #[account(mut)]
+    pub authority: Signer<'info>,
+    #[account(
+        init_if_needed,
+        payer = authority,
+        space = 8 + Bank::INIT_SPACE,
+        seeds = [b"bank"],
+        bump,
+    )]
+    pub bank: Account<'info, Bank>,
+    pub system_program: Program<'info, System>,
+}
+
+#[derive(Accounts)]
+pub struct Withdraw<'info> {
+    #[account(mut)]
+    pub authority: Signer<'info>,
+    #[account(
+        mut,
+        has_one = authority,
+        seeds = [b"bank"],
+        bump,
+    )]
+    pub bank: Account<'info, Bank>,
+}
+
+#[account]
+#[derive(InitSpace, Debug)]
+pub struct Bank {
+    pub authority: Pubkey,
+    pub bank_balance: u64,
+    pub bump: u8,
+}

project-12-attack-the-bank/bank_one/tests/bank_one.ts

@@ -0,0 +1,77 @@
+import * as anchor from "@coral-xyz/anchor";
+import { Program } from "@coral-xyz/anchor";
+import { BankOne } from "../target/types/bank_one";
+import { assert } from "chai";
+
+describe("bank_one", () => {
+  const provider = anchor.AnchorProvider.env();
+  const connection = provider.connection;
+  const wallet = provider.wallet as anchor.Wallet;
+  anchor.setProvider(provider);
+
+  const program = anchor.workspace.BankOne as Program<BankOne>;
+
+  const authority = new anchor.web3.Keypair();
+  const amount = 1_000_000; // 1million lamports
+
+  before(async () => {
+    const transferAmount = 1 * anchor.web3.LAMPORTS_PER_SOL;
+    const transferTx = new anchor.web3.Transaction().add(
+      anchor.web3.SystemProgram.transfer({
+        fromPubkey: wallet.publicKey,
+        toPubkey: authority.publicKey,
+        lamports: transferAmount,
+      })
+    );
+    await provider.sendAndConfirm(transferTx);
+  });
+
+  it("Deposit", async () => {
+    const transaction = await program.methods
+      .deposit(new anchor.BN(amount))
+      .accounts({ authority: authority.publicKey })
+      .transaction();
+
+    const transactionSignature = await anchor.web3.sendAndConfirmTransaction(
+      connection,
+      transaction,
+      [authority],
+      { commitment: "confirmed" }
+    );
+    console.log("Your transaction signature", transactionSignature);
+  });
+
+  it("Withdraw", async () => {
+    const walletInitialBalance = await connection.getBalance(wallet.publicKey);
+
+    const depositInstruction = await program.methods
+      .deposit(new anchor.BN(0))
+      .accounts({ authority: wallet.publicKey })
+      .instruction();
+
+    const withdrawInstruction = await program.methods
+      .withdraw(new anchor.BN(amount))
+      .accounts({ authority: wallet.publicKey })
+      .instruction();
+
+    const transaction = new anchor.web3.Transaction().add(
+      depositInstruction,
+      withdrawInstruction
+    );
+
+    const transactionSignature = await anchor.web3.sendAndConfirmTransaction(
+      connection,
+      transaction,
+      [wallet.payer],
+      { commitment: "confirmed" }
+    );
+
+    console.log("Your transaction signature", transactionSignature);
+
+    const walletFinalBalance = await connection.getBalance(wallet.publicKey, {
+      commitment: "confirmed",
+    });
+
+    assert.equal(walletFinalBalance, walletInitialBalance + amount);
+  });
+});