[thirdparty/u-boot.git] / cmd / zfs.c

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

#include <common.h>
#include <part.h>
#include <config.h>
#include <command.h>
#include <env.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;
	struct disk_partition info;
	struct blk_desc *dev_desc;
	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 = env_get("bootfile");
	switch (argc) {
	case 3:
		addr_str = env_get("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 */
	image_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;
	struct disk_partition 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
83d290c5	1	// SPDX-License-Identifier: GPL-2.0+
4d3c95f5 JL	2	/*
	3	*
	4	* ZFS filesystem porting to Uboot by
	5	* Jorgen Lundman <lundman at lundman.net>
	6	*
	7	* zfsfs support
	8	* made from existing GRUB Sources by Sun, GNU and others.
4d3c95f5 JL	9	*/
	10
	11	#include <common.h>
	12	#include <part.h>
	13	#include <config.h>
	14	#include <command.h>
c7694dd4	15	#include <env.h>
4d3c95f5 JL	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;
0528979f	41	struct disk_partition info;
4101f687	42	struct blk_desc *dev_desc;
4d3c95f5 JL	43	unsigned long count;
	44	const char *addr_str;
	45	struct zfs_file zfile;
	46	struct device_s vdev;
	47
	48	if (argc < 3)
	49	return CMD_RET_USAGE;
	50
	51	count = 0;
	52	addr = simple_strtoul(argv[3], NULL, 16);
00caae6d	53	filename = env_get("bootfile");
4d3c95f5 JL	54	switch (argc) {
4d3c95f5 JL	55	case 3:
00caae6d	56	addr_str = env_get("loadaddr");
4d3c95f5 JL	57	if (addr_str != NULL)
	58	addr = simple_strtoul(addr_str, NULL, 16);
	59	else
	60	addr = CONFIG_SYS_LOAD_ADDR;
	61
	62	break;
	63	case 4:
	64	break;
	65	case 5:
	66	filename = argv[4];
	67	break;
	68	case 6:
	69	filename = argv[4];
	70	count = simple_strtoul(argv[5], NULL, 16);
	71	break;
	72
	73	default:
	74	return cmd_usage(cmdtp);
	75	}
	76
	77	if (!filename) {
	78	puts(" No boot file defined \n");
	79	return 1;
	80	}
	81
e35929e4	82	part = blk_get_device_part_str(argv[1], argv[2], &dev_desc, &info, 1);
41204570	83	if (part < 0)
4d3c95f5	84	return 1;
4d3c95f5	85
bcce53d0	86	dev = dev_desc->devnum;
41204570 RH	87	printf("Loading file \"%s\" from %s device %d%c%c\n",
	88	filename, argv[1], dev,
	89	part ? ':' : ' ', part ? part + '0' : ' ');
4d3c95f5	90
41204570 RH	91	zfs_set_blk_dev(dev_desc, &info);
41204570 RH	92	vdev.part_length = info.size;
4d3c95f5 JL	93
	94	memset(&zfile, 0, sizeof(zfile));
	95	zfile.device = &vdev;
	96	if (zfs_open(&zfile, filename)) {
01adbce2	97	printf(" File not found %s \n", filename);
4d3c95f5 JL	98	return 1;
	99	}
	100
	101	if ((count < zfile.size) && (count != 0))
	102	zfile.size = (uint64_t)count;
	103
	104	if (zfs_read(&zfile, (char *)addr, zfile.size) != zfile.size) {
41204570	105	printf(" Unable to read \"%s\" from %s %d:%d \n",
4d3c95f5 JL	106	filename, argv[1], dev, part);
	107	zfs_close(&zfile);
	108	return 1;
	109	}
	110
	111	zfs_close(&zfile);
	112
	113	/* Loading ok, update default load address */
bb872dd9	114	image_load_addr = addr;
4d3c95f5 JL	115
4d3c95f5 JL	116	printf("%llu bytes read\n", zfile.size);
018f5303	117	env_set_hex("filesize", zfile.size);
4d3c95f5 JL	118
	119	return 0;
	120	}
	121
	122
	123	int zfs_print(const char entry, const struct zfs_dirhook_info data)
	124	{
	125	printf("%s %s\n",
	126	data->dir ? "<DIR> " : " ",
	127	entry);
	128	return 0; /* 0 continue, 1 stop */
	129	}
	130
	131
	132
af8d1d3f	133	static int do_zfs_ls(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
4d3c95f5 JL	134	{
4d3c95f5 JL	135	const char *filename = "/";
41204570	136	int part;
4101f687	137	struct blk_desc *dev_desc;
0528979f	138	struct disk_partition info;
4d3c95f5 JL	139	struct device_s vdev;
4d3c95f5 JL	140
41204570	141	if (argc < 2)
4d3c95f5 JL	142	return cmd_usage(cmdtp);
4d3c95f5 JL	143
4d3c95f5 JL	144	if (argc == 4)
	145	filename = argv[3];
	146
e35929e4	147	part = blk_get_device_part_str(argv[1], argv[2], &dev_desc, &info, 1);
41204570	148	if (part < 0)
4d3c95f5	149	return 1;
4d3c95f5	150
41204570 RH	151	zfs_set_blk_dev(dev_desc, &info);
41204570 RH	152	vdev.part_length = info.size;
4d3c95f5 JL	153
	154	zfs_ls(&vdev, filename,
	155	zfs_print);
	156
	157	return 0;
	158	}
	159
	160
	161	U_BOOT_CMD(zfsls, 4, 1, do_zfs_ls,
	162	"list files in a directory (default /)",
	163	"<interface> <dev[:part]> [directory]\n"
	164	" - list files from 'dev' on 'interface' in a '/DATASET/@/$dir/'");
	165
	166	U_BOOT_CMD(zfsload, 6, 0, do_zfs_load,
	167	"load binary file from a ZFS filesystem",
	168	"<interface> <dev[:part]> [addr] [filename] [bytes]\n"
	169	" - load binary file '/DATASET/@/$dir/$file' from 'dev' on 'interface'\n"
	170	" to address 'addr' from ZFS filesystem");