/* SPDX-License-Identifier: GPL-2.0-only */
/*
 * POWER Data Stream Control Register (DSCR)
 *
 * This header file contains helper functions and macros
 * required for all the DSCR related test cases.
 *
 * Copyright 2012, Anton Blanchard, IBM Corporation.
 * Copyright 2015, Anshuman Khandual, IBM Corporation.
 */
#ifndef _SELFTESTS_POWERPC_DSCR_DSCR_H
#define _SELFTESTS_POWERPC_DSCR_DSCR_H

#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <dirent.h>
#include <pthread.h>
#include <sched.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/wait.h>

#include "reg.h"
#include "utils.h"

#define THREADS  100 /* Max threads */
#define COUNT  100 /* Max iterations */
#define DSCR_MAX 16 /* Max DSCR value */
#define LEN_MAX  100 /* Max name length */

#define DSCR_DEFAULT "/sys/devices/system/cpu/dscr_default"
#define CPU_PATH "/sys/devices/system/cpu/"

#define rmb()  asm volatile("lwsync":::"memory")
#define wmb()  asm volatile("lwsync":::"memory")

#define READ_ONCE(x) (*(volatile typeof(x) *)&(x))

/* Prilvilege state DSCR access */
inline unsigned long get_dscr(void)
{
 return mfspr(SPRN_DSCR_PRIV);
}

inline void set_dscr(unsigned long val)
{
 mtspr(SPRN_DSCR_PRIV, val);
}

/* Problem state DSCR access */
inline unsigned long get_dscr_usr(void)
{
 return mfspr(SPRN_DSCR);
}

inline void set_dscr_usr(unsigned long val)
{
 mtspr(SPRN_DSCR, val);
}

/* Default DSCR access */
unsigned long get_default_dscr(void)
{
 int err;
 unsigned long val;

 err = read_ulong(DSCR_DEFAULT, &val, 16);
 if (err) {
  perror("read() failed");
  exit(1);
 }
 return val;
}

void set_default_dscr(unsigned long val)
{
 int err;

 err = write_ulong(DSCR_DEFAULT, val, 16);
 if (err) {
  perror("write() failed");
  exit(1);
 }
}

#endif /* _SELFTESTS_POWERPC_DSCR_DSCR_H */