1*ddb7bfe2SJoseph Chen // SPDX-License-Identifier: GPL-2.0
2*ddb7bfe2SJoseph Chen /*
3*ddb7bfe2SJoseph Chen * Copyright (c) 2021 Rockchip Electronics Co., Ltd
4*ddb7bfe2SJoseph Chen */
5*ddb7bfe2SJoseph Chen
6*ddb7bfe2SJoseph Chen #include <common.h>
7*ddb7bfe2SJoseph Chen #include <crypto.h>
8*ddb7bfe2SJoseph Chen
dump_hash(const char * label,void * hash,u32 nbits,char * addr,int len)9*ddb7bfe2SJoseph Chen static void dump_hash(const char *label, void *hash,
10*ddb7bfe2SJoseph Chen u32 nbits, char *addr, int len)
11*ddb7bfe2SJoseph Chen {
12*ddb7bfe2SJoseph Chen char *buf;
13*ddb7bfe2SJoseph Chen int i;
14*ddb7bfe2SJoseph Chen
15*ddb7bfe2SJoseph Chen printf("%s for %08lx ... %08lx ==> ", label, (ulong)addr, (ulong)addr + len);
16*ddb7bfe2SJoseph Chen buf = hash;
17*ddb7bfe2SJoseph Chen for (i = 0; i < BITS2BYTE(nbits); i++)
18*ddb7bfe2SJoseph Chen printf("%02x", buf[i]);
19*ddb7bfe2SJoseph Chen
20*ddb7bfe2SJoseph Chen printf("\n\n");
21*ddb7bfe2SJoseph Chen }
22*ddb7bfe2SJoseph Chen
do_crypto_csum(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])23*ddb7bfe2SJoseph Chen static int do_crypto_csum(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
24*ddb7bfe2SJoseph Chen {
25*ddb7bfe2SJoseph Chen sha_context csha_ctx;
26*ddb7bfe2SJoseph Chen struct udevice *dev;
27*ddb7bfe2SJoseph Chen char *input_data;
28*ddb7bfe2SJoseph Chen int input_len;
29*ddb7bfe2SJoseph Chen u8 hash[256];
30*ddb7bfe2SJoseph Chen u32 cap;
31*ddb7bfe2SJoseph Chen
32*ddb7bfe2SJoseph Chen if (argc != 4)
33*ddb7bfe2SJoseph Chen return CMD_RET_USAGE;
34*ddb7bfe2SJoseph Chen
35*ddb7bfe2SJoseph Chen if (!strcmp(argv[1], "md5"))
36*ddb7bfe2SJoseph Chen cap = CRYPTO_MD5;
37*ddb7bfe2SJoseph Chen else if (!strcmp(argv[1], "sha1"))
38*ddb7bfe2SJoseph Chen cap = CRYPTO_SHA1;
39*ddb7bfe2SJoseph Chen else if (!strcmp(argv[1], "sha256"))
40*ddb7bfe2SJoseph Chen cap = CRYPTO_SHA256;
41*ddb7bfe2SJoseph Chen else if (!strcmp(argv[1], "sha512"))
42*ddb7bfe2SJoseph Chen cap = CRYPTO_SHA512;
43*ddb7bfe2SJoseph Chen else
44*ddb7bfe2SJoseph Chen return CMD_RET_USAGE;
45*ddb7bfe2SJoseph Chen
46*ddb7bfe2SJoseph Chen input_data = (char *)simple_strtoul(argv[2], NULL, 16);
47*ddb7bfe2SJoseph Chen input_len = simple_strtoul(argv[3], NULL, 16);
48*ddb7bfe2SJoseph Chen
49*ddb7bfe2SJoseph Chen dev = crypto_get_device(cap);
50*ddb7bfe2SJoseph Chen if (!dev) {
51*ddb7bfe2SJoseph Chen printf("Can't find crypto device for %s\n", argv[1]);
52*ddb7bfe2SJoseph Chen return -ENODEV;
53*ddb7bfe2SJoseph Chen }
54*ddb7bfe2SJoseph Chen
55*ddb7bfe2SJoseph Chen if (cap == CRYPTO_MD5) {
56*ddb7bfe2SJoseph Chen csha_ctx.algo = CRYPTO_MD5;
57*ddb7bfe2SJoseph Chen csha_ctx.length = input_len;
58*ddb7bfe2SJoseph Chen memset(hash, 0, sizeof(hash));
59*ddb7bfe2SJoseph Chen crypto_sha_csum(dev, &csha_ctx, input_data, input_len, hash);
60*ddb7bfe2SJoseph Chen dump_hash("md5", hash, crypto_algo_nbits(csha_ctx.algo),
61*ddb7bfe2SJoseph Chen input_data, input_len);
62*ddb7bfe2SJoseph Chen } else if (cap == CRYPTO_SHA1) {
63*ddb7bfe2SJoseph Chen csha_ctx.algo = CRYPTO_SHA1;
64*ddb7bfe2SJoseph Chen csha_ctx.length = input_len;
65*ddb7bfe2SJoseph Chen memset(hash, 0, sizeof(hash));
66*ddb7bfe2SJoseph Chen crypto_sha_csum(dev, &csha_ctx, input_data, input_len, hash);
67*ddb7bfe2SJoseph Chen dump_hash("sha1", hash, crypto_algo_nbits(csha_ctx.algo),
68*ddb7bfe2SJoseph Chen input_data, input_len);
69*ddb7bfe2SJoseph Chen } else if (cap == CRYPTO_SHA256) {
70*ddb7bfe2SJoseph Chen csha_ctx.algo = CRYPTO_SHA256;
71*ddb7bfe2SJoseph Chen csha_ctx.length = input_len;
72*ddb7bfe2SJoseph Chen memset(hash, 0, sizeof(hash));
73*ddb7bfe2SJoseph Chen crypto_sha_csum(dev, &csha_ctx, input_data, input_len, hash);
74*ddb7bfe2SJoseph Chen dump_hash("sha256", hash, crypto_algo_nbits(csha_ctx.algo),
75*ddb7bfe2SJoseph Chen input_data, input_len);
76*ddb7bfe2SJoseph Chen } else if (cap == CRYPTO_SHA512) {
77*ddb7bfe2SJoseph Chen csha_ctx.algo = CRYPTO_SHA512;
78*ddb7bfe2SJoseph Chen csha_ctx.length = input_len;
79*ddb7bfe2SJoseph Chen memset(hash, 0, sizeof(hash));
80*ddb7bfe2SJoseph Chen crypto_sha_csum(dev, &csha_ctx, input_data, input_len, hash);
81*ddb7bfe2SJoseph Chen dump_hash("sha512", hash, crypto_algo_nbits(csha_ctx.algo),
82*ddb7bfe2SJoseph Chen input_data, input_len);
83*ddb7bfe2SJoseph Chen }
84*ddb7bfe2SJoseph Chen
85*ddb7bfe2SJoseph Chen return 0;
86*ddb7bfe2SJoseph Chen }
87*ddb7bfe2SJoseph Chen
88*ddb7bfe2SJoseph Chen U_BOOT_CMD(
89*ddb7bfe2SJoseph Chen crypto_sum, 4, 1, do_crypto_csum,
90*ddb7bfe2SJoseph Chen "crypto checksum engine",
91*ddb7bfe2SJoseph Chen "[md5|sha1|sha256|sha512] [addr] [length]"
92*ddb7bfe2SJoseph Chen );
93*ddb7bfe2SJoseph Chen
94