certusone
/
wormhole
mirror of https://github.com/certusone/wormhole.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add reentrancy protection

This commit is contained in:
Hendrik Hofstadt 2020-08-20 16:58:03 +02:00
parent d1850e4c59
commit 085af1d6b7
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
ethereum/contracts/Wormhole.sol
View File

@ -1,9 +1,7 @@
// contracts/Wormhole.sol
// SPDX-License-Identifier: Apache 2
// TODO(hendrik): reentrancy protection for all methods
// TODO(hendrik): switch-over feature
// TODO(hendrik): add call for retrying a lockup that the guardian set have refused to sign
pragma solidity ^0.6.0;
pragma experimental ABIEncoderV2;
@ -11,10 +9,11 @@ pragma experimental ABIEncoderV2;
import "@openzeppelin/contracts/token/ERC20/IERC20.sol";
import "@openzeppelin/contracts/token/ERC20/SafeERC20.sol";
import "@openzeppelin/contracts/math/SafeMath.sol";
import "@openzeppelin/contracts/utils/ReentrancyGuard.sol";
import "./BytesLib.sol";
import "./WrappedAsset.sol";
contract Wormhole {
contract Wormhole is ReentrancyGuard {
using SafeERC20 for IERC20;
using BytesLib for bytes;
using SafeMath for uint256;
@ -77,7 +76,7 @@ contract Wormhole {
function submitVAA(
bytes calldata vaa
) public {
) public nonReentrant {
uint8 version = vaa.toUint8(0);
require(version == 1, "VAA version incompatible");
@ -208,7 +207,7 @@ contract Wormhole {
uint256 amount,
bytes32 recipient,
uint8 target_chain
) public {
) public nonReentrant {
require(amount != 0, "amount must not be 0");
uint8 asset_chain = CHAIN_ID;
@ -234,7 +233,7 @@ contract Wormhole {
function lockETH(
bytes32 recipient,
uint8 target_chain
) public payable {
) public payable nonReentrant {
require(msg.value != 0, "amount must not be 0");
// Wrap tx value in WETH