2 * dfu.c -- DFU back-end routines
4 * Copyright (C) 2012 Samsung Electronics
5 * author: Lukasz Majewski <l.majewski@samsung.com>
7 * SPDX-License-Identifier: GPL-2.0+
17 static unsigned char __aligned(CONFIG_SYS_CACHELINE_SIZE
)
18 dfu_file_buf
[CONFIG_SYS_DFU_MAX_FILE_SIZE
];
19 static long dfu_file_buf_len
;
21 static int mmc_block_op(enum dfu_op op
, struct dfu_entity
*dfu
,
22 u64 offset
, void *buf
, long *len
)
24 struct mmc
*mmc
= find_mmc_device(dfu
->dev_num
);
25 u32 blk_start
, blk_count
, n
= 0;
28 * We must ensure that we work in lba_blk_size chunks, so ALIGN
31 *len
= ALIGN(*len
, dfu
->data
.mmc
.lba_blk_size
);
33 blk_start
= dfu
->data
.mmc
.lba_start
+
34 (u32
)lldiv(offset
, dfu
->data
.mmc
.lba_blk_size
);
35 blk_count
= *len
/ dfu
->data
.mmc
.lba_blk_size
;
36 if (blk_start
+ blk_count
>
37 dfu
->data
.mmc
.lba_start
+ dfu
->data
.mmc
.lba_size
) {
38 puts("Request would exceed designated area!\n");
42 debug("%s: %s dev: %d start: %d cnt: %d buf: 0x%p\n", __func__
,
43 op
== DFU_OP_READ
? "MMC READ" : "MMC WRITE", dfu
->dev_num
,
44 blk_start
, blk_count
, buf
);
47 n
= mmc
->block_dev
.block_read(dfu
->dev_num
, blk_start
,
51 n
= mmc
->block_dev
.block_write(dfu
->dev_num
, blk_start
,
55 error("Operation not supported\n");
59 error("MMC operation failed");
66 static int mmc_file_buffer(struct dfu_entity
*dfu
, void *buf
, long *len
)
68 if (dfu_file_buf_len
+ *len
> CONFIG_SYS_DFU_MAX_FILE_SIZE
) {
73 /* Add to the current buffer. */
74 memcpy(dfu_file_buf
+ dfu_file_buf_len
, buf
, *len
);
75 dfu_file_buf_len
+= *len
;
80 static int mmc_file_op(enum dfu_op op
, struct dfu_entity
*dfu
,
83 char cmd_buf
[DFU_CMD_BUF_SIZE
];
87 switch (dfu
->layout
) {
89 sprintf(cmd_buf
, "fat%s mmc %d:%d 0x%x %s",
90 op
== DFU_OP_READ
? "load" : "write",
91 dfu
->data
.mmc
.dev
, dfu
->data
.mmc
.part
,
92 (unsigned int) buf
, dfu
->name
);
95 sprintf(cmd_buf
, "ext4%s mmc %d:%d 0x%x /%s",
96 op
== DFU_OP_READ
? "load" : "write",
97 dfu
->data
.mmc
.dev
, dfu
->data
.mmc
.part
,
98 (unsigned int) buf
, dfu
->name
);
101 printf("%s: Layout (%s) not (yet) supported!\n", __func__
,
102 dfu_get_layout(dfu
->layout
));
106 if (op
== DFU_OP_WRITE
)
107 sprintf(cmd_buf
+ strlen(cmd_buf
), " %lx", *len
);
109 debug("%s: %s 0x%p\n", __func__
, cmd_buf
, cmd_buf
);
111 ret
= run_command(cmd_buf
, 0);
113 puts("dfu: Read error!\n");
117 if (dfu
->layout
!= DFU_RAW_ADDR
&& op
== DFU_OP_READ
) {
118 str_env
= getenv("filesize");
119 if (str_env
== NULL
) {
120 puts("dfu: Wrong file size!\n");
123 *len
= simple_strtoul(str_env
, NULL
, 16);
129 int dfu_write_medium_mmc(struct dfu_entity
*dfu
,
130 u64 offset
, void *buf
, long *len
)
134 switch (dfu
->layout
) {
136 ret
= mmc_block_op(DFU_OP_WRITE
, dfu
, offset
, buf
, len
);
140 ret
= mmc_file_buffer(dfu
, buf
, len
);
143 printf("%s: Layout (%s) not (yet) supported!\n", __func__
,
144 dfu_get_layout(dfu
->layout
));
150 int dfu_flush_medium_mmc(struct dfu_entity
*dfu
)
154 if (dfu
->layout
!= DFU_RAW_ADDR
) {
156 ret
= mmc_file_op(DFU_OP_WRITE
, dfu
, &dfu_file_buf
,
159 /* Now that we're done */
160 dfu_file_buf_len
= 0;
166 int dfu_read_medium_mmc(struct dfu_entity
*dfu
, u64 offset
, void *buf
,
171 switch (dfu
->layout
) {
173 ret
= mmc_block_op(DFU_OP_READ
, dfu
, offset
, buf
, len
);
177 ret
= mmc_file_op(DFU_OP_READ
, dfu
, buf
, len
);
180 printf("%s: Layout (%s) not (yet) supported!\n", __func__
,
181 dfu_get_layout(dfu
->layout
));
188 * @param s Parameter string containing space-separated arguments:
190 * raw (raw read/write)
193 * part (partition image)
195 * lba_start and lba_size, for raw write
196 * mmc_dev and mmc_part, for filesystems and part
198 int dfu_fill_entity_mmc(struct dfu_entity
*dfu
, char *s
)
200 const char *entity_type
;
207 const char **parg
= argv
;
209 for (; parg
< argv
+ sizeof(argv
) / sizeof(*argv
); ++parg
) {
210 *parg
= strsep(&s
, " ");
212 error("Invalid number of arguments.\n");
217 entity_type
= argv
[0];
219 * Base 0 means we'll accept (prefixed with 0x or 0) base 16, 8,
222 second_arg
= simple_strtoul(argv
[1], NULL
, 0);
223 third_arg
= simple_strtoul(argv
[2], NULL
, 0);
225 mmc
= find_mmc_device(dfu
->dev_num
);
227 error("Couldn't find MMC device no. %d.\n", dfu
->dev_num
);
232 error("Couldn't init MMC device.\n");
236 if (!strcmp(entity_type
, "raw")) {
237 dfu
->layout
= DFU_RAW_ADDR
;
238 dfu
->data
.mmc
.lba_start
= second_arg
;
239 dfu
->data
.mmc
.lba_size
= third_arg
;
240 dfu
->data
.mmc
.lba_blk_size
= mmc
->read_bl_len
;
241 } else if (!strcmp(entity_type
, "part")) {
242 disk_partition_t partinfo
;
243 block_dev_desc_t
*blk_dev
= &mmc
->block_dev
;
244 int mmcdev
= second_arg
;
245 int mmcpart
= third_arg
;
247 if (get_partition_info(blk_dev
, mmcpart
, &partinfo
) != 0) {
248 error("Couldn't find part #%d on mmc device #%d\n",
253 dfu
->layout
= DFU_RAW_ADDR
;
254 dfu
->data
.mmc
.lba_start
= partinfo
.start
;
255 dfu
->data
.mmc
.lba_size
= partinfo
.size
;
256 dfu
->data
.mmc
.lba_blk_size
= partinfo
.blksz
;
257 } else if (!strcmp(entity_type
, "fat")) {
258 dfu
->layout
= DFU_FS_FAT
;
259 } else if (!strcmp(entity_type
, "ext4")) {
260 dfu
->layout
= DFU_FS_EXT4
;
262 error("Memory layout (%s) not supported!\n", entity_type
);
266 /* if it's NOT a raw write */
267 if (strcmp(entity_type
, "raw")) {
268 dfu
->data
.mmc
.dev
= second_arg
;
269 dfu
->data
.mmc
.part
= third_arg
;
272 dfu
->dev_type
= DFU_DEV_MMC
;
273 dfu
->read_medium
= dfu_read_medium_mmc
;
274 dfu
->write_medium
= dfu_write_medium_mmc
;
275 dfu
->flush_medium
= dfu_flush_medium_mmc
;