// Copyright(c) 2017-2020, Intel Corporation
//
// 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.

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif // HAVE_CONFIG_H

#include "opae/access.h"
#include "opae/utils.h"
#include "opae/umsg.h"
#include "common_int.h"
#include "opae_drv.h"
#include "intel-fpga.h"

#include <sys/types.h>
#include <sys/stat.h>
#include <sys/ioctl.h>
#include <sys/mman.h>
#include <stdint.h>


// Get number of Umsgs
fpga_result __XFPGA_API__
xfpga_fpgaGetNumUmsg(fpga_handle handle, uint64_t *value)
{
	struct _fpga_handle  *_handle = (struct _fpga_handle *)handle;
	fpga_result result            = FPGA_OK;
	int err                       = 0;
	opae_port_info port_info      = { 0 };

	ASSERT_NOT_NULL(value);
	result = handle_check_and_lock(_handle);
	if (result)
		return result;

	if (_handle->fddev < 0) {
		OPAE_ERR("Invalid handle file descriptor");
		result = FPGA_INVALID_PARAM;
		goto out_unlock;
	}


	result = opae_get_port_info(_handle->fddev, &port_info);
	if (!result) {
		*value = port_info.num_umsgs;
	}

out_unlock:
	err = pthread_mutex_unlock(&_handle->lock);
	if (err)
		OPAE_ERR("pthread_mutex_unlock() failed: %s", strerror(err));
	return result;
}

// Set Umsg Attributes
fpga_result __XFPGA_API__
xfpga_fpgaSetUmsgAttributes(fpga_handle handle, uint64_t value)
{
	struct _fpga_handle  *_handle         = (struct _fpga_handle *)handle;
	fpga_result result                    = FPGA_OK;
	int err                               = 0;

	result = handle_check_and_lock(_handle);
	if (result)
		return result;

	if (_handle->fddev < 0) {
		OPAE_ERR("Invalid handle file descriptor");
		result = FPGA_INVALID_PARAM;
		goto out_unlock;
	}


	result = opae_port_umsg_cfg(_handle->fddev, 0, value);

out_unlock:
	err = pthread_mutex_unlock(&_handle->lock);
	if (err)
		OPAE_ERR("pthread_mutex_unlock() failed: %s", strerror(err));
	return result;
}

// Gets Umsg address
fpga_result __XFPGA_API__
xfpga_fpgaGetUmsgPtr(fpga_handle handle, uint64_t **umsg_ptr)
{
	struct _fpga_handle  *_handle            = (struct _fpga_handle *)handle;
	opae_port_info port_info                 = { 0 };

	fpga_result result                       = FPGA_OK;
	uint64_t umsg_count                      = 0;
	uint64_t umsg_size                       = 0;
	int pagesize                             = 0;
	void *umsg_virt                          = NULL;
	int err                                  = 0;
	uint64_t io_addr = 0;

	ASSERT_NOT_NULL(umsg_ptr);
	result = handle_check_and_lock(_handle);
	if (result)
		return result;

	if (_handle->fddev < 0) {
		OPAE_ERR("Invalid handle file descriptor");
		result = FPGA_INVALID_PARAM;
		goto out_unlock;
	}

	if (_handle->umsg_iova != NULL) {
		*umsg_ptr = _handle->umsg_virt;
		goto out_unlock;
	}

	// Page size
	pagesize = sysconf(_SC_PAGESIZE);

	// get umsg count
	result = opae_get_port_info(_handle->fddev, &port_info);
	if (result != FPGA_OK) {
		OPAE_MSG("Failed to get UMSG count");
		goto out_unlock;
	}
	umsg_count = port_info.num_umsgs;
	umsg_size = (uint64_t)umsg_count  * pagesize;
	umsg_virt = alloc_buffer(umsg_size);
	if (umsg_virt == NULL) {
		OPAE_ERR("Failed to allocate memory");
		result = FPGA_NO_MEMORY;
		goto out_unlock;
	}

	// Map Umsg Buffer
	result = opae_port_map(_handle->fddev, umsg_virt, umsg_size, 0, &io_addr);
	if (result != 0) {
		OPAE_ERR("Failed to map UMSG buffer");
		goto umsg_exit;
	}

	// Set Umsg Address
	result = opae_port_umsg_set_base_addr(_handle->fddev, 0, io_addr);
	if (result != 0) {
		OPAE_ERR("Failed to set UMSG base address");
		goto umsg_map_exit;
	}

	result = opae_port_umsg_enable(_handle->fddev);
	if (result != 0) {
		OPAE_ERR("Failed to enable UMSG");
		goto umsg_map_exit;
	}

	*umsg_ptr = (uint64_t *) umsg_virt;
	_handle->umsg_iova = (uint64_t *)io_addr;
	_handle->umsg_virt = umsg_virt;
	_handle->umsg_size = umsg_size;

out_unlock:
	err = pthread_mutex_unlock(&_handle->lock);
	if (err)
		OPAE_ERR("pthread_mutex_unlock() failed: %s", strerror(err));
	return result;

umsg_map_exit:
	result = opae_port_unmap(_handle->fddev, io_addr);
	if (result) {
		OPAE_MSG("Failed to unmap UMSG buffer");
	}

umsg_exit:
	if (umsg_virt != NULL)
		free_buffer(umsg_virt, umsg_size);

	err = pthread_mutex_unlock(&_handle->lock);
	if (err)
		OPAE_ERR("pthread_mutex_unlock() failed: %s", strerror(err));
	return result;
}

fpga_result free_umsg_buffer(fpga_handle handle)
{
	fpga_result result                = FPGA_OK;
	struct _fpga_handle  *_handle     = (struct _fpga_handle *)handle;
	int err                           = 0;


	result = handle_check_and_lock(_handle);
	if (result)
		return result;

	if (_handle->umsg_virt != NULL) {
		if (opae_port_umsg_disable(_handle->fddev)) {
			OPAE_ERR("Failed to disable UMSG");
		}

		if (opae_port_umsg_set_base_addr(_handle->fddev, 0, 0)) {
			OPAE_ERR("Failed to zero UMSG address");
		}

		if (opae_port_unmap(_handle->fddev, (uint64_t)_handle->umsg_iova)) {
			OPAE_ERR("Failed to unmap UMSG Buffer");
		}

		free_buffer(_handle->umsg_virt, _handle->umsg_size);

		_handle->umsg_virt = NULL;
		_handle->umsg_size = 0;
		_handle->umsg_iova = NULL;
	}

	err = pthread_mutex_unlock(&_handle->lock);
	if (err)
		OPAE_ERR("pthread_mutex_unlock() failed: %s", strerror(err));
	return result;
}

// Trigger umsg
fpga_result __XFPGA_API__
xfpga_fpgaTriggerUmsg(fpga_handle handle, uint64_t value)
{
	struct _fpga_handle  *_handle = (struct _fpga_handle *)handle;
	fpga_result result            = FPGA_OK;
	uint64_t *umsg_ptr            = NULL;
	int err                       = 0;

	result = handle_check_and_lock(_handle);
	if (result)
		return result;

	if (_handle->fddev < 0) {
		OPAE_ERR("Invalid handle file descriptor");
		result = FPGA_INVALID_PARAM;
		goto out_unlock;
	}

	result = xfpga_fpgaGetUmsgPtr(handle, &umsg_ptr);
	if (result != FPGA_OK) {
		OPAE_ERR("Failed to get UMsg buffer");
		goto out_unlock;
	}

	// Assign Value to UMsg
	*((volatile uint64_t *) (umsg_ptr)) = value;

out_unlock:
	err = pthread_mutex_unlock(&_handle->lock);
	if (err)
		OPAE_ERR("pthread_mutex_unlock() failed: %s", strerror(err));
	return result;
}