// SPDX-License-Identifier: GPL-2.0
#ifndef TEST_H
#define TEST_H
/*
 * Provides types and variables used when performing the memory tests.
 *
 * Copyright (C) 2020-2022 Martin Whitaker.
 */

#include <stdbool.h>
#include <stdint.h>

#include "pmem.h"
#include "smp.h"

#include "barrier.h"
#include "spinlock.h"

/*
 * A mapping from a CPU core number to the index number of the memory chunk
 * it operates on when performing a memory test in parallel across all the
 * enabled cores.
 */
extern uint8_t chunk_index[MAX_CPUS];

 /*
  * The number of CPU cores being used for the current test. This is always
  * either 1 or the full number of enabled CPU cores.
  */
extern volatile int num_active_cpus;

/*
 * The current master CPU core.
 */
extern volatile int master_cpu;

/*
 * A barrier used when running tests.
 */
extern barrier_t *run_barrier;

/*
 * A mutex used when reporting errors or printing trace information.
 */
extern spinlock_t *error_mutex;

/*
 * The word width (in bits) used for memory testing.
 */
#ifdef __x86_64__
#define TESTWORD_WIDTH  64
#else
#define TESTWORD_WIDTH  32
#endif

/*
 * The number of hex digits needed to display a memory test word.
 */
#define TESTWORD_DIGITS (TESTWORD_WIDTH / 4)

/*
 * The word type used for memory testing.
 */
typedef uintptr_t testword_t;

/*
 * A virtual memory segment descriptor.
 */
typedef struct {
    uintptr_t           pm_base_addr;
    testword_t          *start;
    testword_t          *end;
} vm_map_t;

/*
 * The list of memory segments currently mapped into virtual memory.
 */
extern volatile vm_map_t    vm_map[MAX_MEM_SEGMENTS];
/*
 * The number of memory segments currently mapped into virtual memory.
 */
extern volatile int         vm_map_size;

/*
 * The number of completed test passes.
 */
extern volatile int         pass_num;
/*
 * The current test number.
 */
extern volatile int         test_num;

/*
 * A flag indicating that testing should be restarted due to a configuration
 * change.
 */
extern volatile bool        restart;
/*
 * A flag indicating that the current test should be aborted.
 */
extern volatile bool        bail;

/*
 * The base address of the block of memory currently being tested.
 */
extern volatile uintptr_t   test_addr[MAX_CPUS];

#endif // TEST_H