[people/ms/u-boot.git] / cmd / zfs.c

/*
 *
 * ZFS filesystem porting to Uboot by
 * Jorgen Lundman <lundman at lundman.net>
 *
 * zfsfs support
 * made from existing GRUB Sources by Sun, GNU and others.
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */

#include <common.h>
#include <part.h>
#include <config.h>
#include <command.h>
#include <image.h>
#include <linux/ctype.h>
#include <asm/byteorder.h>
#include <zfs_common.h>
#include <linux/stat.h>
#include <malloc.h>

#if defined(CONFIG_CMD_USB) && defined(CONFIG_USB_STORAGE)
#include <usb.h>
#endif

#if !CONFIG_IS_ENABLED(DOS_PARTITION) && !CONFIG_IS_ENABLED(EFI_PARTITION)
#error DOS or EFI partition support must be selected
#endif

#define DOS_PART_MAGIC_OFFSET	0x1fe
#define DOS_FS_TYPE_OFFSET	0x36
#define DOS_FS32_TYPE_OFFSET	0x52

static int do_zfs_load(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	char *filename = NULL;
	int dev;
	int part;
	ulong addr = 0;
	disk_partition_t info;
	struct blk_desc *dev_desc;
	char buf[12];
	unsigned long count;
	const char *addr_str;
	struct zfs_file zfile;
	struct device_s vdev;

	if (argc < 3)
		return CMD_RET_USAGE;

	count = 0;
	addr = simple_strtoul(argv[3], NULL, 16);
	filename = getenv("bootfile");
	switch (argc) {
	case 3:
		addr_str = getenv("loadaddr");
		if (addr_str != NULL)
			addr = simple_strtoul(addr_str, NULL, 16);
		else
			addr = CONFIG_SYS_LOAD_ADDR;

		break;
	case 4:
		break;
	case 5:
		filename = argv[4];
		break;
	case 6:
		filename = argv[4];
		count = simple_strtoul(argv[5], NULL, 16);
		break;

	default:
		return cmd_usage(cmdtp);
	}

	if (!filename) {
		puts("** No boot file defined **\n");
		return 1;
	}

	part = blk_get_device_part_str(argv[1], argv[2], &dev_desc, &info, 1);
	if (part < 0)
		return 1;

	dev = dev_desc->devnum;
	printf("Loading file \"%s\" from %s device %d%c%c\n",
		filename, argv[1], dev,
		part ? ':' : ' ', part ? part + '0' : ' ');

	zfs_set_blk_dev(dev_desc, &info);
	vdev.part_length = info.size;

	memset(&zfile, 0, sizeof(zfile));
	zfile.device = &vdev;
	if (zfs_open(&zfile, filename)) {
		printf("** File not found %s **\n", filename);
		return 1;
	}

	if ((count < zfile.size) && (count != 0))
		zfile.size = (uint64_t)count;

	if (zfs_read(&zfile, (char *)addr, zfile.size) != zfile.size) {
		printf("** Unable to read \"%s\" from %s %d:%d **\n",
			   filename, argv[1], dev, part);
		zfs_close(&zfile);
		return 1;
	}

	zfs_close(&zfile);

	/* Loading ok, update default load address */
	load_addr = addr;

	printf("%llu bytes read\n", zfile.size);
	env_set_hex("filesize", zfile.size);

	return 0;
}


int zfs_print(const char *entry, const struct zfs_dirhook_info *data)
{
	printf("%s %s\n",
		   data->dir ? "<DIR> " : "		 ",
		   entry);
	return 0; /* 0 continue, 1 stop */
}


static int do_zfs_ls(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	const char *filename = "/";
	int part;
	struct blk_desc *dev_desc;
	disk_partition_t info;
	struct device_s vdev;

	if (argc < 2)
		return cmd_usage(cmdtp);

	if (argc == 4)
		filename = argv[3];

	part = blk_get_device_part_str(argv[1], argv[2], &dev_desc, &info, 1);
	if (part < 0)
		return 1;

	zfs_set_blk_dev(dev_desc, &info);
	vdev.part_length = info.size;

	zfs_ls(&vdev, filename,
		   zfs_print);

	return 0;
}


U_BOOT_CMD(zfsls, 4, 1, do_zfs_ls,
		   "list files in a directory (default /)",
		   "<interface> <dev[:part]> [directory]\n"
		   "	  - list files from 'dev' on 'interface' in a '/DATASET/@/$dir/'");

U_BOOT_CMD(zfsload, 6, 0, do_zfs_load,
		   "load binary file from a ZFS filesystem",
		   "<interface> <dev[:part]> [addr] [filename] [bytes]\n"
		   "	  - load binary file '/DATASET/@/$dir/$file' from 'dev' on 'interface'\n"
		   "		 to address 'addr' from ZFS filesystem");
Commit	Line	Data
4d3c95f5 JL	1	/*
	2	*
	3	* ZFS filesystem porting to Uboot by
	4	* Jorgen Lundman <lundman at lundman.net>
	5	*
	6	* zfsfs support
	7	* made from existing GRUB Sources by Sun, GNU and others.
	8	*
1a459660	9	* SPDX-License-Identifier: GPL-2.0+
4d3c95f5 JL	10	*/
	11
	12	#include <common.h>
	13	#include <part.h>
	14	#include <config.h>
	15	#include <command.h>
	16	#include <image.h>
	17	#include <linux/ctype.h>
	18	#include <asm/byteorder.h>
	19	#include <zfs_common.h>
	20	#include <linux/stat.h>
	21	#include <malloc.h>
	22
	23	#if defined(CONFIG_CMD_USB) && defined(CONFIG_USB_STORAGE)
	24	#include <usb.h>
	25	#endif
	26
bd42a942	27	#if !CONFIG_IS_ENABLED(DOS_PARTITION) && !CONFIG_IS_ENABLED(EFI_PARTITION)
4d3c95f5 JL	28	#error DOS or EFI partition support must be selected
	29	#endif
	30
	31	#define DOS_PART_MAGIC_OFFSET 0x1fe
	32	#define DOS_FS_TYPE_OFFSET 0x36
	33	#define DOS_FS32_TYPE_OFFSET 0x52
	34
af8d1d3f	35	static int do_zfs_load(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
4d3c95f5 JL	36	{
4d3c95f5 JL	37	char *filename = NULL;
4d3c95f5	38	int dev;
41204570	39	int part;
4d3c95f5	40	ulong addr = 0;
4d3c95f5	41	disk_partition_t info;
4101f687	42	struct blk_desc *dev_desc;
4d3c95f5 JL	43	char buf[12];
	44	unsigned long count;
	45	const char *addr_str;
	46	struct zfs_file zfile;
	47	struct device_s vdev;
	48
	49	if (argc < 3)
	50	return CMD_RET_USAGE;
	51
	52	count = 0;
	53	addr = simple_strtoul(argv[3], NULL, 16);
	54	filename = getenv("bootfile");
	55	switch (argc) {
	56	case 3:
	57	addr_str = getenv("loadaddr");
	58	if (addr_str != NULL)
	59	addr = simple_strtoul(addr_str, NULL, 16);
	60	else
	61	addr = CONFIG_SYS_LOAD_ADDR;
	62
	63	break;
	64	case 4:
	65	break;
	66	case 5:
	67	filename = argv[4];
	68	break;
	69	case 6:
	70	filename = argv[4];
	71	count = simple_strtoul(argv[5], NULL, 16);
	72	break;
	73
	74	default:
	75	return cmd_usage(cmdtp);
	76	}
	77
	78	if (!filename) {
	79	puts(" No boot file defined \n");
	80	return 1;
	81	}
	82
e35929e4	83	part = blk_get_device_part_str(argv[1], argv[2], &dev_desc, &info, 1);
41204570	84	if (part < 0)
4d3c95f5	85	return 1;
4d3c95f5	86
bcce53d0	87	dev = dev_desc->devnum;
41204570 RH	88	printf("Loading file \"%s\" from %s device %d%c%c\n",
	89	filename, argv[1], dev,
	90	part ? ':' : ' ', part ? part + '0' : ' ');
4d3c95f5	91
41204570 RH	92	zfs_set_blk_dev(dev_desc, &info);
41204570 RH	93	vdev.part_length = info.size;
4d3c95f5 JL	94
	95	memset(&zfile, 0, sizeof(zfile));
	96	zfile.device = &vdev;
	97	if (zfs_open(&zfile, filename)) {
01adbce2	98	printf(" File not found %s \n", filename);
4d3c95f5 JL	99	return 1;
	100	}
	101
	102	if ((count < zfile.size) && (count != 0))
	103	zfile.size = (uint64_t)count;
	104
	105	if (zfs_read(&zfile, (char *)addr, zfile.size) != zfile.size) {
41204570	106	printf(" Unable to read \"%s\" from %s %d:%d \n",
4d3c95f5 JL	107	filename, argv[1], dev, part);
	108	zfs_close(&zfile);
	109	return 1;
	110	}
	111
	112	zfs_close(&zfile);
	113
	114	/* Loading ok, update default load address */
	115	load_addr = addr;
	116
	117	printf("%llu bytes read\n", zfile.size);
018f5303	118	env_set_hex("filesize", zfile.size);
4d3c95f5 JL	119
	120	return 0;
	121	}
	122
	123
	124	int zfs_print(const char entry, const struct zfs_dirhook_info data)
	125	{
	126	printf("%s %s\n",
	127	data->dir ? "<DIR> " : " ",
	128	entry);
	129	return 0; /* 0 continue, 1 stop */
	130	}
	131
	132
	133
af8d1d3f	134	static int do_zfs_ls(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
4d3c95f5 JL	135	{
4d3c95f5 JL	136	const char *filename = "/";
41204570	137	int part;
4101f687	138	struct blk_desc *dev_desc;
41204570	139	disk_partition_t info;
4d3c95f5 JL	140	struct device_s vdev;
4d3c95f5 JL	141
41204570	142	if (argc < 2)
4d3c95f5 JL	143	return cmd_usage(cmdtp);
4d3c95f5 JL	144
4d3c95f5 JL	145	if (argc == 4)
	146	filename = argv[3];
	147
e35929e4	148	part = blk_get_device_part_str(argv[1], argv[2], &dev_desc, &info, 1);
41204570	149	if (part < 0)
4d3c95f5	150	return 1;
4d3c95f5	151
41204570 RH	152	zfs_set_blk_dev(dev_desc, &info);
41204570 RH	153	vdev.part_length = info.size;
4d3c95f5 JL	154
	155	zfs_ls(&vdev, filename,
	156	zfs_print);
	157
	158	return 0;
	159	}
	160
	161
	162	U_BOOT_CMD(zfsls, 4, 1, do_zfs_ls,
	163	"list files in a directory (default /)",
	164	"<interface> <dev[:part]> [directory]\n"
	165	" - list files from 'dev' on 'interface' in a '/DATASET/@/$dir/'");
	166
	167	U_BOOT_CMD(zfsload, 6, 0, do_zfs_load,
	168	"load binary file from a ZFS filesystem",
	169	"<interface> <dev[:part]> [addr] [filename] [bytes]\n"
	170	" - load binary file '/DATASET/@/$dir/$file' from 'dev' on 'interface'\n"
	171	" to address 'addr' from ZFS filesystem");