i965/drm: Unwrap the unnecessary drm_bacon_reloc_target_info struct.

[mesa.git] / src / mesa / drivers / dri / i965 / intel_bufmgr_gem.c

/**************************************************************************
 *
 * Copyright © 2007 Red Hat Inc.
 * Copyright © 2007-2012 Intel Corporation
 * Copyright 2006 Tungsten Graphics, Inc., Bismarck, ND., USA
 * 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, sub license, 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 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 NON-INFRINGEMENT. IN NO EVENT SHALL
 * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS 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.
 *
 * The above copyright notice and this permission notice (including the
 * next paragraph) shall be included in all copies or substantial portions
 * of the Software.
 *
 *
 **************************************************************************/
/*
 * Authors: Thomas Hellström <thomas-at-tungstengraphics-dot-com>
 *          Keith Whitwell <keithw-at-tungstengraphics-dot-com>
 *          Eric Anholt <eric@anholt.net>
 *          Dave Airlie <airlied@linux.ie>
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <xf86drm.h>
#include <util/u_atomic.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <assert.h>
#include <pthread.h>
#include <sys/ioctl.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <stdbool.h>

#include "errno.h"
#ifndef ETIME
#define ETIME ETIMEDOUT
#endif
#include "common/gen_debug.h"
#include "common/gen_device_info.h"
#include "libdrm_macros.h"
#include "main/macros.h"
#include "util/macros.h"
#include "util/hash_table.h"
#include "util/list.h"
#include "brw_bufmgr.h"
#include "string.h"

#include "i915_drm.h"

#ifdef HAVE_VALGRIND
#include <valgrind.h>
#include <memcheck.h>
#define VG(x) x
#else
#define VG(x)
#endif

#define memclear(s) memset(&s, 0, sizeof(s))

#define FILE_DEBUG_FLAG DEBUG_BUFMGR

static inline int
atomic_add_unless(int *v, int add, int unless)
{
   int c, old;
   c = p_atomic_read(v);
   while (c != unless && (old = p_atomic_cmpxchg(v, c, c + add)) != c)
      c = old;
   return c == unless;
}

/**
 * upper_32_bits - return bits 32-63 of a number
 * @n: the number we're accessing
 *
 * A basic shift-right of a 64- or 32-bit quantity.  Use this to suppress
 * the "right shift count >= width of type" warning when that quantity is
 * 32-bits.
 */
#define upper_32_bits(n) ((__u32)(((n) >> 16) >> 16))

/**
 * lower_32_bits - return bits 0-31 of a number
 * @n: the number we're accessing
 */
#define lower_32_bits(n) ((__u32)(n))

struct _drm_bacon_context {
        unsigned int ctx_id;
        struct _drm_bacon_bufmgr *bufmgr;
};

typedef struct _drm_bacon_bo_gem drm_bacon_bo_gem;

struct drm_bacon_gem_bo_bucket {
        struct list_head head;
        unsigned long size;
};

typedef struct _drm_bacon_bufmgr {
        int refcount;

        int fd;

        int max_relocs;

        pthread_mutex_t lock;

        struct drm_i915_gem_exec_object2 *exec2_objects;
        drm_bacon_bo **exec_bos;
        int exec_size;
        int exec_count;

        /** Array of lists of cached gem objects of power-of-two sizes */
        struct drm_bacon_gem_bo_bucket cache_bucket[14 * 4];
        int num_buckets;
        time_t time;

        struct list_head managers;

        struct hash_table *name_table;
        struct hash_table *handle_table;

        struct list_head vma_cache;
        int vma_count, vma_open, vma_max;

        uint64_t gtt_size;
        unsigned int has_llc : 1;
        unsigned int bo_reuse : 1;
        unsigned int no_exec : 1;
} drm_bacon_bufmgr;

struct _drm_bacon_bo_gem {
        drm_bacon_bo bo;

        int refcount;
        uint32_t gem_handle;
        const char *name;

        /**
         * Kenel-assigned global name for this object
         *
         * List contains both flink named and prime fd'd objects
         */
        unsigned int global_name;

        /**
         * Index of the buffer within the validation list while preparing a
         * batchbuffer execution.
         */
        int validate_index;

        /**
         * Current tiling mode
         */
        uint32_t tiling_mode;
        uint32_t swizzle_mode;
        unsigned long stride;

        time_t free_time;

        /** Array passed to the DRM containing relocation information. */
        struct drm_i915_gem_relocation_entry *relocs;
        /**
         * Array of info structs corresponding to relocs[i].target_handle etc
         */
        drm_bacon_bo **reloc_bos;
        /** Number of entries in relocs */
        int reloc_count;
        /** Mapped address for the buffer, saved across map/unmap cycles */
        void *mem_virtual;
        /** GTT virtual address for the buffer, saved across map/unmap cycles */
        void *gtt_virtual;
        /** WC CPU address for the buffer, saved across map/unmap cycles */
        void *wc_virtual;
        int map_count;
        struct list_head vma_list;

        /** BO cache list */
        struct list_head head;

        /**
         * Boolean of whether this BO and its children have been included in
         * the current drm_bacon_bufmgr_check_aperture_space() total.
         */
        bool included_in_check_aperture;

        /**
         * Boolean of whether this buffer has been used as a relocation
         * target and had its size accounted for, and thus can't have any
         * further relocations added to it.
         */
        bool used_as_reloc_target;

        /**
         * Boolean of whether we have encountered an error whilst building the relocation tree.
         */
        bool has_error;

        /**
         * Boolean of whether this buffer can be re-used
         */
        bool reusable;

        /**
         * Boolean of whether the GPU is definitely not accessing the buffer.
         *
         * This is only valid when reusable, since non-reusable
         * buffers are those that have been shared with other
         * processes, so we don't know their state.
         */
        bool idle;

        /**
         * Size in bytes of this buffer and its relocation descendents.
         *
         * Used to avoid costly tree walking in
         * drm_bacon_bufmgr_check_aperture in the common case.
         */
        int reloc_tree_size;

        /** Flags that we may need to do the SW_FINISH ioctl on unmap. */
        bool mapped_cpu_write;
};

static unsigned int
drm_bacon_gem_estimate_batch_space(drm_bacon_bo ** bo_array, int count);

static unsigned int
drm_bacon_gem_compute_batch_space(drm_bacon_bo ** bo_array, int count);

static int
drm_bacon_gem_bo_set_tiling_internal(drm_bacon_bo *bo,
                                     uint32_t tiling_mode,
                                     uint32_t stride);

static void drm_bacon_gem_bo_unreference_locked_timed(drm_bacon_bo *bo,
                                                      time_t time);

static void drm_bacon_gem_bo_free(drm_bacon_bo *bo);

static inline drm_bacon_bo_gem *to_bo_gem(drm_bacon_bo *bo)
{
        return (drm_bacon_bo_gem *)bo;
}

static uint32_t
key_hash_uint(const void *key)
{
        return _mesa_hash_data(key, 4);
}

static bool
key_uint_equal(const void *a, const void *b)
{
        return *((unsigned *) a) == *((unsigned *) b);
}

static drm_bacon_bo_gem *
hash_find_bo(struct hash_table *ht, unsigned int key)
{
        struct hash_entry *entry = _mesa_hash_table_search(ht, &key);
        return entry ? (drm_bacon_bo_gem *) entry->data : NULL;
}

static unsigned long
drm_bacon_gem_bo_tile_size(drm_bacon_bufmgr *bufmgr, unsigned long size,
                           uint32_t *tiling_mode)
{
        if (*tiling_mode == I915_TILING_NONE)
                return size;

        /* 965+ just need multiples of page size for tiling */
        return ALIGN(size, 4096);
}

/*
 * Round a given pitch up to the minimum required for X tiling on a
 * given chip.  We use 512 as the minimum to allow for a later tiling
 * change.
 */
static unsigned long
drm_bacon_gem_bo_tile_pitch(drm_bacon_bufmgr *bufmgr,
                            unsigned long pitch, uint32_t *tiling_mode)
{
        unsigned long tile_width;

        /* If untiled, then just align it so that we can do rendering
         * to it with the 3D engine.
         */
        if (*tiling_mode == I915_TILING_NONE)
                return ALIGN(pitch, 64);

        if (*tiling_mode == I915_TILING_X)
                tile_width = 512;
        else
                tile_width = 128;

        /* 965 is flexible */
        return ALIGN(pitch, tile_width);
}

static struct drm_bacon_gem_bo_bucket *
drm_bacon_gem_bo_bucket_for_size(drm_bacon_bufmgr *bufmgr,
                                 unsigned long size)
{
        int i;

        for (i = 0; i < bufmgr->num_buckets; i++) {
                struct drm_bacon_gem_bo_bucket *bucket =
                    &bufmgr->cache_bucket[i];
                if (bucket->size >= size) {
                        return bucket;
                }
        }

        return NULL;
}

static void
drm_bacon_gem_dump_validation_list(drm_bacon_bufmgr *bufmgr)
{
        int i, j;

        for (i = 0; i < bufmgr->exec_count; i++) {
                drm_bacon_bo *bo = bufmgr->exec_bos[i];
                drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;

                if (bo_gem->relocs == NULL) {
                        DBG("%2d: %d (%s)\n", i, bo_gem->gem_handle,
                            bo_gem->name);
                        continue;
                }

                for (j = 0; j < bo_gem->reloc_count; j++) {
                        drm_bacon_bo *target_bo = bo_gem->reloc_bos[j];
                        drm_bacon_bo_gem *target_gem =
                            (drm_bacon_bo_gem *) target_bo;

                        DBG("%2d: %d (%s)@0x%08x %08x -> "
                            "%d (%s)@0x%08x %08x + 0x%08x\n",
                            i,
                            bo_gem->gem_handle,
                            bo_gem->name,
                            upper_32_bits(bo_gem->relocs[j].offset),
                            lower_32_bits(bo_gem->relocs[j].offset),
                            target_gem->gem_handle,
                            target_gem->name,
                            upper_32_bits(target_bo->offset64),
                            lower_32_bits(target_bo->offset64),
                            bo_gem->relocs[j].delta);
                }
        }
}

inline void
drm_bacon_bo_reference(drm_bacon_bo *bo)
{
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;

        p_atomic_inc(&bo_gem->refcount);
}

static void
drm_bacon_add_validate_buffer2(drm_bacon_bo *bo)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *)bo;
        int index;

        if (bo_gem->validate_index != -1)
                return;

        /* Extend the array of validation entries as necessary. */
        if (bufmgr->exec_count == bufmgr->exec_size) {
                int new_size = bufmgr->exec_size * 2;

                if (new_size == 0)
                        new_size = 5;

                bufmgr->exec2_objects =
                        realloc(bufmgr->exec2_objects,
                                sizeof(*bufmgr->exec2_objects) * new_size);
                bufmgr->exec_bos =
                        realloc(bufmgr->exec_bos,
                                sizeof(*bufmgr->exec_bos) * new_size);
                bufmgr->exec_size = new_size;
        }

        index = bufmgr->exec_count;
        bo_gem->validate_index = index;
        /* Fill in array entry */
        bufmgr->exec2_objects[index].handle = bo_gem->gem_handle;
        bufmgr->exec2_objects[index].relocation_count = bo_gem->reloc_count;
        bufmgr->exec2_objects[index].relocs_ptr = (uintptr_t)bo_gem->relocs;
        bufmgr->exec2_objects[index].alignment = bo->align;
        bufmgr->exec2_objects[index].offset = bo->offset64;
        bufmgr->exec2_objects[index].flags = 0;
        bufmgr->exec2_objects[index].rsvd1 = 0;
        bufmgr->exec2_objects[index].rsvd2 = 0;
        bufmgr->exec_bos[index] = bo;
        bufmgr->exec_count++;
}

static void
drm_bacon_bo_gem_set_in_aperture_size(drm_bacon_bufmgr *bufmgr,
                                      drm_bacon_bo_gem *bo_gem,
                                      unsigned int alignment)
{
        unsigned int size;

        assert(!bo_gem->used_as_reloc_target);

        /* The older chipsets are far-less flexible in terms of tiling,
         * and require tiled buffer to be size aligned in the aperture.
         * This means that in the worst possible case we will need a hole
         * twice as large as the object in order for it to fit into the
         * aperture. Optimal packing is for wimps.
         */
        size = bo_gem->bo.size;

        bo_gem->reloc_tree_size = size + alignment;
}

static int
drm_bacon_setup_reloc_list(drm_bacon_bo *bo)
{
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        unsigned int max_relocs = bufmgr->max_relocs;

        if (bo->size / 4 < max_relocs)
                max_relocs = bo->size / 4;

        bo_gem->relocs = malloc(max_relocs *
                                sizeof(struct drm_i915_gem_relocation_entry));
        bo_gem->reloc_bos = malloc(max_relocs * sizeof(drm_bacon_bo *));
        if (bo_gem->relocs == NULL || bo_gem->reloc_bos == NULL) {
                bo_gem->has_error = true;

                free (bo_gem->relocs);
                bo_gem->relocs = NULL;

                free (bo_gem->reloc_bos);
                bo_gem->reloc_bos = NULL;

                return 1;
        }

        return 0;
}

int
drm_bacon_bo_busy(drm_bacon_bo *bo)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        struct drm_i915_gem_busy busy;
        int ret;

        if (bo_gem->reusable && bo_gem->idle)
                return false;

        memclear(busy);
        busy.handle = bo_gem->gem_handle;

        ret = drmIoctl(bufmgr->fd, DRM_IOCTL_I915_GEM_BUSY, &busy);
        if (ret == 0) {
                bo_gem->idle = !busy.busy;
                return busy.busy;
        } else {
                return false;
        }
        return (ret == 0 && busy.busy);
}

static int
drm_bacon_gem_bo_madvise_internal(drm_bacon_bufmgr *bufmgr,
                                  drm_bacon_bo_gem *bo_gem, int state)
{
        struct drm_i915_gem_madvise madv;

        memclear(madv);
        madv.handle = bo_gem->gem_handle;
        madv.madv = state;
        madv.retained = 1;
        drmIoctl(bufmgr->fd, DRM_IOCTL_I915_GEM_MADVISE, &madv);

        return madv.retained;
}

int
drm_bacon_bo_madvise(drm_bacon_bo *bo, int madv)
{
        return drm_bacon_gem_bo_madvise_internal(bo->bufmgr,
                                                 (drm_bacon_bo_gem *) bo,
                                                 madv);
}

/* drop the oldest entries that have been purged by the kernel */
static void
drm_bacon_gem_bo_cache_purge_bucket(drm_bacon_bufmgr *bufmgr,
                                    struct drm_bacon_gem_bo_bucket *bucket)
{
        while (!list_empty(&bucket->head)) {
                drm_bacon_bo_gem *bo_gem;

                bo_gem = LIST_ENTRY(drm_bacon_bo_gem,
                                    bucket->head.next, head);
                if (drm_bacon_gem_bo_madvise_internal
                    (bufmgr, bo_gem, I915_MADV_DONTNEED))
                        break;

                list_del(&bo_gem->head);
                drm_bacon_gem_bo_free(&bo_gem->bo);
        }
}

static drm_bacon_bo *
drm_bacon_gem_bo_alloc_internal(drm_bacon_bufmgr *bufmgr,
                                const char *name,
                                unsigned long size,
                                unsigned long flags,
                                uint32_t tiling_mode,
                                unsigned long stride,
                                unsigned int alignment)
{
        drm_bacon_bo_gem *bo_gem;
        unsigned int page_size = getpagesize();
        int ret;
        struct drm_bacon_gem_bo_bucket *bucket;
        bool alloc_from_cache;
        unsigned long bo_size;
        bool for_render = false;

        if (flags & BO_ALLOC_FOR_RENDER)
                for_render = true;

        /* Round the allocated size up to a power of two number of pages. */
        bucket = drm_bacon_gem_bo_bucket_for_size(bufmgr, size);

        /* If we don't have caching at this size, don't actually round the
         * allocation up.
         */
        if (bucket == NULL) {
                bo_size = size;
                if (bo_size < page_size)
                        bo_size = page_size;
        } else {
                bo_size = bucket->size;
        }

        pthread_mutex_lock(&bufmgr->lock);
        /* Get a buffer out of the cache if available */
retry:
        alloc_from_cache = false;
        if (bucket != NULL && !list_empty(&bucket->head)) {
                if (for_render) {
                        /* Allocate new render-target BOs from the tail (MRU)
                         * of the list, as it will likely be hot in the GPU
                         * cache and in the aperture for us.
                         */
                        bo_gem = LIST_ENTRY(drm_bacon_bo_gem,
                                            bucket->head.prev, head);
                        list_del(&bo_gem->head);
                        alloc_from_cache = true;
                        bo_gem->bo.align = alignment;
                } else {
                        assert(alignment == 0);
                        /* For non-render-target BOs (where we're probably
                         * going to map it first thing in order to fill it
                         * with data), check if the last BO in the cache is
                         * unbusy, and only reuse in that case. Otherwise,
                         * allocating a new buffer is probably faster than
                         * waiting for the GPU to finish.
                         */
                        bo_gem = LIST_ENTRY(drm_bacon_bo_gem,
                                            bucket->head.next, head);
                        if (!drm_bacon_bo_busy(&bo_gem->bo)) {
                                alloc_from_cache = true;
                                list_del(&bo_gem->head);
                        }
                }

                if (alloc_from_cache) {
                        if (!drm_bacon_gem_bo_madvise_internal
                            (bufmgr, bo_gem, I915_MADV_WILLNEED)) {
                                drm_bacon_gem_bo_free(&bo_gem->bo);
                                drm_bacon_gem_bo_cache_purge_bucket(bufmgr,
                                                                    bucket);
                                goto retry;
                        }

                        if (drm_bacon_gem_bo_set_tiling_internal(&bo_gem->bo,
                                                                 tiling_mode,
                                                                 stride)) {
                                drm_bacon_gem_bo_free(&bo_gem->bo);
                                goto retry;
                        }
                }
        }

        if (!alloc_from_cache) {
                struct drm_i915_gem_create create;

                bo_gem = calloc(1, sizeof(*bo_gem));
                if (!bo_gem)
                        goto err;

                /* drm_bacon_gem_bo_free calls list_del() for an uninitialized
                   list (vma_list), so better set the list head here */
                list_inithead(&bo_gem->vma_list);

                bo_gem->bo.size = bo_size;

                memclear(create);
                create.size = bo_size;

                ret = drmIoctl(bufmgr->fd,
                               DRM_IOCTL_I915_GEM_CREATE,
                               &create);
                if (ret != 0) {
                        free(bo_gem);
                        goto err;
                }

                bo_gem->gem_handle = create.handle;
                _mesa_hash_table_insert(bufmgr->handle_table,
                                        &bo_gem->gem_handle, bo_gem);

                bo_gem->bo.handle = bo_gem->gem_handle;
                bo_gem->bo.bufmgr = bufmgr;
                bo_gem->bo.align = alignment;

                bo_gem->tiling_mode = I915_TILING_NONE;
                bo_gem->swizzle_mode = I915_BIT_6_SWIZZLE_NONE;
                bo_gem->stride = 0;

                if (drm_bacon_gem_bo_set_tiling_internal(&bo_gem->bo,
                                                         tiling_mode,
                                                         stride))
                        goto err_free;
        }

        bo_gem->name = name;
        p_atomic_set(&bo_gem->refcount, 1);
        bo_gem->validate_index = -1;
        bo_gem->used_as_reloc_target = false;
        bo_gem->has_error = false;
        bo_gem->reusable = true;

        drm_bacon_bo_gem_set_in_aperture_size(bufmgr, bo_gem, alignment);
        pthread_mutex_unlock(&bufmgr->lock);

        DBG("bo_create: buf %d (%s) %ldb\n",
            bo_gem->gem_handle, bo_gem->name, size);

        return &bo_gem->bo;

err_free:
        drm_bacon_gem_bo_free(&bo_gem->bo);
err:
        pthread_mutex_unlock(&bufmgr->lock);
        return NULL;
}

drm_bacon_bo *
drm_bacon_bo_alloc_for_render(drm_bacon_bufmgr *bufmgr,
                              const char *name,
                              unsigned long size,
                              unsigned int alignment)
{
        return drm_bacon_gem_bo_alloc_internal(bufmgr, name, size,
                                               BO_ALLOC_FOR_RENDER,
                                               I915_TILING_NONE, 0,
                                               alignment);
}

drm_bacon_bo *
drm_bacon_bo_alloc(drm_bacon_bufmgr *bufmgr,
                   const char *name,
                   unsigned long size,
                   unsigned int alignment)
{
        return drm_bacon_gem_bo_alloc_internal(bufmgr, name, size, 0,
                                               I915_TILING_NONE, 0, 0);
}

drm_bacon_bo *
drm_bacon_bo_alloc_tiled(drm_bacon_bufmgr *bufmgr, const char *name,
                         int x, int y, int cpp, uint32_t *tiling_mode,
                         unsigned long *pitch, unsigned long flags)
{
        unsigned long size, stride;
        uint32_t tiling;

        do {
                unsigned long aligned_y, height_alignment;

                tiling = *tiling_mode;

                /* If we're tiled, our allocations are in 8 or 32-row blocks,
                 * so failure to align our height means that we won't allocate
                 * enough pages.
                 *
                 * If we're untiled, we still have to align to 2 rows high
                 * because the data port accesses 2x2 blocks even if the
                 * bottom row isn't to be rendered, so failure to align means
                 * we could walk off the end of the GTT and fault.  This is
                 * documented on 965, and may be the case on older chipsets
                 * too so we try to be careful.
                 */
                aligned_y = y;
                height_alignment = 2;

                if (tiling == I915_TILING_X)
                        height_alignment = 8;
                else if (tiling == I915_TILING_Y)
                        height_alignment = 32;
                aligned_y = ALIGN(y, height_alignment);

                stride = x * cpp;
                stride = drm_bacon_gem_bo_tile_pitch(bufmgr, stride, tiling_mode);
                size = stride * aligned_y;
                size = drm_bacon_gem_bo_tile_size(bufmgr, size, tiling_mode);
        } while (*tiling_mode != tiling);
        *pitch = stride;

        if (tiling == I915_TILING_NONE)
                stride = 0;

        return drm_bacon_gem_bo_alloc_internal(bufmgr, name, size, flags,
                                               tiling, stride, 0);
}

/**
 * Returns a drm_bacon_bo wrapping the given buffer object handle.
 *
 * This can be used when one application needs to pass a buffer object
 * to another.
 */
drm_bacon_bo *
drm_bacon_bo_gem_create_from_name(drm_bacon_bufmgr *bufmgr,
                                  const char *name,
                                  unsigned int handle)
{
        drm_bacon_bo_gem *bo_gem;
        int ret;
        struct drm_gem_open open_arg;
        struct drm_i915_gem_get_tiling get_tiling;

        /* At the moment most applications only have a few named bo.
         * For instance, in a DRI client only the render buffers passed
         * between X and the client are named. And since X returns the
         * alternating names for the front/back buffer a linear search
         * provides a sufficiently fast match.
         */
        pthread_mutex_lock(&bufmgr->lock);
        bo_gem = hash_find_bo(bufmgr->name_table, handle);
        if (bo_gem) {
                drm_bacon_bo_reference(&bo_gem->bo);
                goto out;
        }

        memclear(open_arg);
        open_arg.name = handle;
        ret = drmIoctl(bufmgr->fd,
                       DRM_IOCTL_GEM_OPEN,
                       &open_arg);
        if (ret != 0) {
                DBG("Couldn't reference %s handle 0x%08x: %s\n",
                    name, handle, strerror(errno));
                bo_gem = NULL;
                goto out;
        }
        /* Now see if someone has used a prime handle to get this
         * object from the kernel before by looking through the list
         * again for a matching gem_handle
         */
        bo_gem = hash_find_bo(bufmgr->handle_table, open_arg.handle);
        if (bo_gem) {
                drm_bacon_bo_reference(&bo_gem->bo);
                goto out;
        }

        bo_gem = calloc(1, sizeof(*bo_gem));
        if (!bo_gem)
                goto out;

        p_atomic_set(&bo_gem->refcount, 1);
        list_inithead(&bo_gem->vma_list);

        bo_gem->bo.size = open_arg.size;
        bo_gem->bo.offset64 = 0;
        bo_gem->bo.virtual = NULL;
        bo_gem->bo.bufmgr = bufmgr;
        bo_gem->name = name;
        bo_gem->validate_index = -1;
        bo_gem->gem_handle = open_arg.handle;
        bo_gem->bo.handle = open_arg.handle;
        bo_gem->global_name = handle;
        bo_gem->reusable = false;

        _mesa_hash_table_insert(bufmgr->handle_table,
                                &bo_gem->gem_handle, bo_gem);
        _mesa_hash_table_insert(bufmgr->name_table,
                                &bo_gem->global_name, bo_gem);

        memclear(get_tiling);
        get_tiling.handle = bo_gem->gem_handle;
        ret = drmIoctl(bufmgr->fd,
                       DRM_IOCTL_I915_GEM_GET_TILING,
                       &get_tiling);
        if (ret != 0)
                goto err_unref;

        bo_gem->tiling_mode = get_tiling.tiling_mode;
        bo_gem->swizzle_mode = get_tiling.swizzle_mode;
        /* XXX stride is unknown */
        drm_bacon_bo_gem_set_in_aperture_size(bufmgr, bo_gem, 0);
        DBG("bo_create_from_handle: %d (%s)\n", handle, bo_gem->name);

out:
        pthread_mutex_unlock(&bufmgr->lock);
        return &bo_gem->bo;

err_unref:
        drm_bacon_gem_bo_free(&bo_gem->bo);
        pthread_mutex_unlock(&bufmgr->lock);
        return NULL;
}

static void
drm_bacon_gem_bo_free(drm_bacon_bo *bo)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        struct drm_gem_close close;
        struct hash_entry *entry;
        int ret;

        list_del(&bo_gem->vma_list);
        if (bo_gem->mem_virtual) {
                VG(VALGRIND_FREELIKE_BLOCK(bo_gem->mem_virtual, 0));
                drm_munmap(bo_gem->mem_virtual, bo_gem->bo.size);
                bufmgr->vma_count--;
        }
        if (bo_gem->wc_virtual) {
                VG(VALGRIND_FREELIKE_BLOCK(bo_gem->wc_virtual, 0));
                drm_munmap(bo_gem->wc_virtual, bo_gem->bo.size);
                bufmgr->vma_count--;
        }
        if (bo_gem->gtt_virtual) {
                drm_munmap(bo_gem->gtt_virtual, bo_gem->bo.size);
                bufmgr->vma_count--;
        }

        if (bo_gem->global_name) {
                entry = _mesa_hash_table_search(bufmgr->name_table,
                                                &bo_gem->global_name);
                _mesa_hash_table_remove(bufmgr->name_table, entry);
        }
        entry = _mesa_hash_table_search(bufmgr->handle_table,
                                        &bo_gem->gem_handle);
        _mesa_hash_table_remove(bufmgr->handle_table, entry);

        /* Close this object */
        memclear(close);
        close.handle = bo_gem->gem_handle;
        ret = drmIoctl(bufmgr->fd, DRM_IOCTL_GEM_CLOSE, &close);
        if (ret != 0) {
                DBG("DRM_IOCTL_GEM_CLOSE %d failed (%s): %s\n",
                    bo_gem->gem_handle, bo_gem->name, strerror(errno));
        }
        free(bo);
}

static void
drm_bacon_gem_bo_mark_mmaps_incoherent(drm_bacon_bo *bo)
{
#if HAVE_VALGRIND
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;

        if (bo_gem->mem_virtual)
                VALGRIND_MAKE_MEM_NOACCESS(bo_gem->mem_virtual, bo->size);

        if (bo_gem->wc_virtual)
                VALGRIND_MAKE_MEM_NOACCESS(bo_gem->wc_virtual, bo->size);

        if (bo_gem->gtt_virtual)
                VALGRIND_MAKE_MEM_NOACCESS(bo_gem->gtt_virtual, bo->size);
#endif
}

/** Frees all cached buffers significantly older than @time. */
static void
drm_bacon_gem_cleanup_bo_cache(drm_bacon_bufmgr *bufmgr, time_t time)
{
        int i;

        if (bufmgr->time == time)
                return;

        for (i = 0; i < bufmgr->num_buckets; i++) {
                struct drm_bacon_gem_bo_bucket *bucket =
                    &bufmgr->cache_bucket[i];

                while (!list_empty(&bucket->head)) {
                        drm_bacon_bo_gem *bo_gem;

                        bo_gem = LIST_ENTRY(drm_bacon_bo_gem,
                                            bucket->head.next, head);
                        if (time - bo_gem->free_time <= 1)
                                break;

                        list_del(&bo_gem->head);

                        drm_bacon_gem_bo_free(&bo_gem->bo);
                }
        }

        bufmgr->time = time;
}

static void drm_bacon_gem_bo_purge_vma_cache(drm_bacon_bufmgr *bufmgr)
{
        int limit;

        DBG("%s: cached=%d, open=%d, limit=%d\n", __FUNCTION__,
            bufmgr->vma_count, bufmgr->vma_open, bufmgr->vma_max);

        if (bufmgr->vma_max < 0)
                return;

        /* We may need to evict a few entries in order to create new mmaps */
        limit = bufmgr->vma_max - 2*bufmgr->vma_open;
        if (limit < 0)
                limit = 0;

        while (bufmgr->vma_count > limit) {
                drm_bacon_bo_gem *bo_gem;

                bo_gem = LIST_ENTRY(drm_bacon_bo_gem,
                                    bufmgr->vma_cache.next,
                                    vma_list);
                assert(bo_gem->map_count == 0);
                list_delinit(&bo_gem->vma_list);

                if (bo_gem->mem_virtual) {
                        drm_munmap(bo_gem->mem_virtual, bo_gem->bo.size);
                        bo_gem->mem_virtual = NULL;
                        bufmgr->vma_count--;
                }
                if (bo_gem->wc_virtual) {
                        drm_munmap(bo_gem->wc_virtual, bo_gem->bo.size);
                        bo_gem->wc_virtual = NULL;
                        bufmgr->vma_count--;
                }
                if (bo_gem->gtt_virtual) {
                        drm_munmap(bo_gem->gtt_virtual, bo_gem->bo.size);
                        bo_gem->gtt_virtual = NULL;
                        bufmgr->vma_count--;
                }
        }
}

static void drm_bacon_gem_bo_close_vma(drm_bacon_bufmgr *bufmgr,
                                       drm_bacon_bo_gem *bo_gem)
{
        bufmgr->vma_open--;
        list_addtail(&bo_gem->vma_list, &bufmgr->vma_cache);
        if (bo_gem->mem_virtual)
                bufmgr->vma_count++;
        if (bo_gem->wc_virtual)
                bufmgr->vma_count++;
        if (bo_gem->gtt_virtual)
                bufmgr->vma_count++;
        drm_bacon_gem_bo_purge_vma_cache(bufmgr);
}

static void drm_bacon_gem_bo_open_vma(drm_bacon_bufmgr *bufmgr,
                                      drm_bacon_bo_gem *bo_gem)
{
        bufmgr->vma_open++;
        list_del(&bo_gem->vma_list);
        if (bo_gem->mem_virtual)
                bufmgr->vma_count--;
        if (bo_gem->wc_virtual)
                bufmgr->vma_count--;
        if (bo_gem->gtt_virtual)
                bufmgr->vma_count--;
        drm_bacon_gem_bo_purge_vma_cache(bufmgr);
}

static void
drm_bacon_gem_bo_unreference_final(drm_bacon_bo *bo, time_t time)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        struct drm_bacon_gem_bo_bucket *bucket;
        int i;

        /* Unreference all the target buffers */
        for (i = 0; i < bo_gem->reloc_count; i++) {
                if (bo_gem->reloc_bos[i] != bo) {
                        drm_bacon_gem_bo_unreference_locked_timed(bo_gem->
                                                                  reloc_bos[i],
                                                                  time);
                }
        }
        bo_gem->reloc_count = 0;
        bo_gem->used_as_reloc_target = false;

        DBG("bo_unreference final: %d (%s)\n",
            bo_gem->gem_handle, bo_gem->name);

        /* release memory associated with this object */
        if (bo_gem->reloc_bos) {
                free(bo_gem->reloc_bos);
                bo_gem->reloc_bos = NULL;
        }
        if (bo_gem->relocs) {
                free(bo_gem->relocs);
                bo_gem->relocs = NULL;
        }

        /* Clear any left-over mappings */
        if (bo_gem->map_count) {
                DBG("bo freed with non-zero map-count %d\n", bo_gem->map_count);
                bo_gem->map_count = 0;
                drm_bacon_gem_bo_close_vma(bufmgr, bo_gem);
                drm_bacon_gem_bo_mark_mmaps_incoherent(bo);
        }

        bucket = drm_bacon_gem_bo_bucket_for_size(bufmgr, bo->size);
        /* Put the buffer into our internal cache for reuse if we can. */
        if (bufmgr->bo_reuse && bo_gem->reusable && bucket != NULL &&
            drm_bacon_gem_bo_madvise_internal(bufmgr, bo_gem,
                                              I915_MADV_DONTNEED)) {
                bo_gem->free_time = time;

                bo_gem->name = NULL;
                bo_gem->validate_index = -1;

                list_addtail(&bo_gem->head, &bucket->head);
        } else {
                drm_bacon_gem_bo_free(bo);
        }
}

static void drm_bacon_gem_bo_unreference_locked_timed(drm_bacon_bo *bo,
                                                      time_t time)
{
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;

        assert(p_atomic_read(&bo_gem->refcount) > 0);
        if (p_atomic_dec_zero(&bo_gem->refcount))
                drm_bacon_gem_bo_unreference_final(bo, time);
}

void
drm_bacon_bo_unreference(drm_bacon_bo *bo)
{
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;

        if (bo == NULL)
                return;

        assert(p_atomic_read(&bo_gem->refcount) > 0);

        if (atomic_add_unless(&bo_gem->refcount, -1, 1)) {
                drm_bacon_bufmgr *bufmgr = bo->bufmgr;
                struct timespec time;

                clock_gettime(CLOCK_MONOTONIC, &time);

                pthread_mutex_lock(&bufmgr->lock);

                if (p_atomic_dec_zero(&bo_gem->refcount)) {
                        drm_bacon_gem_bo_unreference_final(bo, time.tv_sec);
                        drm_bacon_gem_cleanup_bo_cache(bufmgr, time.tv_sec);
                }

                pthread_mutex_unlock(&bufmgr->lock);
        }
}

int
drm_bacon_bo_map(drm_bacon_bo *bo, int write_enable)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        struct drm_i915_gem_set_domain set_domain;
        int ret;

        pthread_mutex_lock(&bufmgr->lock);

        if (bo_gem->map_count++ == 0)
                drm_bacon_gem_bo_open_vma(bufmgr, bo_gem);

        if (!bo_gem->mem_virtual) {
                struct drm_i915_gem_mmap mmap_arg;

                DBG("bo_map: %d (%s), map_count=%d\n",
                    bo_gem->gem_handle, bo_gem->name, bo_gem->map_count);

                memclear(mmap_arg);
                mmap_arg.handle = bo_gem->gem_handle;
                mmap_arg.size = bo->size;
                ret = drmIoctl(bufmgr->fd,
                               DRM_IOCTL_I915_GEM_MMAP,
                               &mmap_arg);
                if (ret != 0) {
                        ret = -errno;
                        DBG("%s:%d: Error mapping buffer %d (%s): %s .\n",
                            __FILE__, __LINE__, bo_gem->gem_handle,
                            bo_gem->name, strerror(errno));
                        if (--bo_gem->map_count == 0)
                                drm_bacon_gem_bo_close_vma(bufmgr, bo_gem);
                        pthread_mutex_unlock(&bufmgr->lock);
                        return ret;
                }
                VG(VALGRIND_MALLOCLIKE_BLOCK(mmap_arg.addr_ptr, mmap_arg.size, 0, 1));
                bo_gem->mem_virtual = (void *)(uintptr_t) mmap_arg.addr_ptr;
        }
        DBG("bo_map: %d (%s) -> %p\n", bo_gem->gem_handle, bo_gem->name,
            bo_gem->mem_virtual);
        bo->virtual = bo_gem->mem_virtual;

        memclear(set_domain);
        set_domain.handle = bo_gem->gem_handle;
        set_domain.read_domains = I915_GEM_DOMAIN_CPU;
        if (write_enable)
                set_domain.write_domain = I915_GEM_DOMAIN_CPU;
        else
                set_domain.write_domain = 0;
        ret = drmIoctl(bufmgr->fd,
                       DRM_IOCTL_I915_GEM_SET_DOMAIN,
                       &set_domain);
        if (ret != 0) {
                DBG("%s:%d: Error setting to CPU domain %d: %s\n",
                    __FILE__, __LINE__, bo_gem->gem_handle,
                    strerror(errno));
        }

        if (write_enable)
                bo_gem->mapped_cpu_write = true;

        drm_bacon_gem_bo_mark_mmaps_incoherent(bo);
        VG(VALGRIND_MAKE_MEM_DEFINED(bo_gem->mem_virtual, bo->size));
        pthread_mutex_unlock(&bufmgr->lock);

        return 0;
}

static int
map_gtt(drm_bacon_bo *bo)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        int ret;

        if (bo_gem->map_count++ == 0)
                drm_bacon_gem_bo_open_vma(bufmgr, bo_gem);

        /* Get a mapping of the buffer if we haven't before. */
        if (bo_gem->gtt_virtual == NULL) {
                struct drm_i915_gem_mmap_gtt mmap_arg;

                DBG("bo_map_gtt: mmap %d (%s), map_count=%d\n",
                    bo_gem->gem_handle, bo_gem->name, bo_gem->map_count);

                memclear(mmap_arg);
                mmap_arg.handle = bo_gem->gem_handle;

                /* Get the fake offset back... */
                ret = drmIoctl(bufmgr->fd,
                               DRM_IOCTL_I915_GEM_MMAP_GTT,
                               &mmap_arg);
                if (ret != 0) {
                        ret = -errno;
                        DBG("%s:%d: Error preparing buffer map %d (%s): %s .\n",
                            __FILE__, __LINE__,
                            bo_gem->gem_handle, bo_gem->name,
                            strerror(errno));
                        if (--bo_gem->map_count == 0)
                                drm_bacon_gem_bo_close_vma(bufmgr, bo_gem);
                        return ret;
                }

                /* and mmap it */
                bo_gem->gtt_virtual = drm_mmap(0, bo->size, PROT_READ | PROT_WRITE,
                                               MAP_SHARED, bufmgr->fd,
                                               mmap_arg.offset);
                if (bo_gem->gtt_virtual == MAP_FAILED) {
                        bo_gem->gtt_virtual = NULL;
                        ret = -errno;
                        DBG("%s:%d: Error mapping buffer %d (%s): %s .\n",
                            __FILE__, __LINE__,
                            bo_gem->gem_handle, bo_gem->name,
                            strerror(errno));
                        if (--bo_gem->map_count == 0)
                                drm_bacon_gem_bo_close_vma(bufmgr, bo_gem);
                        return ret;
                }
        }

        bo->virtual = bo_gem->gtt_virtual;

        DBG("bo_map_gtt: %d (%s) -> %p\n", bo_gem->gem_handle, bo_gem->name,
            bo_gem->gtt_virtual);

        return 0;
}

int
drm_bacon_gem_bo_map_gtt(drm_bacon_bo *bo)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        struct drm_i915_gem_set_domain set_domain;
        int ret;

        pthread_mutex_lock(&bufmgr->lock);

        ret = map_gtt(bo);
        if (ret) {
                pthread_mutex_unlock(&bufmgr->lock);
                return ret;
        }

        /* Now move it to the GTT domain so that the GPU and CPU
         * caches are flushed and the GPU isn't actively using the
         * buffer.
         *
         * The pagefault handler does this domain change for us when
         * it has unbound the BO from the GTT, but it's up to us to
         * tell it when we're about to use things if we had done
         * rendering and it still happens to be bound to the GTT.
         */
        memclear(set_domain);
        set_domain.handle = bo_gem->gem_handle;
        set_domain.read_domains = I915_GEM_DOMAIN_GTT;
        set_domain.write_domain = I915_GEM_DOMAIN_GTT;
        ret = drmIoctl(bufmgr->fd,
                       DRM_IOCTL_I915_GEM_SET_DOMAIN,
                       &set_domain);
        if (ret != 0) {
                DBG("%s:%d: Error setting domain %d: %s\n",
                    __FILE__, __LINE__, bo_gem->gem_handle,
                    strerror(errno));
        }

        drm_bacon_gem_bo_mark_mmaps_incoherent(bo);
        VG(VALGRIND_MAKE_MEM_DEFINED(bo_gem->gtt_virtual, bo->size));
        pthread_mutex_unlock(&bufmgr->lock);

        return 0;
}

/**
 * Performs a mapping of the buffer object like the normal GTT
 * mapping, but avoids waiting for the GPU to be done reading from or
 * rendering to the buffer.
 *
 * This is used in the implementation of GL_ARB_map_buffer_range: The
 * user asks to create a buffer, then does a mapping, fills some
 * space, runs a drawing command, then asks to map it again without
 * synchronizing because it guarantees that it won't write over the
 * data that the GPU is busy using (or, more specifically, that if it
 * does write over the data, it acknowledges that rendering is
 * undefined).
 */

int
drm_bacon_gem_bo_map_unsynchronized(drm_bacon_bo *bo)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
#ifdef HAVE_VALGRIND
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
#endif
        int ret;

        /* If the CPU cache isn't coherent with the GTT, then use a
         * regular synchronized mapping.  The problem is that we don't
         * track where the buffer was last used on the CPU side in
         * terms of drm_bacon_bo_map vs drm_bacon_gem_bo_map_gtt, so
         * we would potentially corrupt the buffer even when the user
         * does reasonable things.
         */
        if (!bufmgr->has_llc)
                return drm_bacon_gem_bo_map_gtt(bo);

        pthread_mutex_lock(&bufmgr->lock);

        ret = map_gtt(bo);
        if (ret == 0) {
                drm_bacon_gem_bo_mark_mmaps_incoherent(bo);
                VG(VALGRIND_MAKE_MEM_DEFINED(bo_gem->gtt_virtual, bo->size));
        }

        pthread_mutex_unlock(&bufmgr->lock);

        return ret;
}

int
drm_bacon_bo_unmap(drm_bacon_bo *bo)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        int ret = 0;

        if (bo == NULL)
                return 0;

        pthread_mutex_lock(&bufmgr->lock);

        if (bo_gem->map_count <= 0) {
                DBG("attempted to unmap an unmapped bo\n");
                pthread_mutex_unlock(&bufmgr->lock);
                /* Preserve the old behaviour of just treating this as a
                 * no-op rather than reporting the error.
                 */
                return 0;
        }

        if (bo_gem->mapped_cpu_write) {
                struct drm_i915_gem_sw_finish sw_finish;

                /* Cause a flush to happen if the buffer's pinned for
                 * scanout, so the results show up in a timely manner.
                 * Unlike GTT set domains, this only does work if the
                 * buffer should be scanout-related.
                 */
                memclear(sw_finish);
                sw_finish.handle = bo_gem->gem_handle;
                ret = drmIoctl(bufmgr->fd,
                               DRM_IOCTL_I915_GEM_SW_FINISH,
                               &sw_finish);
                ret = ret == -1 ? -errno : 0;

                bo_gem->mapped_cpu_write = false;
        }

        /* We need to unmap after every innovation as we cannot track
         * an open vma for every bo as that will exhaust the system
         * limits and cause later failures.
         */
        if (--bo_gem->map_count == 0) {
                drm_bacon_gem_bo_close_vma(bufmgr, bo_gem);
                drm_bacon_gem_bo_mark_mmaps_incoherent(bo);
                bo->virtual = NULL;
        }
        pthread_mutex_unlock(&bufmgr->lock);

        return ret;
}

int
drm_bacon_bo_subdata(drm_bacon_bo *bo, unsigned long offset,
                     unsigned long size, const void *data)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        struct drm_i915_gem_pwrite pwrite;
        int ret;

        memclear(pwrite);
        pwrite.handle = bo_gem->gem_handle;
        pwrite.offset = offset;
        pwrite.size = size;
        pwrite.data_ptr = (uint64_t) (uintptr_t) data;
        ret = drmIoctl(bufmgr->fd,
                       DRM_IOCTL_I915_GEM_PWRITE,
                       &pwrite);
        if (ret != 0) {
                ret = -errno;
                DBG("%s:%d: Error writing data to buffer %d: (%d %d) %s .\n",
                    __FILE__, __LINE__, bo_gem->gem_handle, (int)offset,
                    (int)size, strerror(errno));
        }

        return ret;
}

int
drm_bacon_bo_get_subdata(drm_bacon_bo *bo, unsigned long offset,
                         unsigned long size, void *data)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        struct drm_i915_gem_pread pread;
        int ret;

        memclear(pread);
        pread.handle = bo_gem->gem_handle;
        pread.offset = offset;
        pread.size = size;
        pread.data_ptr = (uint64_t) (uintptr_t) data;
        ret = drmIoctl(bufmgr->fd,
                       DRM_IOCTL_I915_GEM_PREAD,
                       &pread);
        if (ret != 0) {
                ret = -errno;
                DBG("%s:%d: Error reading data from buffer %d: (%d %d) %s .\n",
                    __FILE__, __LINE__, bo_gem->gem_handle, (int)offset,
                    (int)size, strerror(errno));
        }

        return ret;
}

/** Waits for all GPU rendering with the object to have completed. */
void
drm_bacon_bo_wait_rendering(drm_bacon_bo *bo)
{
        drm_bacon_gem_bo_start_gtt_access(bo, 1);
}

/**
 * Waits on a BO for the given amount of time.
 *
 * @bo: buffer object to wait for
 * @timeout_ns: amount of time to wait in nanoseconds.
 *   If value is less than 0, an infinite wait will occur.
 *
 * Returns 0 if the wait was successful ie. the last batch referencing the
 * object has completed within the allotted time. Otherwise some negative return
 * value describes the error. Of particular interest is -ETIME when the wait has
 * failed to yield the desired result.
 *
 * Similar to drm_bacon_gem_bo_wait_rendering except a timeout parameter allows
 * the operation to give up after a certain amount of time. Another subtle
 * difference is the internal locking semantics are different (this variant does
 * not hold the lock for the duration of the wait). This makes the wait subject
 * to a larger userspace race window.
 *
 * The implementation shall wait until the object is no longer actively
 * referenced within a batch buffer at the time of the call. The wait will
 * not guarantee that the buffer is re-issued via another thread, or an flinked
 * handle. Userspace must make sure this race does not occur if such precision
 * is important.
 *
 * Note that some kernels have broken the inifite wait for negative values
 * promise, upgrade to latest stable kernels if this is the case.
 */
int
drm_bacon_gem_bo_wait(drm_bacon_bo *bo, int64_t timeout_ns)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        struct drm_i915_gem_wait wait;
        int ret;

        memclear(wait);
        wait.bo_handle = bo_gem->gem_handle;
        wait.timeout_ns = timeout_ns;
        ret = drmIoctl(bufmgr->fd, DRM_IOCTL_I915_GEM_WAIT, &wait);
        if (ret == -1)
                return -errno;

        return ret;
}

/**
 * Sets the object to the GTT read and possibly write domain, used by the X
 * 2D driver in the absence of kernel support to do drm_bacon_gem_bo_map_gtt().
 *
 * In combination with drm_bacon_gem_bo_pin() and manual fence management, we
 * can do tiled pixmaps this way.
 */
void
drm_bacon_gem_bo_start_gtt_access(drm_bacon_bo *bo, int write_enable)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        struct drm_i915_gem_set_domain set_domain;
        int ret;

        memclear(set_domain);
        set_domain.handle = bo_gem->gem_handle;
        set_domain.read_domains = I915_GEM_DOMAIN_GTT;
        set_domain.write_domain = write_enable ? I915_GEM_DOMAIN_GTT : 0;
        ret = drmIoctl(bufmgr->fd,
                       DRM_IOCTL_I915_GEM_SET_DOMAIN,
                       &set_domain);
        if (ret != 0) {
                DBG("%s:%d: Error setting memory domains %d (%08x %08x): %s .\n",
                    __FILE__, __LINE__, bo_gem->gem_handle,
                    set_domain.read_domains, set_domain.write_domain,
                    strerror(errno));
        }
}

static void
drm_bacon_bufmgr_gem_destroy(drm_bacon_bufmgr *bufmgr)
{
        free(bufmgr->exec2_objects);
        free(bufmgr->exec_bos);

        pthread_mutex_destroy(&bufmgr->lock);

        /* Free any cached buffer objects we were going to reuse */
        for (int i = 0; i < bufmgr->num_buckets; i++) {
                struct drm_bacon_gem_bo_bucket *bucket =
                    &bufmgr->cache_bucket[i];
                drm_bacon_bo_gem *bo_gem;

                while (!list_empty(&bucket->head)) {
                        bo_gem = LIST_ENTRY(drm_bacon_bo_gem,
                                            bucket->head.next, head);
                        list_del(&bo_gem->head);

                        drm_bacon_gem_bo_free(&bo_gem->bo);
                }
        }

        _mesa_hash_table_destroy(bufmgr->name_table, NULL);
        _mesa_hash_table_destroy(bufmgr->handle_table, NULL);

        free(bufmgr);
}

/**
 * Adds the target buffer to the validation list and adds the relocation
 * to the reloc_buffer's relocation list.
 *
 * The relocation entry at the given offset must already contain the
 * precomputed relocation value, because the kernel will optimize out
 * the relocation entry write when the buffer hasn't moved from the
 * last known offset in target_bo.
 */
int
drm_bacon_bo_emit_reloc(drm_bacon_bo *bo, uint32_t offset,
                        drm_bacon_bo *target_bo, uint32_t target_offset,
                        uint32_t read_domains, uint32_t write_domain)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        drm_bacon_bo_gem *target_bo_gem = (drm_bacon_bo_gem *) target_bo;

        if (bo_gem->has_error)
                return -ENOMEM;

        if (target_bo_gem->has_error) {
                bo_gem->has_error = true;
                return -ENOMEM;
        }

        /* Create a new relocation list if needed */
        if (bo_gem->relocs == NULL && drm_bacon_setup_reloc_list(bo))
                return -ENOMEM;

        /* Check overflow */
        assert(bo_gem->reloc_count < bufmgr->max_relocs);

        /* Check args */
        assert(offset <= bo->size - 4);
        assert((write_domain & (write_domain - 1)) == 0);

        /* Make sure that we're not adding a reloc to something whose size has
         * already been accounted for.
         */
        assert(!bo_gem->used_as_reloc_target);
        if (target_bo_gem != bo_gem) {
                target_bo_gem->used_as_reloc_target = true;
                bo_gem->reloc_tree_size += target_bo_gem->reloc_tree_size;
        }

        bo_gem->reloc_bos[bo_gem->reloc_count] = target_bo;
        if (target_bo != bo)
                drm_bacon_bo_reference(target_bo);

        bo_gem->relocs[bo_gem->reloc_count].offset = offset;
        bo_gem->relocs[bo_gem->reloc_count].delta = target_offset;
        bo_gem->relocs[bo_gem->reloc_count].target_handle =
            target_bo_gem->gem_handle;
        bo_gem->relocs[bo_gem->reloc_count].read_domains = read_domains;
        bo_gem->relocs[bo_gem->reloc_count].write_domain = write_domain;
        bo_gem->relocs[bo_gem->reloc_count].presumed_offset = target_bo->offset64;
        bo_gem->reloc_count++;

        return 0;
}

int
drm_bacon_gem_bo_get_reloc_count(drm_bacon_bo *bo)
{
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;

        return bo_gem->reloc_count;
}

/**
 * Removes existing relocation entries in the BO after "start".
 *
 * This allows a user to avoid a two-step process for state setup with
 * counting up all the buffer objects and doing a
 * drm_bacon_bufmgr_check_aperture_space() before emitting any of the
 * relocations for the state setup.  Instead, save the state of the
 * batchbuffer including drm_bacon_gem_get_reloc_count(), emit all the
 * state, and then check if it still fits in the aperture.
 *
 * Any further drm_bacon_bufmgr_check_aperture_space() queries
 * involving this buffer in the tree are undefined after this call.
 */
void
drm_bacon_gem_bo_clear_relocs(drm_bacon_bo *bo, int start)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        int i;
        struct timespec time;

        clock_gettime(CLOCK_MONOTONIC, &time);

        assert(bo_gem->reloc_count >= start);

        /* Unreference the cleared target buffers */
        pthread_mutex_lock(&bufmgr->lock);

        for (i = start; i < bo_gem->reloc_count; i++) {
                drm_bacon_bo_gem *target_bo_gem = (drm_bacon_bo_gem *) bo_gem->reloc_bos[i];
                if (&target_bo_gem->bo != bo) {
                        drm_bacon_gem_bo_unreference_locked_timed(&target_bo_gem->bo,
                                                                  time.tv_sec);
                }
        }
        bo_gem->reloc_count = start;

        pthread_mutex_unlock(&bufmgr->lock);

}

static void
drm_bacon_gem_bo_process_reloc2(drm_bacon_bo *bo)
{
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *)bo;
        int i;

        if (bo_gem->relocs == NULL)
                return;

        for (i = 0; i < bo_gem->reloc_count; i++) {
                drm_bacon_bo *target_bo = bo_gem->reloc_bos[i];

                if (target_bo == bo)
                        continue;

                drm_bacon_gem_bo_mark_mmaps_incoherent(bo);

                /* Continue walking the tree depth-first. */
                drm_bacon_gem_bo_process_reloc2(target_bo);

                /* Add the target to the validate list */
                drm_bacon_add_validate_buffer2(target_bo);
        }
}

static void
drm_bacon_update_buffer_offsets2 (drm_bacon_bufmgr *bufmgr)
{
        int i;

        for (i = 0; i < bufmgr->exec_count; i++) {
                drm_bacon_bo *bo = bufmgr->exec_bos[i];
                drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *)bo;

                /* Update the buffer offset */
                if (bufmgr->exec2_objects[i].offset != bo->offset64) {
                        DBG("BO %d (%s) migrated: 0x%08x %08x -> 0x%08x %08x\n",
                            bo_gem->gem_handle, bo_gem->name,
                            upper_32_bits(bo->offset64),
                            lower_32_bits(bo->offset64),
                            upper_32_bits(bufmgr->exec2_objects[i].offset),
                            lower_32_bits(bufmgr->exec2_objects[i].offset));
                        bo->offset64 = bufmgr->exec2_objects[i].offset;
                }
        }
}

static int
do_exec2(drm_bacon_bo *bo, int used, drm_bacon_context *ctx,
         int in_fence, int *out_fence,
         unsigned int flags)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        struct drm_i915_gem_execbuffer2 execbuf;
        int ret = 0;
        int i;

        if (to_bo_gem(bo)->has_error)
                return -ENOMEM;

        pthread_mutex_lock(&bufmgr->lock);
        /* Update indices and set up the validate list. */
        drm_bacon_gem_bo_process_reloc2(bo);

        /* Add the batch buffer to the validation list.  There are no relocations
         * pointing to it.
         */
        drm_bacon_add_validate_buffer2(bo);

        memclear(execbuf);
        execbuf.buffers_ptr = (uintptr_t)bufmgr->exec2_objects;
        execbuf.buffer_count = bufmgr->exec_count;
        execbuf.batch_start_offset = 0;
        execbuf.batch_len = used;
        execbuf.cliprects_ptr = 0;
        execbuf.num_cliprects = 0;
        execbuf.DR1 = 0;
        execbuf.DR4 = 0;
        execbuf.flags = flags;
        if (ctx == NULL)
                i915_execbuffer2_set_context_id(execbuf, 0);
        else
                i915_execbuffer2_set_context_id(execbuf, ctx->ctx_id);
        execbuf.rsvd2 = 0;
        if (in_fence != -1) {
                execbuf.rsvd2 = in_fence;
                execbuf.flags |= I915_EXEC_FENCE_IN;
        }
        if (out_fence != NULL) {
                *out_fence = -1;
                execbuf.flags |= I915_EXEC_FENCE_OUT;
        }

        if (bufmgr->no_exec)
                goto skip_execution;

        ret = drmIoctl(bufmgr->fd,
                       DRM_IOCTL_I915_GEM_EXECBUFFER2_WR,
                       &execbuf);
        if (ret != 0) {
                ret = -errno;
                if (ret == -ENOSPC) {
                        DBG("Execbuffer fails to pin. "
                            "Estimate: %u. Actual: %u. Available: %u\n",
                            drm_bacon_gem_estimate_batch_space(bufmgr->exec_bos,
                                                               bufmgr->exec_count),
                            drm_bacon_gem_compute_batch_space(bufmgr->exec_bos,
                                                              bufmgr->exec_count),
                            (unsigned int) bufmgr->gtt_size);
                }
        }
        drm_bacon_update_buffer_offsets2(bufmgr);

        if (ret == 0 && out_fence != NULL)
                *out_fence = execbuf.rsvd2 >> 32;

skip_execution:
        if (INTEL_DEBUG & DEBUG_BUFMGR)
                drm_bacon_gem_dump_validation_list(bufmgr);

        for (i = 0; i < bufmgr->exec_count; i++) {
                drm_bacon_bo_gem *bo_gem = to_bo_gem(bufmgr->exec_bos[i]);

                bo_gem->idle = false;

                /* Disconnect the buffer from the validate list */
                bo_gem->validate_index = -1;
                bufmgr->exec_bos[i] = NULL;
        }
        bufmgr->exec_count = 0;
        pthread_mutex_unlock(&bufmgr->lock);

        return ret;
}

int
drm_bacon_bo_exec(drm_bacon_bo *bo, int used)
{
        return do_exec2(bo, used, NULL, -1, NULL, I915_EXEC_RENDER);
}

int
drm_bacon_bo_mrb_exec(drm_bacon_bo *bo, int used, unsigned int flags)
{
        return do_exec2(bo, used, NULL, -1, NULL, flags);
}

int
drm_bacon_gem_bo_context_exec(drm_bacon_bo *bo, drm_bacon_context *ctx,
                              int used, unsigned int flags)
{
        return do_exec2(bo, used, ctx, -1, NULL, flags);
}

int
drm_bacon_gem_bo_fence_exec(drm_bacon_bo *bo,
                            drm_bacon_context *ctx,
                            int used,
                            int in_fence,
                            int *out_fence,
                            unsigned int flags)
{
        return do_exec2(bo, used, ctx, in_fence, out_fence, flags);
}

static int
drm_bacon_gem_bo_set_tiling_internal(drm_bacon_bo *bo,
                                     uint32_t tiling_mode,
                                     uint32_t stride)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        struct drm_i915_gem_set_tiling set_tiling;
        int ret;

        if (bo_gem->global_name == 0 &&
            tiling_mode == bo_gem->tiling_mode &&
            stride == bo_gem->stride)
                return 0;

        memset(&set_tiling, 0, sizeof(set_tiling));
        do {
                /* set_tiling is slightly broken and overwrites the
                 * input on the error path, so we have to open code
                 * rmIoctl.
                 */
                set_tiling.handle = bo_gem->gem_handle;
                set_tiling.tiling_mode = tiling_mode;
                set_tiling.stride = stride;

                ret = ioctl(bufmgr->fd,
                            DRM_IOCTL_I915_GEM_SET_TILING,
                            &set_tiling);
        } while (ret == -1 && (errno == EINTR || errno == EAGAIN));
        if (ret == -1)
                return -errno;

        bo_gem->tiling_mode = set_tiling.tiling_mode;
        bo_gem->swizzle_mode = set_tiling.swizzle_mode;
        bo_gem->stride = set_tiling.stride;
        return 0;
}

int
drm_bacon_bo_set_tiling(drm_bacon_bo *bo, uint32_t * tiling_mode,
                        uint32_t stride)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        int ret;

        /* Linear buffers have no stride. By ensuring that we only ever use
         * stride 0 with linear buffers, we simplify our code.
         */
        if (*tiling_mode == I915_TILING_NONE)
                stride = 0;

        ret = drm_bacon_gem_bo_set_tiling_internal(bo, *tiling_mode, stride);
        if (ret == 0)
                drm_bacon_bo_gem_set_in_aperture_size(bufmgr, bo_gem, 0);

        *tiling_mode = bo_gem->tiling_mode;
        return ret;
}

int
drm_bacon_bo_get_tiling(drm_bacon_bo *bo, uint32_t * tiling_mode,
                        uint32_t *swizzle_mode)
{
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;

        *tiling_mode = bo_gem->tiling_mode;
        *swizzle_mode = bo_gem->swizzle_mode;
        return 0;
}

drm_bacon_bo *
drm_bacon_bo_gem_create_from_prime(drm_bacon_bufmgr *bufmgr, int prime_fd, int size)
{
        int ret;
        uint32_t handle;
        drm_bacon_bo_gem *bo_gem;
        struct drm_i915_gem_get_tiling get_tiling;

        pthread_mutex_lock(&bufmgr->lock);
        ret = drmPrimeFDToHandle(bufmgr->fd, prime_fd, &handle);
        if (ret) {
                DBG("create_from_prime: failed to obtain handle from fd: %s\n", strerror(errno));
                pthread_mutex_unlock(&bufmgr->lock);
                return NULL;
        }

        /*
         * See if the kernel has already returned this buffer to us. Just as
         * for named buffers, we must not create two bo's pointing at the same
         * kernel object
         */
        bo_gem = hash_find_bo(bufmgr->handle_table, handle);
        if (bo_gem) {
                drm_bacon_bo_reference(&bo_gem->bo);
                goto out;
        }

        bo_gem = calloc(1, sizeof(*bo_gem));
        if (!bo_gem)
                goto out;

        p_atomic_set(&bo_gem->refcount, 1);
        list_inithead(&bo_gem->vma_list);

        /* Determine size of bo.  The fd-to-handle ioctl really should
         * return the size, but it doesn't.  If we have kernel 3.12 or
         * later, we can lseek on the prime fd to get the size.  Older
         * kernels will just fail, in which case we fall back to the
         * provided (estimated or guess size). */
        ret = lseek(prime_fd, 0, SEEK_END);
        if (ret != -1)
                bo_gem->bo.size = ret;
        else
                bo_gem->bo.size = size;

        bo_gem->bo.handle = handle;
        bo_gem->bo.bufmgr = bufmgr;

        bo_gem->gem_handle = handle;
        _mesa_hash_table_insert(bufmgr->handle_table,
                                &bo_gem->gem_handle, bo_gem);

        bo_gem->name = "prime";
        bo_gem->validate_index = -1;
        bo_gem->used_as_reloc_target = false;
        bo_gem->has_error = false;
        bo_gem->reusable = false;

        memclear(get_tiling);
        get_tiling.handle = bo_gem->gem_handle;
        if (drmIoctl(bufmgr->fd,
                     DRM_IOCTL_I915_GEM_GET_TILING,
                     &get_tiling))
                goto err;

        bo_gem->tiling_mode = get_tiling.tiling_mode;
        bo_gem->swizzle_mode = get_tiling.swizzle_mode;
        /* XXX stride is unknown */
        drm_bacon_bo_gem_set_in_aperture_size(bufmgr, bo_gem, 0);

out:
        pthread_mutex_unlock(&bufmgr->lock);
        return &bo_gem->bo;

err:
        drm_bacon_gem_bo_free(&bo_gem->bo);
        pthread_mutex_unlock(&bufmgr->lock);
        return NULL;
}

int
drm_bacon_bo_gem_export_to_prime(drm_bacon_bo *bo, int *prime_fd)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;

        if (drmPrimeHandleToFD(bufmgr->fd, bo_gem->gem_handle,
                               DRM_CLOEXEC, prime_fd) != 0)
                return -errno;

        bo_gem->reusable = false;

        return 0;
}

int
drm_bacon_bo_flink(drm_bacon_bo *bo, uint32_t *name)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;

        if (!bo_gem->global_name) {
                struct drm_gem_flink flink;

                memclear(flink);
                flink.handle = bo_gem->gem_handle;
                if (drmIoctl(bufmgr->fd, DRM_IOCTL_GEM_FLINK, &flink))
                        return -errno;

                pthread_mutex_lock(&bufmgr->lock);
                if (!bo_gem->global_name) {
                        bo_gem->global_name = flink.name;
                        bo_gem->reusable = false;

                        _mesa_hash_table_insert(bufmgr->name_table,
                                                &bo_gem->global_name, bo_gem);
                }
                pthread_mutex_unlock(&bufmgr->lock);
        }

        *name = bo_gem->global_name;
        return 0;
}

/**
 * Enables unlimited caching of buffer objects for reuse.
 *
 * This is potentially very memory expensive, as the cache at each bucket
 * size is only bounded by how many buffers of that size we've managed to have
 * in flight at once.
 */
void
drm_bacon_bufmgr_gem_enable_reuse(drm_bacon_bufmgr *bufmgr)
{
        bufmgr->bo_reuse = true;
}

/**
 * Return the additional aperture space required by the tree of buffer objects
 * rooted at bo.
 */
static int
drm_bacon_gem_bo_get_aperture_space(drm_bacon_bo *bo)
{
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        int i;
        int total = 0;

        if (bo == NULL || bo_gem->included_in_check_aperture)
                return 0;

        total += bo->size;
        bo_gem->included_in_check_aperture = true;

        for (i = 0; i < bo_gem->reloc_count; i++)
                total +=
                    drm_bacon_gem_bo_get_aperture_space(bo_gem->reloc_bos[i]);

        return total;
}

/**
 * Clear the flag set by drm_bacon_gem_bo_get_aperture_space() so we're ready
 * for the next drm_bacon_bufmgr_check_aperture_space() call.
 */
static void
drm_bacon_gem_bo_clear_aperture_space_flag(drm_bacon_bo *bo)
{
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        int i;

        if (bo == NULL || !bo_gem->included_in_check_aperture)
                return;

        bo_gem->included_in_check_aperture = false;

        for (i = 0; i < bo_gem->reloc_count; i++)
                drm_bacon_gem_bo_clear_aperture_space_flag(bo_gem->reloc_bos[i]);
}

/**
 * Return a conservative estimate for the amount of aperture required
 * for a collection of buffers. This may double-count some buffers.
 */
static unsigned int
drm_bacon_gem_estimate_batch_space(drm_bacon_bo **bo_array, int count)
{
        int i;
        unsigned int total = 0;

        for (i = 0; i < count; i++) {
                drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo_array[i];
                if (bo_gem != NULL)
                        total += bo_gem->reloc_tree_size;
        }
        return total;
}

/**
 * Return the amount of aperture needed for a collection of buffers.
 * This avoids double counting any buffers, at the cost of looking
 * at every buffer in the set.
 */
static unsigned int
drm_bacon_gem_compute_batch_space(drm_bacon_bo **bo_array, int count)
{
        int i;
        unsigned int total = 0;

        for (i = 0; i < count; i++) {
                total += drm_bacon_gem_bo_get_aperture_space(bo_array[i]);
                /* For the first buffer object in the array, we get an
                 * accurate count back for its reloc_tree size (since nothing
                 * had been flagged as being counted yet).  We can save that
                 * value out as a more conservative reloc_tree_size that
                 * avoids double-counting target buffers.  Since the first
                 * buffer happens to usually be the batch buffer in our
                 * callers, this can pull us back from doing the tree
                 * walk on every new batch emit.
                 */
                if (i == 0) {
                        drm_bacon_bo_gem *bo_gem =
                            (drm_bacon_bo_gem *) bo_array[i];
                        bo_gem->reloc_tree_size = total;
                }
        }

        for (i = 0; i < count; i++)
                drm_bacon_gem_bo_clear_aperture_space_flag(bo_array[i]);
        return total;
}

/**
 * Return -1 if the batchbuffer should be flushed before attempting to
 * emit rendering referencing the buffers pointed to by bo_array.
 *
 * This is required because if we try to emit a batchbuffer with relocations
 * to a tree of buffers that won't simultaneously fit in the aperture,
 * the rendering will return an error at a point where the software is not
 * prepared to recover from it.
 *
 * However, we also want to emit the batchbuffer significantly before we reach
 * the limit, as a series of batchbuffers each of which references buffers
 * covering almost all of the aperture means that at each emit we end up
 * waiting to evict a buffer from the last rendering, and we get synchronous
 * performance.  By emitting smaller batchbuffers, we eat some CPU overhead to
 * get better parallelism.
 */
int
drm_bacon_bufmgr_check_aperture_space(drm_bacon_bo **bo_array, int count)
{
        drm_bacon_bufmgr *bufmgr = bo_array[0]->bufmgr;
        unsigned int total = 0;
        unsigned int threshold = bufmgr->gtt_size * 3 / 4;

        total = drm_bacon_gem_estimate_batch_space(bo_array, count);

        if (total > threshold)
                total = drm_bacon_gem_compute_batch_space(bo_array, count);

        if (total > threshold) {
                DBG("check_space: overflowed available aperture, "
                    "%dkb vs %dkb\n",
                    total / 1024, (int)bufmgr->gtt_size / 1024);
                return -ENOSPC;
        } else {
                DBG("drm_check_space: total %dkb vs bufgr %dkb\n", total / 1024,
                    (int)bufmgr->gtt_size / 1024);
                return 0;
        }
}

/*
 * Disable buffer reuse for objects which are shared with the kernel
 * as scanout buffers
 */
int
drm_bacon_bo_disable_reuse(drm_bacon_bo *bo)
{
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;

        bo_gem->reusable = false;
        return 0;
}

int
drm_bacon_bo_is_reusable(drm_bacon_bo *bo)
{
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;

        return bo_gem->reusable;
}

static int
_drm_bacon_gem_bo_references(drm_bacon_bo *bo, drm_bacon_bo *target_bo)
{
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;
        int i;

        for (i = 0; i < bo_gem->reloc_count; i++) {
                if (bo_gem->reloc_bos[i] == target_bo)
                        return 1;
                if (bo == bo_gem->reloc_bos[i])
                        continue;
                if (_drm_bacon_gem_bo_references(bo_gem->reloc_bos[i],
                                                target_bo))
                        return 1;
        }

        return 0;
}

/** Return true if target_bo is referenced by bo's relocation tree. */
int
drm_bacon_bo_references(drm_bacon_bo *bo, drm_bacon_bo *target_bo)
{
        drm_bacon_bo_gem *target_bo_gem = (drm_bacon_bo_gem *) target_bo;

        if (bo == NULL || target_bo == NULL)
                return 0;
        if (target_bo_gem->used_as_reloc_target)
                return _drm_bacon_gem_bo_references(bo, target_bo);
        return 0;
}

static void
add_bucket(drm_bacon_bufmgr *bufmgr, int size)
{
        unsigned int i = bufmgr->num_buckets;

        assert(i < ARRAY_SIZE(bufmgr->cache_bucket));

        list_inithead(&bufmgr->cache_bucket[i].head);
        bufmgr->cache_bucket[i].size = size;
        bufmgr->num_buckets++;
}

static void
init_cache_buckets(drm_bacon_bufmgr *bufmgr)
{
        unsigned long size, cache_max_size = 64 * 1024 * 1024;

        /* OK, so power of two buckets was too wasteful of memory.
         * Give 3 other sizes between each power of two, to hopefully
         * cover things accurately enough.  (The alternative is
         * probably to just go for exact matching of sizes, and assume
         * that for things like composited window resize the tiled
         * width/height alignment and rounding of sizes to pages will
         * get us useful cache hit rates anyway)
         */
        add_bucket(bufmgr, 4096);
        add_bucket(bufmgr, 4096 * 2);
        add_bucket(bufmgr, 4096 * 3);

        /* Initialize the linked lists for BO reuse cache. */
        for (size = 4 * 4096; size <= cache_max_size; size *= 2) {
                add_bucket(bufmgr, size);

                add_bucket(bufmgr, size + size * 1 / 4);
                add_bucket(bufmgr, size + size * 2 / 4);
                add_bucket(bufmgr, size + size * 3 / 4);
        }
}

void
drm_bacon_bufmgr_gem_set_vma_cache_size(drm_bacon_bufmgr *bufmgr, int limit)
{
        bufmgr->vma_max = limit;

        drm_bacon_gem_bo_purge_vma_cache(bufmgr);
}

drm_bacon_context *
drm_bacon_gem_context_create(drm_bacon_bufmgr *bufmgr)
{
        struct drm_i915_gem_context_create create;
        drm_bacon_context *context = NULL;
        int ret;

        context = calloc(1, sizeof(*context));
        if (!context)
                return NULL;

        memclear(create);
        ret = drmIoctl(bufmgr->fd, DRM_IOCTL_I915_GEM_CONTEXT_CREATE, &create);
        if (ret != 0) {
                DBG("DRM_IOCTL_I915_GEM_CONTEXT_CREATE failed: %s\n",
                    strerror(errno));
                free(context);
                return NULL;
        }

        context->ctx_id = create.ctx_id;
        context->bufmgr = bufmgr;

        return context;
}

int
drm_bacon_gem_context_get_id(drm_bacon_context *ctx, uint32_t *ctx_id)
{
        if (ctx == NULL)
                return -EINVAL;

        *ctx_id = ctx->ctx_id;

        return 0;
}

void
drm_bacon_gem_context_destroy(drm_bacon_context *ctx)
{
        struct drm_i915_gem_context_destroy destroy;
        int ret;

        if (ctx == NULL)
                return;

        memclear(destroy);

        destroy.ctx_id = ctx->ctx_id;
        ret = drmIoctl(ctx->bufmgr->fd, DRM_IOCTL_I915_GEM_CONTEXT_DESTROY,
                       &destroy);
        if (ret != 0)
                fprintf(stderr, "DRM_IOCTL_I915_GEM_CONTEXT_DESTROY failed: %s\n",
                        strerror(errno));

        free(ctx);
}

int
drm_bacon_get_reset_stats(drm_bacon_context *ctx,
                          uint32_t *reset_count,
                          uint32_t *active,
                          uint32_t *pending)
{
        struct drm_i915_reset_stats stats;
        int ret;

        if (ctx == NULL)
                return -EINVAL;

        memclear(stats);

        stats.ctx_id = ctx->ctx_id;
        ret = drmIoctl(ctx->bufmgr->fd,
                       DRM_IOCTL_I915_GET_RESET_STATS,
                       &stats);
        if (ret == 0) {
                if (reset_count != NULL)
                        *reset_count = stats.reset_count;

                if (active != NULL)
                        *active = stats.batch_active;

                if (pending != NULL)
                        *pending = stats.batch_pending;
        }

        return ret;
}

int
drm_bacon_reg_read(drm_bacon_bufmgr *bufmgr,
                   uint32_t offset,
                   uint64_t *result)
{
        struct drm_i915_reg_read reg_read;
        int ret;

        memclear(reg_read);
        reg_read.offset = offset;

        ret = drmIoctl(bufmgr->fd, DRM_IOCTL_I915_REG_READ, &reg_read);

        *result = reg_read.val;
        return ret;
}

static pthread_mutex_t bufmgr_list_mutex = PTHREAD_MUTEX_INITIALIZER;
static struct list_head bufmgr_list = { &bufmgr_list, &bufmgr_list };

static drm_bacon_bufmgr *
drm_bacon_bufmgr_gem_find(int fd)
{
        list_for_each_entry(drm_bacon_bufmgr,
                            bufmgr, &bufmgr_list, managers) {
                if (bufmgr->fd == fd) {
                        p_atomic_inc(&bufmgr->refcount);
                        return bufmgr;
                }
        }

        return NULL;
}

void
drm_bacon_bufmgr_destroy(drm_bacon_bufmgr *bufmgr)
{
        if (atomic_add_unless(&bufmgr->refcount, -1, 1)) {
                pthread_mutex_lock(&bufmgr_list_mutex);

                if (p_atomic_dec_zero(&bufmgr->refcount)) {
                        list_del(&bufmgr->managers);
                        drm_bacon_bufmgr_gem_destroy(bufmgr);
                }

                pthread_mutex_unlock(&bufmgr_list_mutex);
        }
}

void *drm_bacon_gem_bo_map__gtt(drm_bacon_bo *bo)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;

        if (bo_gem->gtt_virtual)
                return bo_gem->gtt_virtual;

        pthread_mutex_lock(&bufmgr->lock);
        if (bo_gem->gtt_virtual == NULL) {
                struct drm_i915_gem_mmap_gtt mmap_arg;
                void *ptr;

                DBG("bo_map_gtt: mmap %d (%s), map_count=%d\n",
                    bo_gem->gem_handle, bo_gem->name, bo_gem->map_count);

                if (bo_gem->map_count++ == 0)
                        drm_bacon_gem_bo_open_vma(bufmgr, bo_gem);

                memclear(mmap_arg);
                mmap_arg.handle = bo_gem->gem_handle;

                /* Get the fake offset back... */
                ptr = MAP_FAILED;
                if (drmIoctl(bufmgr->fd,
                             DRM_IOCTL_I915_GEM_MMAP_GTT,
                             &mmap_arg) == 0) {
                        /* and mmap it */
                        ptr = drm_mmap(0, bo->size, PROT_READ | PROT_WRITE,
                                       MAP_SHARED, bufmgr->fd,
                                       mmap_arg.offset);
                }
                if (ptr == MAP_FAILED) {
                        if (--bo_gem->map_count == 0)
                                drm_bacon_gem_bo_close_vma(bufmgr, bo_gem);
                        ptr = NULL;
                }

                bo_gem->gtt_virtual = ptr;
        }
        pthread_mutex_unlock(&bufmgr->lock);

        return bo_gem->gtt_virtual;
}

void *drm_bacon_gem_bo_map__cpu(drm_bacon_bo *bo)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;

        if (bo_gem->mem_virtual)
                return bo_gem->mem_virtual;

        pthread_mutex_lock(&bufmgr->lock);
        if (!bo_gem->mem_virtual) {
                struct drm_i915_gem_mmap mmap_arg;

                if (bo_gem->map_count++ == 0)
                        drm_bacon_gem_bo_open_vma(bufmgr, bo_gem);

                DBG("bo_map: %d (%s), map_count=%d\n",
                    bo_gem->gem_handle, bo_gem->name, bo_gem->map_count);

                memclear(mmap_arg);
                mmap_arg.handle = bo_gem->gem_handle;
                mmap_arg.size = bo->size;
                if (drmIoctl(bufmgr->fd,
                             DRM_IOCTL_I915_GEM_MMAP,
                             &mmap_arg)) {
                        DBG("%s:%d: Error mapping buffer %d (%s): %s .\n",
                            __FILE__, __LINE__, bo_gem->gem_handle,
                            bo_gem->name, strerror(errno));
                        if (--bo_gem->map_count == 0)
                                drm_bacon_gem_bo_close_vma(bufmgr, bo_gem);
                } else {
                        VG(VALGRIND_MALLOCLIKE_BLOCK(mmap_arg.addr_ptr, mmap_arg.size, 0, 1));
                        bo_gem->mem_virtual = (void *)(uintptr_t) mmap_arg.addr_ptr;
                }
        }
        pthread_mutex_unlock(&bufmgr->lock);

        return bo_gem->mem_virtual;
}

void *drm_bacon_gem_bo_map__wc(drm_bacon_bo *bo)
{
        drm_bacon_bufmgr *bufmgr = bo->bufmgr;
        drm_bacon_bo_gem *bo_gem = (drm_bacon_bo_gem *) bo;

        if (bo_gem->wc_virtual)
                return bo_gem->wc_virtual;

        pthread_mutex_lock(&bufmgr->lock);
        if (!bo_gem->wc_virtual) {
                struct drm_i915_gem_mmap mmap_arg;

                if (bo_gem->map_count++ == 0)
                        drm_bacon_gem_bo_open_vma(bufmgr, bo_gem);

                DBG("bo_map: %d (%s), map_count=%d\n",
                    bo_gem->gem_handle, bo_gem->name, bo_gem->map_count);

                memclear(mmap_arg);
                mmap_arg.handle = bo_gem->gem_handle;
                mmap_arg.size = bo->size;
                mmap_arg.flags = I915_MMAP_WC;
                if (drmIoctl(bufmgr->fd,
                             DRM_IOCTL_I915_GEM_MMAP,
                             &mmap_arg)) {
                        DBG("%s:%d: Error mapping buffer %d (%s): %s .\n",
                            __FILE__, __LINE__, bo_gem->gem_handle,
                            bo_gem->name, strerror(errno));
                        if (--bo_gem->map_count == 0)
                                drm_bacon_gem_bo_close_vma(bufmgr, bo_gem);
                } else {
                        VG(VALGRIND_MALLOCLIKE_BLOCK(mmap_arg.addr_ptr, mmap_arg.size, 0, 1));
                        bo_gem->wc_virtual = (void *)(uintptr_t) mmap_arg.addr_ptr;
                }
        }
        pthread_mutex_unlock(&bufmgr->lock);

        return bo_gem->wc_virtual;
}

/**
 * Initializes the GEM buffer manager, which uses the kernel to allocate, map,
 * and manage map buffer objections.
 *
 * \param fd File descriptor of the opened DRM device.
 */
drm_bacon_bufmgr *
drm_bacon_bufmgr_gem_init(struct gen_device_info *devinfo,
                          int fd, int batch_size)
{
        drm_bacon_bufmgr *bufmgr;
        struct drm_i915_gem_get_aperture aperture;

        pthread_mutex_lock(&bufmgr_list_mutex);

        bufmgr = drm_bacon_bufmgr_gem_find(fd);
        if (bufmgr)
                goto exit;

        bufmgr = calloc(1, sizeof(*bufmgr));
        if (bufmgr == NULL)
                goto exit;

        bufmgr->fd = fd;
        p_atomic_set(&bufmgr->refcount, 1);

        if (pthread_mutex_init(&bufmgr->lock, NULL) != 0) {
                free(bufmgr);
                bufmgr = NULL;
                goto exit;
        }

        memclear(aperture);
        drmIoctl(bufmgr->fd, DRM_IOCTL_I915_GEM_GET_APERTURE, &aperture);
        bufmgr->gtt_size = aperture.aper_available_size;

        bufmgr->has_llc = devinfo->has_llc;

        /* Let's go with one relocation per every 2 dwords (but round down a bit
         * since a power of two will mean an extra page allocation for the reloc
         * buffer).
         *
         * Every 4 was too few for the blender benchmark.
         */
        bufmgr->max_relocs = batch_size / sizeof(uint32_t) / 2 - 2;

        init_cache_buckets(bufmgr);

        list_inithead(&bufmgr->vma_cache);
        bufmgr->vma_max = -1; /* unlimited by default */

        list_add(&bufmgr->managers, &bufmgr_list);

        bufmgr->name_table =
                _mesa_hash_table_create(NULL, key_hash_uint, key_uint_equal);
        bufmgr->handle_table =
                _mesa_hash_table_create(NULL, key_hash_uint, key_uint_equal);

exit:
        pthread_mutex_unlock(&bufmgr_list_mutex);

        return bufmgr;
}