[u-boot.git] / drivers / mmc / mmc_write.c

// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright 2008, Freescale Semiconductor, Inc
 * Andy Fleming
 *
 * Based vaguely on the Linux code
 */

#include <config.h>
#include <common.h>
#include <dm.h>
#include <part.h>
#include <div64.h>
#include <linux/math64.h>
#include "mmc_private.h"

static ulong mmc_erase_t(struct mmc *mmc, ulong start, lbaint_t blkcnt)
{
	struct mmc_cmd cmd;
	ulong end;
	int err, start_cmd, end_cmd;

	if (mmc->high_capacity) {
		end = start + blkcnt - 1;
	} else {
		end = (start + blkcnt - 1) * mmc->write_bl_len;
		start *= mmc->write_bl_len;
	}

	if (IS_SD(mmc)) {
		start_cmd = SD_CMD_ERASE_WR_BLK_START;
		end_cmd = SD_CMD_ERASE_WR_BLK_END;
	} else {
		start_cmd = MMC_CMD_ERASE_GROUP_START;
		end_cmd = MMC_CMD_ERASE_GROUP_END;
	}

	cmd.cmdidx = start_cmd;
	cmd.cmdarg = start;
	cmd.resp_type = MMC_RSP_R1;

	err = mmc_send_cmd(mmc, &cmd, NULL);
	if (err)
		goto err_out;

	cmd.cmdidx = end_cmd;
	cmd.cmdarg = end;

	err = mmc_send_cmd(mmc, &cmd, NULL);
	if (err)
		goto err_out;

	cmd.cmdidx = MMC_CMD_ERASE;
	cmd.cmdarg = MMC_ERASE_ARG;
	cmd.resp_type = MMC_RSP_R1b;

	err = mmc_send_cmd(mmc, &cmd, NULL);
	if (err)
		goto err_out;

	return 0;

err_out:
	puts("mmc erase failed\n");
	return err;
}

#ifdef CONFIG_BLK
ulong mmc_berase(struct udevice *dev, lbaint_t start, lbaint_t blkcnt)
#else
ulong mmc_berase(struct blk_desc *block_dev, lbaint_t start, lbaint_t blkcnt)
#endif
{
#ifdef CONFIG_BLK
	struct blk_desc *block_dev = dev_get_uclass_platdata(dev);
#endif
	int dev_num = block_dev->devnum;
	int err = 0;
	u32 start_rem, blkcnt_rem;
	struct mmc *mmc = find_mmc_device(dev_num);
	lbaint_t blk = 0, blk_r = 0;
	int timeout = 1000;

	if (!mmc)
		return -1;

	err = blk_select_hwpart_devnum(IF_TYPE_MMC, dev_num,
				       block_dev->hwpart);
	if (err < 0)
		return -1;

	/*
	 * We want to see if the requested start or total block count are
	 * unaligned.  We discard the whole numbers and only care about the
	 * remainder.
	 */
	err = div_u64_rem(start, mmc->erase_grp_size, &start_rem);
	err = div_u64_rem(blkcnt, mmc->erase_grp_size, &blkcnt_rem);
	if (start_rem || blkcnt_rem)
		printf("\n\nCaution! Your devices Erase group is 0x%x\n"
		       "The erase range would be change to "
		       "0x" LBAF "~0x" LBAF "\n\n",
		       mmc->erase_grp_size, start & ~(mmc->erase_grp_size - 1),
		       ((start + blkcnt + mmc->erase_grp_size)
		       & ~(mmc->erase_grp_size - 1)) - 1);

	while (blk < blkcnt) {
		if (IS_SD(mmc) && mmc->ssr.au) {
			blk_r = ((blkcnt - blk) > mmc->ssr.au) ?
				mmc->ssr.au : (blkcnt - blk);
		} else {
			blk_r = ((blkcnt - blk) > mmc->erase_grp_size) ?
				mmc->erase_grp_size : (blkcnt - blk);
		}
		err = mmc_erase_t(mmc, start + blk, blk_r);
		if (err)
			break;

		blk += blk_r;

		/* Waiting for the ready status */
		if (mmc_send_status(mmc, timeout))
			return 0;
	}

	return blk;
}

static ulong mmc_write_blocks(struct mmc *mmc, lbaint_t start,
		lbaint_t blkcnt, const void *src)
{
	struct mmc_cmd cmd;
	struct mmc_data data;
	int timeout = 1000;

	if ((start + blkcnt) > mmc_get_blk_desc(mmc)->lba) {
		printf("MMC: block number 0x" LBAF " exceeds max(0x" LBAF ")\n",
		       start + blkcnt, mmc_get_blk_desc(mmc)->lba);
		return 0;
	}

	if (blkcnt == 0)
		return 0;
	else if (blkcnt == 1)
		cmd.cmdidx = MMC_CMD_WRITE_SINGLE_BLOCK;
	else
		cmd.cmdidx = MMC_CMD_WRITE_MULTIPLE_BLOCK;

	if (mmc->high_capacity)
		cmd.cmdarg = start;
	else
		cmd.cmdarg = start * mmc->write_bl_len;

	cmd.resp_type = MMC_RSP_R1;

	data.src = src;
	data.blocks = blkcnt;
	data.blocksize = mmc->write_bl_len;
	data.flags = MMC_DATA_WRITE;

	if (mmc_send_cmd(mmc, &cmd, &data)) {
		printf("mmc write failed\n");
		return 0;
	}

	/* SPI multiblock writes terminate using a special
	 * token, not a STOP_TRANSMISSION request.
	 */
	if (!mmc_host_is_spi(mmc) && blkcnt > 1) {
		cmd.cmdidx = MMC_CMD_STOP_TRANSMISSION;
		cmd.cmdarg = 0;
		cmd.resp_type = MMC_RSP_R1b;
		if (mmc_send_cmd(mmc, &cmd, NULL)) {
			printf("mmc fail to send stop cmd\n");
			return 0;
		}
	}

	/* Waiting for the ready status */
	if (mmc_send_status(mmc, timeout))
		return 0;

	return blkcnt;
}

#ifdef CONFIG_BLK
ulong mmc_bwrite(struct udevice *dev, lbaint_t start, lbaint_t blkcnt,
		 const void *src)
#else
ulong mmc_bwrite(struct blk_desc *block_dev, lbaint_t start, lbaint_t blkcnt,
		 const void *src)
#endif
{
#ifdef CONFIG_BLK
	struct blk_desc *block_dev = dev_get_uclass_platdata(dev);
#endif
	int dev_num = block_dev->devnum;
	lbaint_t cur, blocks_todo = blkcnt;
	int err;

	struct mmc *mmc = find_mmc_device(dev_num);
	if (!mmc)
		return 0;

	err = blk_select_hwpart_devnum(IF_TYPE_MMC, dev_num, block_dev->hwpart);
	if (err < 0)
		return 0;

	if (mmc_set_blocklen(mmc, mmc->write_bl_len))
		return 0;

	do {
		cur = (blocks_todo > mmc->cfg->b_max) ?
			mmc->cfg->b_max : blocks_todo;
		if (mmc_write_blocks(mmc, start, cur, src) != cur)
			return 0;
		blocks_todo -= cur;
		start += cur;
		src += cur * mmc->write_bl_len;
	} while (blocks_todo > 0);

	return blkcnt;
}
Commit	Line	Data
83d290c5	1	// SPDX-License-Identifier: GPL-2.0+
da61fa5f PB	2	/*
	3	* Copyright 2008, Freescale Semiconductor, Inc
	4	* Andy Fleming
	5	*
	6	* Based vaguely on the Linux code
da61fa5f PB	7	*/
	8
	9	#include <config.h>
	10	#include <common.h>
33fb211d	11	#include <dm.h>
da61fa5f	12	#include <part.h>
180f87fc TR	13	#include <div64.h>
180f87fc TR	14	#include <linux/math64.h>
da61fa5f PB	15	#include "mmc_private.h"
	16
	17	static ulong mmc_erase_t(struct mmc *mmc, ulong start, lbaint_t blkcnt)
	18	{
	19	struct mmc_cmd cmd;
	20	ulong end;
	21	int err, start_cmd, end_cmd;
	22
	23	if (mmc->high_capacity) {
	24	end = start + blkcnt - 1;
	25	} else {
	26	end = (start + blkcnt - 1) * mmc->write_bl_len;
	27	start *= mmc->write_bl_len;
	28	}
	29
	30	if (IS_SD(mmc)) {
	31	start_cmd = SD_CMD_ERASE_WR_BLK_START;
	32	end_cmd = SD_CMD_ERASE_WR_BLK_END;
	33	} else {
	34	start_cmd = MMC_CMD_ERASE_GROUP_START;
	35	end_cmd = MMC_CMD_ERASE_GROUP_END;
	36	}
	37
	38	cmd.cmdidx = start_cmd;
	39	cmd.cmdarg = start;
	40	cmd.resp_type = MMC_RSP_R1;
	41
	42	err = mmc_send_cmd(mmc, &cmd, NULL);
	43	if (err)
	44	goto err_out;
	45
	46	cmd.cmdidx = end_cmd;
	47	cmd.cmdarg = end;
	48
	49	err = mmc_send_cmd(mmc, &cmd, NULL);
	50	if (err)
	51	goto err_out;
	52
	53	cmd.cmdidx = MMC_CMD_ERASE;
1aa2d074	54	cmd.cmdarg = MMC_ERASE_ARG;
da61fa5f PB	55	cmd.resp_type = MMC_RSP_R1b;
	56
	57	err = mmc_send_cmd(mmc, &cmd, NULL);
	58	if (err)
	59	goto err_out;
	60
	61	return 0;
	62
	63	err_out:
	64	puts("mmc erase failed\n");
	65	return err;
	66	}
	67
561e624c SG	68	#ifdef CONFIG_BLK
	69	ulong mmc_berase(struct udevice *dev, lbaint_t start, lbaint_t blkcnt)
	70	#else
	71	ulong mmc_berase(struct blk_desc *block_dev, lbaint_t start, lbaint_t blkcnt)
	72	#endif
da61fa5f	73	{
561e624c SG	74	#ifdef CONFIG_BLK
	75	struct blk_desc *block_dev = dev_get_uclass_platdata(dev);
	76	#endif
bcce53d0	77	int dev_num = block_dev->devnum;
da61fa5f	78	int err = 0;
180f87fc	79	u32 start_rem, blkcnt_rem;
da61fa5f PB	80	struct mmc *mmc = find_mmc_device(dev_num);
	81	lbaint_t blk = 0, blk_r = 0;
	82	int timeout = 1000;
	83
	84	if (!mmc)
	85	return -1;
	86
69f45cd5 SG	87	err = blk_select_hwpart_devnum(IF_TYPE_MMC, dev_num,
69f45cd5 SG	88	block_dev->hwpart);
873cc1d7 SW	89	if (err < 0)
	90	return -1;
	91
180f87fc TR	92	/*
	93	* We want to see if the requested start or total block count are
	94	* unaligned. We discard the whole numbers and only care about the
	95	* remainder.
	96	*/
	97	err = div_u64_rem(start, mmc->erase_grp_size, &start_rem);
	98	err = div_u64_rem(blkcnt, mmc->erase_grp_size, &blkcnt_rem);
	99	if (start_rem \|\| blkcnt_rem)
da61fa5f PB	100	printf("\n\nCaution! Your devices Erase group is 0x%x\n"
	101	"The erase range would be change to "
	102	"0x" LBAF "~0x" LBAF "\n\n",
	103	mmc->erase_grp_size, start & ~(mmc->erase_grp_size - 1),
	104	((start + blkcnt + mmc->erase_grp_size)
	105	& ~(mmc->erase_grp_size - 1)) - 1);
	106
	107	while (blk < blkcnt) {
e492dbb4 PF	108	if (IS_SD(mmc) && mmc->ssr.au) {
	109	blk_r = ((blkcnt - blk) > mmc->ssr.au) ?
	110	mmc->ssr.au : (blkcnt - blk);
	111	} else {
	112	blk_r = ((blkcnt - blk) > mmc->erase_grp_size) ?
	113	mmc->erase_grp_size : (blkcnt - blk);
	114	}
da61fa5f PB	115	err = mmc_erase_t(mmc, start + blk, blk_r);
	116	if (err)
	117	break;
	118
	119	blk += blk_r;
	120
	121	/* Waiting for the ready status */
	122	if (mmc_send_status(mmc, timeout))
	123	return 0;
	124	}
	125
	126	return blk;
	127	}
	128
	129	static ulong mmc_write_blocks(struct mmc *mmc, lbaint_t start,
	130	lbaint_t blkcnt, const void *src)
	131	{
	132	struct mmc_cmd cmd;
	133	struct mmc_data data;
	134	int timeout = 1000;
	135
c40fdca6	136	if ((start + blkcnt) > mmc_get_blk_desc(mmc)->lba) {
da61fa5f	137	printf("MMC: block number 0x" LBAF " exceeds max(0x" LBAF ")\n",
c40fdca6	138	start + blkcnt, mmc_get_blk_desc(mmc)->lba);
da61fa5f PB	139	return 0;
	140	}
	141
	142	if (blkcnt == 0)
	143	return 0;
	144	else if (blkcnt == 1)
	145	cmd.cmdidx = MMC_CMD_WRITE_SINGLE_BLOCK;
	146	else
	147	cmd.cmdidx = MMC_CMD_WRITE_MULTIPLE_BLOCK;
	148
	149	if (mmc->high_capacity)
	150	cmd.cmdarg = start;
	151	else
	152	cmd.cmdarg = start * mmc->write_bl_len;
	153
	154	cmd.resp_type = MMC_RSP_R1;
	155
	156	data.src = src;
	157	data.blocks = blkcnt;
	158	data.blocksize = mmc->write_bl_len;
	159	data.flags = MMC_DATA_WRITE;
	160
	161	if (mmc_send_cmd(mmc, &cmd, &data)) {
	162	printf("mmc write failed\n");
	163	return 0;
	164	}
	165
	166	/* SPI multiblock writes terminate using a special
	167	* token, not a STOP_TRANSMISSION request.
	168	*/
	169	if (!mmc_host_is_spi(mmc) && blkcnt > 1) {
	170	cmd.cmdidx = MMC_CMD_STOP_TRANSMISSION;
	171	cmd.cmdarg = 0;
	172	cmd.resp_type = MMC_RSP_R1b;
	173	if (mmc_send_cmd(mmc, &cmd, NULL)) {
	174	printf("mmc fail to send stop cmd\n");
	175	return 0;
	176	}
	177	}
	178
	179	/* Waiting for the ready status */
	180	if (mmc_send_status(mmc, timeout))
	181	return 0;
	182
	183	return blkcnt;
	184	}
	185
33fb211d SG	186	#ifdef CONFIG_BLK
	187	ulong mmc_bwrite(struct udevice *dev, lbaint_t start, lbaint_t blkcnt,
	188	const void *src)
	189	#else
4101f687	190	ulong mmc_bwrite(struct blk_desc *block_dev, lbaint_t start, lbaint_t blkcnt,
7c4213f6	191	const void *src)
33fb211d	192	#endif
da61fa5f	193	{
33fb211d SG	194	#ifdef CONFIG_BLK
	195	struct blk_desc *block_dev = dev_get_uclass_platdata(dev);
	196	#endif
bcce53d0	197	int dev_num = block_dev->devnum;
da61fa5f	198	lbaint_t cur, blocks_todo = blkcnt;
873cc1d7	199	int err;
da61fa5f PB	200
	201	struct mmc *mmc = find_mmc_device(dev_num);
	202	if (!mmc)
	203	return 0;
	204
69f45cd5	205	err = blk_select_hwpart_devnum(IF_TYPE_MMC, dev_num, block_dev->hwpart);
873cc1d7 SW	206	if (err < 0)
	207	return 0;
	208
da61fa5f PB	209	if (mmc_set_blocklen(mmc, mmc->write_bl_len))
	210	return 0;
	211
	212	do {
93bfd616 PA	213	cur = (blocks_todo > mmc->cfg->b_max) ?
93bfd616 PA	214	mmc->cfg->b_max : blocks_todo;
da61fa5f PB	215	if (mmc_write_blocks(mmc, start, cur, src) != cur)
	216	return 0;
	217	blocks_todo -= cur;
	218	start += cur;
	219	src += cur * mmc->write_bl_len;
	220	} while (blocks_todo > 0);
	221
	222	return blkcnt;
	223	}