xref: /openbmc/linux/kernel/kexec.c (revision 8622a0e5)
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * kexec.c - kexec_load system call
4  * Copyright (C) 2002-2004 Eric Biederman  <ebiederm@xmission.com>
5  */
6 
7 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
8 
9 #include <linux/capability.h>
10 #include <linux/mm.h>
11 #include <linux/file.h>
12 #include <linux/security.h>
13 #include <linux/kexec.h>
14 #include <linux/mutex.h>
15 #include <linux/list.h>
16 #include <linux/syscalls.h>
17 #include <linux/vmalloc.h>
18 #include <linux/slab.h>
19 
20 #include "kexec_internal.h"
21 
22 static int copy_user_segment_list(struct kimage *image,
23 				  unsigned long nr_segments,
24 				  struct kexec_segment __user *segments)
25 {
26 	int ret;
27 	size_t segment_bytes;
28 
29 	/* Read in the segments */
30 	image->nr_segments = nr_segments;
31 	segment_bytes = nr_segments * sizeof(*segments);
32 	ret = copy_from_user(image->segment, segments, segment_bytes);
33 	if (ret)
34 		ret = -EFAULT;
35 
36 	return ret;
37 }
38 
39 static int kimage_alloc_init(struct kimage **rimage, unsigned long entry,
40 			     unsigned long nr_segments,
41 			     struct kexec_segment __user *segments,
42 			     unsigned long flags)
43 {
44 	int ret;
45 	struct kimage *image;
46 	bool kexec_on_panic = flags & KEXEC_ON_CRASH;
47 
48 	if (kexec_on_panic) {
49 		/* Verify we have a valid entry point */
50 		if ((entry < phys_to_boot_phys(crashk_res.start)) ||
51 		    (entry > phys_to_boot_phys(crashk_res.end)))
52 			return -EADDRNOTAVAIL;
53 	}
54 
55 	/* Allocate and initialize a controlling structure */
56 	image = do_kimage_alloc_init();
57 	if (!image)
58 		return -ENOMEM;
59 
60 	image->start = entry;
61 
62 	ret = copy_user_segment_list(image, nr_segments, segments);
63 	if (ret)
64 		goto out_free_image;
65 
66 	if (kexec_on_panic) {
67 		/* Enable special crash kernel control page alloc policy. */
68 		image->control_page = crashk_res.start;
69 		image->type = KEXEC_TYPE_CRASH;
70 	}
71 
72 	ret = sanity_check_segment_list(image);
73 	if (ret)
74 		goto out_free_image;
75 
76 	/*
77 	 * Find a location for the control code buffer, and add it
78 	 * the vector of segments so that it's pages will also be
79 	 * counted as destination pages.
80 	 */
81 	ret = -ENOMEM;
82 	image->control_code_page = kimage_alloc_control_pages(image,
83 					   get_order(KEXEC_CONTROL_PAGE_SIZE));
84 	if (!image->control_code_page) {
85 		pr_err("Could not allocate control_code_buffer\n");
86 		goto out_free_image;
87 	}
88 
89 	if (!kexec_on_panic) {
90 		image->swap_page = kimage_alloc_control_pages(image, 0);
91 		if (!image->swap_page) {
92 			pr_err("Could not allocate swap buffer\n");
93 			goto out_free_control_pages;
94 		}
95 	}
96 
97 	*rimage = image;
98 	return 0;
99 out_free_control_pages:
100 	kimage_free_page_list(&image->control_pages);
101 out_free_image:
102 	kfree(image);
103 	return ret;
104 }
105 
106 static int do_kexec_load(unsigned long entry, unsigned long nr_segments,
107 		struct kexec_segment __user *segments, unsigned long flags)
108 {
109 	struct kimage **dest_image, *image;
110 	unsigned long i;
111 	int ret;
112 
113 	if (flags & KEXEC_ON_CRASH) {
114 		dest_image = &kexec_crash_image;
115 		if (kexec_crash_image)
116 			arch_kexec_unprotect_crashkres();
117 	} else {
118 		dest_image = &kexec_image;
119 	}
120 
121 	if (nr_segments == 0) {
122 		/* Uninstall image */
123 		kimage_free(xchg(dest_image, NULL));
124 		return 0;
125 	}
126 	if (flags & KEXEC_ON_CRASH) {
127 		/*
128 		 * Loading another kernel to switch to if this one
129 		 * crashes.  Free any current crash dump kernel before
130 		 * we corrupt it.
131 		 */
132 		kimage_free(xchg(&kexec_crash_image, NULL));
133 	}
134 
135 	ret = kimage_alloc_init(&image, entry, nr_segments, segments, flags);
136 	if (ret)
137 		return ret;
138 
139 	if (flags & KEXEC_PRESERVE_CONTEXT)
140 		image->preserve_context = 1;
141 
142 	ret = machine_kexec_prepare(image);
143 	if (ret)
144 		goto out;
145 
146 	/*
147 	 * Some architecture(like S390) may touch the crash memory before
148 	 * machine_kexec_prepare(), we must copy vmcoreinfo data after it.
149 	 */
150 	ret = kimage_crash_copy_vmcoreinfo(image);
151 	if (ret)
152 		goto out;
153 
154 	for (i = 0; i < nr_segments; i++) {
155 		ret = kimage_load_segment(image, &image->segment[i]);
156 		if (ret)
157 			goto out;
158 	}
159 
160 	kimage_terminate(image);
161 
162 	ret = machine_kexec_post_load(image);
163 	if (ret)
164 		goto out;
165 
166 	/* Install the new kernel and uninstall the old */
167 	image = xchg(dest_image, image);
168 
169 out:
170 	if ((flags & KEXEC_ON_CRASH) && kexec_crash_image)
171 		arch_kexec_protect_crashkres();
172 
173 	kimage_free(image);
174 	return ret;
175 }
176 
177 /*
178  * Exec Kernel system call: for obvious reasons only root may call it.
179  *
180  * This call breaks up into three pieces.
181  * - A generic part which loads the new kernel from the current
182  *   address space, and very carefully places the data in the
183  *   allocated pages.
184  *
185  * - A generic part that interacts with the kernel and tells all of
186  *   the devices to shut down.  Preventing on-going dmas, and placing
187  *   the devices in a consistent state so a later kernel can
188  *   reinitialize them.
189  *
190  * - A machine specific part that includes the syscall number
191  *   and then copies the image to it's final destination.  And
192  *   jumps into the image at entry.
193  *
194  * kexec does not sync, or unmount filesystems so if you need
195  * that to happen you need to do that yourself.
196  */
197 
198 static inline int kexec_load_check(unsigned long nr_segments,
199 				   unsigned long flags)
200 {
201 	int result;
202 
203 	/* We only trust the superuser with rebooting the system. */
204 	if (!capable(CAP_SYS_BOOT) || kexec_load_disabled)
205 		return -EPERM;
206 
207 	/* Permit LSMs and IMA to fail the kexec */
208 	result = security_kernel_load_data(LOADING_KEXEC_IMAGE);
209 	if (result < 0)
210 		return result;
211 
212 	/*
213 	 * kexec can be used to circumvent module loading restrictions, so
214 	 * prevent loading in that case
215 	 */
216 	result = security_locked_down(LOCKDOWN_KEXEC);
217 	if (result)
218 		return result;
219 
220 	/*
221 	 * Verify we have a legal set of flags
222 	 * This leaves us room for future extensions.
223 	 */
224 	if ((flags & KEXEC_FLAGS) != (flags & ~KEXEC_ARCH_MASK))
225 		return -EINVAL;
226 
227 	/* Put an artificial cap on the number
228 	 * of segments passed to kexec_load.
229 	 */
230 	if (nr_segments > KEXEC_SEGMENT_MAX)
231 		return -EINVAL;
232 
233 	return 0;
234 }
235 
236 SYSCALL_DEFINE4(kexec_load, unsigned long, entry, unsigned long, nr_segments,
237 		struct kexec_segment __user *, segments, unsigned long, flags)
238 {
239 	int result;
240 
241 	result = kexec_load_check(nr_segments, flags);
242 	if (result)
243 		return result;
244 
245 	/* Verify we are on the appropriate architecture */
246 	if (((flags & KEXEC_ARCH_MASK) != KEXEC_ARCH) &&
247 		((flags & KEXEC_ARCH_MASK) != KEXEC_ARCH_DEFAULT))
248 		return -EINVAL;
249 
250 	/* Because we write directly to the reserved memory
251 	 * region when loading crash kernels we need a mutex here to
252 	 * prevent multiple crash  kernels from attempting to load
253 	 * simultaneously, and to prevent a crash kernel from loading
254 	 * over the top of a in use crash kernel.
255 	 *
256 	 * KISS: always take the mutex.
257 	 */
258 	if (!mutex_trylock(&kexec_mutex))
259 		return -EBUSY;
260 
261 	result = do_kexec_load(entry, nr_segments, segments, flags);
262 
263 	mutex_unlock(&kexec_mutex);
264 
265 	return result;
266 }
267 
268 #ifdef CONFIG_COMPAT
269 COMPAT_SYSCALL_DEFINE4(kexec_load, compat_ulong_t, entry,
270 		       compat_ulong_t, nr_segments,
271 		       struct compat_kexec_segment __user *, segments,
272 		       compat_ulong_t, flags)
273 {
274 	struct compat_kexec_segment in;
275 	struct kexec_segment out, __user *ksegments;
276 	unsigned long i, result;
277 
278 	result = kexec_load_check(nr_segments, flags);
279 	if (result)
280 		return result;
281 
282 	/* Don't allow clients that don't understand the native
283 	 * architecture to do anything.
284 	 */
285 	if ((flags & KEXEC_ARCH_MASK) == KEXEC_ARCH_DEFAULT)
286 		return -EINVAL;
287 
288 	ksegments = compat_alloc_user_space(nr_segments * sizeof(out));
289 	for (i = 0; i < nr_segments; i++) {
290 		result = copy_from_user(&in, &segments[i], sizeof(in));
291 		if (result)
292 			return -EFAULT;
293 
294 		out.buf   = compat_ptr(in.buf);
295 		out.bufsz = in.bufsz;
296 		out.mem   = in.mem;
297 		out.memsz = in.memsz;
298 
299 		result = copy_to_user(&ksegments[i], &out, sizeof(out));
300 		if (result)
301 			return -EFAULT;
302 	}
303 
304 	/* Because we write directly to the reserved memory
305 	 * region when loading crash kernels we need a mutex here to
306 	 * prevent multiple crash  kernels from attempting to load
307 	 * simultaneously, and to prevent a crash kernel from loading
308 	 * over the top of a in use crash kernel.
309 	 *
310 	 * KISS: always take the mutex.
311 	 */
312 	if (!mutex_trylock(&kexec_mutex))
313 		return -EBUSY;
314 
315 	result = do_kexec_load(entry, nr_segments, ksegments, flags);
316 
317 	mutex_unlock(&kexec_mutex);
318 
319 	return result;
320 }
321 #endif
322