Input checks

pie-dao · Sep 12, 2020 · cce403c · cce403c
1 parent 5201455
commit cce403c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/contracts/VestedTokenMigration.sol b/contracts/VestedTokenMigration.sol
@@ -28,6 +28,8 @@ contract VestedTokenMigration is AragonApp {
     * @param _outputTokenManager Address of the output token
     */
     function initialize(address _inputTokenManager, address _outputTokenManager) external onlyInit {
+        require(_inputTokenManager != address(0), "INVALID_INPUT_TOKEN_MANAGER");
+        require(_outputTokenManager != address(0), "INVALID_OUTPUT_TOKEN_MANAGER")
         inputTokenManager = ITokenManager(_inputTokenManager);
         outputTokenManager = ITokenManager(_outputTokenManager);
         initialized();