drivers/char/hw_random/atmel-rng.c

   1 /*
   2  * Copyright (c) 2011 Peter Korsgaard <jacmet@sunsite.dk>
   3  *
   4  * This file is licensed under  the terms of the GNU General Public
   5  * License version 2. This program is licensed "as is" without any
   6  * warranty of any kind, whether express or implied.
   7  */
   8
   9 #include <linux/kernel.h>
  10 #include <linux/module.h>
  11 #include <linux/slab.h>
  12 #include <linux/err.h>
  13 #include <linux/clk.h>
  14 #include <linux/io.h>
  15 #include <linux/hw_random.h>
  16 #include <linux/platform_device.h>
  17
  18 #define TRNG_CR         0x00
  19 #define TRNG_ISR        0x1c
  20 #define TRNG_ODATA      0x50
  21
  22 #define TRNG_KEY        0x524e4700 /* RNG */
  23
  24 struct atmel_trng {
  25         struct clk *clk;
  26         void __iomem *base;
  27         struct hwrng rng;
  28 };
  29
  30 static int atmel_trng_read(struct hwrng *rng, void *buf, size_t max,
  31                            bool wait)
  32 {
  33         struct atmel_trng *trng = container_of(rng, struct atmel_trng, rng);
  34         u32 *data = buf;
  35
  36         /* data ready? */
  37         if (readl(trng->base + TRNG_ISR) & 1) {
  38                 *data = readl(trng->base + TRNG_ODATA);
  39                 /*
  40                   ensure data ready is only set again AFTER the next data
  41                   word is ready in case it got set between checking ISR
  42                   and reading ODATA, so we don't risk re-reading the
  43                   same word
  44                 */
  45                 readl(trng->base + TRNG_ISR);
  46                 return 4;
  47         } else
  48                 return 0;
  49 }
  50
  51 static void atmel_trng_enable(struct atmel_trng *trng)
  52 {
  53         writel(TRNG_KEY | 1, trng->base + TRNG_CR);
  54 }
  55
  56 static void atmel_trng_disable(struct atmel_trng *trng)
  57 {
  58         writel(TRNG_KEY, trng->base + TRNG_CR);
  59 }
  60
  61 static int atmel_trng_probe(struct platform_device *pdev)
  62 {
  63         struct atmel_trng *trng;
  64         struct resource *res;
  65         int ret;
  66
  67         trng = devm_kzalloc(&pdev->dev, sizeof(*trng), GFP_KERNEL);
  68         if (!trng)
  69                 return -ENOMEM;
  70
  71         res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
  72         trng->base = devm_ioremap_resource(&pdev->dev, res);
  73         if (IS_ERR(trng->base))
  74                 return PTR_ERR(trng->base);
  75
  76         trng->clk = devm_clk_get(&pdev->dev, NULL);
  77         if (IS_ERR(trng->clk))
  78                 return PTR_ERR(trng->clk);
  79
  80         ret = clk_prepare_enable(trng->clk);
  81         if (ret)
  82                 return ret;
  83
  84         atmel_trng_enable(trng);
  85         trng->rng.name = pdev->name;
  86         trng->rng.read = atmel_trng_read;
  87
  88         ret = hwrng_register(&trng->rng);
  89         if (ret)
  90                 goto err_register;
  91
  92         platform_set_drvdata(pdev, trng);
  93
  94         return 0;
  95
  96 err_register:
  97         clk_disable_unprepare(trng->clk);
  98         atmel_trng_disable(trng);
  99         return ret;
 100 }
 101
 102 static int atmel_trng_remove(struct platform_device *pdev)
 103 {
 104         struct atmel_trng *trng = platform_get_drvdata(pdev);
 105
 106         hwrng_unregister(&trng->rng);
 107
 108         atmel_trng_disable(trng);
 109         clk_disable_unprepare(trng->clk);
 110
 111         return 0;
 112 }
 113
 114 #ifdef CONFIG_PM
 115 static int atmel_trng_suspend(struct device *dev)
 116 {
 117         struct atmel_trng *trng = dev_get_drvdata(dev);
 118
 119         atmel_trng_disable(trng);
 120         clk_disable_unprepare(trng->clk);
 121
 122         return 0;
 123 }
 124
 125 static int atmel_trng_resume(struct device *dev)
 126 {
 127         struct atmel_trng *trng = dev_get_drvdata(dev);
 128         int ret;
 129
 130         ret = clk_prepare_enable(trng->clk);
 131         if (ret)
 132                 return ret;
 133
 134         atmel_trng_enable(trng);
 135
 136         return 0;
 137 }
 138
 139 static const struct dev_pm_ops atmel_trng_pm_ops = {
 140         .suspend        = atmel_trng_suspend,
 141         .resume         = atmel_trng_resume,
 142 };
 143 #endif /* CONFIG_PM */
 144
 145 static const struct of_device_id atmel_trng_dt_ids[] = {
 146         { .compatible = "atmel,at91sam9g45-trng" },
 147         { /* sentinel */ }
 148 };
 149 MODULE_DEVICE_TABLE(of, atmel_trng_dt_ids);
 150
 151 static struct platform_driver atmel_trng_driver = {
 152         .probe          = atmel_trng_probe,
 153         .remove         = atmel_trng_remove,
 154         .driver         = {
 155                 .name   = "atmel-trng",
 156 #ifdef CONFIG_PM
 157                 .pm     = &atmel_trng_pm_ops,
 158 #endif /* CONFIG_PM */
 159                 .of_match_table = atmel_trng_dt_ids,
 160         },
 161 };
 162
 163 module_platform_driver(atmel_trng_driver);
 164
 165 MODULE_LICENSE("GPL");
 166 MODULE_AUTHOR("Peter Korsgaard <jacmet@sunsite.dk>");
 167 MODULE_DESCRIPTION("Atmel true random number generator driver");