/* SPDX-License-Identifier: BSD-2-Clause */

/*
 * Copyright (C) 2014, 2020 embedded brains GmbH & Co. KG
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. 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.
 *
 * 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

#include "sptls02.h"

#include <stdio.h>
#include <stdlib.h>

#include <rtems.h>
#include <rtems/libcsupport.h>
#include <rtems/stackchk.h>

#include <tmacros.h>

const char rtems_test_name[] = "SPTLS 2";

static thread_local long i123 = 123;

alignas(256) static thread_local long a256 = 256;

static thread_local long i0;

alignas(RTEMS_MINIMUM_STACK_SIZE) static thread_local long a;

int seven()
{
	return 7;
}

static void clobber()
{
	extern_int = 0xdead0007;
	i123 = 0xdead0001;
	a256 = 0xdead0002;
	i0 = 0xdead0003;
	a = 0xdead0004;
}

static long f456(bool clobber)
{
	static thread_local long fi456 = 456;

	if (clobber) {
		fi456 = 0xdead0003;
	}

	return fi456;
}

static long f0(bool clobber)
{
	static thread_local long fi0;

	if (clobber) {
		fi0 = 0xdead0004;
	}

	return fi0;
}

class C {
	public:
		static long c789()
		{
			return ci789;
		}

		static long c0()
		{
			return ci0;
		}

		static void clobber()
		{
			ci789 = 0xdead0005;
			ci0 = 0xdead0006;
		}

	private:
		static thread_local long ci789;

		static thread_local long ci0;
};

thread_local long C::ci789 = 789;

thread_local long C::ci0;

class A {
	public:
		A(long i)
			: ii(i), c(gc)
		{
			++gc;
		}

		~A()
		{
			--gc;
		}

		long i() const
		{
			return ii;
		}

		void clobber()
		{
			ii = ~ii;
			c = ~c;
		}

		long counter() const
		{
			return c;
		}

		static long globalCounter()
		{
			return gc;
		}

	private:
		static long gc;

		long ii;

		long c;
};

long A::gc;

static volatile long mc;

static thread_local A a1(mc + 1);
static thread_local A a2(mc + 2);
static thread_local A a3(mc + 3);

static void checkTLSValues()
{
	rtems_test_assert(extern_int == 7);
	rtems_test_assert(i123 == 123);
	rtems_test_assert(a256 == 256);
	uintptr_t addr = reinterpret_cast<uintptr_t>(&a256);
	RTEMS_OBFUSCATE_VARIABLE(addr);
	rtems_test_assert((addr % 256) == 0);
	rtems_test_assert(i0 == 0);
	rtems_test_assert(a == 0);
	addr = reinterpret_cast<uintptr_t>(&a);
	RTEMS_OBFUSCATE_VARIABLE(addr);
	rtems_test_assert((addr % 512) == 0);
	rtems_test_assert(f456(false) == 456);
	rtems_test_assert(f0(false) == 0);
	rtems_test_assert(C::c789() == 789);
	rtems_test_assert(C::c0() == 0);
	rtems_test_assert(a1.i() == 1);
	rtems_test_assert(a2.i() == 2);
	rtems_test_assert(a3.i() == 3);
}

static rtems_id masterTask;

static void wakeUpMaster()
{
	rtems_status_code sc = rtems_event_transient_send(masterTask);
	rtems_test_assert(sc == RTEMS_SUCCESSFUL);
}

static void waitForWorker()
{
	rtems_status_code sc = rtems_event_transient_receive(
		RTEMS_WAIT,
		RTEMS_NO_TIMEOUT
	);
	rtems_test_assert(sc == RTEMS_SUCCESSFUL);
}

static void worker(rtems_task_argument arg)
{
	wakeUpMaster();
	checkTLSValues();

	const long gc = static_cast<long>(arg);

	rtems_test_assert(A::globalCounter() == gc + 3);

	rtems_test_assert(a1.counter() == gc + 0);
	rtems_test_assert(a2.counter() == gc + 1);
	rtems_test_assert(a3.counter() == gc + 2);

	clobber();
	f456(true);
	f0(true);
	C::clobber();
	a1.clobber();
	a2.clobber();
	a3.clobber();

	wakeUpMaster();

	(void) rtems_task_suspend(RTEMS_SELF);
	rtems_test_assert(false);
}

static void testWorkerTask()
{
	checkTLSValues();

	rtems_id id;
	rtems_status_code sc = rtems_task_create(
		rtems_build_name('T', 'A', 'S', 'K'),
		2,
		RTEMS_MINIMUM_STACK_SIZE,
		RTEMS_DEFAULT_MODES,
		RTEMS_DEFAULT_ATTRIBUTES,
		&id
	);
	rtems_test_assert(sc == RTEMS_SUCCESSFUL);

	const long gc = A::globalCounter();

	sc = rtems_task_start(id, worker, gc);
	rtems_test_assert(sc == RTEMS_SUCCESSFUL);

	waitForWorker();
	rtems_test_assert(A::globalCounter() == gc);

	waitForWorker();
	rtems_test_assert(A::globalCounter() == gc + 3);

	sc = rtems_task_restart(id, gc);
	rtems_test_assert(sc == RTEMS_SUCCESSFUL);

	waitForWorker();
	rtems_test_assert(A::globalCounter() == gc);

	waitForWorker();
	rtems_test_assert(A::globalCounter() == gc + 3);

	sc = rtems_task_delete(id);
	rtems_test_assert(sc == RTEMS_SUCCESSFUL);

	rtems_test_assert(A::globalCounter() == gc);

	checkTLSValues();
}

extern "C" void Init(rtems_task_argument arg)
{
	TEST_BEGIN();

	printf("A::globalCounter() = %li\n", A::globalCounter());

	checkTLSValues();

	printf("A::globalCounter() = %li\n", A::globalCounter());

	masterTask = rtems_task_self();

	testWorkerTask();

	rtems_resource_snapshot snapshot;
	rtems_resource_snapshot_take(&snapshot);

	testWorkerTask();

	rtems_test_assert(rtems_resource_snapshot_check(&snapshot));
	rtems_test_assert(!rtems_stack_checker_is_blown());

	TEST_END();

	exit(0);
}

#define CONFIGURE_APPLICATION_NEEDS_CLOCK_DRIVER
#define CONFIGURE_APPLICATION_NEEDS_SIMPLE_CONSOLE_DRIVER

#define CONFIGURE_STACK_CHECKER_ENABLED

#define CONFIGURE_MAXIMUM_TASKS 2
#define CONFIGURE_MAXIMUM_SEMAPHORES 3

#define CONFIGURE_INITIAL_EXTENSIONS RTEMS_TEST_INITIAL_EXTENSION

#define CONFIGURE_RTEMS_INIT_TASKS_TABLE

#define CONFIGURE_MAXIMUM_POSIX_KEYS 2
#define CONFIGURE_MAXIMUM_POSIX_KEY_VALUE_PAIRS 2

#define CONFIGURE_INIT

#include <rtems/confdefs.h>