drivers/crypto/ccp/sp-pci.c

   1 /*
   2  * AMD Secure Processor device driver
   3  *
   4  * Copyright (C) 2013,2016 Advanced Micro Devices, Inc.
   5  *
   6  * Author: Tom Lendacky <[email protected]>
   7  * Author: Gary R Hook <[email protected]>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License version 2 as
  11  * published by the Free Software Foundation.
  12  */
  13
  14 #include <linux/module.h>
  15 #include <linux/kernel.h>
  16 #include <linux/device.h>
  17 #include <linux/pci.h>
  18 #include <linux/pci_ids.h>
  19 #include <linux/dma-mapping.h>
  20 #include <linux/kthread.h>
  21 #include <linux/sched.h>
  22 #include <linux/interrupt.h>
  23 #include <linux/spinlock.h>
  24 #include <linux/delay.h>
  25 #include <linux/ccp.h>
  26
  27 #include "ccp-dev.h"
  28
  29 #define MSIX_VECTORS                    2
  30
  31 struct sp_pci {
  32         int msix_count;
  33         struct msix_entry msix_entry[MSIX_VECTORS];
  34 };
  35
  36 static int sp_get_msix_irqs(struct sp_device *sp)
  37 {
  38         struct sp_pci *sp_pci = sp->dev_specific;
  39         struct device *dev = sp->dev;
  40         struct pci_dev *pdev = to_pci_dev(dev);
  41         int v, ret;
  42
  43         for (v = 0; v < ARRAY_SIZE(sp_pci->msix_entry); v++)
  44                 sp_pci->msix_entry[v].entry = v;
  45
  46         ret = pci_enable_msix_range(pdev, sp_pci->msix_entry, 1, v);
  47         if (ret < 0)
  48                 return ret;
  49
  50         sp_pci->msix_count = ret;
  51         sp->use_tasklet = true;
  52
  53         sp->psp_irq = sp_pci->msix_entry[0].vector;
  54         sp->ccp_irq = (sp_pci->msix_count > 1) ? sp_pci->msix_entry[1].vector
  55                                                : sp_pci->msix_entry[0].vector;
  56         return 0;
  57 }
  58
  59 static int sp_get_msi_irq(struct sp_device *sp)
  60 {
  61         struct device *dev = sp->dev;
  62         struct pci_dev *pdev = to_pci_dev(dev);
  63         int ret;
  64
  65         ret = pci_enable_msi(pdev);
  66         if (ret)
  67                 return ret;
  68
  69         sp->ccp_irq = pdev->irq;
  70         sp->psp_irq = pdev->irq;
  71
  72         return 0;
  73 }
  74
  75 static int sp_get_irqs(struct sp_device *sp)
  76 {
  77         struct device *dev = sp->dev;
  78         int ret;
  79
  80         ret = sp_get_msix_irqs(sp);
  81         if (!ret)
  82                 return 0;
  83
  84         /* Couldn't get MSI-X vectors, try MSI */
  85         dev_notice(dev, "could not enable MSI-X (%d), trying MSI\n", ret);
  86         ret = sp_get_msi_irq(sp);
  87         if (!ret)
  88                 return 0;
  89
  90         /* Couldn't get MSI interrupt */
  91         dev_notice(dev, "could not enable MSI (%d)\n", ret);
  92
  93         return ret;
  94 }
  95
  96 static void sp_free_irqs(struct sp_device *sp)
  97 {
  98         struct sp_pci *sp_pci = sp->dev_specific;
  99         struct device *dev = sp->dev;
 100         struct pci_dev *pdev = to_pci_dev(dev);
 101
 102         if (sp_pci->msix_count)
 103                 pci_disable_msix(pdev);
 104         else if (sp->psp_irq)
 105                 pci_disable_msi(pdev);
 106
 107         sp->ccp_irq = 0;
 108         sp->psp_irq = 0;
 109 }
 110
 111 static int sp_pci_probe(struct pci_dev *pdev, const struct pci_device_id *id)
 112 {
 113         struct sp_device *sp;
 114         struct sp_pci *sp_pci;
 115         struct device *dev = &pdev->dev;
 116         void __iomem * const *iomap_table;
 117         int bar_mask;
 118         int ret;
 119
 120         ret = -ENOMEM;
 121         sp = sp_alloc_struct(dev);
 122         if (!sp)
 123                 goto e_err;
 124
 125         sp_pci = devm_kzalloc(dev, sizeof(*sp_pci), GFP_KERNEL);
 126         if (!sp_pci)
 127                 goto e_err;
 128
 129         sp->dev_specific = sp_pci;
 130         sp->dev_vdata = (struct sp_dev_vdata *)id->driver_data;
 131         if (!sp->dev_vdata) {
 132                 ret = -ENODEV;
 133                 dev_err(dev, "missing driver data\n");
 134                 goto e_err;
 135         }
 136
 137         ret = pcim_enable_device(pdev);
 138         if (ret) {
 139                 dev_err(dev, "pcim_enable_device failed (%d)\n", ret);
 140                 goto e_err;
 141         }
 142
 143         bar_mask = pci_select_bars(pdev, IORESOURCE_MEM);
 144         ret = pcim_iomap_regions(pdev, bar_mask, "ccp");
 145         if (ret) {
 146                 dev_err(dev, "pcim_iomap_regions failed (%d)\n", ret);
 147                 goto e_err;
 148         }
 149
 150         iomap_table = pcim_iomap_table(pdev);
 151         if (!iomap_table) {
 152                 dev_err(dev, "pcim_iomap_table failed\n");
 153                 ret = -ENOMEM;
 154                 goto e_err;
 155         }
 156
 157         sp->io_map = iomap_table[sp->dev_vdata->bar];
 158         if (!sp->io_map) {
 159                 dev_err(dev, "ioremap failed\n");
 160                 ret = -ENOMEM;
 161                 goto e_err;
 162         }
 163
 164         ret = sp_get_irqs(sp);
 165         if (ret)
 166                 goto e_err;
 167
 168         pci_set_master(pdev);
 169
 170         ret = dma_set_mask_and_coherent(dev, DMA_BIT_MASK(48));
 171         if (ret) {
 172                 ret = dma_set_mask_and_coherent(dev, DMA_BIT_MASK(32));
 173                 if (ret) {
 174                         dev_err(dev, "dma_set_mask_and_coherent failed (%d)\n",
 175                                 ret);
 176                         goto e_err;
 177                 }
 178         }
 179
 180         dev_set_drvdata(dev, sp);
 181
 182         ret = sp_init(sp);
 183         if (ret)
 184                 goto e_err;
 185
 186         dev_notice(dev, "enabled\n");
 187
 188         return 0;
 189
 190 e_err:
 191         dev_notice(dev, "initialization failed\n");
 192         return ret;
 193 }
 194
 195 static void sp_pci_remove(struct pci_dev *pdev)
 196 {
 197         struct device *dev = &pdev->dev;
 198         struct sp_device *sp = dev_get_drvdata(dev);
 199
 200         if (!sp)
 201                 return;
 202
 203         sp_destroy(sp);
 204
 205         sp_free_irqs(sp);
 206
 207         dev_notice(dev, "disabled\n");
 208 }
 209
 210 #ifdef CONFIG_PM
 211 static int sp_pci_suspend(struct pci_dev *pdev, pm_message_t state)
 212 {
 213         struct device *dev = &pdev->dev;
 214         struct sp_device *sp = dev_get_drvdata(dev);
 215
 216         return sp_suspend(sp, state);
 217 }
 218
 219 static int sp_pci_resume(struct pci_dev *pdev)
 220 {
 221         struct device *dev = &pdev->dev;
 222         struct sp_device *sp = dev_get_drvdata(dev);
 223
 224         return sp_resume(sp);
 225 }
 226 #endif
 227
 228 static const struct sp_dev_vdata dev_vdata[] = {
 229         {
 230                 .bar = 2,
 231 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
 232                 .ccp_vdata = &ccpv3,
 233 #endif
 234         },
 235         {
 236                 .bar = 2,
 237 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
 238                 .ccp_vdata = &ccpv5a,
 239 #endif
 240         },
 241         {
 242                 .bar = 2,
 243 #ifdef CONFIG_CRYPTO_DEV_SP_CCP
 244                 .ccp_vdata = &ccpv5b,
 245 #endif
 246         },
 247 };
 248 static const struct pci_device_id sp_pci_table[] = {
 249         { PCI_VDEVICE(AMD, 0x1537), (kernel_ulong_t)&dev_vdata[0] },
 250         { PCI_VDEVICE(AMD, 0x1456), (kernel_ulong_t)&dev_vdata[1] },
 251         { PCI_VDEVICE(AMD, 0x1468), (kernel_ulong_t)&dev_vdata[2] },
 252         /* Last entry must be zero */
 253         { 0, }
 254 };
 255 MODULE_DEVICE_TABLE(pci, sp_pci_table);
 256
 257 static struct pci_driver sp_pci_driver = {
 258         .name = "ccp",
 259         .id_table = sp_pci_table,
 260         .probe = sp_pci_probe,
 261         .remove = sp_pci_remove,
 262 #ifdef CONFIG_PM
 263         .suspend = sp_pci_suspend,
 264         .resume = sp_pci_resume,
 265 #endif
 266 };
 267
 268 int sp_pci_init(void)
 269 {
 270         return pci_register_driver(&sp_pci_driver);
 271 }
 272
 273 void sp_pci_exit(void)
 274 {
 275         pci_unregister_driver(&sp_pci_driver);
 276 }