修改内核路径

hc

2024-05-14 bedbef8ad3e75a304af6361af235302bcc61d06b

 kernel/arch/um/drivers/random.c
..
..
@@ -11,6 +11,7 @@
11
11
 #include <linux/fs.h>
12
12
 #include <linux/interrupt.h>
13
13
 #include <linux/miscdevice.h>
14
+#include <linux/hw_random.h>
14
15
 #include <linux/delay.h>
15
16
 #include <linux/uaccess.h>
16
17
 #include <init.h>
..
..
@@ -18,101 +19,45 @@
18
19
 #include <os.h>
19
20
 
20
21
 /*
21
- * core module and version information
22
+ * core module information
22
23
  */
23
-#define RNG_VERSION "1.0.0"
24
24
 #define RNG_MODULE_NAME "hw_random"
25
-
26
-#define RNG_MISCDEV_MINOR		183 /* official */
27
25
 
28
26
 /* Changed at init time, in the non-modular case, and at module load
29
27
  * time, in the module case.  Presumably, the module subsystem
30
28
  * protects against a module being loaded twice at the same time.
31
29
  */
32
30
 static int random_fd = -1;
33
-static DECLARE_WAIT_QUEUE_HEAD(host_read_wait);
31
+static struct hwrng hwrng;
32
+static DECLARE_COMPLETION(have_data);
34
33
 
35
-static int rng_dev_open (struct inode *inode, struct file *filp)
34
+static int rng_dev_read(struct hwrng *rng, void *buf, size_t max, bool block)
36
35
 {
37
-	/* enforce read-only access to this chrdev */
38
-	if ((filp->f_mode & FMODE_READ) == 0)
39
-		return -EINVAL;
40
-	if ((filp->f_mode & FMODE_WRITE) != 0)
41
-		return -EINVAL;
36
+	int ret;
42
37
 
43
-	return 0;
44
-}
45
-
46
-static atomic_t host_sleep_count = ATOMIC_INIT(0);
47
-
48
-static ssize_t rng_dev_read (struct file *filp, char __user *buf, size_t size,
49
-			     loff_t *offp)
50
-{
51
-	u32 data;
52
-	int n, ret = 0, have_data;
53
-
54
-	while (size) {
55
-		n = os_read_file(random_fd, &data, sizeof(data));
56
-		if (n > 0) {
57
-			have_data = n;
58
-			while (have_data && size) {
59
-				if (put_user((u8) data, buf++)) {
60
-					ret = ret ? : -EFAULT;
61
-					break;
62
-				}
63
-				size--;
64
-				ret++;
65
-				have_data--;
66
-				data >>= 8;
67
-			}
68
-		}
69
-		else if (n == -EAGAIN) {
70
-			DECLARE_WAITQUEUE(wait, current);
71
-
72
-			if (filp->f_flags & O_NONBLOCK)
73
-				return ret ? : -EAGAIN;
74
-
75
-			atomic_inc(&host_sleep_count);
76
-			reactivate_fd(random_fd, RANDOM_IRQ);
38
+	for (;;) {
39
+		ret = os_read_file(random_fd, buf, max);
40
+		if (block && ret == -EAGAIN) {
77
41
 			add_sigio_fd(random_fd);
78
42
 
79
-			add_wait_queue(&host_read_wait, &wait);
80
-			set_current_state(TASK_INTERRUPTIBLE);
43
+			ret = wait_for_completion_killable(&have_data);
81
44
 
82
-			schedule();
83
-			remove_wait_queue(&host_read_wait, &wait);
45
+			ignore_sigio_fd(random_fd);
46
+			deactivate_fd(random_fd, RANDOM_IRQ);
84
47
 
85
-			if (atomic_dec_and_test(&host_sleep_count)) {
86
-				ignore_sigio_fd(random_fd);
87
-				deactivate_fd(random_fd, RANDOM_IRQ);
88
-			}
48
+			if (ret < 0)
49
+				break;
50
+		} else {
51
+			break;
89
52
 		}
90
-		else
91
-			return n;
92
-
93
-		if (signal_pending (current))
94
-			return ret ? : -ERESTARTSYS;
95
53
 	}
96
-	return ret;
54
+
55
+	return ret != -EAGAIN ? ret : 0;
97
56
 }
98
-
99
-static const struct file_operations rng_chrdev_ops = {
100
-	.owner		= THIS_MODULE,
101
-	.open		= rng_dev_open,
102
-	.read		= rng_dev_read,
103
-	.llseek		= noop_llseek,
104
-};
105
-
106
-/* rng_init shouldn't be called more than once at boot time */
107
-static struct miscdevice rng_miscdev = {
108
-	RNG_MISCDEV_MINOR,
109
-	RNG_MODULE_NAME,
110
-	&rng_chrdev_ops,
111
-};
112
57
 
113
58
 static irqreturn_t random_interrupt(int irq, void *data)
114
59
 {
115
-	wake_up(&host_read_wait);
60
+	complete(&have_data);
116
61
 
117
62
 	return IRQ_HANDLED;
118
63
 }
..
..
@@ -129,18 +74,19 @@
129
74
 		goto out;
130
75
 
131
76
 	random_fd = err;
132
-
133
77
 	err = um_request_irq(RANDOM_IRQ, random_fd, IRQ_READ, random_interrupt,
134
78
 			     0, "random", NULL);
135
79
 	if (err)
136
80
 		goto err_out_cleanup_hw;
137
81
 
138
82
 	sigio_broken(random_fd, 1);
83
+	hwrng.name = RNG_MODULE_NAME;
84
+	hwrng.read = rng_dev_read;
85
+	hwrng.quality = 1024;
139
86
 
140
-	err = misc_register (&rng_miscdev);
87
+	err = hwrng_register(&hwrng);
141
88
 	if (err) {
142
-		printk (KERN_ERR RNG_MODULE_NAME ": misc device register "
143
-			"failed\n");
89
+		pr_err(RNG_MODULE_NAME " registering failed (%d)\n", err);
144
90
 		goto err_out_cleanup_hw;
145
91
 	}
146
92
 out:
..
..
@@ -164,8 +110,8 @@
164
110
 
165
111
 static void __exit rng_cleanup(void)
166
112
 {
113
+	hwrng_unregister(&hwrng);
167
114
 	os_close_file(random_fd);
168
-	misc_deregister (&rng_miscdev);
169
115
 }
170
116
 
171
117
 module_init (rng_init);