|
| 1 | +pragma solidity 0.5.17; |
| 2 | + |
| 3 | +import {RLPReader} from "solidity-rlp/contracts/RLPReader.sol"; |
| 4 | +import {BytesLib} from "./BytesLib.sol"; |
| 5 | + |
| 6 | +library ExitPayloadReader { |
| 7 | + using RLPReader for bytes; |
| 8 | + using RLPReader for RLPReader.RLPItem; |
| 9 | + |
| 10 | + uint8 constant WORD_SIZE = 32; |
| 11 | + |
| 12 | + struct ExitPayload { |
| 13 | + RLPReader.RLPItem[] data; |
| 14 | + } |
| 15 | + |
| 16 | + struct Receipt { |
| 17 | + RLPReader.RLPItem[] data; |
| 18 | + bytes raw; |
| 19 | + uint256 logIndex; |
| 20 | + } |
| 21 | + |
| 22 | + struct Log { |
| 23 | + RLPReader.RLPItem data; |
| 24 | + RLPReader.RLPItem[] list; |
| 25 | + } |
| 26 | + |
| 27 | + struct LogTopics { |
| 28 | + RLPReader.RLPItem[] data; |
| 29 | + } |
| 30 | + |
| 31 | + function toExitPayload(bytes memory data) |
| 32 | + internal |
| 33 | + pure |
| 34 | + returns (ExitPayload memory) |
| 35 | + { |
| 36 | + RLPReader.RLPItem[] memory payloadData = data |
| 37 | + .toRlpItem() |
| 38 | + .toList(); |
| 39 | + |
| 40 | + return ExitPayload(payloadData); |
| 41 | + } |
| 42 | + |
| 43 | + function copy(uint src, uint dest, uint len) private pure { |
| 44 | + if (len == 0) return; |
| 45 | + |
| 46 | + // copy as many word sizes as possible |
| 47 | + for (; len >= WORD_SIZE; len -= WORD_SIZE) { |
| 48 | + assembly { |
| 49 | + mstore(dest, mload(src)) |
| 50 | + } |
| 51 | + |
| 52 | + src += WORD_SIZE; |
| 53 | + dest += WORD_SIZE; |
| 54 | + } |
| 55 | + |
| 56 | + // left over bytes. Mask is used to remove unwanted bytes from the word |
| 57 | + uint mask = 256 ** (WORD_SIZE - len) - 1; |
| 58 | + assembly { |
| 59 | + let srcpart := and(mload(src), not(mask)) // zero out src |
| 60 | + let destpart := and(mload(dest), mask) // retrieve the bytes |
| 61 | + mstore(dest, or(destpart, srcpart)) |
| 62 | + } |
| 63 | + } |
| 64 | + |
| 65 | + function getHeaderNumber(ExitPayload memory payload) internal pure returns(uint256) { |
| 66 | + return payload.data[0].toUint(); |
| 67 | + } |
| 68 | + |
| 69 | + function getBlockProof(ExitPayload memory payload) internal pure returns(bytes memory) { |
| 70 | + return payload.data[1].toBytes(); |
| 71 | + } |
| 72 | + |
| 73 | + function getBlockNumber(ExitPayload memory payload) internal pure returns(uint256) { |
| 74 | + return payload.data[2].toUint(); |
| 75 | + } |
| 76 | + |
| 77 | + function getBlockTime(ExitPayload memory payload) internal pure returns(uint256) { |
| 78 | + return payload.data[3].toUint(); |
| 79 | + } |
| 80 | + |
| 81 | + function getTxRoot(ExitPayload memory payload) internal pure returns(bytes32) { |
| 82 | + return bytes32(payload.data[4].toUint()); |
| 83 | + } |
| 84 | + |
| 85 | + function getReceiptRoot(ExitPayload memory payload) internal pure returns(bytes32) { |
| 86 | + return bytes32(payload.data[5].toUint()); |
| 87 | + } |
| 88 | + |
| 89 | + function getReceipt(ExitPayload memory payload) internal pure returns(Receipt memory receipt) { |
| 90 | + receipt.raw = payload.data[6].toBytes(); |
| 91 | + RLPReader.RLPItem memory receiptItem = receipt.raw.toRlpItem(); |
| 92 | + |
| 93 | + if (receiptItem.isList()) { |
| 94 | + // legacy tx |
| 95 | + receipt.data = receiptItem.toList(); |
| 96 | + } else { |
| 97 | + // pop first byte before parsting receipt |
| 98 | + bytes memory typedBytes = receipt.raw; |
| 99 | + bytes memory result = new bytes(typedBytes.length - 1); |
| 100 | + uint256 srcPtr; |
| 101 | + uint256 destPtr; |
| 102 | + assembly { |
| 103 | + srcPtr := add(33, typedBytes) |
| 104 | + destPtr := add(0x20, result) |
| 105 | + } |
| 106 | + |
| 107 | + copy(srcPtr, destPtr, result.length); |
| 108 | + receipt.data = result.toRlpItem().toList(); |
| 109 | + } |
| 110 | + |
| 111 | + receipt.logIndex = getReceiptLogIndex(payload); |
| 112 | + return receipt; |
| 113 | + } |
| 114 | + |
| 115 | + function getReceiptProof(ExitPayload memory payload) internal pure returns(bytes memory) { |
| 116 | + return payload.data[7].toBytes(); |
| 117 | + } |
| 118 | + |
| 119 | + function getBranchMaskAsBytes(ExitPayload memory payload) internal pure returns(bytes memory) { |
| 120 | + return payload.data[8].toBytes(); |
| 121 | + } |
| 122 | + |
| 123 | + function getBranchMaskAsUint(ExitPayload memory payload) internal pure returns(uint256) { |
| 124 | + return payload.data[8].toUint(); |
| 125 | + } |
| 126 | + |
| 127 | + function getReceiptLogIndex(ExitPayload memory payload) internal pure returns(uint256) { |
| 128 | + return payload.data[9].toUint(); |
| 129 | + } |
| 130 | + |
| 131 | + function getTx(ExitPayload memory payload) internal pure returns(bytes memory) { |
| 132 | + return payload.data[10].toBytes(); |
| 133 | + } |
| 134 | + |
| 135 | + function getTxProof(ExitPayload memory payload) internal pure returns(bytes memory) { |
| 136 | + return payload.data[11].toBytes(); |
| 137 | + } |
| 138 | + |
| 139 | + // Receipt methods |
| 140 | + function toBytes(Receipt memory receipt) internal pure returns(bytes memory) { |
| 141 | + return receipt.raw; |
| 142 | + } |
| 143 | + |
| 144 | + function getLog(Receipt memory receipt) internal pure returns(Log memory) { |
| 145 | + RLPReader.RLPItem memory logData = receipt.data[3].toList()[receipt.logIndex]; |
| 146 | + return Log(logData, logData.toList()); |
| 147 | + } |
| 148 | + |
| 149 | + // Log methods |
| 150 | + function getEmitter(Log memory log) internal pure returns(address) { |
| 151 | + return RLPReader.toAddress(log.list[0]); |
| 152 | + } |
| 153 | + |
| 154 | + function getTopics(Log memory log) internal pure returns(LogTopics memory) { |
| 155 | + return LogTopics(log.list[1].toList()); |
| 156 | + } |
| 157 | + |
| 158 | + function getData(Log memory log) internal pure returns(bytes memory) { |
| 159 | + return log.list[2].toBytes(); |
| 160 | + } |
| 161 | + |
| 162 | + function toRlpBytes(Log memory log) internal pure returns(bytes memory) { |
| 163 | + return log.data.toRlpBytes(); |
| 164 | + } |
| 165 | + |
| 166 | + // LogTopics methods |
| 167 | + function getField(LogTopics memory topics, uint256 index) internal pure returns(RLPReader.RLPItem memory) { |
| 168 | + return topics.data[index]; |
| 169 | + } |
| 170 | +} |
0 commit comments