zfs.c 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170
  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. #include <common.h>
  11. #include <part.h>
  12. #include <config.h>
  13. #include <command.h>
  14. #include <env.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. #if defined(CONFIG_CMD_USB) && defined(CONFIG_USB_STORAGE)
  22. #include <usb.h>
  23. #endif
  24. #if !CONFIG_IS_ENABLED(DOS_PARTITION) && !CONFIG_IS_ENABLED(EFI_PARTITION)
  25. #error DOS or EFI partition support must be selected
  26. #endif
  27. #define DOS_PART_MAGIC_OFFSET 0x1fe
  28. #define DOS_FS_TYPE_OFFSET 0x36
  29. #define DOS_FS32_TYPE_OFFSET 0x52
  30. static int do_zfs_load(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  31. {
  32. char *filename = NULL;
  33. int dev;
  34. int part;
  35. ulong addr = 0;
  36. struct disk_partition info;
  37. struct blk_desc *dev_desc;
  38. unsigned long count;
  39. const char *addr_str;
  40. struct zfs_file zfile;
  41. struct device_s vdev;
  42. if (argc < 3)
  43. return CMD_RET_USAGE;
  44. count = 0;
  45. addr = simple_strtoul(argv[3], NULL, 16);
  46. filename = env_get("bootfile");
  47. switch (argc) {
  48. case 3:
  49. addr_str = env_get("loadaddr");
  50. if (addr_str != NULL)
  51. addr = simple_strtoul(addr_str, NULL, 16);
  52. else
  53. addr = CONFIG_SYS_LOAD_ADDR;
  54. break;
  55. case 4:
  56. break;
  57. case 5:
  58. filename = argv[4];
  59. break;
  60. case 6:
  61. filename = argv[4];
  62. count = simple_strtoul(argv[5], NULL, 16);
  63. break;
  64. default:
  65. return cmd_usage(cmdtp);
  66. }
  67. if (!filename) {
  68. puts("** No boot file defined **\n");
  69. return 1;
  70. }
  71. part = blk_get_device_part_str(argv[1], argv[2], &dev_desc, &info, 1);
  72. if (part < 0)
  73. return 1;
  74. dev = dev_desc->devnum;
  75. printf("Loading file \"%s\" from %s device %d%c%c\n",
  76. filename, argv[1], dev,
  77. part ? ':' : ' ', part ? part + '0' : ' ');
  78. zfs_set_blk_dev(dev_desc, &info);
  79. vdev.part_length = info.size;
  80. memset(&zfile, 0, sizeof(zfile));
  81. zfile.device = &vdev;
  82. if (zfs_open(&zfile, filename)) {
  83. printf("** File not found %s **\n", filename);
  84. return 1;
  85. }
  86. if ((count < zfile.size) && (count != 0))
  87. zfile.size = (uint64_t)count;
  88. if (zfs_read(&zfile, (char *)addr, zfile.size) != zfile.size) {
  89. printf("** Unable to read \"%s\" from %s %d:%d **\n",
  90. filename, argv[1], dev, part);
  91. zfs_close(&zfile);
  92. return 1;
  93. }
  94. zfs_close(&zfile);
  95. /* Loading ok, update default load address */
  96. image_load_addr = addr;
  97. printf("%llu bytes read\n", zfile.size);
  98. env_set_hex("filesize", zfile.size);
  99. return 0;
  100. }
  101. int zfs_print(const char *entry, const struct zfs_dirhook_info *data)
  102. {
  103. printf("%s %s\n",
  104. data->dir ? "<DIR> " : " ",
  105. entry);
  106. return 0; /* 0 continue, 1 stop */
  107. }
  108. static int do_zfs_ls(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  109. {
  110. const char *filename = "/";
  111. int part;
  112. struct blk_desc *dev_desc;
  113. struct disk_partition info;
  114. struct device_s vdev;
  115. if (argc < 2)
  116. return cmd_usage(cmdtp);
  117. if (argc == 4)
  118. filename = argv[3];
  119. part = blk_get_device_part_str(argv[1], argv[2], &dev_desc, &info, 1);
  120. if (part < 0)
  121. return 1;
  122. zfs_set_blk_dev(dev_desc, &info);
  123. vdev.part_length = info.size;
  124. zfs_ls(&vdev, filename,
  125. zfs_print);
  126. return 0;
  127. }
  128. U_BOOT_CMD(zfsls, 4, 1, do_zfs_ls,
  129. "list files in a directory (default /)",
  130. "<interface> <dev[:part]> [directory]\n"
  131. " - list files from 'dev' on 'interface' in a '/DATASET/@/$dir/'");
  132. U_BOOT_CMD(zfsload, 6, 0, do_zfs_load,
  133. "load binary file from a ZFS filesystem",
  134. "<interface> <dev[:part]> [addr] [filename] [bytes]\n"
  135. " - load binary file '/DATASET/@/$dir/$file' from 'dev' on 'interface'\n"
  136. " to address 'addr' from ZFS filesystem");