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 <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
  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 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
 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 e_err;
 226                 }
 227         }
 228
 229         dev_set_drvdata(dev, sp);
 230
 231         ret = sp_init(sp);
 232         if (ret)
 233                 goto e_err;
 234
 235         return 0;
 236
 237 e_err:
 238         dev_notice(dev, "initialization failed\n");
 239         return ret;
 240 }
 241
 242 static void sp_pci_remove(struct pci_dev *pdev)
 243 {
 244         struct device *dev = &pdev->dev;
 245         struct sp_device *sp = dev_get_drvdata(dev);
 246
 247         if (!sp)
 248                 return;
 249
 250         sp_destroy(sp);
 251
 252         sp_free_irqs(sp);
 253 }
 254
 255 #ifdef CONFIG_PM
 256 static int sp_pci_suspend(struct pci_dev *pdev, pm_message_t state)
 257 {
 258         struct device *dev = &pdev->dev;
 259         struct sp_device *sp = dev_get_drvdata(dev);
 260
 261         return sp_suspend(sp, state);
 262 }
 263
 264 static int sp_pci_resume(struct pci_dev *pdev)
 265 {
 266         struct device *dev = &pdev->dev;
 267         struct sp_device *sp = dev_get_drvdata(dev);
 268
 269         return sp_resume(sp);
 270 }
 271 #endif
 272
 273 #ifdef CONFIG_CRYPTO_DEV_SP_PSP
 274 static const struct sev_vdata sevv1 = {
 275         .cmdresp_reg            = 0x10580,
 276         .cmdbuff_addr_lo_reg    = 0x105e0,
 277         .cmdbuff_addr_hi_reg    = 0x105e4,
 278 };
 279
 280 static const struct sev_vdata sevv2 = {
 281         .cmdresp_reg            = 0x10980,
 282         .cmdbuff_addr_lo_reg    = 0x109e0,
 283         .cmdbuff_addr_hi_reg    = 0x109e4,
 284 };
 285
 286 static const struct tee_vdata teev1 = {
 287         .cmdresp_reg            = 0x10544,
 288         .cmdbuff_addr_lo_reg    = 0x10548,
 289         .cmdbuff_addr_hi_reg    = 0x1054c,
 290         .ring_wptr_reg          = 0x10550,
 291         .ring_rptr_reg          = 0x10554,
 292 };
 293
 294 static const struct psp_vdata pspv1 = {
 295         .sev                    = &sevv1,
 296         .feature_reg            = 0x105fc,
 297         .inten_reg              = 0x10610,
 298         .intsts_reg             = 0x10614,
 299 };
 300
 301 static const struct psp_vdata pspv2 = {
 302         .sev                    = &sevv2,
 303         .feature_reg            = 0x109fc,
 304         .inten_reg              = 0x10690,
 305         .intsts_reg             = 0x10694,
 306 };
 307
 308 static const struct psp_vdata pspv3 = {
 309         .tee                    = &teev1,
 310         .feature_reg            = 0x109fc,
 311         .inten_reg              = 0x10690,
 312         .intsts_reg             = 0x10694,
 313 };
 314 #endif
 315
 316 static const struct sp_dev_vdata dev_vdata[] = {
 317         {       /* 0 */
 318                 .bar = 2,
 319 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
 320                 .ccp_vdata = &ccpv3,
 321 #endif
 322         },
 323         {       /* 1 */
 324                 .bar = 2,
 325 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
 326                 .ccp_vdata = &ccpv5a,
 327 #endif
 328 #ifdef CONFIG_CRYPTO_DEV_SP_PSP
 329                 .psp_vdata = &pspv1,
 330 #endif
 331         },
 332         {       /* 2 */
 333                 .bar = 2,
 334 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
 335                 .ccp_vdata = &ccpv5b,
 336 #endif
 337         },
 338         {       /* 3 */
 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 = &pspv2,
 345 #endif
 346         },
 347         {       /* 4 */
 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 = &pspv3,
 354 #endif
 355         },
 356 };
 357 static const struct pci_device_id sp_pci_table[] = {
 358         { PCI_VDEVICE(AMD, 0x1537), (kernel_ulong_t)&dev_vdata[0] },
 359         { PCI_VDEVICE(AMD, 0x1456), (kernel_ulong_t)&dev_vdata[1] },
 360         { PCI_VDEVICE(AMD, 0x1468), (kernel_ulong_t)&dev_vdata[2] },
 361         { PCI_VDEVICE(AMD, 0x1486), (kernel_ulong_t)&dev_vdata[3] },
 362         { PCI_VDEVICE(AMD, 0x15DF), (kernel_ulong_t)&dev_vdata[4] },
 363         /* Last entry must be zero */
 364         { 0, }
 365 };
 366 MODULE_DEVICE_TABLE(pci, sp_pci_table);
 367
 368 static struct pci_driver sp_pci_driver = {
 369         .name = "ccp",
 370         .id_table = sp_pci_table,
 371         .probe = sp_pci_probe,
 372         .remove = sp_pci_remove,
 373 #ifdef CONFIG_PM
 374         .suspend = sp_pci_suspend,
 375         .resume = sp_pci_resume,
 376 #endif
 377 };
 378
 379 int sp_pci_init(void)
 380 {
 381         return pci_register_driver(&sp_pci_driver);
 382 }
 383
 384 void sp_pci_exit(void)
 385 {
 386         pci_unregister_driver(&sp_pci_driver);
 387 }