#define drm_for_each_lessee(lessee, lessor) \
        list_for_each_entry((lessee), &(lessor)->lessees, lessee_list)
 
+static uint64_t drm_lease_idr_object;
+
 /**
  * drm_lease_owner - return ancestor owner drm_master
  * @master: drm_master somewhere within tree of lessees and lessors
        _drm_lease_revoke(top);
        mutex_unlock(&top->dev->mode_config.idr_mutex);
 }
+
+static int validate_lease(struct drm_device *dev,
+                         struct drm_file *lessor_priv,
+                         int object_count,
+                         struct drm_mode_object **objects)
+{
+       int o;
+       int has_crtc = -1;
+       int has_connector = -1;
+       int has_plane = -1;
+
+       /* we want to confirm that there is at least one crtc, plane
+          connector object. */
+
+       for (o = 0; o < object_count; o++) {
+               if (objects[o]->type == DRM_MODE_OBJECT_CRTC && has_crtc == -1) {
+                       has_crtc = o;
+               }
+               if (objects[o]->type == DRM_MODE_OBJECT_CONNECTOR && has_connector == -1)
+                       has_connector = o;
+
+               if (lessor_priv->universal_planes) {
+                       if (objects[o]->type == DRM_MODE_OBJECT_PLANE && has_plane == -1)
+                               has_plane = o;
+               }
+       }
+       if (has_crtc == -1 || has_connector == -1)
+               return -EINVAL;
+       if (lessor_priv->universal_planes && has_plane == -1)
+               return -EINVAL;
+       return 0;
+}
+
+static int fill_object_idr(struct drm_device *dev,
+                          struct drm_file *lessor_priv,
+                          struct idr *leases,
+                          int object_count,
+                          u32 *object_ids)
+{
+       struct drm_mode_object **objects;
+       u32 o;
+       int ret;
+       objects = kcalloc(object_count, sizeof(struct drm_mode_object *),
+                         GFP_KERNEL);
+       if (!objects)
+               return -ENOMEM;
+
+       /* step one - get references to all the mode objects
+          and check for validity. */
+       for (o = 0; o < object_count; o++) {
+               if ((int) object_ids[o] < 0) {
+                       ret = -EINVAL;
+                       goto out_free_objects;
+               }
+
+               objects[o] = drm_mode_object_find(dev, lessor_priv,
+                                                 object_ids[o],
+                                                 DRM_MODE_OBJECT_ANY);
+               if (!objects[o]) {
+                       ret = -ENOENT;
+                       goto out_free_objects;
+               }
+
+               if (!drm_mode_object_lease_required(objects[o]->type)) {
+                       ret = -EINVAL;
+                       goto out_free_objects;
+               }
+       }
+
+       ret = validate_lease(dev, lessor_priv, object_count, objects);
+       if (ret)
+               goto out_free_objects;
+
+       /* add their IDs to the lease request - taking into account
+          universal planes */
+       for (o = 0; o < object_count; o++) {
+               struct drm_mode_object *obj = objects[o];
+               u32 object_id = objects[o]->id;
+               DRM_DEBUG_LEASE("Adding object %d to lease\n", object_id);
+
+               /*
+                * We're using an IDR to hold the set of leased
+                * objects, but we don't need to point at the object's
+                * data structure from the lease as the main crtc_idr
+                * will be used to actually find that. Instead, all we
+                * really want is a 'leased/not-leased' result, for
+                * which any non-NULL pointer will work fine.
+                */
+               ret = idr_alloc(leases, &drm_lease_idr_object , object_id, object_id + 1, GFP_KERNEL);
+               if (ret < 0) {
+                       DRM_DEBUG_LEASE("Object %d cannot be inserted into leases (%d)\n",
+                                       object_id, ret);
+                       goto out_free_objects;
+               }
+               if (obj->type == DRM_MODE_OBJECT_CRTC && !lessor_priv->universal_planes) {
+                       struct drm_crtc *crtc = obj_to_crtc(obj);
+                       ret = idr_alloc(leases, &drm_lease_idr_object, crtc->primary->base.id, crtc->primary->base.id + 1, GFP_KERNEL);
+                       if (ret < 0) {
+                               DRM_DEBUG_LEASE("Object primary plane %d cannot be inserted into leases (%d)\n",
+                                               object_id, ret);
+                               goto out_free_objects;
+                       }
+                       if (crtc->cursor) {
+                               ret = idr_alloc(leases, &drm_lease_idr_object, crtc->cursor->base.id, crtc->cursor->base.id + 1, GFP_KERNEL);
+                               if (ret < 0) {
+                                       DRM_DEBUG_LEASE("Object cursor plane %d cannot be inserted into leases (%d)\n",
+                                                       object_id, ret);
+                                       goto out_free_objects;
+                               }
+                       }
+               }
+       }
+
+       ret = 0;
+out_free_objects:
+       for (o = 0; o < object_count; o++) {
+               if (objects[o])
+                       drm_mode_object_put(objects[o]);
+       }
+       kfree(objects);
+       return ret;
+}
+
+/**
+ * drm_mode_create_lease_ioctl - create a new lease
+ * @dev: the drm device
+ * @data: pointer to struct drm_mode_create_lease
+ * @file_priv: the file being manipulated
+ *
+ * The master associated with the specified file will have a lease
+ * created containing the objects specified in the ioctl structure.
+ * A file descriptor will be allocated for that and returned to the
+ * application.
+ */
+int drm_mode_create_lease_ioctl(struct drm_device *dev,
+                               void *data, struct drm_file *lessor_priv)
+{
+       struct drm_mode_create_lease *cl = data;
+       size_t object_count;
+       int ret = 0;
+       struct idr leases;
+       struct drm_master *lessor = lessor_priv->master;
+       struct drm_master *lessee = NULL;
+       struct file *lessee_file = NULL;
+       struct file *lessor_file = lessor_priv->filp;
+       struct drm_file *lessee_priv;
+       int fd = -1;
+       uint32_t *object_ids;
+
+       /* Can't lease without MODESET */
+       if (!drm_core_check_feature(dev, DRIVER_MODESET))
+               return -EINVAL;
+
+       /* Do not allow sub-leases */
+       if (lessor->lessor)
+               return -EINVAL;
+
+       /* need some objects */
+       if (cl->object_count == 0)
+               return -EINVAL;
+
+       if (cl->flags && (cl->flags & ~(O_CLOEXEC | O_NONBLOCK)))
+               return -EINVAL;
+
+       object_count = cl->object_count;
+
+       object_ids = memdup_user(u64_to_user_ptr(cl->object_ids), object_count * sizeof(__u32));
+       if (IS_ERR(object_ids))
+               return PTR_ERR(object_ids);
+
+       idr_init(&leases);
+
+       /* fill and validate the object idr */
+       ret = fill_object_idr(dev, lessor_priv, &leases,
+                             object_count, object_ids);
+       kfree(object_ids);
+       if (ret) {
+               idr_destroy(&leases);
+               return ret;
+       }
+
+       /* Allocate a file descriptor for the lease */
+       fd = get_unused_fd_flags(cl->flags & (O_CLOEXEC | O_NONBLOCK));
+       if (fd < 0) {
+               idr_destroy(&leases);
+               return fd;
+       }
+
+       DRM_DEBUG_LEASE("Creating lease\n");
+       lessee = drm_lease_create(lessor, &leases);
+
+       if (IS_ERR(lessee)) {
+               ret = PTR_ERR(lessee);
+               goto out_leases;
+       }
+
+       /* Clone the lessor file to create a new file for us */
+       DRM_DEBUG_LEASE("Allocating lease file\n");
+       path_get(&lessor_file->f_path);
+       lessee_file = alloc_file(&lessor_file->f_path,
+                                lessor_file->f_mode,
+                                fops_get(lessor_file->f_inode->i_fop));
+
+       if (IS_ERR(lessee_file)) {
+               ret = PTR_ERR(lessee_file);
+               goto out_lessee;
+       }
+
+       /* Initialize the new file for DRM */
+       DRM_DEBUG_LEASE("Initializing the file with %p\n", lessee_file->f_op->open);
+       ret = lessee_file->f_op->open(lessee_file->f_inode, lessee_file);
+       if (ret)
+               goto out_lessee_file;
+
+       lessee_priv = lessee_file->private_data;
+
+       /* Change the file to a master one */
+       drm_master_put(&lessee_priv->master);
+       lessee_priv->master = lessee;
+       lessee_priv->is_master = 1;
+       lessee_priv->authenticated = 1;
+
+       /* Hook up the fd */
+       fd_install(fd, lessee_file);
+
+       /* Pass fd back to userspace */
+       DRM_DEBUG_LEASE("Returning fd %d id %d\n", fd, lessee->lessee_id);
+       cl->fd = fd;
+       cl->lessee_id = lessee->lessee_id;
+
+       DRM_DEBUG_LEASE("drm_mode_create_lease_ioctl succeeded\n");
+       return 0;
+
+out_lessee_file:
+       fput(lessee_file);
+
+out_lessee:
+       drm_master_put(&lessee);
+
+out_leases:
+       put_unused_fd(fd);
+       idr_destroy(&leases);
+
+       DRM_DEBUG_LEASE("drm_mode_create_lease_ioctl failed: %d\n", ret);
+       return ret;
+}
+
+/**
+ * drm_mode_list_lessees_ioctl - list lessee ids
+ * @dev: the drm device
+ * @data: pointer to struct drm_mode_list_lessees
+ * @lessor_priv: the file being manipulated
+ *
+ * Starting from the master associated with the specified file,
+ * the master with the provided lessee_id is found, and then
+ * an array of lessee ids associated with leases from that master
+ * are returned.
+ */
+
+int drm_mode_list_lessees_ioctl(struct drm_device *dev,
+                              void *data, struct drm_file *lessor_priv)
+{
+       struct drm_mode_list_lessees *arg = data;
+       __u32 __user *lessee_ids = (__u32 __user *) (uintptr_t) (arg->lessees_ptr);
+       __u32 count_lessees = arg->count_lessees;
+       struct drm_master *lessor = lessor_priv->master, *lessee;
+       int count;
+       int ret = 0;
+
+       if (arg->pad)
+               return -EINVAL;
+
+       /* Can't lease without MODESET */
+       if (!drm_core_check_feature(dev, DRIVER_MODESET))
+               return -EINVAL;
+
+       DRM_DEBUG_LEASE("List lessees for %d\n", lessor->lessee_id);
+
+       mutex_lock(&dev->mode_config.idr_mutex);
+
+       count = 0;
+       drm_for_each_lessee(lessee, lessor) {
+               /* Only list un-revoked leases */
+               if (!idr_is_empty(&lessee->leases)) {
+                       if (count_lessees > count) {
+                               DRM_DEBUG_LEASE("Add lessee %d\n", lessee->lessee_id);
+                               ret = put_user(lessee->lessee_id, lessee_ids + count);
+                               if (ret)
+                                       break;
+                       }
+                       count++;
+               }
+       }
+
+       DRM_DEBUG_LEASE("Lessor leases to %d\n", count);
+       if (ret == 0)
+               arg->count_lessees = count;
+
+       mutex_unlock(&dev->mode_config.idr_mutex);
+
+       return ret;
+}
+
+/**
+ * drm_mode_get_lease_ioctl - list leased objects
+ * @dev: the drm device
+ * @data: pointer to struct drm_mode_get_lease
+ * @file_priv: the file being manipulated
+ *
+ * Return the list of leased objects for the specified lessee
+ */
+
+int drm_mode_get_lease_ioctl(struct drm_device *dev,
+                            void *data, struct drm_file *lessee_priv)
+{
+       struct drm_mode_get_lease *arg = data;
+       __u32 __user *object_ids = (__u32 __user *) (uintptr_t) (arg->objects_ptr);
+       __u32 count_objects = arg->count_objects;
+       struct drm_master *lessee = lessee_priv->master;
+       struct idr *object_idr;
+       int count;
+       void *entry;
+       int object;
+       int ret = 0;
+
+       if (arg->pad)
+               return -EINVAL;
+
+       /* Can't lease without MODESET */
+       if (!drm_core_check_feature(dev, DRIVER_MODESET))
+               return -EINVAL;
+
+       DRM_DEBUG_LEASE("get lease for %d\n", lessee->lessee_id);
+
+       mutex_lock(&dev->mode_config.idr_mutex);
+
+       if (lessee->lessor == NULL)
+               /* owner can use all objects */
+               object_idr = &lessee->dev->mode_config.crtc_idr;
+       else
+               /* lessee can only use allowed object */
+               object_idr = &lessee->leases;
+
+       count = 0;
+       idr_for_each_entry(object_idr, entry, object) {
+               if (count_objects > count) {
+                       DRM_DEBUG_LEASE("adding object %d\n", object);
+                       ret = put_user(object, object_ids + count);
+                       if (ret)
+                               break;
+               }
+               count++;
+       }
+
+       DRM_DEBUG("lease holds %d objects\n", count);
+       if (ret == 0)
+               arg->count_objects = count;
+
+       mutex_unlock(&dev->mode_config.idr_mutex);
+
+       return ret;
+}
+
+/**
+ * drm_mode_revoke_lease_ioctl - revoke lease
+ * @dev: the drm device
+ * @data: pointer to struct drm_mode_revoke_lease
+ * @file_priv: the file being manipulated
+ *
+ * This removes all of the objects from the lease without
+ * actually getting rid of the lease itself; that way all
+ * references to it still work correctly
+ */
+int drm_mode_revoke_lease_ioctl(struct drm_device *dev,
+                               void *data, struct drm_file *lessor_priv)
+{
+       struct drm_mode_revoke_lease *arg = data;
+       struct drm_master *lessor = lessor_priv->master;
+       struct drm_master *lessee;
+       int ret = 0;
+
+       DRM_DEBUG_LEASE("revoke lease for %d\n", arg->lessee_id);
+
+       /* Can't lease without MODESET */
+       if (!drm_core_check_feature(dev, DRIVER_MODESET))
+               return -EINVAL;
+
+       mutex_lock(&dev->mode_config.idr_mutex);
+
+       lessee = _drm_find_lessee(lessor, arg->lessee_id);
+
+       /* No such lessee */
+       if (!lessee) {
+               ret = -ENOENT;
+               goto fail;
+       }
+
+       /* Lease is not held by lessor */
+       if (lessee->lessor != lessor) {
+               ret = -EACCES;
+               goto fail;
+       }
+
+       _drm_lease_revoke(lessee);
+
+fail:
+       mutex_unlock(&dev->mode_config.idr_mutex);
+
+       return ret;
+}