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