root/drivers/gpu/drm/xe/tests/xe_guc_relay_test.c
// SPDX-License-Identifier: GPL-2.0 AND MIT
/*
 * Copyright © 2023 Intel Corporation
 */

#include <kunit/static_stub.h>
#include <kunit/test.h>
#include <kunit/test-bug.h>

#include "xe_device.h"
#include "xe_kunit_helpers.h"
#include "xe_pci_test.h"

#define TEST_RID        1234
#define TEST_VFID       5
#define TEST_LEN        6
#define TEST_ACTION     0xa
#define TEST_DATA(n)    (0xd0 + (n))

static int replacement_relay_get_totalvfs(struct xe_guc_relay *relay)
{
        return TEST_VFID;
}

static int relay_test_init(struct kunit *test)
{
        struct xe_pci_fake_data fake = {
                .sriov_mode = XE_SRIOV_MODE_PF,
                .platform = XE_TIGERLAKE, /* some random platform */
                .subplatform = XE_SUBPLATFORM_NONE,
        };
        struct xe_guc_relay *relay;
        struct xe_device *xe;

        test->priv = &fake;
        xe_kunit_helper_xe_device_test_init(test);

        xe = test->priv;
        KUNIT_ASSERT_EQ(test, xe_sriov_init(xe), 0);

        relay = &xe_device_get_gt(xe, 0)->uc.guc.relay;
        kunit_activate_static_stub(test, relay_get_totalvfs,
                                   replacement_relay_get_totalvfs);

        KUNIT_ASSERT_EQ(test, xe_guc_relay_init(relay), 0);
        KUNIT_EXPECT_TRUE(test, relay_is_ready(relay));
        relay->last_rid = TEST_RID - 1;

        test->priv = relay;
        return 0;
}

static const u32 TEST_MSG[TEST_LEN] = {
        FIELD_PREP_CONST(GUC_HXG_MSG_0_ORIGIN, GUC_HXG_ORIGIN_HOST) |
        FIELD_PREP_CONST(GUC_HXG_MSG_0_TYPE, GUC_HXG_TYPE_EVENT) |
        FIELD_PREP_CONST(GUC_HXG_EVENT_MSG_0_ACTION, TEST_ACTION) |
        FIELD_PREP_CONST(GUC_HXG_EVENT_MSG_0_DATA0, TEST_DATA(0)),
        TEST_DATA(1), TEST_DATA(2), TEST_DATA(3), TEST_DATA(4),
};

static int replacement_xe_guc_ct_send_recv_always_fails(struct xe_guc_ct *ct,
                                                        const u32 *msg, u32 len,
                                                        u32 *response_buffer)
{
        struct kunit *test = kunit_get_current_test();

        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ct);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, msg);
        KUNIT_ASSERT_GE(test, len, GUC_HXG_MSG_MIN_LEN);

        return -ECOMM;
}

static int replacement_xe_guc_ct_send_recv_expects_pf2guc_relay(struct xe_guc_ct *ct,
                                                                const u32 *msg, u32 len,
                                                                u32 *response_buffer)
{
        struct kunit *test = kunit_get_current_test();

        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ct);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, msg);
        KUNIT_ASSERT_GE(test, len, PF2GUC_RELAY_TO_VF_REQUEST_MSG_MIN_LEN);
        KUNIT_ASSERT_EQ(test, len, PF2GUC_RELAY_TO_VF_REQUEST_MSG_MIN_LEN + TEST_LEN);
        KUNIT_EXPECT_EQ(test, GUC_HXG_ORIGIN_HOST, FIELD_GET(GUC_HXG_MSG_0_ORIGIN, msg[0]));
        KUNIT_EXPECT_EQ(test, GUC_HXG_TYPE_REQUEST, FIELD_GET(GUC_HXG_MSG_0_TYPE, msg[0]));
        KUNIT_EXPECT_EQ(test, XE_GUC_ACTION_PF2GUC_RELAY_TO_VF,
                        FIELD_GET(GUC_HXG_REQUEST_MSG_0_ACTION, msg[0]));
        KUNIT_EXPECT_EQ(test, TEST_VFID,
                        FIELD_GET(PF2GUC_RELAY_TO_VF_REQUEST_MSG_1_VFID, msg[1]));
        KUNIT_EXPECT_EQ(test, TEST_RID,
                        FIELD_GET(PF2GUC_RELAY_TO_VF_REQUEST_MSG_2_RELAY_ID, msg[2]));
        KUNIT_EXPECT_MEMEQ(test, TEST_MSG, msg + PF2GUC_RELAY_TO_VF_REQUEST_MSG_MIN_LEN,
                           sizeof(u32) * TEST_LEN);
        return 0;
}

static const u32 test_guc2pf[GUC2PF_RELAY_FROM_VF_EVENT_MSG_MAX_LEN] = {
        /* transport */
        FIELD_PREP_CONST(GUC_HXG_MSG_0_ORIGIN, GUC_HXG_ORIGIN_GUC) |
        FIELD_PREP_CONST(GUC_HXG_MSG_0_TYPE, GUC_HXG_TYPE_EVENT) |
        FIELD_PREP_CONST(GUC_HXG_EVENT_MSG_0_ACTION, XE_GUC_ACTION_GUC2PF_RELAY_FROM_VF),
        FIELD_PREP_CONST(GUC2PF_RELAY_FROM_VF_EVENT_MSG_1_VFID, TEST_VFID),
        FIELD_PREP_CONST(GUC2PF_RELAY_FROM_VF_EVENT_MSG_2_RELAY_ID, TEST_RID),
        /* payload */
        FIELD_PREP_CONST(GUC_HXG_MSG_0_ORIGIN, GUC_HXG_ORIGIN_HOST) |
        FIELD_PREP_CONST(GUC_HXG_MSG_0_TYPE, GUC_HXG_TYPE_RESPONSE_SUCCESS),
};

static const u32 test_guc2vf[GUC2VF_RELAY_FROM_PF_EVENT_MSG_MAX_LEN] = {
        /* transport */
        FIELD_PREP_CONST(GUC_HXG_MSG_0_ORIGIN, GUC_HXG_ORIGIN_GUC) |
        FIELD_PREP_CONST(GUC_HXG_MSG_0_TYPE, GUC_HXG_TYPE_EVENT) |
        FIELD_PREP_CONST(GUC_HXG_EVENT_MSG_0_ACTION, XE_GUC_ACTION_GUC2VF_RELAY_FROM_PF),
        FIELD_PREP_CONST(GUC2VF_RELAY_FROM_PF_EVENT_MSG_1_RELAY_ID, TEST_RID),
        /* payload */
        FIELD_PREP_CONST(GUC_HXG_MSG_0_ORIGIN, GUC_HXG_ORIGIN_HOST) |
        FIELD_PREP_CONST(GUC_HXG_MSG_0_TYPE, GUC_HXG_TYPE_RESPONSE_SUCCESS),
};

static void pf_rejects_guc2pf_too_short(struct kunit *test)
{
        const u32 len = GUC2PF_RELAY_FROM_VF_EVENT_MSG_MIN_LEN - 1;
        struct xe_guc_relay *relay = test->priv;
        const u32 *msg = test_guc2pf;

        KUNIT_ASSERT_EQ(test, -EPROTO, xe_guc_relay_process_guc2pf(relay, msg, len));
}

static void pf_rejects_guc2pf_too_long(struct kunit *test)
{
        const u32 len = GUC2PF_RELAY_FROM_VF_EVENT_MSG_MAX_LEN + 1;
        struct xe_guc_relay *relay = test->priv;
        const u32 *msg = test_guc2pf;

        KUNIT_ASSERT_EQ(test, -EMSGSIZE, xe_guc_relay_process_guc2pf(relay, msg, len));
}

static void pf_rejects_guc2pf_no_payload(struct kunit *test)
{
        const u32 len = GUC2PF_RELAY_FROM_VF_EVENT_MSG_MIN_LEN;
        struct xe_guc_relay *relay = test->priv;
        const u32 *msg = test_guc2pf;

        KUNIT_ASSERT_EQ(test, -EPROTO, xe_guc_relay_process_guc2pf(relay, msg, len));
}

static void pf_fails_no_payload(struct kunit *test)
{
        struct xe_guc_relay *relay = test->priv;
        const u32 msg = 0;

        KUNIT_ASSERT_EQ(test, -EPROTO, relay_process_msg(relay, TEST_VFID, TEST_RID, &msg, 0));
}

static void pf_fails_bad_origin(struct kunit *test)
{
        struct xe_guc_relay *relay = test->priv;
        static const u32 msg[] = {
                FIELD_PREP_CONST(GUC_HXG_MSG_0_ORIGIN, GUC_HXG_ORIGIN_GUC) |
                FIELD_PREP_CONST(GUC_HXG_MSG_0_TYPE, GUC_HXG_TYPE_RESPONSE_SUCCESS),
        };
        u32 len = ARRAY_SIZE(msg);

        KUNIT_ASSERT_EQ(test, -EPROTO, relay_process_msg(relay, TEST_VFID, TEST_RID, msg, len));
}

static void pf_fails_bad_type(struct kunit *test)
{
        struct xe_guc_relay *relay = test->priv;
        const u32 msg[] = {
                FIELD_PREP(GUC_HXG_MSG_0_ORIGIN, GUC_HXG_ORIGIN_HOST) |
                FIELD_PREP(GUC_HXG_MSG_0_TYPE, 4), /* only 4 is undefined */
        };
        u32 len = ARRAY_SIZE(msg);

        KUNIT_ASSERT_EQ(test, -EBADRQC, relay_process_msg(relay, TEST_VFID, TEST_RID, msg, len));
}

static void pf_txn_reports_error(struct kunit *test)
{
        struct xe_guc_relay *relay = test->priv;
        struct relay_transaction *txn;

        txn = __relay_get_transaction(relay, false, TEST_VFID, TEST_RID,
                                      TEST_MSG, TEST_LEN, NULL, 0);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, txn);

        kunit_activate_static_stub(test, xe_guc_ct_send_recv,
                                   replacement_xe_guc_ct_send_recv_always_fails);
        KUNIT_EXPECT_EQ(test, -ECOMM, relay_send_transaction(relay, txn));

        relay_release_transaction(relay, txn);
}

static void pf_txn_sends_pf2guc(struct kunit *test)
{
        struct xe_guc_relay *relay = test->priv;
        struct relay_transaction *txn;

        txn = __relay_get_transaction(relay, false, TEST_VFID, TEST_RID,
                                      TEST_MSG, TEST_LEN, NULL, 0);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, txn);

        kunit_activate_static_stub(test, xe_guc_ct_send_recv,
                                   replacement_xe_guc_ct_send_recv_expects_pf2guc_relay);
        KUNIT_ASSERT_EQ(test, 0, relay_send_transaction(relay, txn));

        relay_release_transaction(relay, txn);
}

static void pf_sends_pf2guc(struct kunit *test)
{
        struct xe_guc_relay *relay = test->priv;

        kunit_activate_static_stub(test, xe_guc_ct_send_recv,
                                   replacement_xe_guc_ct_send_recv_expects_pf2guc_relay);
        KUNIT_ASSERT_EQ(test, 0,
                        xe_guc_relay_send_to_vf(relay, TEST_VFID,
                                                TEST_MSG, TEST_LEN, NULL, 0));
}

static int replacement_xe_guc_ct_send_recv_loopback_relay(struct xe_guc_ct *ct,
                                                          const u32 *msg, u32 len,
                                                          u32 *response_buffer)
{
        struct kunit *test = kunit_get_current_test();
        struct xe_guc_relay *relay = test->priv;
        u32 *reply = kunit_kzalloc(test, len * sizeof(u32), GFP_KERNEL);
        int (*guc2relay)(struct xe_guc_relay *, const u32 *, u32);
        u32 action;
        int err;

        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ct);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, msg);
        KUNIT_ASSERT_GE(test, len, GUC_HXG_MSG_MIN_LEN);
        KUNIT_ASSERT_EQ(test, GUC_HXG_TYPE_REQUEST,
                        FIELD_GET(GUC_HXG_MSG_0_TYPE, msg[0]));
        KUNIT_ASSERT_GE(test, len, GUC_HXG_REQUEST_MSG_MIN_LEN);
        KUNIT_ASSERT_NOT_NULL(test, reply);

        switch (FIELD_GET(GUC_HXG_REQUEST_MSG_0_ACTION, msg[0])) {
        case XE_GUC_ACTION_PF2GUC_RELAY_TO_VF:
                KUNIT_ASSERT_GE(test, len, PF2GUC_RELAY_TO_VF_REQUEST_MSG_MIN_LEN);
                action = XE_GUC_ACTION_GUC2PF_RELAY_FROM_VF;
                guc2relay = xe_guc_relay_process_guc2pf;
                break;
        case XE_GUC_ACTION_VF2GUC_RELAY_TO_PF:
                KUNIT_ASSERT_GE(test, len, VF2GUC_RELAY_TO_PF_REQUEST_MSG_MIN_LEN);
                action = XE_GUC_ACTION_GUC2VF_RELAY_FROM_PF;
                guc2relay = xe_guc_relay_process_guc2vf;
                break;
        default:
                KUNIT_FAIL(test, "bad RELAY action %#x", msg[0]);
                return -EINVAL;
        }

        reply[0] = FIELD_PREP(GUC_HXG_MSG_0_ORIGIN, GUC_HXG_ORIGIN_GUC) |
                   FIELD_PREP(GUC_HXG_MSG_0_TYPE, GUC_HXG_TYPE_EVENT) |
                   FIELD_PREP(GUC_HXG_EVENT_MSG_0_ACTION, action);
        memcpy(reply + 1, msg + 1, sizeof(u32) * (len - 1));

        err = guc2relay(relay, reply, len);
        KUNIT_EXPECT_EQ(test, err, 0);

        return err;
}

static void test_requires_relay_testloop(struct kunit *test)
{
        /*
         * The debug relay action GUC_RELAY_ACTION_VFXPF_TESTLOOP is available
         * only on builds with CONFIG_DRM_XE_DEBUG_SRIOV enabled.
         * See "kunit.py --kconfig_add" option if it's missing.
         */
        if (!IS_ENABLED(CONFIG_DRM_XE_DEBUG_SRIOV))
                kunit_skip(test, "requires %s\n", __stringify(CONFIG_DRM_XE_DEBUG_SRIOV));
}

static void pf_loopback_nop(struct kunit *test)
{
        struct xe_guc_relay *relay = test->priv;
        u32 request[] = {
                FIELD_PREP(GUC_HXG_MSG_0_ORIGIN, GUC_HXG_ORIGIN_HOST) |
                FIELD_PREP(GUC_HXG_MSG_0_TYPE, GUC_HXG_TYPE_REQUEST) |
                FIELD_PREP(GUC_HXG_REQUEST_MSG_0_ACTION, GUC_RELAY_ACTION_VFXPF_TESTLOOP) |
                FIELD_PREP(GUC_HXG_REQUEST_MSG_0_DATA0, VFXPF_TESTLOOP_OPCODE_NOP),
        };
        u32 response[GUC_HXG_RESPONSE_MSG_MIN_LEN];
        int ret;

        test_requires_relay_testloop(test);

        kunit_activate_static_stub(test, relay_kick_worker, relay_process_incoming_action);
        kunit_activate_static_stub(test, xe_guc_ct_send_recv,
                                   replacement_xe_guc_ct_send_recv_loopback_relay);
        ret = xe_guc_relay_send_to_vf(relay, TEST_VFID,
                                      request, ARRAY_SIZE(request),
                                      response, ARRAY_SIZE(response));
        KUNIT_ASSERT_EQ(test, ret, GUC_HXG_RESPONSE_MSG_MIN_LEN);
        KUNIT_EXPECT_EQ(test, FIELD_GET(GUC_HXG_MSG_0_ORIGIN, response[0]),
                        GUC_HXG_ORIGIN_HOST);
        KUNIT_EXPECT_EQ(test, FIELD_GET(GUC_HXG_MSG_0_TYPE, response[0]),
                        GUC_HXG_TYPE_RESPONSE_SUCCESS);
        KUNIT_EXPECT_EQ(test, FIELD_GET(GUC_HXG_RESPONSE_MSG_0_DATA0, response[0]), 0);
}

static void pf_loopback_echo(struct kunit *test)
{
        struct xe_guc_relay *relay = test->priv;
        u32 request[] = {
                FIELD_PREP(GUC_HXG_MSG_0_ORIGIN, GUC_HXG_ORIGIN_HOST) |
                FIELD_PREP(GUC_HXG_MSG_0_TYPE, GUC_HXG_TYPE_REQUEST) |
                FIELD_PREP(GUC_HXG_REQUEST_MSG_0_ACTION, GUC_RELAY_ACTION_VFXPF_TESTLOOP) |
                FIELD_PREP(GUC_HXG_REQUEST_MSG_0_DATA0, VFXPF_TESTLOOP_OPCODE_ECHO),
                TEST_DATA(1), TEST_DATA(2), TEST_DATA(3), TEST_DATA(4),
        };
        u32 response[ARRAY_SIZE(request)];
        unsigned int n;
        int ret;

        test_requires_relay_testloop(test);

        kunit_activate_static_stub(test, relay_kick_worker, relay_process_incoming_action);
        kunit_activate_static_stub(test, xe_guc_ct_send_recv,
                                   replacement_xe_guc_ct_send_recv_loopback_relay);
        ret = xe_guc_relay_send_to_vf(relay, TEST_VFID,
                                      request, ARRAY_SIZE(request),
                                      response, ARRAY_SIZE(response));
        KUNIT_ASSERT_EQ(test, ret, ARRAY_SIZE(response));
        KUNIT_EXPECT_EQ(test, FIELD_GET(GUC_HXG_MSG_0_ORIGIN, response[0]),
                        GUC_HXG_ORIGIN_HOST);
        KUNIT_EXPECT_EQ(test, FIELD_GET(GUC_HXG_MSG_0_TYPE, response[0]),
                        GUC_HXG_TYPE_RESPONSE_SUCCESS);
        KUNIT_EXPECT_EQ(test, FIELD_GET(GUC_HXG_RESPONSE_MSG_0_DATA0, response[0]),
                        ARRAY_SIZE(response));
        for (n = GUC_HXG_RESPONSE_MSG_MIN_LEN; n < ret; n++)
                KUNIT_EXPECT_EQ(test, request[n], response[n]);
}

static void pf_loopback_fail(struct kunit *test)
{
        struct xe_guc_relay *relay = test->priv;
        u32 request[] = {
                FIELD_PREP(GUC_HXG_MSG_0_ORIGIN, GUC_HXG_ORIGIN_HOST) |
                FIELD_PREP(GUC_HXG_MSG_0_TYPE, GUC_HXG_TYPE_REQUEST) |
                FIELD_PREP(GUC_HXG_REQUEST_MSG_0_ACTION, GUC_RELAY_ACTION_VFXPF_TESTLOOP) |
                FIELD_PREP(GUC_HXG_REQUEST_MSG_0_DATA0, VFXPF_TESTLOOP_OPCODE_FAIL),
        };
        u32 response[GUC_HXG_RESPONSE_MSG_MIN_LEN];
        int ret;

        test_requires_relay_testloop(test);

        kunit_activate_static_stub(test, relay_kick_worker, relay_process_incoming_action);
        kunit_activate_static_stub(test, xe_guc_ct_send_recv,
                                   replacement_xe_guc_ct_send_recv_loopback_relay);
        ret = xe_guc_relay_send_to_vf(relay, TEST_VFID,
                                      request, ARRAY_SIZE(request),
                                      response, ARRAY_SIZE(response));
        KUNIT_ASSERT_EQ(test, ret, -EREMOTEIO);
}

static void pf_loopback_busy(struct kunit *test)
{
        struct xe_guc_relay *relay = test->priv;
        u32 request[] = {
                FIELD_PREP(GUC_HXG_MSG_0_ORIGIN, GUC_HXG_ORIGIN_HOST) |
                FIELD_PREP(GUC_HXG_MSG_0_TYPE, GUC_HXG_TYPE_REQUEST) |
                FIELD_PREP(GUC_HXG_REQUEST_MSG_0_ACTION, GUC_RELAY_ACTION_VFXPF_TESTLOOP) |
                FIELD_PREP(GUC_HXG_REQUEST_MSG_0_DATA0, VFXPF_TESTLOOP_OPCODE_BUSY),
                TEST_DATA(0xb),
        };
        u32 response[GUC_HXG_RESPONSE_MSG_MIN_LEN];
        int ret;

        test_requires_relay_testloop(test);

        kunit_activate_static_stub(test, relay_testonly_nop, relay_process_incoming_action);
        kunit_activate_static_stub(test, relay_kick_worker, relay_process_incoming_action);
        kunit_activate_static_stub(test, xe_guc_ct_send_recv,
                                   replacement_xe_guc_ct_send_recv_loopback_relay);
        ret = xe_guc_relay_send_to_vf(relay, TEST_VFID,
                                      request, ARRAY_SIZE(request),
                                      response, ARRAY_SIZE(response));
        KUNIT_ASSERT_EQ(test, ret, GUC_HXG_RESPONSE_MSG_MIN_LEN);
}

static void pf_loopback_retry(struct kunit *test)
{
        struct xe_guc_relay *relay = test->priv;
        u32 request[] = {
                FIELD_PREP(GUC_HXG_MSG_0_ORIGIN, GUC_HXG_ORIGIN_HOST) |
                FIELD_PREP(GUC_HXG_MSG_0_TYPE, GUC_HXG_TYPE_REQUEST) |
                FIELD_PREP(GUC_HXG_REQUEST_MSG_0_ACTION, GUC_RELAY_ACTION_VFXPF_TESTLOOP) |
                FIELD_PREP(GUC_HXG_REQUEST_MSG_0_DATA0, VFXPF_TESTLOOP_OPCODE_RETRY),
                TEST_DATA(0xd), TEST_DATA(0xd),
        };
        u32 response[GUC_HXG_RESPONSE_MSG_MIN_LEN];
        int ret;

        test_requires_relay_testloop(test);

        kunit_activate_static_stub(test, relay_kick_worker, relay_process_incoming_action);
        kunit_activate_static_stub(test, xe_guc_ct_send_recv,
                                   replacement_xe_guc_ct_send_recv_loopback_relay);
        ret = xe_guc_relay_send_to_vf(relay, TEST_VFID,
                                      request, ARRAY_SIZE(request),
                                      response, ARRAY_SIZE(response));
        KUNIT_ASSERT_EQ(test, ret, GUC_HXG_RESPONSE_MSG_MIN_LEN);
}

static struct kunit_case pf_relay_test_cases[] = {
        KUNIT_CASE(pf_rejects_guc2pf_too_short),
        KUNIT_CASE(pf_rejects_guc2pf_too_long),
        KUNIT_CASE(pf_rejects_guc2pf_no_payload),
        KUNIT_CASE(pf_fails_no_payload),
        KUNIT_CASE(pf_fails_bad_origin),
        KUNIT_CASE(pf_fails_bad_type),
        KUNIT_CASE(pf_txn_reports_error),
        KUNIT_CASE(pf_txn_sends_pf2guc),
        KUNIT_CASE(pf_sends_pf2guc),
        KUNIT_CASE(pf_loopback_nop),
        KUNIT_CASE(pf_loopback_echo),
        KUNIT_CASE(pf_loopback_fail),
        KUNIT_CASE_SLOW(pf_loopback_busy),
        KUNIT_CASE_SLOW(pf_loopback_retry),
        {}
};

static struct kunit_suite pf_relay_suite = {
        .name = "pf_relay",
        .test_cases = pf_relay_test_cases,
        .init = relay_test_init,
};

static void vf_rejects_guc2vf_too_short(struct kunit *test)
{
        const u32 len = GUC2VF_RELAY_FROM_PF_EVENT_MSG_MIN_LEN - 1;
        struct xe_guc_relay *relay = test->priv;
        const u32 *msg = test_guc2vf;

        KUNIT_ASSERT_EQ(test, -EPROTO, xe_guc_relay_process_guc2vf(relay, msg, len));
}

static void vf_rejects_guc2vf_too_long(struct kunit *test)
{
        const u32 len = GUC2VF_RELAY_FROM_PF_EVENT_MSG_MAX_LEN + 1;
        struct xe_guc_relay *relay = test->priv;
        const u32 *msg = test_guc2vf;

        KUNIT_ASSERT_EQ(test, -EMSGSIZE, xe_guc_relay_process_guc2vf(relay, msg, len));
}

static void vf_rejects_guc2vf_no_payload(struct kunit *test)
{
        const u32 len = GUC2VF_RELAY_FROM_PF_EVENT_MSG_MIN_LEN;
        struct xe_guc_relay *relay = test->priv;
        const u32 *msg = test_guc2vf;

        KUNIT_ASSERT_EQ(test, -EPROTO, xe_guc_relay_process_guc2vf(relay, msg, len));
}

static struct kunit_case vf_relay_test_cases[] = {
        KUNIT_CASE(vf_rejects_guc2vf_too_short),
        KUNIT_CASE(vf_rejects_guc2vf_too_long),
        KUNIT_CASE(vf_rejects_guc2vf_no_payload),
        {}
};

static struct kunit_suite vf_relay_suite = {
        .name = "vf_relay",
        .test_cases = vf_relay_test_cases,
        .init = relay_test_init,
};

static void xe_drops_guc2pf_if_not_ready(struct kunit *test)
{
        struct xe_device *xe = test->priv;
        struct xe_guc_relay *relay = &xe_device_get_gt(xe, 0)->uc.guc.relay;
        const u32 *msg = test_guc2pf;
        u32 len = GUC2PF_RELAY_FROM_VF_EVENT_MSG_MIN_LEN + GUC_RELAY_MSG_MIN_LEN;

        KUNIT_ASSERT_EQ(test, -ENODEV, xe_guc_relay_process_guc2pf(relay, msg, len));
}

static void xe_drops_guc2vf_if_not_ready(struct kunit *test)
{
        struct xe_device *xe = test->priv;
        struct xe_guc_relay *relay = &xe_device_get_gt(xe, 0)->uc.guc.relay;
        const u32 *msg = test_guc2vf;
        u32 len = GUC2VF_RELAY_FROM_PF_EVENT_MSG_MIN_LEN + GUC_RELAY_MSG_MIN_LEN;

        KUNIT_ASSERT_EQ(test, -ENODEV, xe_guc_relay_process_guc2vf(relay, msg, len));
}

static void xe_rejects_send_if_not_ready(struct kunit *test)
{
        struct xe_device *xe = test->priv;
        struct xe_guc_relay *relay = &xe_device_get_gt(xe, 0)->uc.guc.relay;
        u32 msg[GUC_RELAY_MSG_MIN_LEN];
        u32 len = ARRAY_SIZE(msg);

        KUNIT_ASSERT_EQ(test, -ENODEV, xe_guc_relay_send_to_pf(relay, msg, len, NULL, 0));
        KUNIT_ASSERT_EQ(test, -ENODEV, relay_send_to(relay, TEST_VFID, msg, len, NULL, 0));
}

static struct kunit_case no_relay_test_cases[] = {
        KUNIT_CASE(xe_drops_guc2pf_if_not_ready),
        KUNIT_CASE(xe_drops_guc2vf_if_not_ready),
        KUNIT_CASE(xe_rejects_send_if_not_ready),
        {}
};

static struct kunit_suite no_relay_suite = {
        .name = "no_relay",
        .test_cases = no_relay_test_cases,
        .init = xe_kunit_helper_xe_device_test_init,
};

kunit_test_suites(&no_relay_suite,
                  &pf_relay_suite,
                  &vf_relay_suite);