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