udelay(1);
 
        if (!timeout) {
-               dev_warn(host->pdata->dev, "timeout waiting for SD bus idle\n");
+               dev_warn(&host->pdev->dev, "timeout waiting for SD bus idle\n");
                return -EBUSY;
        }
 
                goto eclkget;
        }
 
+       host = tmio_mmc_host_alloc(pdev);
+       if (!host) {
+               ret = -ENOMEM;
+               goto eprobe;
+       }
+
        mmc_data->clk_enable = sh_mobile_sdhi_clk_enable;
        mmc_data->clk_disable = sh_mobile_sdhi_clk_disable;
        mmc_data->capabilities = MMC_CAP_MMC_HIGHSPEED;
        /* SD control register space size is 0x100, 0x200 for bus_shift=1 */
        mmc_data->bus_shift = resource_size(res) >> 9;
 
-       ret = tmio_mmc_host_probe(&host, pdev, mmc_data);
+       ret = tmio_mmc_host_probe(host, mmc_data);
        if (ret < 0)
-               goto eprobe;
+               goto efree;
 
        /*
         * FIXME:
 
 eirq:
        tmio_mmc_host_remove(host);
+efree:
+       tmio_mmc_host_free(host);
 eprobe:
 eclkget:
        if (p && p->cleanup)
 
        pdata->bus_shift = resource_size(res) >> 10;
        pdata->flags |= TMIO_MMC_HAVE_HIGH_REG;
 
-       ret = tmio_mmc_host_probe(&host, pdev, pdata);
-       if (ret)
+       host = tmio_mmc_host_alloc(pdev);
+       if (!host)
                goto cell_disable;
 
+       ret = tmio_mmc_host_probe(host, pdata);
+       if (ret)
+               goto host_free;
+
        ret = request_irq(irq, tmio_mmc_irq, IRQF_TRIGGER_FALLING,
                                dev_name(&pdev->dev), host);
        if (ret)
 
 host_remove:
        tmio_mmc_host_remove(host);
+host_free:
+       tmio_mmc_host_free(host);
 cell_disable:
        if (cell->disable)
                cell->disable(pdev);
 
        bool                    sdio_irq_enabled;
 };
 
-int tmio_mmc_host_probe(struct tmio_mmc_host **host,
-                       struct platform_device *pdev,
+struct tmio_mmc_host *tmio_mmc_host_alloc(struct platform_device *pdev);
+void tmio_mmc_host_free(struct tmio_mmc_host *host);
+int tmio_mmc_host_probe(struct tmio_mmc_host *host,
                        struct tmio_mmc_data *pdata);
 void tmio_mmc_host_remove(struct tmio_mmc_host *host);
 void tmio_mmc_do_data_irq(struct tmio_mmc_host *host);
 
                pdata->flags |= TMIO_MMC_WRPROTECT_DISABLE;
 }
 
-int tmio_mmc_host_probe(struct tmio_mmc_host **host,
-                                 struct platform_device *pdev,
-                                 struct tmio_mmc_data *pdata)
+struct tmio_mmc_host*
+tmio_mmc_host_alloc(struct platform_device *pdev)
 {
-       struct tmio_mmc_host *_host;
+       struct tmio_mmc_host *host;
        struct mmc_host *mmc;
+
+       mmc = mmc_alloc_host(sizeof(struct tmio_mmc_host), &pdev->dev);
+       if (!mmc)
+               return NULL;
+
+       host = mmc_priv(mmc);
+       host->mmc = mmc;
+       host->pdev = pdev;
+
+       return host;
+}
+EXPORT_SYMBOL(tmio_mmc_host_alloc);
+
+void tmio_mmc_host_free(struct tmio_mmc_host *host)
+{
+       mmc_free_host(host->mmc);
+
+       host->mmc = NULL;
+}
+EXPORT_SYMBOL(tmio_mmc_host_free);
+
+int tmio_mmc_host_probe(struct tmio_mmc_host *_host,
+                       struct tmio_mmc_data *pdata)
+{
+       struct platform_device *pdev = _host->pdev;
+       struct mmc_host *mmc = _host->mmc;
        struct resource *res_ctl;
        int ret;
        u32 irq_mask = TMIO_MASK_CMD;
        if (!res_ctl)
                return -EINVAL;
 
-       mmc = mmc_alloc_host(sizeof(struct tmio_mmc_host), &pdev->dev);
-       if (!mmc)
-               return -ENOMEM;
-
        ret = mmc_of_parse(mmc);
        if (ret < 0)
                goto host_free;
 
-       pdata->dev = &pdev->dev;
-       _host = mmc_priv(mmc);
        _host->pdata = pdata;
-       _host->mmc = mmc;
-       _host->pdev = pdev;
        platform_set_drvdata(pdev, mmc);
 
        _host->set_pwr = pdata->set_pwr;
                mmc_gpiod_request_cd_irq(mmc);
        }
 
-       *host = _host;
-
        return 0;
 
 host_free:
-       mmc_free_host(mmc);
 
        return ret;
 }
        pm_runtime_disable(&pdev->dev);
 
        iounmap(host->ctl);
-       mmc_free_host(mmc);
 }
 EXPORT_SYMBOL(tmio_mmc_host_remove);
 
 
        unsigned long                   bus_shift;
        u32                             ocr_mask;       /* available voltages */
        struct tmio_mmc_dma             *dma;
-       struct device                   *dev;
        unsigned int                    cd_gpio;
        void (*set_pwr)(struct platform_device *host, int state);
        void (*set_clk_div)(struct platform_device *host, int state);