From 3e75ebf7b05cc9c1d626203fc16ef0c41e61d61b Mon Sep 17 00:00:00 2001
From: Kerber0x <94062656+kerber0x@users.noreply.github.com>
Date: Tue, 25 Jun 2024 14:31:42 +0100
Subject: [PATCH] fix: fee distributor claim floor fix

---
 Cargo.lock                                              | 2 +-
 contracts/liquidity_hub/fee_distributor/Cargo.toml      | 2 +-
 contracts/liquidity_hub/fee_distributor/src/commands.rs | 4 +++-
 contracts/liquidity_hub/fee_distributor/src/error.rs    | 7 ++++++-
 4 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index eccf1a3d3..64085af9c 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -619,7 +619,7 @@ dependencies = [
 
 [[package]]
 name = "fee_distributor"
-version = "0.9.3"
+version = "0.9.4"
 dependencies = [
  "cosmwasm-schema",
  "cosmwasm-std",
diff --git a/contracts/liquidity_hub/fee_distributor/Cargo.toml b/contracts/liquidity_hub/fee_distributor/Cargo.toml
index 977cf0689..95705d66c 100644
--- a/contracts/liquidity_hub/fee_distributor/Cargo.toml
+++ b/contracts/liquidity_hub/fee_distributor/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "fee_distributor"
-version = "0.9.3"
+version = "0.9.4"
 authors = ["Kerber0x <kerber0x@protonmail.com>"]
 edition.workspace = true
 description = "Contract to distribute the fees collected by the Fee Collector."
diff --git a/contracts/liquidity_hub/fee_distributor/src/commands.rs b/contracts/liquidity_hub/fee_distributor/src/commands.rs
index 554dafb3d..d6d1d07fc 100644
--- a/contracts/liquidity_hub/fee_distributor/src/commands.rs
+++ b/contracts/liquidity_hub/fee_distributor/src/commands.rs
@@ -97,7 +97,9 @@ pub fn claim(deps: DepsMut, info: MessageInfo) -> Result<Response, ContractError
             }))?;
 
         for fee in epoch.total.iter() {
-            let reward = fee.amount * bonding_weight_response.share;
+            let reward = fee
+                .amount
+                .checked_mul_floor(bonding_weight_response.share)?;
 
             if reward.is_zero() {
                 // nothing to claim
diff --git a/contracts/liquidity_hub/fee_distributor/src/error.rs b/contracts/liquidity_hub/fee_distributor/src/error.rs
index 26d8fe17e..87112270f 100644
--- a/contracts/liquidity_hub/fee_distributor/src/error.rs
+++ b/contracts/liquidity_hub/fee_distributor/src/error.rs
@@ -1,4 +1,6 @@
-use cosmwasm_std::{DivideByZeroError, OverflowError, StdError, Uint64};
+use cosmwasm_std::{
+    CheckedMultiplyFractionError, DivideByZeroError, OverflowError, StdError, Uint64,
+};
 use cw_utils::ParseReplyError;
 use semver::Version;
 use thiserror::Error;
@@ -53,6 +55,9 @@ pub enum ContractError {
     #[error("{0}")]
     ParseReplyError(#[from] ParseReplyError),
 
+    #[error("{0}")]
+    CheckedMultiplyFractionError(#[from] CheckedMultiplyFractionError),
+
     #[error("Attempt to migrate to version {new_version}, but contract is on a higher version {current_version}")]
     MigrateInvalidVersion {
         new_version: Version,