/* * Copyright (c) 2015-2022, NVIDIA CORPORATION. All rights reserved. * * Permission is hereby granted, free of charge, to any person obtaining a * copy of this software and associated documentation files (the "Software"), * to deal in the Software without restriction, including without limitation * the rights to use, copy, modify, merge, publish, distribute, sublicense, * and/or sell copies of the Software, and to permit persons to whom the * Software is furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice shall be included in * all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER * DEALINGS IN THE SOFTWARE. */ #include "nvidia-drm-conftest.h" /* NV_DRM_AVAILABLE and NV_DRM_DRM_GEM_H_PRESENT */ #include "nvidia-drm-priv.h" #include "nvidia-drm-drv.h" #include "nvidia-drm-fb.h" #include "nvidia-drm-modeset.h" #include "nvidia-drm-encoder.h" #include "nvidia-drm-connector.h" #include "nvidia-drm-gem.h" #include "nvidia-drm-crtc.h" #include "nvidia-drm-fence.h" #include "nvidia-drm-helper.h" #include "nvidia-drm-gem-nvkms-memory.h" #include "nvidia-drm-gem-user-memory.h" #include "nvidia-drm-gem-dma-buf.h" #if defined(NV_DRM_AVAILABLE) #include "nvidia-drm-ioctl.h" #if defined(NV_DRM_DRMP_H_PRESENT) #include #endif #if defined(NV_DRM_DRM_VBLANK_H_PRESENT) #include #endif #if defined(NV_DRM_DRM_FILE_H_PRESENT) #include #endif #if defined(NV_DRM_DRM_PRIME_H_PRESENT) #include #endif #if defined(NV_DRM_DRM_IOCTL_H_PRESENT) #include #endif #include /* * Commit fcd70cd36b9b ("drm: Split out drm_probe_helper.h") * moves a number of helper function definitions from * drm/drm_crtc_helper.h to a new drm_probe_helper.h. */ #if defined(NV_DRM_DRM_PROBE_HELPER_H_PRESENT) #include #endif #include #if defined(NV_DRM_DRM_GEM_H_PRESENT) #include #endif #if defined(NV_DRM_DRM_AUTH_H_PRESENT) #include #endif #if defined(NV_DRM_ATOMIC_MODESET_AVAILABLE) #include #endif static struct nv_drm_device *dev_list = NULL; static const char* nv_get_input_colorspace_name( enum NvKmsInputColorSpace colorSpace) { switch (colorSpace) { case NVKMS_INPUT_COLORSPACE_NONE: return "None"; case NVKMS_INPUT_COLORSPACE_SCRGB_LINEAR: return "IEC 61966-2-2 linear FP"; case NVKMS_INPUT_COLORSPACE_BT2100_PQ: return "ITU-R BT.2100-PQ YCbCr"; default: /* We shoudn't hit this */ WARN_ON("Unsupported input colorspace"); return "None"; } }; #if defined(NV_DRM_ATOMIC_MODESET_AVAILABLE) #if defined(NV_DRM_OUTPUT_POLL_CHANGED_PRESENT) static void nv_drm_output_poll_changed(struct drm_device *dev) { struct drm_connector *connector = NULL; struct drm_mode_config *config = &dev->mode_config; #if defined(NV_DRM_CONNECTOR_LIST_ITER_PRESENT) struct drm_connector_list_iter conn_iter; nv_drm_connector_list_iter_begin(dev, &conn_iter); #endif /* * Here drm_mode_config::mutex has been acquired unconditionally: * * - In the non-NV_DRM_CONNECTOR_LIST_ITER_PRESENT case, the mutex must * be held for the duration of walking over the connectors. * * - In the NV_DRM_CONNECTOR_LIST_ITER_PRESENT case, the mutex must be * held for the duration of a fill_modes() call chain: * connector->funcs->fill_modes() * |-> drm_helper_probe_single_connector_modes() * * It is easiest to always acquire the mutext for the entire connector * loop. */ mutex_lock(&config->mutex); nv_drm_for_each_connector(connector, &conn_iter, dev) { struct nv_drm_connector *nv_connector = to_nv_connector(connector); if (!nv_drm_connector_check_connection_status_dirty_and_clear( nv_connector)) { continue; } connector->funcs->fill_modes( connector, dev->mode_config.max_width, dev->mode_config.max_height); } mutex_unlock(&config->mutex); #if defined(NV_DRM_CONNECTOR_LIST_ITER_PRESENT) nv_drm_connector_list_iter_end(&conn_iter); #endif } #endif /* NV_DRM_OUTPUT_POLL_CHANGED_PRESENT */ static struct drm_framebuffer *nv_drm_framebuffer_create( struct drm_device *dev, struct drm_file *file, #if defined(NV_DRM_HELPER_MODE_FILL_FB_STRUCT_HAS_CONST_MODE_CMD_ARG) const struct drm_mode_fb_cmd2 *cmd #else struct drm_mode_fb_cmd2 *cmd #endif ) { struct drm_mode_fb_cmd2 local_cmd; struct drm_framebuffer *fb; local_cmd = *cmd; fb = nv_drm_internal_framebuffer_create( dev, file, &local_cmd); #if !defined(NV_DRM_HELPER_MODE_FILL_FB_STRUCT_HAS_CONST_MODE_CMD_ARG) *cmd = local_cmd; #endif return fb; } static const struct drm_mode_config_funcs nv_mode_config_funcs = { .fb_create = nv_drm_framebuffer_create, .atomic_state_alloc = nv_drm_atomic_state_alloc, .atomic_state_clear = nv_drm_atomic_state_clear, .atomic_state_free = nv_drm_atomic_state_free, .atomic_check = nv_drm_atomic_check, .atomic_commit = nv_drm_atomic_commit, #if defined(NV_DRM_OUTPUT_POLL_CHANGED_PRESENT) .output_poll_changed = nv_drm_output_poll_changed, #endif }; static void nv_drm_event_callback(const struct NvKmsKapiEvent *event) { struct nv_drm_device *nv_dev = event->privateData; mutex_lock(&nv_dev->lock); if (!atomic_read(&nv_dev->enable_event_handling)) { goto done; } switch (event->type) { case NVKMS_EVENT_TYPE_DPY_CHANGED: nv_drm_handle_display_change( nv_dev, event->u.displayChanged.display); break; case NVKMS_EVENT_TYPE_DYNAMIC_DPY_CONNECTED: nv_drm_handle_dynamic_display_connected( nv_dev, event->u.dynamicDisplayConnected.display); break; case NVKMS_EVENT_TYPE_FLIP_OCCURRED: nv_drm_handle_flip_occurred( nv_dev, event->u.flipOccurred.head, event->u.flipOccurred.layer); break; default: break; } done: mutex_unlock(&nv_dev->lock); } /* * Helper function to initialize drm_device::mode_config from * NvKmsKapiDevice's resource information. */ static void nv_drm_init_mode_config(struct nv_drm_device *nv_dev, const struct NvKmsKapiDeviceResourcesInfo *pResInfo) { struct drm_device *dev = nv_dev->dev; drm_mode_config_init(dev); drm_mode_create_dvi_i_properties(dev); dev->mode_config.funcs = &nv_mode_config_funcs; dev->mode_config.min_width = pResInfo->caps.minWidthInPixels; dev->mode_config.min_height = pResInfo->caps.minHeightInPixels; dev->mode_config.max_width = pResInfo->caps.maxWidthInPixels; dev->mode_config.max_height = pResInfo->caps.maxHeightInPixels; dev->mode_config.cursor_width = pResInfo->caps.maxCursorSizeInPixels; dev->mode_config.cursor_height = pResInfo->caps.maxCursorSizeInPixels; /* * NVIDIA GPUs have no preferred depth. Arbitrarily report 24, to be * consistent with other DRM drivers. */ dev->mode_config.preferred_depth = 24; dev->mode_config.prefer_shadow = 1; #if defined(NV_DRM_CRTC_STATE_HAS_ASYNC_FLIP) || \ defined(NV_DRM_CRTC_STATE_HAS_PAGEFLIP_FLAGS) dev->mode_config.async_page_flip = true; #else dev->mode_config.async_page_flip = false; #endif #if defined(NV_DRM_FORMAT_MODIFIERS_PRESENT) && \ defined(NV_DRM_MODE_CONFIG_HAS_ALLOW_FB_MODIFIERS) /* Allow clients to define framebuffer layouts using DRM format modifiers */ dev->mode_config.allow_fb_modifiers = true; #endif /* Initialize output polling support */ drm_kms_helper_poll_init(dev); /* Disable output polling, because we don't support it yet */ drm_kms_helper_poll_disable(dev); } /* * Helper function to enumerate encoders/connectors from NvKmsKapiDevice. */ static void nv_drm_enumerate_encoders_and_connectors ( struct nv_drm_device *nv_dev ) { struct drm_device *dev = nv_dev->dev; NvU32 nDisplays = 0; if (!nvKms->getDisplays(nv_dev->pDevice, &nDisplays, NULL)) { NV_DRM_DEV_LOG_ERR( nv_dev, "Failed to enumurate NvKmsKapiDisplay count"); } if (nDisplays != 0) { NvKmsKapiDisplay *hDisplays = nv_drm_calloc(nDisplays, sizeof(*hDisplays)); if (hDisplays != NULL) { if (!nvKms->getDisplays(nv_dev->pDevice, &nDisplays, hDisplays)) { NV_DRM_DEV_LOG_ERR( nv_dev, "Failed to enumurate NvKmsKapiDisplay handles"); } else { NvU32 i; for (i = 0; i < nDisplays; i++) { struct drm_encoder *encoder = nv_drm_add_encoder(dev, hDisplays[i]); if (IS_ERR(encoder)) { NV_DRM_DEV_LOG_ERR( nv_dev, "Failed to add connector for NvKmsKapiDisplay 0x%08x", hDisplays[i]); } } } nv_drm_free(hDisplays); } else { NV_DRM_DEV_LOG_ERR( nv_dev, "Failed to allocate memory for NvKmsKapiDisplay array"); } } } #endif /* NV_DRM_ATOMIC_MODESET_AVAILABLE */ /*! * 'NV_DRM_OUT_FENCE_PTR' is an atomic per-plane property that clients can use * to request an out-fence fd for a particular plane that's being flipped. * 'NV_DRM_OUT_FENCE_PTR' does NOT have the same behavior as the standard * 'OUT_FENCE_PTR' property - the fd that's returned via 'NV_DRM_OUT_FENCE_PTR' * will only be signaled once the buffers in the corresponding flip are flipped * away from. * In order to use this property, client needs to call set property function * with user mode pointer as value. Once driver have post syncpt fd from flip reply, * it will copy post syncpt fd at location pointed by user mode pointer. */ static int nv_drm_create_properties(struct nv_drm_device *nv_dev) { struct drm_prop_enum_list enum_list[3] = { }; int i, len = 0; for (i = 0; i < 3; i++) { enum_list[len].type = i; enum_list[len].name = nv_get_input_colorspace_name(i); len++; } #if defined(NV_LINUX_NVHOST_H_PRESENT) && defined(CONFIG_TEGRA_GRHOST) if (!nv_dev->supportsSyncpts) { return 0; } nv_dev->nv_out_fence_property = drm_property_create_range(nv_dev->dev, DRM_MODE_PROP_ATOMIC, "NV_DRM_OUT_FENCE_PTR", 0, U64_MAX); if (nv_dev->nv_out_fence_property == NULL) { return -ENOMEM; } #endif nv_dev->nv_input_colorspace_property = drm_property_create_enum(nv_dev->dev, 0, "NV_INPUT_COLORSPACE", enum_list, len); if (nv_dev->nv_input_colorspace_property == NULL) { NV_DRM_LOG_ERR("Failed to create NV_INPUT_COLORSPACE property"); return -ENOMEM; } #if defined(NV_DRM_HAS_HDR_OUTPUT_METADATA) nv_dev->nv_hdr_output_metadata_property = drm_property_create(nv_dev->dev, DRM_MODE_PROP_BLOB, "NV_HDR_STATIC_METADATA", 0); if (nv_dev->nv_hdr_output_metadata_property == NULL) { return -ENOMEM; } #endif return 0; } static int nv_drm_load(struct drm_device *dev, unsigned long flags) { #if defined(NV_DRM_ATOMIC_MODESET_AVAILABLE) struct NvKmsKapiDevice *pDevice; struct NvKmsKapiAllocateDeviceParams allocateDeviceParams; struct NvKmsKapiDeviceResourcesInfo resInfo; #endif #if defined(NV_DRM_FORMAT_MODIFIERS_PRESENT) NvU64 kind; NvU64 gen; int i; #endif int ret; struct nv_drm_device *nv_dev = to_nv_device(dev); NV_DRM_DEV_LOG_INFO(nv_dev, "Loading driver"); #if defined(NV_DRM_ATOMIC_MODESET_AVAILABLE) if (!drm_core_check_feature(dev, DRIVER_MODESET)) { return 0; } /* Allocate NvKmsKapiDevice from GPU ID */ memset(&allocateDeviceParams, 0, sizeof(allocateDeviceParams)); allocateDeviceParams.gpuId = nv_dev->gpu_info.gpu_id; allocateDeviceParams.privateData = nv_dev; allocateDeviceParams.eventCallback = nv_drm_event_callback; pDevice = nvKms->allocateDevice(&allocateDeviceParams); if (pDevice == NULL) { NV_DRM_DEV_LOG_ERR(nv_dev, "Failed to allocate NvKmsKapiDevice"); return -ENODEV; } /* Query information of resources available on device */ if (!nvKms->getDeviceResourcesInfo(pDevice, &resInfo)) { nvKms->freeDevice(pDevice); NV_DRM_DEV_LOG_ERR( nv_dev, "Failed to query NvKmsKapiDevice resources info"); return -ENODEV; } mutex_lock(&nv_dev->lock); /* Set NvKmsKapiDevice */ nv_dev->pDevice = pDevice; nv_dev->pitchAlignment = resInfo.caps.pitchAlignment; nv_dev->hasVideoMemory = resInfo.caps.hasVideoMemory; nv_dev->genericPageKind = resInfo.caps.genericPageKind; // Fermi-Volta use generation 0, Turing+ uses generation 2. nv_dev->pageKindGeneration = (nv_dev->genericPageKind == 0x06) ? 2 : 0; // Desktop GPUs and mobile GPUs Xavier and later use the same sector layout nv_dev->sectorLayout = 1; nv_dev->supportsSyncpts = resInfo.caps.supportsSyncpts; #if defined(NV_DRM_FORMAT_MODIFIERS_PRESENT) gen = nv_dev->pageKindGeneration; kind = nv_dev->genericPageKind; for (i = 0; i <= 5; i++) { nv_dev->modifiers[i] = /* Log2(block height) ----------------------------------+ * * Page Kind ------------------------------------+ | * * Gob Height/Page Kind Generation --------+ | | * * Sector layout ---------------------+ | | | * * Compression --------------------+ | | | | * * | | | | | */ DRM_FORMAT_MOD_NVIDIA_BLOCK_LINEAR_2D(0, 1, gen, kind, 5 - i); } nv_dev->modifiers[i++] = DRM_FORMAT_MOD_LINEAR; nv_dev->modifiers[i++] = DRM_FORMAT_MOD_INVALID; #endif /* defined(NV_DRM_FORMAT_MODIFIERS_PRESENT) */ /* Initialize drm_device::mode_config */ nv_drm_init_mode_config(nv_dev, &resInfo); ret = nv_drm_create_properties(nv_dev); if (ret < 0) { return -ENODEV; } if (!nvKms->declareEventInterest( nv_dev->pDevice, ((1 << NVKMS_EVENT_TYPE_DPY_CHANGED) | (1 << NVKMS_EVENT_TYPE_DYNAMIC_DPY_CONNECTED) | (1 << NVKMS_EVENT_TYPE_FLIP_OCCURRED)))) { NV_DRM_DEV_LOG_ERR(nv_dev, "Failed to register event mask"); } /* Add crtcs */ nv_drm_enumerate_crtcs_and_planes(nv_dev, &resInfo); /* Add connectors and encoders */ nv_drm_enumerate_encoders_and_connectors(nv_dev); #if !defined(NV_DRM_CRTC_STATE_HAS_NO_VBLANK) drm_vblank_init(dev, dev->mode_config.num_crtc); #endif /* * Trigger hot-plug processing, to update connection status of * all HPD supported connectors. */ drm_helper_hpd_irq_event(dev); /* Enable event handling */ atomic_set(&nv_dev->enable_event_handling, true); init_waitqueue_head(&nv_dev->flip_event_wq); mutex_unlock(&nv_dev->lock); #endif /* NV_DRM_ATOMIC_MODESET_AVAILABLE */ return 0; } static void __nv_drm_unload(struct drm_device *dev) { #if defined(NV_DRM_ATOMIC_MODESET_AVAILABLE) struct NvKmsKapiDevice *pDevice = NULL; #endif struct nv_drm_device *nv_dev = to_nv_device(dev); NV_DRM_DEV_LOG_INFO(nv_dev, "Unloading driver"); #if defined(NV_DRM_ATOMIC_MODESET_AVAILABLE) if (!drm_core_check_feature(dev, DRIVER_MODESET)) { return; } mutex_lock(&nv_dev->lock); /* Disable event handling */ atomic_set(&nv_dev->enable_event_handling, false); /* Clean up output polling */ drm_kms_helper_poll_fini(dev); /* Clean up mode configuration */ drm_mode_config_cleanup(dev); if (!nvKms->declareEventInterest(nv_dev->pDevice, 0x0)) { NV_DRM_DEV_LOG_ERR(nv_dev, "Failed to stop event listening"); } /* Unset NvKmsKapiDevice */ pDevice = nv_dev->pDevice; nv_dev->pDevice = NULL; mutex_unlock(&nv_dev->lock); nvKms->freeDevice(pDevice); #endif /* NV_DRM_ATOMIC_MODESET_AVAILABLE */ } #if defined(NV_DRM_DRIVER_UNLOAD_HAS_INT_RETURN_TYPE) static int nv_drm_unload(struct drm_device *dev) { __nv_drm_unload(dev); return 0; } #else static void nv_drm_unload(struct drm_device *dev) { __nv_drm_unload(dev); } #endif #if defined(NV_DRM_ATOMIC_MODESET_AVAILABLE) static int __nv_drm_master_set(struct drm_device *dev, struct drm_file *file_priv, bool from_open) { struct nv_drm_device *nv_dev = to_nv_device(dev); if (!nvKms->grabOwnership(nv_dev->pDevice)) { return -EINVAL; } return 0; } #if defined(NV_DRM_DRIVER_SET_MASTER_HAS_INT_RETURN_TYPE) static int nv_drm_master_set(struct drm_device *dev, struct drm_file *file_priv, bool from_open) { return __nv_drm_master_set(dev, file_priv, from_open); } #else static void nv_drm_master_set(struct drm_device *dev, struct drm_file *file_priv, bool from_open) { if (__nv_drm_master_set(dev, file_priv, from_open) != 0) { NV_DRM_DEV_LOG_ERR(to_nv_device(dev), "Failed to grab modeset ownership"); } } #endif #if defined(NV_DRM_MASTER_DROP_HAS_FROM_RELEASE_ARG) static void nv_drm_master_drop(struct drm_device *dev, struct drm_file *file_priv, bool from_release) #else static void nv_drm_master_drop(struct drm_device *dev, struct drm_file *file_priv) #endif { struct nv_drm_device *nv_dev = to_nv_device(dev); int err; /* * After dropping nvkms modeset onwership, it is not guaranteed that * drm and nvkms modeset state will remain in sync. Therefore, disable * all outputs and crtcs before dropping nvkms modeset ownership. * * First disable all active outputs atomically and then disable each crtc one * by one, there is not helper function available to disable all crtcs * atomically. */ drm_modeset_lock_all(dev); if ((err = nv_drm_atomic_helper_disable_all( dev, dev->mode_config.acquire_ctx)) != 0) { NV_DRM_DEV_LOG_ERR( nv_dev, "nv_drm_atomic_helper_disable_all failed with error code %d !", err); } drm_modeset_unlock_all(dev); nvKms->releaseOwnership(nv_dev->pDevice); } #endif /* NV_DRM_ATOMIC_MODESET_AVAILABLE */ #if defined(NV_DRM_BUS_PRESENT) || defined(NV_DRM_DRIVER_HAS_SET_BUSID) static int nv_drm_pci_set_busid(struct drm_device *dev, struct drm_master *master) { struct nv_drm_device *nv_dev = to_nv_device(dev); master->unique = nv_drm_asprintf("pci:%04x:%02x:%02x.%d", nv_dev->gpu_info.pci_info.domain, nv_dev->gpu_info.pci_info.bus, nv_dev->gpu_info.pci_info.slot, nv_dev->gpu_info.pci_info.function); if (master->unique == NULL) { return -ENOMEM; } master->unique_len = strlen(master->unique); return 0; } #endif static int nv_drm_get_dev_info_ioctl(struct drm_device *dev, void *data, struct drm_file *filep) { struct nv_drm_device *nv_dev = to_nv_device(dev); struct drm_nvidia_get_dev_info_params *params = data; if (dev->primary == NULL) { return -ENOENT; } params->gpu_id = nv_dev->gpu_info.gpu_id; params->primary_index = dev->primary->index; #if defined(NV_DRM_ATOMIC_MODESET_AVAILABLE) params->generic_page_kind = nv_dev->genericPageKind; params->page_kind_generation = nv_dev->pageKindGeneration; params->sector_layout = nv_dev->sectorLayout; #else params->generic_page_kind = 0; params->page_kind_generation = 0; params->sector_layout = 0; #endif return 0; } static int nv_drm_dmabuf_supported_ioctl(struct drm_device *dev, void *data, struct drm_file *filep) { /* check the pDevice since this only gets set if modeset = 1 * which is a requirement for the dma_buf extension to work */ struct nv_drm_device *nv_dev = to_nv_device(dev); return nv_dev->pDevice ? 0 : -EINVAL; } static int nv_drm_get_client_capability_ioctl(struct drm_device *dev, void *data, struct drm_file *filep) { struct drm_nvidia_get_client_capability_params *params = data; switch (params->capability) { #if defined(DRM_CLIENT_CAP_STEREO_3D) case DRM_CLIENT_CAP_STEREO_3D: params->value = filep->stereo_allowed; break; #endif #if defined(DRM_CLIENT_CAP_UNIVERSAL_PLANES) case DRM_CLIENT_CAP_UNIVERSAL_PLANES: params->value = filep->universal_planes; break; #endif #if defined(DRM_CLIENT_CAP_ATOMIC) case DRM_CLIENT_CAP_ATOMIC: params->value = filep->atomic; break; #endif default: return -EINVAL; } return 0; } #if defined(NV_DRM_ATOMIC_MODESET_AVAILABLE) static bool nv_drm_connector_is_dpy_id(struct drm_connector *connector, NvU32 dpyId) { struct nv_drm_connector *nv_connector = to_nv_connector(connector); return nv_connector->nv_detected_encoder && nv_connector->nv_detected_encoder->hDisplay == dpyId; } static int nv_drm_get_dpy_id_for_connector_id_ioctl(struct drm_device *dev, void *data, struct drm_file *filep) { struct drm_nvidia_get_dpy_id_for_connector_id_params *params = data; // Importantly, drm_connector_lookup (with filep) will only return the // connector if we are master, a lessee with the connector, or not master at // all. It will return NULL if we are a lessee with other connectors. struct drm_connector *connector = nv_drm_connector_lookup(dev, filep, params->connectorId); struct nv_drm_connector *nv_connector; int ret = 0; if (!connector) { return -EINVAL; } nv_connector = to_nv_connector(connector); if (!nv_connector) { ret = -EINVAL; goto done; } if (!nv_connector->nv_detected_encoder) { ret = -EINVAL; goto done; } params->dpyId = nv_connector->nv_detected_encoder->hDisplay; done: nv_drm_connector_put(connector); return ret; } static int nv_drm_get_connector_id_for_dpy_id_ioctl(struct drm_device *dev, void *data, struct drm_file *filep) { struct drm_nvidia_get_connector_id_for_dpy_id_params *params = data; struct drm_connector *connector; int ret = -EINVAL; #if defined(NV_DRM_CONNECTOR_LIST_ITER_PRESENT) struct drm_connector_list_iter conn_iter; nv_drm_connector_list_iter_begin(dev, &conn_iter); #endif /* Lookup for existing connector with same dpyId */ nv_drm_for_each_connector(connector, &conn_iter, dev) { if (nv_drm_connector_is_dpy_id(connector, params->dpyId)) { params->connectorId = connector->base.id; ret = 0; break; } } #if defined(NV_DRM_CONNECTOR_LIST_ITER_PRESENT) nv_drm_connector_list_iter_end(&conn_iter); #endif return ret; } static NvU32 nv_drm_get_head_bit_from_connector(struct drm_connector *connector) { struct nv_drm_connector *nv_connector = to_nv_connector(connector); if (connector->state && connector->state->crtc) { struct nv_drm_crtc *nv_crtc = to_nv_crtc(connector->state->crtc); return NVBIT(nv_crtc->head); } else if (nv_connector->nv_detected_encoder && nv_connector->nv_detected_encoder->base.crtc) { struct nv_drm_crtc *nv_crtc = to_nv_crtc(nv_connector->nv_detected_encoder->base.crtc); return NVBIT(nv_crtc->head); } return 0; } static int nv_drm_grant_permission_ioctl(struct drm_device *dev, void *data, struct drm_file *filep) { struct drm_nvidia_grant_permissions_params *params = data; struct nv_drm_device *nv_dev = to_nv_device(dev); struct nv_drm_connector *target_nv_connector = NULL; struct nv_drm_crtc *target_nv_crtc = NULL; struct drm_connector *connector, *target_connector = NULL; struct drm_crtc *crtc; NvU32 head = 0, freeHeadBits, targetHeadBit, possible_crtcs; int ret = 0; #if defined(NV_DRM_CONNECTOR_LIST_ITER_PRESENT) struct drm_connector_list_iter conn_iter; #endif #if NV_DRM_MODESET_LOCK_ALL_END_ARGUMENT_COUNT == 3 struct drm_modeset_acquire_ctx ctx; DRM_MODESET_LOCK_ALL_BEGIN(dev, ctx, DRM_MODESET_ACQUIRE_INTERRUPTIBLE, ret); #else mutex_lock(&dev->mode_config.mutex); #endif /* Get the connector for the dpyId. */ #if defined(NV_DRM_CONNECTOR_LIST_ITER_PRESENT) nv_drm_connector_list_iter_begin(dev, &conn_iter); #endif nv_drm_for_each_connector(connector, &conn_iter, dev) { if (nv_drm_connector_is_dpy_id(connector, params->dpyId)) { target_connector = nv_drm_connector_lookup(dev, filep, connector->base.id); break; } } #if defined(NV_DRM_CONNECTOR_LIST_ITER_PRESENT) nv_drm_connector_list_iter_end(&conn_iter); #endif // Importantly, drm_connector_lookup/drm_crtc_find (with filep) will only // return the object if we are master, a lessee with the object, or not // master at all. It will return NULL if we are a lessee with other objects. if (!target_connector) { ret = -EINVAL; goto done; } target_nv_connector = to_nv_connector(target_connector); possible_crtcs = target_nv_connector->nv_detected_encoder->base.possible_crtcs; /* Target connector must not be previously granted. */ if (target_nv_connector->modeset_permission_filep) { ret = -EINVAL; goto done; } /* Add all heads that are owned and not already granted. */ freeHeadBits = 0; nv_drm_for_each_crtc(crtc, dev) { struct nv_drm_crtc *nv_crtc = to_nv_crtc(crtc); if (nv_drm_crtc_find(dev, filep, crtc->base.id) && !nv_crtc->modeset_permission_filep && (drm_crtc_mask(crtc) & possible_crtcs)) { freeHeadBits |= NVBIT(nv_crtc->head); } } targetHeadBit = nv_drm_get_head_bit_from_connector(target_connector); if (targetHeadBit & freeHeadBits) { /* If a crtc is already being used by this connector, use it. */ freeHeadBits = targetHeadBit; } else { /* Otherwise, remove heads that are in use by other connectors. */ #if defined(NV_DRM_CONNECTOR_LIST_ITER_PRESENT) nv_drm_connector_list_iter_begin(dev, &conn_iter); #endif nv_drm_for_each_connector(connector, &conn_iter, dev) { freeHeadBits &= ~nv_drm_get_head_bit_from_connector(connector); } #if defined(NV_DRM_CONNECTOR_LIST_ITER_PRESENT) nv_drm_connector_list_iter_end(&conn_iter); #endif } /* Fail if no heads are available. */ if (!freeHeadBits) { ret = -EINVAL; goto done; } /* * Loop through the crtc again and find a matching head. * Record the filep that is using the crtc and the connector. */ nv_drm_for_each_crtc(crtc, dev) { struct nv_drm_crtc *nv_crtc = to_nv_crtc(crtc); if (freeHeadBits & NVBIT(nv_crtc->head)) { target_nv_crtc = nv_crtc; head = nv_crtc->head; break; } } if (!nvKms->grantPermissions(params->fd, nv_dev->pDevice, head, params->dpyId)) { ret = -EINVAL; goto done; } target_nv_connector->modeset_permission_crtc = target_nv_crtc; target_nv_connector->modeset_permission_filep = filep; target_nv_crtc->modeset_permission_filep = filep; done: if (target_connector) { nv_drm_connector_put(target_connector); } #if NV_DRM_MODESET_LOCK_ALL_END_ARGUMENT_COUNT == 3 DRM_MODESET_LOCK_ALL_END(dev, ctx, ret); #else mutex_unlock(&dev->mode_config.mutex); #endif return ret; } static bool nv_drm_revoke_connector(struct nv_drm_device *nv_dev, struct nv_drm_connector *nv_connector) { bool ret = true; if (nv_connector->modeset_permission_crtc) { if (nv_connector->nv_detected_encoder) { ret = nvKms->revokePermissions( nv_dev->pDevice, nv_connector->modeset_permission_crtc->head, nv_connector->nv_detected_encoder->hDisplay); } nv_connector->modeset_permission_crtc->modeset_permission_filep = NULL; nv_connector->modeset_permission_crtc = NULL; } nv_connector->modeset_permission_filep = NULL; return ret; } static int nv_drm_revoke_permission(struct drm_device *dev, struct drm_file *filep, NvU32 dpyId) { struct drm_connector *connector; struct drm_crtc *crtc; int ret = 0; #if defined(NV_DRM_CONNECTOR_LIST_ITER_PRESENT) struct drm_connector_list_iter conn_iter; #endif #if NV_DRM_MODESET_LOCK_ALL_END_ARGUMENT_COUNT == 3 struct drm_modeset_acquire_ctx ctx; DRM_MODESET_LOCK_ALL_BEGIN(dev, ctx, DRM_MODESET_ACQUIRE_INTERRUPTIBLE, ret); #else mutex_lock(&dev->mode_config.mutex); #endif /* * If dpyId is set, only revoke those specific resources. Otherwise, * it is from closing the file so revoke all resources for that filep. */ #if defined(NV_DRM_CONNECTOR_LIST_ITER_PRESENT) nv_drm_connector_list_iter_begin(dev, &conn_iter); #endif nv_drm_for_each_connector(connector, &conn_iter, dev) { struct nv_drm_connector *nv_connector = to_nv_connector(connector); if (nv_connector->modeset_permission_filep == filep && (!dpyId || nv_drm_connector_is_dpy_id(connector, dpyId))) { if (!nv_drm_connector_revoke_permissions(dev, nv_connector)) { ret = -EINVAL; // Continue trying to revoke as much as possible. } } } #if defined(NV_DRM_CONNECTOR_LIST_ITER_PRESENT) nv_drm_connector_list_iter_end(&conn_iter); #endif nv_drm_for_each_crtc(crtc, dev) { struct nv_drm_crtc *nv_crtc = to_nv_crtc(crtc); if (nv_crtc->modeset_permission_filep == filep && !dpyId) { nv_crtc->modeset_permission_filep = NULL; } } #if NV_DRM_MODESET_LOCK_ALL_END_ARGUMENT_COUNT == 3 DRM_MODESET_LOCK_ALL_END(dev, ctx, ret); #else mutex_unlock(&dev->mode_config.mutex); #endif return ret; } static int nv_drm_revoke_permission_ioctl(struct drm_device *dev, void *data, struct drm_file *filep) { struct drm_nvidia_revoke_permissions_params *params = data; if (!params->dpyId) { return -EINVAL; } return nv_drm_revoke_permission(dev, filep, params->dpyId); } static void nv_drm_postclose(struct drm_device *dev, struct drm_file *filep) { /* * Some systems like android can reach here without initializing the * device, so check for that. */ if (dev->mode_config.num_crtc > 0 && dev->mode_config.crtc_list.next != NULL && dev->mode_config.crtc_list.prev != NULL && dev->mode_config.num_connector > 0 && dev->mode_config.connector_list.next != NULL && dev->mode_config.connector_list.prev != NULL) { nv_drm_revoke_permission(dev, filep, 0); } } #endif /* NV_DRM_ATOMIC_MODESET_AVAILABLE */ #if defined(NV_DRM_MASTER_HAS_LEASES) static struct drm_master *nv_drm_find_lessee(struct drm_master *master, int lessee_id) { int object; void *entry; while (master->lessor != NULL) { master = master->lessor; } idr_for_each_entry(&master->lessee_idr, entry, object) { if (object == lessee_id) { return entry; } } return NULL; } static void nv_drm_get_revoked_objects(struct drm_device *dev, struct drm_file *filep, unsigned int cmd, unsigned long arg, int **objects, int *objects_count) { unsigned int ioc_size; struct drm_mode_revoke_lease revoke_lease; struct drm_master *lessor, *lessee; void *entry; int *objs; int obj, obj_count, obj_i; ioc_size = _IOC_SIZE(cmd); if (ioc_size > sizeof(revoke_lease)) { return; } if (copy_from_user(&revoke_lease, (void __user *)arg, ioc_size) != 0) { return; } lessor = nv_drm_file_get_master(filep); if (lessor == NULL) { return; } mutex_lock(&dev->mode_config.idr_mutex); lessee = nv_drm_find_lessee(lessor, revoke_lease.lessee_id); if (lessee == NULL) { goto done; } obj_count = 0; idr_for_each_entry(&lessee->leases, entry, obj) { ++obj_count; } if (obj_count == 0) { goto done; } objs = nv_drm_calloc(obj_count, sizeof(int)); if (objs == NULL) { goto done; } obj_i = 0; idr_for_each_entry(&lessee->leases, entry, obj) { objs[obj_i++] = obj; } *objects = objs; *objects_count = obj_count; done: mutex_unlock(&dev->mode_config.idr_mutex); drm_master_put(&lessor); } static bool nv_drm_is_in_objects(int object, int *objects, int objects_count) { int i; for (i = 0; i < objects_count; ++i) { if (objects[i] == object) { return true; } } return false; } static void nv_drm_finish_revoking_objects(struct drm_device *dev, struct drm_file *filep, int *objects, int objects_count) { struct drm_connector *connector; struct drm_crtc *crtc; #if defined(NV_DRM_CONNECTOR_LIST_ITER_PRESENT) struct drm_connector_list_iter conn_iter; #endif #if NV_DRM_MODESET_LOCK_ALL_END_ARGUMENT_COUNT == 3 int ret = 0; struct drm_modeset_acquire_ctx ctx; DRM_MODESET_LOCK_ALL_BEGIN(dev, ctx, DRM_MODESET_ACQUIRE_INTERRUPTIBLE, ret); #else mutex_lock(&dev->mode_config.mutex); #endif #if defined(NV_DRM_CONNECTOR_LIST_ITER_PRESENT) nv_drm_connector_list_iter_begin(dev, &conn_iter); #endif nv_drm_for_each_connector(connector, &conn_iter, dev) { struct nv_drm_connector *nv_connector = to_nv_connector(connector); if (nv_connector->modeset_permission_filep && nv_drm_is_in_objects(connector->base.id, objects, objects_count)) { nv_drm_connector_revoke_permissions(dev, nv_connector); } } #if defined(NV_DRM_CONNECTOR_LIST_ITER_PRESENT) nv_drm_connector_list_iter_end(&conn_iter); #endif nv_drm_for_each_crtc(crtc, dev) { struct nv_drm_crtc *nv_crtc = to_nv_crtc(crtc); if (nv_crtc->modeset_permission_filep && nv_drm_is_in_objects(crtc->base.id, objects, objects_count)) { nv_crtc->modeset_permission_filep = NULL; } } #if NV_DRM_MODESET_LOCK_ALL_END_ARGUMENT_COUNT == 3 DRM_MODESET_LOCK_ALL_END(dev, ctx, ret); #else mutex_unlock(&dev->mode_config.mutex); #endif } #endif /* NV_DRM_MASTER_HAS_LEASES */ #if defined(NV_DRM_BUS_PRESENT) #if defined(NV_DRM_BUS_HAS_GET_IRQ) static int nv_drm_bus_get_irq(struct drm_device *dev) { return 0; } #endif #if defined(NV_DRM_BUS_HAS_GET_NAME) static const char *nv_drm_bus_get_name(struct drm_device *dev) { return "nvidia-drm"; } #endif static struct drm_bus nv_drm_bus = { #if defined(NV_DRM_BUS_HAS_BUS_TYPE) .bus_type = DRIVER_BUS_PCI, #endif #if defined(NV_DRM_BUS_HAS_GET_IRQ) .get_irq = nv_drm_bus_get_irq, #endif #if defined(NV_DRM_BUS_HAS_GET_NAME) .get_name = nv_drm_bus_get_name, #endif .set_busid = nv_drm_pci_set_busid, }; #endif /* NV_DRM_BUS_PRESENT */ /* * Wrapper around drm_ioctl to hook in to upstream ioctl. * * Currently used to add additional handling to REVOKE_LEASE. */ static long nv_drm_ioctl(struct file *filp, unsigned int cmd, unsigned long arg) { long retcode; #if defined(NV_DRM_MASTER_HAS_LEASES) struct drm_file *file_priv = filp->private_data; struct drm_device *dev = file_priv->minor->dev; int *objects = NULL; int objects_count = 0; if (cmd == DRM_IOCTL_MODE_REVOKE_LEASE) { // Save the revoked objects before revoking. nv_drm_get_revoked_objects(dev, file_priv, cmd, arg, &objects, &objects_count); } #endif retcode = drm_ioctl(filp, cmd, arg); #if defined(NV_DRM_MASTER_HAS_LEASES) if (cmd == DRM_IOCTL_MODE_REVOKE_LEASE && objects) { if (retcode == 0) { // If revoking was successful, finish revoking the objects. nv_drm_finish_revoking_objects(dev, file_priv, objects, objects_count); } nv_drm_free(objects); } #endif return retcode; } static const struct file_operations nv_drm_fops = { .owner = THIS_MODULE, .open = drm_open, .release = drm_release, .unlocked_ioctl = nv_drm_ioctl, #if defined(CONFIG_COMPAT) .compat_ioctl = drm_compat_ioctl, #endif #if defined(NV_DRM_ATOMIC_MODESET_AVAILABLE) .mmap = nv_drm_mmap, #endif .poll = drm_poll, .read = drm_read, .llseek = noop_llseek, #if defined(NV_FILE_OPERATIONS_FOP_UNSIGNED_OFFSET_PRESENT) .fop_flags = FOP_UNSIGNED_OFFSET, #endif }; static const struct drm_ioctl_desc nv_drm_ioctls[] = { #if defined(NV_DRM_ATOMIC_MODESET_AVAILABLE) DRM_IOCTL_DEF_DRV(NVIDIA_GEM_IMPORT_NVKMS_MEMORY, nv_drm_gem_import_nvkms_memory_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), #endif /* NV_DRM_ATOMIC_MODESET_AVAILABLE */ DRM_IOCTL_DEF_DRV(NVIDIA_GEM_IMPORT_USERSPACE_MEMORY, nv_drm_gem_import_userspace_memory_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), DRM_IOCTL_DEF_DRV(NVIDIA_GEM_MAP_OFFSET, nv_drm_gem_map_offset_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), DRM_IOCTL_DEF_DRV(NVIDIA_GET_DEV_INFO, nv_drm_get_dev_info_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), #if defined(NV_DRM_FENCE_AVAILABLE) DRM_IOCTL_DEF_DRV(NVIDIA_FENCE_SUPPORTED, nv_drm_fence_supported_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), DRM_IOCTL_DEF_DRV(NVIDIA_PRIME_FENCE_CONTEXT_CREATE, nv_drm_prime_fence_context_create_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), DRM_IOCTL_DEF_DRV(NVIDIA_GEM_PRIME_FENCE_ATTACH, nv_drm_gem_prime_fence_attach_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), #endif /* * DRM_UNLOCKED is implicit for all non-legacy DRM driver IOCTLs since Linux * v4.10 commit fa5386459f06 "drm: Used DRM_LEGACY for all legacy functions" * (Linux v4.4 commit ea487835e887 "drm: Enforce unlocked ioctl operation * for kms driver ioctls" previously did it only for drivers that set the * DRM_MODESET flag), so this will race with SET_CLIENT_CAP. Linux v4.11 * commit dcf727ab5d17 "drm: setclientcap doesn't need the drm BKL" also * removed locking from SET_CLIENT_CAP so there is no use attempting to lock * manually. The latter commit acknowledges that this can expose userspace * to inconsistent behavior when racing with itself, but accepts that risk. */ DRM_IOCTL_DEF_DRV(NVIDIA_GET_CLIENT_CAPABILITY, nv_drm_get_client_capability_ioctl, 0), #if defined(NV_DRM_ATOMIC_MODESET_AVAILABLE) DRM_IOCTL_DEF_DRV(NVIDIA_GET_CRTC_CRC32, nv_drm_get_crtc_crc32_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), DRM_IOCTL_DEF_DRV(NVIDIA_GET_CRTC_CRC32_V2, nv_drm_get_crtc_crc32_v2_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), DRM_IOCTL_DEF_DRV(NVIDIA_GEM_EXPORT_NVKMS_MEMORY, nv_drm_gem_export_nvkms_memory_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), DRM_IOCTL_DEF_DRV(NVIDIA_GEM_ALLOC_NVKMS_MEMORY, nv_drm_gem_alloc_nvkms_memory_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), DRM_IOCTL_DEF_DRV(NVIDIA_GEM_EXPORT_DMABUF_MEMORY, nv_drm_gem_export_dmabuf_memory_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), DRM_IOCTL_DEF_DRV(NVIDIA_GEM_IDENTIFY_OBJECT, nv_drm_gem_identify_object_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), DRM_IOCTL_DEF_DRV(NVIDIA_DMABUF_SUPPORTED, nv_drm_dmabuf_supported_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), DRM_IOCTL_DEF_DRV(NVIDIA_GET_DPY_ID_FOR_CONNECTOR_ID, nv_drm_get_dpy_id_for_connector_id_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), DRM_IOCTL_DEF_DRV(NVIDIA_GET_CONNECTOR_ID_FOR_DPY_ID, nv_drm_get_connector_id_for_dpy_id_ioctl, DRM_RENDER_ALLOW|DRM_UNLOCKED), DRM_IOCTL_DEF_DRV(NVIDIA_GRANT_PERMISSIONS, nv_drm_grant_permission_ioctl, DRM_UNLOCKED|DRM_MASTER), DRM_IOCTL_DEF_DRV(NVIDIA_REVOKE_PERMISSIONS, nv_drm_revoke_permission_ioctl, DRM_UNLOCKED|DRM_MASTER), #endif /* NV_DRM_ATOMIC_MODESET_AVAILABLE */ }; static struct drm_driver nv_drm_driver = { .driver_features = #if defined(NV_DRM_DRIVER_PRIME_FLAG_PRESENT) DRIVER_PRIME | #endif DRIVER_GEM | DRIVER_RENDER, #if defined(NV_DRM_DRIVER_HAS_GEM_FREE_OBJECT) .gem_free_object = nv_drm_gem_free, #endif .ioctls = nv_drm_ioctls, .num_ioctls = ARRAY_SIZE(nv_drm_ioctls), /* * linux-next commit 71a7974ac701 ("drm/prime: Unexport helpers for fd/handle * conversion") unexports drm_gem_prime_handle_to_fd() and * drm_gem_prime_fd_to_handle(). * * Prior linux-next commit 6b85aa68d9d5 ("drm: Enable PRIME import/export for * all drivers") made these helpers the default when .prime_handle_to_fd / * .prime_fd_to_handle are unspecified, so it's fine to just skip specifying * them if the helpers aren't present. */ #if NV_IS_EXPORT_SYMBOL_PRESENT_drm_gem_prime_handle_to_fd .prime_handle_to_fd = drm_gem_prime_handle_to_fd, #endif #if NV_IS_EXPORT_SYMBOL_PRESENT_drm_gem_prime_fd_to_handle .prime_fd_to_handle = drm_gem_prime_fd_to_handle, #endif .gem_prime_import = nv_drm_gem_prime_import, .gem_prime_import_sg_table = nv_drm_gem_prime_import_sg_table, #if defined(NV_DRM_DRIVER_HAS_GEM_PRIME_CALLBACKS) .gem_prime_export = drm_gem_prime_export, .gem_prime_get_sg_table = nv_drm_gem_prime_get_sg_table, .gem_prime_vmap = nv_drm_gem_prime_vmap, .gem_prime_vunmap = nv_drm_gem_prime_vunmap, .gem_vm_ops = &nv_drm_gem_vma_ops, #endif #if defined(NV_DRM_DRIVER_HAS_GEM_PRIME_RES_OBJ) .gem_prime_res_obj = nv_drm_gem_prime_res_obj, #endif #if defined(NV_DRM_DRIVER_HAS_SET_BUSID) .set_busid = nv_drm_pci_set_busid, #endif .load = nv_drm_load, .unload = nv_drm_unload, #if defined(NV_DRM_ATOMIC_MODESET_AVAILABLE) .postclose = nv_drm_postclose, #endif .fops = &nv_drm_fops, #if defined(NV_DRM_BUS_PRESENT) .bus = &nv_drm_bus, #endif .name = "nvidia-drm", .desc = "NVIDIA DRM driver", .date = "20160202", #if defined(NV_DRM_DRIVER_HAS_DEVICE_LIST) .device_list = LIST_HEAD_INIT(nv_drm_driver.device_list), #elif defined(NV_DRM_DRIVER_HAS_LEGACY_DEV_LIST) .legacy_dev_list = LIST_HEAD_INIT(nv_drm_driver.legacy_dev_list), #endif }; /* * Update the global nv_drm_driver for the intended features. * * It defaults to PRIME-only, but is upgraded to atomic modeset if the * kernel supports atomic modeset and the 'modeset' kernel module * parameter is true. */ static void nv_drm_update_drm_driver_features(void) { #if defined(NV_DRM_ATOMIC_MODESET_AVAILABLE) if (!nv_drm_modeset_module_param) { return; } nv_drm_driver.driver_features |= DRIVER_MODESET | DRIVER_ATOMIC; nv_drm_driver.master_set = nv_drm_master_set; nv_drm_driver.master_drop = nv_drm_master_drop; nv_drm_driver.dumb_create = nv_drm_dumb_create; nv_drm_driver.dumb_map_offset = nv_drm_dumb_map_offset; #if defined(NV_DRM_DRIVER_HAS_DUMB_DESTROY) nv_drm_driver.dumb_destroy = nv_drm_dumb_destroy; #endif /* NV_DRM_DRIVER_HAS_DUMB_DESTROY */ #endif /* NV_DRM_ATOMIC_MODESET_AVAILABLE */ } /* * Helper function for allocate/register DRM device for given NVIDIA GPU ID. */ static void nv_drm_register_drm_device(const nv_gpu_info_t *gpu_info) { struct nv_drm_device *nv_dev = NULL; struct drm_device *dev = NULL; struct device *device = gpu_info->os_device_ptr; DRM_DEBUG( "Registering device for NVIDIA GPU ID 0x08%x", gpu_info->gpu_id); /* Allocate NVIDIA-DRM device */ nv_dev = nv_drm_calloc(1, sizeof(*nv_dev)); if (nv_dev == NULL) { NV_DRM_LOG_ERR( "Failed to allocate memory for NVIDIA-DRM device object"); return; } nv_dev->gpu_info = *gpu_info; #if defined(NV_DRM_ATOMIC_MODESET_AVAILABLE) mutex_init(&nv_dev->lock); #endif /* Allocate DRM device */ dev = drm_dev_alloc(&nv_drm_driver, device); if (dev == NULL) { NV_DRM_DEV_LOG_ERR(nv_dev, "Failed to allocate device"); goto failed_drm_alloc; } dev->dev_private = nv_dev; nv_dev->dev = dev; #if defined(NV_DRM_DEVICE_HAS_PDEV) if (device->bus == &pci_bus_type) { dev->pdev = to_pci_dev(device); } #endif /* Register DRM device to DRM sub-system */ if (drm_dev_register(dev, 0) != 0) { NV_DRM_DEV_LOG_ERR(nv_dev, "Failed to register device"); goto failed_drm_register; } /* Add NVIDIA-DRM device into list */ nv_dev->next = dev_list; dev_list = nv_dev; return; /* Success */ failed_drm_register: nv_drm_dev_free(dev); failed_drm_alloc: nv_drm_free(nv_dev); } /* * Enumerate NVIDIA GPUs and allocate/register DRM device for each of them. */ int nv_drm_probe_devices(void) { nv_gpu_info_t *gpu_info = NULL; NvU32 gpu_count = 0; NvU32 i; int ret = 0; nv_drm_update_drm_driver_features(); /* Enumerate NVIDIA GPUs */ gpu_info = nv_drm_calloc(NV_MAX_GPUS, sizeof(*gpu_info)); if (gpu_info == NULL) { ret = -ENOMEM; NV_DRM_LOG_ERR("Failed to allocate gpu ids arrays"); goto done; } gpu_count = nvKms->enumerateGpus(gpu_info); if (gpu_count == 0) { NV_DRM_LOG_INFO("Not found NVIDIA GPUs"); goto done; } WARN_ON(gpu_count > NV_MAX_GPUS); /* Register DRM device for each NVIDIA GPU */ for (i = 0; i < gpu_count; i++) { nv_drm_register_drm_device(&gpu_info[i]); } done: nv_drm_free(gpu_info); return ret; } /* * Unregister all NVIDIA DRM devices. */ void nv_drm_remove_devices(void) { while (dev_list != NULL) { struct nv_drm_device *next = dev_list->next; drm_dev_unregister(dev_list->dev); nv_drm_dev_free(dev_list->dev); nv_drm_free(dev_list); dev_list = next; } } #endif /* NV_DRM_AVAILABLE */