*/
 
 #include <linux/kernel.h>
+#include <linux/module.h>
 #include <linux/platform_device.h>
 #include <linux/slab.h>
 
 #include "dss.h"
 
 static LIST_HEAD(output_list);
+static DEFINE_MUTEX(output_lock);
+
+int omapdss_output_set_device(struct omap_dss_output *out,
+               struct omap_dss_device *dssdev)
+{
+       int r;
+
+       mutex_lock(&output_lock);
+
+       if (out->device) {
+               DSSERR("output already has device %s connected to it\n",
+                       out->device->name);
+               r = -EINVAL;
+               goto err;
+       }
+
+       if (out->type != dssdev->type) {
+               DSSERR("output type and display type don't match\n");
+               r = -EINVAL;
+               goto err;
+       }
+
+       out->device = dssdev;
+       dssdev->output = out;
+
+       mutex_unlock(&output_lock);
+
+       return 0;
+err:
+       mutex_unlock(&output_lock);
+
+       return r;
+}
+EXPORT_SYMBOL(omapdss_output_set_device);
+
+int omapdss_output_unset_device(struct omap_dss_output *out)
+{
+       int r;
+
+       mutex_lock(&output_lock);
+
+       if (!out->device) {
+               DSSERR("output doesn't have a device connected to it\n");
+               r = -EINVAL;
+               goto err;
+       }
+
+       if (out->device->state != OMAP_DSS_DISPLAY_DISABLED) {
+               DSSERR("device %s is not disabled, cannot unset device\n",
+                               out->device->name);
+               r = -EINVAL;
+               goto err;
+       }
+
+       out->device->output = NULL;
+       out->device = NULL;
+
+       mutex_unlock(&output_lock);
+
+       return 0;
+err:
+       mutex_unlock(&output_lock);
+
+       return r;
+}
+EXPORT_SYMBOL(omapdss_output_unset_device);
 
 void dss_register_output(struct omap_dss_output *out)
 {
 
        enum omap_display_caps caps;
 
        struct omap_overlay_manager *manager;
+       struct omap_dss_output *output;
 
        enum omap_dss_display_state state;
 
 struct omap_overlay *omap_dss_get_overlay(int num);
 
 struct omap_dss_output *omap_dss_get_output(enum omap_dss_output_id id);
+int omapdss_output_set_device(struct omap_dss_output *out,
+               struct omap_dss_device *dssdev);
+int omapdss_output_unset_device(struct omap_dss_output *out);
 
 void omapdss_default_get_resolution(struct omap_dss_device *dssdev,
                u16 *xres, u16 *yres);