[gem] Catch -EINTR from blocking ioctls and restart them.
[profile/ivi/libdrm.git] / libdrm / intel / intel_bufmgr_gem.c
1 /**************************************************************************
2  *
3  * Copyright © 2007 Red Hat Inc.
4  * Copyright © 2007 Intel Corporation
5  * Copyright 2006 Tungsten Graphics, Inc., Bismarck, ND., USA
6  * All Rights Reserved.
7  *
8  * Permission is hereby granted, free of charge, to any person obtaining a
9  * copy of this software and associated documentation files (the
10  * "Software"), to deal in the Software without restriction, including
11  * without limitation the rights to use, copy, modify, merge, publish,
12  * distribute, sub license, and/or sell copies of the Software, and to
13  * permit persons to whom the Software is furnished to do so, subject to
14  * the following conditions:
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 NON-INFRINGEMENT. IN NO EVENT SHALL
19  * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM,
20  * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
21  * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
22  * USE OR OTHER DEALINGS IN THE SOFTWARE.
23  *
24  * The above copyright notice and this permission notice (including the
25  * next paragraph) shall be included in all copies or substantial portions
26  * of the Software.
27  *
28  *
29  **************************************************************************/
30 /*
31  * Authors: Thomas Hellström <thomas-at-tungstengraphics-dot-com>
32  *          Keith Whitwell <keithw-at-tungstengraphics-dot-com>
33  *          Eric Anholt <eric@anholt.net>
34  *          Dave Airlie <airlied@linux.ie>
35  */
36
37 #include <xf86drm.h>
38 #include <stdio.h>
39 #include <stdlib.h>
40 #include <string.h>
41 #include <unistd.h>
42 #include <assert.h>
43 #include <sys/ioctl.h>
44 #include <sys/mman.h>
45
46 #include "errno.h"
47 #include "dri_bufmgr.h"
48 #include "intel_bufmgr.h"
49 #include "string.h"
50
51 #include "i915_drm.h"
52
53 #define DBG(...) do {                                   \
54    if (bufmgr_gem->bufmgr.debug)                        \
55       fprintf(stderr, __VA_ARGS__);                     \
56 } while (0)
57
58 typedef struct _dri_bo_gem dri_bo_gem;
59
60 struct dri_gem_bo_bucket {
61    dri_bo_gem *head, **tail;
62    /**
63     * Limit on the number of entries in this bucket.
64     *
65     * 0 means that this caching at this bucket size is disabled.
66     * -1 means that there is no limit to caching at this size.
67     */
68    int max_entries;
69    int num_entries;
70 };
71
72 /* Arbitrarily chosen, 16 means that the maximum size we'll cache for reuse
73  * is 1 << 16 pages, or 256MB.
74  */
75 #define INTEL_GEM_BO_BUCKETS    16
76 typedef struct _dri_bufmgr_gem {
77     dri_bufmgr bufmgr;
78
79     struct intel_bufmgr intel_bufmgr;
80
81     int fd;
82
83     int max_relocs;
84
85     struct drm_i915_gem_exec_object *exec_objects;
86     dri_bo **exec_bos;
87     int exec_size;
88     int exec_count;
89
90     /** Array of lists of cached gem objects of power-of-two sizes */
91     struct dri_gem_bo_bucket cache_bucket[INTEL_GEM_BO_BUCKETS];
92
93     struct drm_i915_gem_execbuffer exec_arg;
94 } dri_bufmgr_gem;
95
96 struct _dri_bo_gem {
97     dri_bo bo;
98
99     int refcount;
100     /** Boolean whether the mmap ioctl has been called for this buffer yet. */
101     int mapped;
102     uint32_t gem_handle;
103     const char *name;
104
105     /**
106      * Index of the buffer within the validation list while preparing a
107      * batchbuffer execution.
108      */
109     int validate_index;
110
111     /**
112      * Boolean whether set_domain to CPU is current
113      * Set when set_domain has been called
114      * Cleared when a batch has been submitted
115      */
116     int cpu_domain_set;
117
118     /** Array passed to the DRM containing relocation information. */
119     struct drm_i915_gem_relocation_entry *relocs;
120     /** Array of bos corresponding to relocs[i].target_handle */
121     dri_bo **reloc_target_bo;
122     /** Number of entries in relocs */
123     int reloc_count;
124     /** Mapped address for the buffer */
125     void *virtual;
126
127     /** free list */
128     dri_bo_gem *next;
129 };
130
131 static int
132 logbase2(int n)
133 {
134    int i = 1;
135    int log2 = 0;
136
137    while (n > i) {
138       i *= 2;
139       log2++;
140    }
141
142    return log2;
143 }
144
145 static struct dri_gem_bo_bucket *
146 dri_gem_bo_bucket_for_size(dri_bufmgr_gem *bufmgr_gem, unsigned long size)
147 {
148     int i;
149
150     /* We only do buckets in power of two increments */
151     if ((size & (size - 1)) != 0)
152         return NULL;
153
154     /* We should only see sizes rounded to pages. */
155     assert((size % 4096) == 0);
156
157     /* We always allocate in units of pages */
158     i = ffs(size / 4096) - 1;
159     if (i >= INTEL_GEM_BO_BUCKETS)
160         return NULL;
161
162     return &bufmgr_gem->cache_bucket[i];
163 }
164
165
166 static void dri_gem_dump_validation_list(dri_bufmgr_gem *bufmgr_gem)
167 {
168     int i, j;
169
170     for (i = 0; i < bufmgr_gem->exec_count; i++) {
171         dri_bo *bo = bufmgr_gem->exec_bos[i];
172         dri_bo_gem *bo_gem = (dri_bo_gem *)bo;
173
174         if (bo_gem->relocs == NULL) {
175             DBG("%2d: %d (%s)\n", i, bo_gem->gem_handle, bo_gem->name);
176             continue;
177         }
178
179         for (j = 0; j < bo_gem->reloc_count; j++) {
180             dri_bo *target_bo = bo_gem->reloc_target_bo[j];
181             dri_bo_gem *target_gem = (dri_bo_gem *)target_bo;
182
183             DBG("%2d: %d (%s)@0x%08llx -> %d (%s)@0x%08lx + 0x%08x\n",
184                 i,
185                 bo_gem->gem_handle, bo_gem->name, bo_gem->relocs[j].offset,
186                 target_gem->gem_handle, target_gem->name, target_bo->offset,
187                 bo_gem->relocs[j].delta);
188         }
189     }
190 }
191
192 /**
193  * Adds the given buffer to the list of buffers to be validated (moved into the
194  * appropriate memory type) with the next batch submission.
195  *
196  * If a buffer is validated multiple times in a batch submission, it ends up
197  * with the intersection of the memory type flags and the union of the
198  * access flags.
199  */
200 static void
201 intel_add_validate_buffer(dri_bo *bo)
202 {
203     dri_bufmgr_gem *bufmgr_gem = (dri_bufmgr_gem *)bo->bufmgr;
204     dri_bo_gem *bo_gem = (dri_bo_gem *)bo;
205     int index;
206
207     if (bo_gem->validate_index != -1)
208         return;
209
210     /* Extend the array of validation entries as necessary. */
211     if (bufmgr_gem->exec_count == bufmgr_gem->exec_size) {
212         int new_size = bufmgr_gem->exec_size * 2;
213
214         if (new_size == 0)
215             new_size = 5;
216
217         bufmgr_gem->exec_objects =
218             realloc(bufmgr_gem->exec_objects,
219                     sizeof(*bufmgr_gem->exec_objects) * new_size);
220         bufmgr_gem->exec_bos =
221             realloc(bufmgr_gem->exec_bos,
222                     sizeof(*bufmgr_gem->exec_bos) * new_size);
223         bufmgr_gem->exec_size = new_size;
224     }
225
226     index = bufmgr_gem->exec_count;
227     bo_gem->validate_index = index;
228     /* Fill in array entry */
229     bufmgr_gem->exec_objects[index].handle = bo_gem->gem_handle;
230     bufmgr_gem->exec_objects[index].relocation_count = bo_gem->reloc_count;
231     bufmgr_gem->exec_objects[index].relocs_ptr = (uintptr_t)bo_gem->relocs;
232     bufmgr_gem->exec_objects[index].alignment = 0;
233     bufmgr_gem->exec_objects[index].offset = 0;
234     bufmgr_gem->exec_bos[index] = bo;
235     dri_bo_reference(bo);
236     bufmgr_gem->exec_count++;
237 }
238
239
240 #define RELOC_BUF_SIZE(x) ((I915_RELOC_HEADER + x * I915_RELOC0_STRIDE) * \
241         sizeof(uint32_t))
242
243 static int
244 intel_setup_reloc_list(dri_bo *bo)
245 {
246     dri_bo_gem *bo_gem = (dri_bo_gem *)bo;
247     dri_bufmgr_gem *bufmgr_gem = (dri_bufmgr_gem *)bo->bufmgr;
248
249     bo_gem->relocs = malloc(bufmgr_gem->max_relocs *
250                             sizeof(struct drm_i915_gem_relocation_entry));
251     bo_gem->reloc_target_bo = malloc(bufmgr_gem->max_relocs * sizeof(dri_bo *));
252
253     return 0;
254 }
255
256 static dri_bo *
257 dri_gem_bo_alloc(dri_bufmgr *bufmgr, const char *name,
258                  unsigned long size, unsigned int alignment)
259 {
260     dri_bufmgr_gem *bufmgr_gem = (dri_bufmgr_gem *)bufmgr;
261     dri_bo_gem *bo_gem;
262     unsigned int page_size = getpagesize();
263     int ret;
264     struct dri_gem_bo_bucket *bucket;
265     int alloc_from_cache = 0;
266     unsigned long bo_size;
267
268     /* Round the allocated size up to a power of two number of pages. */
269     bo_size = 1 << logbase2(size);
270     if (bo_size < page_size)
271         bo_size = page_size;
272     bucket = dri_gem_bo_bucket_for_size(bufmgr_gem, bo_size);
273
274     /* If we don't have caching at this size, don't actually round the
275      * allocation up.
276      */
277     if (bucket == NULL || bucket->max_entries == 0) {
278         bo_size = size;
279         if (bo_size < page_size)
280             bo_size = page_size;
281     }
282
283     /* Get a buffer out of the cache if available */
284     if (bucket != NULL && bucket->num_entries > 0) {
285         struct drm_i915_gem_busy busy;
286         
287         bo_gem = bucket->head;
288         busy.handle = bo_gem->gem_handle;
289
290         ret = ioctl(bufmgr_gem->fd, DRM_IOCTL_I915_GEM_BUSY, &busy);
291         alloc_from_cache = (ret == 0 && busy.busy == 0);
292
293         if (alloc_from_cache) {
294             bucket->head = bo_gem->next;
295             if (bo_gem->next == NULL)
296                 bucket->tail = &bucket->head;
297             bucket->num_entries--;
298         }
299     }
300
301     if (!alloc_from_cache) {
302         struct drm_i915_gem_create create;
303
304         bo_gem = calloc(1, sizeof(*bo_gem));
305         if (!bo_gem)
306             return NULL;
307
308         bo_gem->bo.size = bo_size;
309         memset(&create, 0, sizeof(create));
310         create.size = bo_size;
311
312         ret = ioctl(bufmgr_gem->fd, DRM_IOCTL_I915_GEM_CREATE, &create);
313         bo_gem->gem_handle = create.handle;
314         if (ret != 0) {
315             free(bo_gem);
316             return NULL;
317         }
318         bo_gem->bo.bufmgr = bufmgr;
319     }
320
321     bo_gem->name = name;
322     bo_gem->refcount = 1;
323     bo_gem->validate_index = -1;
324
325     DBG("bo_create: buf %d (%s) %ldb\n",
326         bo_gem->gem_handle, bo_gem->name, size);
327
328     return &bo_gem->bo;
329 }
330
331 /**
332  * Returns a dri_bo wrapping the given buffer object handle.
333  *
334  * This can be used when one application needs to pass a buffer object
335  * to another.
336  */
337 dri_bo *
338 intel_bo_gem_create_from_name(dri_bufmgr *bufmgr, const char *name,
339                               unsigned int handle)
340 {
341     dri_bufmgr_gem *bufmgr_gem = (dri_bufmgr_gem *)bufmgr;
342     dri_bo_gem *bo_gem;
343     int ret;
344     struct drm_gem_open open_arg;
345
346     bo_gem = calloc(1, sizeof(*bo_gem));
347     if (!bo_gem)
348         return NULL;
349
350     memset(&open_arg, 0, sizeof(open_arg));
351     open_arg.name = handle;
352     ret = ioctl(bufmgr_gem->fd, DRM_IOCTL_GEM_OPEN, &open_arg);
353     if (ret != 0) {
354         fprintf(stderr, "Couldn't reference %s handle 0x%08x: %s\n",
355                name, handle, strerror(-ret));
356         free(bo_gem);
357         return NULL;
358     }
359     bo_gem->bo.size = open_arg.size;
360     bo_gem->bo.offset = 0;
361     bo_gem->bo.virtual = NULL;
362     bo_gem->bo.bufmgr = bufmgr;
363     bo_gem->name = name;
364     bo_gem->refcount = 1;
365     bo_gem->validate_index = -1;
366     bo_gem->gem_handle = open_arg.handle;
367
368     DBG("bo_create_from_handle: %d (%s)\n", handle, bo_gem->name);
369
370     return &bo_gem->bo;
371 }
372
373 static void
374 dri_gem_bo_reference(dri_bo *bo)
375 {
376     dri_bo_gem *bo_gem = (dri_bo_gem *)bo;
377
378     bo_gem->refcount++;
379 }
380
381 static void
382 dri_gem_bo_free(dri_bo *bo)
383 {
384     dri_bufmgr_gem *bufmgr_gem = (dri_bufmgr_gem *)bo->bufmgr;
385     dri_bo_gem *bo_gem = (dri_bo_gem *)bo;
386     struct drm_gem_close close;
387     int ret;
388
389     if (bo_gem->mapped)
390         munmap (bo_gem->virtual, bo_gem->bo.size);
391
392     /* Close this object */
393     close.handle = bo_gem->gem_handle;
394     ret = ioctl(bufmgr_gem->fd, DRM_IOCTL_GEM_CLOSE, &close);
395     if (ret != 0) {
396         fprintf(stderr,
397                 "DRM_IOCTL_GEM_CLOSE %d failed (%s): %s\n",
398                 bo_gem->gem_handle, bo_gem->name, strerror(-ret));
399     }
400     free(bo);
401 }
402
403 static void
404 dri_gem_bo_unreference(dri_bo *bo)
405 {
406     dri_bufmgr_gem *bufmgr_gem = (dri_bufmgr_gem *)bo->bufmgr;
407     dri_bo_gem *bo_gem = (dri_bo_gem *)bo;
408
409     if (!bo)
410         return;
411
412     if (--bo_gem->refcount == 0) {
413         struct dri_gem_bo_bucket *bucket;
414
415         if (bo_gem->relocs != NULL) {
416             int i;
417
418             /* Unreference all the target buffers */
419             for (i = 0; i < bo_gem->reloc_count; i++)
420                  dri_bo_unreference(bo_gem->reloc_target_bo[i]);
421             free(bo_gem->reloc_target_bo);
422             free(bo_gem->relocs);
423         }
424
425         DBG("bo_unreference final: %d (%s)\n",
426             bo_gem->gem_handle, bo_gem->name);
427
428         bucket = dri_gem_bo_bucket_for_size(bufmgr_gem, bo->size);
429         /* Put the buffer into our internal cache for reuse if we can. */
430         if (bucket != NULL &&
431             (bucket->max_entries == -1 ||
432              (bucket->max_entries > 0 &&
433               bucket->num_entries < bucket->max_entries)))
434         {
435             bo_gem->name = 0;
436             bo_gem->validate_index = -1;
437             bo_gem->relocs = NULL;
438             bo_gem->reloc_target_bo = NULL;
439             bo_gem->reloc_count = 0;
440
441             bo_gem->next = NULL;
442             *bucket->tail = bo_gem;
443             bucket->tail = &bo_gem->next;
444             bucket->num_entries++;
445         } else {
446             dri_gem_bo_free(bo);
447         }
448
449         return;
450     }
451 }
452
453 static int
454 dri_gem_bo_map(dri_bo *bo, int write_enable)
455 {
456     dri_bufmgr_gem *bufmgr_gem;
457     dri_bo_gem *bo_gem = (dri_bo_gem *)bo;
458     struct drm_i915_gem_set_domain set_domain;
459     int ret;
460
461     bufmgr_gem = (dri_bufmgr_gem *)bo->bufmgr;
462
463     /* Allow recursive mapping. Mesa may recursively map buffers with
464      * nested display loops.
465      */
466     if (!bo_gem->mapped) {
467     
468         assert(bo->virtual == NULL);
469     
470         DBG("bo_map: %d (%s)\n", bo_gem->gem_handle, bo_gem->name);
471     
472         if (bo_gem->virtual == NULL) {
473             struct drm_i915_gem_mmap mmap_arg;
474     
475             memset(&mmap_arg, 0, sizeof(mmap_arg));
476             mmap_arg.handle = bo_gem->gem_handle;
477             mmap_arg.offset = 0;
478             mmap_arg.size = bo->size;
479             ret = ioctl(bufmgr_gem->fd, DRM_IOCTL_I915_GEM_MMAP, &mmap_arg);
480             if (ret != 0) {
481                 fprintf(stderr, "%s:%d: Error mapping buffer %d (%s): %s .\n",
482                         __FILE__, __LINE__,
483                         bo_gem->gem_handle, bo_gem->name, strerror(errno));
484             }
485             bo_gem->virtual = (void *)(uintptr_t)mmap_arg.addr_ptr;
486         }
487         bo->virtual = bo_gem->virtual;
488         bo_gem->mapped = 1;
489         DBG("bo_map: %d (%s) -> %p\n", bo_gem->gem_handle, bo_gem->name, bo_gem->virtual);
490     }
491
492     if (!bo_gem->cpu_domain_set) {
493         set_domain.handle = bo_gem->gem_handle;
494         set_domain.read_domains = I915_GEM_DOMAIN_CPU;
495         set_domain.write_domain = write_enable ? I915_GEM_DOMAIN_CPU : 0;
496         do {
497             ret = ioctl(bufmgr_gem->fd, DRM_IOCTL_I915_GEM_SET_DOMAIN,
498                         &set_domain);
499         } while (ret == -1 && errno == EINTR);
500         if (ret != 0) {
501             fprintf (stderr, "%s:%d: Error setting memory domains %d (%08x %08x): %s .\n",
502                      __FILE__, __LINE__,
503                      bo_gem->gem_handle, set_domain.read_domains, set_domain.write_domain,
504                      strerror (errno));
505         }
506         bo_gem->cpu_domain_set = 1;
507     }
508
509     return 0;
510 }
511
512 static int
513 dri_gem_bo_unmap(dri_bo *bo)
514 {
515     dri_bo_gem *bo_gem = (dri_bo_gem *)bo;
516
517     if (bo == NULL)
518         return 0;
519
520     assert(bo_gem->mapped);
521
522     return 0;
523 }
524
525 static int
526 dri_gem_bo_subdata (dri_bo *bo, unsigned long offset,
527                     unsigned long size, const void *data)
528 {
529     dri_bufmgr_gem *bufmgr_gem = (dri_bufmgr_gem *)bo->bufmgr;
530     dri_bo_gem *bo_gem = (dri_bo_gem *)bo;
531     struct drm_i915_gem_pwrite pwrite;
532     int ret;
533
534     memset (&pwrite, 0, sizeof (pwrite));
535     pwrite.handle = bo_gem->gem_handle;
536     pwrite.offset = offset;
537     pwrite.size = size;
538     pwrite.data_ptr = (uint64_t) (uintptr_t) data;
539     do {
540         ret = ioctl (bufmgr_gem->fd, DRM_IOCTL_I915_GEM_PWRITE, &pwrite);
541     } while (ret == -1 && errno == EINTR);
542     if (ret != 0) {
543         fprintf (stderr, "%s:%d: Error writing data to buffer %d: (%d %d) %s .\n",
544                  __FILE__, __LINE__,
545                  bo_gem->gem_handle, (int) offset, (int) size,
546                  strerror (errno));
547     }
548     return 0;
549 }
550
551 static int
552 dri_gem_bo_get_subdata (dri_bo *bo, unsigned long offset,
553                         unsigned long size, void *data)
554 {
555     dri_bufmgr_gem *bufmgr_gem = (dri_bufmgr_gem *)bo->bufmgr;
556     dri_bo_gem *bo_gem = (dri_bo_gem *)bo;
557     struct drm_i915_gem_pread pread;
558     int ret;
559
560     memset (&pread, 0, sizeof (pread));
561     pread.handle = bo_gem->gem_handle;
562     pread.offset = offset;
563     pread.size = size;
564     pread.data_ptr = (uint64_t) (uintptr_t) data;
565     do {
566         ret = ioctl (bufmgr_gem->fd, DRM_IOCTL_I915_GEM_PREAD, &pread);
567     } while (ret == -1 && errno == EINTR);
568     if (ret != 0) {
569         fprintf (stderr, "%s:%d: Error reading data from buffer %d: (%d %d) %s .\n",
570                  __FILE__, __LINE__,
571                  bo_gem->gem_handle, (int) offset, (int) size,
572                  strerror (errno));
573     }
574     return 0;
575 }
576
577 static void
578 dri_gem_bo_wait_rendering(dri_bo *bo)
579 {
580     dri_bufmgr_gem *bufmgr_gem = (dri_bufmgr_gem *)bo->bufmgr;
581     dri_bo_gem *bo_gem = (dri_bo_gem *)bo;
582     struct drm_i915_gem_set_domain set_domain;
583     int ret;
584
585     set_domain.handle = bo_gem->gem_handle;
586     set_domain.read_domains = I915_GEM_DOMAIN_CPU;
587     set_domain.write_domain = 0;
588     ret = ioctl (bufmgr_gem->fd, DRM_IOCTL_I915_GEM_SET_DOMAIN, &set_domain);
589     if (ret != 0) {
590         fprintf (stderr, "%s:%d: Error setting memory domains %d (%08x %08x): %s .\n",
591                  __FILE__, __LINE__,
592                  bo_gem->gem_handle, set_domain.read_domains, set_domain.write_domain,
593                  strerror (errno));
594     }
595 }
596
597 static void
598 dri_bufmgr_gem_destroy(dri_bufmgr *bufmgr)
599 {
600     dri_bufmgr_gem *bufmgr_gem = (dri_bufmgr_gem *)bufmgr;
601     int i;
602
603     free(bufmgr_gem->exec_objects);
604     free(bufmgr_gem->exec_bos);
605
606     /* Free any cached buffer objects we were going to reuse */
607     for (i = 0; i < INTEL_GEM_BO_BUCKETS; i++) {
608         struct dri_gem_bo_bucket *bucket = &bufmgr_gem->cache_bucket[i];
609         dri_bo_gem *bo_gem;
610
611         while ((bo_gem = bucket->head) != NULL) {
612             bucket->head = bo_gem->next;
613             if (bo_gem->next == NULL)
614                 bucket->tail = &bucket->head;
615             bucket->num_entries--;
616
617             dri_gem_bo_free(&bo_gem->bo);
618         }
619     }
620
621     free(bufmgr);
622 }
623
624 /**
625  * Adds the target buffer to the validation list and adds the relocation
626  * to the reloc_buffer's relocation list.
627  *
628  * The relocation entry at the given offset must already contain the
629  * precomputed relocation value, because the kernel will optimize out
630  * the relocation entry write when the buffer hasn't moved from the
631  * last known offset in target_bo.
632  */
633 static int
634 dri_gem_emit_reloc(dri_bo *bo, uint32_t read_domains, uint32_t write_domain,
635                    uint32_t delta, uint32_t offset, dri_bo *target_bo)
636 {
637     dri_bufmgr_gem *bufmgr_gem = (dri_bufmgr_gem *)bo->bufmgr;
638     dri_bo_gem *bo_gem = (dri_bo_gem *)bo;
639     dri_bo_gem *target_bo_gem = (dri_bo_gem *)target_bo;
640
641     /* Create a new relocation list if needed */
642     if (bo_gem->relocs == NULL)
643         intel_setup_reloc_list(bo);
644
645     /* Check overflow */
646     assert(bo_gem->reloc_count < bufmgr_gem->max_relocs);
647
648     /* Check args */
649     assert (offset <= bo->size - 4);
650     assert ((write_domain & (write_domain-1)) == 0);
651
652     bo_gem->relocs[bo_gem->reloc_count].offset = offset;
653     bo_gem->relocs[bo_gem->reloc_count].delta = delta;
654     bo_gem->relocs[bo_gem->reloc_count].target_handle =
655         target_bo_gem->gem_handle;
656     bo_gem->relocs[bo_gem->reloc_count].read_domains = read_domains;
657     bo_gem->relocs[bo_gem->reloc_count].write_domain = write_domain;
658     bo_gem->relocs[bo_gem->reloc_count].presumed_offset = target_bo->offset;
659
660     bo_gem->reloc_target_bo[bo_gem->reloc_count] = target_bo;
661     dri_bo_reference(target_bo);
662
663     bo_gem->reloc_count++;
664     return 0;
665 }
666
667 /**
668  * Walk the tree of relocations rooted at BO and accumulate the list of
669  * validations to be performed and update the relocation buffers with
670  * index values into the validation list.
671  */
672 static void
673 dri_gem_bo_process_reloc(dri_bo *bo)
674 {
675     dri_bo_gem *bo_gem = (dri_bo_gem *)bo;
676     int i;
677
678     if (bo_gem->relocs == NULL)
679         return;
680
681     for (i = 0; i < bo_gem->reloc_count; i++) {
682         dri_bo *target_bo = bo_gem->reloc_target_bo[i];
683
684         /* Continue walking the tree depth-first. */
685         dri_gem_bo_process_reloc(target_bo);
686
687         /* Add the target to the validate list */
688         intel_add_validate_buffer(target_bo);
689     }
690 }
691
692 static void *
693 dri_gem_process_reloc(dri_bo *batch_buf)
694 {
695     dri_bufmgr_gem *bufmgr_gem = (dri_bufmgr_gem *) batch_buf->bufmgr;
696
697     /* Update indices and set up the validate list. */
698     dri_gem_bo_process_reloc(batch_buf);
699
700     /* Add the batch buffer to the validation list.  There are no relocations
701      * pointing to it.
702      */
703     intel_add_validate_buffer(batch_buf);
704
705     bufmgr_gem->exec_arg.buffers_ptr = (uintptr_t)bufmgr_gem->exec_objects;
706     bufmgr_gem->exec_arg.buffer_count = bufmgr_gem->exec_count;
707     bufmgr_gem->exec_arg.batch_start_offset = 0;
708     bufmgr_gem->exec_arg.batch_len = 0; /* written in intel_exec_ioctl */
709
710     return &bufmgr_gem->exec_arg;
711 }
712
713 static void
714 intel_update_buffer_offsets (dri_bufmgr_gem *bufmgr_gem)
715 {
716     int i;
717
718     for (i = 0; i < bufmgr_gem->exec_count; i++) {
719         dri_bo *bo = bufmgr_gem->exec_bos[i];
720         dri_bo_gem *bo_gem = (dri_bo_gem *)bo;
721
722         /* Update the buffer offset */
723         if (bufmgr_gem->exec_objects[i].offset != bo->offset) {
724             DBG("BO %d (%s) migrated: 0x%08lx -> 0x%08llx\n",
725                 bo_gem->gem_handle, bo_gem->name, bo->offset,
726                 bufmgr_gem->exec_objects[i].offset);
727             bo->offset = bufmgr_gem->exec_objects[i].offset;
728         }
729     }
730 }
731
732 static void
733 dri_gem_post_submit(dri_bo *batch_buf)
734 {
735     dri_bufmgr_gem *bufmgr_gem = (dri_bufmgr_gem *)batch_buf->bufmgr;
736     int i;
737
738     intel_update_buffer_offsets (bufmgr_gem);
739
740     if (bufmgr_gem->bufmgr.debug)
741         dri_gem_dump_validation_list(bufmgr_gem);
742
743     for (i = 0; i < bufmgr_gem->exec_count; i++) {
744         dri_bo *bo = bufmgr_gem->exec_bos[i];
745         dri_bo_gem *bo_gem = (dri_bo_gem *)bo;
746
747         /* Need to call set_domain on next bo_map */
748         bo_gem->cpu_domain_set = 0;
749
750         /* Disconnect the buffer from the validate list */
751         bo_gem->validate_index = -1;
752         dri_bo_unreference(bo);
753         bufmgr_gem->exec_bos[i] = NULL;
754     }
755     bufmgr_gem->exec_count = 0;
756 }
757
758 /**
759  * Enables unlimited caching of buffer objects for reuse.
760  *
761  * This is potentially very memory expensive, as the cache at each bucket
762  * size is only bounded by how many buffers of that size we've managed to have
763  * in flight at once.
764  */
765 void
766 intel_bufmgr_gem_enable_reuse(dri_bufmgr *bufmgr)
767 {
768     dri_bufmgr_gem *bufmgr_gem = (dri_bufmgr_gem *)bufmgr;
769     int i;
770
771     for (i = 0; i < INTEL_GEM_BO_BUCKETS; i++) {
772         bufmgr_gem->cache_bucket[i].max_entries = -1;
773     }
774 }
775
776 /*
777  *
778  */
779 static int
780 dri_gem_check_aperture_space(dri_bo *bo)
781 {
782     return 0;
783 }
784
785 /**
786  * Initializes the GEM buffer manager, which uses the kernel to allocate, map,
787  * and manage map buffer objections.
788  *
789  * \param fd File descriptor of the opened DRM device.
790  */
791 dri_bufmgr *
792 intel_bufmgr_gem_init(int fd, int batch_size)
793 {
794     dri_bufmgr_gem *bufmgr_gem;
795     int i;
796
797     bufmgr_gem = calloc(1, sizeof(*bufmgr_gem));
798     bufmgr_gem->fd = fd;
799
800     /* Let's go with one relocation per every 2 dwords (but round down a bit
801      * since a power of two will mean an extra page allocation for the reloc
802      * buffer).
803      *
804      * Every 4 was too few for the blender benchmark.
805      */
806     bufmgr_gem->max_relocs = batch_size / sizeof(uint32_t) / 2 - 2;
807
808     bufmgr_gem->bufmgr.bo_alloc = dri_gem_bo_alloc;
809     bufmgr_gem->bufmgr.bo_reference = dri_gem_bo_reference;
810     bufmgr_gem->bufmgr.bo_unreference = dri_gem_bo_unreference;
811     bufmgr_gem->bufmgr.bo_map = dri_gem_bo_map;
812     bufmgr_gem->bufmgr.bo_unmap = dri_gem_bo_unmap;
813     bufmgr_gem->bufmgr.bo_subdata = dri_gem_bo_subdata;
814     bufmgr_gem->bufmgr.bo_get_subdata = dri_gem_bo_get_subdata;
815     bufmgr_gem->bufmgr.bo_wait_rendering = dri_gem_bo_wait_rendering;
816     bufmgr_gem->bufmgr.destroy = dri_bufmgr_gem_destroy;
817     bufmgr_gem->bufmgr.process_relocs = dri_gem_process_reloc;
818     bufmgr_gem->bufmgr.post_submit = dri_gem_post_submit;
819     bufmgr_gem->bufmgr.debug = 0;
820     bufmgr_gem->bufmgr.check_aperture_space = dri_gem_check_aperture_space;
821     bufmgr_gem->intel_bufmgr.emit_reloc = dri_gem_emit_reloc;
822     /* Initialize the linked lists for BO reuse cache. */
823     for (i = 0; i < INTEL_GEM_BO_BUCKETS; i++)
824         bufmgr_gem->cache_bucket[i].tail = &bufmgr_gem->cache_bucket[i].head;
825
826     return &bufmgr_gem->bufmgr;
827 }
828
829 int
830 intel_bo_emit_reloc(dri_bo *reloc_buf,
831                     uint32_t read_domains, uint32_t write_domain,
832                     uint32_t delta, uint32_t offset, dri_bo *target_buf)
833 {
834     struct intel_bufmgr *intel_bufmgr;
835
836     intel_bufmgr = (struct intel_bufmgr *)(reloc_buf->bufmgr + 1);
837
838     return intel_bufmgr->emit_reloc(reloc_buf, read_domains, write_domain,
839                                     delta, offset, target_buf);
840 }