Update for fractional congestion window

L4STeam · May 16, 2024 · dc111c5 · dc111c5
1 parent da73e6f
commit dc111c5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/net/ipv4/tcp_prague.c b/net/ipv4/tcp_prague.c
@@ -554,7 +554,7 @@ static void prague_update_cwnd(struct sock *sk, const struct rate_sample *rs)
 	struct tcp_sock *tp = tcp_sk(sk);
 	u64 increase;
 	s64 acked;
-	u32 new_cwnd;
+	u64 new_cwnd;
 	u64 divisor;
 	u64 mtu_used;
 
@@ -589,9 +589,9 @@ static void prague_update_cwnd(struct sock *sk, const struct rate_sample *rs)
 		ca->frac_cwnd = max_t(u64, ca->frac_cwnd + acked, prague_pacing_rate_to_frac_cwnd(sk));
 	} else {
 		increase = acked * ca->ai_ack_increase;
-		new_cwnd = tp->snd_cwnd;
+		new_cwnd = ca->frac_cwnd;
 		if (likely(new_cwnd))
-			increase = div_u64(increase + (new_cwnd >> 1), new_cwnd);
+			increase = div64_u64((increase << CWND_UNIT) + (new_cwnd >> 1), new_cwnd);
 		ca->frac_cwnd += max_t(u64, acked, increase);
 	}