zfs.c 3.6 KB

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