1
2
3
4
5
6
7
8
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
29struct sp_pci {
30 int msix_count;
31 struct msix_entry msix_entry[MSIX_VECTORS];
32};
33static struct sp_device *sp_dev_master;
34
35static 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
58static 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
74static 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
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
90 dev_notice(dev, "could not enable MSI (%d)\n", ret);
91
92 return ret;
93}
94
95static 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
110static 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
133static 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
144static struct sp_device *psp_get_master(void)
145{
146 return sp_dev_master;
147}
148
149static 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
157static 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
237free_irqs:
238 sp_free_irqs(sp);
239e_err:
240 dev_notice(dev, "initialization failed\n");
241 return ret;
242}
243
244static 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
255static 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
268static 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
275static 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
283static const struct sev_vdata sevv1 = {
284 .cmdresp_reg = 0x10580,
285 .cmdbuff_addr_lo_reg = 0x105e0,
286 .cmdbuff_addr_hi_reg = 0x105e4,
287};
288
289static const struct sev_vdata sevv2 = {
290 .cmdresp_reg = 0x10980,
291 .cmdbuff_addr_lo_reg = 0x109e0,
292 .cmdbuff_addr_hi_reg = 0x109e4,
293};
294
295static 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
303static const struct psp_vdata pspv1 = {
304 .sev = &sevv1,
305 .feature_reg = 0x105fc,
306 .inten_reg = 0x10610,
307 .intsts_reg = 0x10614,
308};
309
310static const struct psp_vdata pspv2 = {
311 .sev = &sevv2,
312 .feature_reg = 0x109fc,
313 .inten_reg = 0x10690,
314 .intsts_reg = 0x10694,
315};
316
317static const struct psp_vdata pspv3 = {
318 .tee = &teev1,
319 .feature_reg = 0x109fc,
320 .inten_reg = 0x10690,
321 .intsts_reg = 0x10694,
322};
323#endif
324
325static const struct sp_dev_vdata dev_vdata[] = {
326 {
327 .bar = 2,
328#ifdef CONFIG_CRYPTO_DEV_SP_CCP
329 .ccp_vdata = &ccpv3,
330#endif
331 },
332 {
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 {
342 .bar = 2,
343#ifdef CONFIG_CRYPTO_DEV_SP_CCP
344 .ccp_vdata = &ccpv5b,
345#endif
346 },
347 {
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 {
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 .bar = 2,
367#ifdef CONFIG_CRYPTO_DEV_SP_PSP
368 .psp_vdata = &pspv2,
369#endif
370 },
371};
372static const struct pci_device_id sp_pci_table[] = {
373 { PCI_VDEVICE(AMD, 0x1537), (kernel_ulong_t)&dev_vdata[0] },
374 { PCI_VDEVICE(AMD, 0x1456), (kernel_ulong_t)&dev_vdata[1] },
375 { PCI_VDEVICE(AMD, 0x1468), (kernel_ulong_t)&dev_vdata[2] },
376 { PCI_VDEVICE(AMD, 0x1486), (kernel_ulong_t)&dev_vdata[3] },
377 { PCI_VDEVICE(AMD, 0x15DF), (kernel_ulong_t)&dev_vdata[4] },
378 { PCI_VDEVICE(AMD, 0x1649), (kernel_ulong_t)&dev_vdata[4] },
379 { PCI_VDEVICE(AMD, 0x14CA), (kernel_ulong_t)&dev_vdata[5] },
380
381 { 0, }
382};
383MODULE_DEVICE_TABLE(pci, sp_pci_table);
384
385static SIMPLE_DEV_PM_OPS(sp_pci_pm_ops, sp_pci_suspend, sp_pci_resume);
386
387static struct pci_driver sp_pci_driver = {
388 .name = "ccp",
389 .id_table = sp_pci_table,
390 .probe = sp_pci_probe,
391 .remove = sp_pci_remove,
392 .shutdown = sp_pci_shutdown,
393 .driver.pm = &sp_pci_pm_ops,
394};
395
396int sp_pci_init(void)
397{
398 return pci_register_driver(&sp_pci_driver);
399}
400
401void sp_pci_exit(void)
402{
403 pci_unregister_driver(&sp_pci_driver);
404}
405
