rte_spinlock.h

Go to the documentation of this file.

/*-
 *   BSD LICENSE
 *
 *   Copyright(c) 2010-2014 Intel Corporation. All rights reserved.
 *   All rights reserved.
 *
 *   Redistribution and use in source and binary forms, with or without
 *   modification, are permitted provided that the following conditions
 *   are met:
 *
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in
 *       the documentation and/or other materials provided with the
 *       distribution.
 *     * Neither the name of Intel Corporation nor the names of its
 *       contributors may be used to endorse or promote products derived
 *       from this software without specific prior written permission.
 *
 *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#ifndef _RTE_SPINLOCK_H_
#define _RTE_SPINLOCK_H_

#include <rte_lcore.h>
#ifdef RTE_FORCE_INTRINSICS
#include <rte_common.h>
#endif
#include <rte_pause.h>

typedef struct {
    volatile int locked; 
} rte_spinlock_t;

#define RTE_SPINLOCK_INITIALIZER { 0 }

static inline void
rte_spinlock_init(rte_spinlock_t *sl)
{
    sl->locked = 0;
}

static inline void
rte_spinlock_lock(rte_spinlock_t *sl);

#ifdef RTE_FORCE_INTRINSICS
static inline void
rte_spinlock_lock(rte_spinlock_t *sl)
{
    int exp = 0;

    while (!__atomic_compare_exchange_n(&sl->locked, &exp, 1, 0,
                __ATOMIC_ACQUIRE, __ATOMIC_RELAXED)) {
        while (__atomic_load_n(&sl->locked, __ATOMIC_RELAXED))
            rte_pause();
        exp = 0;
    }
}
#endif

static inline void
rte_spinlock_unlock (rte_spinlock_t *sl);

#ifdef RTE_FORCE_INTRINSICS
static inline void
rte_spinlock_unlock (rte_spinlock_t *sl)
{
    __atomic_store_n(&sl->locked, 0, __ATOMIC_RELEASE);
}
#endif

static inline int
rte_spinlock_trylock (rte_spinlock_t *sl);

#ifdef RTE_FORCE_INTRINSICS
static inline int
rte_spinlock_trylock (rte_spinlock_t *sl)
{
    int exp = 0;
    return __atomic_compare_exchange_n(&sl->locked, &exp, 1,
                0, /* disallow spurious failure */
                __ATOMIC_ACQUIRE, __ATOMIC_RELAXED);
}
#endif

static inline int rte_spinlock_is_locked (rte_spinlock_t *sl)
{
    return __atomic_load_n(&sl->locked, __ATOMIC_ACQUIRE);
}

static inline int rte_tm_supported(void);

static inline void
rte_spinlock_lock_tm(rte_spinlock_t *sl);

static inline void
rte_spinlock_unlock_tm(rte_spinlock_t *sl);

static inline int
rte_spinlock_trylock_tm(rte_spinlock_t *sl);

typedef struct {
    rte_spinlock_t sl; 
    volatile int user; 
    volatile int count; 
} rte_spinlock_recursive_t;

#define RTE_SPINLOCK_RECURSIVE_INITIALIZER {RTE_SPINLOCK_INITIALIZER, -1, 0}

static inline void rte_spinlock_recursive_init(rte_spinlock_recursive_t *slr)
{
    rte_spinlock_init(&slr->sl);
    slr->user = -1;
    slr->count = 0;
}

static inline void rte_spinlock_recursive_lock(rte_spinlock_recursive_t *slr)
{
    int id = rte_gettid();

    if (slr->user != id) {
        rte_spinlock_lock(&slr->sl);
        slr->user = id;
    }
    slr->count++;
}
static inline void rte_spinlock_recursive_unlock(rte_spinlock_recursive_t *slr)
{
    if (--(slr->count) == 0) {
        slr->user = -1;
        rte_spinlock_unlock(&slr->sl);
    }

}

static inline int rte_spinlock_recursive_trylock(rte_spinlock_recursive_t *slr)
{
    int id = rte_gettid();

    if (slr->user != id) {
        if (rte_spinlock_trylock(&slr->sl) == 0)
            return 0;
        slr->user = id;
    }
    slr->count++;
    return 1;
}


static inline void rte_spinlock_recursive_lock_tm(
    rte_spinlock_recursive_t *slr);

static inline void rte_spinlock_recursive_unlock_tm(
    rte_spinlock_recursive_t *slr);

static inline int rte_spinlock_recursive_trylock_tm(
    rte_spinlock_recursive_t *slr);

#endif /* _RTE_SPINLOCK_H_ */