[people/ms/u-boot.git] / fs / fs.c

/*
 * Copyright (c) 2012, NVIDIA CORPORATION.  All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <config.h>
#include <common.h>
#include <part.h>
#include <ext4fs.h>
#include <fat.h>
#include <fs.h>
#include <sandboxfs.h>
#include <asm/io.h>

DECLARE_GLOBAL_DATA_PTR;

static block_dev_desc_t *fs_dev_desc;
static disk_partition_t fs_partition;
static int fs_type = FS_TYPE_ANY;

static inline int fs_probe_unsupported(block_dev_desc_t *fs_dev_desc,
				      disk_partition_t *fs_partition)
{
	printf("** Unrecognized filesystem type **\n");
	return -1;
}

static inline int fs_ls_unsupported(const char *dirname)
{
	return -1;
}

static inline int fs_read_unsupported(const char *filename, void *buf,
				      int offset, int len)
{
	return -1;
}

static inline int fs_write_unsupported(const char *filename, void *buf,
				      int offset, int len)
{
	return -1;
}

static inline void fs_close_unsupported(void)
{
}

struct fstype_info {
	int fstype;
	int (*probe)(block_dev_desc_t *fs_dev_desc,
		     disk_partition_t *fs_partition);
	int (*ls)(const char *dirname);
	int (*read)(const char *filename, void *buf, int offset, int len);
	int (*write)(const char *filename, void *buf, int offset, int len);
	void (*close)(void);
};

static struct fstype_info fstypes[] = {
#ifdef CONFIG_FS_FAT
	{
		.fstype = FS_TYPE_FAT,
		.probe = fat_set_blk_dev,
		.close = fat_close,
		.ls = file_fat_ls,
		.read = fat_read_file,
	},
#endif
#ifdef CONFIG_FS_EXT4
	{
		.fstype = FS_TYPE_EXT,
		.probe = ext4fs_probe,
		.close = ext4fs_close,
		.ls = ext4fs_ls,
		.read = ext4_read_file,
	},
#endif
#ifdef CONFIG_SANDBOX
	{
		.fstype = FS_TYPE_SANDBOX,
		.probe = sandbox_fs_set_blk_dev,
		.close = sandbox_fs_close,
		.ls = sandbox_fs_ls,
		.read = fs_read_sandbox,
		.write = fs_write_sandbox,
	},
#endif
	{
		.fstype = FS_TYPE_ANY,
		.probe = fs_probe_unsupported,
		.close = fs_close_unsupported,
		.ls = fs_ls_unsupported,
		.read = fs_read_unsupported,
		.write = fs_write_unsupported,
	},
};

static struct fstype_info *fs_get_info(int fstype)
{
	struct fstype_info *info;
	int i;

	for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes) - 1; i++, info++) {
		if (fstype == info->fstype)
			return info;
	}

	/* Return the 'unsupported' sentinel */
	return info;
}

int fs_set_blk_dev(const char *ifname, const char *dev_part_str, int fstype)
{
	struct fstype_info *info;
	int part, i;
#ifdef CONFIG_NEEDS_MANUAL_RELOC
	static int relocated;

	if (!relocated) {
		for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes);
				i++, info++) {
			info->probe += gd->reloc_off;
			info->close += gd->reloc_off;
			info->ls += gd->reloc_off;
			info->read += gd->reloc_off;
			info->write += gd->reloc_off;
		}
		relocated = 1;
	}
#endif

	part = get_device_and_partition(ifname, dev_part_str, &fs_dev_desc,
					&fs_partition, 1);
	if (part < 0)
		return -1;

	for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes); i++, info++) {
		if (fstype != FS_TYPE_ANY && info->fstype != FS_TYPE_ANY &&
				fstype != info->fstype)
			continue;

		if (!info->probe(fs_dev_desc, &fs_partition)) {
			fs_type = info->fstype;
			return 0;
		}
	}

	return -1;
}

static void fs_close(void)
{
	struct fstype_info *info = fs_get_info(fs_type);

	info->close();

	fs_type = FS_TYPE_ANY;
}

int fs_ls(const char *dirname)
{
	int ret;

	struct fstype_info *info = fs_get_info(fs_type);

	ret = info->ls(dirname);

	fs_type = FS_TYPE_ANY;
	fs_close();

	return ret;
}

int fs_read(const char *filename, ulong addr, int offset, int len)
{
	struct fstype_info *info = fs_get_info(fs_type);
	void *buf;
	int ret;

	/*
	 * We don't actually know how many bytes are being read, since len==0
	 * means read the whole file.
	 */
	buf = map_sysmem(addr, len);
	ret = info->read(filename, buf, offset, len);
	unmap_sysmem(buf);

	/* If we requested a specific number of bytes, check we got it */
	if (ret >= 0 && len && ret != len) {
		printf("** Unable to read file %s **\n", filename);
		ret = -1;
	}
	fs_close();

	return ret;
}

int fs_write(const char *filename, ulong addr, int offset, int len)
{
	struct fstype_info *info = fs_get_info(fs_type);
	void *buf;
	int ret;

	/*
	 * We don't actually know how many bytes are being read, since len==0
	 * means read the whole file.
	 */
	buf = map_sysmem(addr, len);
	ret = info->write(filename, buf, offset, len);
	unmap_sysmem(buf);

	/* If we requested a specific number of bytes, check we got it */
	if (ret >= 0 && len && ret != len) {
		printf("** Unable to write file %s **\n", filename);
		ret = -1;
	}
	fs_close();

	return ret;
}

int do_load(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
		int fstype, int cmdline_base)
{
	unsigned long addr;
	const char *addr_str;
	const char *filename;
	unsigned long bytes;
	unsigned long pos;
	int len_read;
	unsigned long time;

	if (argc < 2)
		return CMD_RET_USAGE;
	if (argc > 7)
		return CMD_RET_USAGE;

	if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype))
		return 1;

	if (argc >= 4) {
		addr = simple_strtoul(argv[3], NULL, cmdline_base);
	} else {
		addr_str = getenv("loadaddr");
		if (addr_str != NULL)
			addr = simple_strtoul(addr_str, NULL, 16);
		else
			addr = CONFIG_SYS_LOAD_ADDR;
	}
	if (argc >= 5) {
		filename = argv[4];
	} else {
		filename = getenv("bootfile");
		if (!filename) {
			puts("** No boot file defined **\n");
			return 1;
		}
	}
	if (argc >= 6)
		bytes = simple_strtoul(argv[5], NULL, cmdline_base);
	else
		bytes = 0;
	if (argc >= 7)
		pos = simple_strtoul(argv[6], NULL, cmdline_base);
	else
		pos = 0;

	time = get_timer(0);
	len_read = fs_read(filename, addr, pos, bytes);
	time = get_timer(time);
	if (len_read <= 0)
		return 1;

	printf("%d bytes read in %lu ms", len_read, time);
	if (time > 0) {
		puts(" (");
		print_size(len_read / time * 1000, "/s");
		puts(")");
	}
	puts("\n");

	setenv_hex("filesize", len_read);

	return 0;
}

int do_ls(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
	int fstype)
{
	if (argc < 2)
		return CMD_RET_USAGE;
	if (argc > 4)
		return CMD_RET_USAGE;

	if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype))
		return 1;

	if (fs_ls(argc >= 4 ? argv[3] : "/"))
		return 1;

	return 0;
}

int do_save(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[],
		int fstype, int cmdline_base)
{
	unsigned long addr;
	const char *filename;
	unsigned long bytes;
	unsigned long pos;
	int len;
	unsigned long time;

	if (argc < 6 || argc > 7)
		return CMD_RET_USAGE;

	if (fs_set_blk_dev(argv[1], argv[2], fstype))
		return 1;

	filename = argv[3];
	addr = simple_strtoul(argv[4], NULL, cmdline_base);
	bytes = simple_strtoul(argv[5], NULL, cmdline_base);
	if (argc >= 7)
		pos = simple_strtoul(argv[6], NULL, cmdline_base);
	else
		pos = 0;

	time = get_timer(0);
	len = fs_write(filename, addr, pos, bytes);
	time = get_timer(time);
	if (len <= 0)
		return 1;

	printf("%d bytes written in %lu ms", len, time);
	if (time > 0) {
		puts(" (");
		print_size(len / time * 1000, "/s");
		puts(")");
	}
	puts("\n");

	return 0;
}
Commit	Line	Data
045fa1e1 SW	1	/*
	2	* Copyright (c) 2012, NVIDIA CORPORATION. All rights reserved.
	3	*
	4	* This program is free software; you can redistribute it and/or modify it
	5	* under the terms and conditions of the GNU General Public License,
	6	* version 2, as published by the Free Software Foundation.
	7	*
	8	* This program is distributed in the hope it will be useful, but WITHOUT
	9	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
	10	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
	11	* more details.
	12	*
	13	* You should have received a copy of the GNU General Public License
	14	* along with this program. If not, see <http://www.gnu.org/licenses/>.
	15	*/
	16
	17	#include <config.h>
	18	#include <common.h>
	19	#include <part.h>
	20	#include <ext4fs.h>
	21	#include <fat.h>
	22	#include <fs.h>
92ccc96b	23	#include <sandboxfs.h>
117e0507	24	#include <asm/io.h>
045fa1e1	25
a1b231ce SW	26	DECLARE_GLOBAL_DATA_PTR;
a1b231ce SW	27
045fa1e1 SW	28	static block_dev_desc_t *fs_dev_desc;
	29	static disk_partition_t fs_partition;
	30	static int fs_type = FS_TYPE_ANY;
	31
2ded0d47 SG	32	static inline int fs_probe_unsupported(block_dev_desc_t *fs_dev_desc,
2ded0d47 SG	33	disk_partition_t *fs_partition)
436e2b73 SG	34	{
	35	printf(" Unrecognized filesystem type \n");
	36	return -1;
	37	}
	38
045fa1e1 SW	39	static inline int fs_ls_unsupported(const char *dirname)
045fa1e1 SW	40	{
045fa1e1 SW	41	return -1;
	42	}
	43
117e0507	44	static inline int fs_read_unsupported(const char filename, void buf,
045fa1e1 SW	45	int offset, int len)
045fa1e1 SW	46	{
045fa1e1 SW	47	return -1;
	48	}
	49
a8f6ab52 SG	50	static inline int fs_write_unsupported(const char filename, void buf,
	51	int offset, int len)
	52	{
	53	return -1;
	54	}
	55
436e2b73 SG	56	static inline void fs_close_unsupported(void)
	57	{
	58	}
	59
436e2b73	60	struct fstype_info {
045fa1e1	61	int fstype;
2ded0d47 SG	62	int (probe)(block_dev_desc_t fs_dev_desc,
2ded0d47 SG	63	disk_partition_t *fs_partition);
436e2b73	64	int (ls)(const char dirname);
117e0507	65	int (read)(const char filename, void *buf, int offset, int len);
a8f6ab52	66	int (write)(const char filename, void *buf, int offset, int len);
436e2b73 SG	67	void (*close)(void);
	68	};
	69
	70	static struct fstype_info fstypes[] = {
	71	#ifdef CONFIG_FS_FAT
045fa1e1 SW	72	{
045fa1e1 SW	73	.fstype = FS_TYPE_FAT,
e6d52415 SG	74	.probe = fat_set_blk_dev,
e6d52415 SG	75	.close = fat_close,
436e2b73	76	.ls = file_fat_ls,
e6d52415	77	.read = fat_read_file,
045fa1e1	78	},
436e2b73 SG	79	#endif
436e2b73 SG	80	#ifdef CONFIG_FS_EXT4
045fa1e1 SW	81	{
045fa1e1 SW	82	.fstype = FS_TYPE_EXT,
e6d52415 SG	83	.probe = ext4fs_probe,
e6d52415 SG	84	.close = ext4fs_close,
436e2b73	85	.ls = ext4fs_ls,
e6d52415	86	.read = ext4_read_file,
436e2b73	87	},
92ccc96b SG	88	#endif
	89	#ifdef CONFIG_SANDBOX
	90	{
	91	.fstype = FS_TYPE_SANDBOX,
	92	.probe = sandbox_fs_set_blk_dev,
	93	.close = sandbox_fs_close,
	94	.ls = sandbox_fs_ls,
	95	.read = fs_read_sandbox,
7eb2c8d5	96	.write = fs_write_sandbox,
92ccc96b	97	},
436e2b73 SG	98	#endif
	99	{
	100	.fstype = FS_TYPE_ANY,
	101	.probe = fs_probe_unsupported,
	102	.close = fs_close_unsupported,
	103	.ls = fs_ls_unsupported,
	104	.read = fs_read_unsupported,
a8f6ab52	105	.write = fs_write_unsupported,
045fa1e1 SW	106	},
	107	};
	108
c6f548d2 SG	109	static struct fstype_info *fs_get_info(int fstype)
	110	{
	111	struct fstype_info *info;
	112	int i;
	113
	114	for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes) - 1; i++, info++) {
	115	if (fstype == info->fstype)
	116	return info;
	117	}
	118
	119	/* Return the 'unsupported' sentinel */
	120	return info;
	121	}
	122
045fa1e1 SW	123	int fs_set_blk_dev(const char ifname, const char dev_part_str, int fstype)
045fa1e1 SW	124	{
436e2b73	125	struct fstype_info *info;
045fa1e1	126	int part, i;
a1b231ce SW	127	#ifdef CONFIG_NEEDS_MANUAL_RELOC
	128	static int relocated;
	129
	130	if (!relocated) {
436e2b73 SG	131	for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes);
	132	i++, info++) {
	133	info->probe += gd->reloc_off;
	134	info->close += gd->reloc_off;
	135	info->ls += gd->reloc_off;
	136	info->read += gd->reloc_off;
a8f6ab52	137	info->write += gd->reloc_off;
436e2b73	138	}
a1b231ce SW	139	relocated = 1;
	140	}
	141	#endif
045fa1e1 SW	142
	143	part = get_device_and_partition(ifname, dev_part_str, &fs_dev_desc,
	144	&fs_partition, 1);
	145	if (part < 0)
	146	return -1;
	147
436e2b73 SG	148	for (i = 0, info = fstypes; i < ARRAY_SIZE(fstypes); i++, info++) {
	149	if (fstype != FS_TYPE_ANY && info->fstype != FS_TYPE_ANY &&
	150	fstype != info->fstype)
045fa1e1 SW	151	continue;
045fa1e1 SW	152
2ded0d47	153	if (!info->probe(fs_dev_desc, &fs_partition)) {
436e2b73	154	fs_type = info->fstype;
045fa1e1 SW	155	return 0;
	156	}
	157	}
	158
045fa1e1 SW	159	return -1;
	160	}
	161
	162	static void fs_close(void)
	163	{
c6f548d2	164	struct fstype_info *info = fs_get_info(fs_type);
045fa1e1	165
c6f548d2	166	info->close();
e6d52415	167
045fa1e1 SW	168	fs_type = FS_TYPE_ANY;
	169	}
	170
	171	int fs_ls(const char *dirname)
	172	{
	173	int ret;
	174
c6f548d2 SG	175	struct fstype_info *info = fs_get_info(fs_type);
	176
	177	ret = info->ls(dirname);
045fa1e1	178
e6d52415	179	fs_type = FS_TYPE_ANY;
045fa1e1 SW	180	fs_close();
	181
	182	return ret;
	183	}
	184
	185	int fs_read(const char *filename, ulong addr, int offset, int len)
	186	{
c6f548d2	187	struct fstype_info *info = fs_get_info(fs_type);
117e0507	188	void *buf;
045fa1e1 SW	189	int ret;
045fa1e1 SW	190
117e0507 SG	191	/*
	192	* We don't actually know how many bytes are being read, since len==0
	193	* means read the whole file.
	194	*/
	195	buf = map_sysmem(addr, len);
	196	ret = info->read(filename, buf, offset, len);
	197	unmap_sysmem(buf);
045fa1e1	198
c6f548d2 SG	199	/* If we requested a specific number of bytes, check we got it */
	200	if (ret >= 0 && len && ret != len) {
	201	printf(" Unable to read file %s \n", filename);
	202	ret = -1;
	203	}
045fa1e1 SW	204	fs_close();
	205
	206	return ret;
	207	}
	208
a8f6ab52 SG	209	int fs_write(const char *filename, ulong addr, int offset, int len)
	210	{
	211	struct fstype_info *info = fs_get_info(fs_type);
	212	void *buf;
	213	int ret;
	214
	215	/*
	216	* We don't actually know how many bytes are being read, since len==0
	217	* means read the whole file.
	218	*/
	219	buf = map_sysmem(addr, len);
	220	ret = info->write(filename, buf, offset, len);
	221	unmap_sysmem(buf);
	222
	223	/* If we requested a specific number of bytes, check we got it */
	224	if (ret >= 0 && len && ret != len) {
	225	printf(" Unable to write file %s \n", filename);
	226	ret = -1;
	227	}
	228	fs_close();
	229
	230	return ret;
	231	}
	232
f9b55e22	233	int do_load(cmd_tbl_t cmdtp, int flag, int argc, char const argv[],
3f83c87e	234	int fstype, int cmdline_base)
045fa1e1 SW	235	{
	236	unsigned long addr;
	237	const char *addr_str;
	238	const char *filename;
	239	unsigned long bytes;
	240	unsigned long pos;
	241	int len_read;
da1fd96c	242	unsigned long time;
045fa1e1	243
e9b0f99e SW	244	if (argc < 2)
	245	return CMD_RET_USAGE;
	246	if (argc > 7)
045fa1e1 SW	247	return CMD_RET_USAGE;
045fa1e1 SW	248
e9b0f99e	249	if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype))
045fa1e1 SW	250	return 1;
	251
	252	if (argc >= 4) {
3f83c87e	253	addr = simple_strtoul(argv[3], NULL, cmdline_base);
045fa1e1 SW	254	} else {
	255	addr_str = getenv("loadaddr");
	256	if (addr_str != NULL)
	257	addr = simple_strtoul(addr_str, NULL, 16);
	258	else
	259	addr = CONFIG_SYS_LOAD_ADDR;
	260	}
	261	if (argc >= 5) {
	262	filename = argv[4];
	263	} else {
	264	filename = getenv("bootfile");
	265	if (!filename) {
	266	puts(" No boot file defined \n");
	267	return 1;
	268	}
	269	}
	270	if (argc >= 6)
3f83c87e	271	bytes = simple_strtoul(argv[5], NULL, cmdline_base);
045fa1e1 SW	272	else
	273	bytes = 0;
	274	if (argc >= 7)
3f83c87e	275	pos = simple_strtoul(argv[6], NULL, cmdline_base);
045fa1e1 SW	276	else
	277	pos = 0;
	278
da1fd96c	279	time = get_timer(0);
045fa1e1	280	len_read = fs_read(filename, addr, pos, bytes);
da1fd96c	281	time = get_timer(time);
045fa1e1 SW	282	if (len_read <= 0)
	283	return 1;
	284
da1fd96c AB	285	printf("%d bytes read in %lu ms", len_read, time);
	286	if (time > 0) {
	287	puts(" (");
	288	print_size(len_read / time * 1000, "/s");
	289	puts(")");
	290	}
	291	puts("\n");
045fa1e1	292
49c4f037	293	setenv_hex("filesize", len_read);
045fa1e1 SW	294
	295	return 0;
	296	}
	297
	298	int do_ls(cmd_tbl_t cmdtp, int flag, int argc, char const argv[],
	299	int fstype)
	300	{
	301	if (argc < 2)
	302	return CMD_RET_USAGE;
e9b0f99e SW	303	if (argc > 4)
e9b0f99e SW	304	return CMD_RET_USAGE;
045fa1e1 SW	305
	306	if (fs_set_blk_dev(argv[1], (argc >= 3) ? argv[2] : NULL, fstype))
	307	return 1;
	308
e9b0f99e	309	if (fs_ls(argc >= 4 ? argv[3] : "/"))
045fa1e1 SW	310	return 1;
	311
	312	return 0;
	313	}
a8f6ab52 SG	314
	315	int do_save(cmd_tbl_t cmdtp, int flag, int argc, char const argv[],
	316	int fstype, int cmdline_base)
	317	{
	318	unsigned long addr;
	319	const char *filename;
	320	unsigned long bytes;
	321	unsigned long pos;
	322	int len;
	323	unsigned long time;
	324
	325	if (argc < 6 \|\| argc > 7)
	326	return CMD_RET_USAGE;
	327
	328	if (fs_set_blk_dev(argv[1], argv[2], fstype))
	329	return 1;
	330
	331	filename = argv[3];
	332	addr = simple_strtoul(argv[4], NULL, cmdline_base);
	333	bytes = simple_strtoul(argv[5], NULL, cmdline_base);
	334	if (argc >= 7)
	335	pos = simple_strtoul(argv[6], NULL, cmdline_base);
	336	else
	337	pos = 0;
	338
	339	time = get_timer(0);
	340	len = fs_write(filename, addr, pos, bytes);
	341	time = get_timer(time);
	342	if (len <= 0)
	343	return 1;
	344
	345	printf("%d bytes written in %lu ms", len, time);
	346	if (time > 0) {
	347	puts(" (");
	348	print_size(len / time * 1000, "/s");
	349	puts(")");
	350	}
	351	puts("\n");
	352
	353	return 0;
	354	}