1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3 * s390 TRNG device driver
4 *
5 * Driver for the TRNG (true random number generation) command
6 * available via CPACF extension MSA 7 on the s390 arch.
7
8 * Copyright IBM Corp. 2017
9 * Author(s): Harald Freudenberger <freude@de.ibm.com>
10 */
11
12 #define KMSG_COMPONENT "trng"
13 #define pr_fmt(fmt) KMSG_COMPONENT ": " fmt
14
15 #include <linux/hw_random.h>
16 #include <linux/kernel.h>
17 #include <linux/module.h>
18 #include <linux/cpufeature.h>
19 #include <linux/miscdevice.h>
20 #include <linux/debugfs.h>
21 #include <linux/atomic.h>
22 #include <linux/random.h>
23 #include <linux/sched/signal.h>
24 #include <asm/debug.h>
25 #include <asm/cpacf.h>
26 #include <asm/archrandom.h>
27
28 MODULE_LICENSE("GPL v2");
29 MODULE_AUTHOR("IBM Corporation");
30 MODULE_DESCRIPTION("s390 CPACF TRNG device driver");
31
32
33 /* trng related debug feature things */
34
35 static debug_info_t *debug_info;
36
37 #define DEBUG_DBG(...) debug_sprintf_event(debug_info, 6, ##__VA_ARGS__)
38 #define DEBUG_INFO(...) debug_sprintf_event(debug_info, 5, ##__VA_ARGS__)
39 #define DEBUG_WARN(...) debug_sprintf_event(debug_info, 4, ##__VA_ARGS__)
40 #define DEBUG_ERR(...) debug_sprintf_event(debug_info, 3, ##__VA_ARGS__)
41
42
43 /* trng helpers */
44
45 static atomic64_t trng_dev_counter = ATOMIC64_INIT(0);
46 static atomic64_t trng_hwrng_counter = ATOMIC64_INIT(0);
47
48
49 /* file io functions */
50
trng_open(struct inode * inode,struct file * file)51 static int trng_open(struct inode *inode, struct file *file)
52 {
53 return nonseekable_open(inode, file);
54 }
55
trng_read(struct file * file,char __user * ubuf,size_t nbytes,loff_t * ppos)56 static ssize_t trng_read(struct file *file, char __user *ubuf,
57 size_t nbytes, loff_t *ppos)
58 {
59 u8 buf[32];
60 u8 *p = buf;
61 unsigned int n;
62 ssize_t ret = 0;
63
64 /*
65 * use buf for requests <= sizeof(buf),
66 * otherwise allocate one page and fetch
67 * pagewise.
68 */
69
70 if (nbytes > sizeof(buf)) {
71 p = (u8 *) __get_free_page(GFP_KERNEL);
72 if (!p)
73 return -ENOMEM;
74 }
75
76 while (nbytes) {
77 if (need_resched()) {
78 if (signal_pending(current)) {
79 if (ret == 0)
80 ret = -ERESTARTSYS;
81 break;
82 }
83 schedule();
84 }
85 n = nbytes > PAGE_SIZE ? PAGE_SIZE : nbytes;
86 cpacf_trng(NULL, 0, p, n);
87 atomic64_add(n, &trng_dev_counter);
88 if (copy_to_user(ubuf, p, n)) {
89 ret = -EFAULT;
90 break;
91 }
92 nbytes -= n;
93 ubuf += n;
94 ret += n;
95 }
96
97 if (p != buf)
98 free_page((unsigned long) p);
99
100 DEBUG_DBG("trng_read()=%zd\n", ret);
101 return ret;
102 }
103
104
105 /* sysfs */
106
trng_counter_show(struct device * dev,struct device_attribute * attr,char * buf)107 static ssize_t trng_counter_show(struct device *dev,
108 struct device_attribute *attr, char *buf)
109 {
110 u64 dev_counter = atomic64_read(&trng_dev_counter);
111 u64 hwrng_counter = atomic64_read(&trng_hwrng_counter);
112 u64 arch_counter = atomic64_read(&s390_arch_random_counter);
113
114 return sysfs_emit(buf,
115 "trng: %llu\n"
116 "hwrng: %llu\n"
117 "arch: %llu\n"
118 "total: %llu\n",
119 dev_counter, hwrng_counter, arch_counter,
120 dev_counter + hwrng_counter + arch_counter);
121 }
122 static DEVICE_ATTR(byte_counter, 0444, trng_counter_show, NULL);
123
124 static struct attribute *trng_dev_attrs[] = {
125 &dev_attr_byte_counter.attr,
126 NULL
127 };
128
129 static const struct attribute_group trng_dev_attr_group = {
130 .attrs = trng_dev_attrs
131 };
132
133 static const struct attribute_group *trng_dev_attr_groups[] = {
134 &trng_dev_attr_group,
135 NULL
136 };
137
138 static const struct file_operations trng_fops = {
139 .owner = THIS_MODULE,
140 .open = &trng_open,
141 .release = NULL,
142 .read = &trng_read,
143 .llseek = noop_llseek,
144 };
145
146 static struct miscdevice trng_dev = {
147 .name = "trng",
148 .minor = MISC_DYNAMIC_MINOR,
149 .mode = 0444,
150 .fops = &trng_fops,
151 .groups = trng_dev_attr_groups,
152 };
153
154
155 /* hwrng_register */
156
_trng_hwrng_read(u8 * buf,size_t len)157 static inline void _trng_hwrng_read(u8 *buf, size_t len)
158 {
159 cpacf_trng(NULL, 0, buf, len);
160 atomic64_add(len, &trng_hwrng_counter);
161 }
162
trng_hwrng_data_read(struct hwrng * rng,u32 * data)163 static int trng_hwrng_data_read(struct hwrng *rng, u32 *data)
164 {
165 size_t len = sizeof(*data);
166
167 _trng_hwrng_read((u8 *) data, len);
168
169 DEBUG_DBG("trng_hwrng_data_read()=%zu\n", len);
170
171 return len;
172 }
173
trng_hwrng_read(struct hwrng * rng,void * data,size_t max,bool wait)174 static int trng_hwrng_read(struct hwrng *rng, void *data, size_t max, bool wait)
175 {
176 size_t len = max <= PAGE_SIZE ? max : PAGE_SIZE;
177
178 _trng_hwrng_read((u8 *) data, len);
179
180 DEBUG_DBG("trng_hwrng_read()=%zu\n", len);
181
182 return len;
183 }
184
185 /*
186 * hwrng register struct
187 * The trng is supposed to have 100% entropy, and thus we register with a very
188 * high quality value. If we ever have a better driver in the future, we should
189 * change this value again when we merge this driver.
190 */
191 static struct hwrng trng_hwrng_dev = {
192 .name = "s390-trng",
193 .data_read = trng_hwrng_data_read,
194 .read = trng_hwrng_read,
195 };
196
197
198 /* init and exit */
199
trng_debug_init(void)200 static void __init trng_debug_init(void)
201 {
202 debug_info = debug_register("trng", 1, 1, 4 * sizeof(long));
203 debug_register_view(debug_info, &debug_sprintf_view);
204 debug_set_level(debug_info, 3);
205 }
206
trng_debug_exit(void)207 static void trng_debug_exit(void)
208 {
209 debug_unregister(debug_info);
210 }
211
trng_init(void)212 static int __init trng_init(void)
213 {
214 int ret;
215
216 trng_debug_init();
217
218 /* check if subfunction CPACF_PRNO_TRNG is available */
219 if (!cpacf_query_func(CPACF_PRNO, CPACF_PRNO_TRNG)) {
220 DEBUG_INFO("trng_init CPACF_PRNO_TRNG not available\n");
221 ret = -ENODEV;
222 goto out_dbg;
223 }
224
225 ret = misc_register(&trng_dev);
226 if (ret) {
227 DEBUG_WARN("trng_init misc_register() failed rc=%d\n", ret);
228 goto out_dbg;
229 }
230
231 ret = hwrng_register(&trng_hwrng_dev);
232 if (ret) {
233 DEBUG_WARN("trng_init hwrng_register() failed rc=%d\n", ret);
234 goto out_misc;
235 }
236
237 DEBUG_DBG("trng_init successful\n");
238
239 return 0;
240
241 out_misc:
242 misc_deregister(&trng_dev);
243 out_dbg:
244 trng_debug_exit();
245 return ret;
246 }
247
trng_exit(void)248 static void __exit trng_exit(void)
249 {
250 hwrng_unregister(&trng_hwrng_dev);
251 misc_deregister(&trng_dev);
252 trng_debug_exit();
253 }
254
255 module_cpu_feature_match(S390_CPU_FEATURE_MSA, trng_init);
256 module_exit(trng_exit);
257