이번에는 위의 예제를 응용하여 owner에 대한 권한을 넘겨받는 부분을 Pull over Push 형태로 바꾸어 보겠다. 기존의 코드에서는 ownership을 보내버리면 한 번의 프로세스로 체크 없이 바로 보내(push)버리지만. ownable에서도 Pull over Push를 적용하면 new owner가 먼저 제안이 되고 그 후에 new owner가 ownership을 accept 하는 식(pull)으로 변경 시킬 수 있다.

이해하기 쉬운 설명을 위하여 최소한의 부분만을 보여주는 코드이기 때문에, 실제 owner에 대한 부분은 open-zepplin의 Ownable.sol 컨트렉트를 사용하는 것을 추천한다.

기존 코드의 프로세스

Owner가 transferOwnership을 실행하면 newOwner한테 Ownership이 넘어간다.

하나의 프로세스로 이루어지면 newOwner가 pull을 하는 부분이 없음을 볼 수 있다.

기존 코드

pragma solidity ^0.4.24;

contract Ownable {
  address private _owner;

  event OwnershipTransferred(address indexed previousOwner, address indexed newOwner);

  constructor () internal {
    _owner = msg.sender;
    emit OwnershipTransferred(address(0), _owner);
  }

  modifier onlyOwner() {
    require(isOwner(), "Ownable: caller is not the owner");
    _;
  }

  function transferOwnership(address newOwner) public onlyOwner {
    _transferOwnership(newOwner);
  }

  function _transferOwnership(address newOwner) internal {
    require(newOwner != address(0), "Ownable: new owner is the zero address");
    emit OwnershipTransferred(_owner, newOwner);
    _owner = newOwner;
  }
} 

위의 코드를 pull로 수정한다면 아래와 같은 문제를 해결할 수 있다.

transferOwnership에서 newOwner를 잘못 입력했을 때 주소값이 0인 경우만 예외로 처리하기 때문에 잘못된 주소를 입력받아서 컨트랙트의 Owner가 잘못 지정되거나 존재하지 않는 주소로 Owner의 권한이 가서 Owner가 존재하지 않는 주소로 지정되는 케이스가 발생할 수 있다.
⇒ pull 방식에서는 newOwner가 accept를 pull 하는 과정을 추가하여 Owner가 제대로 승인을 했는지 체크하고 transferOwnership을 다시 보내므로 제대로 된 검증되고 유효한 newOwner만 새로운 Owner가 될 수 있다.

수정할 코드의 프로세스

1. Owner가 newOwner에게 askOwnership을 보낸다.
2. newOwner가 Owner에게 acceptOwnership를 보낸다.
3. Owner가 transferOwnership을 실행한다.

3개의 프로세스로 이루어지며 newOwner가 pull을 하고 나서 owner가 변경되는 것을 볼 수 있다.

추가할 부분 1 구조체 & mapping

push 방식으로 바꾸기 위하여 owner를 바꾸는 것에 대한 정보를 기록해둘 mapping과 struct를 새로 선언해주어야 한다.

/**
   * @dev OwnerInfo and Owner is used for transferOwnership function
   */
  struct OwnerInfo {
    bool OwnerAsked;
    bool OwnerAccept;
  } 

  mapping(address => OwnerInfo) private Owner;

추가할 부분 2 – 함수에서 (askOwnership, acceptOwnership) 부분을 순차적으로 체크하고 난 후 push에 해당하는 _transferOwnership 함수를 실행한다.

/**
   * @dev Transfers control of the contract to a newOwner.
   * @param newOwner The address to transfer ownership to.
   */
  function _transferOwnership(address newOwner) internal {
    require(newOwner != address(0), "Ownable: new owner is the zero address");
    require(Owner[newOwner].OwnerAccept == true, "Ownable: newOwner didn't accept the request");
    emit OwnershipTransferred(_owner, newOwner);
    _owner = newOwner;
  }
  
  /**
   * @dev to adjust Pull-Over-Push Pattern new owner should be proposed first and stored in an intermediate variable
   */
  function askOwnership(address newOwner) public onlyOwner{
    OwnerInfo storage ownerInfo = Owner[newOwner]; 
    ownerInfo.OwnerAsked = true;
  }
    /**
   * @dev to adjust Pull-Over-Push Pattern proposed owner should be able to accept ownership of the contract
   */
  function acceptOwnership() public {
    OwnerInfo storage ownerInfo = Owner[msg.sender];
    require(ownerInfo.OwnerAsked == true, "Ownable: Owner did not make a request");
    ownerInfo.OwnerAccept = true;
  }

수정된 코드

pragma solidity ^0.4.24;

contract Ownable {
  address private _owner;
  
  struct OwnerInfo {
    bool OwnerAsked;
    bool OwnerAccept;
  } 

  mapping(address => OwnerInfo) private Owner;

  event OwnershipTransferred(address indexed previousOwner, address indexed newOwner);

  constructor () internal {
    _owner = msg.sender;
    emit OwnershipTransferred(address(0), _owner);
  }

  modifier onlyOwner() {
    require(isOwner(), "Ownable: caller is not the owner");
    _;
  }

  function isAsked() public view returns (bool) {
    return Owner[msg.sender].OwnerAsked;
  }

  function transferOwnership(address newOwner) public onlyOwner {
    _transferOwnership(newOwner);
  }

  function _transferOwnership(address newOwner) internal {
    require(newOwner != address(0), "Ownable: new owner is the zero address");
    require(Owner[newOwner].OwnerAccept == true, "Ownable: newOwner didn't accept the request");
    emit OwnershipTransferred(_owner, newOwner);
    _owner = newOwner;
  }
  
  /**
   * @dev to adjust Pull-Over-Push Pattern new owner should be proposed first and stored in an intermediate variable
   */
  function askOwnership(address newOwner) public onlyOwner{
    OwnerInfo storage ownerInfo = Owner[newOwner]; 
    ownerInfo.OwnerAsked = true;
  }
    /**
   * @dev to adjust Pull-Over-Push Pattern proposed owner should be able to accept ownership of the contract
   */
  function acceptOwnership() public {
    OwnerInfo storage ownerInfo = Owner[msg.sender];
    require(ownerInfo.OwnerAsked == true, "Ownable: Owner did not make a request");
    ownerInfo.OwnerAccept = true;
  }
  
}