--- /dev/null
+/******************************************************************************
+*
+* Copyright (c) 2019 Intel.
+*
+* 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.
+*
+*******************************************************************************/
+
+#include <stdio.h>
+#include <unistd.h>
+#include <sys/mman.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <stdint.h>
+#include <stdlib.h>
+#include <string.h>
+#include <errno.h>
+#include <pthread.h>
+
+#include <sys/ipc.h>
+#include <sys/shm.h>
+
+#include <rte_eal.h>
+#include <rte_errno.h>
+#include <rte_lcore.h>
+#include <rte_memory.h>
+#include <rte_memzone.h>
+#include <rte_branch_prediction.h>
+
+#include "ttypes.h"
+#include "wls_lib.h"
+#include "wls.h"
+#include "syslib.h"
+
+#define WLS_MAP_SHM 1
+
+#define WLS_PHY_SHM_FILE_NAME "hp_"
+
+#define HUGE_PAGE_FILE_NAME "/mnt/huge/page"
+
+#define DIV_ROUND_OFFSET(X,Y) ( X/Y + ((X%Y)?1:0) )
+
+#define WLS_LIB_USER_SPACE_CTX_SIZE DMA_MAP_MAX_BLOCK_SIZE
+
+#define PLIB_ERR(x, args...) printf("wls_lib: "x, ## args);
+#define PLIB_INFO(x, args...) printf("wls_lib: "x, ## args);
+
+#ifdef _DEBUG_
+#define PLIB_DEBUG(x, args...) printf("wls_lib debug: "x, ## args);
+#else
+#define PLIB_DEBUG(x, args...) do { } while(0)
+#endif
+
+extern int gethugepagesizes(long pagesizes[], int n_elem);
+
+
+static uint32_t get_hugepagesz_flag(uint64_t hugepage_sz)
+{
+ unsigned size_flag = 0;
+
+ switch (hugepage_sz) {
+ case RTE_PGSIZE_256K:
+ size_flag = RTE_MEMZONE_256KB;
+ break;
+ case RTE_PGSIZE_2M:
+ size_flag = RTE_MEMZONE_2MB;
+ break;
+ case RTE_PGSIZE_16M:
+ size_flag = RTE_MEMZONE_16MB;
+ break;
+ case RTE_PGSIZE_256M:
+ size_flag = RTE_MEMZONE_256MB;
+ break;
+ case RTE_PGSIZE_512M:
+ size_flag = RTE_MEMZONE_512MB;
+ break;
+ case RTE_PGSIZE_1G:
+ size_flag = RTE_MEMZONE_1GB;
+ break;
+ case RTE_PGSIZE_4G:
+ size_flag = RTE_MEMZONE_4GB;
+ break;
+ case RTE_PGSIZE_16G:
+ size_flag = RTE_MEMZONE_16GB;
+ break;
+ default:
+ PLIB_INFO("Unknown hugepage size %lu\n", hugepage_sz);
+ break;
+ }
+ return size_flag;
+}
+
+static pthread_mutex_t wls_put_lock;
+static pthread_mutex_t wls_get_lock;
+static pthread_mutex_t wls_put_lock1;
+static pthread_mutex_t wls_get_lock1;
+
+static wls_us_ctx_t* wls_us_ctx = NULL;
+static wls_us_ctx_t* wls_us_ctx1 = NULL;
+static const struct rte_memzone *hp_memzone = NULL;
+static long hugePageSize = WLS_HUGE_DEF_PAGE_SIZE;
+static uint64_t gWlsMemorySize = 0;
+
+static inline int wls_check_ctx(void *h)
+{
+ if (h != wls_us_ctx) {
+ PLIB_ERR("Incorrect user space context\n");
+ return -1;
+ }
+ return 0;
+}
+
+static inline int wls_check_ctx1(void *h)
+{
+ if (h != wls_us_ctx1) {
+ PLIB_ERR("Incorrect user space context1\n");
+ return -1;
+ }
+ return 0;
+}
+
+static int wls_VirtToPhys(const void* virtAddr, uint64_t* physAddr)
+{
+ int mapFd;
+ uint64_t page;
+ unsigned int pageSize;
+ unsigned long virtualPageNumber;
+
+ mapFd = open("/proc/self/pagemap", O_RDONLY);
+ if (mapFd < 0) {
+ PLIB_ERR("Could't open pagemap file\n");
+ return -1;
+ }
+
+ /*get standard page size*/
+ pageSize = getpagesize();
+
+ virtualPageNumber = (unsigned long) virtAddr / pageSize;
+
+ lseek(mapFd, virtualPageNumber * sizeof (uint64_t), SEEK_SET);
+
+ if (read(mapFd, &page, sizeof (uint64_t)) < 0) {
+ close(mapFd);
+ PLIB_ERR("Could't read pagemap file\n");
+ return -1;
+ }
+
+ *physAddr = ((page & 0x007fffffffffffffULL) * pageSize);
+
+ close(mapFd);
+
+ return 0;
+}
+
+static void wls_mutex_destroy(pthread_mutex_t* pMutex)
+{
+ pthread_mutex_destroy(pMutex);
+}
+
+static void wls_mutex_init(pthread_mutex_t* pMutex)
+{
+ pthread_mutexattr_t prior;
+ pthread_mutexattr_init(&prior);
+ pthread_mutexattr_setprotocol(&prior, PTHREAD_PRIO_INHERIT);
+ pthread_mutex_init(pMutex, &prior);
+ pthread_mutexattr_destroy(&prior);
+}
+
+static void wls_mutex_lock(pthread_mutex_t* pMutex)
+{
+ pthread_mutex_lock(pMutex);
+}
+
+static void wls_mutex_unlock(pthread_mutex_t* pMutex)
+{
+ pthread_mutex_unlock(pMutex);
+}
+
+static int wls_initialize(const char *ifacename, uint64_t nWlsMemorySize)
+{
+ int ret;
+ pthread_mutexattr_t attr;
+
+ uint64_t nSize = nWlsMemorySize + sizeof(wls_drv_ctx_t);
+ uint8_t *pMemZone;
+ const struct rte_memzone *mng_ctx_memzone;
+ wls_drv_ctx_t *mng_ctx;
+
+ mng_ctx_memzone = rte_memzone_reserve_aligned(ifacename, nSize, rte_socket_id(), get_hugepagesz_flag(hugePageSize), hugePageSize);
+ if (mng_ctx_memzone == NULL) {
+ PLIB_ERR("Cannot reserve memory zone[%s]: %s\n", ifacename, rte_strerror(rte_errno));
+ return -1;
+ }
+
+ pMemZone = ((uint8_t *)mng_ctx_memzone->addr) + nWlsMemorySize;
+ memset(pMemZone, 0, sizeof(wls_drv_ctx_t));
+ mng_ctx = (wls_drv_ctx_t *)pMemZone;
+
+ pthread_mutexattr_init(&attr);
+ pthread_mutexattr_setpshared(&attr, PTHREAD_PROCESS_SHARED);
+ if (ret = pthread_mutex_init(&mng_ctx->mng_mutex, &attr)) {
+ PLIB_ERR("Failed to initialize mng_mutex %d\n", ret);
+ pthread_mutexattr_destroy(&attr);
+ return ret;
+ }
+ pthread_mutexattr_destroy(&attr);
+ PLIB_DEBUG("Run wls_initialized\n");
+ return 0;
+}
+
+static wls_us_ctx_t* wls_create_us_ctx(wls_drv_ctx_t *pDrv_ctx)
+{
+ int idx;
+
+ wls_mutex_lock(&pDrv_ctx->mng_mutex);
+
+ if (unlikely(pDrv_ctx->nWlsClients >= WLS_US_CLIENTS_MAX)) {
+ PLIB_ERR("Maximum number of clients reached");
+ wls_mutex_unlock(&pDrv_ctx->mng_mutex);
+ return NULL;
+ }
+
+ wls_us_ctx_t *pUsCtx = &pDrv_ctx->p_wls_us_ctx[pDrv_ctx->nWlsClients];
+ wls_us_priv_t *pUs_priv = &pUsCtx->wls_us_private;
+
+ PLIB_DEBUG("wls_create_us_ctx for %d client\n", pDrv_ctx->nWlsClients);
+ memset(pUsCtx, 0, sizeof (wls_us_ctx_t));
+
+ SFL_DefQueue(&pUsCtx->ul_free_block_pq, pUsCtx->ul_free_block_storage,
+ UL_FREE_BLOCK_QUEUE_SIZE * sizeof (void*));
+ WLS_MsgDefineQueue(&pUsCtx->get_queue, pUsCtx->get_storage, WLS_GET_QUEUE_N_ELEMENTS, 0);
+ WLS_MsgDefineQueue(&pUsCtx->put_queue, pUsCtx->put_storage, WLS_PUT_QUEUE_N_ELEMENTS, 0);
+
+ memset(pUs_priv, 0, sizeof (wls_us_priv_t));
+ if (sem_init(&pUs_priv->sema.sem, 1, 0)) {
+ PLIB_ERR("Failed to initialize semaphore %s\n", strerror(errno));
+ memset(pUsCtx, 0, sizeof (wls_us_ctx_t));
+ wls_mutex_unlock(&pDrv_ctx->mng_mutex);
+ return NULL;
+ }
+ rte_atomic16_init(&pUs_priv->sema.is_irq);
+
+ idx = pDrv_ctx->nWlsClients;
+ pDrv_ctx->p_wls_us_ctx[idx].dst_user_va = (uint64_t) & pDrv_ctx->p_wls_us_ctx[idx ^ 1];
+ PLIB_INFO("link: %d <-> %d\n", idx, idx ^ 1);
+
+ pUs_priv->pid = getpid();
+ pDrv_ctx->nWlsClients++;
+
+ wls_mutex_unlock(&pDrv_ctx->mng_mutex);
+ return pUsCtx;
+}
+
+static int wls_destroy_us_ctx(wls_us_ctx_t *pUsCtx, wls_drv_ctx_t *pDrv_ctx)
+{
+ wls_us_priv_t* pUs_priv = NULL;
+
+ wls_mutex_lock(&pDrv_ctx->mng_mutex);
+ if (pDrv_ctx->p_wls_us_ctx[0].wls_us_private.pid
+ && pDrv_ctx->p_wls_us_ctx[1].wls_us_private.pid) {
+ PLIB_INFO("un-link: 0 <-> 1\n");
+ pDrv_ctx->p_wls_us_ctx[0].dst_user_va = 0ULL;
+ pDrv_ctx->p_wls_us_ctx[1].dst_user_va = 0ULL;
+ }
+
+
+ if (pUsCtx) {
+ pUs_priv = (wls_us_priv_t*) & pUsCtx->wls_us_private;
+ if (pUs_priv) {
+ sem_destroy(&pUs_priv->sema.sem);
+ memset(pUsCtx, 0, sizeof (wls_us_ctx_t));
+ pDrv_ctx->nWlsClients--;
+ }
+ }
+
+ wls_mutex_unlock(&pDrv_ctx->mng_mutex);
+ return 0;
+}
+
+static int wls_destroy_us_ctx1(wls_us_ctx_t *pUsCtx, wls_drv_ctx_t *pDrv_ctx)
+{
+ wls_us_priv_t* pUs_priv = NULL;
+
+ wls_mutex_lock(&pDrv_ctx->mng_mutex);
+ if (pDrv_ctx->p_wls_us_ctx[2].wls_us_private.pid
+ && pDrv_ctx->p_wls_us_ctx[3].wls_us_private.pid) {
+ PLIB_INFO("un-link: 2 <-> 3\n");
+ pDrv_ctx->p_wls_us_ctx[2].dst_user_va = 0ULL;
+ pDrv_ctx->p_wls_us_ctx[3].dst_user_va = 0ULL;
+ }
+
+
+ if (pUsCtx) {
+ pUs_priv = (wls_us_priv_t*) & pUsCtx->wls_us_private;
+ if (pUs_priv) {
+ sem_destroy(&pUs_priv->sema.sem);
+ memset(pUsCtx, 0, sizeof (wls_us_ctx_t));
+ pDrv_ctx->nWlsClients--;
+ }
+ }
+
+ wls_mutex_unlock(&pDrv_ctx->mng_mutex);
+ return 0;
+}
+
+static int wls_wake_up_user_thread(char *buf, wls_sema_priv_t *semap)
+{
+ if (unlikely(rte_atomic16_read(&semap->is_irq) >= FIFO_LEN))
+ return 0;
+
+ unsigned int put = semap->drv_block_put + 1;
+ if (put >= FIFO_LEN)
+ put = 0;
+ memcpy(&semap->drv_block[put], buf, sizeof (wls_wait_req_t));
+ semap->drv_block_put = put;
+ rte_atomic16_inc(&semap->is_irq);
+ PLIB_DEBUG("PUT: put=%d get=%d T=%lu is_irq=%d\n",
+ semap->drv_block_put, semap->drv_block_get,
+ semap->drv_block[put].start_time, rte_atomic16_read(&semap->is_irq));
+ sem_post(&semap->sem);
+ return 0;
+}
+
+static int wls_process_put(wls_us_ctx_t *src, wls_us_ctx_t *dst)
+{
+ int ret = 0;
+ WLS_MSG_HANDLE hMsg;
+ int n = 0;
+
+ wls_us_priv_t* pDstPriv = NULL;
+ wls_wait_req_t drv_block;
+
+ if (NULL == src || NULL == dst) {
+ PLIB_DEBUG("Bad input addresses\n");
+ return -1;
+ }
+
+ n = WLS_GetNumItemsInTheQueue(&src->put_queue);
+
+ while (n--) {
+ if (WLS_MsgDequeue(&src->put_queue, &hMsg, NULL, (void*) src) == FALSE) {
+ PLIB_ERR("WLS_MsgDequeue src failed\n");
+ ret = -1;
+ break;
+ }
+ PLIB_DEBUG("WLS_Get %lx %d type %d\n", (U64) hMsg.pIaPaMsg, hMsg.MsgSize, hMsg.TypeID);
+ if (WLS_MsgEnqueue(&dst->get_queue, hMsg.pIaPaMsg, hMsg.MsgSize, hMsg.TypeID,
+ hMsg.flags, NULL, (void*) dst) == FALSE) { // try to send
+ if (WLS_MsgEnqueue(&src->put_queue, hMsg.pIaPaMsg, hMsg.MsgSize, hMsg.TypeID,
+ hMsg.flags, NULL, (void*) src) == FALSE) { // return back
+ PLIB_ERR("wls_process_put: Cannot return block to back to queue \n");
+ ret = -1;
+ }
+ break;
+ }
+ }
+
+ if (dst->wls_us_private.pid) {
+ pDstPriv = (wls_us_priv_t*) & dst->wls_us_private;
+
+ drv_block.start_time = wls_rdtsc();
+ pDstPriv->NeedToWakeUp = 1;
+ wls_wake_up_user_thread((char *) &drv_block, &pDstPriv->sema);
+ } else
+ ret = -1;
+
+ return ret;
+}
+
+static int wls_process_wakeup(void* h)
+{
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+ int ret = 0;
+ wls_wait_req_t drv_block;
+ if (wls_check_ctx(h))
+ return -1;
+ if (!pWls_us->wls_us_private.pid) {
+ PLIB_ERR("wakeup failed");
+ return -1;
+ }
+ drv_block.start_time = wls_rdtsc();
+ wls_wake_up_user_thread((char *) &drv_block, &pWls_us->wls_us_private.sema);
+
+ return ret;
+}
+
+static int wls_process_wakeup1(void* h)
+{
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+ int ret = 0;
+ wls_wait_req_t drv_block;
+ if (wls_check_ctx1(h))
+ return -1;
+ if (!pWls_us->wls_us_private.pid) {
+ PLIB_ERR("wakeup failed");
+ return -1;
+ }
+ drv_block.start_time = wls_rdtsc();
+ wls_wake_up_user_thread((char *) &drv_block, &pWls_us->wls_us_private.sema);
+
+ return ret;
+}
+
+static int wls_wait(wls_sema_priv_t *priv)
+{
+ if (!rte_atomic16_read(&priv->is_irq)) {
+ if (sem_wait(&priv->sem) || !rte_atomic16_read(&priv->is_irq)) {
+ return -1;
+ }
+ }
+
+ rte_atomic16_dec(&priv->is_irq);
+
+ if (priv->drv_block_put != priv->drv_block_get) {
+ unsigned int get = priv->drv_block_get + 1;
+
+ if (get >= FIFO_LEN)
+ get = 0;
+
+ priv->drv_block_get = get;
+
+ PLIB_DEBUG("GET: put=%d get=%d T=%lu is_irq=%d\n",
+ priv->drv_block_put, priv->drv_block_get,
+ priv->drv_block[get].start_time, rte_atomic16_read(&priv->is_irq));
+ } else {
+ PLIB_DEBUG("[wrong computation of queueing\n");
+ }
+
+ return 0;
+}
+
+static int wls_process_wait(void* h)
+{
+ wls_us_ctx_t* pUsCtx = (wls_us_ctx_t*) h;
+
+ if (wls_check_ctx(h))
+ return -1;
+
+ if (pUsCtx == NULL) {
+ PLIB_ERR("Wait failed on User context");
+ return -1;
+ }
+
+
+ if (!pUsCtx->wls_us_private.pid) {
+ PLIB_ERR("Wait failed");
+ return -1;
+ }
+ pUsCtx->wls_us_private.isWait = 1;
+ wls_wait(&pUsCtx->wls_us_private.sema);
+ pUsCtx->wls_us_private.isWait = 0;
+ return WLS_GetNumItemsInTheQueue(&pUsCtx->get_queue);
+}
+
+static int wls_process_wait1(void* h)
+{
+ wls_us_ctx_t* pUsCtx = (wls_us_ctx_t*) h;
+
+ if (wls_check_ctx1(h))
+ return -1;
+
+ if (pUsCtx == NULL) {
+ PLIB_ERR("Wait failed on User context");
+ return -1;
+ }
+
+
+ if (!pUsCtx->wls_us_private.pid) {
+ PLIB_ERR("Wait failed");
+ return -1;
+ }
+ pUsCtx->wls_us_private.isWait = 1;
+ wls_wait(&pUsCtx->wls_us_private.sema);
+ pUsCtx->wls_us_private.isWait = 0;
+ return WLS_GetNumItemsInTheQueue(&pUsCtx->get_queue);
+}
+
+
+void* WLS_Open(const char *ifacename, unsigned int mode, unsigned long long nWlsMemorySize)
+{
+ wls_us_ctx_t* pWls_us = NULL;
+ int i, len;
+ char temp[WLS_DEV_SHM_NAME_LEN] = {0};
+ uint8_t *pMemZone;
+
+ gethugepagesizes(&hugePageSize, 1);
+
+ if (wls_us_ctx)
+ return wls_us_ctx;
+
+ strncpy(temp, ifacename, WLS_DEV_SHM_NAME_LEN - 1);
+ PLIB_INFO("Open %s (DPDK memzone)\n", temp);
+
+ static const struct rte_memzone *mng_memzone;
+ mng_memzone = rte_memzone_lookup(temp);
+ if ((mng_memzone == NULL)&&(RTE_PROC_PRIMARY==rte_eal_process_type())) {
+ wls_initialize(temp, nWlsMemorySize);
+ }
+
+ mng_memzone = rte_memzone_lookup(temp);
+ if (mng_memzone == NULL) {
+ PLIB_ERR("Cannot initialize wls shared memory: %s\n", temp);
+ return NULL;
+ }
+
+ pMemZone = ((uint8_t *)mng_memzone->addr) + nWlsMemorySize;
+
+ PLIB_INFO("WLS_Open %p\n", pMemZone);
+ if ((pWls_us = wls_create_us_ctx((wls_drv_ctx_t *)pMemZone)) == NULL) {
+ PLIB_ERR("WLS_Open failed to create context\n");
+ return NULL;
+ }
+
+ PLIB_DEBUG("Local: pWls_us %p\n", pWls_us);
+
+ pWls_us->padding_wls_us_user_space_va = 0LL;
+ pWls_us->wls_us_user_space_va = pWls_us;
+ pWls_us->wls_us_ctx_size = sizeof (*pWls_us);
+ gWlsMemorySize = nWlsMemorySize;
+
+ wls_mutex_init(&wls_put_lock);
+ wls_mutex_init(&wls_get_lock);
+
+ pWls_us->mode = mode;
+ pWls_us->secmode = WLS_SEC_NA;
+ pWls_us->dualMode = WLS_SINGLE_MODE;
+ PLIB_INFO("Mode %d\n", pWls_us->mode);
+
+ PLIB_INFO("WLS shared management memzone: %s\n", temp);
+ strncpy(pWls_us->wls_dev_name, temp, WLS_DEV_SHM_NAME_LEN - 1);
+ wls_us_ctx = pWls_us;
+
+ return wls_us_ctx;
+}
+
+void* WLS_Open_Dual(const char *ifacename, unsigned int mode, unsigned long long nWlsMemorySize, void** handle1)
+{
+ wls_us_ctx_t* pWls_us = NULL;
+ wls_us_ctx_t* pWls_us1 = NULL;
+ int i, len;
+ char temp[WLS_DEV_SHM_NAME_LEN] = {0};
+ uint8_t *pMemZone;
+
+ gethugepagesizes(&hugePageSize, 1);
+
+ if (wls_us_ctx)
+ return wls_us_ctx;
+
+ strncpy(temp, ifacename, WLS_DEV_SHM_NAME_LEN - 1);
+ PLIB_INFO("Open %s (DPDK memzone)\n", temp);
+
+ static const struct rte_memzone *mng_memzone;
+ mng_memzone = rte_memzone_lookup(temp);
+ if ((mng_memzone == NULL)&&(RTE_PROC_PRIMARY==rte_eal_process_type())) {
+ wls_initialize(temp, nWlsMemorySize);
+ }
+
+ mng_memzone = rte_memzone_lookup(temp);
+ if (mng_memzone == NULL) {
+ PLIB_ERR("Cannot initialize wls shared memory: %s\n", temp);
+ return NULL;
+ }
+
+ pMemZone = ((uint8_t *)mng_memzone->addr) + nWlsMemorySize;
+ PLIB_INFO("nWlsMemorySize is %llu\n", nWlsMemorySize);
+ PLIB_INFO("WLS_Open Dual 1 %p\n", pMemZone);
+ if ((pWls_us = wls_create_us_ctx((wls_drv_ctx_t *)pMemZone)) == NULL) {
+ PLIB_ERR("WLS_Open Dual 1 failed to create context\n");
+ return NULL;
+ }
+ else
+ {
+ PLIB_DEBUG("Local: pWls_us %p\n", pWls_us);
+
+ pWls_us->padding_wls_us_user_space_va = 0LL;
+ pWls_us->wls_us_user_space_va = pWls_us;
+ pWls_us->wls_us_ctx_size = sizeof (*pWls_us);
+ gWlsMemorySize = nWlsMemorySize;
+
+ wls_mutex_init(&wls_put_lock);
+ wls_mutex_init(&wls_get_lock);
+
+ pWls_us->mode = mode;
+ pWls_us->secmode = WLS_SEC_MASTER;
+ pWls_us->dualMode = WLS_DUAL_MODE;
+ PLIB_INFO("Mode %d SecMode %d \n", pWls_us->mode, pWls_us->secmode);
+
+ PLIB_INFO("WLS shared management memzone 1: %s\n", temp);
+ strncpy(pWls_us->wls_dev_name, temp, WLS_DEV_SHM_NAME_LEN - 1);
+ wls_us_ctx = pWls_us;
+ PLIB_INFO("pWLs_us is %p\n", wls_us_ctx);
+ *handle1 = pWls_us; // Now the first context is for L1-FT_iapi
+ }
+
+ // Create second context to support the second wls shared memory interface
+ if ((pWls_us1 = wls_create_us_ctx((wls_drv_ctx_t *)pMemZone)) == NULL) {
+ PLIB_ERR("WLS_Open Dual failed to create context 1\n");
+ return NULL;
+ }
+ else
+ {
+ PLIB_DEBUG("Local: pWls_us1 %p\n", pWls_us1);
+ pWls_us1->padding_wls_us_user_space_va = 0LL;
+ pWls_us1->wls_us_user_space_va = pWls_us1;
+ pWls_us1->wls_us_ctx_size = sizeof (*pWls_us1);
+ gWlsMemorySize = nWlsMemorySize;
+
+ wls_mutex_init(&wls_put_lock1);
+ wls_mutex_init(&wls_get_lock1);
+
+ pWls_us1->mode = mode;
+ pWls_us1->secmode = WLS_SEC_NA;
+ PLIB_INFO("Mode %d Secmode %d\n", pWls_us1->mode, pWls_us1->secmode);
+
+ PLIB_INFO("WLS shared management memzone 2: %s\n", temp);
+ strncpy(pWls_us1->wls_dev_name, temp, WLS_DEV_SHM_NAME_LEN - 1);
+ wls_us_ctx1 = pWls_us1; // Now the second context is for the L2-FT_fapi
+ PLIB_INFO("pWLs_us1 is %p\n", wls_us_ctx1);
+
+ }
+
+ return wls_us_ctx1; // returning second context preserves the L2 legacy code
+}
+
+int WLS_Ready(void* h)
+{
+ wls_us_ctx_t *pWls_us = (wls_us_ctx_t*) h;
+ wls_us_ctx_t *pWls_usRem = (wls_us_ctx_t*) pWls_us->dst_user_va;
+
+ if (!wls_us_ctx) {
+ PLIB_ERR("Library was not opened\n");
+ return -1;
+ }
+
+ if (pWls_usRem->wls_us_private.pid) {
+ return 0;
+ }
+ return -1;
+}
+
+int WLS_Ready1(void* h)
+{
+ wls_us_ctx_t *pWls_us = (wls_us_ctx_t*) h;
+ wls_us_ctx_t *pWls_usRem = (wls_us_ctx_t*) pWls_us->dst_user_va;
+
+ if (!wls_us_ctx1) {
+ PLIB_ERR("Library was not opened for Context 1\n");
+ return -1;
+ }
+
+ if (pWls_usRem->wls_us_private.pid) {
+ return 0;
+ }
+ return -1;
+}
+
+int WLS_Close(void* h)
+{
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+ int ret = 0;
+ uint8_t *pMemZone;
+ wls_drv_ctx_t *pDrv_ctx;
+
+ if (!wls_us_ctx) {
+ PLIB_ERR("Library was not opened\n");
+ return -1;
+ }
+
+ if (wls_check_ctx(h))
+ return -1;
+
+ static const struct rte_memzone *mng_memzone;
+ mng_memzone = rte_memzone_lookup(pWls_us->wls_dev_name);
+ if (mng_memzone == NULL) {
+ PLIB_ERR("Cannot find mng memzone: %s %s\n",
+ pWls_us->wls_dev_name, rte_strerror(rte_errno));
+ return -1;
+ }
+
+ pMemZone = ((uint8_t *)mng_memzone->addr) + gWlsMemorySize;
+ pDrv_ctx = (wls_drv_ctx_t *)pMemZone;
+
+ PLIB_INFO("WLS_Close\n");
+ if ((ret = wls_destroy_us_ctx(pWls_us, pDrv_ctx)) < 0) {
+ PLIB_ERR("Close failed [%d]\n", ret);
+ return ret;
+ }
+
+ wls_mutex_destroy(&wls_put_lock);
+ wls_mutex_destroy(&wls_get_lock);
+
+ wls_us_ctx = NULL;
+
+ if (0 == pDrv_ctx->nWlsClients) {
+ wls_mutex_destroy(&pDrv_ctx->mng_mutex);
+ rte_memzone_free(mng_memzone);
+ }
+ return 0;
+}
+
+int WLS_Close1(void* h)
+{
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+ int ret = 0;
+ uint8_t *pMemZone;
+ wls_drv_ctx_t *pDrv_ctx;
+
+ if (!wls_us_ctx1) {
+ PLIB_ERR("Library was not opened\n");
+ return -1;
+ }
+
+ if (wls_check_ctx1(h))
+ return -1;
+
+ static const struct rte_memzone *mng_memzone;
+ mng_memzone = rte_memzone_lookup(pWls_us->wls_dev_name);
+ if (mng_memzone == NULL) {
+ PLIB_ERR("Cannot find mng memzone: %s %s\n",
+ pWls_us->wls_dev_name, rte_strerror(rte_errno));
+ return -1;
+ }
+
+ pMemZone = ((uint8_t *)mng_memzone->addr) + gWlsMemorySize;
+ pDrv_ctx = (wls_drv_ctx_t *)pMemZone;
+
+ PLIB_INFO("WLS_Close1\n");
+ if ((ret = wls_destroy_us_ctx1(pWls_us, pDrv_ctx)) < 0) {
+ PLIB_ERR("Close failed [%d]\n", ret);
+ return ret;
+ }
+
+ wls_mutex_destroy(&wls_put_lock1);
+ wls_mutex_destroy(&wls_get_lock1);
+
+ wls_us_ctx1 = NULL;
+
+ if (0 == pDrv_ctx->nWlsClients) {
+ wls_mutex_destroy(&pDrv_ctx->mng_mutex);
+ rte_memzone_free(mng_memzone);
+ }
+ return 0;
+}
+
+void* WLS_Alloc(void* h, unsigned int size)
+{
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+ static const struct rte_memzone *mng_memzone;
+ long nHugePage;
+ void *pvirtAddr = NULL;
+ int count;
+
+ if ((NULL != hp_memzone)&&(pWls_us->dualMode != WLS_DUAL_MODE)) {
+ PLIB_ERR("Memory zone already reserved\n");
+ return hp_memzone->addr;
+ }
+
+ hugepage_tabl_t* pHugePageTlb = &pWls_us->hugepageTbl[0];
+ hugepage_tabl_t* pHugePageTlb1 = &pWls_us->hugepageTbl[1];
+ hugepage_tabl_t* pHugePageTlb2 = &pWls_us->hugepageTbl[2];
+
+ PLIB_INFO("hugePageSize on the system is %ld\n", hugePageSize);
+
+ /* calculate total number of hugepages */
+ nHugePage = DIV_ROUND_OFFSET(size, hugePageSize);
+
+ if (nHugePage >= MAX_N_HUGE_PAGES) {
+ PLIB_INFO("not enough hugepages: need %ld system has %d\n", nHugePage, MAX_N_HUGE_PAGES);
+ return NULL;
+ }
+
+ mng_memzone = rte_memzone_lookup(pWls_us->wls_dev_name);
+ if (mng_memzone == NULL) {
+ PLIB_ERR("Cannot initialize wls shared memory: %s\n", pWls_us->wls_dev_name);
+ return NULL;
+ }
+
+ hp_memzone = (struct rte_memzone *)mng_memzone;
+ pvirtAddr = (void *)hp_memzone->addr;
+ PLIB_DEBUG("pvirtAddr is %p\n", pvirtAddr);
+ if (rte_eal_process_type() == RTE_PROC_PRIMARY) {
+ memset(pvirtAddr, 0, sizeof (wls_drv_ctx_t));
+ }
+
+ for (count = 0; count < nHugePage; count++) {
+ /*Increment virtual address to next hugepage to create table*/
+ pHugePageTlb[count].pageVa = ((unsigned char*) pvirtAddr + \
+ (count * hugePageSize));
+ /*Creating dummy page fault in process for each page
+ inorder to get pagemap*/
+ *(unsigned char*) pHugePageTlb[count].pageVa = 1;
+
+ if (wls_VirtToPhys((uint64_t*) pHugePageTlb[count].pageVa,
+ &pHugePageTlb[count].pagePa) == -1) {
+ PLIB_ERR("Virtual to physical conversion failed\n");
+ return NULL;
+ }
+ }
+
+ PLIB_DEBUG("count is %d, pHugePageTlb->pageVa is %p pHugePageTlb1->pageVa is %p pHugePageTlb2->pageVa is %p\n",count, pHugePageTlb->pageVa, pHugePageTlb1->pageVa, pHugePageTlb2->pageVa);
+ PLIB_INFO("WLS_Alloc [%d] bytes\n", size);
+
+ pWls_us->HugePageSize = (uint32_t) hugePageSize;
+ pWls_us->alloc_buffer = pvirtAddr;
+ pWls_us->alloc_size = (uint32_t) (nHugePage * hugePageSize);
+
+ if ((pWls_us->mode == WLS_MASTER_CLIENT)||(pWls_us->secmode == WLS_SEC_MASTER)) {
+ wls_us_ctx_t *pWls_usRem = (wls_us_ctx_t*) pWls_us->dst_user_va;
+ PLIB_INFO("Connecting to remote peer ...\n");
+ while (pWls_usRem->wls_us_private.pid == 0) { // wait for slave
+ }
+
+ PLIB_INFO("Connected to remote peer\n");
+ pWls_us->dst_user_va = (uint64_t) pWls_usRem;
+ }
+ return pvirtAddr;
+}
+
+int WLS_Free(void* h, PVOID pMsg)
+{
+ static const struct rte_memzone *mng_memzone;
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+ wls_drv_ctx_t *pDrv_ctx;
+
+ mng_memzone = rte_memzone_lookup(pWls_us->wls_dev_name);
+ if (mng_memzone == NULL) {
+ PLIB_ERR("Cannot find mng memzone: %s %s\n",
+ pWls_us->wls_dev_name, rte_strerror(rte_errno));
+ return -1;
+ }
+ pDrv_ctx = mng_memzone->addr;
+
+ if (pMsg != pWls_us->alloc_buffer) {
+ PLIB_ERR("incorrect pMsg %p [expected %p]\n", pMsg, pWls_us->alloc_buffer);
+ return -1;
+ }
+
+ if ((pWls_us->mode == WLS_MASTER_CLIENT)||(pWls_us->mode == WLS_SEC_MASTER)) {
+ if (pWls_us->dst_user_va) {
+ pWls_us->dst_user_va = 0;
+ }
+ }
+
+ PLIB_DEBUG("WLS_Free %s\n", shm_name);
+ if ( (1 == pDrv_ctx->nWlsClients) && hp_memzone)
+ hp_memzone = NULL;
+ return 0;
+}
+
+int WLS_Put(void *h, unsigned long long pMsg, unsigned int MsgSize, unsigned short MsgTypeID, unsigned short Flags)
+{
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+ int ret = 0;
+
+ if (wls_check_ctx(h))
+ return -1;
+
+ if (!WLS_IS_ONE_HUGE_PAGE(pMsg, MsgSize, hugePageSize)) {
+ PLIB_ERR("WLS_Put input error: buffer is crossing 2MB page boundary %lx size %u\n",
+ (U64) pMsg, MsgSize);
+ }
+
+ wls_mutex_lock(&wls_put_lock);
+
+ if ((WLS_FLAGS_MASK & Flags)) { // multi block transaction
+ if (Flags & WLS_TF_SYN) {
+ PLIB_DEBUG("WLS_SG_FIRST\n");
+ if (WLS_MsgEnqueue(&pWls_us->put_queue, pMsg, MsgSize, MsgTypeID,
+ Flags, NULL, (void*) pWls_us)) {
+ PLIB_DEBUG("WLS_Get %lx %d type %d\n", (U64) pMsg, MsgSize, MsgTypeID);
+ }
+ } else if ((Flags & WLS_TF_SCATTER_GATHER)
+ && !(Flags & WLS_TF_SYN)
+ && !(Flags & WLS_TF_FIN)) {
+ PLIB_DEBUG("WLS_SG_NEXT\n");
+ if (WLS_MsgEnqueue(&pWls_us->put_queue, pMsg, MsgSize, MsgTypeID,
+ Flags, NULL, (void*) pWls_us)) {
+ PLIB_DEBUG("WLS_Put %lx %d type %d\n", (U64) pMsg, MsgSize, MsgTypeID);
+ }
+ } else if (Flags & WLS_TF_FIN) {
+ if (WLS_MsgEnqueue(&pWls_us->put_queue, pMsg, MsgSize, MsgTypeID,
+ Flags, NULL, (void*) pWls_us)) {
+ PLIB_DEBUG("WLS_Put %lx %d type %d\n", (U64) pMsg, MsgSize, MsgTypeID);
+ }
+
+ PLIB_DEBUG("List: call wls_process_put\n");
+ if (pWls_us->dst_user_va) {
+ if ((ret = wls_process_put(pWls_us, (wls_us_ctx_t*) pWls_us->dst_user_va)) < 0) {
+ PLIB_ERR("Put failed [%d]\n", ret);
+ wls_mutex_unlock(&wls_put_lock);
+ return -1;
+ }
+ }
+ } else
+ PLIB_ERR("unsaported flags %x\n", WLS_FLAGS_MASK & Flags);
+ } else { // one block transaction
+ if (WLS_MsgEnqueue(&pWls_us->put_queue, pMsg, MsgSize, MsgTypeID,
+ Flags, NULL, (void*) pWls_us)) {
+ PLIB_DEBUG("WLS_Put %lx %d type %d\n", (U64) pMsg, MsgSize, MsgTypeID);
+ }
+
+ PLIB_DEBUG("One block: call wls_process_put\n");
+ if (likely(pWls_us->dst_user_va)) {
+ if ((ret = wls_process_put(pWls_us, (wls_us_ctx_t*) pWls_us->dst_user_va)) < 0) {
+ PLIB_ERR("Put failed [%d]\n", ret);
+ wls_mutex_unlock(&wls_put_lock);
+ return -1;
+ }
+ } else {
+ PLIB_ERR("Destination address is empty\n");
+ wls_mutex_unlock(&wls_put_lock);
+ return -1;
+ }
+ }
+ wls_mutex_unlock(&wls_put_lock);
+
+ return 0;
+}
+
+int WLS_Put1(void *h, unsigned long long pMsg, unsigned int MsgSize, unsigned short MsgTypeID, unsigned short Flags)
+{
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+ int ret = 0;
+
+ if (wls_check_ctx1(h))
+ return -1;
+
+ if (!WLS_IS_ONE_HUGE_PAGE(pMsg, MsgSize, hugePageSize)) {
+ PLIB_ERR("WLS_Put input error: buffer is crossing 2MB page boundary %lx size %u\n",
+ (U64) pMsg, MsgSize);
+ }
+
+ wls_mutex_lock(&wls_put_lock1);
+
+ if ((WLS_FLAGS_MASK & Flags)) { // multi block transaction
+ if (Flags & WLS_TF_SYN) {
+ PLIB_DEBUG("WLS_SG_FIRST\n");
+ if (WLS_MsgEnqueue(&pWls_us->put_queue, pMsg, MsgSize, MsgTypeID,
+ Flags, NULL, (void*) pWls_us)) {
+ PLIB_DEBUG("WLS_Get %lx %d type %d\n", (U64) pMsg, MsgSize, MsgTypeID);
+ }
+ } else if ((Flags & WLS_TF_SCATTER_GATHER)
+ && !(Flags & WLS_TF_SYN)
+ && !(Flags & WLS_TF_FIN)) {
+ PLIB_DEBUG("WLS_SG_NEXT\n");
+ if (WLS_MsgEnqueue(&pWls_us->put_queue, pMsg, MsgSize, MsgTypeID,
+ Flags, NULL, (void*) pWls_us)) {
+ PLIB_DEBUG("WLS_Put %lx %d type %d\n", (U64) pMsg, MsgSize, MsgTypeID);
+ }
+ } else if (Flags & WLS_TF_FIN) {
+ if (WLS_MsgEnqueue(&pWls_us->put_queue, pMsg, MsgSize, MsgTypeID,
+ Flags, NULL, (void*) pWls_us)) {
+ PLIB_DEBUG("WLS_Put %lx %d type %d\n", (U64) pMsg, MsgSize, MsgTypeID);
+ }
+
+ PLIB_DEBUG("List: call wls_process_put\n");
+ if (pWls_us->dst_user_va) {
+ if ((ret = wls_process_put(pWls_us, (wls_us_ctx_t*) pWls_us->dst_user_va)) < 0) {
+ PLIB_ERR("Put failed [%d]\n", ret);
+ wls_mutex_unlock(&wls_put_lock1);
+ return -1;
+ }
+ }
+ } else
+ PLIB_ERR("unsupported flags %x\n", WLS_FLAGS_MASK & Flags);
+ } else { // one block transaction
+ if (WLS_MsgEnqueue(&pWls_us->put_queue, pMsg, MsgSize, MsgTypeID,
+ Flags, NULL, (void*) pWls_us)) {
+ PLIB_DEBUG("WLS_Put %lx %d type %d\n", (U64) pMsg, MsgSize, MsgTypeID);
+ }
+
+ PLIB_DEBUG("One block: call wls_process_put\n");
+ if (likely(pWls_us->dst_user_va)) {
+ if ((ret = wls_process_put(pWls_us, (wls_us_ctx_t*) pWls_us->dst_user_va)) < 0) {
+ PLIB_ERR("Put failed [%d]\n", ret);
+ wls_mutex_unlock(&wls_put_lock1);
+ return -1;
+ }
+ } else {
+ PLIB_ERR("Destination address is empty\n");
+ wls_mutex_unlock(&wls_put_lock1);
+ return -1;
+ }
+ }
+ wls_mutex_unlock(&wls_put_lock1);
+
+ return 0;
+}
+
+int WLS_Check(void* h)
+{
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+
+ if (wls_check_ctx(h))
+ return 0;
+
+ return WLS_GetNumItemsInTheQueue(&pWls_us->get_queue);
+}
+
+int WLS_Check1(void* h)
+{
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+
+ if (wls_check_ctx1(h))
+ return 0;
+
+ return WLS_GetNumItemsInTheQueue(&pWls_us->get_queue);
+}
+
+unsigned long long WLS_Get(void* h, unsigned int *MsgSize, unsigned short *MsgTypeID, unsigned short *Flags)
+{
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+ WLS_MSG_HANDLE hMsg;
+ uint64_t pMsg = (uint64_t) NULL;
+
+ if (wls_check_ctx(h))
+ {
+ PLIB_ERR("WLS_Get fails wls_check_ctx\n");
+ return 0;
+ }
+
+ wls_mutex_lock(&wls_get_lock);
+
+ if (WLS_MsgDequeue(&pWls_us->get_queue, &hMsg, NULL, (void*) pWls_us)) {
+ PLIB_DEBUG("WLS_Get %lx %d type %d\n", (U64) hMsg.pIaPaMsg, hMsg.MsgSize, hMsg.TypeID);
+ pMsg = hMsg.pIaPaMsg;
+ *MsgSize = hMsg.MsgSize;
+ *MsgTypeID = hMsg.TypeID;
+ *Flags = hMsg.flags;
+ }
+
+ wls_mutex_unlock(&wls_get_lock);
+
+ return pMsg;
+}
+
+unsigned long long WLS_Get1(void* h, unsigned int *MsgSize, unsigned short *MsgTypeID, unsigned short *Flags)
+{
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+ WLS_MSG_HANDLE hMsg;
+ uint64_t pMsg = (uint64_t) NULL;
+
+ if (wls_check_ctx1(h))
+ return 0;
+
+ wls_mutex_lock(&wls_get_lock1);
+
+ if (WLS_MsgDequeue(&pWls_us->get_queue, &hMsg, NULL, (void*) pWls_us)) {
+ PLIB_DEBUG("WLS_Get %lx %d type %d\n", (U64) hMsg.pIaPaMsg, hMsg.MsgSize, hMsg.TypeID);
+ pMsg = hMsg.pIaPaMsg;
+ *MsgSize = hMsg.MsgSize;
+ *MsgTypeID = hMsg.TypeID;
+ *Flags = hMsg.flags;
+ }
+
+ wls_mutex_unlock(&wls_get_lock1);
+
+ return pMsg;
+}
+
+int WLS_WakeUp(void* h)
+{
+ if (!wls_us_ctx) {
+ PLIB_ERR("Library was not opened\n");
+ return -1;
+ }
+ if (wls_check_ctx(h))
+ return -1;
+
+ PLIB_DEBUG("WLS_WakeUp\n");
+
+ return wls_process_wakeup(h);
+
+
+ return 0;
+}
+
+int WLS_WakeUp1(void* h)
+{
+ if (!wls_us_ctx1) {
+ PLIB_ERR("Library was not opened\n");
+ return -1;
+ }
+ if (wls_check_ctx1(h))
+ return -1;
+
+ PLIB_DEBUG("WLS_WakeUp1\n");
+
+ return wls_process_wakeup1(h);
+
+
+ return 0;
+}
+
+int WLS_Wait(void* h)
+{
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+
+ if (!wls_us_ctx || (wls_us_ctx != pWls_us)) {
+ PLIB_ERR("Library was not opened\n");
+ return -1;
+ }
+
+ return wls_process_wait(h);
+}
+
+int WLS_Wait1(void* h)
+{
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+
+ if (!wls_us_ctx1 || (wls_us_ctx1 != pWls_us)) {
+ PLIB_ERR("Library was not opened\n");
+ return -1;
+ }
+
+ return wls_process_wait1(h);
+}
+
+unsigned long long WLS_WGet(void* h, unsigned int *MsgSize, unsigned short *MsgTypeID, unsigned short *Flags)
+{
+ uint64_t pRxMsg = WLS_Get(h, MsgSize, MsgTypeID, Flags);
+
+ if (pRxMsg)
+ return pRxMsg;
+
+ WLS_Wait(h);
+ return WLS_Get(h, MsgSize, MsgTypeID, Flags);
+}
+
+unsigned long long WLS_WGet1(void* h, unsigned int *MsgSize, unsigned short *MsgTypeID, unsigned short *Flags)
+{
+ uint64_t pRxMsg = WLS_Get1(h, MsgSize, MsgTypeID, Flags);
+
+ if (pRxMsg)
+ return pRxMsg;
+
+ WLS_Wait1(h);
+ return WLS_Get1(h, MsgSize, MsgTypeID, Flags);
+}
+
+unsigned long long WLS_VA2PA(void* h, PVOID pMsg)
+{
+ uint64_t ret = 0;
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+
+ unsigned long alloc_base;
+ hugepage_tabl_t* pHugePageTlb;
+ uint64_t hugePageBase;
+ uint64_t hugePageOffet;
+ unsigned int count = 0;
+
+ uint64_t HugePageMask = ((unsigned long) pWls_us->HugePageSize - 1);
+ if (pWls_us->alloc_buffer == NULL) {
+ PLIB_ERR("WLS_VA2PA: nothing was allocated [%ld]\n", ret);
+ return (uint64_t) ret;
+ }
+
+ alloc_base = (unsigned long) pWls_us->alloc_buffer;
+
+ pHugePageTlb = &pWls_us->hugepageTbl[0];
+
+ hugePageBase = (uint64_t) pMsg & ~HugePageMask;
+ hugePageOffet = (uint64_t) pMsg & HugePageMask;
+
+ count = (hugePageBase - alloc_base) / pWls_us->HugePageSize;
+ PLIB_DEBUG("WLS_VA2PA %lx base %llx off %llx count %u\n", (unsigned long) pMsg,
+ (uint64_t) hugePageBase, (uint64_t) hugePageOffet, count);
+
+ ret = pHugePageTlb[count].pagePa + hugePageOffet;
+
+ //printf(" WLS_VA2PA: %p -> %p HugePageSize[%d] HugePageMask[%p] count[%d] pagePa[%p] hugePageBase[%p] alloc_buffer[%p] hugePageOffet[%lld]\n",
+ // pMsg, (void*)ret, pWls_us->HugePageSize, (void*)HugePageMask, count, (void*)pHugePageTlb[count].pagePa, (void*)hugePageBase, pWls_us->alloc_buffer, hugePageOffet);
+
+ return (uint64_t) ret;
+}
+
+void* WLS_PA2VA(void* h, unsigned long long pMsg)
+{
+ unsigned long ret = 0;
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+
+ hugepage_tabl_t* pHugePageTlb;
+ uint64_t hugePageBase;
+ uint64_t hugePageOffet;
+ unsigned int count;
+ int i;
+ uint64_t HugePageMask = ((uint64_t) pWls_us->HugePageSize - 1);
+
+ if (pWls_us->alloc_buffer == NULL) {
+ PLIB_ERR("WLS_PA2VA: nothing was allocated [%ld]\n", ret);
+ return (void*) ret;
+ }
+
+ pHugePageTlb = &pWls_us->hugepageTbl[0];
+
+ hugePageBase = (uint64_t) pMsg & ~HugePageMask;
+ hugePageOffet = (uint64_t) pMsg & HugePageMask;
+
+ count = pWls_us->alloc_size / pWls_us->HugePageSize;
+
+ PLIB_DEBUG("WLS_PA2VA %llx base %llx off %llx count %d\n", (uint64_t) pMsg,
+ (uint64_t) hugePageBase, (uint64_t) hugePageOffet, count);
+
+ for (i = 0; i < count; i++) {
+ if (pHugePageTlb[i].pagePa == hugePageBase) {
+ ret = (unsigned long) pHugePageTlb[i].pageVa;
+ ret += hugePageOffet;
+ return (void*) ret;
+ }
+ }
+
+ //printf(" WLS_VA2PA: %p -> %p HugePageSize[%d] HugePageMask[%p] count[%d] pagePa[%p] hugePageBase[%p] alloc_buffer[%p] hugePageOffet[%lld]\n",
+ // (void*)pMsg, (void*)ret, pWls_us->HugePageSize, (void*)HugePageMask, count, (void*)pHugePageTlb[count].pagePa, (void*)hugePageBase, pWls_us->alloc_buffer, hugePageOffet);
+
+ return (void*) (ret);
+}
+
+int WLS_EnqueueBlock(void* h, unsigned long long pMsg)
+{
+ int ret = 0;
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+
+ if (!wls_us_ctx) {
+ PLIB_ERR("Library was not opened\n");
+ return -1;
+ }
+
+
+ if (pMsg == 0) {
+ PLIB_ERR("WLS_EnqueueBlock: Null\n");
+ return -1;
+ }
+
+ if (pWls_us->dst_user_va) {
+ wls_us_ctx_t* pDstWls_us = (wls_us_ctx_t*) pWls_us->dst_user_va;
+ ret = SFL_WlsEnqueue(&pDstWls_us->ul_free_block_pq, pMsg, NULL, pWls_us);
+ if (ret == 1) {
+ unsigned long* ptr = (unsigned long*) WLS_PA2VA(pWls_us, pMsg);
+ if (ptr) {
+ *ptr = 0xFFFFFFFFFFFFFFFF;
+ }
+ }
+ } else
+ ret = -1;
+
+ PLIB_DEBUG("SFL_WlsEnqueue %d\n", ret);
+ return ret;
+}
+
+int WLS_EnqueueBlock1(void* h, unsigned long long pMsg)
+{
+ int ret = 0;
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+
+ if (!wls_us_ctx1) {
+ PLIB_ERR("Library was not opened for a second context\n");
+ return -1;
+ }
+
+ if (pWls_us->mode == WLS_SLAVE_CLIENT) {
+ PLIB_ERR("Slave doesn't support memory allocation\n");
+ return -1;
+ }
+
+ if (pMsg == 0) {
+ PLIB_ERR("WLS_EnqueueBlock: Null\n");
+ return -1;
+ }
+
+ if (pWls_us->dst_user_va) {
+ wls_us_ctx_t* pDstWls_us = (wls_us_ctx_t*) pWls_us->dst_user_va;
+ ret = SFL_WlsEnqueue(&pDstWls_us->ul_free_block_pq, pMsg, NULL, pWls_us);
+ if (ret == 1) {
+ unsigned long* ptr = (unsigned long*) WLS_PA2VA(pWls_us, pMsg);
+ if (ptr) {
+ *ptr = 0xFFFFFFFFFFFFFFFF;
+ }
+ }
+ } else
+ ret = -1;
+
+ PLIB_DEBUG("SFL_WlsEnqueue %d\n", ret);
+ return ret;
+}
+
+
+unsigned long long WLS_DequeueBlock(void* h)
+{
+ unsigned long long retval = 0;
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+
+ if ((pWls_us->mode == WLS_SLAVE_CLIENT)&&(pWls_us->secmode==WLS_SEC_NA))
+ {
+ // local
+ return SFL_WlsDequeue(&pWls_us->ul_free_block_pq, NULL, h);
+ }
+ if (!pWls_us->dst_user_va)
+ {
+ return retval;
+ }
+ // remote
+ wls_us_ctx_t* pDstWls_us = (wls_us_ctx_t*) pWls_us->dst_user_va;
+ retval = SFL_WlsDequeue(&pDstWls_us->ul_free_block_pq, NULL, pDstWls_us);
+ if (retval) {
+ unsigned long* ptr = (unsigned long*) WLS_PA2VA(pWls_us, retval);
+ if (ptr) {
+ if (*ptr != 0xFFFFFFFFFFFFFFFF) {
+ PLIB_ERR("WLS_EnqueueBlock: incorrect content pa: 0x%016lx: 0x%016lx\n",
+ (unsigned long) retval, *ptr);
+ }
+ }
+ }
+
+ return retval;
+}
+
+int WLS_NumBlocks(void* h)
+{
+ wls_us_ctx_t* pWls_us = (wls_us_ctx_t*) h;
+ int n = 0;
+
+ if (pWls_us->mode == WLS_SLAVE_CLIENT) {
+ // local
+ n = SFL_GetNumItemsInTheQueue(&pWls_us->ul_free_block_pq);
+ } else if (pWls_us->dst_user_va) {
+ // remote
+ wls_us_ctx_t* pDstWls_us = (wls_us_ctx_t*) pWls_us->dst_user_va;
+ n = SFL_GetNumItemsInTheQueue(&pDstWls_us->ul_free_block_pq);
+ }
+
+ return n;
+}