// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright 2016, Michael Ellerman, IBM Corp.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/mman.h>
#include <unistd.h>

#include <asm/cputable.h>

#include "utils.h"

#define SIZE (64 * 1024)

int test_prot_sao(void)
{
 char *p;

 /*
 * SAO was introduced in 2.06 and removed in 3.1. It's disabled in
 * guests/LPARs by default, so also skip if we are running in a guest.
 */
 SKIP_IF(!have_hwcap(PPC_FEATURE_ARCH_2_06) ||
  have_hwcap2(PPC_FEATURE2_ARCH_3_1) ||
  access("/proc/device-tree/rtas/ibm,hypertas-functions", F_OK) == 0);

 /*
 * Ensure we can ask for PROT_SAO.
 * We can't really verify that it does the right thing, but at least we
 * confirm the kernel will accept it.
 */
 p = mmap(NULL, SIZE, PROT_READ | PROT_WRITE | PROT_SAO,
   MAP_ANONYMOUS | MAP_PRIVATE, -1, 0);
 FAIL_IF(p == MAP_FAILED);

 /* Write to the mapping, to at least cause a fault */
 memset(p, 0xaa, SIZE);

 return 0;
}

int main(void)
{
 return test_harness(test_prot_sao, "prot-sao");
}