1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125

/// # Fee market change for ETH 1.0 chain
/// A transaction pricing mechanism that includes fixed-per-block network fee that is burned and dynamically expands/contracts block sizes to deal with transient congestion.
///
/// [Fee market change for ETH 1.0 chain](https://eips.ethereum.org/EIPS/eip-1559)
pub mod eip1559 {
    use num_bigint::BigUint;
    use num_traits::One;
    use std::cmp::{self, Ordering};

    const BASEFEE_MAX_CHANGE_DENOMINATOR: u64 = 8;

    ///
    /// Computes the base fee of a block given parent block header data.
    /// # Arguments
    /// * `parent_base_fee` - The value of the parent block base fee
    /// * `parent_gas_used` - The value of the parent block gas used
    /// * `parent_target_gas_used` - The value of the parent block target gas used
    ///
    /// # Examples
    ///
    /// ```
    /// use num_bigint::BigUint;
    /// use eip_utils::eip1559;
    /// let parent_base_fee = BigUint::from(1000000000 as u64);
    /// let parent_gas_used = BigUint::from(10000000 as u64);
    /// let parent_target_gas_used = BigUint::from(5000000 as u64);
    /// let base_fee = eip1559::compute_base_fee(&parent_base_fee, &parent_gas_used, &parent_target_gas_used);
    /// assert_eq!(base_fee, BigUint::from(1125000000 as u64));
    /// ```
    pub fn compute_base_fee(
        parent_base_fee: &BigUint,
        parent_gas_used: &BigUint,
        parent_target_gas_used: &BigUint,
    ) -> BigUint {
        match parent_gas_used.cmp(&parent_target_gas_used) {
            Ordering::Equal => parent_base_fee.clone(),
            Ordering::Greater => {
                compute_base_fee_increase(parent_base_fee, parent_gas_used, parent_target_gas_used)
            }
            Ordering::Less => {
                compute_base_fee_decrease(parent_base_fee, parent_gas_used, parent_target_gas_used)
            }
        }
    }

    fn compute_base_fee_increase(
        parent_base_fee: &BigUint,
        parent_gas_used: &BigUint,
        parent_target_gas_used: &BigUint,
    ) -> BigUint {
        let gas_delta = parent_gas_used - parent_target_gas_used;
        let fee_delta = cmp::max(
            (parent_base_fee * gas_delta) / parent_target_gas_used / BASEFEE_MAX_CHANGE_DENOMINATOR,
            One::one(),
        );
        parent_base_fee + fee_delta
    }

    fn compute_base_fee_decrease(
        parent_base_fee: &BigUint,
        parent_gas_used: &BigUint,
        parent_target_gas_used: &BigUint,
    ) -> BigUint {
        let gas_delta = parent_target_gas_used - parent_gas_used;
        let fee_delta =
            (parent_base_fee * gas_delta) / parent_target_gas_used / BASEFEE_MAX_CHANGE_DENOMINATOR;
        parent_base_fee - fee_delta
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use num_bigint::BigUint;
    use serde::{Deserialize, Serialize};
    use std::{cmp::Ordering, fs, path::PathBuf};

    #[test]
    fn assert_basefee_computation_works() {
        let mut d = PathBuf::from(env!("CARGO_MANIFEST_DIR"));
        d.push("resources/test/eip1559/basefee/reference-test.json");
        let path = d.as_path().to_str().unwrap();
        let data = fs::read_to_string(path);
        let data = match data {
            Ok(c) => c,
            Err(error) => panic!("Problem opening the file: {:?}", error),
        };

        let test_cases = json_to_testcase(&data);
        for case in &test_cases {
            let parent_base_fee = BigUint::from(case.parent_base_fee);
            let parent_gas_used = BigUint::from(case.parent_gas_used);
            let parent_target_gas_used = BigUint::from(case.parent_target_gas_used);
            let base_fee = eip1559::compute_base_fee(
                &parent_base_fee,
                &parent_gas_used,
                &parent_target_gas_used,
            );
            assert_eq!(
                true,
                base_fee.cmp(&BigUint::from(case.expected_base_fee)) == Ordering::Equal
            );
        }
    }

    pub fn json_to_testcase(data: &str) -> Vec<TestCase> {
        let t = serde_json::from_str(data);
        match t {
            Ok(c) => c,
            Err(error) => panic!("Problem opening the file: {:?}", error),
        }
    }

    #[derive(Debug, Serialize, Deserialize)]
    pub struct TestCase {
        #[serde(rename = "parentBaseFee")]
        pub parent_base_fee: u64,
        #[serde(rename = "parentGasUsed")]
        pub parent_gas_used: u64,
        #[serde(rename = "parentTargetGasUsed")]
        pub parent_target_gas_used: u64,
        #[serde(rename = "expectedBaseFee")]
        pub expected_base_fee: u64,
    }
}