cmd/zfs.c

   1 // SPDX-License-Identifier: GPL-2.0+
   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.
   9  */
  10
  11 #include <common.h>
  12 #include <part.h>
  13 #include <config.h>
  14 #include <command.h>
  15 #include <image.h>
  16 #include <linux/ctype.h>
  17 #include <asm/byteorder.h>
  18 #include <zfs_common.h>
  19 #include <linux/stat.h>
  20 #include <malloc.h>
  21
  22 #if defined(CONFIG_CMD_USB) && defined(CONFIG_USB_STORAGE)
  23 #include <usb.h>
  24 #endif
  25
  26 #if !CONFIG_IS_ENABLED(DOS_PARTITION) && !CONFIG_IS_ENABLED(EFI_PARTITION)
  27 #error DOS or EFI partition support must be selected
  28 #endif
  29
  30 #define DOS_PART_MAGIC_OFFSET   0x1fe
  31 #define DOS_FS_TYPE_OFFSET      0x36
  32 #define DOS_FS32_TYPE_OFFSET    0x52
  33
  34 static int do_zfs_load(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  35 {
  36         char *filename = NULL;
  37         int dev;
  38         int part;
  39         ulong addr = 0;
  40         disk_partition_t info;
  41         struct blk_desc *dev_desc;
  42         char buf[12];
  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);
  53         filename = env_get("bootfile");
  54         switch (argc) {
  55         case 3:
  56                 addr_str = env_get("loadaddr");
  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
  82         part = blk_get_device_part_str(argv[1], argv[2], &dev_desc, &info, 1);
  83         if (part < 0)
  84                 return 1;
  85
  86         dev = dev_desc->devnum;
  87         printf("Loading file \"%s\" from %s device %d%c%c\n",
  88                 filename, argv[1], dev,
  89                 part ? ':' : ' ', part ? part + '0' : ' ');
  90
  91         zfs_set_blk_dev(dev_desc, &info);
  92         vdev.part_length = info.size;
  93
  94         memset(&zfile, 0, sizeof(zfile));
  95         zfile.device = &vdev;
  96         if (zfs_open(&zfile, filename)) {
  97                 printf("** File not found %s **\n", filename);
  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) {
 105                 printf("** Unable to read \"%s\" from %s %d:%d **\n",
 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 */
 114         load_addr = addr;
 115
 116         printf("%llu bytes read\n", zfile.size);
 117         env_set_hex("filesize", zfile.size);
 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
 133 static int do_zfs_ls(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 134 {
 135         const char *filename = "/";
 136         int part;
 137         struct blk_desc *dev_desc;
 138         disk_partition_t info;
 139         struct device_s vdev;
 140
 141         if (argc < 2)
 142                 return cmd_usage(cmdtp);
 143
 144         if (argc == 4)
 145                 filename = argv[3];
 146
 147         part = blk_get_device_part_str(argv[1], argv[2], &dev_desc, &info, 1);
 148         if (part < 0)
 149                 return 1;
 150
 151         zfs_set_blk_dev(dev_desc, &info);
 152         vdev.part_length = info.size;
 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");