Eric Lee / smarc-uboot

Blame view

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