1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 * Copyright (c) 2021 Aspeed Technology Inc.
4 */
5
6 #include "aspeed-hace.h"
7 #include <crypto/engine.h>
8 #include <linux/clk.h>
9 #include <linux/dma-mapping.h>
10 #include <linux/err.h>
11 #include <linux/interrupt.h>
12 #include <linux/io.h>
13 #include <linux/kernel.h>
14 #include <linux/module.h>
15 #include <linux/of.h>
16 #include <linux/platform_device.h>
17 #include <linux/property.h>
18
19 #ifdef CONFIG_CRYPTO_DEV_ASPEED_DEBUG
20 #define HACE_DBG(d, fmt, ...) \
21 dev_info((d)->dev, "%s() " fmt, __func__, ##__VA_ARGS__)
22 #else
23 #define HACE_DBG(d, fmt, ...) \
24 dev_dbg((d)->dev, "%s() " fmt, __func__, ##__VA_ARGS__)
25 #endif
26
27 /* HACE interrupt service routine */
aspeed_hace_irq(int irq,void * dev)28 static irqreturn_t aspeed_hace_irq(int irq, void *dev)
29 {
30 struct aspeed_hace_dev *hace_dev = (struct aspeed_hace_dev *)dev;
31 struct aspeed_engine_crypto *crypto_engine = &hace_dev->crypto_engine;
32 struct aspeed_engine_hash *hash_engine = &hace_dev->hash_engine;
33 u32 sts;
34
35 sts = ast_hace_read(hace_dev, ASPEED_HACE_STS);
36 ast_hace_write(hace_dev, sts, ASPEED_HACE_STS);
37
38 HACE_DBG(hace_dev, "irq status: 0x%x\n", sts);
39
40 if (sts & HACE_HASH_ISR) {
41 if (hash_engine->flags & CRYPTO_FLAGS_BUSY)
42 tasklet_schedule(&hash_engine->done_task);
43 else
44 dev_warn(hace_dev->dev, "HASH no active requests.\n");
45 }
46
47 if (sts & HACE_CRYPTO_ISR) {
48 if (crypto_engine->flags & CRYPTO_FLAGS_BUSY)
49 tasklet_schedule(&crypto_engine->done_task);
50 else
51 dev_warn(hace_dev->dev, "CRYPTO no active requests.\n");
52 }
53
54 return IRQ_HANDLED;
55 }
56
aspeed_hace_crypto_done_task(unsigned long data)57 static void aspeed_hace_crypto_done_task(unsigned long data)
58 {
59 struct aspeed_hace_dev *hace_dev = (struct aspeed_hace_dev *)data;
60 struct aspeed_engine_crypto *crypto_engine = &hace_dev->crypto_engine;
61
62 crypto_engine->resume(hace_dev);
63 }
64
aspeed_hace_hash_done_task(unsigned long data)65 static void aspeed_hace_hash_done_task(unsigned long data)
66 {
67 struct aspeed_hace_dev *hace_dev = (struct aspeed_hace_dev *)data;
68 struct aspeed_engine_hash *hash_engine = &hace_dev->hash_engine;
69
70 hash_engine->resume(hace_dev);
71 }
72
aspeed_hace_register(struct aspeed_hace_dev * hace_dev)73 static void aspeed_hace_register(struct aspeed_hace_dev *hace_dev)
74 {
75 #ifdef CONFIG_CRYPTO_DEV_ASPEED_HACE_HASH
76 aspeed_register_hace_hash_algs(hace_dev);
77 #endif
78 #ifdef CONFIG_CRYPTO_DEV_ASPEED_HACE_CRYPTO
79 aspeed_register_hace_crypto_algs(hace_dev);
80 #endif
81 }
82
aspeed_hace_unregister(struct aspeed_hace_dev * hace_dev)83 static void aspeed_hace_unregister(struct aspeed_hace_dev *hace_dev)
84 {
85 #ifdef CONFIG_CRYPTO_DEV_ASPEED_HACE_HASH
86 aspeed_unregister_hace_hash_algs(hace_dev);
87 #endif
88 #ifdef CONFIG_CRYPTO_DEV_ASPEED_HACE_CRYPTO
89 aspeed_unregister_hace_crypto_algs(hace_dev);
90 #endif
91 }
92
93 static const struct of_device_id aspeed_hace_of_matches[] = {
94 { .compatible = "aspeed,ast2500-hace", .data = (void *)5, },
95 { .compatible = "aspeed,ast2600-hace", .data = (void *)6, },
96 {},
97 };
98
aspeed_hace_probe(struct platform_device * pdev)99 static int aspeed_hace_probe(struct platform_device *pdev)
100 {
101 struct aspeed_engine_crypto *crypto_engine;
102 struct aspeed_engine_hash *hash_engine;
103 struct aspeed_hace_dev *hace_dev;
104 int rc;
105
106 hace_dev = devm_kzalloc(&pdev->dev, sizeof(struct aspeed_hace_dev),
107 GFP_KERNEL);
108 if (!hace_dev)
109 return -ENOMEM;
110
111 hace_dev->version = (uintptr_t)device_get_match_data(&pdev->dev);
112 if (!hace_dev->version) {
113 dev_err(&pdev->dev, "Failed to match hace dev id\n");
114 return -EINVAL;
115 }
116
117 hace_dev->dev = &pdev->dev;
118 hash_engine = &hace_dev->hash_engine;
119 crypto_engine = &hace_dev->crypto_engine;
120
121 platform_set_drvdata(pdev, hace_dev);
122
123 hace_dev->regs = devm_platform_get_and_ioremap_resource(pdev, 0, NULL);
124 if (IS_ERR(hace_dev->regs))
125 return PTR_ERR(hace_dev->regs);
126
127 /* Get irq number and register it */
128 hace_dev->irq = platform_get_irq(pdev, 0);
129 if (hace_dev->irq < 0)
130 return -ENXIO;
131
132 rc = devm_request_irq(&pdev->dev, hace_dev->irq, aspeed_hace_irq, 0,
133 dev_name(&pdev->dev), hace_dev);
134 if (rc) {
135 dev_err(&pdev->dev, "Failed to request interrupt\n");
136 return rc;
137 }
138
139 /* Get clk and enable it */
140 hace_dev->clk = devm_clk_get(&pdev->dev, NULL);
141 if (IS_ERR(hace_dev->clk)) {
142 dev_err(&pdev->dev, "Failed to get clk\n");
143 return -ENODEV;
144 }
145
146 rc = clk_prepare_enable(hace_dev->clk);
147 if (rc) {
148 dev_err(&pdev->dev, "Failed to enable clock 0x%x\n", rc);
149 return rc;
150 }
151
152 /* Initialize crypto hardware engine structure for hash */
153 hace_dev->crypt_engine_hash = crypto_engine_alloc_init(hace_dev->dev,
154 true);
155 if (!hace_dev->crypt_engine_hash) {
156 rc = -ENOMEM;
157 goto clk_exit;
158 }
159
160 rc = crypto_engine_start(hace_dev->crypt_engine_hash);
161 if (rc)
162 goto err_engine_hash_start;
163
164 tasklet_init(&hash_engine->done_task, aspeed_hace_hash_done_task,
165 (unsigned long)hace_dev);
166
167 /* Initialize crypto hardware engine structure for crypto */
168 hace_dev->crypt_engine_crypto = crypto_engine_alloc_init(hace_dev->dev,
169 true);
170 if (!hace_dev->crypt_engine_crypto) {
171 rc = -ENOMEM;
172 goto err_engine_hash_start;
173 }
174
175 rc = crypto_engine_start(hace_dev->crypt_engine_crypto);
176 if (rc)
177 goto err_engine_crypto_start;
178
179 tasklet_init(&crypto_engine->done_task, aspeed_hace_crypto_done_task,
180 (unsigned long)hace_dev);
181
182 /* Allocate DMA buffer for hash engine input used */
183 hash_engine->ahash_src_addr =
184 dmam_alloc_coherent(&pdev->dev,
185 ASPEED_HASH_SRC_DMA_BUF_LEN,
186 &hash_engine->ahash_src_dma_addr,
187 GFP_KERNEL);
188 if (!hash_engine->ahash_src_addr) {
189 dev_err(&pdev->dev, "Failed to allocate dma buffer\n");
190 rc = -ENOMEM;
191 goto err_engine_crypto_start;
192 }
193
194 /* Allocate DMA buffer for crypto engine context used */
195 crypto_engine->cipher_ctx =
196 dmam_alloc_coherent(&pdev->dev,
197 PAGE_SIZE,
198 &crypto_engine->cipher_ctx_dma,
199 GFP_KERNEL);
200 if (!crypto_engine->cipher_ctx) {
201 dev_err(&pdev->dev, "Failed to allocate cipher ctx dma\n");
202 rc = -ENOMEM;
203 goto err_engine_crypto_start;
204 }
205
206 /* Allocate DMA buffer for crypto engine input used */
207 crypto_engine->cipher_addr =
208 dmam_alloc_coherent(&pdev->dev,
209 ASPEED_CRYPTO_SRC_DMA_BUF_LEN,
210 &crypto_engine->cipher_dma_addr,
211 GFP_KERNEL);
212 if (!crypto_engine->cipher_addr) {
213 dev_err(&pdev->dev, "Failed to allocate cipher addr dma\n");
214 rc = -ENOMEM;
215 goto err_engine_crypto_start;
216 }
217
218 /* Allocate DMA buffer for crypto engine output used */
219 if (hace_dev->version == AST2600_VERSION) {
220 crypto_engine->dst_sg_addr =
221 dmam_alloc_coherent(&pdev->dev,
222 ASPEED_CRYPTO_DST_DMA_BUF_LEN,
223 &crypto_engine->dst_sg_dma_addr,
224 GFP_KERNEL);
225 if (!crypto_engine->dst_sg_addr) {
226 dev_err(&pdev->dev, "Failed to allocate dst_sg dma\n");
227 rc = -ENOMEM;
228 goto err_engine_crypto_start;
229 }
230 }
231
232 aspeed_hace_register(hace_dev);
233
234 dev_info(&pdev->dev, "Aspeed Crypto Accelerator successfully registered\n");
235
236 return 0;
237
238 err_engine_crypto_start:
239 crypto_engine_exit(hace_dev->crypt_engine_crypto);
240 err_engine_hash_start:
241 crypto_engine_exit(hace_dev->crypt_engine_hash);
242 clk_exit:
243 clk_disable_unprepare(hace_dev->clk);
244
245 return rc;
246 }
247
aspeed_hace_remove(struct platform_device * pdev)248 static void aspeed_hace_remove(struct platform_device *pdev)
249 {
250 struct aspeed_hace_dev *hace_dev = platform_get_drvdata(pdev);
251 struct aspeed_engine_crypto *crypto_engine = &hace_dev->crypto_engine;
252 struct aspeed_engine_hash *hash_engine = &hace_dev->hash_engine;
253
254 aspeed_hace_unregister(hace_dev);
255
256 crypto_engine_exit(hace_dev->crypt_engine_hash);
257 crypto_engine_exit(hace_dev->crypt_engine_crypto);
258
259 tasklet_kill(&hash_engine->done_task);
260 tasklet_kill(&crypto_engine->done_task);
261
262 clk_disable_unprepare(hace_dev->clk);
263 }
264
265 MODULE_DEVICE_TABLE(of, aspeed_hace_of_matches);
266
267 static struct platform_driver aspeed_hace_driver = {
268 .probe = aspeed_hace_probe,
269 .remove_new = aspeed_hace_remove,
270 .driver = {
271 .name = KBUILD_MODNAME,
272 .of_match_table = aspeed_hace_of_matches,
273 },
274 };
275
276 module_platform_driver(aspeed_hace_driver);
277
278 MODULE_AUTHOR("Neal Liu <neal_liu@aspeedtech.com>");
279 MODULE_DESCRIPTION("Aspeed HACE driver Crypto Accelerator");
280 MODULE_LICENSE("GPL");
281