1caab277bSThomas Gleixner // SPDX-License-Identifier: GPL-2.0-only 2667ce33eSRob Clark /* 3667ce33eSRob Clark * Copyright (C) 2016 Red Hat 4667ce33eSRob Clark * Author: Rob Clark <robdclark@gmail.com> 5667ce33eSRob Clark */ 6667ce33eSRob Clark 7667ce33eSRob Clark #include "msm_drv.h" 8667ce33eSRob Clark #include "msm_gem.h" 9667ce33eSRob Clark #include "msm_mmu.h" 10667ce33eSRob Clark 11ee546cd3SJordan Crouse static void 12ee546cd3SJordan Crouse msm_gem_address_space_destroy(struct kref *kref) 13ee546cd3SJordan Crouse { 14ee546cd3SJordan Crouse struct msm_gem_address_space *aspace = container_of(kref, 15ee546cd3SJordan Crouse struct msm_gem_address_space, kref); 16ee546cd3SJordan Crouse 17ee546cd3SJordan Crouse drm_mm_takedown(&aspace->mm); 18ee546cd3SJordan Crouse if (aspace->mmu) 19ee546cd3SJordan Crouse aspace->mmu->funcs->destroy(aspace->mmu); 20ee546cd3SJordan Crouse kfree(aspace); 21ee546cd3SJordan Crouse } 22ee546cd3SJordan Crouse 23ee546cd3SJordan Crouse 24ee546cd3SJordan Crouse void msm_gem_address_space_put(struct msm_gem_address_space *aspace) 25ee546cd3SJordan Crouse { 26ee546cd3SJordan Crouse if (aspace) 27ee546cd3SJordan Crouse kref_put(&aspace->kref, msm_gem_address_space_destroy); 28ee546cd3SJordan Crouse } 29ee546cd3SJordan Crouse 30933415e2SJordan Crouse struct msm_gem_address_space * 31933415e2SJordan Crouse msm_gem_address_space_get(struct msm_gem_address_space *aspace) 32933415e2SJordan Crouse { 33933415e2SJordan Crouse if (!IS_ERR_OR_NULL(aspace)) 34933415e2SJordan Crouse kref_get(&aspace->kref); 35933415e2SJordan Crouse 36933415e2SJordan Crouse return aspace; 37933415e2SJordan Crouse } 38933415e2SJordan Crouse 397ad0e8cfSJordan Crouse /* Actually unmap memory for the vma */ 407ad0e8cfSJordan Crouse void msm_gem_purge_vma(struct msm_gem_address_space *aspace, 4170dc51b4SJordan Crouse struct msm_gem_vma *vma) 42667ce33eSRob Clark { 437ad0e8cfSJordan Crouse unsigned size = vma->node.size << PAGE_SHIFT; 447ad0e8cfSJordan Crouse 457ad0e8cfSJordan Crouse /* Print a message if we try to purge a vma in use */ 467ad0e8cfSJordan Crouse if (WARN_ON(vma->inuse > 0)) 47667ce33eSRob Clark return; 48667ce33eSRob Clark 497ad0e8cfSJordan Crouse /* Don't do anything if the memory isn't mapped */ 507ad0e8cfSJordan Crouse if (!vma->mapped) 517ad0e8cfSJordan Crouse return; 527ad0e8cfSJordan Crouse 537ad0e8cfSJordan Crouse if (aspace->mmu) 5470dc51b4SJordan Crouse aspace->mmu->funcs->unmap(aspace->mmu, vma->iova, size); 557ad0e8cfSJordan Crouse 567ad0e8cfSJordan Crouse vma->mapped = false; 57667ce33eSRob Clark } 58667ce33eSRob Clark 597ad0e8cfSJordan Crouse /* Remove reference counts for the mapping */ 607ad0e8cfSJordan Crouse void msm_gem_unmap_vma(struct msm_gem_address_space *aspace, 617ad0e8cfSJordan Crouse struct msm_gem_vma *vma) 627ad0e8cfSJordan Crouse { 637ad0e8cfSJordan Crouse if (!WARN_ON(!vma->iova)) 647ad0e8cfSJordan Crouse vma->inuse--; 65667ce33eSRob Clark } 66667ce33eSRob Clark 67667ce33eSRob Clark int 68667ce33eSRob Clark msm_gem_map_vma(struct msm_gem_address_space *aspace, 69bbc2cd07SRob Clark struct msm_gem_vma *vma, int prot, 70bbc2cd07SRob Clark struct sg_table *sgt, int npages) 71667ce33eSRob Clark { 72c0ee9794SJordan Crouse unsigned size = npages << PAGE_SHIFT; 73c0ee9794SJordan Crouse int ret = 0; 74667ce33eSRob Clark 75c0ee9794SJordan Crouse if (WARN_ON(!vma->iova)) 76c0ee9794SJordan Crouse return -EINVAL; 77c0ee9794SJordan Crouse 787ad0e8cfSJordan Crouse /* Increase the usage counter */ 797ad0e8cfSJordan Crouse vma->inuse++; 807ad0e8cfSJordan Crouse 81c0ee9794SJordan Crouse if (vma->mapped) 82667ce33eSRob Clark return 0; 83c0ee9794SJordan Crouse 84c0ee9794SJordan Crouse vma->mapped = true; 85c0ee9794SJordan Crouse 867603df38SLuca Weiss if (aspace && aspace->mmu) 87c0ee9794SJordan Crouse ret = aspace->mmu->funcs->map(aspace->mmu, vma->iova, sgt, 88bbc2cd07SRob Clark size, prot); 89c0ee9794SJordan Crouse 90c0ee9794SJordan Crouse if (ret) 91c0ee9794SJordan Crouse vma->mapped = false; 92c0ee9794SJordan Crouse 93c0ee9794SJordan Crouse return ret; 940e08270aSSushmita Susheelendra } 95667ce33eSRob Clark 967ad0e8cfSJordan Crouse /* Close an iova. Warn if it is still in use */ 977ad0e8cfSJordan Crouse void msm_gem_close_vma(struct msm_gem_address_space *aspace, 987ad0e8cfSJordan Crouse struct msm_gem_vma *vma) 997ad0e8cfSJordan Crouse { 1007ad0e8cfSJordan Crouse if (WARN_ON(vma->inuse > 0 || vma->mapped)) 1017ad0e8cfSJordan Crouse return; 1027ad0e8cfSJordan Crouse 1037ad0e8cfSJordan Crouse spin_lock(&aspace->lock); 1047ad0e8cfSJordan Crouse if (vma->iova) 1057ad0e8cfSJordan Crouse drm_mm_remove_node(&vma->node); 1067ad0e8cfSJordan Crouse spin_unlock(&aspace->lock); 1077ad0e8cfSJordan Crouse 1087ad0e8cfSJordan Crouse vma->iova = 0; 1097ad0e8cfSJordan Crouse 1107ad0e8cfSJordan Crouse msm_gem_address_space_put(aspace); 1117ad0e8cfSJordan Crouse } 1127ad0e8cfSJordan Crouse 113c0ee9794SJordan Crouse /* Initialize a new vma and allocate an iova for it */ 114c0ee9794SJordan Crouse int msm_gem_init_vma(struct msm_gem_address_space *aspace, 115d3b8877eSJonathan Marek struct msm_gem_vma *vma, int npages, 116d3b8877eSJonathan Marek u64 range_start, u64 range_end) 117c0ee9794SJordan Crouse { 118c0ee9794SJordan Crouse int ret; 119c0ee9794SJordan Crouse 120c0ee9794SJordan Crouse if (WARN_ON(vma->iova)) 121c0ee9794SJordan Crouse return -EBUSY; 122c0ee9794SJordan Crouse 123c0ee9794SJordan Crouse spin_lock(&aspace->lock); 124d3b8877eSJonathan Marek ret = drm_mm_insert_node_in_range(&aspace->mm, &vma->node, npages, 0, 125d3b8877eSJonathan Marek 0, range_start, range_end, 0); 1260e08270aSSushmita Susheelendra spin_unlock(&aspace->lock); 1270e08270aSSushmita Susheelendra 128667ce33eSRob Clark if (ret) 129667ce33eSRob Clark return ret; 130667ce33eSRob Clark 131667ce33eSRob Clark vma->iova = vma->node.start << PAGE_SHIFT; 132c0ee9794SJordan Crouse vma->mapped = false; 133667ce33eSRob Clark 134ee546cd3SJordan Crouse kref_get(&aspace->kref); 135667ce33eSRob Clark 136c0ee9794SJordan Crouse return 0; 137667ce33eSRob Clark } 138667ce33eSRob Clark 139667ce33eSRob Clark struct msm_gem_address_space * 140ccac7ce3SJordan Crouse msm_gem_address_space_create(struct msm_mmu *mmu, const char *name, 141ccac7ce3SJordan Crouse u64 va_start, u64 size) 142667ce33eSRob Clark { 143667ce33eSRob Clark struct msm_gem_address_space *aspace; 144ccac7ce3SJordan Crouse 145ccac7ce3SJordan Crouse if (IS_ERR(mmu)) 146ccac7ce3SJordan Crouse return ERR_CAST(mmu); 147667ce33eSRob Clark 148667ce33eSRob Clark aspace = kzalloc(sizeof(*aspace), GFP_KERNEL); 149667ce33eSRob Clark if (!aspace) 150667ce33eSRob Clark return ERR_PTR(-ENOMEM); 151667ce33eSRob Clark 1520e08270aSSushmita Susheelendra spin_lock_init(&aspace->lock); 153667ce33eSRob Clark aspace->name = name; 154ccac7ce3SJordan Crouse aspace->mmu = mmu; 155667ce33eSRob Clark 156ccac7ce3SJordan Crouse drm_mm_init(&aspace->mm, va_start >> PAGE_SHIFT, size >> PAGE_SHIFT); 157c2052a4eSJonathan Marek 158c2052a4eSJonathan Marek kref_init(&aspace->kref); 159c2052a4eSJonathan Marek 160c2052a4eSJonathan Marek return aspace; 161c2052a4eSJonathan Marek } 162