Merge branch 'master' of git://git.denx.de/u-boot
[oweals/u-boot.git] / drivers / net / fsl-mc / mc.c
index d2ca5c657c9c206b4503fb255421e8045f9a8d63..46b8a6bc69912f4da1868fe6ca3f87d4fe9f8b78 100644 (file)
@@ -5,6 +5,7 @@
  */
 #include <common.h>
 #include <errno.h>
+#include <linux/bug.h>
 #include <asm/io.h>
 #include <libfdt.h>
 #include <fdt_support.h>
@@ -31,11 +32,15 @@ static int mc_dpl_applied = -1;
 #ifdef CONFIG_SYS_LS_MC_DRAM_AIOP_IMG_OFFSET
 static int mc_aiop_applied = -1;
 #endif
-struct fsl_mc_io *dflt_mc_io = NULL;
+struct fsl_mc_io *root_mc_io = NULL;
+struct fsl_mc_io *dflt_mc_io = NULL; /* child container */
+uint16_t root_dprc_handle = 0;
 uint16_t dflt_dprc_handle = 0;
+int child_dprc_id;
 struct fsl_dpbp_obj *dflt_dpbp = NULL;
 struct fsl_dpio_obj *dflt_dpio = NULL;
-uint16_t dflt_dpio_handle = 0;
+struct fsl_dpni_obj *dflt_dpni = NULL;
+static u64 mc_lazy_dpl_addr;
 
 #ifdef DEBUG
 void dump_ram_words(const char *title, void *addr)
@@ -352,6 +357,12 @@ static unsigned long get_mc_boot_timeout_ms(void)
 }
 
 #ifdef CONFIG_SYS_LS_MC_DRAM_AIOP_IMG_OFFSET
+
+__weak bool soc_has_aiop(void)
+{
+       return false;
+}
+
 static int load_mc_aiop_img(u64 aiop_fw_addr)
 {
        u64 mc_ram_addr = mc_get_dram_addr();
@@ -359,6 +370,9 @@ static int load_mc_aiop_img(u64 aiop_fw_addr)
        void *aiop_img;
 #endif
 
+       /* Check if AIOP is available */
+       if (!soc_has_aiop())
+               return -ENODEV;
        /*
         * Load the MC AIOP image in the MC private DRAM block:
         */
@@ -451,7 +465,7 @@ int mc_init(u64 mc_fw_addr, u64 mc_dpc_addr)
 
        /*
         * Management Complex cores should be held at reset out of POR.
-        * U-boot should be the first software to touch MC. To be safe,
+        * U-Boot should be the first software to touch MC. To be safe,
         * we reset all cores again by setting GCR1 to 0. It doesn't do
         * anything if they are held at reset. After we setup the firmware
         * we kick off MC by deasserting the reset bit for core 0, and
@@ -521,36 +535,23 @@ int mc_init(u64 mc_fw_addr, u64 mc_dpc_addr)
         * Initialize the global default MC portal
         * And check that the MC firmware is responding portal commands:
         */
-       dflt_mc_io = (struct fsl_mc_io *)malloc(sizeof(struct fsl_mc_io));
-       if (!dflt_mc_io) {
+       root_mc_io = (struct fsl_mc_io *)malloc(sizeof(struct fsl_mc_io));
+       if (!root_mc_io) {
                printf(" No memory: malloc() failed\n");
                return -ENOMEM;
        }
 
-       dflt_mc_io->mmio_regs = SOC_MC_PORTAL_ADDR(portal_id);
+       root_mc_io->mmio_regs = SOC_MC_PORTAL_ADDR(portal_id);
        debug("Checking access to MC portal of root DPRC container (portal_id %d, portal physical addr %p)\n",
-             portal_id, dflt_mc_io->mmio_regs);
+             portal_id, root_mc_io->mmio_regs);
 
-       error = mc_get_version(dflt_mc_io, MC_CMD_NO_FLAGS, &mc_ver_info);
+       error = mc_get_version(root_mc_io, MC_CMD_NO_FLAGS, &mc_ver_info);
        if (error != 0) {
                printf("fsl-mc: ERROR: Firmware version check failed (error: %d)\n",
                       error);
                goto out;
        }
 
-       if (MC_VER_MAJOR != mc_ver_info.major) {
-               printf("fsl-mc: ERROR: Firmware major version mismatch (found: %d, expected: %d)\n",
-                      mc_ver_info.major, MC_VER_MAJOR);
-               printf("fsl-mc: Update the Management Complex firmware\n");
-
-               error = -ENODEV;
-               goto out;
-       }
-
-       if (MC_VER_MINOR != mc_ver_info.minor)
-               printf("fsl-mc: WARNING: Firmware minor version mismatch (found: %d, expected: %d)\n",
-                      mc_ver_info.minor, MC_VER_MINOR);
-
        printf("fsl-mc: Management Complex booted (version: %d.%d.%d, boot status: %#x)\n",
               mc_ver_info.major, mc_ver_info.minor, mc_ver_info.revision,
               reg_gsr & GSR_FS_MASK);
@@ -572,6 +573,9 @@ int mc_apply_dpl(u64 mc_dpl_addr)
        u64 mc_ram_addr = mc_get_dram_addr();
        size_t mc_ram_size = mc_get_dram_block_size();
 
+       if (!mc_dpl_addr)
+               return -1;
+
        error = load_mc_dpl(mc_ram_addr, mc_ram_size, mc_dpl_addr);
        if (error != 0)
                return error;
@@ -654,36 +658,84 @@ unsigned long mc_get_dram_block_size(void)
        return dram_block_size;
 }
 
-int dpio_init(struct dprc_obj_desc obj_desc)
+int fsl_mc_ldpaa_init(bd_t *bis)
+{
+       int i;
+
+       for (i = WRIOP1_DPMAC1; i < NUM_WRIOP_PORTS; i++)
+               if ((wriop_is_enabled_dpmac(i) == 1) &&
+                   (wriop_get_phy_address(i) != -1))
+                       ldpaa_eth_init(i, wriop_get_enet_if(i));
+       return 0;
+}
+
+static int dprc_version_check(struct fsl_mc_io *mc_io, uint16_t handle)
+{
+       struct dprc_attributes attr;
+       int error;
+
+       memset(&attr, 0, sizeof(struct dprc_attributes));
+       error = dprc_get_attributes(mc_io, MC_CMD_NO_FLAGS, handle, &attr);
+       if (error == 0) {
+               if ((attr.version.major != DPRC_VER_MAJOR) ||
+                   (attr.version.minor != DPRC_VER_MINOR)) {
+                       printf("DPRC version mismatch found %u.%u,",
+                              attr.version.major,
+                              attr.version.minor);
+                       printf("supported version is %u.%u\n",
+                              DPRC_VER_MAJOR, DPRC_VER_MINOR);
+               }
+       }
+       return error;
+}
+
+static int dpio_init(void)
 {
        struct qbman_swp_desc p_des;
        struct dpio_attr attr;
+       struct dpio_cfg dpio_cfg;
        int err = 0;
 
        dflt_dpio = (struct fsl_dpio_obj *)malloc(sizeof(struct fsl_dpio_obj));
        if (!dflt_dpio) {
-               printf(" No memory: malloc() failed\n");
-               return -ENOMEM;
+               printf("No memory: malloc() failed\n");
+               err = -ENOMEM;
+               goto err_malloc;
        }
 
-       dflt_dpio->dpio_id = obj_desc.id;
+       dpio_cfg.channel_mode = DPIO_LOCAL_CHANNEL;
+       dpio_cfg.num_priorities = 8;
 
-       err = dpio_open(dflt_mc_io, MC_CMD_NO_FLAGS, obj_desc.id,
-                       &dflt_dpio_handle);
-       if (err) {
-               printf("dpio_open() failed\n");
-               goto err_open;
+       err = dpio_create(dflt_mc_io, MC_CMD_NO_FLAGS, &dpio_cfg,
+                         &dflt_dpio->dpio_handle);
+       if (err < 0) {
+               printf("dpio_create() failed: %d\n", err);
+               err = -ENODEV;
+               goto err_create;
        }
 
+       memset(&attr, 0, sizeof(struct dpio_attr));
        err = dpio_get_attributes(dflt_mc_io, MC_CMD_NO_FLAGS,
-                                 dflt_dpio_handle, &attr);
-       if (err) {
-               printf("dpio_get_attributes() failed %d\n", err);
+                                 dflt_dpio->dpio_handle, &attr);
+       if (err < 0) {
+               printf("dpio_get_attributes() failed: %d\n", err);
                goto err_get_attr;
        }
 
-       err = dpio_enable(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpio_handle);
-       if (err) {
+       if ((attr.version.major != DPIO_VER_MAJOR) ||
+           (attr.version.minor != DPIO_VER_MINOR)) {
+               printf("DPIO version mismatch found %u.%u,",
+                      attr.version.major, attr.version.minor);
+               printf("supported version is %u.%u\n",
+                      DPIO_VER_MAJOR, DPIO_VER_MINOR);
+       }
+
+       dflt_dpio->dpio_id = attr.id;
+#ifdef DEBUG
+       printf("Init: DPIO id=0x%d\n", dflt_dpio->dpio_id);
+#endif
+       err = dpio_enable(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpio->dpio_handle);
+       if (err < 0) {
                printf("dpio_enable() failed %d\n", err);
                goto err_get_enable;
        }
@@ -706,36 +758,453 @@ int dpio_init(struct dprc_obj_desc obj_desc)
        return 0;
 
 err_get_swp_init:
+       dpio_disable(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpio->dpio_handle);
 err_get_enable:
-       dpio_disable(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpio_handle);
 err_get_attr:
-       dpio_close(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpio_handle);
-err_open:
+       dpio_close(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpio->dpio_handle);
+       dpio_destroy(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpio->dpio_handle);
+err_create:
        free(dflt_dpio);
+err_malloc:
+       return err;
+}
+
+static int dpio_exit(void)
+{
+       int err;
+
+       err = dpio_disable(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpio->dpio_handle);
+       if (err < 0) {
+               printf("dpio_disable() failed: %d\n", err);
+               goto err;
+       }
+
+       err = dpio_destroy(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpio->dpio_handle);
+       if (err < 0) {
+               printf("dpio_destroy() failed: %d\n", err);
+               goto err;
+       }
+
+#ifdef DEBUG
+       printf("Exit: DPIO id=0x%d\n", dflt_dpio->dpio_id);
+#endif
+
+       if (dflt_dpio)
+               free(dflt_dpio);
+
+       return 0;
+err:
+       return err;
+}
+
+static int dprc_init(void)
+{
+       int err, child_portal_id, container_id;
+       struct dprc_cfg cfg;
+       uint64_t mc_portal_offset;
+
+       /* Open root container */
+       err = dprc_get_container_id(root_mc_io, MC_CMD_NO_FLAGS, &container_id);
+       if (err < 0) {
+               printf("dprc_get_container_id(): Root failed: %d\n", err);
+               goto err_root_container_id;
+       }
+
+#ifdef DEBUG
+       printf("Root container id = %d\n", container_id);
+#endif
+       err = dprc_open(root_mc_io, MC_CMD_NO_FLAGS, container_id,
+                       &root_dprc_handle);
+       if (err < 0) {
+               printf("dprc_open(): Root Container failed: %d\n", err);
+               goto err_root_open;
+       }
+
+       if (!root_dprc_handle) {
+               printf("dprc_open(): Root Container Handle is not valid\n");
+               goto err_root_open;
+       }
+
+       err = dprc_version_check(root_mc_io, root_dprc_handle);
+       if (err < 0) {
+               printf("dprc_version_check() failed: %d\n", err);
+               goto err_root_open;
+       }
+
+       memset(&cfg, 0, sizeof(struct dprc_cfg));
+       cfg.options = DPRC_CFG_OPT_TOPOLOGY_CHANGES_ALLOWED |
+                     DPRC_CFG_OPT_OBJ_CREATE_ALLOWED |
+                     DPRC_CFG_OPT_ALLOC_ALLOWED;
+       cfg.icid = DPRC_GET_ICID_FROM_POOL;
+       cfg.portal_id = DPRC_GET_PORTAL_ID_FROM_POOL;
+       err = dprc_create_container(root_mc_io, MC_CMD_NO_FLAGS,
+                       root_dprc_handle,
+                       &cfg,
+                       &child_dprc_id,
+                       &mc_portal_offset);
+       if (err < 0) {
+               printf("dprc_create_container() failed: %d\n", err);
+               goto err_create;
+       }
+
+       dflt_mc_io = (struct fsl_mc_io *)malloc(sizeof(struct fsl_mc_io));
+       if (!dflt_mc_io) {
+               err  = -ENOMEM;
+               printf(" No memory: malloc() failed\n");
+               goto err_malloc;
+       }
+
+       child_portal_id = MC_PORTAL_OFFSET_TO_PORTAL_ID(mc_portal_offset);
+       dflt_mc_io->mmio_regs = SOC_MC_PORTAL_ADDR(child_portal_id);
+#ifdef DEBUG
+       printf("MC portal of child DPRC container: %d, physical addr %p)\n",
+              child_dprc_id, dflt_mc_io->mmio_regs);
+#endif
+
+       err = dprc_open(dflt_mc_io, MC_CMD_NO_FLAGS, child_dprc_id,
+                       &dflt_dprc_handle);
+       if (err < 0) {
+               printf("dprc_open(): Child container failed: %d\n", err);
+               goto err_child_open;
+       }
+
+       if (!dflt_dprc_handle) {
+               printf("dprc_open(): Child container Handle is not valid\n");
+               goto err_child_open;
+       }
+
+       return 0;
+err_child_open:
+       free(dflt_mc_io);
+err_malloc:
+       dprc_destroy_container(root_mc_io, MC_CMD_NO_FLAGS,
+                              root_dprc_handle, child_dprc_id);
+err_create:
+       dprc_close(root_mc_io, MC_CMD_NO_FLAGS, root_dprc_handle);
+err_root_open:
+err_root_container_id:
        return err;
 }
 
-int dpbp_init(struct dprc_obj_desc obj_desc)
+static int dprc_exit(void)
 {
+       int err;
+
+       err = dprc_close(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dprc_handle);
+       if (err < 0) {
+               printf("dprc_close(): Child failed: %d\n", err);
+               goto err;
+       }
+
+       err = dprc_destroy_container(root_mc_io, MC_CMD_NO_FLAGS,
+                                    root_dprc_handle, child_dprc_id);
+       if (err < 0) {
+               printf("dprc_destroy_container() failed: %d\n", err);
+               goto err;
+       }
+
+       err = dprc_close(root_mc_io, MC_CMD_NO_FLAGS, root_dprc_handle);
+       if (err < 0) {
+               printf("dprc_close(): Root failed: %d\n", err);
+               goto err;
+       }
+
+       if (dflt_mc_io)
+               free(dflt_mc_io);
+
+       if (root_mc_io)
+               free(root_mc_io);
+
+       return 0;
+
+err:
+       return err;
+}
+
+static int dpbp_init(void)
+{
+       int err;
+       struct dpbp_attr dpbp_attr;
+       struct dpbp_cfg dpbp_cfg;
+
        dflt_dpbp = (struct fsl_dpbp_obj *)malloc(sizeof(struct fsl_dpbp_obj));
        if (!dflt_dpbp) {
-               printf(" No memory: malloc() failed\n");
-               return -ENOMEM;
+               printf("No memory: malloc() failed\n");
+               err = -ENOMEM;
+               goto err_malloc;
+       }
+
+       dpbp_cfg.options = 512;
+
+       err = dpbp_create(dflt_mc_io, MC_CMD_NO_FLAGS, &dpbp_cfg,
+                         &dflt_dpbp->dpbp_handle);
+
+       if (err < 0) {
+               err = -ENODEV;
+               printf("dpbp_create() failed: %d\n", err);
+               goto err_create;
+       }
+
+       memset(&dpbp_attr, 0, sizeof(struct dpbp_attr));
+       err = dpbp_get_attributes(dflt_mc_io, MC_CMD_NO_FLAGS,
+                                 dflt_dpbp->dpbp_handle,
+                                 &dpbp_attr);
+       if (err < 0) {
+               printf("dpbp_get_attributes() failed: %d\n", err);
+               goto err_get_attr;
+       }
+
+       if ((dpbp_attr.version.major != DPBP_VER_MAJOR) ||
+           (dpbp_attr.version.minor != DPBP_VER_MINOR)) {
+               printf("DPBP version mismatch found %u.%u,",
+                      dpbp_attr.version.major, dpbp_attr.version.minor);
+               printf("supported version is %u.%u\n",
+                      DPBP_VER_MAJOR, DPBP_VER_MINOR);
+       }
+
+       dflt_dpbp->dpbp_attr.id = dpbp_attr.id;
+#ifdef DEBUG
+       printf("Init: DPBP id=0x%d\n", dflt_dpbp->dpbp_attr.id);
+#endif
+
+       err = dpbp_close(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpbp->dpbp_handle);
+       if (err < 0) {
+               printf("dpbp_close() failed: %d\n", err);
+               goto err_close;
        }
-       dflt_dpbp->dpbp_attr.id = obj_desc.id;
 
        return 0;
+
+err_close:
+       free(dflt_dpbp);
+err_get_attr:
+       dpbp_close(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpbp->dpbp_handle);
+       dpbp_destroy(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpbp->dpbp_handle);
+err_create:
+err_malloc:
+       return err;
 }
 
-int fsl_mc_ldpaa_init(bd_t *bis)
+static int dpbp_exit(void)
+{
+       int err;
+
+       err = dpbp_open(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpbp->dpbp_attr.id,
+                       &dflt_dpbp->dpbp_handle);
+       if (err < 0) {
+               printf("dpbp_open() failed: %d\n", err);
+               goto err;
+       }
+
+       err = dpbp_destroy(dflt_mc_io, MC_CMD_NO_FLAGS,
+                          dflt_dpbp->dpbp_handle);
+       if (err < 0) {
+               printf("dpbp_destroy() failed: %d\n", err);
+               goto err;
+       }
+
+#ifdef DEBUG
+       printf("Exit: DPBP id=0x%d\n", dflt_dpbp->dpbp_attr.id);
+#endif
+
+       if (dflt_dpbp)
+               free(dflt_dpbp);
+       return 0;
+
+err:
+       return err;
+}
+
+static int dpni_init(void)
+{
+       int err;
+       struct dpni_attr dpni_attr;
+       uint8_t ext_cfg_buf[256] = {0};
+       struct dpni_extended_cfg dpni_extended_cfg;
+       struct dpni_cfg dpni_cfg;
+
+       dflt_dpni = (struct fsl_dpni_obj *)malloc(sizeof(struct fsl_dpni_obj));
+       if (!dflt_dpni) {
+               printf("No memory: malloc() failed\n");
+               err = -ENOMEM;
+               goto err_malloc;
+       }
+
+       memset(&dpni_extended_cfg, 0, sizeof(dpni_extended_cfg));
+       err = dpni_prepare_extended_cfg(&dpni_extended_cfg, &ext_cfg_buf[0]);
+       if (err < 0) {
+               err = -ENODEV;
+               printf("dpni_prepare_extended_cfg() failed: %d\n", err);
+               goto err_prepare_extended_cfg;
+       }
+
+       memset(&dpni_cfg, 0, sizeof(dpni_cfg));
+       dpni_cfg.adv.options = DPNI_OPT_UNICAST_FILTER |
+                              DPNI_OPT_MULTICAST_FILTER;
+
+       dpni_cfg.adv.ext_cfg_iova = (uint64_t)&ext_cfg_buf[0];
+       err = dpni_create(dflt_mc_io, MC_CMD_NO_FLAGS, &dpni_cfg,
+                         &dflt_dpni->dpni_handle);
+
+       if (err < 0) {
+               err = -ENODEV;
+               printf("dpni_create() failed: %d\n", err);
+               goto err_create;
+       }
+
+       memset(&dpni_attr, 0, sizeof(struct dpni_attr));
+       err = dpni_get_attributes(dflt_mc_io, MC_CMD_NO_FLAGS,
+                                 dflt_dpni->dpni_handle,
+                                 &dpni_attr);
+       if (err < 0) {
+               printf("dpni_get_attributes() failed: %d\n", err);
+               goto err_get_attr;
+       }
+
+       if ((dpni_attr.version.major != DPNI_VER_MAJOR) ||
+           (dpni_attr.version.minor != DPNI_VER_MINOR)) {
+               printf("DPNI version mismatch found %u.%u,",
+                      dpni_attr.version.major, dpni_attr.version.minor);
+               printf("supported version is %u.%u\n",
+                      DPNI_VER_MAJOR, DPNI_VER_MINOR);
+       }
+
+       dflt_dpni->dpni_id = dpni_attr.id;
+#ifdef DEBUG
+       printf("Init: DPNI id=0x%d\n", dflt_dpni->dpni_id);
+#endif
+
+       err = dpni_close(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpni->dpni_handle);
+       if (err < 0) {
+               printf("dpni_close() failed: %d\n", err);
+               goto err_close;
+       }
+
+       return 0;
+
+err_close:
+err_get_attr:
+       dpni_close(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpni->dpni_handle);
+       dpni_destroy(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpni->dpni_handle);
+err_create:
+err_prepare_extended_cfg:
+       free(dflt_dpni);
+err_malloc:
+       return err;
+}
+
+static int dpni_exit(void)
 {
+       int err;
+
+       err = dpni_open(dflt_mc_io, MC_CMD_NO_FLAGS, dflt_dpni->dpni_id,
+                       &dflt_dpni->dpni_handle);
+       if (err < 0) {
+               printf("dpni_open() failed: %d\n", err);
+               goto err;
+       }
 
+       err = dpni_destroy(dflt_mc_io, MC_CMD_NO_FLAGS,
+                          dflt_dpni->dpni_handle);
+       if (err < 0) {
+               printf("dpni_destroy() failed: %d\n", err);
+               goto err;
+       }
+
+#ifdef DEBUG
+       printf("Exit: DPNI id=0x%d\n", dflt_dpni->dpni_id);
+#endif
+
+       if (dflt_dpni)
+               free(dflt_dpni);
        return 0;
+
+err:
+       return err;
+}
+
+static int mc_init_object(void)
+{
+       int err = 0;
+
+       err = dprc_init();
+       if (err < 0) {
+               printf("dprc_init() failed: %d\n", err);
+               goto err;
+       }
+
+       err = dpbp_init();
+       if (err < 0) {
+               printf("dpbp_init() failed: %d\n", err);
+               goto err;
+       }
+
+       err = dpio_init();
+       if (err < 0) {
+               printf("dpio_init() failed: %d\n", err);
+               goto err;
+       }
+
+       err = dpni_init();
+       if (err < 0) {
+               printf("dpni_init() failed: %d\n", err);
+               goto err;
+       }
+
+       return 0;
+err:
+       return err;
 }
 
-void fsl_mc_ldpaa_exit(bd_t *bis)
+int fsl_mc_ldpaa_exit(bd_t *bd)
 {
-       return;
+       int err = 0;
+
+       if (bd && mc_lazy_dpl_addr && !fsl_mc_ldpaa_exit(NULL)) {
+               mc_apply_dpl(mc_lazy_dpl_addr);
+               mc_lazy_dpl_addr = 0;
+       }
+
+       /* MC is not loaded intentionally, So return success. */
+       if (bd && get_mc_boot_status() != 0)
+               return 0;
+
+       if (bd && !get_mc_boot_status() && get_dpl_apply_status() == -1) {
+               printf("ERROR: fsl-mc: DPL is not applied\n");
+               err = -ENODEV;
+               return err;
+       }
+
+       if (bd && !get_mc_boot_status() && !get_dpl_apply_status())
+               return err;
+
+       err = dpbp_exit();
+       if (err < 0) {
+               printf("dpbp_exit() failed: %d\n", err);
+               goto err;
+       }
+
+       err = dpio_exit();
+       if (err < 0) {
+               printf("dpio_exit() failed: %d\n", err);
+               goto err;
+       }
+
+       err = dpni_exit();
+       if (err < 0) {
+               printf("dpni_exit() failed: %d\n", err);
+               goto err;
+       }
+
+       err = dprc_exit();
+       if (err < 0) {
+               printf("dprc_exit() failed: %d\n", err);
+               goto err;
+       }
+
+       return 0;
+err:
+       return err;
 }
 
 static int do_fsl_mc(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
@@ -747,7 +1216,10 @@ static int do_fsl_mc(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
        switch (argv[1][0]) {
        case 's': {
                        char sub_cmd;
-                       u64 mc_fw_addr, mc_dpc_addr, aiop_fw_addr;
+                       u64 mc_fw_addr, mc_dpc_addr;
+#ifdef CONFIG_SYS_LS_MC_DRAM_AIOP_IMG_OFFSET
+                       u64 aiop_fw_addr;
+#endif
 
                        sub_cmd = argv[2][0];
                        switch (sub_cmd) {
@@ -763,7 +1235,9 @@ static int do_fsl_mc(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
                                mc_fw_addr = simple_strtoull(argv[3], NULL, 16);
                                mc_dpc_addr = simple_strtoull(argv[4], NULL,
                                                              16);
-                               err = mc_init(mc_fw_addr, mc_dpc_addr);
+
+                               if (!mc_init(mc_fw_addr, mc_dpc_addr))
+                                       err = mc_init_object();
                                break;
 
 #ifdef CONFIG_SYS_LS_MC_DRAM_AIOP_IMG_OFFSET
@@ -779,6 +1253,7 @@ static int do_fsl_mc(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
                                aiop_fw_addr = simple_strtoull(argv[3], NULL,
                                                               16);
 
+                               /* if SoC doesn't have AIOP, err = -ENODEV */
                                err = load_mc_aiop_img(aiop_fw_addr);
                                if (!err)
                                        printf("fsl-mc: AIOP FW applied\n");
@@ -793,6 +1268,7 @@ static int do_fsl_mc(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
                }
                break;
 
+       case 'l':
        case 'a': {
                        u64 mc_dpl_addr;
 
@@ -806,12 +1282,24 @@ static int do_fsl_mc(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 
                        mc_dpl_addr = simple_strtoull(argv[3], NULL,
                                                              16);
+
                        if (get_mc_boot_status() != 0) {
                                printf("fsl-mc: Deploying data path layout ..");
                                printf("ERROR (MC is not booted)\n");
                                return -ENODEV;
                        }
-                       err = mc_apply_dpl(mc_dpl_addr);
+
+                       if (argv[1][0] == 'l') {
+                               /*
+                                * We will do the actual dpaa exit and dpl apply
+                                * later from announce_and_cleanup().
+                                */
+                               mc_lazy_dpl_addr = mc_dpl_addr;
+                       } else {
+                               /* The user wants it applied now */
+                               if (!fsl_mc_ldpaa_exit(NULL))
+                                       err = mc_apply_dpl(mc_dpl_addr);
+                       }
                        break;
                }
        default:
@@ -829,5 +1317,6 @@ U_BOOT_CMD(
        "DPAA2 command to manage Management Complex (MC)",
        "start mc [FW_addr] [DPC_addr] - Start Management Complex\n"
        "fsl_mc apply DPL [DPL_addr] - Apply DPL file\n"
+       "fsl_mc lazyapply DPL [DPL_addr] - Apply DPL file on exit\n"
        "fsl_mc start aiop [FW_addr] - Start AIOP\n"
 );