/* Delivery Rate Sampling */

#include <haproxy/pool.h>
#include <haproxy/quic_cc-t.h>
#include <haproxy/quic_cc_drs.h>
#include <haproxy/quic_tx-t.h>
#include <haproxy/task.h>
#include <haproxy/ticks.h>
#include <haproxy/window_filter.h>

static void quic_cc_rs_init(struct quic_cc_rs *rs)
{
	rs->interval_us = UINT32_MAX;
	rs->delivered = 0;
	rs->prior_delivered = 0;
	rs->prior_time_ns = 0;
	rs->tx_in_flight = 0;
	rs->lost = 0;
	rs->prior_lost = 0;
	rs->send_elapsed_us = 0;
	rs->ack_elapsed_us = 0;
	rs->last_end_seq = -1;
	rs->is_app_limited = 0;
}

void quic_cc_drs_init(struct quic_cc_drs *drs)
{
	quic_cc_rs_init(&drs->rs);
	drs->round_count = 0;
	drs->next_round_delivered = 0;
	drs->delivered = 0;
	drs->lost = 0;
	drs->last_seq = -1;
	drs->delivered_time_ns = 0;
	drs->first_sent_time_ns = 0;
	drs->app_limited = 0;
	drs->is_cwnd_limited = 0;
}

/* Update <pkt> TX packet rate sampling information.
 * Must be called after <pkt> has just been sent.
 */
void quic_cc_drs_on_pkt_sent(struct quic_cc_path *path,
                             struct quic_tx_packet *pkt, struct quic_cc_drs *drs)
{
	if (!path->in_flight)
		drs->first_sent_time_ns = drs->delivered_time_ns = pkt->time_sent_ns;

	pkt->rs.first_sent_time_ns = drs->first_sent_time_ns;
	pkt->rs.delivered_time_ns  = drs->delivered_time_ns;
	pkt->rs.delivered          = drs->delivered;
	pkt->rs.is_app_limited     = drs->app_limited != 0;

	pkt->rs.tx_in_flight = path->in_flight + pkt->len;
	pkt->rs.lost = drs->lost;
	pkt->rs.end_seq = ++drs->last_seq;
}

/* Return 1 if <pkt> TX packet is the most recently sent packet
 * that has been delivered, 0 if not.
 */
static inline int quic_cc_drs_is_newest_packet(struct quic_cc_drs *drs,
                                               struct quic_tx_packet *pkt)
{
	return drs->first_sent_time_ns < pkt->time_sent_ns ||
		(pkt->time_sent_ns == drs->first_sent_time_ns &&
		 pkt->rs.end_seq > drs->rs.last_end_seq);
}

/* RFC https://datatracker.ietf.org/doc/draft-ietf-ccwg-bbr/
 * 4.5.2.3.3.  Upon receiving an ACK
 *
 * When an ACK arrives, the sender invokes GenerateRateSample() to fill
 * in a rate sample.  For each packet that was newly SACKed or ACKed,
 * UpdateRateSample() updates the rate sample based on a snapshot of
 * connection delivery information from the time at which the packet was
 * last transmitted.  UpdateRateSample() is invoked multiple times when
 * a stretched ACK acknowledges multiple data packets.  In this case we
 * use the information from the most recently sent packet, i.e., the
 * packet with the highest "P.delivered" value.
 *
 * haproxy implementation: quic_cc_drs_update_rate_sample() matches with
 * RFC UpdateRateSample() called from first part of GenerateRateSample().
 */
void quic_cc_drs_update_rate_sample(struct quic_cc_drs *drs,
                                    struct quic_tx_packet *pkt, uint64_t time_ns)
{
	struct quic_cc_rs *rs = &drs->rs;

	if (!pkt->rs.delivered_time_ns)
		return;

	drs->delivered += pkt->len;
	drs->delivered_time_ns = time_ns;
	/* Update info using the newest packet. */
	if (rs->prior_time_ns && !quic_cc_drs_is_newest_packet(drs, pkt))
		return;

	rs->prior_delivered  = pkt->rs.delivered;
	rs->prior_time_ns    = pkt->rs.delivered_time_ns;
	rs->is_app_limited   = pkt->rs.is_app_limited;
	rs->send_elapsed_us  = (pkt->time_sent_ns - pkt->rs.first_sent_time_ns) / 1000;
	rs->ack_elapsed_us   = (drs->delivered_time_ns - pkt->rs.delivered_time_ns) / 1000;
	rs->tx_in_flight     = pkt->rs.tx_in_flight;
	rs->prior_lost       = pkt->rs.lost;
	rs->last_end_seq     = pkt->rs.end_seq;
	drs->first_sent_time_ns = pkt->time_sent_ns;
	/* Mark the packet as delivered once it's SACKed to
	 * avoid being used again when it's cumulatively acked.
	 */
	pkt->rs.delivered_time_ns = 0;
}

/* RFC https://datatracker.ietf.org/doc/draft-ietf-ccwg-bbr/
 * 4.5.2.3.3.  Upon receiving an ACK
 *
 * haproxy implementation: second part of GenerateRateSample(). Follows the
 * first one above.
 */
void quic_cc_drs_on_ack_recv(struct quic_cc_drs *drs, struct quic_cc_path *path,
                             uint64_t pkt_delivered)
{
	struct quic_cc_rs *rs = &drs->rs;

	if (drs->app_limited && drs->delivered > drs->app_limited)
		drs->app_limited = 0;

	if (pkt_delivered >= drs->next_round_delivered) {
		drs->next_round_delivered = pkt_delivered;
		++drs->round_count;
	}

	if (!rs->prior_time_ns)
		return;

	rs->interval_us = MAX(rs->send_elapsed_us, rs->ack_elapsed_us);

	BUG_ON(drs->delivered <= rs->prior_delivered);
	rs->delivered = drs->delivered - rs->prior_delivered;
	BUG_ON(drs->lost < rs->prior_lost);
	rs->lost = drs->lost - rs->prior_lost;

	if (rs->interval_us < path->loss.rtt_min * 1000) {
		rs->interval_us = UINT32_MAX;
		return;
	}

	if (!rs->interval_us)
		return;

	/* <delivery_rate> is in bytes/s. */
	path->delivery_rate = rs->delivered * 1000000 / rs->interval_us;
}