1/*
2 * Copyright 2008 Jerome Glisse.
3 * All Rights Reserved.
4 *
5 * Permission is hereby granted, free of charge, to any person obtaining a
6 * copy of this software and associated documentation files (the "Software"),
7 * to deal in the Software without restriction, including without limitation
8 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
9 * and/or sell copies of the Software, and to permit persons to whom the
10 * Software is furnished to do so, subject to the following conditions:
11 *
12 * The above copyright notice and this permission notice (including the next
13 * paragraph) shall be included in all copies or substantial portions of the
14 * Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
19 * PRECISION INSIGHT AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
20 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
21 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
22 * DEALINGS IN THE SOFTWARE.
23 *
24 * Authors:
25 *    Jerome Glisse <glisse@freedesktop.org>
26 */
27#include "drmP.h"
28#include "radeon_drm.h"
29#include "radeon_reg.h"
30#include "radeon.h"
31
32void r100_cs_dump_packet(struct radeon_cs_parser *p,
33			 struct radeon_cs_packet *pkt);
34
35int radeon_cs_parser_relocs(struct radeon_cs_parser *p)
36{
37	struct drm_device *ddev = p->rdev->ddev;
38	struct radeon_cs_chunk *chunk;
39	unsigned i, j;
40	bool duplicate;
41
42	if (p->chunk_relocs_idx == -1) {
43		return 0;
44	}
45	chunk = &p->chunks[p->chunk_relocs_idx];
46	/* FIXME: we assume that each relocs use 4 dwords */
47	p->nrelocs = chunk->length_dw / 4;
48	p->relocs_ptr = kcalloc(p->nrelocs, sizeof(void *), GFP_KERNEL);
49	if (p->relocs_ptr == NULL) {
50		return -ENOMEM;
51	}
52	p->relocs = kcalloc(p->nrelocs, sizeof(struct radeon_cs_reloc), GFP_KERNEL);
53	if (p->relocs == NULL) {
54		return -ENOMEM;
55	}
56	for (i = 0; i < p->nrelocs; i++) {
57		struct drm_radeon_cs_reloc *r;
58
59		duplicate = false;
60		r = (struct drm_radeon_cs_reloc *)&chunk->kdata[i*4];
61		for (j = 0; j < i; j++) {
62			if (r->handle == p->relocs[j].handle) {
63				p->relocs_ptr[i] = &p->relocs[j];
64				duplicate = true;
65				break;
66			}
67		}
68		if (!duplicate) {
69			p->relocs[i].gobj = drm_gem_object_lookup(ddev,
70								  p->filp,
71								  r->handle);
72			if (p->relocs[i].gobj == NULL) {
73				DRM_ERROR("gem object lookup failed 0x%x\n",
74					  r->handle);
75				return -ENOENT;
76			}
77			p->relocs_ptr[i] = &p->relocs[i];
78			p->relocs[i].robj = gem_to_radeon_bo(p->relocs[i].gobj);
79			p->relocs[i].lobj.bo = p->relocs[i].robj;
80			p->relocs[i].lobj.wdomain = r->write_domain;
81			p->relocs[i].lobj.rdomain = r->read_domains;
82			p->relocs[i].lobj.tv.bo = &p->relocs[i].robj->tbo;
83			p->relocs[i].handle = r->handle;
84			p->relocs[i].flags = r->flags;
85			radeon_bo_list_add_object(&p->relocs[i].lobj,
86						  &p->validated);
87
88			if (p->relocs[i].robj->tbo.sync_obj && !(r->flags & RADEON_RELOC_DONT_SYNC)) {
89				struct radeon_fence *fence = p->relocs[i].robj->tbo.sync_obj;
90				if (!radeon_fence_signaled(fence)) {
91					p->sync_to_ring[fence->ring] = true;
92				}
93			}
94		} else
95			p->relocs[i].handle = 0;
96	}
97	return radeon_bo_list_validate(&p->validated);
98}
99
100static int radeon_cs_get_ring(struct radeon_cs_parser *p, u32 ring, s32 priority)
101{
102	p->priority = priority;
103
104	switch (ring) {
105	default:
106		DRM_ERROR("unknown ring id: %d\n", ring);
107		return -EINVAL;
108	case RADEON_CS_RING_GFX:
109		p->ring = RADEON_RING_TYPE_GFX_INDEX;
110		break;
111	case RADEON_CS_RING_COMPUTE:
112		/* for now */
113		p->ring = RADEON_RING_TYPE_GFX_INDEX;
114		break;
115	}
116	return 0;
117}
118
119static int radeon_cs_sync_rings(struct radeon_cs_parser *p)
120{
121	int i, r;
122
123	for (i = 0; i < RADEON_NUM_RINGS; ++i) {
124		/* no need to sync to our own or unused rings */
125		if (i == p->ring || !p->sync_to_ring[i] || !p->rdev->ring[i].ready)
126			continue;
127
128		if (!p->ib->fence->semaphore) {
129			r = radeon_semaphore_create(p->rdev, &p->ib->fence->semaphore);
130			if (r)
131				return r;
132		}
133
134		r = radeon_ring_lock(p->rdev, &p->rdev->ring[i], 3);
135		if (r)
136			return r;
137		radeon_semaphore_emit_signal(p->rdev, i, p->ib->fence->semaphore);
138		radeon_ring_unlock_commit(p->rdev, &p->rdev->ring[i]);
139
140		r = radeon_ring_lock(p->rdev, &p->rdev->ring[p->ring], 3);
141		if (r)
142			return r;
143		radeon_semaphore_emit_wait(p->rdev, p->ring, p->ib->fence->semaphore);
144		radeon_ring_unlock_commit(p->rdev, &p->rdev->ring[p->ring]);
145	}
146	return 0;
147}
148
149int radeon_cs_parser_init(struct radeon_cs_parser *p, void *data)
150{
151	struct drm_radeon_cs *cs = data;
152	uint64_t *chunk_array_ptr;
153	unsigned size, i;
154	u32 ring = RADEON_CS_RING_GFX;
155	s32 priority = 0;
156
157	if (!cs->num_chunks) {
158		return 0;
159	}
160	/* get chunks */
161	INIT_LIST_HEAD(&p->validated);
162	p->idx = 0;
163	p->chunk_ib_idx = -1;
164	p->chunk_relocs_idx = -1;
165	p->chunk_flags_idx = -1;
166	p->chunks_array = kcalloc(cs->num_chunks, sizeof(uint64_t), GFP_KERNEL);
167	if (p->chunks_array == NULL) {
168		return -ENOMEM;
169	}
170	chunk_array_ptr = (uint64_t *)(unsigned long)(cs->chunks);
171	if (DRM_COPY_FROM_USER(p->chunks_array, chunk_array_ptr,
172			       sizeof(uint64_t)*cs->num_chunks)) {
173		return -EFAULT;
174	}
175	p->cs_flags = 0;
176	p->nchunks = cs->num_chunks;
177	p->chunks = kcalloc(p->nchunks, sizeof(struct radeon_cs_chunk), GFP_KERNEL);
178	if (p->chunks == NULL) {
179		return -ENOMEM;
180	}
181	for (i = 0; i < p->nchunks; i++) {
182		struct drm_radeon_cs_chunk __user **chunk_ptr = NULL;
183		struct drm_radeon_cs_chunk user_chunk;
184		uint32_t __user *cdata;
185
186		chunk_ptr = (void __user*)(unsigned long)p->chunks_array[i];
187		if (DRM_COPY_FROM_USER(&user_chunk, chunk_ptr,
188				       sizeof(struct drm_radeon_cs_chunk))) {
189			return -EFAULT;
190		}
191		p->chunks[i].length_dw = user_chunk.length_dw;
192		p->chunks[i].kdata = NULL;
193		p->chunks[i].chunk_id = user_chunk.chunk_id;
194
195		if (p->chunks[i].chunk_id == RADEON_CHUNK_ID_RELOCS) {
196			p->chunk_relocs_idx = i;
197		}
198		if (p->chunks[i].chunk_id == RADEON_CHUNK_ID_IB) {
199			p->chunk_ib_idx = i;
200			/* zero length IB isn't useful */
201			if (p->chunks[i].length_dw == 0)
202				return -EINVAL;
203		}
204		if (p->chunks[i].chunk_id == RADEON_CHUNK_ID_FLAGS) {
205			p->chunk_flags_idx = i;
206			/* zero length flags aren't useful */
207			if (p->chunks[i].length_dw == 0)
208				return -EINVAL;
209		}
210
211		p->chunks[i].length_dw = user_chunk.length_dw;
212		p->chunks[i].user_ptr = (void __user *)(unsigned long)user_chunk.chunk_data;
213
214		cdata = (uint32_t *)(unsigned long)user_chunk.chunk_data;
215		if ((p->chunks[i].chunk_id == RADEON_CHUNK_ID_RELOCS) ||
216		    (p->chunks[i].chunk_id == RADEON_CHUNK_ID_FLAGS)) {
217			size = p->chunks[i].length_dw * sizeof(uint32_t);
218			p->chunks[i].kdata = kmalloc(size, GFP_KERNEL);
219			if (p->chunks[i].kdata == NULL) {
220				return -ENOMEM;
221			}
222			if (DRM_COPY_FROM_USER(p->chunks[i].kdata,
223					       p->chunks[i].user_ptr, size)) {
224				return -EFAULT;
225			}
226			if (p->chunks[i].chunk_id == RADEON_CHUNK_ID_FLAGS) {
227				p->cs_flags = p->chunks[i].kdata[0];
228				if (p->chunks[i].length_dw > 1)
229					ring = p->chunks[i].kdata[1];
230				if (p->chunks[i].length_dw > 2)
231					priority = (s32)p->chunks[i].kdata[2];
232			}
233		}
234	}
235
236	if ((p->cs_flags & RADEON_CS_USE_VM) &&
237	    !p->rdev->vm_manager.enabled) {
238		DRM_ERROR("VM not active on asic!\n");
239		if (p->chunk_relocs_idx != -1)
240			kfree(p->chunks[p->chunk_relocs_idx].kdata);
241		if (p->chunk_flags_idx != -1)
242			kfree(p->chunks[p->chunk_flags_idx].kdata);
243		return -EINVAL;
244	}
245
246	if (radeon_cs_get_ring(p, ring, priority)) {
247		if (p->chunk_relocs_idx != -1)
248			kfree(p->chunks[p->chunk_relocs_idx].kdata);
249		if (p->chunk_flags_idx != -1)
250			kfree(p->chunks[p->chunk_flags_idx].kdata);
251		return -EINVAL;
252	}
253
254
255	/* deal with non-vm */
256	if ((p->chunk_ib_idx != -1) &&
257	    ((p->cs_flags & RADEON_CS_USE_VM) == 0) &&
258	    (p->chunks[p->chunk_ib_idx].chunk_id == RADEON_CHUNK_ID_IB)) {
259		if (p->chunks[p->chunk_ib_idx].length_dw > (16 * 1024)) {
260			DRM_ERROR("cs IB too big: %d\n",
261				  p->chunks[p->chunk_ib_idx].length_dw);
262			return -EINVAL;
263		}
264		p->chunks[p->chunk_ib_idx].kpage[0] = kmalloc(PAGE_SIZE, GFP_KERNEL);
265		p->chunks[p->chunk_ib_idx].kpage[1] = kmalloc(PAGE_SIZE, GFP_KERNEL);
266		if (p->chunks[p->chunk_ib_idx].kpage[0] == NULL ||
267		    p->chunks[p->chunk_ib_idx].kpage[1] == NULL) {
268			kfree(p->chunks[p->chunk_ib_idx].kpage[0]);
269			kfree(p->chunks[p->chunk_ib_idx].kpage[1]);
270			return -ENOMEM;
271		}
272		p->chunks[p->chunk_ib_idx].kpage_idx[0] = -1;
273		p->chunks[p->chunk_ib_idx].kpage_idx[1] = -1;
274		p->chunks[p->chunk_ib_idx].last_copied_page = -1;
275		p->chunks[p->chunk_ib_idx].last_page_index =
276			((p->chunks[p->chunk_ib_idx].length_dw * 4) - 1) / PAGE_SIZE;
277	}
278
279	return 0;
280}
281
282/**
283 * cs_parser_fini() - clean parser states
284 * @parser:	parser structure holding parsing context.
285 * @error:	error number
286 *
287 * If error is set than unvalidate buffer, otherwise just free memory
288 * used by parsing context.
289 **/
290static void radeon_cs_parser_fini(struct radeon_cs_parser *parser, int error)
291{
292	unsigned i;
293
294
295	if (!error && parser->ib)
296		ttm_eu_fence_buffer_objects(&parser->validated,
297					    parser->ib->fence);
298	else
299		ttm_eu_backoff_reservation(&parser->validated);
300
301	if (parser->relocs != NULL) {
302		for (i = 0; i < parser->nrelocs; i++) {
303			if (parser->relocs[i].gobj)
304				drm_gem_object_unreference_unlocked(parser->relocs[i].gobj);
305		}
306	}
307	kfree(parser->track);
308	kfree(parser->relocs);
309	kfree(parser->relocs_ptr);
310	for (i = 0; i < parser->nchunks; i++) {
311		kfree(parser->chunks[i].kdata);
312		kfree(parser->chunks[i].kpage[0]);
313		kfree(parser->chunks[i].kpage[1]);
314	}
315	kfree(parser->chunks);
316	kfree(parser->chunks_array);
317	radeon_ib_free(parser->rdev, &parser->ib);
318}
319
320static int radeon_cs_ib_chunk(struct radeon_device *rdev,
321			      struct radeon_cs_parser *parser)
322{
323	struct radeon_cs_chunk *ib_chunk;
324	int r;
325
326	if (parser->chunk_ib_idx == -1)
327		return 0;
328
329	if (parser->cs_flags & RADEON_CS_USE_VM)
330		return 0;
331
332	ib_chunk = &parser->chunks[parser->chunk_ib_idx];
333	/* Copy the packet into the IB, the parser will read from the
334	 * input memory (cached) and write to the IB (which can be
335	 * uncached).
336	 */
337	r =  radeon_ib_get(rdev, parser->ring, &parser->ib,
338			   ib_chunk->length_dw * 4);
339	if (r) {
340		DRM_ERROR("Failed to get ib !\n");
341		return r;
342	}
343	parser->ib->length_dw = ib_chunk->length_dw;
344	r = radeon_cs_parse(parser);
345	if (r || parser->parser_error) {
346		DRM_ERROR("Invalid command stream !\n");
347		return r;
348	}
349	r = radeon_cs_finish_pages(parser);
350	if (r) {
351		DRM_ERROR("Invalid command stream !\n");
352		return r;
353	}
354	r = radeon_cs_sync_rings(parser);
355	if (r) {
356		DRM_ERROR("Failed to synchronize rings !\n");
357	}
358	parser->ib->vm_id = 0;
359	r = radeon_ib_schedule(rdev, parser->ib);
360	if (r) {
361		DRM_ERROR("Failed to schedule IB !\n");
362	}
363	return 0;
364}
365
366static int radeon_bo_vm_update_pte(struct radeon_cs_parser *parser,
367				   struct radeon_vm *vm)
368{
369	struct radeon_bo_list *lobj;
370	struct radeon_bo *bo;
371	int r;
372
373	list_for_each_entry(lobj, &parser->validated, tv.head) {
374		bo = lobj->bo;
375		r = radeon_vm_bo_update_pte(parser->rdev, vm, bo, &bo->tbo.mem);
376		if (r) {
377			return r;
378		}
379	}
380	return 0;
381}
382
383static int radeon_cs_ib_vm_chunk(struct radeon_device *rdev,
384				 struct radeon_cs_parser *parser)
385{
386	struct radeon_cs_chunk *ib_chunk;
387	struct radeon_fpriv *fpriv = parser->filp->driver_priv;
388	struct radeon_vm *vm = &fpriv->vm;
389	int r;
390
391	if (parser->chunk_ib_idx == -1)
392		return 0;
393
394	if ((parser->cs_flags & RADEON_CS_USE_VM) == 0)
395		return 0;
396
397	ib_chunk = &parser->chunks[parser->chunk_ib_idx];
398	if (ib_chunk->length_dw > RADEON_IB_VM_MAX_SIZE) {
399		DRM_ERROR("cs IB too big: %d\n", ib_chunk->length_dw);
400		return -EINVAL;
401	}
402	r =  radeon_ib_get(rdev, parser->ring, &parser->ib,
403			   ib_chunk->length_dw * 4);
404	if (r) {
405		DRM_ERROR("Failed to get ib !\n");
406		return r;
407	}
408	parser->ib->length_dw = ib_chunk->length_dw;
409	/* Copy the packet into the IB */
410	if (DRM_COPY_FROM_USER(parser->ib->ptr, ib_chunk->user_ptr,
411			       ib_chunk->length_dw * 4)) {
412		return -EFAULT;
413	}
414	r = radeon_ring_ib_parse(rdev, parser->ring, parser->ib);
415	if (r) {
416		return r;
417	}
418
419	mutex_lock(&vm->mutex);
420	r = radeon_vm_bind(rdev, vm);
421	if (r) {
422		goto out;
423	}
424	r = radeon_bo_vm_update_pte(parser, vm);
425	if (r) {
426		goto out;
427	}
428	r = radeon_cs_sync_rings(parser);
429	if (r) {
430		DRM_ERROR("Failed to synchronize rings !\n");
431	}
432	parser->ib->vm_id = vm->id;
433	/* ib pool is bind at 0 in virtual address space to gpu_addr is the
434	 * offset inside the pool bo
435	 */
436	parser->ib->gpu_addr = parser->ib->sa_bo.offset;
437	r = radeon_ib_schedule(rdev, parser->ib);
438out:
439	if (!r) {
440		if (vm->fence) {
441			radeon_fence_unref(&vm->fence);
442		}
443		vm->fence = radeon_fence_ref(parser->ib->fence);
444	}
445	mutex_unlock(&fpriv->vm.mutex);
446	return r;
447}
448
449int radeon_cs_ioctl(struct drm_device *dev, void *data, struct drm_file *filp)
450{
451	struct radeon_device *rdev = dev->dev_private;
452	struct radeon_cs_parser parser;
453	int r;
454
455	radeon_mutex_lock(&rdev->cs_mutex);
456	if (!rdev->accel_working) {
457		radeon_mutex_unlock(&rdev->cs_mutex);
458		return -EBUSY;
459	}
460	/* initialize parser */
461	memset(&parser, 0, sizeof(struct radeon_cs_parser));
462	parser.filp = filp;
463	parser.rdev = rdev;
464	parser.dev = rdev->dev;
465	parser.family = rdev->family;
466	r = radeon_cs_parser_init(&parser, data);
467	if (r) {
468		DRM_ERROR("Failed to initialize parser !\n");
469		radeon_cs_parser_fini(&parser, r);
470		radeon_mutex_unlock(&rdev->cs_mutex);
471		return r;
472	}
473	r = radeon_cs_parser_relocs(&parser);
474	if (r) {
475		if (r != -ERESTARTSYS)
476			DRM_ERROR("Failed to parse relocation %d!\n", r);
477		radeon_cs_parser_fini(&parser, r);
478		radeon_mutex_unlock(&rdev->cs_mutex);
479		return r;
480	}
481	r = radeon_cs_ib_chunk(rdev, &parser);
482	if (r) {
483		goto out;
484	}
485	r = radeon_cs_ib_vm_chunk(rdev, &parser);
486	if (r) {
487		goto out;
488	}
489out:
490	radeon_cs_parser_fini(&parser, r);
491	radeon_mutex_unlock(&rdev->cs_mutex);
492	return r;
493}
494
495int radeon_cs_finish_pages(struct radeon_cs_parser *p)
496{
497	struct radeon_cs_chunk *ibc = &p->chunks[p->chunk_ib_idx];
498	int i;
499	int size = PAGE_SIZE;
500
501	for (i = ibc->last_copied_page + 1; i <= ibc->last_page_index; i++) {
502		if (i == ibc->last_page_index) {
503			size = (ibc->length_dw * 4) % PAGE_SIZE;
504			if (size == 0)
505				size = PAGE_SIZE;
506		}
507
508		if (DRM_COPY_FROM_USER(p->ib->ptr + (i * (PAGE_SIZE/4)),
509				       ibc->user_ptr + (i * PAGE_SIZE),
510				       size))
511			return -EFAULT;
512	}
513	return 0;
514}
515
516int radeon_cs_update_pages(struct radeon_cs_parser *p, int pg_idx)
517{
518	int new_page;
519	struct radeon_cs_chunk *ibc = &p->chunks[p->chunk_ib_idx];
520	int i;
521	int size = PAGE_SIZE;
522
523	for (i = ibc->last_copied_page + 1; i < pg_idx; i++) {
524		if (DRM_COPY_FROM_USER(p->ib->ptr + (i * (PAGE_SIZE/4)),
525				       ibc->user_ptr + (i * PAGE_SIZE),
526				       PAGE_SIZE)) {
527			p->parser_error = -EFAULT;
528			return 0;
529		}
530	}
531
532	new_page = ibc->kpage_idx[0] < ibc->kpage_idx[1] ? 0 : 1;
533
534	if (pg_idx == ibc->last_page_index) {
535		size = (ibc->length_dw * 4) % PAGE_SIZE;
536			if (size == 0)
537				size = PAGE_SIZE;
538	}
539
540	if (DRM_COPY_FROM_USER(ibc->kpage[new_page],
541			       ibc->user_ptr + (pg_idx * PAGE_SIZE),
542			       size)) {
543		p->parser_error = -EFAULT;
544		return 0;
545	}
546
547	/* copy to IB here */
548	memcpy((void *)(p->ib->ptr+(pg_idx*(PAGE_SIZE/4))), ibc->kpage[new_page], size);
549
550	ibc->last_copied_page = pg_idx;
551	ibc->kpage_idx[new_page] = pg_idx;
552
553	return new_page;
554}
555