rte_ring_rts_elem_pvt.h

Go to the documentation of this file.

/* SPDX-License-Identifier: BSD-3-Clause
 *
 * Copyright (c) 2010-2020 Intel Corporation
 * Copyright (c) 2007-2009 Kip Macy kmacy@freebsd.org
 * All rights reserved.
 * Derived from FreeBSD's bufring.h
 * Used as BSD-3 Licensed with permission from Kip Macy.
 */

#ifndef _RTE_RING_RTS_ELEM_PVT_H_
#define _RTE_RING_RTS_ELEM_PVT_H_

static __rte_always_inline void
__rte_ring_rts_update_tail(struct rte_ring_rts_headtail *ht)
{
    union __rte_ring_rts_poscnt h, ot, nt;

    /*
     * If there are other enqueues/dequeues in progress that
     * might preceded us, then don't update tail with new value.
     */

    ot.raw = __atomic_load_n(&ht->tail.raw, __ATOMIC_ACQUIRE);

    do {
        /* on 32-bit systems we have to do atomic read here */
        h.raw = __atomic_load_n(&ht->head.raw, __ATOMIC_RELAXED);

        nt.raw = ot.raw;
        if (++nt.val.cnt == h.val.cnt)
            nt.val.pos = h.val.pos;

    } while (__atomic_compare_exchange_n(&ht->tail.raw, &ot.raw, nt.raw,
            0, __ATOMIC_RELEASE, __ATOMIC_ACQUIRE) == 0);
}

static __rte_always_inline void
__rte_ring_rts_head_wait(const struct rte_ring_rts_headtail *ht,
    union __rte_ring_rts_poscnt *h)
{
    uint32_t max;

    max = ht->htd_max;

    while (h->val.pos - ht->tail.val.pos > max) {
        rte_pause();
        h->raw = __atomic_load_n(&ht->head.raw, __ATOMIC_ACQUIRE);
    }
}

static __rte_always_inline uint32_t
__rte_ring_rts_move_prod_head(struct rte_ring *r, uint32_t num,
    enum rte_ring_queue_behavior behavior, uint32_t *old_head,
    uint32_t *free_entries)
{
    uint32_t n;
    union __rte_ring_rts_poscnt nh, oh;

    const uint32_t capacity = r->capacity;

    oh.raw = __atomic_load_n(&r->rts_prod.head.raw, __ATOMIC_ACQUIRE);

    do {
        /* Reset n to the initial burst count */
        n = num;

        /*
         * wait for prod head/tail distance,
         * make sure that we read prod head *before*
         * reading cons tail.
         */
        __rte_ring_rts_head_wait(&r->rts_prod, &oh);

        /*
         *  The subtraction is done between two unsigned 32bits value
         * (the result is always modulo 32 bits even if we have
         * *old_head > cons_tail). So 'free_entries' is always between 0
         * and capacity (which is < size).
         */
        *free_entries = capacity + r->cons.tail - oh.val.pos;

        /* check that we have enough room in ring */
        if (unlikely(n > *free_entries))
            n = (behavior == RTE_RING_QUEUE_FIXED) ?
                    0 : *free_entries;

        if (n == 0)
            break;

        nh.val.pos = oh.val.pos + n;
        nh.val.cnt = oh.val.cnt + 1;

    /*
     * this CAS(ACQUIRE, ACQUIRE) serves as a hoist barrier to prevent:
     *  - OOO reads of cons tail value
     *  - OOO copy of elems to the ring
     */
    } while (__atomic_compare_exchange_n(&r->rts_prod.head.raw,
            &oh.raw, nh.raw,
            0, __ATOMIC_ACQUIRE, __ATOMIC_ACQUIRE) == 0);

    *old_head = oh.val.pos;
    return n;
}

static __rte_always_inline unsigned int
__rte_ring_rts_move_cons_head(struct rte_ring *r, uint32_t num,
    enum rte_ring_queue_behavior behavior, uint32_t *old_head,
    uint32_t *entries)
{
    uint32_t n;
    union __rte_ring_rts_poscnt nh, oh;

    oh.raw = __atomic_load_n(&r->rts_cons.head.raw, __ATOMIC_ACQUIRE);

    /* move cons.head atomically */
    do {
        /* Restore n as it may change every loop */
        n = num;

        /*
         * wait for cons head/tail distance,
         * make sure that we read cons head *before*
         * reading prod tail.
         */
        __rte_ring_rts_head_wait(&r->rts_cons, &oh);

        /* The subtraction is done between two unsigned 32bits value
         * (the result is always modulo 32 bits even if we have
         * cons_head > prod_tail). So 'entries' is always between 0
         * and size(ring)-1.
         */
        *entries = r->prod.tail - oh.val.pos;

        /* Set the actual entries for dequeue */
        if (n > *entries)
            n = (behavior == RTE_RING_QUEUE_FIXED) ? 0 : *entries;

        if (unlikely(n == 0))
            break;

        nh.val.pos = oh.val.pos + n;
        nh.val.cnt = oh.val.cnt + 1;

    /*
     * this CAS(ACQUIRE, ACQUIRE) serves as a hoist barrier to prevent:
     *  - OOO reads of prod tail value
     *  - OOO copy of elems from the ring
     */
    } while (__atomic_compare_exchange_n(&r->rts_cons.head.raw,
            &oh.raw, nh.raw,
            0, __ATOMIC_ACQUIRE, __ATOMIC_ACQUIRE) == 0);

    *old_head = oh.val.pos;
    return n;
}

static __rte_always_inline unsigned int
__rte_ring_do_rts_enqueue_elem(struct rte_ring *r, const void *obj_table,
    uint32_t esize, uint32_t n, enum rte_ring_queue_behavior behavior,
    uint32_t *free_space)
{
    uint32_t free, head;

    n =  __rte_ring_rts_move_prod_head(r, n, behavior, &head, &free);

    if (n != 0) {
        __rte_ring_enqueue_elems(r, head, obj_table, esize, n);
        __rte_ring_rts_update_tail(&r->rts_prod);
    }

    if (free_space != NULL)
        *free_space = free - n;
    return n;
}

static __rte_always_inline unsigned int
__rte_ring_do_rts_dequeue_elem(struct rte_ring *r, void *obj_table,
    uint32_t esize, uint32_t n, enum rte_ring_queue_behavior behavior,
    uint32_t *available)
{
    uint32_t entries, head;

    n = __rte_ring_rts_move_cons_head(r, n, behavior, &head, &entries);

    if (n != 0) {
        __rte_ring_dequeue_elems(r, head, obj_table, esize, n);
        __rte_ring_rts_update_tail(&r->rts_cons);
    }

    if (available != NULL)
        *available = entries - n;
    return n;
}

#endif /* _RTE_RING_RTS_ELEM_PVT_H_ */