1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3 * AMD Secure Processor device driver
4 *
5 * Copyright (C) 2013,2019 Advanced Micro Devices, Inc.
6 *
7 * Author: Tom Lendacky <thomas.lendacky@amd.com>
8 * Author: Gary R Hook <gary.hook@amd.com>
9 */
10
11 #include <linux/module.h>
12 #include <linux/kernel.h>
13 #include <linux/device.h>
14 #include <linux/pci.h>
15 #include <linux/pci_ids.h>
16 #include <linux/dma-mapping.h>
17 #include <linux/kthread.h>
18 #include <linux/sched.h>
19 #include <linux/interrupt.h>
20 #include <linux/spinlock.h>
21 #include <linux/delay.h>
22 #include <linux/ccp.h>
23
24 #include "ccp-dev.h"
25 #include "psp-dev.h"
26
27 #define MSIX_VECTORS 2
28
29 struct sp_pci {
30 int msix_count;
31 struct msix_entry msix_entry[MSIX_VECTORS];
32 };
33 static struct sp_device *sp_dev_master;
34
sp_get_msix_irqs(struct sp_device * sp)35 static int sp_get_msix_irqs(struct sp_device *sp)
36 {
37 struct sp_pci *sp_pci = sp->dev_specific;
38 struct device *dev = sp->dev;
39 struct pci_dev *pdev = to_pci_dev(dev);
40 int v, ret;
41
42 for (v = 0; v < ARRAY_SIZE(sp_pci->msix_entry); v++)
43 sp_pci->msix_entry[v].entry = v;
44
45 ret = pci_enable_msix_range(pdev, sp_pci->msix_entry, 1, v);
46 if (ret < 0)
47 return ret;
48
49 sp_pci->msix_count = ret;
50 sp->use_tasklet = true;
51
52 sp->psp_irq = sp_pci->msix_entry[0].vector;
53 sp->ccp_irq = (sp_pci->msix_count > 1) ? sp_pci->msix_entry[1].vector
54 : sp_pci->msix_entry[0].vector;
55 return 0;
56 }
57
sp_get_msi_irq(struct sp_device * sp)58 static int sp_get_msi_irq(struct sp_device *sp)
59 {
60 struct device *dev = sp->dev;
61 struct pci_dev *pdev = to_pci_dev(dev);
62 int ret;
63
64 ret = pci_enable_msi(pdev);
65 if (ret)
66 return ret;
67
68 sp->ccp_irq = pdev->irq;
69 sp->psp_irq = pdev->irq;
70
71 return 0;
72 }
73
sp_get_irqs(struct sp_device * sp)74 static int sp_get_irqs(struct sp_device *sp)
75 {
76 struct device *dev = sp->dev;
77 int ret;
78
79 ret = sp_get_msix_irqs(sp);
80 if (!ret)
81 return 0;
82
83 /* Couldn't get MSI-X vectors, try MSI */
84 dev_notice(dev, "could not enable MSI-X (%d), trying MSI\n", ret);
85 ret = sp_get_msi_irq(sp);
86 if (!ret)
87 return 0;
88
89 /* Couldn't get MSI interrupt */
90 dev_notice(dev, "could not enable MSI (%d)\n", ret);
91
92 return ret;
93 }
94
sp_free_irqs(struct sp_device * sp)95 static void sp_free_irqs(struct sp_device *sp)
96 {
97 struct sp_pci *sp_pci = sp->dev_specific;
98 struct device *dev = sp->dev;
99 struct pci_dev *pdev = to_pci_dev(dev);
100
101 if (sp_pci->msix_count)
102 pci_disable_msix(pdev);
103 else if (sp->psp_irq)
104 pci_disable_msi(pdev);
105
106 sp->ccp_irq = 0;
107 sp->psp_irq = 0;
108 }
109
sp_pci_is_master(struct sp_device * sp)110 static bool sp_pci_is_master(struct sp_device *sp)
111 {
112 struct device *dev_cur, *dev_new;
113 struct pci_dev *pdev_cur, *pdev_new;
114
115 dev_new = sp->dev;
116 dev_cur = sp_dev_master->dev;
117
118 pdev_new = to_pci_dev(dev_new);
119 pdev_cur = to_pci_dev(dev_cur);
120
121 if (pdev_new->bus->number < pdev_cur->bus->number)
122 return true;
123
124 if (PCI_SLOT(pdev_new->devfn) < PCI_SLOT(pdev_cur->devfn))
125 return true;
126
127 if (PCI_FUNC(pdev_new->devfn) < PCI_FUNC(pdev_cur->devfn))
128 return true;
129
130 return false;
131 }
132
psp_set_master(struct sp_device * sp)133 static void psp_set_master(struct sp_device *sp)
134 {
135 if (!sp_dev_master) {
136 sp_dev_master = sp;
137 return;
138 }
139
140 if (sp_pci_is_master(sp))
141 sp_dev_master = sp;
142 }
143
psp_get_master(void)144 static struct sp_device *psp_get_master(void)
145 {
146 return sp_dev_master;
147 }
148
psp_clear_master(struct sp_device * sp)149 static void psp_clear_master(struct sp_device *sp)
150 {
151 if (sp == sp_dev_master) {
152 sp_dev_master = NULL;
153 dev_dbg(sp->dev, "Cleared sp_dev_master\n");
154 }
155 }
156
sp_pci_probe(struct pci_dev * pdev,const struct pci_device_id * id)157 static int sp_pci_probe(struct pci_dev *pdev, const struct pci_device_id *id)
158 {
159 struct sp_device *sp;
160 struct sp_pci *sp_pci;
161 struct device *dev = &pdev->dev;
162 void __iomem * const *iomap_table;
163 int bar_mask;
164 int ret;
165
166 ret = -ENOMEM;
167 sp = sp_alloc_struct(dev);
168 if (!sp)
169 goto e_err;
170
171 sp_pci = devm_kzalloc(dev, sizeof(*sp_pci), GFP_KERNEL);
172 if (!sp_pci)
173 goto e_err;
174
175 sp->dev_specific = sp_pci;
176 sp->dev_vdata = (struct sp_dev_vdata *)id->driver_data;
177 if (!sp->dev_vdata) {
178 ret = -ENODEV;
179 dev_err(dev, "missing driver data\n");
180 goto e_err;
181 }
182
183 ret = pcim_enable_device(pdev);
184 if (ret) {
185 dev_err(dev, "pcim_enable_device failed (%d)\n", ret);
186 goto e_err;
187 }
188
189 bar_mask = pci_select_bars(pdev, IORESOURCE_MEM);
190 ret = pcim_iomap_regions(pdev, bar_mask, "ccp");
191 if (ret) {
192 dev_err(dev, "pcim_iomap_regions failed (%d)\n", ret);
193 goto e_err;
194 }
195
196 iomap_table = pcim_iomap_table(pdev);
197 if (!iomap_table) {
198 dev_err(dev, "pcim_iomap_table failed\n");
199 ret = -ENOMEM;
200 goto e_err;
201 }
202
203 sp->io_map = iomap_table[sp->dev_vdata->bar];
204 if (!sp->io_map) {
205 dev_err(dev, "ioremap failed\n");
206 ret = -ENOMEM;
207 goto e_err;
208 }
209
210 ret = sp_get_irqs(sp);
211 if (ret)
212 goto e_err;
213
214 pci_set_master(pdev);
215 sp->set_psp_master_device = psp_set_master;
216 sp->get_psp_master_device = psp_get_master;
217 sp->clear_psp_master_device = psp_clear_master;
218
219 ret = dma_set_mask_and_coherent(dev, DMA_BIT_MASK(48));
220 if (ret) {
221 ret = dma_set_mask_and_coherent(dev, DMA_BIT_MASK(32));
222 if (ret) {
223 dev_err(dev, "dma_set_mask_and_coherent failed (%d)\n",
224 ret);
225 goto free_irqs;
226 }
227 }
228
229 dev_set_drvdata(dev, sp);
230
231 ret = sp_init(sp);
232 if (ret)
233 goto free_irqs;
234
235 return 0;
236
237 free_irqs:
238 sp_free_irqs(sp);
239 e_err:
240 dev_notice(dev, "initialization failed\n");
241 return ret;
242 }
243
sp_pci_shutdown(struct pci_dev * pdev)244 static void sp_pci_shutdown(struct pci_dev *pdev)
245 {
246 struct device *dev = &pdev->dev;
247 struct sp_device *sp = dev_get_drvdata(dev);
248
249 if (!sp)
250 return;
251
252 sp_destroy(sp);
253 }
254
sp_pci_remove(struct pci_dev * pdev)255 static void sp_pci_remove(struct pci_dev *pdev)
256 {
257 struct device *dev = &pdev->dev;
258 struct sp_device *sp = dev_get_drvdata(dev);
259
260 if (!sp)
261 return;
262
263 sp_destroy(sp);
264
265 sp_free_irqs(sp);
266 }
267
sp_pci_suspend(struct device * dev)268 static int __maybe_unused sp_pci_suspend(struct device *dev)
269 {
270 struct sp_device *sp = dev_get_drvdata(dev);
271
272 return sp_suspend(sp);
273 }
274
sp_pci_resume(struct device * dev)275 static int __maybe_unused sp_pci_resume(struct device *dev)
276 {
277 struct sp_device *sp = dev_get_drvdata(dev);
278
279 return sp_resume(sp);
280 }
281
282 #ifdef CONFIG_CRYPTO_DEV_SP_PSP
283 static const struct sev_vdata sevv1 = {
284 .cmdresp_reg = 0x10580,
285 .cmdbuff_addr_lo_reg = 0x105e0,
286 .cmdbuff_addr_hi_reg = 0x105e4,
287 };
288
289 static const struct sev_vdata sevv2 = {
290 .cmdresp_reg = 0x10980,
291 .cmdbuff_addr_lo_reg = 0x109e0,
292 .cmdbuff_addr_hi_reg = 0x109e4,
293 };
294
295 static const struct tee_vdata teev1 = {
296 .cmdresp_reg = 0x10544,
297 .cmdbuff_addr_lo_reg = 0x10548,
298 .cmdbuff_addr_hi_reg = 0x1054c,
299 .ring_wptr_reg = 0x10550,
300 .ring_rptr_reg = 0x10554,
301 };
302
303 static const struct psp_vdata pspv1 = {
304 .sev = &sevv1,
305 .feature_reg = 0x105fc,
306 .inten_reg = 0x10610,
307 .intsts_reg = 0x10614,
308 };
309
310 static const struct psp_vdata pspv2 = {
311 .sev = &sevv2,
312 .feature_reg = 0x109fc,
313 .inten_reg = 0x10690,
314 .intsts_reg = 0x10694,
315 };
316
317 static const struct psp_vdata pspv3 = {
318 .tee = &teev1,
319 .feature_reg = 0x109fc,
320 .inten_reg = 0x10690,
321 .intsts_reg = 0x10694,
322 };
323 #endif
324
325 static const struct sp_dev_vdata dev_vdata[] = {
326 { /* 0 */
327 .bar = 2,
328 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
329 .ccp_vdata = &ccpv3,
330 #endif
331 },
332 { /* 1 */
333 .bar = 2,
334 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
335 .ccp_vdata = &ccpv5a,
336 #endif
337 #ifdef CONFIG_CRYPTO_DEV_SP_PSP
338 .psp_vdata = &pspv1,
339 #endif
340 },
341 { /* 2 */
342 .bar = 2,
343 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
344 .ccp_vdata = &ccpv5b,
345 #endif
346 },
347 { /* 3 */
348 .bar = 2,
349 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
350 .ccp_vdata = &ccpv5a,
351 #endif
352 #ifdef CONFIG_CRYPTO_DEV_SP_PSP
353 .psp_vdata = &pspv2,
354 #endif
355 },
356 { /* 4 */
357 .bar = 2,
358 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
359 .ccp_vdata = &ccpv5a,
360 #endif
361 #ifdef CONFIG_CRYPTO_DEV_SP_PSP
362 .psp_vdata = &pspv3,
363 #endif
364 },
365 };
366 static const struct pci_device_id sp_pci_table[] = {
367 { PCI_VDEVICE(AMD, 0x1537), (kernel_ulong_t)&dev_vdata[0] },
368 { PCI_VDEVICE(AMD, 0x1456), (kernel_ulong_t)&dev_vdata[1] },
369 { PCI_VDEVICE(AMD, 0x1468), (kernel_ulong_t)&dev_vdata[2] },
370 { PCI_VDEVICE(AMD, 0x1486), (kernel_ulong_t)&dev_vdata[3] },
371 { PCI_VDEVICE(AMD, 0x15DF), (kernel_ulong_t)&dev_vdata[4] },
372 /* Last entry must be zero */
373 { 0, }
374 };
375 MODULE_DEVICE_TABLE(pci, sp_pci_table);
376
377 static SIMPLE_DEV_PM_OPS(sp_pci_pm_ops, sp_pci_suspend, sp_pci_resume);
378
379 static struct pci_driver sp_pci_driver = {
380 .name = "ccp",
381 .id_table = sp_pci_table,
382 .probe = sp_pci_probe,
383 .remove = sp_pci_remove,
384 .shutdown = sp_pci_shutdown,
385 .driver.pm = &sp_pci_pm_ops,
386 };
387
sp_pci_init(void)388 int sp_pci_init(void)
389 {
390 return pci_register_driver(&sp_pci_driver);
391 }
392
sp_pci_exit(void)393 void sp_pci_exit(void)
394 {
395 pci_unregister_driver(&sp_pci_driver);
396 }
397