const u32 *sdr_addr;
        dma_addr_t msi_phys;
        void *msi_virt;
+       int err;
 
        sdr_addr = of_get_property(dev->dev.of_node, "sdr-base", NULL);
        if (!sdr_addr)
-               return -1;
+               return -EINVAL;
 
-       mtdcri(SDR0, *sdr_addr, upper_32_bits(res.start));      /*HIGH addr */
-       mtdcri(SDR0, *sdr_addr + 1, lower_32_bits(res.start));  /* Low addr */
+       msi_data = of_get_property(dev->dev.of_node, "msi-data", NULL);
+       if (!msi_data)
+               return -EINVAL;
+
+       msi_mask = of_get_property(dev->dev.of_node, "msi-mask", NULL);
+       if (!msi_mask)
+               return -EINVAL;
 
        msi->msi_dev = of_find_node_by_name(NULL, "ppc4xx-msi");
        if (!msi->msi_dev)
 
        msi->msi_regs = of_iomap(msi->msi_dev, 0);
        if (!msi->msi_regs) {
-               dev_err(&dev->dev, "of_iomap problem failed\n");
-               return -ENOMEM;
+               dev_err(&dev->dev, "of_iomap failed\n");
+               err = -ENOMEM;
+               goto node_put;
        }
        dev_dbg(&dev->dev, "PCIE-MSI: msi register mapped 0x%x 0x%x\n",
                (u32) (msi->msi_regs + PEIH_TERMADH), (u32) (msi->msi_regs));
 
        msi_virt = dma_alloc_coherent(&dev->dev, 64, &msi_phys, GFP_KERNEL);
-       if (!msi_virt)
-               return -ENOMEM;
+       if (!msi_virt) {
+               err = -ENOMEM;
+               goto iounmap;
+       }
        msi->msi_addr_hi = upper_32_bits(msi_phys);
        msi->msi_addr_lo = lower_32_bits(msi_phys & 0xffffffff);
        dev_dbg(&dev->dev, "PCIE-MSI: msi address high 0x%x, low 0x%x\n",
                msi->msi_addr_hi, msi->msi_addr_lo);
 
+       mtdcri(SDR0, *sdr_addr, upper_32_bits(res.start));      /*HIGH addr */
+       mtdcri(SDR0, *sdr_addr + 1, lower_32_bits(res.start));  /* Low addr */
+
        /* Progam the Interrupt handler Termination addr registers */
        out_be32(msi->msi_regs + PEIH_TERMADH, msi->msi_addr_hi);
        out_be32(msi->msi_regs + PEIH_TERMADL, msi->msi_addr_lo);
 
-       msi_data = of_get_property(dev->dev.of_node, "msi-data", NULL);
-       if (!msi_data)
-               return -1;
-       msi_mask = of_get_property(dev->dev.of_node, "msi-mask", NULL);
-       if (!msi_mask)
-               return -1;
        /* Program MSI Expected data and Mask bits */
        out_be32(msi->msi_regs + PEIH_MSIED, *msi_data);
        out_be32(msi->msi_regs + PEIH_MSIMK, *msi_mask);
        dma_free_coherent(&dev->dev, 64, msi_virt, msi_phys);
 
        return 0;
+
+iounmap:
+       iounmap(msi->msi_regs);
+node_put:
+       of_node_put(msi->msi_dev);
+       return err;
 }
 
 static int ppc4xx_of_msi_remove(struct platform_device *dev)
                msi_bitmap_free(&msi->bitmap);
        iounmap(msi->msi_regs);
        of_node_put(msi->msi_dev);
-       kfree(msi);
 
        return 0;
 }
 
        dev_dbg(&dev->dev, "PCIE-MSI: Setting up MSI support...\n");
 
-       msi = kzalloc(sizeof(*msi), GFP_KERNEL);
-       if (!msi) {
-               dev_err(&dev->dev, "No memory for MSI structure\n");
+       msi = devm_kzalloc(&dev->dev, sizeof(*msi), GFP_KERNEL);
+       if (!msi)
                return -ENOMEM;
-       }
        dev->dev.platform_data = msi;
 
        /* Get MSI ranges */
        err = of_address_to_resource(dev->dev.of_node, 0, &res);
        if (err) {
                dev_err(&dev->dev, "%pOF resource error!\n", dev->dev.of_node);
-               goto error_out;
+               return err;
        }
 
        msi_irqs = of_irq_count(dev->dev.of_node);
 
        err = ppc4xx_setup_pcieh_hw(dev, res, msi);
        if (err)
-               goto error_out;
+               return err;
 
        err = ppc4xx_msi_init_allocator(dev, msi);
        if (err) {
                phb->controller_ops.setup_msi_irqs = ppc4xx_setup_msi_irqs;
                phb->controller_ops.teardown_msi_irqs = ppc4xx_teardown_msi_irqs;
        }
-       return err;
+       return 0;
 
 error_out:
        ppc4xx_of_msi_remove(dev);