/*
 * Copyright (c) 2016  Intel Corporation.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at:
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef _TCP_TXQ_H_
#define _TCP_TXQ_H_

#ifdef __cplusplus
extern "C" {
#endif

static inline struct rte_mbuf **
tcp_txq_get_nxt_objs(const struct tle_tcp_stream *s, uint32_t *num)
{
	uint32_t cnt, head, mask, sz, tail;
	struct rte_ring *r;

	r = s->tx.q;
	sz = _rte_ring_get_size(r);
	mask = _rte_ring_get_mask(r);
	head = r->cons.head & mask;
	tail = r->prod.tail & mask;

	cnt = (tail >= head) ? tail - head : sz - head;

	*num = cnt;
	return (struct rte_mbuf **)(_rte_ring_get_data(r) + head);
}

static inline struct rte_mbuf **
tcp_txq_get_una_objs(const struct tle_tcp_stream *s, uint32_t *num)
{
	uint32_t cnt, head, mask, sz, tail;
	struct rte_ring *r;

	r = s->tx.q;
	sz = _rte_ring_get_size(r);
	mask = _rte_ring_get_mask(r);
	head = r->prod.tail & mask;
	tail = r->cons.tail & mask;

	cnt = (head >= tail) ? head - tail : sz - tail;

	*num = cnt;
	return (struct rte_mbuf **)(_rte_ring_get_data(r) + tail);
}

static inline void
tcp_txq_set_nxt_head(struct tle_tcp_stream *s, uint32_t num)
{
	struct rte_ring *r;

	r = s->tx.q;
	r->cons.head += num;
}

static inline void
tcp_txq_rst_nxt_head(struct tle_tcp_stream *s)
{
	struct rte_ring *r = s->tx.q;
	struct rte_mbuf *m;
	uint32_t offset, data_len;

	if (s->tcb.snd.nxt_pkt != NULL) {
		s->tcb.snd.nxt_offset = 0;
		s->tcb.snd.nxt_pkt = NULL;
	}

	offset = s->tcb.snd.una_offset;
	if (offset) {
		m = (struct rte_mbuf *)(_rte_ring_get_data(r)[r->cons.tail & r->mask]);
		data_len = m->data_len - PKT_L234_HLEN(m);
		while (offset >= data_len) {
			offset -= data_len;
			m = m->next;
			data_len = m->data_len;
		}
		s->tcb.snd.nxt_pkt = m;
		s->tcb.snd.nxt_offset = offset;
	}

	r->cons.head = r->cons.tail;
}

static inline void
tcp_txq_set_una_tail(struct tle_tcp_stream *s, uint32_t num)
{
	struct rte_ring *r;

	r = s->tx.q;
	rte_smp_rmb();
	r->cons.tail += num;
}

static inline uint32_t
tcp_txq_nxt_cnt(struct tle_tcp_stream *s)
{
	struct rte_ring *r;

	r = s->tx.q;
	return (r->prod.tail - r->cons.head) & _rte_ring_get_mask(r);
}

static inline void
txs_enqueue(struct tle_ctx *ctx, struct tle_tcp_stream *s)
{
	struct rte_ring *r;
	uint32_t n;

	if (rte_atomic32_add_return(&s->tx.arm, 1) == 1) {
		r = CTX_TCP_TSQ(ctx);
		n = _rte_ring_enqueue_burst(r, (void * const *)&s, 1);
		RTE_VERIFY(n == 1);
	}
}

static inline uint32_t
txs_dequeue_bulk(struct tle_ctx *ctx, struct tle_tcp_stream *s[], uint32_t num)
{
	struct rte_ring *r;
	uint32_t n, i;

	r = CTX_TCP_TSQ(ctx);
	n = _rte_ring_dequeue_burst(r, (void **)s, num);
	for (i = 0; i < n; i++)
		rte_atomic32_clear(&s[i]->tx.arm);
	return n;
}

#ifdef __cplusplus
}
#endif

#endif /* _TCP_TXQ_H_ */