1  /* SPDX-License-Identifier: GPL-2.0 */
2  /*
3   * sun8i-ce.h - hardware cryptographic offloader for
4   * Allwinner H3/A64/H5/H2+/H6 SoC
5   *
6   * Copyright (C) 2016-2019 Corentin LABBE <clabbe.montjoie@gmail.com>
7   */
8  #include <crypto/aes.h>
9  #include <crypto/des.h>
10  #include <crypto/engine.h>
11  #include <crypto/skcipher.h>
12  #include <linux/atomic.h>
13  #include <linux/debugfs.h>
14  #include <linux/crypto.h>
15  #include <linux/hw_random.h>
16  #include <crypto/internal/hash.h>
17  #include <crypto/md5.h>
18  #include <crypto/rng.h>
19  #include <crypto/sha1.h>
20  #include <crypto/sha2.h>
21  
22  /* CE Registers */
23  #define CE_TDQ	0x00
24  #define CE_CTR	0x04
25  #define CE_ICR	0x08
26  #define CE_ISR	0x0C
27  #define CE_TLR	0x10
28  #define CE_TSR	0x14
29  #define CE_ESR	0x18
30  #define CE_CSSGR	0x1C
31  #define CE_CDSGR	0x20
32  #define CE_CSAR	0x24
33  #define CE_CDAR	0x28
34  #define CE_TPR	0x2C
35  
36  /* Used in struct ce_task */
37  /* ce_task common */
38  #define CE_ENCRYPTION		0
39  #define CE_DECRYPTION		BIT(8)
40  
41  #define CE_COMM_INT		BIT(31)
42  
43  /* ce_task symmetric */
44  #define CE_AES_128BITS 0
45  #define CE_AES_192BITS 1
46  #define CE_AES_256BITS 2
47  
48  #define CE_OP_ECB	0
49  #define CE_OP_CBC	(1 << 8)
50  
51  #define CE_ALG_AES		0
52  #define CE_ALG_DES		1
53  #define CE_ALG_3DES		2
54  #define CE_ALG_MD5              16
55  #define CE_ALG_SHA1             17
56  #define CE_ALG_SHA224           18
57  #define CE_ALG_SHA256           19
58  #define CE_ALG_SHA384           20
59  #define CE_ALG_SHA512           21
60  #define CE_ALG_TRNG		48
61  #define CE_ALG_PRNG		49
62  #define CE_ALG_TRNG_V2		0x1c
63  #define CE_ALG_PRNG_V2		0x1d
64  
65  /* Used in ce_variant */
66  #define CE_ID_NOTSUPP		0xFF
67  
68  #define CE_ID_CIPHER_AES	0
69  #define CE_ID_CIPHER_DES	1
70  #define CE_ID_CIPHER_DES3	2
71  #define CE_ID_CIPHER_MAX	3
72  
73  #define CE_ID_HASH_MD5		0
74  #define CE_ID_HASH_SHA1		1
75  #define CE_ID_HASH_SHA224	2
76  #define CE_ID_HASH_SHA256	3
77  #define CE_ID_HASH_SHA384	4
78  #define CE_ID_HASH_SHA512	5
79  #define CE_ID_HASH_MAX		6
80  
81  #define CE_ID_OP_ECB	0
82  #define CE_ID_OP_CBC	1
83  #define CE_ID_OP_MAX	2
84  
85  /* Used in CE registers */
86  #define CE_ERR_ALGO_NOTSUP	BIT(0)
87  #define CE_ERR_DATALEN		BIT(1)
88  #define CE_ERR_KEYSRAM		BIT(2)
89  #define CE_ERR_ADDR_INVALID	BIT(5)
90  #define CE_ERR_KEYLADDER	BIT(6)
91  
92  #define ESR_H3	0
93  #define ESR_A64	1
94  #define ESR_R40	2
95  #define ESR_H5	3
96  #define ESR_H6	4
97  #define ESR_D1	5
98  
99  #define PRNG_DATA_SIZE (160 / 8)
100  #define PRNG_SEED_SIZE DIV_ROUND_UP(175, 8)
101  #define PRNG_LD BIT(17)
102  
103  #define CE_DIE_ID_SHIFT	16
104  #define CE_DIE_ID_MASK	0x07
105  
106  #define MAX_SG 8
107  
108  #define CE_MAX_CLOCKS 4
109  
110  #define MAXFLOW 4
111  
112  /*
113   * struct ce_clock - Describe clocks used by sun8i-ce
114   * @name:	Name of clock needed by this variant
115   * @freq:	Frequency to set for each clock
116   * @max_freq:	Maximum frequency for each clock (generally given by datasheet)
117   */
118  struct ce_clock {
119  	const char *name;
120  	unsigned long freq;
121  	unsigned long max_freq;
122  };
123  
124  /*
125   * struct ce_variant - Describe CE capability for each variant hardware
126   * @alg_cipher:	list of supported ciphers. for each CE_ID_ this will give the
127   *              coresponding CE_ALG_XXX value
128   * @alg_hash:	list of supported hashes. for each CE_ID_ this will give the
129   *              corresponding CE_ALG_XXX value
130   * @op_mode:	list of supported block modes
131   * @cipher_t_dlen_in_bytes:	Does the request size for cipher is in
132   *				bytes or words
133   * @hash_t_dlen_in_bytes:	Does the request size for hash is in
134   *				bits or words
135   * @prng_t_dlen_in_bytes:	Does the request size for PRNG is in
136   *				bytes or words
137   * @trng_t_dlen_in_bytes:	Does the request size for TRNG is in
138   *				bytes or words
139   * @ce_clks:	list of clocks needed by this variant
140   * @esr:	The type of error register
141   * @prng:	The CE_ALG_XXX value for the PRNG
142   * @trng:	The CE_ALG_XXX value for the TRNG
143   */
144  struct ce_variant {
145  	char alg_cipher[CE_ID_CIPHER_MAX];
146  	char alg_hash[CE_ID_HASH_MAX];
147  	u32 op_mode[CE_ID_OP_MAX];
148  	bool cipher_t_dlen_in_bytes;
149  	bool hash_t_dlen_in_bits;
150  	bool prng_t_dlen_in_bytes;
151  	bool trng_t_dlen_in_bytes;
152  	bool needs_word_addresses;
153  	struct ce_clock ce_clks[CE_MAX_CLOCKS];
154  	int esr;
155  	unsigned char prng;
156  	unsigned char trng;
157  };
158  
159  struct sginfo {
160  	__le32 addr;
161  	__le32 len;
162  } __packed;
163  
164  /*
165   * struct ce_task - CE Task descriptor
166   * The structure of this descriptor could be found in the datasheet
167   */
168  struct ce_task {
169  	__le32 t_id;
170  	__le32 t_common_ctl;
171  	__le32 t_sym_ctl;
172  	__le32 t_asym_ctl;
173  	__le32 t_key;
174  	__le32 t_iv;
175  	__le32 t_ctr;
176  	__le32 t_dlen;
177  	struct sginfo t_src[MAX_SG];
178  	struct sginfo t_dst[MAX_SG];
179  	__le32 next;
180  	__le32 reserved[3];
181  } __packed __aligned(8);
182  
183  /*
184   * struct sun8i_ce_flow - Information used by each flow
185   * @engine:	ptr to the crypto_engine for this flow
186   * @complete:	completion for the current task on this flow
187   * @status:	set to 1 by interrupt if task is done
188   * @t_phy:	Physical address of task
189   * @tl:		pointer to the current ce_task for this flow
190   * @backup_iv:		buffer which contain the next IV to store
191   * @bounce_iv:		buffer which contain the IV
192   * @stat_req:	number of request done by this flow
193   */
194  struct sun8i_ce_flow {
195  	struct crypto_engine *engine;
196  	struct completion complete;
197  	int status;
198  	dma_addr_t t_phy;
199  	int timeout;
200  	struct ce_task *tl;
201  	void *backup_iv;
202  	void *bounce_iv;
203  #ifdef CONFIG_CRYPTO_DEV_SUN8I_CE_DEBUG
204  	unsigned long stat_req;
205  #endif
206  };
207  
208  /*
209   * struct sun8i_ce_dev - main container for all this driver information
210   * @base:	base address of CE
211   * @ceclks:	clocks used by CE
212   * @reset:	pointer to reset controller
213   * @dev:	the platform device
214   * @mlock:	Control access to device registers
215   * @rnglock:	Control access to the RNG (dedicated channel 3)
216   * @chanlist:	array of all flow
217   * @flow:	flow to use in next request
218   * @variant:	pointer to variant specific data
219   * @dbgfs_dir:	Debugfs dentry for statistic directory
220   * @dbgfs_stats: Debugfs dentry for statistic counters
221   */
222  struct sun8i_ce_dev {
223  	void __iomem *base;
224  	struct clk *ceclks[CE_MAX_CLOCKS];
225  	struct reset_control *reset;
226  	struct device *dev;
227  	struct mutex mlock;
228  	struct mutex rnglock;
229  	struct sun8i_ce_flow *chanlist;
230  	atomic_t flow;
231  	const struct ce_variant *variant;
232  #ifdef CONFIG_CRYPTO_DEV_SUN8I_CE_DEBUG
233  	struct dentry *dbgfs_dir;
234  	struct dentry *dbgfs_stats;
235  #endif
236  #ifdef CONFIG_CRYPTO_DEV_SUN8I_CE_TRNG
237  	struct hwrng trng;
238  #ifdef CONFIG_CRYPTO_DEV_SUN8I_CE_DEBUG
239  	unsigned long hwrng_stat_req;
240  	unsigned long hwrng_stat_bytes;
241  #endif
242  #endif
243  };
244  
desc_addr_val(struct sun8i_ce_dev * dev,dma_addr_t addr)245  static inline u32 desc_addr_val(struct sun8i_ce_dev *dev, dma_addr_t addr)
246  {
247  	if (dev->variant->needs_word_addresses)
248  		return addr / 4;
249  
250  	return addr;
251  }
252  
desc_addr_val_le32(struct sun8i_ce_dev * dev,dma_addr_t addr)253  static inline __le32 desc_addr_val_le32(struct sun8i_ce_dev *dev,
254  					dma_addr_t addr)
255  {
256  	return cpu_to_le32(desc_addr_val(dev, addr));
257  }
258  
259  /*
260   * struct sun8i_cipher_req_ctx - context for a skcipher request
261   * @op_dir:		direction (encrypt vs decrypt) for this request
262   * @flow:		the flow to use for this request
263   * @ivlen:		size of bounce_iv
264   * @nr_sgs:		The number of source SG (as given by dma_map_sg())
265   * @nr_sgd:		The number of destination SG (as given by dma_map_sg())
266   * @addr_iv:		The IV addr returned by dma_map_single, need to unmap later
267   * @addr_key:		The key addr returned by dma_map_single, need to unmap later
268   * @fallback_req:	request struct for invoking the fallback skcipher TFM
269   */
270  struct sun8i_cipher_req_ctx {
271  	u32 op_dir;
272  	int flow;
273  	unsigned int ivlen;
274  	int nr_sgs;
275  	int nr_sgd;
276  	dma_addr_t addr_iv;
277  	dma_addr_t addr_key;
278  	struct skcipher_request fallback_req;   // keep at the end
279  };
280  
281  /*
282   * struct sun8i_cipher_tfm_ctx - context for a skcipher TFM
283   * @key:		pointer to key data
284   * @keylen:		len of the key
285   * @ce:			pointer to the private data of driver handling this TFM
286   * @fallback_tfm:	pointer to the fallback TFM
287   */
288  struct sun8i_cipher_tfm_ctx {
289  	u32 *key;
290  	u32 keylen;
291  	struct sun8i_ce_dev *ce;
292  	struct crypto_skcipher *fallback_tfm;
293  };
294  
295  /*
296   * struct sun8i_ce_hash_tfm_ctx - context for an ahash TFM
297   * @ce:			pointer to the private data of driver handling this TFM
298   * @fallback_tfm:	pointer to the fallback TFM
299   */
300  struct sun8i_ce_hash_tfm_ctx {
301  	struct sun8i_ce_dev *ce;
302  	struct crypto_ahash *fallback_tfm;
303  };
304  
305  /*
306   * struct sun8i_ce_hash_reqctx - context for an ahash request
307   * @fallback_req:	pre-allocated fallback request
308   * @flow:	the flow to use for this request
309   */
310  struct sun8i_ce_hash_reqctx {
311  	struct ahash_request fallback_req;
312  	int flow;
313  };
314  
315  /*
316   * struct sun8i_ce_prng_ctx - context for PRNG TFM
317   * @seed:	The seed to use
318   * @slen:	The size of the seed
319   */
320  struct sun8i_ce_rng_tfm_ctx {
321  	void *seed;
322  	unsigned int slen;
323  };
324  
325  /*
326   * struct sun8i_ce_alg_template - crypto_alg template
327   * @type:		the CRYPTO_ALG_TYPE for this template
328   * @ce_algo_id:		the CE_ID for this template
329   * @ce_blockmode:	the type of block operation CE_ID
330   * @ce:			pointer to the sun8i_ce_dev structure associated with
331   *			this template
332   * @alg:		one of sub struct must be used
333   * @stat_req:		number of request done on this template
334   * @stat_fb:		number of request which has fallbacked
335   * @stat_bytes:		total data size done by this template
336   */
337  struct sun8i_ce_alg_template {
338  	u32 type;
339  	u32 ce_algo_id;
340  	u32 ce_blockmode;
341  	struct sun8i_ce_dev *ce;
342  	union {
343  		struct skcipher_engine_alg skcipher;
344  		struct ahash_engine_alg hash;
345  		struct rng_alg rng;
346  	} alg;
347  	unsigned long stat_req;
348  	unsigned long stat_fb;
349  	unsigned long stat_bytes;
350  	unsigned long stat_fb_maxsg;
351  	unsigned long stat_fb_leniv;
352  	unsigned long stat_fb_len0;
353  	unsigned long stat_fb_mod16;
354  	unsigned long stat_fb_srcali;
355  	unsigned long stat_fb_srclen;
356  	unsigned long stat_fb_dstali;
357  	unsigned long stat_fb_dstlen;
358  	char fbname[CRYPTO_MAX_ALG_NAME];
359  };
360  
361  int sun8i_ce_aes_setkey(struct crypto_skcipher *tfm, const u8 *key,
362  			unsigned int keylen);
363  int sun8i_ce_des3_setkey(struct crypto_skcipher *tfm, const u8 *key,
364  			 unsigned int keylen);
365  int sun8i_ce_cipher_init(struct crypto_tfm *tfm);
366  void sun8i_ce_cipher_exit(struct crypto_tfm *tfm);
367  int sun8i_ce_cipher_do_one(struct crypto_engine *engine, void *areq);
368  int sun8i_ce_skdecrypt(struct skcipher_request *areq);
369  int sun8i_ce_skencrypt(struct skcipher_request *areq);
370  
371  int sun8i_ce_get_engine_number(struct sun8i_ce_dev *ce);
372  
373  int sun8i_ce_run_task(struct sun8i_ce_dev *ce, int flow, const char *name);
374  
375  int sun8i_ce_hash_init_tfm(struct crypto_ahash *tfm);
376  void sun8i_ce_hash_exit_tfm(struct crypto_ahash *tfm);
377  int sun8i_ce_hash_init(struct ahash_request *areq);
378  int sun8i_ce_hash_export(struct ahash_request *areq, void *out);
379  int sun8i_ce_hash_import(struct ahash_request *areq, const void *in);
380  int sun8i_ce_hash_final(struct ahash_request *areq);
381  int sun8i_ce_hash_update(struct ahash_request *areq);
382  int sun8i_ce_hash_finup(struct ahash_request *areq);
383  int sun8i_ce_hash_digest(struct ahash_request *areq);
384  int sun8i_ce_hash_run(struct crypto_engine *engine, void *breq);
385  
386  int sun8i_ce_prng_generate(struct crypto_rng *tfm, const u8 *src,
387  			   unsigned int slen, u8 *dst, unsigned int dlen);
388  int sun8i_ce_prng_seed(struct crypto_rng *tfm, const u8 *seed, unsigned int slen);
389  void sun8i_ce_prng_exit(struct crypto_tfm *tfm);
390  int sun8i_ce_prng_init(struct crypto_tfm *tfm);
391  
392  int sun8i_ce_hwrng_register(struct sun8i_ce_dev *ce);
393  void sun8i_ce_hwrng_unregister(struct sun8i_ce_dev *ce);
394