/*
 * Copyright (c) 2011 Texas Instruments, Inc.
 * Copyright (c) 2011 Trusted Logic S.A.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * version 2 as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 */

/*
 * This file implements the non-secure rproc and smc interface/integration
 */

#include <linux/types.h>
#include <linux/module.h>

#include "tee_client_api.h"
#include "tf_defs.h"

/*C2537CC3-36F0-48D9-820E-559601478029*/
#define COMMON_SECURE_DRIVER_UUID {0xC2537CC3, 0x36F0, 0x48D9, \
			{0x82, 0x0E, 0x55, 0x96, 0x01, 0x47, 0x80, 0x29} }

#define ENTER_SECURE_PLAYBACK	0x00003000

#define EXIT_SECURE_PLAYBACK	0x00003001

enum rproc_drm_s_state {
	RPROC_DRM_SECURE_LEAVE,
	RPROC_DRM_SECURE_ENTER
};

static enum rproc_drm_s_state s_state;

static TEEC_Result rproc_drm_initialize(TEEC_Context *teec_context,
					TEEC_Session *teec_session)
{
	static const TEEC_UUID drm_uuid = COMMON_SECURE_DRIVER_UUID;
	static u32 drm_gid = 1019;
	TEEC_Result result;

	result = TEEC_InitializeContext(NULL, teec_context);
	if (result != TEEC_SUCCESS)
		goto exit;

	result = TEEC_OpenSession(teec_context, teec_session, &drm_uuid,
			TEEC_LOGIN_PRIVILEGED, &drm_gid, NULL, NULL);
	if (result != TEEC_SUCCESS)
		TEEC_FinalizeContext(teec_context);

exit:
	return result;
}

static TEEC_Result rproc_drm_finalize(TEEC_Context *teec_context,
					TEEC_Session *teec_session)
{
	TEEC_CloseSession(teec_session);
	TEEC_FinalizeContext(teec_context);
	return TEEC_SUCCESS;
}

static TEEC_Result _rproc_drm_invoke_secure_service(bool enable)
{
	TEEC_Result result;
	TEEC_Operation operation;
	TEEC_Context teec_context;
	TEEC_Session teec_session;
	u32 command;

	result = rproc_drm_initialize(&teec_context, &teec_session);
	if (result != TEEC_SUCCESS)
		goto out;

	operation.paramTypes = TEEC_PARAM_TYPES(TEEC_NONE, TEEC_NONE,
						TEEC_NONE, TEEC_NONE);
	command = (enable ? ENTER_SECURE_PLAYBACK :
				EXIT_SECURE_PLAYBACK);
	result = TEEC_InvokeCommand(&teec_session, command, &operation, NULL);
	rproc_drm_finalize(&teec_context, &teec_session);
out:
	return result;
}

int rproc_drm_invoke_service(bool enable)
{
	int ret;

	if ((s_state == RPROC_DRM_SECURE_ENTER && enable) ||
		(s_state == RPROC_DRM_SECURE_LEAVE && !enable))
		return 0;

	ret = _rproc_drm_invoke_secure_service(enable);
	s_state = (enum rproc_drm_s_state) enable;

	return ret == TEEC_SUCCESS ? 0 : -EACCES;
}
EXPORT_SYMBOL(rproc_drm_invoke_service);