/*
 * Copyright 2018 Rockchip Electronics Co. LTD
 *
 * 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 <sys/mman.h>

#include "mpp_mem.h"
#include "mpp_debug.h"

#include "allocator_ext_dma.h"

typedef struct {
    size_t alignment;
} allocator_ctx;

static MPP_RET allocator_ext_dma_open(void **ctx, MppAllocatorCfg *cfg)
{
    MPP_RET ret = MPP_OK;
    allocator_ctx *p = NULL;

    if (NULL == ctx) {
        mpp_err_f("do not accept NULL input\n");
        return MPP_ERR_NULL_PTR;
    }

    p = mpp_malloc(allocator_ctx, 1);
    if (NULL == p) {
        mpp_err_f("failed to allocate context\n");
        ret = MPP_ERR_MALLOC;
    } else {
        p->alignment = cfg->alignment;
    }

    *ctx = p;
    return ret;
}

static MPP_RET allocator_ext_dma_alloc(void *ctx, MppBufferInfo *info)
{
    if (!ctx || !info) {
        mpp_err_f("found NULL context input\n");
        return MPP_ERR_VALUE;
    }

    return MPP_ERR_PERM;
}

static MPP_RET allocator_ext_dma_free(void *ctx, MppBufferInfo *info)
{
    if (!ctx || !info) {
        mpp_err_f("found NULL context input\n");
        return MPP_ERR_VALUE;
    }

    return MPP_ERR_PERM;
}

static MPP_RET allocator_ext_dma_import(void *ctx, MppBufferInfo *info)
{
    allocator_ctx *p = (allocator_ctx *)ctx;
    mpp_assert(p);
    mpp_assert(info->size);

    if (info->ptr) {
        mpp_err_f("The ext_dma is not used for userptr\n");
        return MPP_ERR_VALUE;
    }

    return ((info->fd < 0) ? MPP_ERR_VALUE : MPP_OK);
}

static MPP_RET allocator_ext_dma_mmap(void *ctx, MppBufferInfo *info)
{
    void *ptr = NULL;
    unsigned long offset = 0L;
    mpp_assert(ctx);
    mpp_assert(info->size);
    mpp_assert(info->fd >= 0);

    if (info->ptr)
        return MPP_OK;

    /*
     * It is insecure to access the first memory page,
     * usually system doesn't allow this behavior.
     */
    ptr = mmap(NULL, info->size, PROT_READ | PROT_WRITE,
               MAP_SHARED, info->fd, offset);
    if (ptr == MAP_FAILED)
        return MPP_ERR_NULL_PTR;

    info->ptr = ptr;

    return MPP_OK;
}

static MPP_RET allocator_ext_dma_release(void *ctx, MppBufferInfo *info)
{
    mpp_assert(ctx);
    mpp_assert(info->size);

    if (info->ptr)
        munmap(info->ptr, info->size);

    info->ptr   = NULL;
    info->hnd   = NULL;
    info->fd    = -1;
    info->size  = 0;

    return MPP_OK;
}

static MPP_RET allocator_ext_dma_close(void *ctx)
{
    if (ctx) {
        mpp_free(ctx);
        return MPP_OK;
    }

    mpp_err_f("found NULL context input\n");
    return MPP_ERR_VALUE;
}

os_allocator allocator_ext_dma = {
    .type = MPP_BUFFER_TYPE_EXT_DMA,
    .open = allocator_ext_dma_open,
    .close = allocator_ext_dma_close,
    .alloc = allocator_ext_dma_alloc,
    .free = allocator_ext_dma_free,
    .import = allocator_ext_dma_import,
    .release = allocator_ext_dma_release,
    .mmap = allocator_ext_dma_mmap,
};
