]>
Commit | Line | Data |
---|---|---|
1 | /* | |
2 | * Access kernel memory without faulting. | |
3 | */ | |
4 | #include <linux/module.h> | |
5 | #include <linux/mm.h> | |
6 | #include <linux/uaccess.h> | |
7 | ||
8 | /** | |
9 | * probe_kernel_read(): safely attempt to read from a location | |
10 | * @dst: pointer to the buffer that shall take the data | |
11 | * @src: address to read from | |
12 | * @size: size of the data chunk | |
13 | * | |
14 | * Safely read from address @src to the buffer at @dst. If a kernel fault | |
15 | * happens, handle that and return -EFAULT. | |
16 | */ | |
17 | ||
18 | long __weak probe_kernel_read(void *dst, const void *src, size_t size) | |
19 | __attribute__((alias("__probe_kernel_read"))); | |
20 | ||
21 | long __probe_kernel_read(void *dst, const void *src, size_t size) | |
22 | { | |
23 | long ret; | |
24 | mm_segment_t old_fs = get_fs(); | |
25 | ||
26 | set_fs(KERNEL_DS); | |
27 | pagefault_disable(); | |
28 | ret = __copy_from_user_inatomic(dst, | |
29 | (__force const void __user *)src, size); | |
30 | pagefault_enable(); | |
31 | set_fs(old_fs); | |
32 | ||
33 | return ret ? -EFAULT : 0; | |
34 | } | |
35 | EXPORT_SYMBOL_GPL(probe_kernel_read); | |
36 | ||
37 | /** | |
38 | * probe_kernel_write(): safely attempt to write to a location | |
39 | * @dst: address to write to | |
40 | * @src: pointer to the data that shall be written | |
41 | * @size: size of the data chunk | |
42 | * | |
43 | * Safely write to address @dst from the buffer at @src. If a kernel fault | |
44 | * happens, handle that and return -EFAULT. | |
45 | */ | |
46 | long __weak probe_kernel_write(void *dst, const void *src, size_t size) | |
47 | __attribute__((alias("__probe_kernel_write"))); | |
48 | ||
49 | long __probe_kernel_write(void *dst, const void *src, size_t size) | |
50 | { | |
51 | long ret; | |
52 | mm_segment_t old_fs = get_fs(); | |
53 | ||
54 | set_fs(KERNEL_DS); | |
55 | pagefault_disable(); | |
56 | ret = __copy_to_user_inatomic((__force void __user *)dst, src, size); | |
57 | pagefault_enable(); | |
58 | set_fs(old_fs); | |
59 | ||
60 | return ret ? -EFAULT : 0; | |
61 | } | |
62 | EXPORT_SYMBOL_GPL(probe_kernel_write); |