add zero-address checks to Ticket and ControlledToken (#306)

pooltogether · Jul 12, 2021 · 0af7e60 · 0af7e60
1 parent 65f81f8
commit 0af7e60
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/contracts/token/ControlledToken.sol b/contracts/token/ControlledToken.sol
@@ -37,6 +37,7 @@ contract ControlledToken is ERC20PermitUpgradeable, ControlledTokenInterface {
     virtual
     initializer
   {
+    require(address(_controller) != address(0), "ControlledToken/controller-not-zero");
     __ERC20_init(_name, _symbol);
     __ERC20Permit_init("PoolTogether ControlledToken");
     controller = _controller;

diff --git a/contracts/token/Ticket.sol b/contracts/token/Ticket.sol
@@ -41,9 +41,9 @@ contract Ticket is ControlledToken, TicketInterface {
     override
     initializer
   {
+    require(address(_controller) != address(0), "Ticket/controller-not-zero");
     ControlledToken.initialize(_name, _symbol, _decimals, _controller);
     sortitionSumTrees.createTree(TREE_KEY, MAX_TREE_LEAVES);
-
     emit Initialized(
       _name,
       _symbol,