/*
 * Copyright (c) 2022, Arm Limited. All rights reserved.
 *
 * SPDX-License-Identifier: BSD-3-Clause
 */

#include <arch.h>
#include <arch_helpers.h>

static bool brbe_supported(void)
{
	uint64_t features;

	features = read_id_aa64dfr0_el1() >> ID_AA64DFR0_BRBE_SHIFT;
	return ((features & ID_AA64DFR0_BRBE_MASK) ==
		ID_AA64DFR0_BRBE_SUPPORTED);
}

void brbe_enable(void)
{
	uint64_t val;

	if (brbe_supported()) {
		/*
		 * MDCR_EL3.SBRBE = 0b01
		 *
		 * Allows BRBE usage in non-secure world and prohibited in
		 * secure world.
		 */
		val = read_mdcr_el3();
		val &= ~(MDCR_SBRBE_MASK << MDCR_SBRBE_SHIFT);
		val |= (0x1UL << MDCR_SBRBE_SHIFT);
		write_mdcr_el3(val);
	}
}