/*
 * Copyright (c) 2022 Nordic Semiconductor ASA
 *
 * SPDX-License-Identifier: Apache-2.0
 */

#include "testing_common_defs.h"

#include <zephyr/bluetooth/hci.h>
#include <zephyr/fff.h>
#include <zephyr/kernel.h>

#include <host/hci_core.h>
#include <host/id.h>
#include <mocks/addr.h>
#include <mocks/addr_expects.h>

DEFINE_FFF_GLOBALS;

static void fff_reset_rule_before(const struct ztest_unit_test *test, void *fixture)
{
	memset(&bt_dev, 0x00, sizeof(struct bt_dev));
	ADDR_FFF_FAKES_LIST(RESET_FAKE);
}

ZTEST_RULE(fff_reset_rule, fff_reset_rule_before, NULL);

ZTEST_SUITE(bt_id_create, NULL, NULL, NULL, NULL, NULL);

static int bt_addr_le_create_static_custom_fake(bt_addr_le_t *addr)
{
	const char *func_name = "bt_addr_le_create_static";

	zassert_not_null(addr, "'%s()' was called with incorrect '%s' value", func_name, "addr");

	/* This is required to test when the generated address already exists in the ID list */
	if (bt_addr_le_create_static_fake.call_count == 1) {
		bt_addr_le_copy(addr, BT_STATIC_RANDOM_LE_ADDR_1);
	} else {
		bt_addr_le_copy(addr, BT_STATIC_RANDOM_LE_ADDR_2);
	}

	return 0;
}

/*
 *  Test creating a new identity.
 *  As a NULL is passed to bt_id_create() for the address and 'BT_DEV_ENABLE' is set,
 *  a new random address is generated.
 *
 *  Constraints:
 *   - Input address is NULL
 *   - Input IRK is NULL
 *   - 'BT_DEV_ENABLE' flag is set in bt_dev.flags
 *   - bt_addr_le_create_static() returns a zero error code (success)
 *
 *  Expected behaviour:
 *   - A new identity is created and the address is loaded to bt_dev.id_addr[]
 *   - bt_dev.id_count is incremented
 */
ZTEST(bt_id_create, test_create_id_null_address)
{
	int id_count, new_id;

	id_count = bt_dev.id_count;
	atomic_set_bit(bt_dev.flags, BT_DEV_ENABLE);
	bt_addr_le_create_static_fake.custom_fake = bt_addr_le_create_static_custom_fake;

	new_id = bt_id_create(NULL, NULL);

	expect_call_count_bt_addr_le_create_static(1);

	zassert_true(new_id >= 0, "Unexpected error code '%d' was returned", new_id);
	zassert_true(bt_dev.id_count == (id_count + 1), "Incorrect ID count %d was set",
		     bt_dev.id_count);
	zassert_mem_equal(&bt_dev.id_addr[new_id], BT_STATIC_RANDOM_LE_ADDR_1, sizeof(bt_addr_le_t),
			  "Incorrect address was set");
}

/*
 *  Test creating a new identity and ensure that the generated address isn't in the ID list.
 *  As a NULL is passed to bt_id_create() for the address and 'BT_DEV_ENABLE' is set,
 *  a new random address is generated.
 *
 *  Constraints:
 *   - Input address is NULL
 *   - Input IRK is NULL
 *   - 'BT_DEV_ENABLE' flag is set in bt_dev.flags
 *   - bt_addr_le_create_static() returns a zero error code (success)
 *
 *  Expected behaviour:
 *   - A new identity is created and the address is loaded to bt_dev.id_addr[]
 *   - bt_dev.id_count is incremented
 */
ZTEST(bt_id_create, test_create_id_null_address_with_no_duplication)
{
	int id_count, new_id;

	bt_dev.id_count = 1;
	bt_addr_le_copy(&bt_dev.id_addr[0], BT_STATIC_RANDOM_LE_ADDR_1);

	id_count = bt_dev.id_count;
	atomic_set_bit(bt_dev.flags, BT_DEV_ENABLE);
	bt_addr_le_create_static_fake.custom_fake = bt_addr_le_create_static_custom_fake;

	new_id = bt_id_create(NULL, NULL);

	expect_call_count_bt_addr_le_create_static(2);

	zassert_true(new_id >= 0, "Unexpected error code '%d' was returned", new_id);
	zassert_true(bt_dev.id_count == (id_count + 1), "Incorrect ID count %d was set",
		     bt_dev.id_count);
	zassert_mem_equal(&bt_dev.id_addr[new_id], BT_STATIC_RANDOM_LE_ADDR_2, sizeof(bt_addr_le_t),
			  "Incorrect address was set");
}

/*
 *  Test creating a new identity.
 *  As an initialized address to BT_ADDR_LE_ANY is passed to bt_id_create() for the address and
 * 'BT_DEV_ENABLE' is set, a new random address is generated.
 *  The generated address should be copied to the address reference passed
 *
 *  Constraints:
 *   - Input address is NULL
 *   - Input IRK is NULL
 *   - 'BT_DEV_ENABLE' flag is set in bt_dev.flags
 *   - bt_addr_le_create_static() returns a zero error code (success)
 *
 *  Expected behaviour:
 *   - A new identity is created and the address is loaded to bt_dev.id_addr[]
 *   - bt_dev.id_count is incremented
 */
ZTEST(bt_id_create, test_create_id_bt_addr_le_any_address)
{
	int id_count, new_id;
	bt_addr_le_t addr = bt_addr_le_any;

	id_count = bt_dev.id_count;
	atomic_set_bit(bt_dev.flags, BT_DEV_ENABLE);
	bt_addr_le_create_static_fake.custom_fake = bt_addr_le_create_static_custom_fake;

	new_id = bt_id_create(&addr, NULL);

	expect_call_count_bt_addr_le_create_static(1);

	zassert_true(new_id >= 0, "Unexpected error code '%d' was returned", new_id);
	zassert_true(bt_dev.id_count == (id_count + 1), "Incorrect ID count %d was set",
		     bt_dev.id_count);
	zassert_mem_equal(&bt_dev.id_addr[new_id], BT_STATIC_RANDOM_LE_ADDR_1, sizeof(bt_addr_le_t),
			  "Incorrect address was set");
	zassert_mem_equal(&addr, BT_STATIC_RANDOM_LE_ADDR_1, sizeof(bt_addr_le_t),
			  "Incorrect address was set");
}

/*
 *  Test creating a new identity, but bt_addr_le_create_static() returns an error.
 *
 *  Constraints:
 *   - Input address is NULL
 *   - Input IRK is NULL
 *   - 'BT_DEV_ENABLE' flag is set in bt_dev.flags
 *   - bt_addr_le_create_static() returns a non-zero error code (failure)
 *
 *  Expected behaviour:
 *   - No new identity is created
 *   - bt_dev.id_count is kept unchanged
 */
ZTEST(bt_id_create, test_create_id_null_address_fails)
{
	int id_count, err;

	id_count = bt_dev.id_count;
	atomic_set_bit(bt_dev.flags, BT_DEV_ENABLE);
	bt_addr_le_create_static_fake.return_val = -1;

	err = bt_id_create(NULL, NULL);

	expect_call_count_bt_addr_le_create_static(1);

	zassert_true(err == -1, "Unexpected error code '%d' was returned", err);
	zassert_true(bt_dev.id_count == id_count, "Incorrect ID count %d was set", bt_dev.id_count);
}

/*
 *  Test creating a new identity.
 *  A valid random static address is passed to bt_id_create() for the address and 'BT_DEV_ENABLE' is
 *  set, the same address is used and copied to bt_dev.id_addr[].
 *
 *  Constraints:
 *   - Valid private random address is used
 *   - Input IRK is NULL
 *   - 'BT_DEV_ENABLE' flag is set in bt_dev.flags
 *
 *  Expected behaviour:
 *   - The same address is used and loaded to bt_dev.id_addr[]
 *   - bt_dev.id_count is incremented
 */
ZTEST(bt_id_create, test_create_id_valid_input_address)
{
	int id_count, new_id;
	bt_addr_le_t addr = *BT_STATIC_RANDOM_LE_ADDR_1;

	id_count = bt_dev.id_count;
	atomic_set_bit(bt_dev.flags, BT_DEV_ENABLE);
	/* Calling bt_addr_le_create_static() isn't expected */
	bt_addr_le_create_static_fake.return_val = -1;

	new_id = bt_id_create(&addr, NULL);

	expect_not_called_bt_addr_le_create_static();

	zassert_true(new_id >= 0, "Unexpected error code '%d' was returned", new_id);
	zassert_true(bt_dev.id_count == (id_count + 1), "Incorrect ID count %d was set",
		     bt_dev.id_count);
	zassert_mem_equal(&bt_dev.id_addr[new_id], BT_STATIC_RANDOM_LE_ADDR_1, sizeof(bt_addr_le_t),
			  "Incorrect address was set");
}

/*
 *  Test creating a new public identity.
 *
 *  Constraints:
 *   - A valid address of type public is used
 *   - Input IRK is NULL
 *   - 'BT_DEV_ENABLE' flag is set in bt_dev.flags
 *
 *  Expected behaviour:
 *   - The public address is loaded to bt_dev.id_addr[BT_ID_DEFAULT]
 *   - bt_dev.id_count is incremented
 */
ZTEST(bt_id_create, test_public_address)
{
	int id_count, new_id;
	bt_addr_le_t addr = *BT_LE_ADDR;

	if (!IS_ENABLED(CONFIG_BT_HCI_SET_PUBLIC_ADDR)) {
		ztest_test_skip();
	}

	id_count = bt_dev.id_count;
	atomic_set_bit(bt_dev.flags, BT_DEV_ENABLE);
	/* Calling bt_addr_le_create_static() isn't expected */
	bt_addr_le_create_static_fake.return_val = -1;

	new_id = bt_id_create(&addr, NULL);

	expect_not_called_bt_addr_le_create_static();

	zassert_true(new_id == BT_ID_DEFAULT, "Unexpected error code '%d' was returned", new_id);
	zassert_true(bt_dev.id_count == (id_count + 1), "Incorrect ID count %d was set",
		     bt_dev.id_count);
	zassert_mem_equal(&bt_dev.id_addr[new_id], BT_LE_ADDR, sizeof(bt_addr_le_t),
			  "Incorrect address was set");
}