Linux Audio

Check our new training course

Loading...
v4.6
  1/**************************************************************************
  2 *
  3 * Copyright (c) 2006-2009 VMware, Inc., Palo Alto, CA., USA
  4 * All Rights Reserved.
  5 *
  6 * Permission is hereby granted, free of charge, to any person obtaining a
  7 * copy of this software and associated documentation files (the
  8 * "Software"), to deal in the Software without restriction, including
  9 * without limitation the rights to use, copy, modify, merge, publish,
 10 * distribute, sub license, and/or sell copies of the Software, and to
 11 * permit persons to whom the Software is furnished to do so, subject to
 12 * the following conditions:
 13 *
 14 * The above copyright notice and this permission notice (including the
 15 * next paragraph) shall be included in all copies or substantial portions
 16 * of the Software.
 17 *
 18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 19 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 20 * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
 21 * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM,
 22 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
 23 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
 24 * USE OR OTHER DEALINGS IN THE SOFTWARE.
 25 *
 26 **************************************************************************/
 27/*
 28 * Authors: Thomas Hellstrom <thellstrom-at-vmware-dot-com>
 29 */
 30
 31#define pr_fmt(fmt) "[TTM] " fmt
 32
 33#include <linux/sched.h>
 34#include <linux/highmem.h>
 35#include <linux/pagemap.h>
 36#include <linux/shmem_fs.h>
 37#include <linux/file.h>
 38#include <linux/swap.h>
 39#include <linux/slab.h>
 40#include <linux/export.h>
 41#include <drm/drm_cache.h>
 42#include <drm/drm_mem_util.h>
 43#include <drm/ttm/ttm_module.h>
 44#include <drm/ttm/ttm_bo_driver.h>
 45#include <drm/ttm/ttm_placement.h>
 46#include <drm/ttm/ttm_page_alloc.h>
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 47
 48/**
 49 * Allocates storage for pointers to the pages that back the ttm.
 50 */
 51static void ttm_tt_alloc_page_directory(struct ttm_tt *ttm)
 52{
 53	ttm->pages = drm_calloc_large(ttm->num_pages, sizeof(void*));
 
 
 
 
 54}
 55
 56static void ttm_dma_tt_alloc_page_directory(struct ttm_dma_tt *ttm)
 57{
 58	ttm->ttm.pages = drm_calloc_large(ttm->ttm.num_pages,
 59					  sizeof(*ttm->ttm.pages) +
 60					  sizeof(*ttm->dma_address) +
 61					  sizeof(*ttm->cpu_address));
 62	ttm->cpu_address = (void *) (ttm->ttm.pages + ttm->ttm.num_pages);
 63	ttm->dma_address = (void *) (ttm->cpu_address + ttm->ttm.num_pages);
 
 
 
 
 
 
 
 
 
 
 
 
 64}
 65
 66#ifdef CONFIG_X86
 67static inline int ttm_tt_set_page_caching(struct page *p,
 68					  enum ttm_caching_state c_old,
 69					  enum ttm_caching_state c_new)
 70{
 71	int ret = 0;
 72
 73	if (PageHighMem(p))
 74		return 0;
 75
 76	if (c_old != tt_cached) {
 77		/* p isn't in the default caching state, set it to
 78		 * writeback first to free its current memtype. */
 79
 80		ret = set_pages_wb(p, 1);
 81		if (ret)
 82			return ret;
 83	}
 84
 85	if (c_new == tt_wc)
 86		ret = set_memory_wc((unsigned long) page_address(p), 1);
 87	else if (c_new == tt_uncached)
 88		ret = set_pages_uc(p, 1);
 89
 90	return ret;
 91}
 92#else /* CONFIG_X86 */
 93static inline int ttm_tt_set_page_caching(struct page *p,
 94					  enum ttm_caching_state c_old,
 95					  enum ttm_caching_state c_new)
 96{
 97	return 0;
 98}
 99#endif /* CONFIG_X86 */
100
101/*
102 * Change caching policy for the linear kernel map
103 * for range of pages in a ttm.
104 */
105
106static int ttm_tt_set_caching(struct ttm_tt *ttm,
107			      enum ttm_caching_state c_state)
108{
109	int i, j;
110	struct page *cur_page;
111	int ret;
112
113	if (ttm->caching_state == c_state)
114		return 0;
115
116	if (ttm->state == tt_unpopulated) {
117		/* Change caching but don't populate */
118		ttm->caching_state = c_state;
119		return 0;
120	}
121
122	if (ttm->caching_state == tt_cached)
123		drm_clflush_pages(ttm->pages, ttm->num_pages);
124
125	for (i = 0; i < ttm->num_pages; ++i) {
126		cur_page = ttm->pages[i];
127		if (likely(cur_page != NULL)) {
128			ret = ttm_tt_set_page_caching(cur_page,
129						      ttm->caching_state,
130						      c_state);
131			if (unlikely(ret != 0))
132				goto out_err;
133		}
134	}
135
136	ttm->caching_state = c_state;
137
138	return 0;
139
140out_err:
141	for (j = 0; j < i; ++j) {
142		cur_page = ttm->pages[j];
143		if (likely(cur_page != NULL)) {
144			(void)ttm_tt_set_page_caching(cur_page, c_state,
145						      ttm->caching_state);
146		}
147	}
148
149	return ret;
150}
151
152int ttm_tt_set_placement_caching(struct ttm_tt *ttm, uint32_t placement)
153{
154	enum ttm_caching_state state;
155
156	if (placement & TTM_PL_FLAG_WC)
157		state = tt_wc;
158	else if (placement & TTM_PL_FLAG_UNCACHED)
159		state = tt_uncached;
160	else
161		state = tt_cached;
162
163	return ttm_tt_set_caching(ttm, state);
164}
165EXPORT_SYMBOL(ttm_tt_set_placement_caching);
166
167void ttm_tt_destroy(struct ttm_tt *ttm)
168{
169	if (unlikely(ttm == NULL))
170		return;
171
172	if (ttm->state == tt_bound) {
173		ttm_tt_unbind(ttm);
174	}
175
176	if (ttm->state == tt_unbound)
177		ttm_tt_unpopulate(ttm);
178
179	if (!(ttm->page_flags & TTM_PAGE_FLAG_PERSISTENT_SWAP) &&
180	    ttm->swap_storage)
181		fput(ttm->swap_storage);
182
183	ttm->swap_storage = NULL;
184	ttm->func->destroy(ttm);
185}
186
187int ttm_tt_init(struct ttm_tt *ttm, struct ttm_bo_device *bdev,
188		unsigned long size, uint32_t page_flags,
189		struct page *dummy_read_page)
190{
191	ttm->bdev = bdev;
192	ttm->glob = bdev->glob;
193	ttm->num_pages = (size + PAGE_SIZE - 1) >> PAGE_SHIFT;
194	ttm->caching_state = tt_cached;
195	ttm->page_flags = page_flags;
196	ttm->dummy_read_page = dummy_read_page;
197	ttm->state = tt_unpopulated;
198	ttm->swap_storage = NULL;
 
 
199
200	ttm_tt_alloc_page_directory(ttm);
201	if (!ttm->pages) {
 
 
 
 
202		ttm_tt_destroy(ttm);
203		pr_err("Failed allocating page table\n");
204		return -ENOMEM;
205	}
206	return 0;
207}
208EXPORT_SYMBOL(ttm_tt_init);
209
210void ttm_tt_fini(struct ttm_tt *ttm)
211{
212	drm_free_large(ttm->pages);
213	ttm->pages = NULL;
214}
215EXPORT_SYMBOL(ttm_tt_fini);
216
217int ttm_dma_tt_init(struct ttm_dma_tt *ttm_dma, struct ttm_bo_device *bdev,
218		unsigned long size, uint32_t page_flags,
219		struct page *dummy_read_page)
220{
221	struct ttm_tt *ttm = &ttm_dma->ttm;
222
223	ttm->bdev = bdev;
224	ttm->glob = bdev->glob;
225	ttm->num_pages = (size + PAGE_SIZE - 1) >> PAGE_SHIFT;
226	ttm->caching_state = tt_cached;
227	ttm->page_flags = page_flags;
228	ttm->dummy_read_page = dummy_read_page;
229	ttm->state = tt_unpopulated;
230	ttm->swap_storage = NULL;
231
232	INIT_LIST_HEAD(&ttm_dma->pages_list);
233	ttm_dma_tt_alloc_page_directory(ttm_dma);
234	if (!ttm->pages) {
235		ttm_tt_destroy(ttm);
236		pr_err("Failed allocating page table\n");
237		return -ENOMEM;
238	}
239	return 0;
240}
241EXPORT_SYMBOL(ttm_dma_tt_init);
242
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
243void ttm_dma_tt_fini(struct ttm_dma_tt *ttm_dma)
244{
245	struct ttm_tt *ttm = &ttm_dma->ttm;
246
247	drm_free_large(ttm->pages);
 
 
 
248	ttm->pages = NULL;
249	ttm_dma->cpu_address = NULL;
250	ttm_dma->dma_address = NULL;
251}
252EXPORT_SYMBOL(ttm_dma_tt_fini);
253
254void ttm_tt_unbind(struct ttm_tt *ttm)
255{
256	int ret;
257
258	if (ttm->state == tt_bound) {
259		ret = ttm->func->unbind(ttm);
260		BUG_ON(ret);
261		ttm->state = tt_unbound;
262	}
263}
264
265int ttm_tt_bind(struct ttm_tt *ttm, struct ttm_mem_reg *bo_mem)
 
266{
267	int ret = 0;
268
269	if (!ttm)
270		return -EINVAL;
271
272	if (ttm->state == tt_bound)
273		return 0;
274
275	ret = ttm->bdev->driver->ttm_tt_populate(ttm);
276	if (ret)
277		return ret;
278
279	ret = ttm->func->bind(ttm, bo_mem);
280	if (unlikely(ret != 0))
281		return ret;
282
283	ttm->state = tt_bound;
284
285	return 0;
286}
287EXPORT_SYMBOL(ttm_tt_bind);
288
289int ttm_tt_swapin(struct ttm_tt *ttm)
290{
291	struct address_space *swap_space;
292	struct file *swap_storage;
293	struct page *from_page;
294	struct page *to_page;
295	int i;
296	int ret = -ENOMEM;
297
298	swap_storage = ttm->swap_storage;
299	BUG_ON(swap_storage == NULL);
300
301	swap_space = file_inode(swap_storage)->i_mapping;
302
303	for (i = 0; i < ttm->num_pages; ++i) {
304		from_page = shmem_read_mapping_page(swap_space, i);
 
 
 
 
305		if (IS_ERR(from_page)) {
306			ret = PTR_ERR(from_page);
307			goto out_err;
308		}
309		to_page = ttm->pages[i];
310		if (unlikely(to_page == NULL))
311			goto out_err;
312
313		copy_highpage(to_page, from_page);
314		put_page(from_page);
315	}
316
317	if (!(ttm->page_flags & TTM_PAGE_FLAG_PERSISTENT_SWAP))
318		fput(swap_storage);
319	ttm->swap_storage = NULL;
320	ttm->page_flags &= ~TTM_PAGE_FLAG_SWAPPED;
321
322	return 0;
323out_err:
324	return ret;
325}
326
327int ttm_tt_swapout(struct ttm_tt *ttm, struct file *persistent_swap_storage)
328{
329	struct address_space *swap_space;
330	struct file *swap_storage;
331	struct page *from_page;
332	struct page *to_page;
333	int i;
334	int ret = -ENOMEM;
335
336	BUG_ON(ttm->state != tt_unbound && ttm->state != tt_unpopulated);
337	BUG_ON(ttm->caching_state != tt_cached);
338
339	if (!persistent_swap_storage) {
340		swap_storage = shmem_file_setup("ttm swap",
341						ttm->num_pages << PAGE_SHIFT,
342						0);
343		if (IS_ERR(swap_storage)) {
344			pr_err("Failed allocating swap storage\n");
345			return PTR_ERR(swap_storage);
346		}
347	} else
348		swap_storage = persistent_swap_storage;
 
349
350	swap_space = file_inode(swap_storage)->i_mapping;
351
352	for (i = 0; i < ttm->num_pages; ++i) {
 
 
 
 
353		from_page = ttm->pages[i];
354		if (unlikely(from_page == NULL))
355			continue;
356		to_page = shmem_read_mapping_page(swap_space, i);
 
357		if (IS_ERR(to_page)) {
358			ret = PTR_ERR(to_page);
359			goto out_err;
360		}
361		copy_highpage(to_page, from_page);
362		set_page_dirty(to_page);
363		mark_page_accessed(to_page);
364		put_page(to_page);
365	}
366
367	ttm_tt_unpopulate(ttm);
368	ttm->swap_storage = swap_storage;
369	ttm->page_flags |= TTM_PAGE_FLAG_SWAPPED;
370	if (persistent_swap_storage)
371		ttm->page_flags |= TTM_PAGE_FLAG_PERSISTENT_SWAP;
372
373	return 0;
374out_err:
375	if (!persistent_swap_storage)
376		fput(swap_storage);
377
378	return ret;
379}
380
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
381static void ttm_tt_clear_mapping(struct ttm_tt *ttm)
382{
383	pgoff_t i;
384	struct page **page = ttm->pages;
385
386	if (ttm->page_flags & TTM_PAGE_FLAG_SG)
387		return;
388
389	for (i = 0; i < ttm->num_pages; ++i) {
390		(*page)->mapping = NULL;
391		(*page++)->index = 0;
392	}
393}
394
395void ttm_tt_unpopulate(struct ttm_tt *ttm)
396{
397	if (ttm->state == tt_unpopulated)
398		return;
399
400	ttm_tt_clear_mapping(ttm);
401	ttm->bdev->driver->ttm_tt_unpopulate(ttm);
 
 
 
402}
v4.17
  1/**************************************************************************
  2 *
  3 * Copyright (c) 2006-2009 VMware, Inc., Palo Alto, CA., USA
  4 * All Rights Reserved.
  5 *
  6 * Permission is hereby granted, free of charge, to any person obtaining a
  7 * copy of this software and associated documentation files (the
  8 * "Software"), to deal in the Software without restriction, including
  9 * without limitation the rights to use, copy, modify, merge, publish,
 10 * distribute, sub license, and/or sell copies of the Software, and to
 11 * permit persons to whom the Software is furnished to do so, subject to
 12 * the following conditions:
 13 *
 14 * The above copyright notice and this permission notice (including the
 15 * next paragraph) shall be included in all copies or substantial portions
 16 * of the Software.
 17 *
 18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 19 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 20 * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
 21 * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM,
 22 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
 23 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
 24 * USE OR OTHER DEALINGS IN THE SOFTWARE.
 25 *
 26 **************************************************************************/
 27/*
 28 * Authors: Thomas Hellstrom <thellstrom-at-vmware-dot-com>
 29 */
 30
 31#define pr_fmt(fmt) "[TTM] " fmt
 32
 33#include <linux/sched.h>
 
 34#include <linux/pagemap.h>
 35#include <linux/shmem_fs.h>
 36#include <linux/file.h>
 
 
 
 37#include <drm/drm_cache.h>
 
 
 38#include <drm/ttm/ttm_bo_driver.h>
 
 39#include <drm/ttm/ttm_page_alloc.h>
 40#ifdef CONFIG_X86
 41#include <asm/set_memory.h>
 42#endif
 43
 44/**
 45 * Allocates a ttm structure for the given BO.
 46 */
 47int ttm_tt_create(struct ttm_buffer_object *bo, bool zero_alloc)
 48{
 49	struct ttm_bo_device *bdev = bo->bdev;
 50	uint32_t page_flags = 0;
 51
 52	reservation_object_assert_held(bo->resv);
 53
 54	if (bdev->need_dma32)
 55		page_flags |= TTM_PAGE_FLAG_DMA32;
 56
 57	if (bdev->no_retry)
 58		page_flags |= TTM_PAGE_FLAG_NO_RETRY;
 59
 60	switch (bo->type) {
 61	case ttm_bo_type_device:
 62		if (zero_alloc)
 63			page_flags |= TTM_PAGE_FLAG_ZERO_ALLOC;
 64		break;
 65	case ttm_bo_type_kernel:
 66		break;
 67	case ttm_bo_type_sg:
 68		page_flags |= TTM_PAGE_FLAG_SG;
 69		break;
 70	default:
 71		bo->ttm = NULL;
 72		pr_err("Illegal buffer object type\n");
 73		return -EINVAL;
 74	}
 75
 76	bo->ttm = bdev->driver->ttm_tt_create(bo, page_flags);
 77	if (unlikely(bo->ttm == NULL))
 78		return -ENOMEM;
 79
 80	return 0;
 81}
 82
 83/**
 84 * Allocates storage for pointers to the pages that back the ttm.
 85 */
 86static int ttm_tt_alloc_page_directory(struct ttm_tt *ttm)
 87{
 88	ttm->pages = kvmalloc_array(ttm->num_pages, sizeof(void*),
 89			GFP_KERNEL | __GFP_ZERO);
 90	if (!ttm->pages)
 91		return -ENOMEM;
 92	return 0;
 93}
 94
 95static int ttm_dma_tt_alloc_page_directory(struct ttm_dma_tt *ttm)
 96{
 97	ttm->ttm.pages = kvmalloc_array(ttm->ttm.num_pages,
 98					  sizeof(*ttm->ttm.pages) +
 99					  sizeof(*ttm->dma_address),
100					  GFP_KERNEL | __GFP_ZERO);
101	if (!ttm->ttm.pages)
102		return -ENOMEM;
103	ttm->dma_address = (void *) (ttm->ttm.pages + ttm->ttm.num_pages);
104	return 0;
105}
106
107static int ttm_sg_tt_alloc_page_directory(struct ttm_dma_tt *ttm)
108{
109	ttm->dma_address = kvmalloc_array(ttm->ttm.num_pages,
110					  sizeof(*ttm->dma_address),
111					  GFP_KERNEL | __GFP_ZERO);
112	if (!ttm->dma_address)
113		return -ENOMEM;
114	return 0;
115}
116
117#ifdef CONFIG_X86
118static inline int ttm_tt_set_page_caching(struct page *p,
119					  enum ttm_caching_state c_old,
120					  enum ttm_caching_state c_new)
121{
122	int ret = 0;
123
124	if (PageHighMem(p))
125		return 0;
126
127	if (c_old != tt_cached) {
128		/* p isn't in the default caching state, set it to
129		 * writeback first to free its current memtype. */
130
131		ret = set_pages_wb(p, 1);
132		if (ret)
133			return ret;
134	}
135
136	if (c_new == tt_wc)
137		ret = set_memory_wc((unsigned long) page_address(p), 1);
138	else if (c_new == tt_uncached)
139		ret = set_pages_uc(p, 1);
140
141	return ret;
142}
143#else /* CONFIG_X86 */
144static inline int ttm_tt_set_page_caching(struct page *p,
145					  enum ttm_caching_state c_old,
146					  enum ttm_caching_state c_new)
147{
148	return 0;
149}
150#endif /* CONFIG_X86 */
151
152/*
153 * Change caching policy for the linear kernel map
154 * for range of pages in a ttm.
155 */
156
157static int ttm_tt_set_caching(struct ttm_tt *ttm,
158			      enum ttm_caching_state c_state)
159{
160	int i, j;
161	struct page *cur_page;
162	int ret;
163
164	if (ttm->caching_state == c_state)
165		return 0;
166
167	if (ttm->state == tt_unpopulated) {
168		/* Change caching but don't populate */
169		ttm->caching_state = c_state;
170		return 0;
171	}
172
173	if (ttm->caching_state == tt_cached)
174		drm_clflush_pages(ttm->pages, ttm->num_pages);
175
176	for (i = 0; i < ttm->num_pages; ++i) {
177		cur_page = ttm->pages[i];
178		if (likely(cur_page != NULL)) {
179			ret = ttm_tt_set_page_caching(cur_page,
180						      ttm->caching_state,
181						      c_state);
182			if (unlikely(ret != 0))
183				goto out_err;
184		}
185	}
186
187	ttm->caching_state = c_state;
188
189	return 0;
190
191out_err:
192	for (j = 0; j < i; ++j) {
193		cur_page = ttm->pages[j];
194		if (likely(cur_page != NULL)) {
195			(void)ttm_tt_set_page_caching(cur_page, c_state,
196						      ttm->caching_state);
197		}
198	}
199
200	return ret;
201}
202
203int ttm_tt_set_placement_caching(struct ttm_tt *ttm, uint32_t placement)
204{
205	enum ttm_caching_state state;
206
207	if (placement & TTM_PL_FLAG_WC)
208		state = tt_wc;
209	else if (placement & TTM_PL_FLAG_UNCACHED)
210		state = tt_uncached;
211	else
212		state = tt_cached;
213
214	return ttm_tt_set_caching(ttm, state);
215}
216EXPORT_SYMBOL(ttm_tt_set_placement_caching);
217
218void ttm_tt_destroy(struct ttm_tt *ttm)
219{
220	if (ttm == NULL)
221		return;
222
223	ttm_tt_unbind(ttm);
 
 
224
225	if (ttm->state == tt_unbound)
226		ttm_tt_unpopulate(ttm);
227
228	if (!(ttm->page_flags & TTM_PAGE_FLAG_PERSISTENT_SWAP) &&
229	    ttm->swap_storage)
230		fput(ttm->swap_storage);
231
232	ttm->swap_storage = NULL;
233	ttm->func->destroy(ttm);
234}
235
236void ttm_tt_init_fields(struct ttm_tt *ttm, struct ttm_buffer_object *bo,
237			uint32_t page_flags)
238{
239	ttm->bdev = bo->bdev;
240	ttm->num_pages = bo->num_pages;
 
 
241	ttm->caching_state = tt_cached;
242	ttm->page_flags = page_flags;
 
243	ttm->state = tt_unpopulated;
244	ttm->swap_storage = NULL;
245	ttm->sg = bo->sg;
246}
247
248int ttm_tt_init(struct ttm_tt *ttm, struct ttm_buffer_object *bo,
249		uint32_t page_flags)
250{
251	ttm_tt_init_fields(ttm, bo, page_flags);
252
253	if (ttm_tt_alloc_page_directory(ttm)) {
254		ttm_tt_destroy(ttm);
255		pr_err("Failed allocating page table\n");
256		return -ENOMEM;
257	}
258	return 0;
259}
260EXPORT_SYMBOL(ttm_tt_init);
261
262void ttm_tt_fini(struct ttm_tt *ttm)
263{
264	kvfree(ttm->pages);
265	ttm->pages = NULL;
266}
267EXPORT_SYMBOL(ttm_tt_fini);
268
269int ttm_dma_tt_init(struct ttm_dma_tt *ttm_dma, struct ttm_buffer_object *bo,
270		    uint32_t page_flags)
 
271{
272	struct ttm_tt *ttm = &ttm_dma->ttm;
273
274	ttm_tt_init_fields(ttm, bo, page_flags);
 
 
 
 
 
 
 
275
276	INIT_LIST_HEAD(&ttm_dma->pages_list);
277	if (ttm_dma_tt_alloc_page_directory(ttm_dma)) {
 
278		ttm_tt_destroy(ttm);
279		pr_err("Failed allocating page table\n");
280		return -ENOMEM;
281	}
282	return 0;
283}
284EXPORT_SYMBOL(ttm_dma_tt_init);
285
286int ttm_sg_tt_init(struct ttm_dma_tt *ttm_dma, struct ttm_buffer_object *bo,
287		   uint32_t page_flags)
288{
289	struct ttm_tt *ttm = &ttm_dma->ttm;
290	int ret;
291
292	ttm_tt_init_fields(ttm, bo, page_flags);
293
294	INIT_LIST_HEAD(&ttm_dma->pages_list);
295	if (page_flags & TTM_PAGE_FLAG_SG)
296		ret = ttm_sg_tt_alloc_page_directory(ttm_dma);
297	else
298		ret = ttm_dma_tt_alloc_page_directory(ttm_dma);
299	if (ret) {
300		ttm_tt_destroy(ttm);
301		pr_err("Failed allocating page table\n");
302		return -ENOMEM;
303	}
304	return 0;
305}
306EXPORT_SYMBOL(ttm_sg_tt_init);
307
308void ttm_dma_tt_fini(struct ttm_dma_tt *ttm_dma)
309{
310	struct ttm_tt *ttm = &ttm_dma->ttm;
311
312	if (ttm->pages)
313		kvfree(ttm->pages);
314	else
315		kvfree(ttm_dma->dma_address);
316	ttm->pages = NULL;
 
317	ttm_dma->dma_address = NULL;
318}
319EXPORT_SYMBOL(ttm_dma_tt_fini);
320
321void ttm_tt_unbind(struct ttm_tt *ttm)
322{
323	int ret;
324
325	if (ttm->state == tt_bound) {
326		ret = ttm->func->unbind(ttm);
327		BUG_ON(ret);
328		ttm->state = tt_unbound;
329	}
330}
331
332int ttm_tt_bind(struct ttm_tt *ttm, struct ttm_mem_reg *bo_mem,
333		struct ttm_operation_ctx *ctx)
334{
335	int ret = 0;
336
337	if (!ttm)
338		return -EINVAL;
339
340	if (ttm->state == tt_bound)
341		return 0;
342
343	ret = ttm_tt_populate(ttm, ctx);
344	if (ret)
345		return ret;
346
347	ret = ttm->func->bind(ttm, bo_mem);
348	if (unlikely(ret != 0))
349		return ret;
350
351	ttm->state = tt_bound;
352
353	return 0;
354}
355EXPORT_SYMBOL(ttm_tt_bind);
356
357int ttm_tt_swapin(struct ttm_tt *ttm)
358{
359	struct address_space *swap_space;
360	struct file *swap_storage;
361	struct page *from_page;
362	struct page *to_page;
363	int i;
364	int ret = -ENOMEM;
365
366	swap_storage = ttm->swap_storage;
367	BUG_ON(swap_storage == NULL);
368
369	swap_space = swap_storage->f_mapping;
370
371	for (i = 0; i < ttm->num_pages; ++i) {
372		gfp_t gfp_mask = mapping_gfp_mask(swap_space);
373
374		gfp_mask |= (ttm->page_flags & TTM_PAGE_FLAG_NO_RETRY ? __GFP_RETRY_MAYFAIL : 0);
375		from_page = shmem_read_mapping_page_gfp(swap_space, i, gfp_mask);
376
377		if (IS_ERR(from_page)) {
378			ret = PTR_ERR(from_page);
379			goto out_err;
380		}
381		to_page = ttm->pages[i];
382		if (unlikely(to_page == NULL))
383			goto out_err;
384
385		copy_highpage(to_page, from_page);
386		put_page(from_page);
387	}
388
389	if (!(ttm->page_flags & TTM_PAGE_FLAG_PERSISTENT_SWAP))
390		fput(swap_storage);
391	ttm->swap_storage = NULL;
392	ttm->page_flags &= ~TTM_PAGE_FLAG_SWAPPED;
393
394	return 0;
395out_err:
396	return ret;
397}
398
399int ttm_tt_swapout(struct ttm_tt *ttm, struct file *persistent_swap_storage)
400{
401	struct address_space *swap_space;
402	struct file *swap_storage;
403	struct page *from_page;
404	struct page *to_page;
405	int i;
406	int ret = -ENOMEM;
407
408	BUG_ON(ttm->state != tt_unbound && ttm->state != tt_unpopulated);
409	BUG_ON(ttm->caching_state != tt_cached);
410
411	if (!persistent_swap_storage) {
412		swap_storage = shmem_file_setup("ttm swap",
413						ttm->num_pages << PAGE_SHIFT,
414						0);
415		if (IS_ERR(swap_storage)) {
416			pr_err("Failed allocating swap storage\n");
417			return PTR_ERR(swap_storage);
418		}
419	} else {
420		swap_storage = persistent_swap_storage;
421	}
422
423	swap_space = swap_storage->f_mapping;
424
425	for (i = 0; i < ttm->num_pages; ++i) {
426		gfp_t gfp_mask = mapping_gfp_mask(swap_space);
427
428		gfp_mask |= (ttm->page_flags & TTM_PAGE_FLAG_NO_RETRY ? __GFP_RETRY_MAYFAIL : 0);
429
430		from_page = ttm->pages[i];
431		if (unlikely(from_page == NULL))
432			continue;
433
434		to_page = shmem_read_mapping_page_gfp(swap_space, i, gfp_mask);
435		if (IS_ERR(to_page)) {
436			ret = PTR_ERR(to_page);
437			goto out_err;
438		}
439		copy_highpage(to_page, from_page);
440		set_page_dirty(to_page);
441		mark_page_accessed(to_page);
442		put_page(to_page);
443	}
444
445	ttm_tt_unpopulate(ttm);
446	ttm->swap_storage = swap_storage;
447	ttm->page_flags |= TTM_PAGE_FLAG_SWAPPED;
448	if (persistent_swap_storage)
449		ttm->page_flags |= TTM_PAGE_FLAG_PERSISTENT_SWAP;
450
451	return 0;
452out_err:
453	if (!persistent_swap_storage)
454		fput(swap_storage);
455
456	return ret;
457}
458
459static void ttm_tt_add_mapping(struct ttm_tt *ttm)
460{
461	pgoff_t i;
462
463	if (ttm->page_flags & TTM_PAGE_FLAG_SG)
464		return;
465
466	for (i = 0; i < ttm->num_pages; ++i)
467		ttm->pages[i]->mapping = ttm->bdev->dev_mapping;
468}
469
470int ttm_tt_populate(struct ttm_tt *ttm, struct ttm_operation_ctx *ctx)
471{
472	int ret;
473
474	if (ttm->state != tt_unpopulated)
475		return 0;
476
477	if (ttm->bdev->driver->ttm_tt_populate)
478		ret = ttm->bdev->driver->ttm_tt_populate(ttm, ctx);
479	else
480		ret = ttm_pool_populate(ttm, ctx);
481	if (!ret)
482		ttm_tt_add_mapping(ttm);
483	return ret;
484}
485
486static void ttm_tt_clear_mapping(struct ttm_tt *ttm)
487{
488	pgoff_t i;
489	struct page **page = ttm->pages;
490
491	if (ttm->page_flags & TTM_PAGE_FLAG_SG)
492		return;
493
494	for (i = 0; i < ttm->num_pages; ++i) {
495		(*page)->mapping = NULL;
496		(*page++)->index = 0;
497	}
498}
499
500void ttm_tt_unpopulate(struct ttm_tt *ttm)
501{
502	if (ttm->state == tt_unpopulated)
503		return;
504
505	ttm_tt_clear_mapping(ttm);
506	if (ttm->bdev->driver->ttm_tt_unpopulate)
507		ttm->bdev->driver->ttm_tt_unpopulate(ttm);
508	else
509		ttm_pool_unpopulate(ttm);
510}