drivers/crypto/ccp/sp-pci.c

   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 <[email protected]>
   8  * Author: Gary R Hook <[email protected]>
   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
  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
  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
  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
  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
 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
 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
 144 static struct sp_device *psp_get_master(void)
 145 {
 146         return sp_dev_master;
 147 }
 148
 149 static int sp_pci_probe(struct pci_dev *pdev, const struct pci_device_id *id)
 150 {
 151         struct sp_device *sp;
 152         struct sp_pci *sp_pci;
 153         struct device *dev = &pdev->dev;
 154         void __iomem * const *iomap_table;
 155         int bar_mask;
 156         int ret;
 157
 158         ret = -ENOMEM;
 159         sp = sp_alloc_struct(dev);
 160         if (!sp)
 161                 goto e_err;
 162
 163         sp_pci = devm_kzalloc(dev, sizeof(*sp_pci), GFP_KERNEL);
 164         if (!sp_pci)
 165                 goto e_err;
 166
 167         sp->dev_specific = sp_pci;
 168         sp->dev_vdata = (struct sp_dev_vdata *)id->driver_data;
 169         if (!sp->dev_vdata) {
 170                 ret = -ENODEV;
 171                 dev_err(dev, "missing driver data\n");
 172                 goto e_err;
 173         }
 174
 175         ret = pcim_enable_device(pdev);
 176         if (ret) {
 177                 dev_err(dev, "pcim_enable_device failed (%d)\n", ret);
 178                 goto e_err;
 179         }
 180
 181         bar_mask = pci_select_bars(pdev, IORESOURCE_MEM);
 182         ret = pcim_iomap_regions(pdev, bar_mask, "ccp");
 183         if (ret) {
 184                 dev_err(dev, "pcim_iomap_regions failed (%d)\n", ret);
 185                 goto e_err;
 186         }
 187
 188         iomap_table = pcim_iomap_table(pdev);
 189         if (!iomap_table) {
 190                 dev_err(dev, "pcim_iomap_table failed\n");
 191                 ret = -ENOMEM;
 192                 goto e_err;
 193         }
 194
 195         sp->io_map = iomap_table[sp->dev_vdata->bar];
 196         if (!sp->io_map) {
 197                 dev_err(dev, "ioremap failed\n");
 198                 ret = -ENOMEM;
 199                 goto e_err;
 200         }
 201
 202         ret = sp_get_irqs(sp);
 203         if (ret)
 204                 goto e_err;
 205
 206         pci_set_master(pdev);
 207         sp->set_psp_master_device = psp_set_master;
 208         sp->get_psp_master_device = psp_get_master;
 209
 210         ret = dma_set_mask_and_coherent(dev, DMA_BIT_MASK(48));
 211         if (ret) {
 212                 ret = dma_set_mask_and_coherent(dev, DMA_BIT_MASK(32));
 213                 if (ret) {
 214                         dev_err(dev, "dma_set_mask_and_coherent failed (%d)\n",
 215                                 ret);
 216                         goto e_err;
 217                 }
 218         }
 219
 220         dev_set_drvdata(dev, sp);
 221
 222         ret = sp_init(sp);
 223         if (ret)
 224                 goto e_err;
 225
 226         return 0;
 227
 228 e_err:
 229         dev_notice(dev, "initialization failed\n");
 230         return ret;
 231 }
 232
 233 static void sp_pci_remove(struct pci_dev *pdev)
 234 {
 235         struct device *dev = &pdev->dev;
 236         struct sp_device *sp = dev_get_drvdata(dev);
 237
 238         if (!sp)
 239                 return;
 240
 241         sp_destroy(sp);
 242
 243         sp_free_irqs(sp);
 244 }
 245
 246 #ifdef CONFIG_PM
 247 static int sp_pci_suspend(struct pci_dev *pdev, pm_message_t state)
 248 {
 249         struct device *dev = &pdev->dev;
 250         struct sp_device *sp = dev_get_drvdata(dev);
 251
 252         return sp_suspend(sp, state);
 253 }
 254
 255 static int sp_pci_resume(struct pci_dev *pdev)
 256 {
 257         struct device *dev = &pdev->dev;
 258         struct sp_device *sp = dev_get_drvdata(dev);
 259
 260         return sp_resume(sp);
 261 }
 262 #endif
 263
 264 #ifdef CONFIG_CRYPTO_DEV_SP_PSP
 265 static const struct sev_vdata sevv1 = {
 266         .cmdresp_reg            = 0x10580,
 267         .cmdbuff_addr_lo_reg    = 0x105e0,
 268         .cmdbuff_addr_hi_reg    = 0x105e4,
 269 };
 270
 271 static const struct sev_vdata sevv2 = {
 272         .cmdresp_reg            = 0x10980,
 273         .cmdbuff_addr_lo_reg    = 0x109e0,
 274         .cmdbuff_addr_hi_reg    = 0x109e4,
 275 };
 276
 277 static const struct tee_vdata teev1 = {
 278         .cmdresp_reg            = 0x10544,
 279         .cmdbuff_addr_lo_reg    = 0x10548,
 280         .cmdbuff_addr_hi_reg    = 0x1054c,
 281         .ring_wptr_reg          = 0x10550,
 282         .ring_rptr_reg          = 0x10554,
 283 };
 284
 285 static const struct psp_vdata pspv1 = {
 286         .sev                    = &sevv1,
 287         .feature_reg            = 0x105fc,
 288         .inten_reg              = 0x10610,
 289         .intsts_reg             = 0x10614,
 290 };
 291
 292 static const struct psp_vdata pspv2 = {
 293         .sev                    = &sevv2,
 294         .feature_reg            = 0x109fc,
 295         .inten_reg              = 0x10690,
 296         .intsts_reg             = 0x10694,
 297 };
 298
 299 static const struct psp_vdata pspv3 = {
 300         .tee                    = &teev1,
 301         .feature_reg            = 0x109fc,
 302         .inten_reg              = 0x10690,
 303         .intsts_reg             = 0x10694,
 304 };
 305 #endif
 306
 307 static const struct sp_dev_vdata dev_vdata[] = {
 308         {       /* 0 */
 309                 .bar = 2,
 310 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
 311                 .ccp_vdata = &ccpv3,
 312 #endif
 313         },
 314         {       /* 1 */
 315                 .bar = 2,
 316 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
 317                 .ccp_vdata = &ccpv5a,
 318 #endif
 319 #ifdef CONFIG_CRYPTO_DEV_SP_PSP
 320                 .psp_vdata = &pspv1,
 321 #endif
 322         },
 323         {       /* 2 */
 324                 .bar = 2,
 325 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
 326                 .ccp_vdata = &ccpv5b,
 327 #endif
 328         },
 329         {       /* 3 */
 330                 .bar = 2,
 331 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
 332                 .ccp_vdata = &ccpv5a,
 333 #endif
 334 #ifdef CONFIG_CRYPTO_DEV_SP_PSP
 335                 .psp_vdata = &pspv2,
 336 #endif
 337         },
 338         {       /* 4 */
 339                 .bar = 2,
 340 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
 341                 .ccp_vdata = &ccpv5a,
 342 #endif
 343 #ifdef CONFIG_CRYPTO_DEV_SP_PSP
 344                 .psp_vdata = &pspv3,
 345 #endif
 346         },
 347 };
 348 static const struct pci_device_id sp_pci_table[] = {
 349         { PCI_VDEVICE(AMD, 0x1537), (kernel_ulong_t)&dev_vdata[0] },
 350         { PCI_VDEVICE(AMD, 0x1456), (kernel_ulong_t)&dev_vdata[1] },
 351         { PCI_VDEVICE(AMD, 0x1468), (kernel_ulong_t)&dev_vdata[2] },
 352         { PCI_VDEVICE(AMD, 0x1486), (kernel_ulong_t)&dev_vdata[3] },
 353         { PCI_VDEVICE(AMD, 0x15DF), (kernel_ulong_t)&dev_vdata[4] },
 354         /* Last entry must be zero */
 355         { 0, }
 356 };
 357 MODULE_DEVICE_TABLE(pci, sp_pci_table);
 358
 359 static struct pci_driver sp_pci_driver = {
 360         .name = "ccp",
 361         .id_table = sp_pci_table,
 362         .probe = sp_pci_probe,
 363         .remove = sp_pci_remove,
 364 #ifdef CONFIG_PM
 365         .suspend = sp_pci_suspend,
 366         .resume = sp_pci_resume,
 367 #endif
 368 };
 369
 370 int sp_pci_init(void)
 371 {
 372         return pci_register_driver(&sp_pci_driver);
 373 }
 374
 375 void sp_pci_exit(void)
 376 {
 377         pci_unregister_driver(&sp_pci_driver);
 378 }