diff --git a/pkg/abi/nvgpu/ctrl.go b/pkg/abi/nvgpu/ctrl.go index 5ba1199d31..50837dd32c 100644 --- a/pkg/abi/nvgpu/ctrl.go +++ b/pkg/abi/nvgpu/ctrl.go @@ -444,12 +444,14 @@ const ( // From src/common/sdk/nvidia/inc/ctrl/ctrl2080/ctrl2080fb.h: const ( - NV2080_CTRL_CMD_FB_GET_INFO = 0x20801301 - NV2080_CTRL_CMD_FB_GET_INFO_V2 = 0x20801303 - NV2080_CTRL_CMD_FB_GET_GPU_CACHE_INFO = 0x20801315 - NV2080_CTRL_CMD_FB_GET_FB_REGION_INFO = 0x20801320 - NV2080_CTRL_CMD_FB_GET_SEMAPHORE_SURFACE_LAYOUT = 0x20801352 - NV2080_CTRL_CMD_FB_QUERY_DRAM_ENCRYPTION_INFOROM_SUPPORT = 0x20801358 + NV2080_CTRL_CMD_FB_GET_INFO = 0x20801301 + NV2080_CTRL_CMD_FB_GET_INFO_V2 = 0x20801303 + NV2080_CTRL_CMD_FB_GET_GPU_CACHE_INFO = 0x20801315 + NV2080_CTRL_CMD_FB_GET_FB_REGION_INFO = 0x20801320 + NV2080_CTRL_CMD_FB_GET_SEMAPHORE_SURFACE_LAYOUT = 0x20801352 + NV2080_CTRL_CMD_FB_QUERY_DRAM_ENCRYPTION_INFOROM_SUPPORT = 0x20801358 + NV2080_CTRL_CMD_FB_QUERY_DRAM_ENCRYPTION_INFOROM_SUPPORT_575_51_02 = 0x20801357 + NV2080_CTRL_CMD_FB_QUERY_DRAM_ENCRYPTION_STATUS = 0x20801358 ) // From src/common/sdk/nvidia/inc/ctrl/ctrl2080/ctrl2080fifo.h: diff --git a/pkg/sentry/devices/nvproxy/version.go b/pkg/sentry/devices/nvproxy/version.go index 78bbe56033..b201a4280c 100644 --- a/pkg/sentry/devices/nvproxy/version.go +++ b/pkg/sentry/devices/nvproxy/version.go @@ -871,10 +871,15 @@ func Init() { v575_51_02 := func() *driverABI { abi := v570_133_20() abi.controlCmd[nvgpu.NV2080_CTRL_CMD_THERMAL_SYSTEM_EXECUTE_V2] = ctrlHandler(rmControlSimple, compUtil) + abi.controlCmd[nvgpu.NV2080_CTRL_CMD_FB_QUERY_DRAM_ENCRYPTION_INFOROM_SUPPORT_575_51_02] = ctrlHandler(rmControlSimple, compUtil) + abi.controlCmd[nvgpu.NV2080_CTRL_CMD_FB_QUERY_DRAM_ENCRYPTION_STATUS] = ctrlHandler(rmControlSimple, compUtil) + prevStructs := abi.getStructs abi.getStructs = func() *driverABIStructs { structs := prevStructs() structs.controlStructs[nvgpu.NV2080_CTRL_CMD_THERMAL_SYSTEM_EXECUTE_V2] = simpleDriverStruct("NV2080_CTRL_THERMAL_SYSTEM_EXECUTE_V2_PARAMS") + structs.controlStructs[nvgpu.NV2080_CTRL_CMD_FB_QUERY_DRAM_ENCRYPTION_INFOROM_SUPPORT_575_51_02] = simpleDriverStruct("NV2080_CTRL_FB_DRAM_ENCRYPTION_INFOROM_SUPPORT_PARAMS") + structs.controlStructs[nvgpu.NV2080_CTRL_CMD_FB_QUERY_DRAM_ENCRYPTION_STATUS] = simpleDriverStruct("NV2080_CTRL_FB_DRAM_ENCRYPTION_STATUS") return structs } return abi