文件 1 的 25:CarefulMath.sol
pragma solidity ^0.7.6;
contract CarefulMath {
enum MathError {
NO_ERROR,
DIVISION_BY_ZERO,
INTEGER_OVERFLOW,
INTEGER_UNDERFLOW
}
function mulUInt(uint a, uint b) internal pure returns (MathError, uint) {
if (a == 0) {
return (MathError.NO_ERROR, 0);
}
uint c = a * b;
if (c / a != b) {
return (MathError.INTEGER_OVERFLOW, 0);
} else {
return (MathError.NO_ERROR, c);
}
}
function divUInt(uint a, uint b) internal pure returns (MathError, uint) {
if (b == 0) {
return (MathError.DIVISION_BY_ZERO, 0);
}
return (MathError.NO_ERROR, a / b);
}
function subUInt(uint a, uint b) internal pure returns (MathError, uint) {
if (b <= a) {
return (MathError.NO_ERROR, a - b);
} else {
return (MathError.INTEGER_UNDERFLOW, 0);
}
}
function addUInt(uint a, uint b) internal pure returns (MathError, uint) {
uint c = a + b;
if (c >= a) {
return (MathError.NO_ERROR, c);
} else {
return (MathError.INTEGER_OVERFLOW, 0);
}
}
function addThenSubUInt(uint a, uint b, uint c) internal pure returns (MathError, uint) {
(MathError err0, uint sum) = addUInt(a, b);
if (err0 != MathError.NO_ERROR) {
return (err0, 0);
}
return subUInt(sum, c);
}
}
文件 2 的 25:Controller.sol
pragma solidity ^0.7.6;
import "./ErrorReporter.sol";
import "./Exponential.sol";
import "./PriceOracle.sol";
import "./ControllerInterface.sol";
import "./ControllerStorage.sol";
import "./PTokenInterfaces.sol";
import "./EIP20Interface.sol";
import "./Unitroller.sol";
contract Controller is ControllerStorage, ControllerInterface, ControllerErrorReporter, Exponential {
event MarketListed(address pToken);
event MarketEntered(address pToken, address account);
event MarketExited(address pToken, address account);
event NewCloseFactor(uint oldCloseFactorMantissa, uint newCloseFactorMantissa);
event NewCollateralFactor(address pToken, uint oldCollateralFactorMantissa, uint newCollateralFactorMantissa);
event NewLiquidationIncentive(uint oldLiquidationIncentiveMantissa, uint newLiquidationIncentiveMantissa);
event NewMaxAssets(uint oldMaxAssets, uint newMaxAssets);
event NewPriceOracle(PriceOracle oldPriceOracle, PriceOracle newPriceOracle);
event NewPauseGuardian(address oldPauseGuardian, address newPauseGuardian);
event ActionPaused(string action, bool pauseState);
event ActionPaused(address pToken, string action, bool pauseState);
event PieSpeedUpdated(address indexed pToken, uint newSpeed);
event DistributedSupplierPie(address indexed pToken, address indexed supplier, uint pieDelta, uint pieSupplyIndex);
event DistributedBorrowerPie(address indexed pToken, address indexed borrower, uint pieDelta, uint pieBorrowIndex);
uint public constant pieClaimThreshold = 0.001e18;
uint224 public constant pieInitialIndex = 1e36;
uint internal constant closeFactorMinMantissa = 0.05e18;
uint internal constant closeFactorMaxMantissa = 0.9e18;
uint internal constant collateralFactorMaxMantissa = 0.9e18;
uint internal constant liquidationIncentiveMinMantissa = 1.0e18;
uint internal constant liquidationIncentiveMaxMantissa = 1.5e18;
constructor() {
admin = msg.sender;
}
function getAssetsIn(address account) external view returns (address[] memory) {
address[] memory assetsIn = accountAssets[account];
return assetsIn;
}
function checkMembership(address account, address pToken) external view returns (bool) {
return markets[pToken].accountMembership[account];
}
function enterMarkets(address[] memory pTokens) public override returns (uint[] memory) {
uint len = pTokens.length;
uint[] memory results = new uint[](len);
for (uint i = 0; i < len; i++) {
address pToken = pTokens[i];
results[i] = uint(addToMarketInternal(pToken, msg.sender));
}
return results;
}
function addToMarketInternal(address pToken, address borrower) internal returns (Error) {
Market storage marketToJoin = markets[pToken];
if (!marketToJoin.isListed) {
return Error.MARKET_NOT_LISTED;
}
if (marketToJoin.accountMembership[borrower] == true) {
return Error.NO_ERROR;
}
if (accountAssets[borrower].length >= maxAssets) {
return Error.TOO_MANY_ASSETS;
}
marketToJoin.accountMembership[borrower] = true;
accountAssets[borrower].push(pToken);
emit MarketEntered(pToken, borrower);
return Error.NO_ERROR;
}
function exitMarket(address pTokenAddress) external override returns (uint) {
address pToken = pTokenAddress;
(uint oErr, uint tokensHeld, uint amountOwed, ) = PTokenInterface(pToken).getAccountSnapshot(msg.sender);
require(oErr == 0, "exitMarket: getAccountSnapshot failed");
if (amountOwed != 0) {
return fail(Error.NONZERO_BORROW_BALANCE, FailureInfo.EXIT_MARKET_BALANCE_OWED);
}
uint allowed = redeemAllowedInternal(pTokenAddress, msg.sender, tokensHeld);
if (allowed != 0) {
return failOpaque(Error.REJECTION, FailureInfo.EXIT_MARKET_REJECTION, allowed);
}
Market storage marketToExit = markets[pToken];
if (!marketToExit.accountMembership[msg.sender]) {
return uint(Error.NO_ERROR);
}
delete marketToExit.accountMembership[msg.sender];
address[] memory userAssetList = accountAssets[msg.sender];
uint len = userAssetList.length;
uint assetIndex = len;
for (uint i = 0; i < len; i++) {
if (userAssetList[i] == pToken) {
assetIndex = i;
break;
}
}
assert(assetIndex < len);
address[] storage storedList = accountAssets[msg.sender];
storedList[assetIndex] = storedList[storedList.length - 1];
storedList.pop();
emit MarketExited(pToken, msg.sender);
return uint(Error.NO_ERROR);
}
function mintAllowed(address pToken, address minter, uint mintAmount) external override returns (uint) {
require(!mintGuardianPaused[pToken], "mint is paused");
minter;
mintAmount;
if (!markets[pToken].isListed) {
return uint(Error.MARKET_NOT_LISTED);
}
updatePieSupplyIndex(pToken);
distributeSupplierPie(pToken, minter, false);
return uint(Error.NO_ERROR);
}
function redeemAllowed(address pToken, address redeemer, uint redeemTokens) external override returns (uint) {
uint allowed = redeemAllowedInternal(pToken, redeemer, redeemTokens);
if (allowed != uint(Error.NO_ERROR)) {
return allowed;
}
updatePieSupplyIndex(pToken);
distributeSupplierPie(pToken, redeemer, false);
return uint(Error.NO_ERROR);
}
function redeemAllowedInternal(address pToken, address redeemer, uint redeemTokens) internal view returns (uint) {
if (!markets[pToken].isListed) {
return uint(Error.MARKET_NOT_LISTED);
}
if (!markets[pToken].accountMembership[redeemer]) {
return uint(Error.NO_ERROR);
}
(Error err, , uint shortfall) = getHypotheticalAccountLiquidityInternal(redeemer, pToken, redeemTokens, 0);
if (err != Error.NO_ERROR) {
return uint(err);
}
if (shortfall > 0) {
return uint(Error.INSUFFICIENT_LIQUIDITY);
}
return uint(Error.NO_ERROR);
}
function redeemVerify(address pToken, address redeemer, uint redeemAmount, uint redeemTokens) external override {
pToken;
redeemer;
if (redeemTokens == 0 && redeemAmount > 0) {
revert("redeemTokens zero");
}
}
function borrowAllowed(address pToken, address borrower, uint borrowAmount) external override returns (uint) {
require(!borrowGuardianPaused[pToken], "borrow is paused");
if (!markets[pToken].isListed) {
return uint(Error.MARKET_NOT_LISTED);
}
Error err;
if (!markets[pToken].accountMembership[borrower]) {
require(msg.sender == pToken, "sender must be pToken");
err = addToMarketInternal(msg.sender, borrower);
if (err != Error.NO_ERROR) {
return uint(err);
}
assert(markets[pToken].accountMembership[borrower]);
}
if (oracle.getUnderlyingPrice(pToken) == 0) {
return uint(Error.PRICE_ERROR);
}
uint shortfall;
(err, , shortfall) = getHypotheticalAccountLiquidityInternal(borrower, pToken, 0, borrowAmount);
if (err != Error.NO_ERROR) {
return uint(err);
}
if (shortfall > 0) {
return uint(Error.INSUFFICIENT_LIQUIDITY);
}
Exp memory borrowIndex = Exp({mantissa: PTokenInterface(pToken).borrowIndex()});
updatePieBorrowIndex(pToken, borrowIndex);
distributeBorrowerPie(pToken, borrower, borrowIndex, false);
return uint(Error.NO_ERROR);
}
function repayBorrowAllowed(
address pToken,
address payer,
address borrower,
uint repayAmount
) external override returns (uint) {
payer;
borrower;
repayAmount;
if (!markets[pToken].isListed) {
return uint(Error.MARKET_NOT_LISTED);
}
Exp memory borrowIndex = Exp({mantissa: PTokenInterface(pToken).borrowIndex()});
updatePieBorrowIndex(pToken, borrowIndex);
distributeBorrowerPie(pToken, borrower, borrowIndex, false);
return uint(Error.NO_ERROR);
}
function liquidateBorrowAllowed(
address pTokenBorrowed,
address pTokenCollateral,
address liquidator,
address borrower,
uint repayAmount
) external override returns (uint) {
liquidator;
if (!markets[pTokenBorrowed].isListed || !markets[pTokenCollateral].isListed) {
return uint(Error.MARKET_NOT_LISTED);
}
(Error err, , uint shortfall) = getAccountLiquidityInternal(borrower);
if (err != Error.NO_ERROR) {
return uint(err);
}
if (shortfall == 0) {
return uint(Error.INSUFFICIENT_SHORTFALL);
}
uint borrowBalance = PTokenInterface(pTokenBorrowed).borrowBalanceStored(borrower);
(MathError mathErr, uint maxClose) = mulScalarTruncate(Exp({mantissa: closeFactorMantissa}), borrowBalance);
if (mathErr != MathError.NO_ERROR) {
return uint(Error.MATH_ERROR);
}
if (repayAmount > maxClose) {
return uint(Error.TOO_MUCH_REPAY);
}
return uint(Error.NO_ERROR);
}
function seizeAllowed(
address pTokenCollateral,
address pTokenBorrowed,
address liquidator,
address borrower,
uint seizeTokens
) external override returns (uint) {
require(!seizeGuardianPaused, "seize is paused");
seizeTokens;
if (!markets[pTokenCollateral].isListed || !markets[pTokenBorrowed].isListed) {
return uint(Error.MARKET_NOT_LISTED);
}
if (PTokenInterface(pTokenCollateral).controller() != PTokenInterface(pTokenBorrowed).controller()) {
return uint(Error.CONTROLLER_MISMATCH);
}
updatePieSupplyIndex(pTokenCollateral);
distributeSupplierPie(pTokenCollateral, borrower, false);
distributeSupplierPie(pTokenCollateral, liquidator, false);
return uint(Error.NO_ERROR);
}
function transferAllowed(
address pToken,
address src,
address dst,
uint transferTokens
) external override returns (uint) {
require(!transferGuardianPaused, "transfer is paused");
uint allowed = redeemAllowedInternal(pToken, src, transferTokens);
if (allowed != uint(Error.NO_ERROR)) {
return allowed;
}
updatePieSupplyIndex(pToken);
distributeSupplierPie(pToken, src, false);
distributeSupplierPie(pToken, dst, false);
return uint(Error.NO_ERROR);
}
struct AccountLiquidityLocalVars {
uint sumCollateral;
uint sumBorrowPlusEffects;
uint pTokenBalance;
uint borrowBalance;
uint exchangeRateMantissa;
uint oraclePriceMantissa;
Exp collateralFactor;
Exp exchangeRate;
Exp oraclePrice;
Exp tokensToDenom;
}
function getAccountLiquidity(address account) public view returns (uint, uint, uint) {
(Error err, uint liquidity, uint shortfall) = getHypotheticalAccountLiquidityInternal(account, address(0), 0, 0);
return (uint(err), liquidity, shortfall);
}
function getAccountLiquidityInternal(address account) internal view returns (Error, uint, uint) {
return getHypotheticalAccountLiquidityInternal(account, address(0), 0, 0);
}
function getHypotheticalAccountLiquidity(
address account,
address pTokenModify,
uint redeemTokens,
uint borrowAmount
) public view virtual returns (uint, uint, uint) {
(Error err, uint liquidity, uint shortfall) = getHypotheticalAccountLiquidityInternal(account, pTokenModify, redeemTokens, borrowAmount);
return (uint(err), liquidity, shortfall);
}
function getHypotheticalAccountLiquidityInternal(
address account,
address pTokenModify,
uint redeemTokens,
uint borrowAmount
) internal view returns (Error, uint, uint) {
AccountLiquidityLocalVars memory vars;
uint oErr;
MathError mErr;
address[] memory assets = accountAssets[account];
for (uint i = 0; i < assets.length; i++) {
address asset = assets[i];
(oErr, vars.pTokenBalance, vars.borrowBalance, vars.exchangeRateMantissa) = PTokenInterface(asset).getAccountSnapshot(account);
if (oErr != 0) {
return (Error.SNAPSHOT_ERROR, 0, 0);
}
vars.collateralFactor = Exp({mantissa: markets[address(asset)].collateralFactorMantissa});
vars.exchangeRate = Exp({mantissa: vars.exchangeRateMantissa});
vars.oraclePriceMantissa = oracle.getUnderlyingPrice(asset);
if (vars.oraclePriceMantissa == 0) {
return (Error.PRICE_ERROR, 0, 0);
}
vars.oraclePrice = Exp({mantissa: vars.oraclePriceMantissa});
(mErr, vars.tokensToDenom) = mulExp3(vars.collateralFactor, vars.exchangeRate, vars.oraclePrice);
if (mErr != MathError.NO_ERROR) {
return (Error.MATH_ERROR, 0, 0);
}
(mErr, vars.sumCollateral) = mulScalarTruncateAddUInt(vars.tokensToDenom, vars.pTokenBalance, vars.sumCollateral);
if (mErr != MathError.NO_ERROR) {
return (Error.MATH_ERROR, 0, 0);
}
(mErr, vars.sumBorrowPlusEffects) = mulScalarTruncateAddUInt(vars.oraclePrice, vars.borrowBalance, vars.sumBorrowPlusEffects);
if (mErr != MathError.NO_ERROR) {
return (Error.MATH_ERROR, 0, 0);
}
if (asset == pTokenModify) {
(mErr, vars.sumBorrowPlusEffects) = mulScalarTruncateAddUInt(vars.tokensToDenom, redeemTokens, vars.sumBorrowPlusEffects);
if (mErr != MathError.NO_ERROR) {
return (Error.MATH_ERROR, 0, 0);
}
(mErr, vars.sumBorrowPlusEffects) = mulScalarTruncateAddUInt(vars.oraclePrice, borrowAmount, vars.sumBorrowPlusEffects);
if (mErr != MathError.NO_ERROR) {
return (Error.MATH_ERROR, 0, 0);
}
}
}
if (vars.sumCollateral > vars.sumBorrowPlusEffects) {
return (Error.NO_ERROR, vars.sumCollateral - vars.sumBorrowPlusEffects, 0);
} else {
return (Error.NO_ERROR, 0, vars.sumBorrowPlusEffects - vars.sumCollateral);
}
}
function liquidateCalculateSeizeTokens(
address pTokenBorrowed,
address pTokenCollateral,
uint actualRepayAmount
) external view override returns (uint, uint) {
uint priceBorrowedMantissa = oracle.getUnderlyingPrice(pTokenBorrowed);
uint priceCollateralMantissa = oracle.getUnderlyingPrice(pTokenCollateral);
if (priceBorrowedMantissa == 0 || priceCollateralMantissa == 0) {
return (uint(Error.PRICE_ERROR), 0);
}
uint exchangeRateMantissa = PTokenInterface(pTokenCollateral).exchangeRateStored();
uint seizeTokens;
Exp memory numerator;
Exp memory denominator;
Exp memory ratio;
MathError mathErr;
(mathErr, numerator) = mulExp(liquidationIncentiveMantissa, priceBorrowedMantissa);
if (mathErr != MathError.NO_ERROR) {
return (uint(Error.MATH_ERROR), 0);
}
(mathErr, denominator) = mulExp(priceCollateralMantissa, exchangeRateMantissa);
if (mathErr != MathError.NO_ERROR) {
return (uint(Error.MATH_ERROR), 0);
}
(mathErr, ratio) = divExp(numerator, denominator);
if (mathErr != MathError.NO_ERROR) {
return (uint(Error.MATH_ERROR), 0);
}
(mathErr, seizeTokens) = mulScalarTruncate(ratio, actualRepayAmount);
if (mathErr != MathError.NO_ERROR) {
return (uint(Error.MATH_ERROR), 0);
}
return (uint(Error.NO_ERROR), seizeTokens);
}
function _setPriceOracle(PriceOracle newOracle) public returns (uint) {
if (msg.sender != admin) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_PRICE_ORACLE_OWNER_CHECK);
}
PriceOracle oldOracle = oracle;
oracle = newOracle;
emit NewPriceOracle(oldOracle, newOracle);
return uint(Error.NO_ERROR);
}
function _setPieAddress(address pieAddress_) public returns (uint) {
require(msg.sender == admin && pieAddress == address(0),"pie address may only be initialized once");
pieAddress = pieAddress_;
return uint(Error.NO_ERROR);
}
function _setCloseFactor(uint newCloseFactorMantissa) external returns (uint) {
if (msg.sender != admin) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_CLOSE_FACTOR_OWNER_CHECK);
}
Exp memory newCloseFactorExp = Exp({mantissa: newCloseFactorMantissa});
Exp memory lowLimit = Exp({mantissa: closeFactorMinMantissa});
if (lessThanOrEqualExp(newCloseFactorExp, lowLimit)) {
return fail(Error.INVALID_CLOSE_FACTOR, FailureInfo.SET_CLOSE_FACTOR_VALIDATION);
}
Exp memory highLimit = Exp({mantissa: closeFactorMaxMantissa});
if (lessThanExp(highLimit, newCloseFactorExp)) {
return fail(Error.INVALID_CLOSE_FACTOR, FailureInfo.SET_CLOSE_FACTOR_VALIDATION);
}
uint oldCloseFactorMantissa = closeFactorMantissa;
closeFactorMantissa = newCloseFactorMantissa;
emit NewCloseFactor(oldCloseFactorMantissa, closeFactorMantissa);
return uint(Error.NO_ERROR);
}
function _setCollateralFactor(address pToken, uint newCollateralFactorMantissa) external returns (uint) {
if (msg.sender != admin) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_COLLATERAL_FACTOR_OWNER_CHECK);
}
Market storage market = markets[pToken];
if (!market.isListed) {
return fail(Error.MARKET_NOT_LISTED, FailureInfo.SET_COLLATERAL_FACTOR_NO_EXISTS);
}
Exp memory newCollateralFactorExp = Exp({mantissa: newCollateralFactorMantissa});
Exp memory highLimit = Exp({mantissa: collateralFactorMaxMantissa});
if (lessThanExp(highLimit, newCollateralFactorExp)) {
return fail(Error.INVALID_COLLATERAL_FACTOR, FailureInfo.SET_COLLATERAL_FACTOR_VALIDATION);
}
oracle.updateUnderlyingPrice(pToken);
if (newCollateralFactorMantissa != 0 && oracle.getUnderlyingPrice(pToken) == 0) {
return fail(Error.PRICE_ERROR, FailureInfo.SET_COLLATERAL_FACTOR_WITHOUT_PRICE);
}
uint oldCollateralFactorMantissa = market.collateralFactorMantissa;
market.collateralFactorMantissa = newCollateralFactorMantissa;
emit NewCollateralFactor(pToken, oldCollateralFactorMantissa, newCollateralFactorMantissa);
return uint(Error.NO_ERROR);
}
function _setMaxAssets(uint newMaxAssets) external returns (uint) {
if (msg.sender != admin) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_MAX_ASSETS_OWNER_CHECK);
}
uint oldMaxAssets = maxAssets;
maxAssets = newMaxAssets;
emit NewMaxAssets(oldMaxAssets, newMaxAssets);
return uint(Error.NO_ERROR);
}
function _setLiquidationIncentive(uint newLiquidationIncentiveMantissa) external returns (uint) {
if (msg.sender != admin) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_LIQUIDATION_INCENTIVE_OWNER_CHECK);
}
Exp memory newLiquidationIncentive = Exp({mantissa: newLiquidationIncentiveMantissa});
Exp memory minLiquidationIncentive = Exp({mantissa: liquidationIncentiveMinMantissa});
if (lessThanExp(newLiquidationIncentive, minLiquidationIncentive)) {
return fail(Error.INVALID_LIQUIDATION_INCENTIVE, FailureInfo.SET_LIQUIDATION_INCENTIVE_VALIDATION);
}
Exp memory maxLiquidationIncentive = Exp({mantissa: liquidationIncentiveMaxMantissa});
if (lessThanExp(maxLiquidationIncentive, newLiquidationIncentive)) {
return fail(Error.INVALID_LIQUIDATION_INCENTIVE, FailureInfo.SET_LIQUIDATION_INCENTIVE_VALIDATION);
}
uint oldLiquidationIncentiveMantissa = liquidationIncentiveMantissa;
liquidationIncentiveMantissa = newLiquidationIncentiveMantissa;
emit NewLiquidationIncentive(oldLiquidationIncentiveMantissa, newLiquidationIncentiveMantissa);
return uint(Error.NO_ERROR);
}
function _supportMarket(address pToken) external returns (uint) {
if (msg.sender != admin && msg.sender != factory) {
return fail(Error.UNAUTHORIZED, FailureInfo.SUPPORT_MARKET_OWNER_CHECK);
}
if (markets[pToken].isListed) {
return fail(Error.MARKET_ALREADY_LISTED, FailureInfo.SUPPORT_MARKET_EXISTS);
}
PTokenInterface(pToken).isPToken();
_addMarketInternal(pToken);
Market storage newMarket = markets[pToken];
newMarket.isListed = true;
emit MarketListed(pToken);
return uint(Error.NO_ERROR);
}
function _addMarketInternal(address pToken) internal {
require(markets[pToken].isListed == false, "market already added");
allMarkets.push(pToken);
}
function _setPauseGuardian(address newPauseGuardian) public returns (uint) {
if (msg.sender != admin) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_PAUSE_GUARDIAN_OWNER_CHECK);
}
address oldPauseGuardian = pauseGuardian;
pauseGuardian = newPauseGuardian;
emit NewPauseGuardian(oldPauseGuardian, pauseGuardian);
return uint(Error.NO_ERROR);
}
function _setMintPaused(address pToken, bool state) public returns (bool) {
require(markets[pToken].isListed, "cannot pause a market that is not listed");
require(msg.sender == pauseGuardian || msg.sender == admin, "only pause guardian and admin can pause");
require(msg.sender == admin || state == true, "only admin can unpause");
mintGuardianPaused[pToken] = state;
emit ActionPaused(pToken, "Mint", state);
return state;
}
function _setBorrowPaused(address pToken, bool state) public returns (bool) {
require(markets[pToken].isListed, "cannot pause a market that is not listed");
require(msg.sender == pauseGuardian || msg.sender == admin, "only pause guardian and admin can pause");
require(msg.sender == admin || state == true, "only admin can unpause");
borrowGuardianPaused[pToken] = state;
emit ActionPaused(pToken, "Borrow", state);
return state;
}
function _setTransferPaused(bool state) public returns (bool) {
require(msg.sender == pauseGuardian || msg.sender == admin, "only pause guardian and admin can pause");
require(msg.sender == admin || state == true, "only admin can unpause");
transferGuardianPaused = state;
emit ActionPaused("Transfer", state);
return state;
}
function _setSeizePaused(bool state) public returns (bool) {
require(msg.sender == pauseGuardian || msg.sender == admin, "only pause guardian and admin can pause");
require(msg.sender == admin || state == true, "only admin can unpause");
seizeGuardianPaused = state;
emit ActionPaused("Seize", state);
return state;
}
function _setFactoryContract(address _factory) external returns (uint) {
if (msg.sender != admin) {
return uint(Error.UNAUTHORIZED);
}
factory = _factory;
return uint(Error.NO_ERROR);
}
function _become(address payable unitroller) public {
require(msg.sender == Unitroller(unitroller).admin(), "only unitroller admin can change brains");
require(Unitroller(unitroller)._acceptImplementation() == 0, "change not authorized");
}
function setPieSpeedInternal(address pToken, uint pieSpeed) internal {
uint currentPieSpeed = pieSpeeds[pToken];
if (currentPieSpeed != 0) {
Exp memory borrowIndex = Exp({mantissa: PTokenInterface(pToken).borrowIndex()});
updatePieSupplyIndex(pToken);
updatePieBorrowIndex(pToken, borrowIndex);
} else if (pieSpeed != 0) {
Market storage market = markets[pToken];
require(market.isListed == true, "pie market is not listed");
if (pieSupplyState[pToken].index == 0) {
pieSupplyState[pToken] = PieMarketState({
index: pieInitialIndex,
block: safe32(getBlockNumber(), "block number exceeds 32 bits")
});
} else {
pieSupplyState[pToken].block = safe32(getBlockNumber(), "block number exceeds 32 bits");
}
if (pieBorrowState[pToken].index == 0) {
pieBorrowState[pToken] = PieMarketState({
index: pieInitialIndex,
block: safe32(getBlockNumber(), "block number exceeds 32 bits")
});
} else {
pieBorrowState[pToken].block = safe32(getBlockNumber(), "block number exceeds 32 bits");
}
}
if (currentPieSpeed != pieSpeed) {
pieSpeeds[pToken] = pieSpeed;
emit PieSpeedUpdated(pToken, pieSpeed);
}
}
function updatePieSupplyIndex(address pToken) internal {
PieMarketState storage supplyState = pieSupplyState[pToken];
uint supplySpeed = pieSpeeds[pToken];
uint blockNumber = getBlockNumber();
uint deltaBlocks = sub_(blockNumber, uint(supplyState.block));
if (deltaBlocks > 0 && supplySpeed > 0) {
uint supplyTokens = PTokenInterface(pToken).totalSupply();
uint pieAccrued = mul_(deltaBlocks, supplySpeed);
Double memory ratio = supplyTokens > 0 ? fraction(pieAccrued, supplyTokens) : Double({mantissa: 0});
Double memory index = add_(Double({mantissa: supplyState.index}), ratio);
pieSupplyState[pToken] = PieMarketState({
index: safe224(index.mantissa, "new index exceeds 224 bits"),
block: safe32(blockNumber, "block number exceeds 32 bits")
});
}
}
function updatePieBorrowIndex(address pToken, Exp memory marketBorrowIndex) internal {
PieMarketState storage borrowState = pieBorrowState[pToken];
uint borrowSpeed = pieSpeeds[pToken];
uint blockNumber = getBlockNumber();
uint deltaBlocks = sub_(blockNumber, uint(borrowState.block));
if (deltaBlocks > 0 && borrowSpeed > 0) {
uint borrowAmount = div_(PTokenInterface(pToken).totalBorrows(), marketBorrowIndex);
uint pieAccrued = mul_(deltaBlocks, borrowSpeed);
Double memory ratio = borrowAmount > 0 ? fraction(pieAccrued, borrowAmount) : Double({mantissa: 0});
Double memory index = add_(Double({mantissa: borrowState.index}), ratio);
pieBorrowState[pToken] = PieMarketState({
index: safe224(index.mantissa, "new index exceeds 224 bits"),
block: safe32(blockNumber, "block number exceeds 32 bits")
});
}
}
function distributeSupplierPie(address pToken, address supplier, bool distributeAll) internal {
PieMarketState storage supplyState = pieSupplyState[pToken];
Double memory supplyIndex = Double({mantissa: supplyState.index});
Double memory supplierIndex = Double({mantissa: pieSupplierIndex[pToken][supplier]});
pieSupplierIndex[pToken][supplier] = supplyIndex.mantissa;
if (supplierIndex.mantissa == 0 && supplyIndex.mantissa > 0) {
supplierIndex.mantissa = pieInitialIndex;
}
Double memory deltaIndex = sub_(supplyIndex, supplierIndex);
uint supplierTokens = PTokenInterface(pToken).balanceOf(supplier);
uint supplierDelta = mul_(supplierTokens, deltaIndex);
uint supplierAccrued = add_(pieAccrued[supplier], supplierDelta);
pieAccrued[supplier] = transferPie(supplier, supplierAccrued, distributeAll ? 0 : pieClaimThreshold);
emit DistributedSupplierPie(pToken, supplier, supplierDelta, supplyIndex.mantissa);
}
function distributeBorrowerPie(
address pToken,
address borrower,
Exp memory marketBorrowIndex,
bool distributeAll
) internal {
PieMarketState storage borrowState = pieBorrowState[pToken];
Double memory borrowIndex = Double({mantissa: borrowState.index});
Double memory borrowerIndex = Double({mantissa: pieBorrowerIndex[pToken][borrower]});
pieBorrowerIndex[pToken][borrower] = borrowIndex.mantissa;
if (borrowerIndex.mantissa > 0) {
Double memory deltaIndex = sub_(borrowIndex, borrowerIndex);
uint borrowerAmount = div_(PTokenInterface(pToken).borrowBalanceStored(borrower), marketBorrowIndex);
uint borrowerDelta = mul_(borrowerAmount, deltaIndex);
uint borrowerAccrued = add_(pieAccrued[borrower], borrowerDelta);
pieAccrued[borrower] = transferPie(borrower, borrowerAccrued, distributeAll ? 0 : pieClaimThreshold);
emit DistributedBorrowerPie(pToken, borrower, borrowerDelta, borrowIndex.mantissa);
}
}
function claimPie(address holder) public {
claimPie(holder, allMarkets);
}
function claimPie(address holder, address[] memory pTokens) public {
address[] memory holders = new address[](1);
holders[0] = holder;
claimPie(holders, pTokens, true, true);
}
function claimPie(address[] memory holders, address[] memory pTokens, bool borrowers, bool suppliers) public {
for (uint i = 0; i < pTokens.length; i++) {
address pToken = pTokens[i];
require(markets[pToken].isListed, "market must be listed");
if (borrowers == true) {
Exp memory borrowIndex = Exp({mantissa: PTokenInterface(pToken).borrowIndex()});
updatePieBorrowIndex(pToken, borrowIndex);
for (uint j = 0; j < holders.length; j++) {
distributeBorrowerPie(pToken, holders[j], borrowIndex, true);
}
}
if (suppliers == true) {
updatePieSupplyIndex(pToken);
for (uint j = 0; j < holders.length; j++) {
distributeSupplierPie(pToken, holders[j], true);
}
}
}
}
function transferPie(address user, uint userAccrued, uint threshold) internal returns (uint) {
if (userAccrued >= threshold && userAccrued > 0) {
address pie = getPieAddress();
uint pieRemaining = EIP20Interface(pie).balanceOf(address(this));
if (userAccrued <= pieRemaining) {
EIP20Interface(pie).transfer(user, userAccrued);
return 0;
}
}
return userAccrued;
}
function _setPieSpeed(address pToken, uint pieSpeed) public {
require(msg.sender == admin, "only admin can set pie speed");
setPieSpeedInternal(pToken, pieSpeed);
}
function getAllMarkets() public view returns (address[] memory) {
return allMarkets;
}
function getBlockNumber() public view virtual returns (uint) {
return block.number;
}
function getPieAddress() public view virtual returns (address) {
return pieAddress;
}
function getOracle() public view override returns (PriceOracle) {
return oracle;
}
}
文件 3 的 25:ControllerInterface.sol
pragma solidity ^0.7.6;
import "./PriceOracle.sol";
abstract contract ControllerInterface {
bool public constant isController = true;
function enterMarkets(address[] calldata pTokens) external virtual returns (uint[] memory);
function exitMarket(address pToken) external virtual returns (uint);
function mintAllowed(address pToken, address minter, uint mintAmount) external virtual returns (uint);
function redeemAllowed(address pToken, address redeemer, uint redeemTokens) external virtual returns (uint);
function redeemVerify(address pToken, address redeemer, uint redeemAmount, uint redeemTokens) external virtual;
function borrowAllowed(address pToken, address borrower, uint borrowAmount) external virtual returns (uint);
function repayBorrowAllowed(
address pToken,
address payer,
address borrower,
uint repayAmount) external virtual returns (uint);
function liquidateBorrowAllowed(
address pTokenBorrowed,
address pTokenCollateral,
address liquidator,
address borrower,
uint repayAmount) external virtual returns (uint);
function seizeAllowed(
address pTokenCollateral,
address pTokenBorrowed,
address liquidator,
address borrower,
uint seizeTokens) external virtual returns (uint);
function transferAllowed(address pToken, address src, address dst, uint transferTokens) external virtual returns (uint);
function liquidateCalculateSeizeTokens(
address pTokenBorrowed,
address pTokenCollateral,
uint repayAmount) external view virtual returns (uint, uint);
function getOracle() external view virtual returns (PriceOracle);
}
文件 4 的 25:ControllerStorage.sol
pragma solidity ^0.7.6;
import "./PriceOracle.sol";
contract UnitrollerAdminStorage {
address public admin;
address public pendingAdmin;
address public controllerImplementation;
address public pendingControllerImplementation;
}
contract ControllerStorage is UnitrollerAdminStorage {
PriceOracle public oracle;
uint public closeFactorMantissa;
uint public liquidationIncentiveMantissa;
uint public maxAssets;
mapping(address => address[]) public accountAssets;
struct Market {
bool isListed;
uint collateralFactorMantissa;
mapping(address => bool) accountMembership;
bool isPied;
}
mapping(address => Market) public markets;
address public pauseGuardian;
bool public _mintGuardianPaused;
bool public _borrowGuardianPaused;
bool public transferGuardianPaused;
bool public seizeGuardianPaused;
mapping(address => bool) public mintGuardianPaused;
mapping(address => bool) public borrowGuardianPaused;
struct PieMarketState {
uint224 index;
uint32 block;
}
address[] public allMarkets;
uint public pieRate;
address public pieAddress;
address public factory;
mapping(address => uint) public pieSpeeds;
mapping(address => PieMarketState) public pieSupplyState;
mapping(address => PieMarketState) public pieBorrowState;
mapping(address => mapping(address => uint)) public pieSupplierIndex;
mapping(address => mapping(address => uint)) public pieBorrowerIndex;
mapping(address => uint) public pieAccrued;
}
文件 5 的 25:EIP20Interface.sol
pragma solidity ^0.7.6;
interface EIP20Interface {
function name() external view returns (string memory);
function symbol() external view returns (string memory);
function decimals() external view returns (uint8);
function totalSupply() external view returns (uint256);
function balanceOf(address owner) external view returns (uint256);
function transfer(address dst, uint256 amount) external returns (bool);
function transferFrom(address src, address dst, uint256 amount) external returns (bool);
function approve(address spender, uint256 amount) external returns (bool);
function allowance(address owner, address spender) external view returns (uint256);
event Transfer(address indexed from, address indexed to, uint256 amount);
event Approval(address indexed owner, address indexed spender, uint256 amount);
}
文件 6 的 25:ErrorReporter.sol
pragma solidity ^0.7.6;
contract ControllerErrorReporter {
enum Error {
NO_ERROR,
UNAUTHORIZED,
CONTROLLER_MISMATCH,
INSUFFICIENT_SHORTFALL,
INSUFFICIENT_LIQUIDITY,
INVALID_CLOSE_FACTOR,
INVALID_COLLATERAL_FACTOR,
INVALID_LIQUIDATION_INCENTIVE,
MARKET_NOT_ENTERED,
MARKET_NOT_LISTED,
MARKET_ALREADY_LISTED,
MATH_ERROR,
NONZERO_BORROW_BALANCE,
PRICE_ERROR,
PRICE_UPDATE_ERROR,
REJECTION,
SNAPSHOT_ERROR,
TOO_MANY_ASSETS,
TOO_MUCH_REPAY
}
enum FailureInfo {
ACCEPT_ADMIN_PENDING_ADMIN_CHECK,
ACCEPT_PENDING_IMPLEMENTATION_ADDRESS_CHECK,
EXIT_MARKET_BALANCE_OWED,
EXIT_MARKET_REJECTION,
SET_CLOSE_FACTOR_OWNER_CHECK,
SET_CLOSE_FACTOR_VALIDATION,
SET_COLLATERAL_FACTOR_OWNER_CHECK,
SET_COLLATERAL_FACTOR_NO_EXISTS,
SET_COLLATERAL_FACTOR_VALIDATION,
SET_COLLATERAL_FACTOR_WITHOUT_PRICE,
SET_IMPLEMENTATION_OWNER_CHECK,
SET_LIQUIDATION_INCENTIVE_OWNER_CHECK,
SET_LIQUIDATION_INCENTIVE_VALIDATION,
SET_MAX_ASSETS_OWNER_CHECK,
SET_PAUSE_GUARDIAN_OWNER_CHECK,
SET_PENDING_ADMIN_OWNER_CHECK,
SET_PENDING_IMPLEMENTATION_OWNER_CHECK,
SET_PRICE_ORACLE_OWNER_CHECK,
SUPPORT_MARKET_EXISTS,
SUPPORT_MARKET_OWNER_CHECK
}
event Failure(uint error, uint info, uint detail);
function fail(Error err, FailureInfo info) internal returns (uint) {
emit Failure(uint(err), uint(info), 0);
return uint(err);
}
function failOpaque(Error err, FailureInfo info, uint opaqueError) internal returns (uint) {
emit Failure(uint(err), uint(info), opaqueError);
return uint(err);
}
}
contract TokenErrorReporter {
enum Error {
NO_ERROR,
UNAUTHORIZED,
BAD_INPUT,
CONTROLLER_REJECTION,
CONTROLLER_CALCULATION_ERROR,
INTEREST_RATE_MODEL_ERROR,
INVALID_ACCOUNT_PAIR,
INVALID_CLOSE_AMOUNT_REQUESTED,
INVALID_COLLATERAL_FACTOR,
MATH_ERROR,
MARKET_NOT_FRESH,
MARKET_NOT_LISTED,
TOKEN_INSUFFICIENT_ALLOWANCE,
TOKEN_INSUFFICIENT_BALANCE,
TOKEN_INSUFFICIENT_CASH,
TOKEN_TRANSFER_IN_FAILED,
TOKEN_TRANSFER_OUT_FAILED
}
enum FailureInfo {
ACCEPT_ADMIN_PENDING_ADMIN_CHECK,
ACCRUE_INTEREST_ACCUMULATED_INTEREST_CALCULATION_FAILED,
ACCRUE_INTEREST_BORROW_RATE_CALCULATION_FAILED,
ACCRUE_INTEREST_NEW_BORROW_INDEX_CALCULATION_FAILED,
ACCRUE_INTEREST_NEW_TOTAL_BORROWS_CALCULATION_FAILED,
ACCRUE_INTEREST_NEW_TOTAL_RESERVES_CALCULATION_FAILED,
ACCRUE_INTEREST_SIMPLE_INTEREST_FACTOR_CALCULATION_FAILED,
BORROW_ACCUMULATED_BALANCE_CALCULATION_FAILED,
BORROW_ACCRUE_INTEREST_FAILED,
BORROW_CASH_NOT_AVAILABLE,
BORROW_FRESHNESS_CHECK,
BORROW_NEW_TOTAL_BALANCE_CALCULATION_FAILED,
BORROW_NEW_ACCOUNT_BORROW_BALANCE_CALCULATION_FAILED,
BORROW_MARKET_NOT_LISTED,
BORROW_CONTROLLER_REJECTION,
LIQUIDATE_ACCRUE_BORROW_INTEREST_FAILED,
LIQUIDATE_ACCRUE_COLLATERAL_INTEREST_FAILED,
LIQUIDATE_COLLATERAL_FRESHNESS_CHECK,
LIQUIDATE_CONTROLLER_REJECTION,
LIQUIDATE_CONTROLLER_CALCULATE_AMOUNT_SEIZE_FAILED,
LIQUIDATE_CLOSE_AMOUNT_IS_UINT_MAX,
LIQUIDATE_CLOSE_AMOUNT_IS_ZERO,
LIQUIDATE_FRESHNESS_CHECK,
LIQUIDATE_LIQUIDATOR_IS_BORROWER,
LIQUIDATE_REPAY_BORROW_FRESH_FAILED,
LIQUIDATE_SEIZE_BALANCE_INCREMENT_FAILED,
LIQUIDATE_SEIZE_BALANCE_DECREMENT_FAILED,
LIQUIDATE_SEIZE_CONTROLLER_REJECTION,
LIQUIDATE_SEIZE_LIQUIDATOR_IS_BORROWER,
LIQUIDATE_SEIZE_TOO_MUCH,
MINT_ACCRUE_INTEREST_FAILED,
MINT_CONTROLLER_REJECTION,
MINT_EXCHANGE_CALCULATION_FAILED,
MINT_EXCHANGE_RATE_READ_FAILED,
MINT_FRESHNESS_CHECK,
MINT_NEW_ACCOUNT_BALANCE_CALCULATION_FAILED,
MINT_NEW_TOTAL_SUPPLY_CALCULATION_FAILED,
MINT_TRANSFER_IN_FAILED,
MINT_TRANSFER_IN_NOT_POSSIBLE,
REDEEM_ACCRUE_INTEREST_FAILED,
REDEEM_CONTROLLER_REJECTION,
REDEEM_EXCHANGE_TOKENS_CALCULATION_FAILED,
REDEEM_EXCHANGE_AMOUNT_CALCULATION_FAILED,
REDEEM_EXCHANGE_RATE_READ_FAILED,
REDEEM_FRESHNESS_CHECK,
REDEEM_NEW_ACCOUNT_BALANCE_CALCULATION_FAILED,
REDEEM_NEW_TOTAL_SUPPLY_CALCULATION_FAILED,
REDEEM_TRANSFER_OUT_NOT_POSSIBLE,
REDUCE_RESERVES_ACCRUE_INTEREST_FAILED,
REDUCE_RESERVES_ADMIN_CHECK,
REDUCE_RESERVES_CASH_NOT_AVAILABLE,
REDUCE_RESERVES_FRESH_CHECK,
REDUCE_RESERVES_VALIDATION,
REPAY_BEHALF_ACCRUE_INTEREST_FAILED,
REPAY_BORROW_ACCRUE_INTEREST_FAILED,
REPAY_BORROW_ACCUMULATED_BALANCE_CALCULATION_FAILED,
REPAY_BORROW_CONTROLLER_REJECTION,
REPAY_BORROW_FRESHNESS_CHECK,
REPAY_BORROW_NEW_ACCOUNT_BORROW_BALANCE_CALCULATION_FAILED,
REPAY_BORROW_NEW_TOTAL_BALANCE_CALCULATION_FAILED,
REPAY_BORROW_TRANSFER_IN_NOT_POSSIBLE,
SET_COLLATERAL_FACTOR_OWNER_CHECK,
SET_COLLATERAL_FACTOR_VALIDATION,
SET_CONTROLLER_OWNER_CHECK,
SET_INTEREST_RATE_MODEL_ACCRUE_INTEREST_FAILED,
SET_INTEREST_RATE_MODEL_FRESH_CHECK,
SET_INTEREST_RATE_MODEL_OWNER_CHECK,
SET_MAX_ASSETS_OWNER_CHECK,
SET_ORACLE_MARKET_NOT_LISTED,
SET_PENDING_ADMIN_OWNER_CHECK,
SET_RESERVE_FACTOR_ACCRUE_INTEREST_FAILED,
SET_RESERVE_FACTOR_ADMIN_CHECK,
SET_RESERVE_FACTOR_FRESH_CHECK,
SET_RESERVE_FACTOR_BOUNDS_CHECK,
TRANSFER_CONTROLLER_REJECTION,
TRANSFER_NOT_ALLOWED,
TRANSFER_NOT_ENOUGH,
TRANSFER_TOO_MUCH,
ADD_RESERVES_ACCRUE_INTEREST_FAILED,
ADD_RESERVES_FRESH_CHECK,
ADD_RESERVES_TRANSFER_IN_NOT_POSSIBLE,
SET_NEW_IMPLEMENTATION
}
event Failure(uint error, uint info, uint detail);
function fail(Error err, FailureInfo info) internal returns (uint) {
emit Failure(uint(err), uint(info), 0);
return uint(err);
}
function failOpaque(Error err, FailureInfo info, uint opaqueError) internal returns (uint) {
emit Failure(uint(err), uint(info), opaqueError);
return uint(err);
}
}
contract OracleErrorReporter {
enum Error {
NO_ERROR,
POOL_OR_COIN_EXIST,
UNAUTHORIZED,
UPDATE_PRICE
}
enum FailureInfo {
ADD_POOL_OR_COIN,
NO_PAIR,
NO_RESERVES,
PERIOD_NOT_ELAPSED,
SET_NEW_IMPLEMENTATION,
UPDATE_DATA
}
event Failure(uint error, uint info, uint detail);
function fail(Error err, FailureInfo info) internal returns (uint) {
emit Failure(uint(err), uint(info), 0);
return uint(err);
}
}
contract FactoryErrorReporter {
enum Error {
NO_ERROR,
INVALID_POOL,
MARKET_NOT_LISTED,
UNAUTHORIZED
}
enum FailureInfo {
ACCEPT_ADMIN_PENDING_ADMIN_CHECK,
CREATE_PETH_POOL,
CREATE_PPIE_POOL,
DEFICIENCY_LIQUIDITY_IN_POOL_OR_PAIR_IS_NOT_EXIST,
SET_MIN_LIQUIDITY_OWNER_CHECK,
SET_NEW_CONTROLLER,
SET_NEW_DECIMALS,
SET_NEW_EXCHANGE_RATE,
SET_NEW_IMPLEMENTATION,
SET_NEW_INTEREST_RATE_MODEL,
SET_NEW_ORACLE,
SET_NEW_RESERVE_FACTOR,
SET_PENDING_ADMIN_OWNER_CHECK,
SUPPORT_MARKET_BAD_RESULT
}
event Failure(uint error, uint info, uint detail);
function fail(Error err, FailureInfo info) internal returns (uint) {
emit Failure(uint(err), uint(info), 0);
return uint(err);
}
}
contract RegistryErrorReporter {
enum Error {
NO_ERROR,
UNAUTHORIZED
}
enum FailureInfo {
ACCEPT_ADMIN_PENDING_ADMIN_CHECK,
SET_NEW_IMPLEMENTATION,
SET_PENDING_ADMIN_OWNER_CHECK,
SET_NEW_FACTORY
}
event Failure(uint error, uint info, uint detail);
function fail(Error err, FailureInfo info) internal returns (uint) {
emit Failure(uint(err), uint(info), 0);
return uint(err);
}
}
文件 7 的 25:Exponential.sol
pragma solidity ^0.7.6;
import "./CarefulMath.sol";
contract Exponential is CarefulMath {
uint constant expScale = 1e18;
uint constant doubleScale = 1e36;
uint constant halfExpScale = expScale/2;
uint constant mantissaOne = expScale;
struct Exp {
uint mantissa;
}
struct Double {
uint mantissa;
}
function getExp(uint num, uint denom) pure internal returns (MathError, Exp memory) {
(MathError err0, uint scaledNumerator) = mulUInt(num, expScale);
if (err0 != MathError.NO_ERROR) {
return (err0, Exp({mantissa: 0}));
}
(MathError err1, uint rational) = divUInt(scaledNumerator, denom);
if (err1 != MathError.NO_ERROR) {
return (err1, Exp({mantissa: 0}));
}
return (MathError.NO_ERROR, Exp({mantissa: rational}));
}
function addExp(Exp memory a, Exp memory b) pure internal returns (MathError, Exp memory) {
(MathError error, uint result) = addUInt(a.mantissa, b.mantissa);
return (error, Exp({mantissa: result}));
}
function subExp(Exp memory a, Exp memory b) pure internal returns (MathError, Exp memory) {
(MathError error, uint result) = subUInt(a.mantissa, b.mantissa);
return (error, Exp({mantissa: result}));
}
function mulScalar(Exp memory a, uint scalar) pure internal returns (MathError, Exp memory) {
(MathError err0, uint scaledMantissa) = mulUInt(a.mantissa, scalar);
if (err0 != MathError.NO_ERROR) {
return (err0, Exp({mantissa: 0}));
}
return (MathError.NO_ERROR, Exp({mantissa: scaledMantissa}));
}
function mulScalarTruncate(Exp memory a, uint scalar) pure internal returns (MathError, uint) {
(MathError err, Exp memory product) = mulScalar(a, scalar);
if (err != MathError.NO_ERROR) {
return (err, 0);
}
return (MathError.NO_ERROR, truncate(product));
}
function mulScalarTruncateAddUInt(Exp memory a, uint scalar, uint addend) pure internal returns (MathError, uint) {
(MathError err, Exp memory product) = mulScalar(a, scalar);
if (err != MathError.NO_ERROR) {
return (err, 0);
}
return addUInt(truncate(product), addend);
}
function divScalar(Exp memory a, uint scalar) pure internal returns (MathError, Exp memory) {
(MathError err0, uint descaledMantissa) = divUInt(a.mantissa, scalar);
if (err0 != MathError.NO_ERROR) {
return (err0, Exp({mantissa: 0}));
}
return (MathError.NO_ERROR, Exp({mantissa: descaledMantissa}));
}
function divScalarByExp(uint scalar, Exp memory divisor) pure internal returns (MathError, Exp memory) {
(MathError err0, uint numerator) = mulUInt(expScale, scalar);
if (err0 != MathError.NO_ERROR) {
return (err0, Exp({mantissa: 0}));
}
return getExp(numerator, divisor.mantissa);
}
function divScalarByExpTruncate(uint scalar, Exp memory divisor) pure internal returns (MathError, uint) {
(MathError err, Exp memory fraction_) = divScalarByExp(scalar, divisor);
if (err != MathError.NO_ERROR) {
return (err, 0);
}
return (MathError.NO_ERROR, truncate(fraction_));
}
function mulExp(Exp memory a, Exp memory b) pure internal returns (MathError, Exp memory) {
(MathError err0, uint doubleScaledProduct) = mulUInt(a.mantissa, b.mantissa);
if (err0 != MathError.NO_ERROR) {
return (err0, Exp({mantissa: 0}));
}
(MathError err1, uint doubleScaledProductWithHalfScale) = addUInt(halfExpScale, doubleScaledProduct);
if (err1 != MathError.NO_ERROR) {
return (err1, Exp({mantissa: 0}));
}
(MathError err2, uint product) = divUInt(doubleScaledProductWithHalfScale, expScale);
assert(err2 == MathError.NO_ERROR);
return (MathError.NO_ERROR, Exp({mantissa: product}));
}
function mulExp(uint a, uint b) pure internal returns (MathError, Exp memory) {
return mulExp(Exp({mantissa: a}), Exp({mantissa: b}));
}
function mulExp3(Exp memory a, Exp memory b, Exp memory c) pure internal returns (MathError, Exp memory) {
(MathError err, Exp memory ab) = mulExp(a, b);
if (err != MathError.NO_ERROR) {
return (err, ab);
}
return mulExp(ab, c);
}
function divExp(Exp memory a, Exp memory b) pure internal returns (MathError, Exp memory) {
return getExp(a.mantissa, b.mantissa);
}
function truncate(Exp memory exp) pure internal returns (uint) {
return exp.mantissa / expScale;
}
function lessThanExp(Exp memory left, Exp memory right) pure internal returns (bool) {
return left.mantissa < right.mantissa;
}
function lessThanOrEqualExp(Exp memory left, Exp memory right) pure internal returns (bool) {
return left.mantissa <= right.mantissa;
}
function greaterThanExp(Exp memory left, Exp memory right) pure internal returns (bool) {
return left.mantissa > right.mantissa;
}
function isZeroExp(Exp memory value) pure internal returns (bool) {
return value.mantissa == 0;
}
function safe224(uint n, string memory errorMessage) pure internal returns (uint224) {
require(n < 2**224, errorMessage);
return uint224(n);
}
function safe32(uint n, string memory errorMessage) pure internal returns (uint32) {
require(n < 2**32, errorMessage);
return uint32(n);
}
function add_(Exp memory a, Exp memory b) pure internal returns (Exp memory) {
return Exp({mantissa: add_(a.mantissa, b.mantissa)});
}
function add_(Double memory a, Double memory b) pure internal returns (Double memory) {
return Double({mantissa: add_(a.mantissa, b.mantissa)});
}
function add_(uint a, uint b) pure internal returns (uint) {
return add_(a, b, "addition overflow");
}
function add_(uint a, uint b, string memory errorMessage) pure internal returns (uint) {
uint c = a + b;
require(c >= a, errorMessage);
return c;
}
function sub_(Exp memory a, Exp memory b) pure internal returns (Exp memory) {
return Exp({mantissa: sub_(a.mantissa, b.mantissa)});
}
function sub_(Double memory a, Double memory b) pure internal returns (Double memory) {
return Double({mantissa: sub_(a.mantissa, b.mantissa)});
}
function sub_(uint a, uint b) pure internal returns (uint) {
return sub_(a, b, "subtraction underflow");
}
function sub_(uint a, uint b, string memory errorMessage) pure internal returns (uint) {
require(b <= a, errorMessage);
return a - b;
}
function mul_(Exp memory a, Exp memory b) pure internal returns (Exp memory) {
return Exp({mantissa: mul_(a.mantissa, b.mantissa) / expScale});
}
function mul_(Exp memory a, uint b) pure internal returns (Exp memory) {
return Exp({mantissa: mul_(a.mantissa, b)});
}
function mul_(uint a, Exp memory b) pure internal returns (uint) {
return mul_(a, b.mantissa) / expScale;
}
function mul_(Double memory a, Double memory b) pure internal returns (Double memory) {
return Double({mantissa: mul_(a.mantissa, b.mantissa) / doubleScale});
}
function mul_(Double memory a, uint b) pure internal returns (Double memory) {
return Double({mantissa: mul_(a.mantissa, b)});
}
function mul_(uint a, Double memory b) pure internal returns (uint) {
return mul_(a, b.mantissa) / doubleScale;
}
function mul_(uint a, uint b) pure internal returns (uint) {
return mul_(a, b, "multiplication overflow");
}
function mul_(uint a, uint b, string memory errorMessage) pure internal returns (uint) {
if (a == 0 || b == 0) {
return 0;
}
uint c = a * b;
require(c / a == b, errorMessage);
return c;
}
function div_(Exp memory a, Exp memory b) pure internal returns (Exp memory) {
return Exp({mantissa: div_(mul_(a.mantissa, expScale), b.mantissa)});
}
function div_(Exp memory a, uint b) pure internal returns (Exp memory) {
return Exp({mantissa: div_(a.mantissa, b)});
}
function div_(uint a, Exp memory b) pure internal returns (uint) {
return div_(mul_(a, expScale), b.mantissa);
}
function div_(Double memory a, Double memory b) pure internal returns (Double memory) {
return Double({mantissa: div_(mul_(a.mantissa, doubleScale), b.mantissa)});
}
function div_(Double memory a, uint b) pure internal returns (Double memory) {
return Double({mantissa: div_(a.mantissa, b)});
}
function div_(uint a, Double memory b) pure internal returns (uint) {
return div_(mul_(a, doubleScale), b.mantissa);
}
function div_(uint a, uint b) pure internal returns (uint) {
return div_(a, b, "divide by zero");
}
function div_(uint a, uint b, string memory errorMessage) pure internal returns (uint) {
require(b > 0, errorMessage);
return a / b;
}
function fraction(uint a, uint b) pure internal returns (Double memory) {
return Double({mantissa: div_(mul_(a, doubleScale), b)});
}
}
文件 8 的 25:IPriceFeeds.sol
pragma solidity ^0.7.6;
import "./SafeMath.sol";
interface AggregatorInterface {
function latestAnswer() external view returns (int256);
}
library UQ112x112 {
uint224 constant Q112 = 2**112;
function encode(uint112 y) internal pure returns (uint224 z) {
z = uint224(y) * Q112;
}
function uqdiv(uint224 x, uint112 y) internal pure returns (uint224 z) {
z = x / uint224(y);
}
}
library FixedPoint {
struct uq112x112 {
uint224 _x;
}
struct uq144x112 {
uint _x;
}
uint8 private constant RESOLUTION = 112;
function mul(uq112x112 memory self, uint y) internal pure returns (uq144x112 memory) {
uint z;
require(y == 0 || (z = uint(self._x) * y) / y == uint(self._x), "FixedPoint: MULTIPLICATION_OVERFLOW");
return uq144x112(z);
}
function decode144(uq144x112 memory self) internal pure returns (uint144) {
return uint144(self._x >> RESOLUTION);
}
}
interface IUniswapV2Pair {
function token0() external view returns (address);
function token1() external view returns (address);
function getReserves() external view returns (uint112 reserve0, uint112 reserve1, uint32 blockTimestampLast);
function price0CumulativeLast() external view returns (uint);
function price1CumulativeLast() external view returns (uint);
}
interface IUniswapV2Factory {
function getPair(address tokenA, address tokenB) external view returns (address pair);
}
文件 9 的 25:InterestRateModel.sol
pragma solidity ^0.7.6;
abstract contract InterestRateModel {
bool public constant isInterestRateModel = true;
function getBorrowRate(uint cash, uint borrows, uint reserves) external view virtual returns (uint);
function getSupplyRate(uint cash, uint borrows, uint reserves, uint reserveFactorMantissa) external view virtual returns (uint);
}
文件 10 的 25:PErc20Delegator.sol
pragma solidity ^0.7.6;
import "./ProxyWithRegistry.sol";
import "./RegistryInterface.sol";
contract PErc20Delegator is ProxyWithRegistry {
constructor(
address underlying_,
address controller_,
address interestRateModel_,
uint initialExchangeRateMantissa_,
uint initialReserveFactorMantissa_,
string memory name_,
string memory symbol_,
uint8 decimals_,
address registry_
) {
_setRegistry(registry_);
delegateTo(_pTokenImplementation(), abi.encodeWithSignature("initialize(address,address,address,address,uint256,uint256,string,string,uint8)",
underlying_,
registry_,
controller_,
interestRateModel_,
initialExchangeRateMantissa_,
initialReserveFactorMantissa_,
name_,
symbol_,
decimals_));
}
function delegateTo(address callee, bytes memory data) internal returns (bytes memory) {
(bool success, bytes memory returnData) = callee.delegatecall(data);
assembly {
if eq(success, 0) {
revert(add(returnData, 0x20), returndatasize())
}
}
return returnData;
}
function delegateAndReturn() internal returns (bytes memory) {
(bool success, ) = _pTokenImplementation().delegatecall(msg.data);
assembly {
let free_mem_ptr := mload(0x40)
returndatacopy(free_mem_ptr, 0, returndatasize())
switch success
case 0 { revert(free_mem_ptr, returndatasize()) }
default { return(free_mem_ptr, returndatasize()) }
}
}
fallback() external {
delegateAndReturn();
}
}
文件 11 的 25:PEtherDelegator.sol
pragma solidity ^0.7.6;
import "./ProxyWithRegistry.sol";
import "./RegistryInterface.sol";
import "./ErrorReporter.sol";
contract PETHDelegator is ImplementationStorage, ProxyWithRegistry, TokenErrorReporter {
event NewImplementation(address oldImplementation, address newImplementation);
constructor(
address pETHImplementation_,
address controller_,
address interestRateModel_,
uint initialExchangeRateMantissa_,
uint initialReserveFactorMantissa_,
string memory name_,
string memory symbol_,
uint8 decimals_,
address registry_
) {
_setRegistry(registry_);
_setImplementation(pETHImplementation_);
delegateTo(implementation, abi.encodeWithSignature("initialize(address,address,address,uint256,uint256,string,string,uint8)",
registry_,
controller_,
interestRateModel_,
initialExchangeRateMantissa_,
initialReserveFactorMantissa_,
name_,
symbol_,
decimals_));
}
function delegateTo(address callee, bytes memory data) internal returns (bytes memory) {
(bool success, bytes memory returnData) = callee.delegatecall(data);
assembly {
if eq(success, 0) {
revert(add(returnData, 0x20), returndatasize())
}
}
return returnData;
}
function delegateAndReturn() private returns (bytes memory) {
(bool success, ) = implementation.delegatecall(msg.data);
assembly {
let free_mem_ptr := mload(0x40)
returndatacopy(free_mem_ptr, 0, returndatasize())
switch success
case 0 { revert(free_mem_ptr, returndatasize()) }
default { return(free_mem_ptr, returndatasize()) }
}
}
fallback() external payable {
delegateAndReturn();
}
receive() external payable {
delegateAndReturn();
}
function setImplementation(address newImplementation) external returns(uint) {
if (msg.sender != RegistryInterface(registry).admin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_NEW_IMPLEMENTATION);
}
address oldImplementation = implementation;
_setImplementation(newImplementation);
emit NewImplementation(oldImplementation, implementation);
return(uint(Error.NO_ERROR));
}
}
文件 12 的 25:PPIEDelegator.sol
pragma solidity ^0.7.6;
import "./ProxyWithRegistry.sol";
import "./RegistryInterface.sol";
import "./ErrorReporter.sol";
contract PPIEDelegator is ImplementationStorage, ProxyWithRegistry, TokenErrorReporter {
event NewImplementation(address oldImplementation, address newImplementation);
constructor(
address underlying_,
address pPIEImplementation_,
address controller_,
address interestRateModel_,
uint initialExchangeRateMantissa_,
uint initialReserveFactorMantissa_,
string memory name_,
string memory symbol_,
uint8 decimals_,
address registry_
) {
_setRegistry(registry_);
_setImplementation(pPIEImplementation_);
delegateTo(implementation, abi.encodeWithSignature("initialize(address,address,address,address,uint256,uint256,string,string,uint8)",
underlying_,
registry_,
controller_,
interestRateModel_,
initialExchangeRateMantissa_,
initialReserveFactorMantissa_,
name_,
symbol_,
decimals_));
}
function delegateTo(address callee, bytes memory data) internal returns (bytes memory) {
(bool success, bytes memory returnData) = callee.delegatecall(data);
assembly {
if eq(success, 0) {
revert(add(returnData, 0x20), returndatasize())
}
}
return returnData;
}
function delegateAndReturn() internal returns (bytes memory) {
(bool success, ) = implementation.delegatecall(msg.data);
assembly {
let free_mem_ptr := mload(0x40)
returndatacopy(free_mem_ptr, 0, returndatasize())
switch success
case 0 { revert(free_mem_ptr, returndatasize()) }
default { return(free_mem_ptr, returndatasize()) }
}
}
fallback() external {
delegateAndReturn();
}
function setImplementation(address newImplementation) external returns(uint) {
if (msg.sender != RegistryInterface(registry).admin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_NEW_IMPLEMENTATION);
}
address oldImplementation = implementation;
_setImplementation(newImplementation);
emit NewImplementation(oldImplementation, implementation);
return(uint(Error.NO_ERROR));
}
}
文件 13 的 25:PTokenFactory.sol
pragma solidity ^0.7.6;
import './PErc20Delegator.sol';
import './RegistryInterface.sol';
import './EIP20Interface.sol';
import './Strings.sol';
import "./IPriceFeeds.sol";
import "./ErrorReporter.sol";
import "./SafeMath.sol";
import "./PEtherDelegator.sol";
import "./PPIEDelegator.sol";
import "./Controller.sol";
import "./UniswapPriceOracle.sol";
contract PTokenFactory is FactoryErrorReporter {
using strings for *;
using SafeMath for uint;
UniswapPriceOracle public oracle;
uint public minUniswapLiquidity;
uint8 public decimals = 8;
address public controller;
address public interestRateModel;
uint256 public initialExchangeRateMantissa;
uint256 public initialReserveFactorMantissa;
event PTokenCreated(address newPToken);
RegistryInterface public registry;
constructor(
RegistryInterface registry_,
uint minUniswapLiquidity_,
address oracle_,
address _controller,
address _interestRateModel,
uint256 _initialExchangeRateMantissa,
uint256 _initialReserveFactorMantissa
) {
registry = registry_;
minUniswapLiquidity = minUniswapLiquidity_;
oracle = UniswapPriceOracle(oracle_);
controller = _controller;
interestRateModel = _interestRateModel;
initialExchangeRateMantissa = _initialExchangeRateMantissa;
initialReserveFactorMantissa = _initialReserveFactorMantissa;
}
function createPToken(address underlying_) external returns (uint) {
if (!checkPair(underlying_)) {
return fail(Error.INVALID_POOL, FailureInfo.DEFICIENCY_LIQUIDITY_IN_POOL_OR_PAIR_IS_NOT_EXIST);
}
(string memory name, string memory symbol) = _createPTokenNameAndSymbol(underlying_);
uint power = EIP20Interface(underlying_).decimals();
uint exchangeRateMantissa = calcExchangeRate(power);
PErc20Delegator newPToken = new PErc20Delegator(underlying_, controller, interestRateModel, exchangeRateMantissa, initialReserveFactorMantissa, name, symbol, decimals, address(registry));
uint256 result = Controller(controller)._supportMarket(address(newPToken));
if (result != 0) {
return fail(Error.MARKET_NOT_LISTED, FailureInfo.SUPPORT_MARKET_BAD_RESULT);
}
registry.addPToken(underlying_, address(newPToken));
emit PTokenCreated(address(newPToken));
oracle.update(underlying_);
return uint(Error.NO_ERROR);
}
function createPETH(address pETHImplementation_) external virtual returns (uint) {
if (msg.sender != getAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.CREATE_PETH_POOL);
}
string memory name = "DeFiPie ETH";
string memory symbol = "pETH";
uint power = 18;
uint exchangeRateMantissa = calcExchangeRate(power);
PETHDelegator newPETH = new PETHDelegator(pETHImplementation_, controller, interestRateModel, exchangeRateMantissa, initialReserveFactorMantissa, name, symbol, decimals, address(registry));
uint256 result = Controller(controller)._supportMarket(address(newPETH));
if (result != 0) {
return fail(Error.MARKET_NOT_LISTED, FailureInfo.SUPPORT_MARKET_BAD_RESULT);
}
registry.addPETH(address(newPETH));
emit PTokenCreated(address(newPETH));
return uint(Error.NO_ERROR);
}
function createPPIE(address underlying_, address pPIEImplementation_) external virtual returns (uint) {
if (msg.sender != getAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.CREATE_PPIE_POOL);
}
string memory name = "DeFiPie PIE";
string memory symbol = "pPIE";
uint power = EIP20Interface(underlying_).decimals();
uint exchangeRateMantissa = calcExchangeRate(power);
PPIEDelegator newPPIE = new PPIEDelegator(underlying_, pPIEImplementation_, controller, interestRateModel, exchangeRateMantissa, initialReserveFactorMantissa, name, symbol, decimals, address(registry));
uint256 result = Controller(controller)._supportMarket(address(newPPIE));
if (result != 0) {
return fail(Error.MARKET_NOT_LISTED, FailureInfo.SUPPORT_MARKET_BAD_RESULT);
}
registry.addPPIE(address(newPPIE));
emit PTokenCreated(address(newPPIE));
oracle.update(underlying_);
return uint(Error.NO_ERROR);
}
function checkPair(address asset) public view returns (bool) {
(address pair, uint112 ethEquivalentReserves) = oracle.searchPair(asset);
return bool(pair != address(0) && ethEquivalentReserves >= minUniswapLiquidity);
}
function setMinUniswapLiquidity(uint minUniswapLiquidity_) public returns (uint) {
if (msg.sender != getAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_MIN_LIQUIDITY_OWNER_CHECK);
}
minUniswapLiquidity = minUniswapLiquidity_;
return uint(Error.NO_ERROR);
}
function setOracle(address oracle_) public returns (uint) {
if (msg.sender != getAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_NEW_ORACLE);
}
oracle = UniswapPriceOracle(oracle_);
return uint(Error.NO_ERROR);
}
function setController(address newController) external returns (uint) {
if (msg.sender != getAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_NEW_CONTROLLER);
}
controller = newController;
return(uint(Error.NO_ERROR));
}
function setInterestRateModel(address newInterestRateModel) external returns (uint) {
if (msg.sender != getAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_NEW_INTEREST_RATE_MODEL);
}
interestRateModel = newInterestRateModel;
return(uint(Error.NO_ERROR));
}
function setInitialExchangeRateMantissa(uint _initialExchangeRateMantissa) external returns (uint) {
if (msg.sender != getAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_NEW_EXCHANGE_RATE);
}
initialExchangeRateMantissa = _initialExchangeRateMantissa;
return(uint(Error.NO_ERROR));
}
function setInitialReserveFactorMantissa(uint _initialReserveFactorMantissa) external returns (uint) {
if (msg.sender != getAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_NEW_RESERVE_FACTOR);
}
initialReserveFactorMantissa = _initialReserveFactorMantissa;
return(uint(Error.NO_ERROR));
}
function setPTokenDecimals(uint _decimals) external returns (uint) {
if (msg.sender != getAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_NEW_DECIMALS);
}
decimals = uint8(_decimals);
return(uint(Error.NO_ERROR));
}
function getAdmin() public view returns(address payable) {
return registry.admin();
}
function _createPTokenNameAndSymbol(address underlying) internal view returns (string memory, string memory) {
string memory name = ("DeFiPie ".toSlice().concat(EIP20Interface(underlying).name().toSlice()));
string memory symbol = ("p".toSlice().concat(EIP20Interface(underlying).symbol().toSlice()));
return (name, symbol);
}
function calcExchangeRate(uint power) internal view returns (uint) {
uint factor;
if (decimals >= power) {
factor = 10**(decimals - power);
return initialExchangeRateMantissa.div(factor);
} else {
factor = 10**(power - decimals);
return initialExchangeRateMantissa.mul(factor);
}
}
}
文件 14 的 25:PTokenInterfaces.sol
pragma solidity ^0.7.6;
import "./ControllerInterface.sol";
import "./InterestRateModel.sol";
import "./ProxyWithRegistry.sol";
contract PTokenStorage is ProxyWithRegistryStorage {
bool internal _notEntered;
string public name;
string public symbol;
uint8 public decimals;
uint internal constant borrowRateMaxMantissa = 0.0005e16;
uint internal constant reserveFactorMaxMantissa = 1e18;
ControllerInterface public controller;
InterestRateModel public interestRateModel;
uint internal initialExchangeRateMantissa;
uint public reserveFactorMantissa;
uint public accrualBlockNumber;
uint public borrowIndex;
uint public totalBorrows;
uint public totalReserves;
uint public totalSupply;
mapping (address => uint) internal accountTokens;
mapping (address => mapping (address => uint)) internal transferAllowances;
struct BorrowSnapshot {
uint principal;
uint interestIndex;
}
mapping(address => BorrowSnapshot) internal accountBorrows;
}
abstract contract PTokenInterface is PTokenStorage {
bool public constant isPToken = true;
event AccrueInterest(uint cashPrior, uint interestAccumulated, uint borrowIndex, uint totalBorrows, uint totalReserves);
event Mint(address minter, uint mintAmount, uint mintTokens);
event Redeem(address redeemer, uint redeemAmount, uint redeemTokens);
event Borrow(address borrower, uint borrowAmount, uint accountBorrows, uint totalBorrows);
event RepayBorrow(address payer, address borrower, uint repayAmount, uint accountBorrows, uint totalBorrows);
event LiquidateBorrow(address liquidator, address borrower, uint repayAmount, address pTokenCollateral, uint seizeTokens);
event NewController(ControllerInterface oldController, ControllerInterface newController);
event NewMarketInterestRateModel(InterestRateModel oldInterestRateModel, InterestRateModel newInterestRateModel);
event NewReserveFactor(uint oldReserveFactorMantissa, uint newReserveFactorMantissa);
event ReservesAdded(address benefactor, uint addAmount, uint newTotalReserves);
event ReservesReduced(address admin, uint reduceAmount, uint newTotalReserves);
event Transfer(address indexed from, address indexed to, uint amount);
event Approval(address indexed owner, address indexed spender, uint amount);
function transfer(address dst, uint amount) external virtual returns (bool);
function transferFrom(address src, address dst, uint amount) external virtual returns (bool);
function approve(address spender, uint amount) external virtual returns (bool);
function allowance(address owner, address spender) external view virtual returns (uint);
function balanceOf(address owner) external view virtual returns (uint);
function balanceOfUnderlying(address owner) external virtual returns (uint);
function getAccountSnapshot(address account) external view virtual returns (uint, uint, uint, uint);
function borrowRatePerBlock() external view virtual returns (uint);
function supplyRatePerBlock() external view virtual returns (uint);
function totalBorrowsCurrent() external virtual returns (uint);
function borrowBalanceCurrent(address account) external virtual returns (uint);
function borrowBalanceStored(address account) public view virtual returns (uint);
function exchangeRateCurrent() public virtual returns (uint);
function exchangeRateStored() public view virtual returns (uint);
function getCash() external view virtual returns (uint);
function accrueInterest() public virtual returns (uint);
function seize(address liquidator, address borrower, uint seizeTokens) external virtual returns (uint);
function _setController(ControllerInterface newController) public virtual returns (uint);
function _setReserveFactor(uint newReserveFactorMantissa) external virtual returns (uint);
function _reduceReserves(uint reduceAmount) external virtual returns (uint);
function _setInterestRateModel(InterestRateModel newInterestRateModel) public virtual returns (uint);
}
contract PErc20Storage {
address public underlying;
}
abstract contract PErc20Interface is PErc20Storage {
function mint(uint mintAmount) external virtual returns (uint);
function redeem(uint redeemTokens) external virtual returns (uint);
function redeemUnderlying(uint redeemAmount) external virtual returns (uint);
function borrow(uint borrowAmount) external virtual returns (uint);
function repayBorrow(uint repayAmount) external virtual returns (uint);
function repayBorrowBehalf(address borrower, uint repayAmount) external virtual returns (uint);
function liquidateBorrow(address borrower, uint repayAmount, PTokenInterface pTokenCollateral) external virtual returns (uint);
function _addReserves(uint addAmount) external virtual returns (uint);
}
contract PPIEStorage {
mapping (address => address) public delegates;
struct Checkpoint {
uint32 fromBlock;
uint96 votes;
}
mapping (address => mapping (uint32 => Checkpoint)) public checkpoints;
mapping (address => uint32) public numCheckpoints;
bytes32 public constant DOMAIN_TYPEHASH = keccak256("EIP712Domain(string name,uint256 chainId,address verifyingContract)");
bytes32 public constant DELEGATION_TYPEHASH = keccak256("Delegation(address delegatee,uint256 nonce,uint256 expiry)");
mapping (address => uint) public nonces;
}
abstract contract PPIEInterface is PPIEStorage {
event DelegateChanged(address indexed delegator, address indexed fromDelegate, address indexed toDelegate);
event DelegateVotesChanged(address indexed delegate, uint previousBalance, uint newBalance);
function delegate(address delegatee) external virtual;
function delegateBySig(address delegatee, uint nonce, uint expiry, uint8 v, bytes32 r, bytes32 s) external virtual;
function getCurrentVotes(address account) external view virtual returns (uint96);
function getPriorVotes(address account, uint blockNumber) external view virtual returns (uint96);
}
文件 15 的 25:PriceOracle.sol
pragma solidity ^0.7.6;
abstract contract PriceOracle {
bool public constant isPriceOracle = true;
event PriceUpdated(address asset, uint price);
function getUnderlyingPrice(address pToken) external view virtual returns (uint);
function updateUnderlyingPrice(address pToken) external virtual returns (uint);
}
文件 16 的 25:ProxyWithRegistry.sol
pragma solidity ^0.7.6;
import "./RegistryInterface.sol";
contract ProxyWithRegistryStorage {
address public registry;
}
abstract contract ProxyWithRegistryInterface is ProxyWithRegistryStorage {
function _setRegistry(address _registry) internal virtual;
function _pTokenImplementation() internal view virtual returns (address);
}
contract ProxyWithRegistry is ProxyWithRegistryInterface {
function _pTokenImplementation() internal view override returns (address) {
return RegistryInterface(registry).pTokenImplementation();
}
function _setRegistry(address _registry) internal override {
registry = _registry;
}
}
contract ImplementationStorage {
address public implementation;
function _setImplementation(address implementation_) internal {
implementation = implementation_;
}
}
文件 17 的 25:Registry.sol
pragma solidity ^0.7.6;
import "./PTokenInterfaces.sol";
import './RegistryStorage.sol';
import "./ErrorReporter.sol";
import "./Controller.sol";
import "./PTokenFactory.sol";
contract Registry is RegistryStorage, RegistryErrorReporter {
address public factory;
address public pTokenImplementation;
mapping (address => address) public pTokens;
address public pETH;
address public pPIE;
event NewPendingAdmin(address oldPendingAdmin, address newPendingAdmin);
event NewAdmin(address oldAdmin, address newAdmin);
event NewPTokenImplementation(address oldImplementation, address newImplementation);
event NewFactory(address oldFactory, address newFactory);
event RemovePToken(address pToken);
constructor() {}
function initialize(address _pTokenImplementation) public {
require(pTokenImplementation == address(0), "Registry may only be initialized once");
pTokenImplementation = _pTokenImplementation;
}
function setPTokenImplementation(address newImplementation) external returns (uint) {
if (msg.sender != admin) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_NEW_IMPLEMENTATION);
}
address oldImplementation = pTokenImplementation;
pTokenImplementation = newImplementation;
emit NewPTokenImplementation(oldImplementation, pTokenImplementation);
return(uint(Error.NO_ERROR));
}
function _setFactoryContract(address _factory) external returns (uint) {
if (msg.sender != admin) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_NEW_FACTORY);
}
address oldFactory = factory;
factory = _factory;
emit NewFactory(oldFactory, factory);
return uint(Error.NO_ERROR);
}
function addPToken(address underlying, address pToken) public returns (uint) {
require(msg.sender == admin || msg.sender == factory, "Only admin or factory can add PTokens");
PTokenInterface(pToken).isPToken();
require(pTokens[underlying] == address(0), "Token already added");
pTokens[underlying] = pToken;
return uint(Error.NO_ERROR);
}
function addPETH(address pETH_) public returns (uint) {
require(msg.sender == admin || msg.sender == factory, "Only admin or factory can add PETH");
PTokenInterface(pETH_).isPToken();
require(pETH == address(0), "ETH already added");
pETH = pETH_;
return uint(Error.NO_ERROR);
}
function addPPIE(address pPIE_) public returns (uint) {
require(msg.sender == admin || msg.sender == factory, "Only admin or factory can add PPIE");
PTokenInterface(pPIE_).isPToken();
require(pPIE == address(0), "PIE already added");
pPIE = pPIE_;
address underlying = PErc20Storage(pPIE).underlying();
pTokens[underlying] = pPIE;
return uint(Error.NO_ERROR);
}
function removePToken(address pToken) public returns (uint) {
require(msg.sender == admin, "Only admin can remove PTokens");
PTokenInterface(pToken).isPToken();
address underlying = PErc20Storage(pToken).underlying();
require(pTokens[underlying] != address(0), "Token not added");
delete pTokens[underlying];
emit RemovePToken(pToken);
return uint(Error.NO_ERROR);
}
}
文件 18 的 25:RegistryInterface.sol
pragma solidity ^0.7.6;
interface RegistryInterface {
function admin() external view returns (address payable);
function pTokenImplementation() external view returns (address);
function addPToken(address underlying, address pToken) external returns(uint);
function addPETH(address pETH_) external returns(uint);
function addPPIE(address pPIE_) external returns(uint);
}
文件 19 的 25:RegistryStorage.sol
pragma solidity ^0.7.6;
contract RegistryStorage {
address public implementation;
address public admin;
address public pendingAdmin;
}
文件 20 的 25:SafeMath.sol
pragma solidity ^0.7.6;
library SafeMath {
function add(uint256 a, uint256 b) internal pure returns (uint256) {
uint256 c = a + b;
require(c >= a, "SafeMath: addition overflow");
return c;
}
function add(uint256 a, uint256 b, string memory errorMessage) internal pure returns (uint256) {
uint256 c = a + b;
require(c >= a, errorMessage);
return c;
}
function sub(uint256 a, uint256 b) internal pure returns (uint256) {
return sub(a, b, "SafeMath: subtraction underflow");
}
function sub(uint256 a, uint256 b, string memory errorMessage) internal pure returns (uint256) {
require(b <= a, errorMessage);
uint256 c = a - b;
return c;
}
function mul(uint256 a, uint256 b) internal pure returns (uint256) {
if (a == 0) {
return 0;
}
uint256 c = a * b;
require(c / a == b, "SafeMath: multiplication overflow");
return c;
}
function mul(uint256 a, uint256 b, string memory errorMessage) internal pure returns (uint256) {
if (a == 0) {
return 0;
}
uint256 c = a * b;
require(c / a == b, errorMessage);
return c;
}
function div(uint256 a, uint256 b) internal pure returns (uint256) {
return div(a, b, "SafeMath: division by zero");
}
function div(uint256 a, uint256 b, string memory errorMessage) internal pure returns (uint256) {
require(b > 0, errorMessage);
uint256 c = a / b;
return c;
}
function mod(uint256 a, uint256 b) internal pure returns (uint256) {
return mod(a, b, "SafeMath: modulo by zero");
}
function mod(uint256 a, uint256 b, string memory errorMessage) internal pure returns (uint256) {
require(b != 0, errorMessage);
return a % b;
}
}
文件 21 的 25:Strings.sol
pragma solidity ^0.7.6;
library strings {
struct slice {
uint _len;
uint _ptr;
}
function memcpy(uint dest, uint src, uint leng) private pure {
for(; leng >= 32; leng -= 32) {
assembly {
mstore(dest, mload(src))
}
dest += 32;
src += 32;
}
uint mask = 256 ** (32 - leng) - 1;
assembly {
let srcpart := and(mload(src), not(mask))
let destpart := and(mload(dest), mask)
mstore(dest, or(destpart, srcpart))
}
}
function toSlice(string memory self) internal pure returns (slice memory) {
uint ptr;
assembly {
ptr := add(self, 0x20)
}
return slice(bytes(self).length, ptr);
}
function len(bytes32 self) internal pure returns (uint) {
uint ret;
if (self == 0)
return 0;
if (uint(self) & 0xffffffffffffffffffffffffffffffff == 0) {
ret += 16;
self = bytes32(uint(self) / 0x100000000000000000000000000000000);
}
if (uint(self) & 0xffffffffffffffff == 0) {
ret += 8;
self = bytes32(uint(self) / 0x10000000000000000);
}
if (uint(self) & 0xffffffff == 0) {
ret += 4;
self = bytes32(uint(self) / 0x100000000);
}
if (uint(self) & 0xffff == 0) {
ret += 2;
self = bytes32(uint(self) / 0x10000);
}
if (uint(self) & 0xff == 0) {
ret += 1;
}
return 32 - ret;
}
function toSliceB32(bytes32 self) internal pure returns (slice memory ret) {
assembly {
let ptr := mload(0x40)
mstore(0x40, add(ptr, 0x20))
mstore(ptr, self)
mstore(add(ret, 0x20), ptr)
}
ret._len = len(self);
}
function copy(slice memory self) internal pure returns (slice memory) {
return slice(self._len, self._ptr);
}
function toString(slice memory self) internal pure returns (string memory) {
string memory ret = new string(self._len);
uint retptr;
assembly { retptr := add(ret, 32) }
memcpy(retptr, self._ptr, self._len);
return ret;
}
function len(slice memory self) internal pure returns (uint l) {
uint ptr = self._ptr - 31;
uint end = ptr + self._len;
for (l = 0; ptr < end; l++) {
uint8 b;
assembly { b := and(mload(ptr), 0xFF) }
if (b < 0x80) {
ptr += 1;
} else if(b < 0xE0) {
ptr += 2;
} else if(b < 0xF0) {
ptr += 3;
} else if(b < 0xF8) {
ptr += 4;
} else if(b < 0xFC) {
ptr += 5;
} else {
ptr += 6;
}
}
}
function empty(slice memory self) internal pure returns (bool) {
return self._len == 0;
}
function compare(slice memory self, slice memory other) internal pure returns (int) {
uint shortest = self._len;
if (other._len < self._len)
shortest = other._len;
uint selfptr = self._ptr;
uint otherptr = other._ptr;
for (uint idx = 0; idx < shortest; idx += 32) {
uint a;
uint b;
assembly {
a := mload(selfptr)
b := mload(otherptr)
}
if (a != b) {
uint256 mask = uint256(-1);
if(shortest < 32) {
mask = ~(2 ** (8 * (32 - shortest + idx)) - 1);
}
uint256 diff = (a & mask) - (b & mask);
if (diff != 0)
return int(diff);
}
selfptr += 32;
otherptr += 32;
}
return int(self._len) - int(other._len);
}
function equals(slice memory self, slice memory other) internal pure returns (bool) {
return compare(self, other) == 0;
}
function nextRune(slice memory self, slice memory rune) internal pure returns (slice memory) {
rune._ptr = self._ptr;
if (self._len == 0) {
rune._len = 0;
return rune;
}
uint l;
uint b;
assembly { b := and(mload(sub(mload(add(self, 32)), 31)), 0xFF) }
if (b < 0x80) {
l = 1;
} else if(b < 0xE0) {
l = 2;
} else if(b < 0xF0) {
l = 3;
} else {
l = 4;
}
if (l > self._len) {
rune._len = self._len;
self._ptr += self._len;
self._len = 0;
return rune;
}
self._ptr += l;
self._len -= l;
rune._len = l;
return rune;
}
function nextRune(slice memory self) internal pure returns (slice memory ret) {
nextRune(self, ret);
}
function ord(slice memory self) internal pure returns (uint ret) {
if (self._len == 0) {
return 0;
}
uint word;
uint length;
uint divisor = 2 ** 248;
assembly { word:= mload(mload(add(self, 32))) }
uint b = word / divisor;
if (b < 0x80) {
ret = b;
length = 1;
} else if(b < 0xE0) {
ret = b & 0x1F;
length = 2;
} else if(b < 0xF0) {
ret = b & 0x0F;
length = 3;
} else {
ret = b & 0x07;
length = 4;
}
if (length > self._len) {
return 0;
}
for (uint i = 1; i < length; i++) {
divisor = divisor / 256;
b = (word / divisor) & 0xFF;
if (b & 0xC0 != 0x80) {
return 0;
}
ret = (ret * 64) | (b & 0x3F);
}
return ret;
}
function keccak(slice memory self) internal pure returns (bytes32 ret) {
assembly {
ret := keccak256(mload(add(self, 32)), mload(self))
}
}
function startsWith(slice memory self, slice memory needle) internal pure returns (bool) {
if (self._len < needle._len) {
return false;
}
if (self._ptr == needle._ptr) {
return true;
}
bool equal;
assembly {
let length := mload(needle)
let selfptr := mload(add(self, 0x20))
let needleptr := mload(add(needle, 0x20))
equal := eq(keccak256(selfptr, length), keccak256(needleptr, length))
}
return equal;
}
function beyond(slice memory self, slice memory needle) internal pure returns (slice memory) {
if (self._len < needle._len) {
return self;
}
bool equal = true;
if (self._ptr != needle._ptr) {
assembly {
let length := mload(needle)
let selfptr := mload(add(self, 0x20))
let needleptr := mload(add(needle, 0x20))
equal := eq(keccak256(selfptr, length), keccak256(needleptr, length))
}
}
if (equal) {
self._len -= needle._len;
self._ptr += needle._len;
}
return self;
}
function endsWith(slice memory self, slice memory needle) internal pure returns (bool) {
if (self._len < needle._len) {
return false;
}
uint selfptr = self._ptr + self._len - needle._len;
if (selfptr == needle._ptr) {
return true;
}
bool equal;
assembly {
let length := mload(needle)
let needleptr := mload(add(needle, 0x20))
equal := eq(keccak256(selfptr, length), keccak256(needleptr, length))
}
return equal;
}
function until(slice memory self, slice memory needle) internal pure returns (slice memory) {
if (self._len < needle._len) {
return self;
}
uint selfptr = self._ptr + self._len - needle._len;
bool equal = true;
if (selfptr != needle._ptr) {
assembly {
let length := mload(needle)
let needleptr := mload(add(needle, 0x20))
equal := eq(keccak256(selfptr, length), keccak256(needleptr, length))
}
}
if (equal) {
self._len -= needle._len;
}
return self;
}
function findPtr(uint selflen, uint selfptr, uint needlelen, uint needleptr) private pure returns (uint) {
uint ptr = selfptr;
uint idx;
if (needlelen <= selflen) {
if (needlelen <= 32) {
bytes32 mask = bytes32(~(2 ** (8 * (32 - needlelen)) - 1));
bytes32 needledata;
assembly { needledata := and(mload(needleptr), mask) }
uint end = selfptr + selflen - needlelen;
bytes32 ptrdata;
assembly { ptrdata := and(mload(ptr), mask) }
while (ptrdata != needledata) {
if (ptr >= end)
return selfptr + selflen;
ptr++;
assembly { ptrdata := and(mload(ptr), mask) }
}
return ptr;
} else {
bytes32 hash;
assembly { hash := keccak256(needleptr, needlelen) }
for (idx = 0; idx <= selflen - needlelen; idx++) {
bytes32 testHash;
assembly { testHash := keccak256(ptr, needlelen) }
if (hash == testHash)
return ptr;
ptr += 1;
}
}
}
return selfptr + selflen;
}
function rfindPtr(uint selflen, uint selfptr, uint needlelen, uint needleptr) private pure returns (uint) {
uint ptr;
if (needlelen <= selflen) {
if (needlelen <= 32) {
bytes32 mask = bytes32(~(2 ** (8 * (32 - needlelen)) - 1));
bytes32 needledata;
assembly { needledata := and(mload(needleptr), mask) }
ptr = selfptr + selflen - needlelen;
bytes32 ptrdata;
assembly { ptrdata := and(mload(ptr), mask) }
while (ptrdata != needledata) {
if (ptr <= selfptr)
return selfptr;
ptr--;
assembly { ptrdata := and(mload(ptr), mask) }
}
return ptr + needlelen;
} else {
bytes32 hash;
assembly { hash := keccak256(needleptr, needlelen) }
ptr = selfptr + (selflen - needlelen);
while (ptr >= selfptr) {
bytes32 testHash;
assembly { testHash := keccak256(ptr, needlelen) }
if (hash == testHash)
return ptr + needlelen;
ptr -= 1;
}
}
}
return selfptr;
}
function find(slice memory self, slice memory needle) internal pure returns (slice memory) {
uint ptr = findPtr(self._len, self._ptr, needle._len, needle._ptr);
self._len -= ptr - self._ptr;
self._ptr = ptr;
return self;
}
function rfind(slice memory self, slice memory needle) internal pure returns (slice memory) {
uint ptr = rfindPtr(self._len, self._ptr, needle._len, needle._ptr);
self._len = ptr - self._ptr;
return self;
}
function split(slice memory self, slice memory needle, slice memory token) internal pure returns (slice memory) {
uint ptr = findPtr(self._len, self._ptr, needle._len, needle._ptr);
token._ptr = self._ptr;
token._len = ptr - self._ptr;
if (ptr == self._ptr + self._len) {
self._len = 0;
} else {
self._len -= token._len + needle._len;
self._ptr = ptr + needle._len;
}
return token;
}
function split(slice memory self, slice memory needle) internal pure returns (slice memory token) {
split(self, needle, token);
}
function rsplit(slice memory self, slice memory needle, slice memory token) internal pure returns (slice memory) {
uint ptr = rfindPtr(self._len, self._ptr, needle._len, needle._ptr);
token._ptr = ptr;
token._len = self._len - (ptr - self._ptr);
if (ptr == self._ptr) {
self._len = 0;
} else {
self._len -= token._len + needle._len;
}
return token;
}
function rsplit(slice memory self, slice memory needle) internal pure returns (slice memory token) {
rsplit(self, needle, token);
}
function count(slice memory self, slice memory needle) internal pure returns (uint cnt) {
uint ptr = findPtr(self._len, self._ptr, needle._len, needle._ptr) + needle._len;
while (ptr <= self._ptr + self._len) {
cnt++;
ptr = findPtr(self._len - (ptr - self._ptr), ptr, needle._len, needle._ptr) + needle._len;
}
}
function contains(slice memory self, slice memory needle) internal pure returns (bool) {
return rfindPtr(self._len, self._ptr, needle._len, needle._ptr) != self._ptr;
}
function concat(slice memory self, slice memory other) internal pure returns (string memory) {
string memory ret = new string(self._len + other._len);
uint retptr;
assembly { retptr := add(ret, 32) }
memcpy(retptr, self._ptr, self._len);
memcpy(retptr + self._len, other._ptr, other._len);
return ret;
}
function join(slice memory self, slice[] memory parts) internal pure returns (string memory) {
if (parts.length == 0)
return "";
uint length = self._len * (parts.length - 1);
for(uint i = 0; i < parts.length; i++)
length += parts[i]._len;
string memory ret = new string(length);
uint retptr;
assembly { retptr := add(ret, 32) }
for(uint i = 0; i < parts.length; i++) {
memcpy(retptr, parts[i]._ptr, parts[i]._len);
retptr += parts[i]._len;
if (i < parts.length - 1) {
memcpy(retptr, self._ptr, self._len);
retptr += self._len;
}
}
return ret;
}
}
文件 22 的 25:UniswapPriceOracle.sol
pragma solidity ^0.7.6;
pragma abicoder v2;
import "./PriceOracle.sol";
import "./ErrorReporter.sol";
import "./PTokenInterfaces.sol";
import "./SafeMath.sol";
import "./UniswapPriceOracleStorage.sol";
import "./EIP20Interface.sol";
import "./Controller.sol";
import "./PTokenFactory.sol";
contract UniswapPriceOracle is UniswapPriceOracleStorageV1, PriceOracle, OracleErrorReporter {
using FixedPoint for *;
using SafeMath for uint;
event PoolAdded(uint id, address poolFactory);
event PoolRemoved(uint id, address poolFactory);
event PoolUpdated(uint id, address poolFactory);
event StableCoinAdded(uint id, address coin);
event StableCoinRemoved(uint id, address coin);
event StableCoinUpdated(uint id, address coin);
event AssetPairUpdated(address asset, address pair);
constructor() {}
function initialize(
address poolFactory_,
address WETHToken_,
address ETHUSDPriceFeed_
)
public
{
require(
WETHToken == address(0) &&
ETHUSDPriceFeed == address(0)
, "Oracle: may only be initialized once"
);
WETHToken = WETHToken_;
ETHUSDPriceFeed = ETHUSDPriceFeed_;
require(
poolFactory_ != address(0)
, 'Oracle: invalid address for factory'
);
poolFactories.push(poolFactory_);
emit PoolAdded(0, poolFactory_);
}
function updateUnderlyingPrice(address pToken) public override returns (uint) {
if (pToken == Registry(registry).pETH()) {
return uint(Error.NO_ERROR);
}
address asset = PErc20Interface(pToken).underlying();
return update(asset);
}
function getPriceInUSD(address asset) public view virtual returns (uint) {
uint ETHUSDPrice = uint(AggregatorInterface(ETHUSDPriceFeed).latestAnswer());
uint AssetETHCourse = getCourseInETH(asset);
return ETHUSDPrice.mul(AssetETHCourse).div(1e8);
}
function getCourseInETH(address asset) public view returns (uint) {
if (asset == Registry(registry).pETH()) {
return 1e18;
}
return averagePrices[asset];
}
function update(address asset) public returns (uint) {
uint112 reserve0;
uint112 reserve1;
uint32 blockTimeStamp;
address pair;
if (isNewAsset(asset)) {
if (assetPair[asset] == address(0)) {
(pair, ) = searchPair(asset);
} else {
pair = assetPair[asset];
}
if (pair != address(0)) {
assetPair[asset] = pair;
(reserve0, reserve1, blockTimeStamp) = getReservesFromPair(asset);
if (reserve1 < minReserveLiquidity) {
return fail(Error.UPDATE_PRICE, FailureInfo.NO_RESERVES);
}
cumulativePrices[pair][asset].priceAverage = FixedPoint.uq112x112(uqdiv(encode(reserve1), reserve0));
} else {
return fail(Error.UPDATE_PRICE, FailureInfo.NO_PAIR);
}
} else {
(, , blockTimeStamp) = getReservesFromPair(asset);
if (reserve1 < minReserveLiquidity) {
cumulativePrices[assetPair[asset]][asset].priceAverage._x = 0;
cumulativePrices[assetPair[asset]][asset].priceCumulativePrevious = 0;
cumulativePrices[assetPair[asset]][asset].blockTimeStampPrevious = 0;
return fail(Error.UPDATE_PRICE, FailureInfo.NO_RESERVES);
}
if (!isPeriodElapsed(asset)) {
return fail(Error.UPDATE_PRICE, FailureInfo.PERIOD_NOT_ELAPSED);
}
pair = assetPair[asset];
uint32 timeElapsed = blockTimeStamp - cumulativePrices[pair][asset].blockTimeStampPrevious;
if (asset == IUniswapV2Pair(pair).token0()) {
cumulativePrices[pair][asset].priceAverage = FixedPoint.uq112x112(uint224((IUniswapV2Pair(pair).price0CumulativeLast() - cumulativePrices[pair][asset].priceCumulativePrevious) / timeElapsed));
} else {
cumulativePrices[pair][asset].priceAverage = FixedPoint.uq112x112(uint224((IUniswapV2Pair(pair).price1CumulativeLast() - cumulativePrices[pair][asset].priceCumulativePrevious) / timeElapsed));
}
}
cumulativePrices[pair][asset].blockTimeStampPrevious = blockTimeStamp;
if (asset == IUniswapV2Pair(pair).token0()) {
cumulativePrices[pair][asset].priceCumulativePrevious = IUniswapV2Pair(pair).price0CumulativeLast();
} else {
cumulativePrices[pair][asset].priceCumulativePrevious = IUniswapV2Pair(pair).price1CumulativeLast();
}
averagePrices[asset] = calcCourseInETH(asset);
emit PriceUpdated(asset, getCourseInETH(asset));
return uint(Error.NO_ERROR);
}
function checkAndUpdateAllNewAssets() public {
PTokenFactory factory = PTokenFactory(Registry(registry).factory());
Controller controller = Controller(factory.controller());
address[] memory allMarkets = Controller(controller).getAllMarkets();
updateNewAssets(allMarkets);
}
function updateNewAssets(address[] memory pTokens) public {
address asset;
for(uint i = 0; i < pTokens.length; i++) {
if (pTokens[i] == Registry(registry).pETH()) {
continue;
}
asset = PErc20Interface(pTokens[i]).underlying();
if (isNewAsset(asset)) {
update(asset);
}
}
}
function getUnderlyingPrice(address pToken) public view override virtual returns (uint) {
if (pToken == Registry(registry).pETH()) {
return getPriceInUSD(Registry(registry).pETH());
}
address asset = PErc20Interface(pToken).underlying();
uint price = getPriceInUSD(asset);
uint decimals = EIP20Interface(asset).decimals();
return price.mul(10 ** (36 - decimals)).div(1e18);
}
function isNewAsset(address asset) public view returns (bool) {
return bool(cumulativePrices[assetPair[asset]][asset].blockTimeStampPrevious == 0);
}
function getPoolPair(address asset, uint poolId) public view returns (address) {
IUniswapV2Factory factory = IUniswapV2Factory(poolFactories[poolId]);
return factory.getPair(WETHToken, asset);
}
function getPoolPairWithStableCoin(address asset, uint poolId, uint stableCoinId) public view returns (address) {
IUniswapV2Factory factory = IUniswapV2Factory(poolFactories[poolId]);
return factory.getPair(stableCoins[stableCoinId], asset);
}
function getReservesFromPair(address asset) public view returns (uint112, uint112, uint32) {
uint112 assetReserve;
uint112 ethOrCoinReserves;
uint32 blockTimeStamp;
IUniswapV2Pair pair = IUniswapV2Pair(assetPair[asset]);
address token0 = pair.token0();
if (token0 == asset) {
(assetReserve, ethOrCoinReserves, blockTimeStamp) = pair.getReserves();
} else {
(ethOrCoinReserves, assetReserve, blockTimeStamp) = pair.getReserves();
}
return (assetReserve, ethOrCoinReserves, blockTimeStamp);
}
function isPeriodElapsed(address asset) public view returns (bool) {
IUniswapV2Pair pair = IUniswapV2Pair(assetPair[asset]);
( , , uint32 blockTimeStamp) = pair.getReserves();
uint timeElapsed = uint(blockTimeStamp).sub(uint(cumulativePrices[assetPair[asset]][asset].blockTimeStampPrevious));
return bool(timeElapsed > period);
}
function calcCourseInETH(address asset) public view returns (uint) {
if (asset == Registry(registry).pETH()) {
return 1e18;
}
uint power = EIP20Interface(asset).decimals();
uint amountIn = 10**power;
return getETHAmount(asset, amountIn);
}
function getETHAmount(address asset, uint amountIn) public view returns (uint) {
address pair = assetPair[asset];
address token0 = IUniswapV2Pair(pair).token0();
address token1 = IUniswapV2Pair(pair).token1();
uint power;
uint result = cumulativePrices[pair][asset].priceAverage.mul(amountIn).decode144();
if (token0 == WETHToken || token1 == WETHToken) {
return result;
} else {
if (token0 == asset) {
power = EIP20Interface(token1).decimals();
return result.mul(getCourseInETH(token1)).div(10**power);
} else {
power = EIP20Interface(token0).decimals();
return result.mul(getCourseInETH(token0)).div(10**power);
}
}
}
function searchPair(address asset) public view returns (address, uint112) {
address pair;
uint112 maxReserves;
IUniswapV2Pair tempPair;
uint112 ETHReserves;
for (uint i = 0; i < poolFactories.length; i++) {
tempPair = IUniswapV2Pair(getPoolPair(asset, i));
if (address(tempPair) != address(0)) {
if (tempPair.token0() == asset) {
(, ETHReserves, ) = tempPair.getReserves();
} else {
(ETHReserves, , ) = tempPair.getReserves();
}
if (ETHReserves > maxReserves) {
maxReserves = ETHReserves;
pair = address(tempPair);
}
}
for (uint j = 0; j < stableCoins.length; j++) {
tempPair = IUniswapV2Pair(getPoolPairWithStableCoin(asset, i, j));
if (address(tempPair) != address(0)) {
uint112 stableCoinReserve;
uint power;
address token0 = tempPair.token0();
address token1 = tempPair.token1();
if (token0 == asset) {
(, stableCoinReserve,) = tempPair.getReserves();
power = EIP20Interface(token1).decimals();
ETHReserves = uint112(getCourseInETH(token1) * stableCoinReserve / (10**power));
} else {
(stableCoinReserve, , ) = tempPair.getReserves();
power = EIP20Interface(token0).decimals();
ETHReserves = uint112(getCourseInETH(token0) * stableCoinReserve / (10**power));
}
if (ETHReserves > maxReserves) {
maxReserves = ETHReserves;
pair = address(tempPair);
}
}
}
}
return (pair, maxReserves);
}
function _setNewAddresses(address WETHToken_, address ETHUSDPriceFeed_) external returns (uint) {
if (msg.sender != getMyAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.UPDATE_DATA);
}
WETHToken = WETHToken_;
ETHUSDPriceFeed = ETHUSDPriceFeed_;
return uint(Error.NO_ERROR);
}
function _setMinReserveLiquidity(uint minReserveLiquidity_) public returns (uint) {
if (msg.sender != getMyAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.UPDATE_DATA);
}
minReserveLiquidity = minReserveLiquidity_;
return uint(Error.NO_ERROR);
}
function _setPeriod(uint period_) public returns (uint) {
if (msg.sender != getMyAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.UPDATE_DATA);
}
period = period_;
return uint(Error.NO_ERROR);
}
function _addPool(address poolFactory_) public returns (uint) {
if (msg.sender != getMyAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.ADD_POOL_OR_COIN);
}
require(
poolFactory_ != address(0)
, 'Oracle: invalid address for factory'
);
for (uint i = 0; i < poolFactories.length; i++) {
if (poolFactories[i] == poolFactory_) {
return fail(Error.POOL_OR_COIN_EXIST, FailureInfo.ADD_POOL_OR_COIN);
}
}
poolFactories.push(poolFactory_);
uint poolId = poolFactories.length - 1;
emit PoolAdded(poolId, poolFactory_);
return uint(Error.NO_ERROR);
}
function _removePool(uint poolId) public returns (uint) {
if (msg.sender != getMyAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.UPDATE_DATA);
}
require(
poolFactories.length > 1
, 'Oracle: must have one pool'
);
uint lastId = poolFactories.length - 1;
address factory = poolFactories[lastId];
poolFactories.pop();
emit PoolRemoved(lastId, factory);
if (lastId != poolId) {
poolFactories[poolId] = factory;
emit PoolUpdated(poolId, factory);
}
return uint(Error.NO_ERROR);
}
function _updatePool(uint poolId, address poolFactory_) public returns (uint) {
if (msg.sender != getMyAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.UPDATE_DATA);
}
require(
poolFactory_ != address(0)
, 'Oracle: invalid address for factory'
);
for (uint i = 0; i < poolFactories.length; i++) {
if (poolFactories[i] == poolFactory_) {
return fail(Error.POOL_OR_COIN_EXIST, FailureInfo.UPDATE_DATA);
}
}
poolFactories[poolId] = poolFactory_;
emit PoolUpdated(poolId, poolFactory_);
return uint(Error.NO_ERROR);
}
function _addStableCoin(address stableCoin_) public returns (uint) {
if (msg.sender != getMyAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.ADD_POOL_OR_COIN);
}
require(
stableCoin_ != address(0)
, 'Oracle: invalid address for stable coin'
);
for (uint i = 0; i < stableCoins.length; i++) {
if (stableCoins[i] == stableCoin_) {
return fail(Error.POOL_OR_COIN_EXIST, FailureInfo.ADD_POOL_OR_COIN);
}
}
stableCoins.push(stableCoin_);
emit StableCoinAdded(stableCoins.length - 1, stableCoin_);
return uint(Error.NO_ERROR);
}
function _removeStableCoin(uint coinId) public returns (uint) {
if (msg.sender != getMyAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.UPDATE_DATA);
}
require(
stableCoins.length > 0
, 'Oracle: stable coins are empty'
);
uint lastId = stableCoins.length - 1;
address stableCoin = stableCoins[lastId];
stableCoins.pop();
emit StableCoinRemoved(lastId, stableCoin);
if (lastId != coinId) {
stableCoins[coinId] = stableCoin;
emit StableCoinUpdated(coinId, stableCoin);
}
return uint(Error.NO_ERROR);
}
function _updateStableCoin(uint coinId, address stableCoin_) public returns (uint) {
if (msg.sender != getMyAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.UPDATE_DATA);
}
require(
stableCoin_ != address(0)
, 'Oracle: invalid address for stable coin'
);
for (uint i = 0; i < stableCoins.length; i++) {
if (stableCoins[i] == stableCoin_) {
return fail(Error.POOL_OR_COIN_EXIST, FailureInfo.UPDATE_DATA);
}
}
stableCoins[coinId] = stableCoin_;
emit StableCoinUpdated(coinId, stableCoin_);
return uint(Error.NO_ERROR);
}
function _updateAssetPair(address asset, address pair) public returns (uint) {
if (msg.sender != getMyAdmin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.UPDATE_DATA);
}
require(
pair != address(0)
, 'Oracle: invalid address for pair'
);
cumulativePrices[assetPair[asset]][asset].priceAverage._x = 0;
cumulativePrices[assetPair[asset]][asset].priceCumulativePrevious = 0;
cumulativePrices[assetPair[asset]][asset].blockTimeStampPrevious = 0;
assetPair[asset] = pair;
emit AssetPairUpdated(asset, pair);
return update(asset);
}
function getAllPoolFactories() public view returns (address[] memory) {
return poolFactories;
}
function getAllStableCoins() public view returns (address[] memory) {
return stableCoins;
}
function getMyAdmin() public view returns (address) {
return Registry(registry).admin();
}
function encode(uint112 y) internal view returns (uint224 z) {
z = uint224(y) * uint224(Q112);
}
function uqdiv(uint224 x, uint112 y) internal pure returns (uint224 z) {
z = x / uint224(y);
}
}
文件 23 的 25:UniswapPriceOracleProxy.sol
pragma solidity ^0.7.6;
pragma abicoder v2;
import "./ErrorReporter.sol";
import "./UniswapPriceOracleStorage.sol";
import "./RegistryInterface.sol";
contract UniswapPriceOracleProxy is UniswapPriceOracleProxyStorage, OracleErrorReporter {
event NewImplementation(address oldImplementation, address newImplementation);
constructor(
address implementation_,
address registry_,
address uniswapFactory_,
address WETHToken_,
address ETHUSDPriceFeed_
) {
implementation = implementation_;
registry = registry_;
delegateTo(implementation, abi.encodeWithSignature("initialize(address,address,address)",
uniswapFactory_,
WETHToken_,
ETHUSDPriceFeed_));
}
function setOracleImplementation(address newImplementation) external returns(uint256) {
if (msg.sender != RegistryInterface(registry).admin()) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_NEW_IMPLEMENTATION);
}
address oldImplementation = implementation;
implementation = newImplementation;
emit NewImplementation(oldImplementation, implementation);
return(uint(Error.NO_ERROR));
}
function delegateTo(address callee, bytes memory data) internal returns (bytes memory) {
(bool success, bytes memory returnData) = callee.delegatecall(data);
assembly {
if eq(success, 0) {
revert(add(returnData, 0x20), returndatasize())
}
}
return returnData;
}
function delegateAndReturn() private returns (bytes memory) {
(bool success, ) = implementation.delegatecall(msg.data);
assembly {
let free_mem_ptr := mload(0x40)
returndatacopy(free_mem_ptr, 0, returndatasize())
switch success
case 0 { revert(free_mem_ptr, returndatasize()) }
default { return(free_mem_ptr, returndatasize()) }
}
}
fallback() external {
delegateAndReturn();
}
}
文件 24 的 25:UniswapPriceOracleStorage.sol
pragma solidity ^0.7.6;
pragma abicoder v2;
import './Registry.sol';
import "./IPriceFeeds.sol";
contract UniswapPriceOracleProxyStorage {
address public implementation;
address public registry;
uint public Q112 = 2**112;
uint public period = 10 minutes;
}
contract UniswapPriceOracleStorageV1 is UniswapPriceOracleProxyStorage {
uint public minReserveLiquidity;
address public WETHToken;
address public ETHUSDPriceFeed;
struct PoolCumulativePrice {
FixedPoint.uq112x112 priceAverage;
uint priceCumulativePrevious;
uint32 blockTimeStampPrevious;
}
mapping(address => mapping (address => PoolCumulativePrice)) public cumulativePrices;
mapping(address => uint) public averagePrices;
mapping(address => address) public assetPair;
address[] public poolFactories;
address[] public stableCoins;
}
文件 25 的 25:Unitroller.sol
pragma solidity ^0.7.6;
import "./ErrorReporter.sol";
import "./ControllerStorage.sol";
contract Unitroller is UnitrollerAdminStorage, ControllerErrorReporter {
event NewPendingImplementation(address oldPendingImplementation, address newPendingImplementation);
event NewImplementation(address oldImplementation, address newImplementation);
event NewPendingAdmin(address oldPendingAdmin, address newPendingAdmin);
event NewAdmin(address oldAdmin, address newAdmin);
constructor() {
admin = msg.sender;
}
function _setPendingImplementation(address newPendingImplementation) public returns (uint) {
if (msg.sender != admin) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_PENDING_IMPLEMENTATION_OWNER_CHECK);
}
address oldPendingImplementation = pendingControllerImplementation;
pendingControllerImplementation = newPendingImplementation;
emit NewPendingImplementation(oldPendingImplementation, pendingControllerImplementation);
return uint(Error.NO_ERROR);
}
function _acceptImplementation() public returns (uint) {
if (msg.sender != pendingControllerImplementation || pendingControllerImplementation == address(0)) {
return fail(Error.UNAUTHORIZED, FailureInfo.ACCEPT_PENDING_IMPLEMENTATION_ADDRESS_CHECK);
}
address oldImplementation = controllerImplementation;
address oldPendingImplementation = pendingControllerImplementation;
controllerImplementation = pendingControllerImplementation;
pendingControllerImplementation = address(0);
emit NewImplementation(oldImplementation, controllerImplementation);
emit NewPendingImplementation(oldPendingImplementation, pendingControllerImplementation);
return uint(Error.NO_ERROR);
}
function _setPendingAdmin(address newPendingAdmin) public returns (uint) {
if (msg.sender != admin) {
return fail(Error.UNAUTHORIZED, FailureInfo.SET_PENDING_ADMIN_OWNER_CHECK);
}
address oldPendingAdmin = pendingAdmin;
pendingAdmin = newPendingAdmin;
emit NewPendingAdmin(oldPendingAdmin, newPendingAdmin);
return uint(Error.NO_ERROR);
}
function _acceptAdmin() public returns (uint) {
if (msg.sender != pendingAdmin) {
return fail(Error.UNAUTHORIZED, FailureInfo.ACCEPT_ADMIN_PENDING_ADMIN_CHECK);
}
address oldAdmin = admin;
address oldPendingAdmin = pendingAdmin;
admin = pendingAdmin;
pendingAdmin = address(0);
emit NewAdmin(oldAdmin, admin);
emit NewPendingAdmin(oldPendingAdmin, pendingAdmin);
return uint(Error.NO_ERROR);
}
fallback() payable external {
(bool success, ) = controllerImplementation.delegatecall(msg.data);
assembly {
let free_mem_ptr := mload(0x40)
returndatacopy(free_mem_ptr, 0, returndatasize())
switch success
case 0 { revert(free_mem_ptr, returndatasize()) }
default { return(free_mem_ptr, returndatasize()) }
}
}
}
{
"compilationTarget": {
"contracts/UniswapPriceOracleProxy.sol": "UniswapPriceOracleProxy"
},
"evmVersion": "istanbul",
"libraries": {},
"metadata": {
"bytecodeHash": "ipfs",
"useLiteralContent": true
},
"optimizer": {
"enabled": true,
"runs": 1000
},
"remappings": []
}
[{"inputs":[{"internalType":"address","name":"implementation_","type":"address"},{"internalType":"address","name":"registry_","type":"address"},{"internalType":"address","name":"uniswapFactory_","type":"address"},{"internalType":"address","name":"WETHToken_","type":"address"},{"internalType":"address","name":"ETHUSDPriceFeed_","type":"address"}],"stateMutability":"nonpayable","type":"constructor"},{"anonymous":false,"inputs":[{"indexed":false,"internalType":"uint256","name":"error","type":"uint256"},{"indexed":false,"internalType":"uint256","name":"info","type":"uint256"},{"indexed":false,"internalType":"uint256","name":"detail","type":"uint256"}],"name":"Failure","type":"event"},{"anonymous":false,"inputs":[{"indexed":false,"internalType":"address","name":"oldImplementation","type":"address"},{"indexed":false,"internalType":"address","name":"newImplementation","type":"address"}],"name":"NewImplementation","type":"event"},{"stateMutability":"nonpayable","type":"fallback"},{"inputs":[],"name":"Q112","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"implementation","outputs":[{"internalType":"address","name":"","type":"address"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"period","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"registry","outputs":[{"internalType":"address","name":"","type":"address"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"address","name":"newImplementation","type":"address"}],"name":"setOracleImplementation","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"nonpayable","type":"function"}]