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

#define _GNU_SOURCE

#include <elf.h>
#include <limits.h>
#include <stdio.h>
#include <stdbool.h>
#include <string.h>
#include <sys/prctl.h>

#include "event.h"
#include "lib.h"
#include "utils.h"

/*
 * Test that per-event excludes work.
 */

static int per_event_excludes(void)
{
        struct event *e, events[4];
        int i;

        SKIP_IF(!have_hwcap2(PPC_FEATURE2_ARCH_2_07));

        /*
         * We need to create the events disabled, otherwise the running/enabled
         * counts don't match up.
         */
        e = &events[0];
        event_init_opts(e, PERF_COUNT_HW_INSTRUCTIONS,
                        PERF_TYPE_HARDWARE, "instructions");
        e->attr.disabled = 1;

        e = &events[1];
        event_init_opts(e, PERF_COUNT_HW_INSTRUCTIONS,
                        PERF_TYPE_HARDWARE, "instructions(k)");
        e->attr.disabled = 1;
        e->attr.exclude_user = 1;
        e->attr.exclude_hv = 1;

        e = &events[2];
        event_init_opts(e, PERF_COUNT_HW_INSTRUCTIONS,
                        PERF_TYPE_HARDWARE, "instructions(h)");
        e->attr.disabled = 1;
        e->attr.exclude_user = 1;
        e->attr.exclude_kernel = 1;

        e = &events[3];
        event_init_opts(e, PERF_COUNT_HW_INSTRUCTIONS,
                        PERF_TYPE_HARDWARE, "instructions(u)");
        e->attr.disabled = 1;
        e->attr.exclude_hv = 1;
        e->attr.exclude_kernel = 1;

        FAIL_IF(event_open(&events[0]));

        /*
         * The open here will fail if we don't have per event exclude support,
         * because the second event has an incompatible set of exclude settings
         * and we're asking for the events to be in a group.
         */
        for (i = 1; i < 4; i++)
                FAIL_IF(event_open_with_group(&events[i], events[0].fd));

        /*
         * Even though the above will fail without per-event excludes we keep
         * testing in order to be thorough.
         */
        prctl(PR_TASK_PERF_EVENTS_ENABLE);

        /* Spin for a while */
        for (i = 0; i < INT_MAX; i++)
                asm volatile("" : : : "memory");

        prctl(PR_TASK_PERF_EVENTS_DISABLE);

        for (i = 0; i < 4; i++) {
                FAIL_IF(event_read(&events[i]));
                event_report(&events[i]);
        }

        /*
         * We should see that all events have enabled == running. That
         * shows that they were all on the PMU at once.
         */
        for (i = 0; i < 4; i++)
                FAIL_IF(events[i].result.running != events[i].result.enabled);

        /*
         * We can also check that the result for instructions is >= all the
         * other counts. That's because it is counting all instructions while
         * the others are counting a subset.
         */
        for (i = 1; i < 4; i++)
                FAIL_IF(events[0].result.value < events[i].result.value);

        for (i = 0; i < 4; i++)
                event_close(&events[i]);

        return 0;
}

int main(void)
{
        return test_harness(per_event_excludes, "per_event_excludes");
}