projects / u-boot.git / blame
| Commit | Line | Data |
|---|---|---|
| 83d290c5 | 1 | // SPDX-License-Identifier: GPL-2.0+ |
| b401b73d MV |
2 | /* |
| 3 | * Copyright (C) 2014 Marek Vasut <[email protected]> | |
| 4 | * | |
| 5 | * Command for en/de-crypting block of memory with AES-128-CBC cipher. | |
| b401b73d MV |
6 | */ |
| 7 | ||
| 8 | #include <common.h> | |
| 9 | #include <command.h> | |
| b80c0b99 | 10 | #include <uboot_aes.h> |
| b401b73d MV |
11 | #include <malloc.h> |
| 12 | #include <asm/byteorder.h> | |
| 13 | #include <linux/compiler.h> | |
| ed7ea058 | 14 | #include <mapmem.h> |
| b401b73d | 15 | |
| b401b73d MV |
16 | /** |
| 17 | * do_aes() - Handle the "aes" command-line command | |
| 18 | * @cmdtp: Command data struct pointer | |
| 19 | * @flag: Command flag | |
| 20 | * @argc: Command-line argument count | |
| 21 | * @argv: Array of command-line arguments | |
| 22 | * | |
| 23 | * Returns zero on success, CMD_RET_USAGE in case of misuse and negative | |
| 24 | * on error. | |
| 25 | */ | |
| 26 | static int do_aes(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) | |
| 27 | { | |
| af09eba6 АМ |
28 | uint32_t key_addr, iv_addr, src_addr, dst_addr, len; |
| 29 | uint8_t *key_ptr, *iv_ptr, *src_ptr, *dst_ptr; | |
| b401b73d MV |
30 | uint8_t key_exp[AES_EXPAND_KEY_LENGTH]; |
| 31 | uint32_t aes_blocks; | |
| 32 | int enc; | |
| 33 | ||
| af09eba6 | 34 | if (argc != 7) |
| b401b73d MV |
35 | return CMD_RET_USAGE; |
| 36 | ||
| 37 | if (!strncmp(argv[1], "enc", 3)) | |
| 38 | enc = 1; | |
| 39 | else if (!strncmp(argv[1], "dec", 3)) | |
| 40 | enc = 0; | |
| 41 | else | |
| 42 | return CMD_RET_USAGE; | |
| 43 | ||
| 44 | key_addr = simple_strtoul(argv[2], NULL, 16); | |
| af09eba6 АМ |
45 | iv_addr = simple_strtoul(argv[3], NULL, 16); |
| 46 | src_addr = simple_strtoul(argv[4], NULL, 16); | |
| 47 | dst_addr = simple_strtoul(argv[5], NULL, 16); | |
| 48 | len = simple_strtoul(argv[6], NULL, 16); | |
| b401b73d | 49 | |
| ed7ea058 PR |
50 | key_ptr = (uint8_t *)map_sysmem(key_addr, 128 / 8); |
| 51 | iv_ptr = (uint8_t *)map_sysmem(iv_addr, 128 / 8); | |
| 52 | src_ptr = (uint8_t *)map_sysmem(src_addr, len); | |
| 53 | dst_ptr = (uint8_t *)map_sysmem(dst_addr, len); | |
| b401b73d MV |
54 | |
| 55 | /* First we expand the key. */ | |
| 56 | aes_expand_key(key_ptr, key_exp); | |
| 57 | ||
| 58 | /* Calculate the number of AES blocks to encrypt. */ | |
| 59 | aes_blocks = DIV_ROUND_UP(len, AES_KEY_LENGTH); | |
| 60 | ||
| 61 | if (enc) | |
| af09eba6 АМ |
62 | aes_cbc_encrypt_blocks(key_exp, iv_ptr, src_ptr, dst_ptr, |
| 63 | aes_blocks); | |
| b401b73d | 64 | else |
| af09eba6 АМ |
65 | aes_cbc_decrypt_blocks(key_exp, iv_ptr, src_ptr, dst_ptr, |
| 66 | aes_blocks); | |
| b401b73d | 67 | |
| ed7ea058 PR |
68 | unmap_sysmem(key_ptr); |
| 69 | unmap_sysmem(iv_ptr); | |
| 70 | unmap_sysmem(src_ptr); | |
| 71 | unmap_sysmem(dst_ptr); | |
| 72 | ||
| b401b73d MV |
73 | return 0; |
| 74 | } | |
| 75 | ||
| 76 | /***************************************************/ | |
| 77 | #ifdef CONFIG_SYS_LONGHELP | |
| 78 | static char aes_help_text[] = | |
| af09eba6 АМ |
79 | "enc key iv src dst len - Encrypt block of data $len bytes long\n" |
| 80 | " at address $src using a key at address\n" | |
| 81 | " $key with initialization vector at address\n" | |
| 82 | " $iv. Store the result at address $dst.\n" | |
| 83 | " The $len size must be multiple of 16 bytes.\n" | |
| 84 | " The $key and $iv must be 16 bytes long.\n" | |
| 85 | "aes dec key iv src dst len - Decrypt block of data $len bytes long\n" | |
| 86 | " at address $src using a key at address\n" | |
| 87 | " $key with initialization vector at address\n" | |
| 88 | " $iv. Store the result at address $dst.\n" | |
| 89 | " The $len size must be multiple of 16 bytes.\n" | |
| 90 | " The $key and $iv must be 16 bytes long."; | |
| b401b73d MV |
91 | #endif |
| 92 | ||
| 93 | U_BOOT_CMD( | |
| af09eba6 | 94 | aes, 7, 1, do_aes, |
| b401b73d MV |
95 | "AES 128 CBC encryption", |
| 96 | aes_help_text | |
| 97 | ); |