ahab.c 7.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347
  1. // SPDX-License-Identifier: GPL-2.0+
  2. /*
  3. * Copyright 2018-2019 NXP
  4. */
  5. #include <common.h>
  6. #include <command.h>
  7. #include <errno.h>
  8. #include <asm/io.h>
  9. #include <asm/arch/sci/sci.h>
  10. #include <asm/mach-imx/sys_proto.h>
  11. #include <asm/arch-imx/cpu.h>
  12. #include <asm/arch/sys_proto.h>
  13. #include <asm/arch/image.h>
  14. #include <console.h>
  15. DECLARE_GLOBAL_DATA_PTR;
  16. #define SEC_SECURE_RAM_BASE (0x31800000UL)
  17. #define SEC_SECURE_RAM_END_BASE (SEC_SECURE_RAM_BASE + 0xFFFFUL)
  18. #define SECO_LOCAL_SEC_SEC_SECURE_RAM_BASE (0x60000000UL)
  19. #define SECO_PT 2U
  20. static inline bool check_in_dram(ulong addr)
  21. {
  22. int i;
  23. bd_t *bd = gd->bd;
  24. for (i = 0; i < CONFIG_NR_DRAM_BANKS; ++i) {
  25. if (bd->bi_dram[i].size) {
  26. if (addr >= bd->bi_dram[i].start &&
  27. addr < (bd->bi_dram[i].start + bd->bi_dram[i].size))
  28. return true;
  29. }
  30. }
  31. return false;
  32. }
  33. int authenticate_os_container(ulong addr)
  34. {
  35. struct container_hdr *phdr;
  36. int i, ret = 0;
  37. int err;
  38. sc_rm_mr_t mr;
  39. sc_faddr_t start, end;
  40. u16 length;
  41. struct boot_img_t *img;
  42. unsigned long s, e;
  43. if (addr % 4) {
  44. puts("Error: Image's address is not 4 byte aligned\n");
  45. return -EINVAL;
  46. }
  47. if (!check_in_dram(addr)) {
  48. puts("Error: Image's address is invalid\n");
  49. return -EINVAL;
  50. }
  51. phdr = (struct container_hdr *)addr;
  52. if (phdr->tag != 0x87 && phdr->version != 0x0) {
  53. printf("Error: Wrong container header\n");
  54. return -EFAULT;
  55. }
  56. if (!phdr->num_images) {
  57. printf("Error: Wrong container, no image found\n");
  58. return -EFAULT;
  59. }
  60. length = phdr->length_lsb + (phdr->length_msb << 8);
  61. debug("container length %u\n", length);
  62. memcpy((void *)SEC_SECURE_RAM_BASE, (const void *)addr,
  63. ALIGN(length, CONFIG_SYS_CACHELINE_SIZE));
  64. err = sc_seco_authenticate(-1, SC_SECO_AUTH_CONTAINER,
  65. SECO_LOCAL_SEC_SEC_SECURE_RAM_BASE);
  66. if (err) {
  67. printf("Authenticate container hdr failed, return %d\n",
  68. err);
  69. ret = -EIO;
  70. goto exit;
  71. }
  72. /* Copy images to dest address */
  73. for (i = 0; i < phdr->num_images; i++) {
  74. img = (struct boot_img_t *)(addr +
  75. sizeof(struct container_hdr) +
  76. i * sizeof(struct boot_img_t));
  77. debug("img %d, dst 0x%x, src 0x%x, size 0x%x\n",
  78. i, (uint32_t) img->dst, img->offset + addr, img->size);
  79. memcpy((void *)img->dst, (const void *)(img->offset + addr),
  80. img->size);
  81. s = img->dst & ~(CONFIG_SYS_CACHELINE_SIZE - 1);
  82. e = ALIGN(img->dst + img->size, CONFIG_SYS_CACHELINE_SIZE) - 1;
  83. flush_dcache_range(s, e);
  84. /* Find the memreg and set permission for seco pt */
  85. err = sc_rm_find_memreg(-1, &mr, s, e);
  86. if (err) {
  87. printf("Error: can't find memreg for image load address 0x%x, error %d\n", img->dst, err);
  88. ret = -ENOMEM;
  89. goto exit;
  90. }
  91. err = sc_rm_get_memreg_info(-1, mr, &start, &end);
  92. if (!err)
  93. debug("memreg %u 0x%llx -- 0x%llx\n", mr, start, end);
  94. err = sc_rm_set_memreg_permissions(-1, mr, SECO_PT,
  95. SC_RM_PERM_FULL);
  96. if (err) {
  97. printf("Set permission failed for img %d, error %d\n",
  98. i, err);
  99. ret = -EPERM;
  100. goto exit;
  101. }
  102. err = sc_seco_authenticate(-1, SC_SECO_VERIFY_IMAGE,
  103. (1 << i));
  104. if (err) {
  105. printf("Authenticate img %d failed, return %d\n",
  106. i, err);
  107. ret = -EIO;
  108. }
  109. err = sc_rm_set_memreg_permissions(-1, mr, SECO_PT,
  110. SC_RM_PERM_NONE);
  111. if (err) {
  112. printf("Remove permission failed for img %d, err %d\n",
  113. i, err);
  114. ret = -EPERM;
  115. }
  116. if (ret)
  117. goto exit;
  118. }
  119. exit:
  120. if (sc_seco_authenticate(-1, SC_SECO_REL_CONTAINER, 0) != SC_ERR_NONE)
  121. printf("Error: release container failed!\n");
  122. return ret;
  123. }
  124. static int do_authenticate(struct cmd_tbl *cmdtp, int flag, int argc,
  125. char *const argv[])
  126. {
  127. ulong addr;
  128. if (argc < 2)
  129. return CMD_RET_USAGE;
  130. addr = simple_strtoul(argv[1], NULL, 16);
  131. printf("Authenticate OS container at 0x%lx\n", addr);
  132. if (authenticate_os_container(addr))
  133. return CMD_RET_FAILURE;
  134. return CMD_RET_SUCCESS;
  135. }
  136. static void display_life_cycle(u16 lc)
  137. {
  138. printf("Lifecycle: 0x%04X, ", lc);
  139. switch (lc) {
  140. case 0x1:
  141. printf("Pristine\n\n");
  142. break;
  143. case 0x2:
  144. printf("Fab\n\n");
  145. break;
  146. case 0x8:
  147. printf("Open\n\n");
  148. break;
  149. case 0x20:
  150. printf("NXP closed\n\n");
  151. break;
  152. case 0x80:
  153. printf("OEM closed\n\n");
  154. break;
  155. case 0x100:
  156. printf("Partial field return\n\n");
  157. break;
  158. case 0x200:
  159. printf("Full field return\n\n");
  160. break;
  161. case 0x400:
  162. printf("No return\n\n");
  163. break;
  164. default:
  165. printf("Unknown\n\n");
  166. break;
  167. }
  168. }
  169. #define AHAB_AUTH_CONTAINER_REQ 0x87
  170. #define AHAB_VERIFY_IMAGE_REQ 0x88
  171. #define AHAB_NO_AUTHENTICATION_IND 0xee
  172. #define AHAB_BAD_KEY_HASH_IND 0xfa
  173. #define AHAB_INVALID_KEY_IND 0xf9
  174. #define AHAB_BAD_SIGNATURE_IND 0xf0
  175. #define AHAB_BAD_HASH_IND 0xf1
  176. static void display_ahab_auth_event(u32 event)
  177. {
  178. u8 cmd = (event >> 16) & 0xff;
  179. u8 resp_ind = (event >> 8) & 0xff;
  180. switch (cmd) {
  181. case AHAB_AUTH_CONTAINER_REQ:
  182. printf("\tCMD = AHAB_AUTH_CONTAINER_REQ (0x%02X)\n", cmd);
  183. printf("\tIND = ");
  184. break;
  185. case AHAB_VERIFY_IMAGE_REQ:
  186. printf("\tCMD = AHAB_VERIFY_IMAGE_REQ (0x%02X)\n", cmd);
  187. printf("\tIND = ");
  188. break;
  189. default:
  190. return;
  191. }
  192. switch (resp_ind) {
  193. case AHAB_NO_AUTHENTICATION_IND:
  194. printf("AHAB_NO_AUTHENTICATION_IND (0x%02X)\n\n", resp_ind);
  195. break;
  196. case AHAB_BAD_KEY_HASH_IND:
  197. printf("AHAB_BAD_KEY_HASH_IND (0x%02X)\n\n", resp_ind);
  198. break;
  199. case AHAB_INVALID_KEY_IND:
  200. printf("AHAB_INVALID_KEY_IND (0x%02X)\n\n", resp_ind);
  201. break;
  202. case AHAB_BAD_SIGNATURE_IND:
  203. printf("AHAB_BAD_SIGNATURE_IND (0x%02X)\n\n", resp_ind);
  204. break;
  205. case AHAB_BAD_HASH_IND:
  206. printf("AHAB_BAD_HASH_IND (0x%02X)\n\n", resp_ind);
  207. break;
  208. default:
  209. printf("Unknown Indicator (0x%02X)\n\n", resp_ind);
  210. break;
  211. }
  212. }
  213. static int do_ahab_status(struct cmd_tbl *cmdtp, int flag, int argc,
  214. char *const argv[])
  215. {
  216. int err;
  217. u8 idx = 0U;
  218. u32 event;
  219. u16 lc;
  220. err = sc_seco_chip_info(-1, &lc, NULL, NULL, NULL);
  221. if (err != SC_ERR_NONE) {
  222. printf("Error in get lifecycle\n");
  223. return -EIO;
  224. }
  225. display_life_cycle(lc);
  226. err = sc_seco_get_event(-1, idx, &event);
  227. while (err == SC_ERR_NONE) {
  228. printf("SECO Event[%u] = 0x%08X\n", idx, event);
  229. display_ahab_auth_event(event);
  230. idx++;
  231. err = sc_seco_get_event(-1, idx, &event);
  232. }
  233. if (idx == 0)
  234. printf("No SECO Events Found!\n\n");
  235. return 0;
  236. }
  237. static int confirm_close(void)
  238. {
  239. puts("Warning: Please ensure your sample is in NXP closed state, "
  240. "OEM SRK hash has been fused, \n"
  241. " and you are able to boot a signed image successfully "
  242. "without any SECO events reported.\n"
  243. " If not, your sample will be unrecoverable.\n"
  244. "\nReally perform this operation? <y/N>\n");
  245. if (confirm_yesno())
  246. return 1;
  247. puts("Ahab close aborted\n");
  248. return 0;
  249. }
  250. static int do_ahab_close(struct cmd_tbl *cmdtp, int flag, int argc,
  251. char *const argv[])
  252. {
  253. int err;
  254. u16 lc;
  255. if (!confirm_close())
  256. return -EACCES;
  257. err = sc_seco_chip_info(-1, &lc, NULL, NULL, NULL);
  258. if (err != SC_ERR_NONE) {
  259. printf("Error in get lifecycle\n");
  260. return -EIO;
  261. }
  262. if (lc != 0x20) {
  263. puts("Current lifecycle is NOT NXP closed, can't move to OEM closed\n");
  264. display_life_cycle(lc);
  265. return -EPERM;
  266. }
  267. err = sc_seco_forward_lifecycle(-1, 16);
  268. if (err != SC_ERR_NONE) {
  269. printf("Error in forward lifecycle to OEM closed\n");
  270. return -EIO;
  271. }
  272. printf("Change to OEM closed successfully\n");
  273. return 0;
  274. }
  275. U_BOOT_CMD(auth_cntr, CONFIG_SYS_MAXARGS, 1, do_authenticate,
  276. "autenticate OS container via AHAB",
  277. "addr\n"
  278. "addr - OS container hex address\n"
  279. );
  280. U_BOOT_CMD(ahab_status, CONFIG_SYS_MAXARGS, 1, do_ahab_status,
  281. "display AHAB lifecycle and events from seco",
  282. ""
  283. );
  284. U_BOOT_CMD(ahab_close, CONFIG_SYS_MAXARGS, 1, do_ahab_close,
  285. "Change AHAB lifecycle to OEM closed",
  286. ""
  287. );