#include <linux/platform_device.h>
 #include <linux/regulator/consumer.h>
 
+#include <drm/drm_aperture.h>
 #include <drm/drm_atomic_state_helper.h>
 #include <drm/drm_connector.h>
 #include <drm/drm_damage_helper.h>
 
 static int simpledrm_device_init_mm(struct simpledrm_device *sdev)
 {
+       struct drm_device *dev = &sdev->dev;
        struct platform_device *pdev = sdev->pdev;
        struct resource *mem;
        void __iomem *screen_base;
+       int ret;
 
        mem = platform_get_resource(pdev, IORESOURCE_MEM, 0);
        if (!mem)
                return -EINVAL;
 
+       ret = devm_aperture_acquire_from_firmware(dev, mem->start, resource_size(mem));
+       if (ret) {
+               drm_err(dev, "could not acquire memory range [0x%llx:0x%llx]: error %d\n",
+                       mem->start, mem->end, ret);
+               return ret;
+       }
+
        screen_base = devm_ioremap_wc(&pdev->dev, mem->start,
                                      resource_size(mem));
        if (!screen_base)
        struct drm_shadow_plane_state *shadow_plane_state = to_drm_shadow_plane_state(plane_state);
        struct drm_framebuffer *fb = plane_state->fb;
        void *vmap = shadow_plane_state->map[0].vaddr; /* TODO: Use mapping abstraction properly */
+       struct drm_device *dev = &sdev->dev;
+       int idx;
 
        if (!fb)
                return;
 
+       if (!drm_dev_enter(dev, &idx))
+               return;
+
        drm_fb_blit_dstclip(sdev->screen_base, sdev->pitch,
                            sdev->format->format, vmap, fb);
+       drm_dev_exit(idx);
 }
 
 static void
        struct drm_shadow_plane_state *shadow_plane_state = to_drm_shadow_plane_state(plane_state);
        void *vmap = shadow_plane_state->map[0].vaddr; /* TODO: Use mapping abstraction properly */
        struct drm_framebuffer *fb = plane_state->fb;
+       struct drm_device *dev = &sdev->dev;
        struct drm_rect clip;
+       int idx;
 
        if (!fb)
                return;
        if (!drm_atomic_helper_damage_merged(old_plane_state, plane_state, &clip))
                return;
 
+       if (!drm_dev_enter(dev, &idx))
+               return;
+
        drm_fb_blit_rect_dstclip(sdev->screen_base, sdev->pitch,
                                 sdev->format->format, vmap, fb, &clip);
+
+       drm_dev_exit(idx);
 }
 
 static const struct drm_simple_display_pipe_funcs
        struct simpledrm_device *sdev = platform_get_drvdata(pdev);
        struct drm_device *dev = &sdev->dev;
 
-       drm_dev_unregister(dev);
+       drm_dev_unplug(dev);
 
        return 0;
 }