mirror of
https://github.com/brain-hackers/u-boot-brain
synced 2024-08-29 05:18:51 +09:00
![Łukasz Majewski](/assets/img/avatar_default.png)
For some time we have been using the run_command() with properly crafted string. Such approach turned to be unreliable and error prone. Switch to "native" mmc subsystem API would allow better type checking and shall improve speed. Also, it seems that this API is changing less often than u-boot commands. The approach similar to env operations on the eMMC has been reused. Signed-off-by: Lukasz Majewski <l.majewski@samsung.com>
251 lines
5.7 KiB
C
251 lines
5.7 KiB
C
/*
|
|
* dfu.c -- DFU back-end routines
|
|
*
|
|
* Copyright (C) 2012 Samsung Electronics
|
|
* author: Lukasz Majewski <l.majewski@samsung.com>
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0+
|
|
*/
|
|
|
|
#include <common.h>
|
|
#include <malloc.h>
|
|
#include <errno.h>
|
|
#include <div64.h>
|
|
#include <dfu.h>
|
|
#include <mmc.h>
|
|
|
|
static unsigned char __aligned(CONFIG_SYS_CACHELINE_SIZE)
|
|
dfu_file_buf[CONFIG_SYS_DFU_MAX_FILE_SIZE];
|
|
static long dfu_file_buf_len;
|
|
|
|
static int mmc_block_op(enum dfu_op op, struct dfu_entity *dfu,
|
|
u64 offset, void *buf, long *len)
|
|
{
|
|
struct mmc *mmc = find_mmc_device(dfu->dev_num);
|
|
u32 blk_start, blk_count, n = 0;
|
|
|
|
/*
|
|
* We must ensure that we work in lba_blk_size chunks, so ALIGN
|
|
* this value.
|
|
*/
|
|
*len = ALIGN(*len, dfu->data.mmc.lba_blk_size);
|
|
|
|
blk_start = dfu->data.mmc.lba_start +
|
|
(u32)lldiv(offset, dfu->data.mmc.lba_blk_size);
|
|
blk_count = *len / dfu->data.mmc.lba_blk_size;
|
|
if (blk_start + blk_count >
|
|
dfu->data.mmc.lba_start + dfu->data.mmc.lba_size) {
|
|
puts("Request would exceed designated area!\n");
|
|
return -EINVAL;
|
|
}
|
|
|
|
debug("%s: %s dev: %d start: %d cnt: %d buf: 0x%p\n", __func__,
|
|
op == DFU_OP_READ ? "MMC READ" : "MMC WRITE", dfu->dev_num,
|
|
blk_start, blk_count, buf);
|
|
switch (op) {
|
|
case DFU_OP_READ:
|
|
n = mmc->block_dev.block_read(dfu->dev_num, blk_start,
|
|
blk_count, buf);
|
|
break;
|
|
case DFU_OP_WRITE:
|
|
n = mmc->block_dev.block_write(dfu->dev_num, blk_start,
|
|
blk_count, buf);
|
|
break;
|
|
default:
|
|
error("Operation not supported\n");
|
|
}
|
|
|
|
if (n != blk_count) {
|
|
error("MMC operation failed");
|
|
return -EIO;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int mmc_file_buffer(struct dfu_entity *dfu, void *buf, long *len)
|
|
{
|
|
if (dfu_file_buf_len + *len > CONFIG_SYS_DFU_MAX_FILE_SIZE) {
|
|
dfu_file_buf_len = 0;
|
|
return -EINVAL;
|
|
}
|
|
|
|
/* Add to the current buffer. */
|
|
memcpy(dfu_file_buf + dfu_file_buf_len, buf, *len);
|
|
dfu_file_buf_len += *len;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int mmc_file_op(enum dfu_op op, struct dfu_entity *dfu,
|
|
void *buf, long *len)
|
|
{
|
|
char cmd_buf[DFU_CMD_BUF_SIZE];
|
|
char *str_env;
|
|
int ret;
|
|
|
|
switch (dfu->layout) {
|
|
case DFU_FS_FAT:
|
|
sprintf(cmd_buf, "fat%s mmc %d:%d 0x%x %s",
|
|
op == DFU_OP_READ ? "load" : "write",
|
|
dfu->data.mmc.dev, dfu->data.mmc.part,
|
|
(unsigned int) buf, dfu->name);
|
|
break;
|
|
case DFU_FS_EXT4:
|
|
sprintf(cmd_buf, "ext4%s mmc %d:%d 0x%x /%s",
|
|
op == DFU_OP_READ ? "load" : "write",
|
|
dfu->data.mmc.dev, dfu->data.mmc.part,
|
|
(unsigned int) buf, dfu->name);
|
|
break;
|
|
default:
|
|
printf("%s: Layout (%s) not (yet) supported!\n", __func__,
|
|
dfu_get_layout(dfu->layout));
|
|
return -1;
|
|
}
|
|
|
|
if (op == DFU_OP_WRITE)
|
|
sprintf(cmd_buf + strlen(cmd_buf), " %lx", *len);
|
|
|
|
debug("%s: %s 0x%p\n", __func__, cmd_buf, cmd_buf);
|
|
|
|
ret = run_command(cmd_buf, 0);
|
|
if (ret) {
|
|
puts("dfu: Read error!\n");
|
|
return ret;
|
|
}
|
|
|
|
if (dfu->layout != DFU_RAW_ADDR && op == DFU_OP_READ) {
|
|
str_env = getenv("filesize");
|
|
if (str_env == NULL) {
|
|
puts("dfu: Wrong file size!\n");
|
|
return -1;
|
|
}
|
|
*len = simple_strtoul(str_env, NULL, 16);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int dfu_write_medium_mmc(struct dfu_entity *dfu,
|
|
u64 offset, void *buf, long *len)
|
|
{
|
|
int ret = -1;
|
|
|
|
switch (dfu->layout) {
|
|
case DFU_RAW_ADDR:
|
|
ret = mmc_block_op(DFU_OP_WRITE, dfu, offset, buf, len);
|
|
break;
|
|
case DFU_FS_FAT:
|
|
case DFU_FS_EXT4:
|
|
ret = mmc_file_buffer(dfu, buf, len);
|
|
break;
|
|
default:
|
|
printf("%s: Layout (%s) not (yet) supported!\n", __func__,
|
|
dfu_get_layout(dfu->layout));
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int dfu_flush_medium_mmc(struct dfu_entity *dfu)
|
|
{
|
|
int ret = 0;
|
|
|
|
if (dfu->layout != DFU_RAW_ADDR) {
|
|
/* Do stuff here. */
|
|
ret = mmc_file_op(DFU_OP_WRITE, dfu, &dfu_file_buf,
|
|
&dfu_file_buf_len);
|
|
|
|
/* Now that we're done */
|
|
dfu_file_buf_len = 0;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int dfu_read_medium_mmc(struct dfu_entity *dfu, u64 offset, void *buf,
|
|
long *len)
|
|
{
|
|
int ret = -1;
|
|
|
|
switch (dfu->layout) {
|
|
case DFU_RAW_ADDR:
|
|
ret = mmc_block_op(DFU_OP_READ, dfu, offset, buf, len);
|
|
break;
|
|
case DFU_FS_FAT:
|
|
case DFU_FS_EXT4:
|
|
ret = mmc_file_op(DFU_OP_READ, dfu, buf, len);
|
|
break;
|
|
default:
|
|
printf("%s: Layout (%s) not (yet) supported!\n", __func__,
|
|
dfu_get_layout(dfu->layout));
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int dfu_fill_entity_mmc(struct dfu_entity *dfu, char *s)
|
|
{
|
|
int dev, part;
|
|
struct mmc *mmc;
|
|
block_dev_desc_t *blk_dev;
|
|
disk_partition_t partinfo;
|
|
char *st;
|
|
|
|
dfu->dev_type = DFU_DEV_MMC;
|
|
st = strsep(&s, " ");
|
|
if (!strcmp(st, "mmc")) {
|
|
dfu->layout = DFU_RAW_ADDR;
|
|
dfu->data.mmc.lba_start = simple_strtoul(s, &s, 16);
|
|
dfu->data.mmc.lba_size = simple_strtoul(++s, &s, 16);
|
|
dfu->data.mmc.lba_blk_size = get_mmc_blk_size(dfu->dev_num);
|
|
} else if (!strcmp(st, "fat")) {
|
|
dfu->layout = DFU_FS_FAT;
|
|
} else if (!strcmp(st, "ext4")) {
|
|
dfu->layout = DFU_FS_EXT4;
|
|
} else if (!strcmp(st, "part")) {
|
|
|
|
dfu->layout = DFU_RAW_ADDR;
|
|
|
|
dev = simple_strtoul(s, &s, 10);
|
|
s++;
|
|
part = simple_strtoul(s, &s, 10);
|
|
|
|
mmc = find_mmc_device(dev);
|
|
if (mmc == NULL || mmc_init(mmc)) {
|
|
printf("%s: could not find mmc device #%d!\n",
|
|
__func__, dev);
|
|
return -ENODEV;
|
|
}
|
|
|
|
blk_dev = &mmc->block_dev;
|
|
if (get_partition_info(blk_dev, part, &partinfo) != 0) {
|
|
printf("%s: could not find partition #%d on mmc device #%d!\n",
|
|
__func__, part, dev);
|
|
return -ENODEV;
|
|
}
|
|
|
|
dfu->data.mmc.lba_start = partinfo.start;
|
|
dfu->data.mmc.lba_size = partinfo.size;
|
|
dfu->data.mmc.lba_blk_size = partinfo.blksz;
|
|
|
|
} else {
|
|
printf("%s: Memory layout (%s) not supported!\n", __func__, st);
|
|
return -ENODEV;
|
|
}
|
|
|
|
if (dfu->layout == DFU_FS_EXT4 || dfu->layout == DFU_FS_FAT) {
|
|
dfu->data.mmc.dev = simple_strtoul(s, &s, 10);
|
|
dfu->data.mmc.part = simple_strtoul(++s, &s, 10);
|
|
}
|
|
|
|
dfu->read_medium = dfu_read_medium_mmc;
|
|
dfu->write_medium = dfu_write_medium_mmc;
|
|
dfu->flush_medium = dfu_flush_medium_mmc;
|
|
|
|
/* initial state */
|
|
dfu->inited = 0;
|
|
|
|
return 0;
|
|
}
|