md5sum.c 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183
  1. // SPDX-License-Identifier: GPL-2.0+
  2. /*
  3. * (C) Copyright 2011
  4. * Joe Hershberger, National Instruments, joe.hershberger@ni.com
  5. *
  6. * (C) Copyright 2000
  7. * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
  8. */
  9. #include <common.h>
  10. #include <command.h>
  11. #include <env.h>
  12. #include <mapmem.h>
  13. #include <u-boot/md5.h>
  14. #include <asm/io.h>
  15. /*
  16. * Store the resulting sum to an address or variable
  17. */
  18. static void store_result(const u8 *sum, const char *dest)
  19. {
  20. unsigned int i;
  21. if (*dest == '*') {
  22. u8 *ptr;
  23. ptr = (u8 *)simple_strtoul(dest + 1, NULL, 16);
  24. for (i = 0; i < 16; i++)
  25. *ptr++ = sum[i];
  26. } else {
  27. char str_output[33];
  28. char *str_ptr = str_output;
  29. for (i = 0; i < 16; i++) {
  30. sprintf(str_ptr, "%02x", sum[i]);
  31. str_ptr += 2;
  32. }
  33. env_set(dest, str_output);
  34. }
  35. }
  36. #ifdef CONFIG_MD5SUM_VERIFY
  37. static int parse_verify_sum(char *verify_str, u8 *vsum)
  38. {
  39. if (*verify_str == '*') {
  40. u8 *ptr;
  41. ptr = (u8 *)simple_strtoul(verify_str + 1, NULL, 16);
  42. memcpy(vsum, ptr, 16);
  43. } else {
  44. unsigned int i;
  45. char *vsum_str;
  46. if (strlen(verify_str) == 32)
  47. vsum_str = verify_str;
  48. else {
  49. vsum_str = env_get(verify_str);
  50. if (vsum_str == NULL || strlen(vsum_str) != 32)
  51. return 1;
  52. }
  53. for (i = 0; i < 16; i++) {
  54. char *nullp = vsum_str + (i + 1) * 2;
  55. char end = *nullp;
  56. *nullp = '\0';
  57. *(u8 *)(vsum + i) =
  58. simple_strtoul(vsum_str + (i * 2), NULL, 16);
  59. *nullp = end;
  60. }
  61. }
  62. return 0;
  63. }
  64. int do_md5sum(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  65. {
  66. ulong addr, len;
  67. unsigned int i;
  68. u8 output[16];
  69. u8 vsum[16];
  70. int verify = 0;
  71. int ac;
  72. char * const *av;
  73. void *buf;
  74. if (argc < 3)
  75. return CMD_RET_USAGE;
  76. av = argv + 1;
  77. ac = argc - 1;
  78. if (strcmp(*av, "-v") == 0) {
  79. verify = 1;
  80. av++;
  81. ac--;
  82. if (ac < 3)
  83. return CMD_RET_USAGE;
  84. }
  85. addr = simple_strtoul(*av++, NULL, 16);
  86. len = simple_strtoul(*av++, NULL, 16);
  87. buf = map_sysmem(addr, len);
  88. md5_wd(buf, len, output, CHUNKSZ_MD5);
  89. unmap_sysmem(buf);
  90. if (!verify) {
  91. printf("md5 for %08lx ... %08lx ==> ", addr, addr + len - 1);
  92. for (i = 0; i < 16; i++)
  93. printf("%02x", output[i]);
  94. printf("\n");
  95. if (ac > 2)
  96. store_result(output, *av);
  97. } else {
  98. char *verify_str = *av++;
  99. if (parse_verify_sum(verify_str, vsum)) {
  100. printf("ERROR: %s does not contain a valid md5 sum\n",
  101. verify_str);
  102. return 1;
  103. }
  104. if (memcmp(output, vsum, 16) != 0) {
  105. printf("md5 for %08lx ... %08lx ==> ", addr,
  106. addr + len - 1);
  107. for (i = 0; i < 16; i++)
  108. printf("%02x", output[i]);
  109. printf(" != ");
  110. for (i = 0; i < 16; i++)
  111. printf("%02x", vsum[i]);
  112. printf(" ** ERROR **\n");
  113. return 1;
  114. }
  115. }
  116. return 0;
  117. }
  118. #else
  119. static int do_md5sum(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  120. {
  121. unsigned long addr, len;
  122. unsigned int i;
  123. u8 output[16];
  124. void *buf;
  125. if (argc < 3)
  126. return CMD_RET_USAGE;
  127. addr = simple_strtoul(argv[1], NULL, 16);
  128. len = simple_strtoul(argv[2], NULL, 16);
  129. buf = map_sysmem(addr, len);
  130. md5_wd(buf, len, output, CHUNKSZ_MD5);
  131. unmap_sysmem(buf);
  132. printf("md5 for %08lx ... %08lx ==> ", addr, addr + len - 1);
  133. for (i = 0; i < 16; i++)
  134. printf("%02x", output[i]);
  135. printf("\n");
  136. if (argc > 3)
  137. store_result(output, argv[3]);
  138. return 0;
  139. }
  140. #endif
  141. #ifdef CONFIG_MD5SUM_VERIFY
  142. U_BOOT_CMD(
  143. md5sum, 5, 1, do_md5sum,
  144. "compute MD5 message digest",
  145. "address count [[*]sum]\n"
  146. " - compute MD5 message digest [save to sum]\n"
  147. "md5sum -v address count [*]sum\n"
  148. " - verify md5sum of memory area"
  149. );
  150. #else
  151. U_BOOT_CMD(
  152. md5sum, 4, 1, do_md5sum,
  153. "compute MD5 message digest",
  154. "address count [[*]sum]\n"
  155. " - compute MD5 message digest [save to sum]"
  156. );
  157. #endif