1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
/*
* Copyright (C) 2022-2025 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#include "shared/test/common/libult/linux/drm_query_mock.h"
#include "shared/source/helpers/ptr_math.h"
#include "shared/source/os_interface/linux/i915.h"
#include "gtest/gtest.h"
DrmQueryMock::DrmQueryMock(RootDeviceEnvironment &rootDeviceEnvironment) : DrmMock(rootDeviceEnvironment) {
context.hwInfo = rootDeviceEnvironment.getMutableHardwareInfo();
callBaseIsVmBindAvailable = true;
this->ioctlHelper = std::make_unique<IoctlHelperPrelim20>(*this);
memoryInfoQueried = false;
EXPECT_TRUE(queryMemoryInfo());
EXPECT_EQ(2u + getBaseIoctlCalls(), ioctlCallsCount);
ioctlCallsCount = 0;
}
int DrmQueryMock::handleRemainingRequests(DrmIoctl request, void *arg) {
if (request == DrmIoctl::query && arg) {
if (i915QuerySuccessCount == 0) {
return EINVAL;
}
i915QuerySuccessCount--;
auto query = static_cast<Query *>(arg);
if (query->itemsPtr == 0) {
return EINVAL;
}
for (auto i = 0u; i < query->numItems; ++i) {
const auto queryItem = reinterpret_cast<QueryItem *>(query->itemsPtr) + i;
if (!this->handleQueryItem(queryItem)) {
return EINVAL;
}
}
return 0;
} else if (request == DrmIoctl::gemContextSetparam && receivedContextParamRequest.param == I915_CONTEXT_PARAM_ENGINES) {
EXPECT_LE(receivedContextParamRequest.size, sizeof(receivedContextParamEngines));
memcpy(&receivedContextParamEngines, reinterpret_cast<const void *>(receivedContextParamRequest.value), receivedContextParamRequest.size);
auto srcBalancer = reinterpret_cast<const I915::i915_context_engines_load_balance *>(receivedContextParamEngines.extensions);
if (srcBalancer) {
EXPECT_EQ(static_cast<__u32>(I915_CONTEXT_ENGINES_EXT_LOAD_BALANCE), srcBalancer->base.name);
auto balancerSize = ptrDiff(srcBalancer->engines + srcBalancer->num_siblings, srcBalancer);
EXPECT_LE(balancerSize, sizeof(receivedContextEnginesLoadBalance));
memcpy(&receivedContextEnginesLoadBalance, srcBalancer, balancerSize);
}
return storedRetValForSetParamEngines;
}
return context.handlePrelimRequest(request, arg);
}
bool DrmQueryMock::handleQueryItem(void *arg) {
return context.handlePrelimQueryItem(arg);
}
|
