patch to prevent price exploitation

Author: avkr003

x/oracle/keeper/abci.go

@@ -10,7 +10,7 @@ func (k Keeper) EndBlock(ctx sdk.Context) {
 	assetInfos := k.GetAllAssetInfo(ctx)
 
 	for _, info := range assetInfos {
-		allAssetPrice := k.GetAllAssetPrice(ctx, info.Display)
+		allAssetPrice := k.GetAllAssetPrice(ctx, info.Display, true)
 		total := len(allAssetPrice)
 
 		// Need to sort it because order fetched from GetAllAssetPrice will not be in ascending order - depending on source

x/oracle/keeper/abci_test.go

@@ -103,11 +103,11 @@ func (suite *KeeperTestSuite) TestEndBlock() {
 		return ethPriceData[i].Timestamp < ethPriceData[j].Timestamp
 	})
 
-	allPrices := suite.app.OracleKeeper.GetAllAssetPrice(suite.ctx, "BTC")
+	allPrices := suite.app.OracleKeeper.GetAllAssetPrice(suite.ctx, "BTC", true)
 	suite.Require().Equal(1, len(allPrices))
 	suite.Require().Equal(btcPriceData[len(btcPriceData)-1], allPrices[0])
 
-	allPrices = suite.app.OracleKeeper.GetAllAssetPrice(suite.ctx, "ETH")
+	allPrices = suite.app.OracleKeeper.GetAllAssetPrice(suite.ctx, "ETH", true)
 	suite.Require().Equal(1, len(allPrices))
 	suite.Require().Equal(ethPriceData[len(ethPriceData)-1], allPrices[0])
 }

x/oracle/keeper/price.go

@@ -12,6 +12,33 @@ import (
 	"sort"
 )
 
+var pricesMap = map[string]math.LegacyDec{
+	"SCRT": math.LegacyMustNewDecFromStr("0.1603"),
+	"KAVA": math.LegacyMustNewDecFromStr("0.1687"),
+	"WETH": math.LegacyMustNewDecFromStr("3797.49"),
+	"SAGA": math.LegacyMustNewDecFromStr("0.1268"),
+	"ATOM": math.LegacyMustNewDecFromStr("3.05"),
+	"NTRN": math.LegacyMustNewDecFromStr("0.05013"),
+	"XRP":  math.LegacyMustNewDecFromStr("2.38"),
+	"FET":  math.LegacyMustNewDecFromStr("0.3877"),
+	"INJ":  math.LegacyMustNewDecFromStr("8.95"),
+	"ONDO": math.LegacyMustNewDecFromStr("0.7362"),
+	"TIA":  math.LegacyMustNewDecFromStr("0.9503"),
+	"USDT": math.LegacyMustNewDecFromStr("1.0"),
+	"OSMO": math.LegacyMustNewDecFromStr("0.1106"),
+	"AKT":  math.LegacyMustNewDecFromStr("0.7323"),
+	"USDC": math.LegacyMustNewDecFromStr("0.9998"),
+	"OM":   math.LegacyMustNewDecFromStr("0.1084"),
+	"PAXG": math.LegacyMustNewDecFromStr("3981.96"),
+	"WBTC": math.LegacyMustNewDecFromStr("112534.90"),
+	"LINK": math.LegacyMustNewDecFromStr("17.51"),
+	"BABY": math.LegacyMustNewDecFromStr("0.03276"),
+	"XION": math.LegacyMustNewDecFromStr("0.5202"),
+}
+
+var _startHeight int64 = 6947415
+var _endHeight = _startHeight + 5
+
 // SetPrice set a specific price in the store from its index
 func (k Keeper) SetPrice(ctx sdk.Context, price types.Price) {
 	store := runtime.KVStoreAdapter(k.storeService.OpenKVStore(ctx))
@@ -82,40 +109,57 @@ func (k Keeper) GetAllPrice(ctx sdk.Context) (list []types.Price) {
 	return
 }
 
-func (k Keeper) GetAllAssetPrice(ctx sdk.Context, asset string) (list []types.Price) {
-	store := prefix.NewStore(runtime.KVStoreAdapter(k.storeService.OpenKVStore(ctx)), types.PriceKeyPrefixAsset(asset))
-	iterator := storetypes.KVStorePrefixIterator(store, []byte{})
+func (k Keeper) GetAllAssetPrice(ctx sdk.Context, asset string, endBlocker bool) (list []types.Price) {
+	if !endBlocker && ctx.BlockHeight() > _startHeight && ctx.BlockHeight() <= _endHeight {
+		for key, value := range pricesMap {
+			p := types.Price{
+				Asset:       key,
+				Price:       value,
+				Source:      "elys",
+				Provider:    "elys16qgewtplahkqwqa0aqv4pxnxa58ulu48k6crhj",
+				Timestamp:   uint64(ctx.BlockTime().Unix()),
+				BlockHeight: uint64(ctx.BlockHeight()),
+			}
+			list = append(list, p)
+		}
+	} else {
+		store := prefix.NewStore(runtime.KVStoreAdapter(k.storeService.OpenKVStore(ctx)), types.PriceKeyPrefixAsset(asset))
+		iterator := storetypes.KVStorePrefixIterator(store, []byte{})
 
-	defer iterator.Close()
+		defer iterator.Close()
 
-	for ; iterator.Valid(); iterator.Next() {
-		var val types.Price
-		k.cdc.MustUnmarshal(iterator.Value(), &val)
-		list = append(list, val)
+		for ; iterator.Valid(); iterator.Next() {
+			var val types.Price
+			k.cdc.MustUnmarshal(iterator.Value(), &val)
+			list = append(list, val)
+		}
 	}
-
 	return
 }
 
 func (k Keeper) GetAssetPrice(ctx sdk.Context, asset string) (math.LegacyDec, bool) {
-	// try out elys source
-	val, found := k.GetLatestPriceFromAssetAndSource(ctx, asset, types.ELYS)
-	if found {
-		return val.Price, true
-	}
+	if ctx.BlockHeight() > _startHeight && ctx.BlockHeight() <= _endHeight {
+		return pricesMap[asset], true
+	} else {
+		// try out elys source
+		val, found := k.GetLatestPriceFromAssetAndSource(ctx, asset, types.ELYS)
+		if found {
+			return val.Price, true
+		}
 
-	// try out band source
-	val, found = k.GetLatestPriceFromAssetAndSource(ctx, asset, types.BAND)
-	if found {
-		return val.Price, true
-	}
+		// try out band source
+		val, found = k.GetLatestPriceFromAssetAndSource(ctx, asset, types.BAND)
+		if found {
+			return val.Price, true
+		}
 
-	// find from any source if band source does not exist
-	price, found := k.GetLatestPriceFromAnySource(ctx, asset)
-	if found {
-		return price.Price, true
+		// find from any source if band source does not exist
+		price, found := k.GetLatestPriceFromAnySource(ctx, asset)
+		if found {
+			return price.Price, true
+		}
+		return math.LegacyDec{}, false
 	}
-	return math.LegacyDec{}, false
 }
 
 func (k Keeper) GetDenomPrice(ctx sdk.Context, denom string) osmomath.BigDec {
@@ -134,7 +178,7 @@ func (k Keeper) GetDenomPrice(ctx sdk.Context, denom string) osmomath.BigDec {
 }
 
 func (k Keeper) DeleteAXLPrices(ctx sdk.Context) {
-	allAssetPrice := k.GetAllAssetPrice(ctx, "AXL")
+	allAssetPrice := k.GetAllAssetPrice(ctx, "AXL", true)
 	total := len(allAssetPrice)
 
 	// Need to sort it because order fetched from GetAllAssetPrice will not be in ascending order - depending on source

x/oracle/keeper/price_test.go

@@ -156,7 +156,7 @@ func (suite *KeeperTestSuite) TestGetAllAssetPrice() {
 	for _, price := range priceData {
 		suite.app.OracleKeeper.SetPrice(suite.ctx, price)
 	}
-	allPrices := suite.app.OracleKeeper.GetAllAssetPrice(suite.ctx, "BTC")
+	allPrices := suite.app.OracleKeeper.GetAllAssetPrice(suite.ctx, "BTC", true)
 	suite.Require().Equal(len(priceData), len(allPrices))
 }