clone.c 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  1. // SPDX-License-Identifier: GPL-2.0+
  2. /*
  3. * Copyright (C) 2020 John Chau <john@harmon.hk>
  4. *
  5. */
  6. #include <common.h>
  7. #include <command.h>
  8. #include <malloc.h>
  9. #include <part.h>
  10. #include <blk.h>
  11. #include <vsprintf.h>
  12. #define BUFSIZE (1 * 1024 * 1024)
  13. static int do_clone(struct cmd_tbl *cmdtp, int flag, int argc, char * const argv[])
  14. {
  15. int srcdev, destdev;
  16. struct blk_desc *srcdesc, *destdesc;
  17. int srcbz, destbz, ret;
  18. char *unit, *buf;
  19. unsigned long wrcnt, rdcnt, requested, srcblk, destblk;
  20. unsigned long timer;
  21. const unsigned long buffersize = 1024 * 1024;
  22. if (argc < 6)
  23. return CMD_RET_USAGE;
  24. srcdev = blk_get_device_by_str(argv[1], argv[2], &srcdesc);
  25. destdev = blk_get_device_by_str(argv[3], argv[4], &destdesc);
  26. if (srcdev < 0) {
  27. printf("Unable to open source device\n");
  28. return 1;
  29. } else if (destdev < 0) {
  30. printf("Unable to open destination device\n");
  31. return 1;
  32. }
  33. requested = dectoul(argv[5], &unit);
  34. srcbz = srcdesc->blksz;
  35. destbz = destdesc->blksz;
  36. if ((srcbz * (buffersize / srcbz) != buffersize) ||
  37. (destbz * (buffersize / destbz) != buffersize)) {
  38. printf("failed: cannot match device block sizes\n");
  39. return 1;
  40. }
  41. if (requested == 0) {
  42. unsigned long a = srcdesc->lba * srcdesc->blksz;
  43. unsigned long b = destdesc->lba * destdesc->blksz;
  44. if (a > b)
  45. requested = a;
  46. else
  47. requested = b;
  48. } else {
  49. switch (unit[0]) {
  50. case 'g':
  51. case 'G':
  52. requested *= 1024 * 1024 * 1024;
  53. break;
  54. case 'm':
  55. case 'M':
  56. requested *= 1024 * 1024;
  57. break;
  58. case 'k':
  59. case 'K':
  60. requested *= 1024;
  61. break;
  62. }
  63. }
  64. printf("Copying %ld bytes from %s:%s to %s:%s\n",
  65. requested, argv[1], argv[2], argv[3], argv[4]);
  66. wrcnt = 0;
  67. rdcnt = 0;
  68. buf = (char *)malloc(BUFSIZE);
  69. srcblk = 0;
  70. destblk = 0;
  71. timer = get_timer(0);
  72. while (wrcnt < requested) {
  73. unsigned long toread = BUFSIZE / srcbz;
  74. unsigned long towrite = BUFSIZE / destbz;
  75. unsigned long offset = 0;
  76. read:
  77. ret = blk_dread(srcdesc, srcblk, toread, buf + offset);
  78. if (ret < 0) {
  79. printf("Src read error @blk %ld\n", srcblk);
  80. goto exit;
  81. }
  82. rdcnt += ret * srcbz;
  83. srcblk += ret;
  84. if (ret < toread) {
  85. toread -= ret;
  86. offset += ret * srcbz;
  87. goto read;
  88. }
  89. offset = 0;
  90. write:
  91. ret = blk_dwrite(destdesc, destblk, towrite, buf + offset);
  92. if (ret < 0) {
  93. printf("Dest write error @blk %ld\n", srcblk);
  94. goto exit;
  95. }
  96. wrcnt += ret * destbz;
  97. destblk += ret;
  98. if (ret < towrite) {
  99. towrite -= ret;
  100. offset += ret * destbz;
  101. goto write;
  102. }
  103. }
  104. exit:
  105. timer = get_timer(timer);
  106. timer = 1000 * timer / CONFIG_SYS_HZ;
  107. printf("%ld read\n", rdcnt);
  108. printf("%ld written\n", wrcnt);
  109. printf("%ldms, %ldkB/s\n", timer, wrcnt / timer);
  110. free(buf);
  111. return 0;
  112. }
  113. U_BOOT_CMD(
  114. clone, 6, 1, do_clone,
  115. "simple storage cloning",
  116. "<src interface> <src dev> <dest interface> <dest dev> <size[K/M/G]>\n"
  117. "clone storage from 'src dev' on 'src interface' to 'dest dev' on 'dest interface' with maximum 'size' bytes (or 0 for clone to end)"
  118. );