/*
 * Copyright (c) 2013-2015, Texas Instruments Incorporated
 * 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 Texas Instruments Incorporated 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.
 */

/*
 *  ======== GateMP_daemon.c ========
 */


/* Standard headers */
#include <ti/ipc/Std.h>

/* Qnx specific header files */
#include <pthread.h>

/* System headers */
#include <sys/mman.h>
#include <stdlib.h>
#include <assert.h>

/* Module level headers */
#include <ti/ipc/GateMP.h>
#include <ti/ipc/NameServer.h>
#include <ti/ipc/MultiProc.h>
#include <_MultiProc.h>
#include <GateMP_config.h>
#include <_GateMP.h>

#include <ti/syslink/utils/IGateProvider.h>
#include <ti/syslink/inc/_GateMP_daemon.h>
#include <_IpcLog.h>

#if defined (__cplusplus)
extern "C" {
#endif

/* How often to poll for default gate during setup (in microseconds) */
#define POLL_INTERVAL         10000

/*
 * Number of times to repeatedly poll for default gate in GateMP_setup before
 * timing out.
 * The default adds up to a total of roughly 2 seconds, which means the slave
 * core that owns SR0 must have setup GateMP within that time. Otherwise the
 * driver will simply move on without setting up GateMP.
 */
#define SETUP_TIMEOUT         200

#define NUM_INFO_FIELDS       6    /* Number of fields in info entry */

/* Values used to populate the resource 'inUse' arrays */
#define UNUSED          ((UInt8)0)
#define USED            ((UInt8)1)
#define RESERVED        ((UInt8)-1)

/* Name of GateMP's nameserver */
#define GateMP_NAMESERVER  "GateMP"

/* =============================================================================
 * Structures & Enums
 * =============================================================================
 */

/* structure for GateMP module state */
typedef struct {
    Int                numRemoteSystem;
    Int                numRemoteCustom1;
    Int                numRemoteCustom2;
    UInt8 *            remoteSystemInUse;
    UInt8 *            remoteCustom1InUse;
    UInt8 *            remoteCustom2InUse;
    GateMP_Handle      defaultGate;
    NameServer_Handle  nameServer;
    Bool               isSetup;
    UInt32             startCount;   /* number of times GateMP is started */
    pthread_mutex_t    lock;
} GateMP_ModuleObject;

/* Internal functions */
static Int GateMP_openDefaultGate(GateMP_Handle *handlePtr,
    UInt16 * creatorProcId);
static Int GateMP_closeDefaultGate(GateMP_Handle *handlePtr);

/* =============================================================================
 *  Globals
 * =============================================================================
 */
/*
 * GateMP_state
 */
static GateMP_ModuleObject GateMP_state = {
    .numRemoteSystem                 = 0,
    .numRemoteCustom1                = 0,
    .numRemoteCustom2                = 0,
    .remoteSystemInUse               = NULL,
    .remoteCustom1InUse              = NULL,
    .remoteCustom2InUse              = NULL,
    .defaultGate                     = NULL,
    .nameServer                      = NULL,
    .isSetup                         = FALSE,
    .startCount                      = 0,
    .lock                            = PTHREAD_RMUTEX_INITIALIZER
};

static GateMP_ModuleObject * GateMP_module = &GateMP_state;

/* =============================================================================
 * APIS
 * =============================================================================
 */

/* Function to setup the gatemp module. */
Int GateMP_setup(Int32 * sr0ProcId)
{
    Int               status = GateMP_S_SUCCESS;
    NameServer_Params params;
    UInt32            nsValue[NUM_INFO_FIELDS];
    UInt32            len;
    UInt              timeout = SETUP_TIMEOUT;
    UInt16            procId;

    NameServer_Params_init(&params);
    params.maxRuntimeEntries = MAX_RUNTIME_ENTRIES;
    params.maxNameLen = MAX_NAME_LEN;

    /* Assume info entry has more fields than other entries */
    params.maxValueLen = NUM_INFO_FIELDS * sizeof(UInt32);

    pthread_mutex_lock(&GateMP_module->lock);

    if (GateMP_module->isSetup) {
        pthread_mutex_unlock(&GateMP_module->lock);
        return (GateMP_S_ALREADYSETUP);
    }

    GateMP_module->nameServer =
                NameServer_create(GateMP_NAMESERVER, &params);

    if (GateMP_module->nameServer == NULL) {
        status = GateMP_E_FAIL;
        LOG0("GateMP_setup: NameServer_create failed\n");
    }

    if (status == GateMP_S_SUCCESS) {
        /* The default gate creator is the owner of SR0 */
        while (((status = GateMP_openDefaultGate(&GateMP_module->defaultGate,
            &procId)) == GateMP_E_NOTFOUND) && (timeout > 0)) {
            usleep(POLL_INTERVAL);
            timeout--;
        }

        if ((status < 0) && (status != GateMP_E_NOTFOUND)) {
            LOG0("GateMP_setup: failed to open default gate\n");
            status = GateMP_E_FAIL;
        }
        else if (status == GateMP_S_SUCCESS) {
            *sr0ProcId = procId;
        }
        else {
            /* If default gate is not found, return GateMP_E_NOTFOUND */
        }
    }

    if (status == GateMP_S_SUCCESS) {
        /* Process global info NameServer entry */
        len = sizeof(nsValue);

        status = NameServer_get(GateMP_module->nameServer, "_GateMP_TI_info",
            &nsValue, &len, NULL);

        if (status < 0) {
            LOG0("GateMP_setup: failed to find info entry\n");
            status = GateMP_E_NOTFOUND;
        }
        else {
            GateMP_module->numRemoteSystem = nsValue[3];
            GateMP_module->numRemoteCustom1 = nsValue[4];
            GateMP_module->numRemoteCustom2 = nsValue[5];

            /* Map InUse arrays to daemon's address space */
            if (GateMP_module->numRemoteSystem != 0) {
                GateMP_module->remoteSystemInUse = mmap(NULL,
                    GateMP_module->numRemoteSystem * sizeof (UInt8),
                    (PROT_READ|PROT_WRITE|PROT_NOCACHE),
                    (MAP_PHYS|MAP_SHARED), NOFD, (off_t)nsValue[0]);
                if (GateMP_module->remoteSystemInUse == MAP_FAILED) {
                    GateMP_module->remoteSystemInUse = NULL;
                    status = GateMP_E_MEMORY;
                    LOG0("Failed to map remoteSystemInUse to host address" \
                         " space!");
                }
            }

            if ((status == GateMP_S_SUCCESS) &&
                (GateMP_module->numRemoteCustom1 != 0)) {
                GateMP_module->remoteCustom1InUse = mmap(NULL,
                    GateMP_module->numRemoteCustom1 * sizeof (UInt8),
                    (PROT_READ|PROT_WRITE|PROT_NOCACHE),
                    (MAP_PHYS|MAP_SHARED), NOFD, (off_t)nsValue[1]);
                if (GateMP_module->remoteCustom1InUse == MAP_FAILED) {
                    GateMP_module->remoteCustom1InUse = NULL;
                    status = GateMP_E_MEMORY;
                    LOG0("Failed to map remoteCustom1InUse to host address" \
                        " space!");
                }
            }

            if ((status == GateMP_S_SUCCESS) &&
                (GateMP_module->numRemoteCustom2 != 0)) {
                GateMP_module->remoteCustom2InUse = mmap(NULL,
                    GateMP_module->numRemoteCustom2 * sizeof (UInt8),
                    (PROT_READ|PROT_WRITE|PROT_NOCACHE),
                    (MAP_PHYS|MAP_SHARED), NOFD, (off_t)nsValue[2]);
                if (GateMP_module->remoteCustom2InUse == MAP_FAILED) {
                    GateMP_module->remoteCustom2InUse = NULL;
                    status = GateMP_E_MEMORY;
                    LOG0("Failed to map remoteCustom2InUse to host address" \
                        " space!");
                }
            }
        }
    }

    /* TODO: setup the proxy map */

    /* clean up if error */
    if (status < 0) {
        GateMP_destroy(TRUE);
    }
    else {
        GateMP_module->isSetup = TRUE;
    }

    pthread_mutex_unlock(&GateMP_module->lock);

    return (status);
}

Int GateMP_destroy(Bool forced)
{
    pthread_mutex_lock(&GateMP_module->lock);

    if ((!forced) && (GateMP_module->startCount != 0)) {
        /* Some process on the host is still using GateMP */
        pthread_mutex_unlock(&GateMP_module->lock);
        return (GateMP_E_FAIL);
    }

    if (GateMP_module->remoteSystemInUse) {
        munmap((unsigned int *)GateMP_module->remoteSystemInUse,
            GateMP_module->numRemoteSystem * sizeof (UInt8));
        GateMP_module->remoteSystemInUse = NULL;
    }

    if (GateMP_module->remoteCustom1InUse) {
        munmap((unsigned int *)GateMP_module->remoteCustom1InUse,
            GateMP_module->numRemoteCustom1 * sizeof (UInt8));
        GateMP_module->remoteCustom1InUse = NULL;
    }

    if (GateMP_module->remoteCustom2InUse) {
        munmap((unsigned int *)GateMP_module->remoteCustom2InUse,
            GateMP_module->numRemoteCustom2 * sizeof (UInt8));
        GateMP_module->remoteCustom2InUse = NULL;
    }

    if (GateMP_module->defaultGate) {
        GateMP_closeDefaultGate(&GateMP_module->defaultGate);
    }

    if (GateMP_module->nameServer) {
        NameServer_delete(&GateMP_module->nameServer);
        GateMP_module->nameServer = NULL;
    }

    GateMP_module->isSetup = FALSE;

    pthread_mutex_unlock(&GateMP_module->lock);

    return (GateMP_S_SUCCESS);
}

/* Open default gate during GateMP_setup. Should only be called once */
static Int GateMP_openDefaultGate(GateMP_Handle *handlePtr,
    UInt16 * creatorProcId)
{
    Int             status = GateMP_S_SUCCESS;
    UInt32          len;
    UInt32          nsValue[4];
    GateMP_Object * obj = NULL;
    UInt32          arg;
    UInt32          mask;

    GateMP_RemoteSystemProxy_Params     systemParams;

    /* assert that a valid pointer has been supplied */
    if (handlePtr == NULL) {
        LOG0("GateMP_open: argument cannot be null\n");
        status = GateMP_E_INVALIDARG;
    }

    if (status == GateMP_S_SUCCESS) {
        len = sizeof(nsValue);

        status = NameServer_get(GateMP_module->nameServer, "_GateMP_TI_dGate",
            &nsValue, &len, NULL);

        if (status < 0) {
            *handlePtr = NULL;
            status = GateMP_E_NOTFOUND;
        }
        else {
            arg = nsValue[2];
            mask = nsValue[3];
            *creatorProcId = nsValue[1] >> 16;
        }
    }

    if (status == GateMP_S_SUCCESS) {
        /* allocate the instance object */
        obj = (GateMP_Object *)calloc(1, sizeof (GateMP_Object));
        if (obj != NULL) {
            obj->localGate  = NULL;  /* TODO: create the local gate instance */
            obj->localProtect  = GETLOCAL(mask);
            obj->remoteProtect = GETREMOTE(mask);
            obj->nsKey         = 0;
            obj->numOpens      = 1;
            obj->objType       = Ipc_ObjType_OPENDYNAMIC;
            obj->resourceId    = arg;

            assert(obj->remoteProtect == GateMP_RemoteProtect_SYSTEM);

            /* create the proxy object */
            GateMP_RemoteSystemProxy_Params_init(&systemParams);
            systemParams.resourceId = obj->resourceId;
            systemParams.openFlag = TRUE;

            /*
             * TODO: Currently passing in localProtect instead of localGate,
             * since GateHWSpinlock owns the local gate
             */
            obj->gateHandle = (IGateProvider_Handle)
                GateMP_RemoteSystemProxy_create(obj->localProtect,
                    &systemParams);

            if (obj->gateHandle == NULL) {
                LOG0("GateMP_openDefaultGate: failed to create proxy\n");
                free(obj);
                obj = NULL;
            }
        }
        else {
            LOG0("GateMP_openDefaultGate: Memory allocation failed")
        }

        if (obj == NULL) {
            status = GateMP_E_FAIL;
        }
    }

    /* Return the "opened" GateMP instance  */
    *handlePtr = (GateMP_Handle)obj;

    return status;
}

static Int GateMP_closeDefaultGate(GateMP_Handle *handlePtr)
{
    Int status = GateMP_S_SUCCESS;
    GateMP_Object * obj = *(GateMP_Object **)handlePtr;

    if (obj->gateHandle != NULL) {
        /* Default gate is always of type System when more than 1 processor */
        GateMP_RemoteSystemProxy_delete(
            (GateMP_RemoteSystemProxy_Handle *)&obj->gateHandle);
    }

    free(*handlePtr);
    *handlePtr = NULL;

    return(status);
}

Int GateMP_getFreeResource(GateMP_RemoteProtect type)
{
    IArg   key;
    Bool   flag = FALSE;
    Int    resourceId = -1;
    UInt8* inUse = NULL;
    Int    num = 0;

    /* Remote case */
    switch (type) {
        /* TODO: currently only support System proxy */
        case GateMP_RemoteProtect_SYSTEM:
        case GateMP_RemoteProtect_CUSTOM1:
        case GateMP_RemoteProtect_CUSTOM2:
            inUse = GateMP_module->remoteSystemInUse;
            num = GateMP_module->numRemoteSystem;
            break;

        default:
            LOG0("GateMP_getFreeResource: Invalid remote protection type\n");
            break;
    }

    if (inUse != NULL) {
        assert(GateMP_module->defaultGate != NULL);
        key = GateMP_enter(GateMP_module->defaultGate);

        /*
         *  Find a free resource id. Note: zero is reserved on the
         *  system proxy for the default gate.
         */
        for (resourceId = 0; resourceId < num; resourceId++) {
            /*
             *  If not in-use, set the inUse to TRUE to prevent other
             *  creates from getting this one.
             */
            if (inUse[resourceId] == UNUSED) {
                flag = TRUE;

                /* Denote in shared memory that the resource is used */
                inUse[resourceId] = USED;
                break;
            }
        }

        GateMP_leave(GateMP_module->defaultGate, key);
    }

    if (flag == FALSE) {
        resourceId = -1;
    }

    return (resourceId);
}

Int GateMP_releaseResource(UInt id, GateMP_RemoteProtect type)
{
    Int    status = GateMP_S_SUCCESS;
    IArg   key;
    UInt8* inUse = NULL;
    Int    num = 0;

    /* Remote case */
    switch (type) {
        /* TODO: currently only support System proxy */
        case GateMP_RemoteProtect_SYSTEM:
        case GateMP_RemoteProtect_CUSTOM1:
        case GateMP_RemoteProtect_CUSTOM2:
            inUse = GateMP_module->remoteSystemInUse;
            num = GateMP_module->numRemoteSystem;
            break;

        default:
            LOG0("GateMP_releaseResource: Invalid remote protection type\n");
            status = GateMP_E_FAIL;
            break;
    }

    if ((inUse != NULL) && (id < num)) {
        assert(GateMP_module->defaultGate != NULL);
        key = GateMP_enter(GateMP_module->defaultGate);
        inUse[id] = UNUSED;
        GateMP_leave(GateMP_module->defaultGate, key);
    }
    else {
        /* Should not happen if module is properly setup */
        status = GateMP_E_FAIL;
    }

    return (status);
}

Int GateMP_getNumResources(GateMP_RemoteProtect type)
{
    Int   num = -1;

    /* Remote case */
    switch (type) {
        /* TODO: currently only support System proxy */
        case GateMP_RemoteProtect_SYSTEM:
        case GateMP_RemoteProtect_CUSTOM1:
        case GateMP_RemoteProtect_CUSTOM2:
            num = GateMP_module->numRemoteSystem;
            break;

        default:
            LOG0("GateMP_getNumResources: Invalid remote protection type\n");
            break;
    }

    return (num);
}

NameServer_Handle GateMP_getNameServer(Void)
{
    return (GateMP_module->nameServer);
}

Int GateMP_start(Void)
{
    Int status = GateMP_S_SUCCESS;

    pthread_mutex_lock(&GateMP_module->lock);

    if (GateMP_isSetup()) {
        GateMP_module->startCount++;
    }
    else {
        /* Cannot start GateMP if it is not setup */
        status = GateMP_E_FAIL;
    }

    pthread_mutex_unlock(&GateMP_module->lock);

    return (status);
}

Int GateMP_stop(Void)
{
    Int status = GateMP_S_SUCCESS;

    pthread_mutex_lock(&GateMP_module->lock);

    if (GateMP_isSetup()) {
        GateMP_module->startCount--;
    }
    else {
        /* Cannot stop GateMP if it is not setup */
        status = GateMP_E_FAIL;
    }

    pthread_mutex_unlock(&GateMP_module->lock);

    return (status);
}

Bool GateMP_isSetup(Void)
{
    return (GateMP_module->isSetup);
}

IArg GateMP_enter(GateMP_Handle handle)
{
    GateMP_Object * obj;
    IArg            key;

    obj = (GateMP_Object *)handle;
    key = IGateProvider_enter(obj->gateHandle);

    return(key);
}

Void GateMP_leave(GateMP_Handle handle, IArg key)
{
    GateMP_Object *obj;

    obj = (GateMP_Object *)handle;
    IGateProvider_leave(obj->gateHandle, key);
}