New upstream version 18.11-rc1

[deb_dpdk.git] / lib / librte_kni / rte_kni_fifo.h

diff --git a/lib/librte_kni/rte_kni_fifo.h b/lib/librte_kni/rte_kni_fifo.h
index ac26a8c..287d7de 100644 (file)
--- a/lib/librte_kni/rte_kni_fifo.h
+++ b/lib/librte_kni/rte_kni_fifo.h
@@ -4,6 +4,36 @@
 
 
 
+/**
+ * @internal when c11 memory model enabled use c11 atomic memory barrier.
+ * when under non c11 memory model use rte_smp_* memory barrier.
+ *
+ * @param src
+ *   Pointer to the source data.
+ * @param dst
+ *   Pointer to the destination data.
+ * @param value
+ *   Data value.
+ */
+#ifdef RTE_USE_C11_MEM_MODEL
+#define __KNI_LOAD_ACQUIRE(src) ({                         \
+               __atomic_load_n((src), __ATOMIC_ACQUIRE);           \
+       })
+#define __KNI_STORE_RELEASE(dst, value) do {               \
+               __atomic_store_n((dst), value, __ATOMIC_RELEASE);   \
+       } while(0)
+#else
+#define __KNI_LOAD_ACQUIRE(src) ({                         \
+               typeof (*(src)) val = *(src);                       \
+               rte_smp_rmb();                                      \
+               val;                                                \
+       })
+#define __KNI_STORE_RELEASE(dst, value) do {               \
+               *(dst) = value;                                     \
+               rte_smp_wmb();                                      \
+       } while(0)
+#endif
+
 /**
  * Initializes the kni fifo structure
  */
@@ -28,8 +58,8 @@ kni_fifo_put(struct rte_kni_fifo *fifo, void **data, unsigned num)
 {
        unsigned i = 0;
        unsigned fifo_write = fifo->write;
-       unsigned fifo_read = fifo->read;
        unsigned new_write = fifo_write;
+       unsigned fifo_read = __KNI_LOAD_ACQUIRE(&fifo->read);
 
        for (i = 0; i < num; i++) {
                new_write = (new_write + 1) & (fifo->len - 1);
@@ -39,7 +69,7 @@ kni_fifo_put(struct rte_kni_fifo *fifo, void **data, unsigned num)
                fifo->buffer[fifo_write] = data[i];
                fifo_write = new_write;
        }
-       fifo->write = fifo_write;
+       __KNI_STORE_RELEASE(&fifo->write, fifo_write);
        return i;
 }
 
@@ -51,7 +81,8 @@ kni_fifo_get(struct rte_kni_fifo *fifo, void **data, unsigned num)
 {
        unsigned i = 0;
        unsigned new_read = fifo->read;
-       unsigned fifo_write = fifo->write;
+       unsigned fifo_write = __KNI_LOAD_ACQUIRE(&fifo->write);
+
        for (i = 0; i < num; i++) {
                if (new_read == fifo_write)
                        break;
@@ -59,7 +90,7 @@ kni_fifo_get(struct rte_kni_fifo *fifo, void **data, unsigned num)
                data[i] = fifo->buffer[new_read];
                new_read = (new_read + 1) & (fifo->len - 1);
        }
-       fifo->read = new_read;
+       __KNI_STORE_RELEASE(&fifo->read, new_read);
        return i;
 }
 
@@ -69,5 +100,7 @@ kni_fifo_get(struct rte_kni_fifo *fifo, void **data, unsigned num)
 static inline uint32_t
 kni_fifo_count(struct rte_kni_fifo *fifo)
 {
-       return (fifo->len + fifo->write - fifo->read) & (fifo->len - 1);
+       unsigned fifo_write = __KNI_LOAD_ACQUIRE(&fifo->write);
+       unsigned fifo_read = __KNI_LOAD_ACQUIRE(&fifo->read);
+       return (fifo->len + fifo_write - fifo_read) & (fifo->len - 1);
 }