summaryrefslogtreecommitdiffstats
path: root/tools/rmq/grctx_oracle.c
blob: a2fc5fa3eb748ddf11cc2919e6abfef4a993f7b6 (plain)
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
/* grctx_oracle - query the host NVIDIA RM for the engine context buffer sizes.
 *
 * Goal: recover the table that GSP-RM serves for
 * NV2080_CTRL_CMD_INTERNAL_STATIC_KGR_GET_CONTEXT_BUFFERS_INFO (0x20800a32),
 * so the emulation layer does not have to guess the per-engine ctx buffer
 * sizes (kernel/gsp_rm_tu104.h: TU104_CTX_SIZE_*).
 *
 * Outcome (2026-09-22): NOT REACHABLE from userspace on a bare-metal PF:
 *   - 0x801707 NV0080_CTRL_CMD_FIFO_GET_ENGINE_CONTEXT_PROPERTIES
 *       hObject=hDevice  -> 0x57 NV_ERR_OBJECT_NOT_FOUND (not exported)
 *       hObject=hClient  -> 0x56 NV_ERR_NOT_SUPPORTED (bound to the
 *                           _92bfc3 hal: the client RM does not manage ctx
 *                           buffers on a PF, only a VF's VGPU_STATIC_INFO)
 *   - 0x2080122d NV2080_CTRL_CMD_GR_GET_ENGINE_CONTEXT_PROPERTIES
 *       (subdevice)      -> 0x57 / 0x56, never returns sizes
 *   - The producing control (0x20800a32) is internal-only.
 * The real values are probed from the ctxsw ucode by the physical RM
 * (grSetupInitialCtxState -> grDiscoverCtxBufferSize) and are internal to
 * GSP-RM, so the emulation currently serves synthesized page-aligned sizes
 * (see tu104_ctx_buffer_info in kernel/gsp_rm_tu104.h).
 *
 * Kept as a reference for the RM client bring-up sequence and for a future
 * attempt via a kernel-side RM client.
 */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/ioctl.h>
#include <stdint.h>

typedef uint32_t NvU32;
typedef uint64_t NvU64;
typedef uint32_t NvHandle;
typedef uint32_t NvV32;
typedef uint64_t NvP64;

#define NV_IOCTL_MAGIC      'F'
#define NV_ESC_RM_ALLOC     0x2B
#define NV_ESC_RM_CONTROL   0x2A

#define NV01_ROOT           0x0
#define NV01_DEVICE_0       0x80
#define NV20_SUBDEVICE_0    0x2080

typedef struct {
    NvHandle hRoot;
    NvHandle hObjectParent;
    NvHandle hObjectNew;
    NvV32    hClass;
    NvP64    pAllocParms __attribute__((aligned(8)));
    NvU32    paramsSize;
    NvV32    status;
} NVOS21_PARAMETERS;

typedef struct {
    NvHandle hRoot;
    NvHandle hObjectParent;
    NvHandle hObjectNew;
    NvV32    hClass;
    NvP64    pAllocParms __attribute__((aligned(8)));
    NvP64    pRightsRequested __attribute__((aligned(8)));
    NvU32    paramsSize;
    NvU32    flags;
    NvV32    status;
} NVOS64_PARAMETERS;

typedef struct {
    NvHandle hClient;
    NvHandle hObject;
    NvV32    cmd;
    NvU32    flags;
    NvP64    params __attribute__((aligned(8)));
    NvU32    paramsSize;
    NvV32    status;
} NVOS54_PARAMETERS;

#define NV_IOCTL_RM_ALLOC   _IOWR(NV_IOCTL_MAGIC, NV_ESC_RM_ALLOC, NVOS21_PARAMETERS)
#define NV_IOCTL_RM_CONTROL _IOWR(NV_IOCTL_MAGIC, NV_ESC_RM_CONTROL, NVOS54_PARAMETERS)

#define NV0080_CTRL_CMD_FIFO_GET_ENGINE_CONTEXT_PROPERTIES 0x801707U
#define NV2080_CTRL_CMD_GR_GET_ENGINE_CONTEXT_PROPERTIES   0x2080122dU
#define ENGINE_ID_COUNT 0x1a

typedef struct {
    NvU32 engineId;
    NvU32 alignment;
    NvU32 size;
} FIFO_GET_ENGINE_CONTEXT_PROPERTIES_PARAMS;

typedef struct {
    NvU32 flags;
    NvU32 pad;
    NvU64 route;
} GR_ROUTE_INFO;

typedef struct {
    GR_ROUTE_INFO grRouteInfo;
    NvU32 engineId;
    NvU32 alignment;
    NvU32 size;
    NvU32 bInfoPopulated;
} GR_GET_ENGINE_CONTEXT_PROPERTIES_PARAMS;

static int fd;

static NvU32 rm_alloc(NvHandle parent, NvHandle client, NvU32 hClass, NvHandle *pOut)
{
    NVOS21_PARAMETERS p = {0};
    p.hRoot = client;
    p.hObjectParent = parent;
    p.hClass = hClass;
    if (ioctl(fd, NV_IOCTL_RM_ALLOC, &p) < 0) { perror("alloc ioctl"); return 1; }
    if (p.status) { fprintf(stderr, "alloc class 0x%x failed: 0x%x\n", hClass, p.status); return p.status; }
    *pOut = p.hObjectNew;
    return 0;
}

static NvU32 rm_alloc64(NvHandle parent, NvHandle client, NvU32 hClass,
                        NvHandle newHandle, void *parms, NvU32 parmsSize, NvHandle *pOut)
{
    NVOS64_PARAMETERS p = {0};
    p.hRoot = client;
    p.hObjectParent = parent;
    p.hObjectNew = newHandle; /* 0 = let RM generate (device); subdevices take an explicit handle */
    p.hClass = hClass;
    p.pAllocParms = (NvP64)parms;
    p.paramsSize = parmsSize;
    p.flags = 0;
    if (ioctl(fd, NV_IOCTL_RM_ALLOC, &p) < 0) { perror("alloc64 ioctl"); return 1; }
    if (p.status) {
        fprintf(stderr, "alloc64 class 0x%x failed: 0x%x\n", hClass, p.status);
        return p.status;
    }
    *pOut = p.hObjectNew;
    return 0;
}

static NvU32 rm_control(NvHandle hClient, NvHandle hObject, NvU32 cmd, void *params, NvU32 size)
{
    NVOS54_PARAMETERS p = {0};
    p.hClient = hClient; p.hObject = hObject; p.cmd = cmd;
    p.params = (NvP64)params; p.paramsSize = size;
    if (ioctl(fd, NV_IOCTL_RM_CONTROL, &p) < 0) { perror("control ioctl"); return 1; }
    return p.status;
}

int main(void)
{
    NvHandle hClient = 0, hDevice = 0, hSubdevice = 0;
    NvU32 deviceParams = 0;    /* deviceId 0 */
    NvU32 subdeviceParams = 0; /* subDeviceId 0 */
    NvU32 st, i;

    fd = open("/dev/nvidiactl", O_RDWR);
    if (fd < 0) { perror("open /dev/nvidiactl"); return 1; }

    if ((st = rm_alloc(0, 0, NV01_ROOT, &hClient))) return 1;
    if ((st = rm_alloc64(hClient, hClient, NV01_DEVICE_0, 0,
                         &deviceParams, sizeof(deviceParams), &hDevice))) return 1;
    if ((st = rm_alloc64(hDevice, hClient, NV20_SUBDEVICE_0, 0xcaf00001,
                         &subdeviceParams, sizeof(subdeviceParams), &hSubdevice))) return 1;
    printf("hClient=0x%x hDevice=0x%x hSubdevice=0x%x\n", hClient, hDevice, hSubdevice);

    printf("=== device-level FIFO_GET_ENGINE_CONTEXT_PROPERTIES (0x801707) ===\n");
    for (i = 0; i < ENGINE_ID_COUNT; i++) {
        FIFO_GET_ENGINE_CONTEXT_PROPERTIES_PARAMS p;
        memset(&p, 0, sizeof(p));
        p.engineId = i;
        st = rm_control(hClient, hDevice, NV0080_CTRL_CMD_FIFO_GET_ENGINE_CONTEXT_PROPERTIES,
                        &p, sizeof(p));
        printf("engine[%2u] status=0x%x size=0x%08x alignment=0x%08x\n",
               i, st, p.size, p.alignment);
    }

    printf("=== subdevice-level GR_GET_ENGINE_CONTEXT_PROPERTIES (0x2080122d) ===\n");
    for (i = 0; i < ENGINE_ID_COUNT; i++) {
        GR_GET_ENGINE_CONTEXT_PROPERTIES_PARAMS p;
        memset(&p, 0, sizeof(p));
        p.engineId = i;
        st = rm_control(hClient, hSubdevice, NV2080_CTRL_CMD_GR_GET_ENGINE_CONTEXT_PROPERTIES,
                        &p, sizeof(p));
        printf("engine[%2u] status=0x%x size=0x%08x alignment=0x%08x\n",
               i, st, p.size, p.alignment);
    }

    return 0;
}