xref: /rk3399_rockchip-uboot/include/crypto.h (revision e7846385ca65a4d31edc00691799fb13d18ab12a)
1cc6ac5d6SJoseph Chen /* SPDX-License-Identifier:     GPL-2.0+ */
2cc6ac5d6SJoseph Chen /*
3cc6ac5d6SJoseph Chen  * (C) Copyright 2019 Rockchip Electronics Co., Ltd
4cc6ac5d6SJoseph Chen  */
5cc6ac5d6SJoseph Chen 
6cc6ac5d6SJoseph Chen #ifndef _CORE_CRYPTO_H_
7cc6ac5d6SJoseph Chen #define _CORE_CRYPTO_H_
8cc6ac5d6SJoseph Chen 
9cc6ac5d6SJoseph Chen /* Algorithms/capability of crypto, works together with crypto_algo_nbits() */
10cc6ac5d6SJoseph Chen #define CRYPTO_MD5		BIT(0)
11cc6ac5d6SJoseph Chen #define CRYPTO_SHA1		BIT(1)
12cc6ac5d6SJoseph Chen #define CRYPTO_SHA256		BIT(2)
13*e7846385SLin Jinhan #define CRYPTO_SHA512		BIT(3)
14*e7846385SLin Jinhan 
15*e7846385SLin Jinhan #define CRYPTO_RSA512		BIT(10)
16*e7846385SLin Jinhan #define CRYPTO_RSA1024		BIT(11)
17*e7846385SLin Jinhan #define CRYPTO_RSA2048		BIT(12)
18*e7846385SLin Jinhan #define CRYPTO_RSA3072		BIT(13)
19*e7846385SLin Jinhan #define CRYPTO_RSA4096		BIT(14)
20cc6ac5d6SJoseph Chen 
21cc6ac5d6SJoseph Chen #define BYTE2WORD(bytes)	((bytes) / 4)
22cc6ac5d6SJoseph Chen #define BITS2BYTE(nbits)	((nbits) / 8)
23cc6ac5d6SJoseph Chen #define BITS2WORD(nbits)	((nbits) / 32)
24cc6ac5d6SJoseph Chen 
25cc6ac5d6SJoseph Chen typedef struct {
26cc6ac5d6SJoseph Chen 	u32 algo;	/* Algorithm: CRYPTO_MD5/CRYPTO_SHA1/CRYPTO_RSA2048... */
27cc6ac5d6SJoseph Chen 	u32 length;	/* Data total length */
28cc6ac5d6SJoseph Chen 
29cc6ac5d6SJoseph Chen } sha_context;
30cc6ac5d6SJoseph Chen 
31cc6ac5d6SJoseph Chen typedef struct {
32cc6ac5d6SJoseph Chen 	u32 algo;	/* Algorithm: CRYPTO_MD5/CRYPTO_SHA1/CRYPTO_RSA2048... */
33cc6ac5d6SJoseph Chen 	u32 *n;		/* Public key factor N */
34cc6ac5d6SJoseph Chen 	u32 *e;		/* Public key factor E */
35cc6ac5d6SJoseph Chen 	u32 *c;		/* Optional, a accelerate factor for some crypto */
36cc6ac5d6SJoseph Chen } rsa_key;
37cc6ac5d6SJoseph Chen 
38cc6ac5d6SJoseph Chen struct dm_crypto_ops {
39cc6ac5d6SJoseph Chen 	/* Hardware algorithm capability */
40cc6ac5d6SJoseph Chen 	u32 (*capability)(struct udevice *dev);
41cc6ac5d6SJoseph Chen 
42cc6ac5d6SJoseph Chen 	/* SHA init/update/final */
43cc6ac5d6SJoseph Chen 	int (*sha_init)(struct udevice *dev, sha_context *ctx);
44cc6ac5d6SJoseph Chen 	int (*sha_update)(struct udevice *dev, u32 *input, u32 len);
45cc6ac5d6SJoseph Chen 	int (*sha_final)(struct udevice *dev, sha_context *ctx, u8 *output);
46cc6ac5d6SJoseph Chen 
47cc6ac5d6SJoseph Chen 	/* RSA verify */
48cc6ac5d6SJoseph Chen 	int (*rsa_verify)(struct udevice *dev, rsa_key *ctx,
49cc6ac5d6SJoseph Chen 			  u8 *sign, u8 *output);
50cc6ac5d6SJoseph Chen };
51cc6ac5d6SJoseph Chen 
52cc6ac5d6SJoseph Chen /**
53cc6ac5d6SJoseph Chen  * crypto_algo_nbits() - Get algorithm bits accroding to algorithm
54cc6ac5d6SJoseph Chen  * @capability: expected algorithm capability, eg. CRYPTO_MD5/RSA2048...
55cc6ac5d6SJoseph Chen  *
56cc6ac5d6SJoseph Chen  * @return algorithm bits
57cc6ac5d6SJoseph Chen  */
58cc6ac5d6SJoseph Chen u32 crypto_algo_nbits(u32 algo);
59cc6ac5d6SJoseph Chen 
60cc6ac5d6SJoseph Chen /**
61cc6ac5d6SJoseph Chen  * crypto_get_device() - Get crypto device by capability
62cc6ac5d6SJoseph Chen  * @capability: expected algorithm capability, eg. CRYPTO_MD5/RSA2048...
63cc6ac5d6SJoseph Chen  *
64cc6ac5d6SJoseph Chen  * @return dev on success, otherwise NULL
65cc6ac5d6SJoseph Chen  */
66cc6ac5d6SJoseph Chen struct udevice *crypto_get_device(u32 capability);
67cc6ac5d6SJoseph Chen 
68cc6ac5d6SJoseph Chen /**
69cc6ac5d6SJoseph Chen  * crypto_sha_init() - Crypto sha init
70cc6ac5d6SJoseph Chen  *
71cc6ac5d6SJoseph Chen  * @dev: crypto device
72cc6ac5d6SJoseph Chen  * @ctx: sha context
73cc6ac5d6SJoseph Chen  *
74cc6ac5d6SJoseph Chen  * @return 0 on success, otherwise failed
75cc6ac5d6SJoseph Chen  */
76cc6ac5d6SJoseph Chen int crypto_sha_init(struct udevice *dev, sha_context *ctx);
77cc6ac5d6SJoseph Chen 
78cc6ac5d6SJoseph Chen /**
79cc6ac5d6SJoseph Chen  * crypto_sha_update() - Crypto sha update
80cc6ac5d6SJoseph Chen  *
81cc6ac5d6SJoseph Chen  * @dev: crypto device
82cc6ac5d6SJoseph Chen  * @input: input data buffer
83cc6ac5d6SJoseph Chen  * @len: input data length
84cc6ac5d6SJoseph Chen  *
85cc6ac5d6SJoseph Chen  * @return 0 on success, otherwise failed
86cc6ac5d6SJoseph Chen  */
87cc6ac5d6SJoseph Chen int crypto_sha_update(struct udevice *dev, u32 *input, u32 len);
88cc6ac5d6SJoseph Chen 
89cc6ac5d6SJoseph Chen /**
90cc6ac5d6SJoseph Chen  * crypto_sha_final() - Crypto sha finish and get result
91cc6ac5d6SJoseph Chen  *
92cc6ac5d6SJoseph Chen  * @dev: crypto device
93cc6ac5d6SJoseph Chen  * @ctx: sha context
94cc6ac5d6SJoseph Chen  * @output: output hash data
95cc6ac5d6SJoseph Chen  *
96cc6ac5d6SJoseph Chen  * @return 0 on success, otherwise failed
97cc6ac5d6SJoseph Chen  */
98cc6ac5d6SJoseph Chen int crypto_sha_final(struct udevice *dev, sha_context *ctx, u8 *output);
99cc6ac5d6SJoseph Chen 
100cc6ac5d6SJoseph Chen /**
101cc6ac5d6SJoseph Chen  * crypto_sha_csum() - Crypto sha hash for one data block only
102cc6ac5d6SJoseph Chen  *
103cc6ac5d6SJoseph Chen  * @dev: crypto device
104cc6ac5d6SJoseph Chen  * @ctx: sha context
105cc6ac5d6SJoseph Chen  * @input: input data buffer
106cc6ac5d6SJoseph Chen  * @input_len: input data length
107cc6ac5d6SJoseph Chen  * @output: output hash data
108cc6ac5d6SJoseph Chen  *
109cc6ac5d6SJoseph Chen  * @return 0 on success, otherwise failed
110cc6ac5d6SJoseph Chen  */
111cc6ac5d6SJoseph Chen int crypto_sha_csum(struct udevice *dev, sha_context *ctx,
112cc6ac5d6SJoseph Chen 		    char *input, u32 input_len, u8 *output);
113cc6ac5d6SJoseph Chen 
114cc6ac5d6SJoseph Chen /**
115cc6ac5d6SJoseph Chen  * crypto_rsa_verify() - Crypto rsa verify
116cc6ac5d6SJoseph Chen  *
117cc6ac5d6SJoseph Chen  * @dev: crypto device
118cc6ac5d6SJoseph Chen  * @ctx: rsa key context
119cc6ac5d6SJoseph Chen  * @sign: signature
120cc6ac5d6SJoseph Chen  * @output: output hash data buffer
121cc6ac5d6SJoseph Chen  *
122cc6ac5d6SJoseph Chen  * @return 0 on success, otherwise failed
123cc6ac5d6SJoseph Chen  */
124cc6ac5d6SJoseph Chen int crypto_rsa_verify(struct udevice *dev, rsa_key *ctx, u8 *sign, u8 *output);
125cc6ac5d6SJoseph Chen 
126cc6ac5d6SJoseph Chen #endif
127