2 * kexec: kexec_file_load system call
4 * Copyright (C) 2014 Red Hat Inc.
6 * Vivek Goyal <vgoyal@redhat.com>
8 * This source code is licensed under the GNU General Public License,
9 * Version 2. See the file COPYING for more details.
12 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
14 #include <linux/capability.h>
16 #include <linux/file.h>
17 #include <linux/slab.h>
18 #include <linux/kexec.h>
19 #include <linux/mutex.h>
20 #include <linux/list.h>
22 #include <crypto/hash.h>
23 #include <crypto/sha.h>
24 #include <linux/syscalls.h>
25 #include <linux/vmalloc.h>
26 #include "kexec_internal.h"
29 * Declare these symbols weak so that if architecture provides a purgatory,
30 * these will be overridden.
32 char __weak kexec_purgatory
[0];
33 size_t __weak kexec_purgatory_size
= 0;
35 static int kexec_calculate_store_digests(struct kimage
*image
);
37 /* Architectures can provide this probe function */
38 int __weak
arch_kexec_kernel_image_probe(struct kimage
*image
, void *buf
,
39 unsigned long buf_len
)
44 void * __weak
arch_kexec_kernel_image_load(struct kimage
*image
)
46 return ERR_PTR(-ENOEXEC
);
49 int __weak
arch_kimage_file_post_load_cleanup(struct kimage
*image
)
54 #ifdef CONFIG_KEXEC_VERIFY_SIG
55 int __weak
arch_kexec_kernel_verify_sig(struct kimage
*image
, void *buf
,
56 unsigned long buf_len
)
62 /* Apply relocations of type RELA */
64 arch_kexec_apply_relocations_add(const Elf_Ehdr
*ehdr
, Elf_Shdr
*sechdrs
,
67 pr_err("RELA relocation unsupported.\n");
71 /* Apply relocations of type REL */
73 arch_kexec_apply_relocations(const Elf_Ehdr
*ehdr
, Elf_Shdr
*sechdrs
,
76 pr_err("REL relocation unsupported.\n");
81 * Free up memory used by kernel, initrd, and command line. This is temporary
82 * memory allocation which is not needed any more after these buffers have
83 * been loaded into separate segments and have been copied elsewhere.
85 void kimage_file_post_load_cleanup(struct kimage
*image
)
87 struct purgatory_info
*pi
= &image
->purgatory_info
;
89 vfree(image
->kernel_buf
);
90 image
->kernel_buf
= NULL
;
92 vfree(image
->initrd_buf
);
93 image
->initrd_buf
= NULL
;
95 kfree(image
->cmdline_buf
);
96 image
->cmdline_buf
= NULL
;
98 vfree(pi
->purgatory_buf
);
99 pi
->purgatory_buf
= NULL
;
104 /* See if architecture has anything to cleanup post load */
105 arch_kimage_file_post_load_cleanup(image
);
108 * Above call should have called into bootloader to free up
109 * any data stored in kimage->image_loader_data. It should
110 * be ok now to free it up.
112 kfree(image
->image_loader_data
);
113 image
->image_loader_data
= NULL
;
117 * In file mode list of segments is prepared by kernel. Copy relevant
118 * data from user space, do error checking, prepare segment list
121 kimage_file_prepare_segments(struct kimage
*image
, int kernel_fd
, int initrd_fd
,
122 const char __user
*cmdline_ptr
,
123 unsigned long cmdline_len
, unsigned flags
)
129 ret
= kernel_read_file_from_fd(kernel_fd
, &image
->kernel_buf
,
130 &size
, INT_MAX
, READING_KEXEC_IMAGE
);
133 image
->kernel_buf_len
= size
;
135 /* Call arch image probe handlers */
136 ret
= arch_kexec_kernel_image_probe(image
, image
->kernel_buf
,
137 image
->kernel_buf_len
);
141 #ifdef CONFIG_KEXEC_VERIFY_SIG
142 ret
= arch_kexec_kernel_verify_sig(image
, image
->kernel_buf
,
143 image
->kernel_buf_len
);
145 pr_debug("kernel signature verification failed.\n");
148 pr_debug("kernel signature verification successful.\n");
150 /* It is possible that there no initramfs is being loaded */
151 if (!(flags
& KEXEC_FILE_NO_INITRAMFS
)) {
152 ret
= kernel_read_file_from_fd(initrd_fd
, &image
->initrd_buf
,
154 READING_KEXEC_INITRAMFS
);
157 image
->initrd_buf_len
= size
;
161 image
->cmdline_buf
= kzalloc(cmdline_len
, GFP_KERNEL
);
162 if (!image
->cmdline_buf
) {
167 ret
= copy_from_user(image
->cmdline_buf
, cmdline_ptr
,
174 image
->cmdline_buf_len
= cmdline_len
;
176 /* command line should be a string with last byte null */
177 if (image
->cmdline_buf
[cmdline_len
- 1] != '\0') {
183 /* Call arch image load handlers */
184 ldata
= arch_kexec_kernel_image_load(image
);
187 ret
= PTR_ERR(ldata
);
191 image
->image_loader_data
= ldata
;
193 /* In case of error, free up all allocated memory in this function */
195 kimage_file_post_load_cleanup(image
);
200 kimage_file_alloc_init(struct kimage
**rimage
, int kernel_fd
,
201 int initrd_fd
, const char __user
*cmdline_ptr
,
202 unsigned long cmdline_len
, unsigned long flags
)
205 struct kimage
*image
;
206 bool kexec_on_panic
= flags
& KEXEC_FILE_ON_CRASH
;
208 image
= do_kimage_alloc_init();
212 image
->file_mode
= 1;
214 if (kexec_on_panic
) {
215 /* Enable special crash kernel control page alloc policy. */
216 image
->control_page
= crashk_res
.start
;
217 image
->type
= KEXEC_TYPE_CRASH
;
220 ret
= kimage_file_prepare_segments(image
, kernel_fd
, initrd_fd
,
221 cmdline_ptr
, cmdline_len
, flags
);
225 ret
= sanity_check_segment_list(image
);
227 goto out_free_post_load_bufs
;
230 image
->control_code_page
= kimage_alloc_control_pages(image
,
231 get_order(KEXEC_CONTROL_PAGE_SIZE
));
232 if (!image
->control_code_page
) {
233 pr_err("Could not allocate control_code_buffer\n");
234 goto out_free_post_load_bufs
;
237 if (!kexec_on_panic
) {
238 image
->swap_page
= kimage_alloc_control_pages(image
, 0);
239 if (!image
->swap_page
) {
240 pr_err("Could not allocate swap buffer\n");
241 goto out_free_control_pages
;
247 out_free_control_pages
:
248 kimage_free_page_list(&image
->control_pages
);
249 out_free_post_load_bufs
:
250 kimage_file_post_load_cleanup(image
);
256 SYSCALL_DEFINE5(kexec_file_load
, int, kernel_fd
, int, initrd_fd
,
257 unsigned long, cmdline_len
, const char __user
*, cmdline_ptr
,
258 unsigned long, flags
)
261 struct kimage
**dest_image
, *image
;
263 /* We only trust the superuser with rebooting the system. */
264 if (!capable(CAP_SYS_BOOT
) || kexec_load_disabled
)
267 /* Make sure we have a legal set of flags */
268 if (flags
!= (flags
& KEXEC_FILE_FLAGS
))
273 if (!mutex_trylock(&kexec_mutex
))
276 dest_image
= &kexec_image
;
277 if (flags
& KEXEC_FILE_ON_CRASH
) {
278 dest_image
= &kexec_crash_image
;
279 if (kexec_crash_image
)
280 arch_kexec_unprotect_crashkres();
283 if (flags
& KEXEC_FILE_UNLOAD
)
287 * In case of crash, new kernel gets loaded in reserved region. It is
288 * same memory where old crash kernel might be loaded. Free any
289 * current crash dump kernel before we corrupt it.
291 if (flags
& KEXEC_FILE_ON_CRASH
)
292 kimage_free(xchg(&kexec_crash_image
, NULL
));
294 ret
= kimage_file_alloc_init(&image
, kernel_fd
, initrd_fd
, cmdline_ptr
,
299 ret
= machine_kexec_prepare(image
);
303 ret
= kexec_calculate_store_digests(image
);
307 for (i
= 0; i
< image
->nr_segments
; i
++) {
308 struct kexec_segment
*ksegment
;
310 ksegment
= &image
->segment
[i
];
311 pr_debug("Loading segment %d: buf=0x%p bufsz=0x%zx mem=0x%lx memsz=0x%zx\n",
312 i
, ksegment
->buf
, ksegment
->bufsz
, ksegment
->mem
,
315 ret
= kimage_load_segment(image
, &image
->segment
[i
]);
320 kimage_terminate(image
);
323 * Free up any temporary buffers allocated which are not needed
324 * after image has been loaded
326 kimage_file_post_load_cleanup(image
);
328 image
= xchg(dest_image
, image
);
330 if ((flags
& KEXEC_FILE_ON_CRASH
) && kexec_crash_image
)
331 arch_kexec_protect_crashkres();
333 mutex_unlock(&kexec_mutex
);
338 static int locate_mem_hole_top_down(unsigned long start
, unsigned long end
,
339 struct kexec_buf
*kbuf
)
341 struct kimage
*image
= kbuf
->image
;
342 unsigned long temp_start
, temp_end
;
344 temp_end
= min(end
, kbuf
->buf_max
);
345 temp_start
= temp_end
- kbuf
->memsz
;
348 /* align down start */
349 temp_start
= temp_start
& (~(kbuf
->buf_align
- 1));
351 if (temp_start
< start
|| temp_start
< kbuf
->buf_min
)
354 temp_end
= temp_start
+ kbuf
->memsz
- 1;
357 * Make sure this does not conflict with any of existing
360 if (kimage_is_destination_range(image
, temp_start
, temp_end
)) {
361 temp_start
= temp_start
- PAGE_SIZE
;
365 /* We found a suitable memory range */
369 /* If we are here, we found a suitable memory range */
370 kbuf
->mem
= temp_start
;
372 /* Success, stop navigating through remaining System RAM ranges */
376 static int locate_mem_hole_bottom_up(unsigned long start
, unsigned long end
,
377 struct kexec_buf
*kbuf
)
379 struct kimage
*image
= kbuf
->image
;
380 unsigned long temp_start
, temp_end
;
382 temp_start
= max(start
, kbuf
->buf_min
);
385 temp_start
= ALIGN(temp_start
, kbuf
->buf_align
);
386 temp_end
= temp_start
+ kbuf
->memsz
- 1;
388 if (temp_end
> end
|| temp_end
> kbuf
->buf_max
)
391 * Make sure this does not conflict with any of existing
394 if (kimage_is_destination_range(image
, temp_start
, temp_end
)) {
395 temp_start
= temp_start
+ PAGE_SIZE
;
399 /* We found a suitable memory range */
403 /* If we are here, we found a suitable memory range */
404 kbuf
->mem
= temp_start
;
406 /* Success, stop navigating through remaining System RAM ranges */
410 static int locate_mem_hole_callback(u64 start
, u64 end
, void *arg
)
412 struct kexec_buf
*kbuf
= (struct kexec_buf
*)arg
;
413 unsigned long sz
= end
- start
+ 1;
415 /* Returning 0 will take to next memory range */
416 if (sz
< kbuf
->memsz
)
419 if (end
< kbuf
->buf_min
|| start
> kbuf
->buf_max
)
423 * Allocate memory top down with-in ram range. Otherwise bottom up
427 return locate_mem_hole_top_down(start
, end
, kbuf
);
428 return locate_mem_hole_bottom_up(start
, end
, kbuf
);
432 * arch_kexec_walk_mem - call func(data) on free memory regions
433 * @kbuf: Context info for the search. Also passed to @func.
434 * @func: Function to call for each memory region.
436 * Return: The memory walk will stop when func returns a non-zero value
437 * and that value will be returned. If all free regions are visited without
438 * func returning non-zero, then zero will be returned.
440 int __weak
arch_kexec_walk_mem(struct kexec_buf
*kbuf
,
441 int (*func
)(u64
, u64
, void *))
443 if (kbuf
->image
->type
== KEXEC_TYPE_CRASH
)
444 return walk_iomem_res_desc(crashk_res
.desc
,
445 IORESOURCE_SYSTEM_RAM
| IORESOURCE_BUSY
,
446 crashk_res
.start
, crashk_res
.end
,
449 return walk_system_ram_res(0, ULONG_MAX
, kbuf
, func
);
453 * kexec_locate_mem_hole - find free memory for the purgatory or the next kernel
454 * @kbuf: Parameters for the memory search.
456 * On success, kbuf->mem will have the start address of the memory region found.
458 * Return: 0 on success, negative errno on error.
460 int kexec_locate_mem_hole(struct kexec_buf
*kbuf
)
464 ret
= arch_kexec_walk_mem(kbuf
, locate_mem_hole_callback
);
466 return ret
== 1 ? 0 : -EADDRNOTAVAIL
;
470 * kexec_add_buffer - place a buffer in a kexec segment
471 * @kbuf: Buffer contents and memory parameters.
473 * This function assumes that kexec_mutex is held.
474 * On successful return, @kbuf->mem will have the physical address of
475 * the buffer in memory.
477 * Return: 0 on success, negative errno on error.
479 int kexec_add_buffer(struct kexec_buf
*kbuf
)
482 struct kexec_segment
*ksegment
;
485 /* Currently adding segment this way is allowed only in file mode */
486 if (!kbuf
->image
->file_mode
)
489 if (kbuf
->image
->nr_segments
>= KEXEC_SEGMENT_MAX
)
493 * Make sure we are not trying to add buffer after allocating
494 * control pages. All segments need to be placed first before
495 * any control pages are allocated. As control page allocation
496 * logic goes through list of segments to make sure there are
497 * no destination overlaps.
499 if (!list_empty(&kbuf
->image
->control_pages
)) {
504 /* Ensure minimum alignment needed for segments. */
505 kbuf
->memsz
= ALIGN(kbuf
->memsz
, PAGE_SIZE
);
506 kbuf
->buf_align
= max(kbuf
->buf_align
, PAGE_SIZE
);
508 /* Walk the RAM ranges and allocate a suitable range for the buffer */
509 ret
= kexec_locate_mem_hole(kbuf
);
513 /* Found a suitable memory range */
514 ksegment
= &kbuf
->image
->segment
[kbuf
->image
->nr_segments
];
515 ksegment
->kbuf
= kbuf
->buffer
;
516 ksegment
->bufsz
= kbuf
->bufsz
;
517 ksegment
->mem
= kbuf
->mem
;
518 ksegment
->memsz
= kbuf
->memsz
;
519 kbuf
->image
->nr_segments
++;
523 /* Calculate and store the digest of segments */
524 static int kexec_calculate_store_digests(struct kimage
*image
)
526 struct crypto_shash
*tfm
;
527 struct shash_desc
*desc
;
528 int ret
= 0, i
, j
, zero_buf_sz
, sha_region_sz
;
529 size_t desc_size
, nullsz
;
532 struct kexec_sha_region
*sha_regions
;
533 struct purgatory_info
*pi
= &image
->purgatory_info
;
535 zero_buf
= __va(page_to_pfn(ZERO_PAGE(0)) << PAGE_SHIFT
);
536 zero_buf_sz
= PAGE_SIZE
;
538 tfm
= crypto_alloc_shash("sha256", 0, 0);
544 desc_size
= crypto_shash_descsize(tfm
) + sizeof(*desc
);
545 desc
= kzalloc(desc_size
, GFP_KERNEL
);
551 sha_region_sz
= KEXEC_SEGMENT_MAX
* sizeof(struct kexec_sha_region
);
552 sha_regions
= vzalloc(sha_region_sz
);
559 ret
= crypto_shash_init(desc
);
561 goto out_free_sha_regions
;
563 digest
= kzalloc(SHA256_DIGEST_SIZE
, GFP_KERNEL
);
566 goto out_free_sha_regions
;
569 for (j
= i
= 0; i
< image
->nr_segments
; i
++) {
570 struct kexec_segment
*ksegment
;
572 ksegment
= &image
->segment
[i
];
574 * Skip purgatory as it will be modified once we put digest
577 if (ksegment
->kbuf
== pi
->purgatory_buf
)
580 ret
= crypto_shash_update(desc
, ksegment
->kbuf
,
586 * Assume rest of the buffer is filled with zero and
587 * update digest accordingly.
589 nullsz
= ksegment
->memsz
- ksegment
->bufsz
;
591 unsigned long bytes
= nullsz
;
593 if (bytes
> zero_buf_sz
)
595 ret
= crypto_shash_update(desc
, zero_buf
, bytes
);
604 sha_regions
[j
].start
= ksegment
->mem
;
605 sha_regions
[j
].len
= ksegment
->memsz
;
610 ret
= crypto_shash_final(desc
, digest
);
612 goto out_free_digest
;
613 ret
= kexec_purgatory_get_set_symbol(image
, "sha_regions",
614 sha_regions
, sha_region_sz
, 0);
616 goto out_free_digest
;
618 ret
= kexec_purgatory_get_set_symbol(image
, "sha256_digest",
619 digest
, SHA256_DIGEST_SIZE
, 0);
621 goto out_free_digest
;
626 out_free_sha_regions
:
636 /* Actually load purgatory. Lot of code taken from kexec-tools */
637 static int __kexec_load_purgatory(struct kimage
*image
, unsigned long min
,
638 unsigned long max
, int top_down
)
640 struct purgatory_info
*pi
= &image
->purgatory_info
;
641 unsigned long align
, bss_align
, bss_sz
, bss_pad
;
642 unsigned long entry
, load_addr
, curr_load_addr
, bss_addr
, offset
;
643 unsigned char *buf_addr
, *src
;
644 int i
, ret
= 0, entry_sidx
= -1;
645 const Elf_Shdr
*sechdrs_c
;
646 Elf_Shdr
*sechdrs
= NULL
;
647 struct kexec_buf kbuf
= { .image
= image
, .bufsz
= 0, .buf_align
= 1,
648 .buf_min
= min
, .buf_max
= max
,
649 .top_down
= top_down
};
652 * sechdrs_c points to section headers in purgatory and are read
653 * only. No modifications allowed.
655 sechdrs_c
= (void *)pi
->ehdr
+ pi
->ehdr
->e_shoff
;
658 * We can not modify sechdrs_c[] and its fields. It is read only.
659 * Copy it over to a local copy where one can store some temporary
660 * data and free it at the end. We need to modify ->sh_addr and
661 * ->sh_offset fields to keep track of permanent and temporary
662 * locations of sections.
664 sechdrs
= vzalloc(pi
->ehdr
->e_shnum
* sizeof(Elf_Shdr
));
668 memcpy(sechdrs
, sechdrs_c
, pi
->ehdr
->e_shnum
* sizeof(Elf_Shdr
));
671 * We seem to have multiple copies of sections. First copy is which
672 * is embedded in kernel in read only section. Some of these sections
673 * will be copied to a temporary buffer and relocated. And these
674 * sections will finally be copied to their final destination at
677 * Use ->sh_offset to reflect section address in memory. It will
678 * point to original read only copy if section is not allocatable.
679 * Otherwise it will point to temporary copy which will be relocated.
681 * Use ->sh_addr to contain final address of the section where it
682 * will go during execution time.
684 for (i
= 0; i
< pi
->ehdr
->e_shnum
; i
++) {
685 if (sechdrs
[i
].sh_type
== SHT_NOBITS
)
688 sechdrs
[i
].sh_offset
= (unsigned long)pi
->ehdr
+
689 sechdrs
[i
].sh_offset
;
693 * Identify entry point section and make entry relative to section
696 entry
= pi
->ehdr
->e_entry
;
697 for (i
= 0; i
< pi
->ehdr
->e_shnum
; i
++) {
698 if (!(sechdrs
[i
].sh_flags
& SHF_ALLOC
))
701 if (!(sechdrs
[i
].sh_flags
& SHF_EXECINSTR
))
704 /* Make entry section relative */
705 if (sechdrs
[i
].sh_addr
<= pi
->ehdr
->e_entry
&&
706 ((sechdrs
[i
].sh_addr
+ sechdrs
[i
].sh_size
) >
707 pi
->ehdr
->e_entry
)) {
709 entry
-= sechdrs
[i
].sh_addr
;
714 /* Determine how much memory is needed to load relocatable object. */
718 for (i
= 0; i
< pi
->ehdr
->e_shnum
; i
++) {
719 if (!(sechdrs
[i
].sh_flags
& SHF_ALLOC
))
722 align
= sechdrs
[i
].sh_addralign
;
723 if (sechdrs
[i
].sh_type
!= SHT_NOBITS
) {
724 if (kbuf
.buf_align
< align
)
725 kbuf
.buf_align
= align
;
726 kbuf
.bufsz
= ALIGN(kbuf
.bufsz
, align
);
727 kbuf
.bufsz
+= sechdrs
[i
].sh_size
;
730 if (bss_align
< align
)
732 bss_sz
= ALIGN(bss_sz
, align
);
733 bss_sz
+= sechdrs
[i
].sh_size
;
737 /* Determine the bss padding required to align bss properly */
739 if (kbuf
.bufsz
& (bss_align
- 1))
740 bss_pad
= bss_align
- (kbuf
.bufsz
& (bss_align
- 1));
742 kbuf
.memsz
= kbuf
.bufsz
+ bss_pad
+ bss_sz
;
744 /* Allocate buffer for purgatory */
745 kbuf
.buffer
= vzalloc(kbuf
.bufsz
);
751 if (kbuf
.buf_align
< bss_align
)
752 kbuf
.buf_align
= bss_align
;
754 /* Add buffer to segment list */
755 ret
= kexec_add_buffer(&kbuf
);
758 pi
->purgatory_load_addr
= kbuf
.mem
;
760 /* Load SHF_ALLOC sections */
761 buf_addr
= kbuf
.buffer
;
762 load_addr
= curr_load_addr
= pi
->purgatory_load_addr
;
763 bss_addr
= load_addr
+ kbuf
.bufsz
+ bss_pad
;
765 for (i
= 0; i
< pi
->ehdr
->e_shnum
; i
++) {
766 if (!(sechdrs
[i
].sh_flags
& SHF_ALLOC
))
769 align
= sechdrs
[i
].sh_addralign
;
770 if (sechdrs
[i
].sh_type
!= SHT_NOBITS
) {
771 curr_load_addr
= ALIGN(curr_load_addr
, align
);
772 offset
= curr_load_addr
- load_addr
;
773 /* We already modifed ->sh_offset to keep src addr */
774 src
= (char *) sechdrs
[i
].sh_offset
;
775 memcpy(buf_addr
+ offset
, src
, sechdrs
[i
].sh_size
);
777 /* Store load address and source address of section */
778 sechdrs
[i
].sh_addr
= curr_load_addr
;
781 * This section got copied to temporary buffer. Update
782 * ->sh_offset accordingly.
784 sechdrs
[i
].sh_offset
= (unsigned long)(buf_addr
+ offset
);
786 /* Advance to the next address */
787 curr_load_addr
+= sechdrs
[i
].sh_size
;
789 bss_addr
= ALIGN(bss_addr
, align
);
790 sechdrs
[i
].sh_addr
= bss_addr
;
791 bss_addr
+= sechdrs
[i
].sh_size
;
795 /* Update entry point based on load address of text section */
797 entry
+= sechdrs
[entry_sidx
].sh_addr
;
799 /* Make kernel jump to purgatory after shutdown */
800 image
->start
= entry
;
802 /* Used later to get/set symbol values */
803 pi
->sechdrs
= sechdrs
;
806 * Used later to identify which section is purgatory and skip it
809 pi
->purgatory_buf
= kbuf
.buffer
;
817 static int kexec_apply_relocations(struct kimage
*image
)
820 struct purgatory_info
*pi
= &image
->purgatory_info
;
821 Elf_Shdr
*sechdrs
= pi
->sechdrs
;
823 /* Apply relocations */
824 for (i
= 0; i
< pi
->ehdr
->e_shnum
; i
++) {
825 Elf_Shdr
*section
, *symtab
;
827 if (sechdrs
[i
].sh_type
!= SHT_RELA
&&
828 sechdrs
[i
].sh_type
!= SHT_REL
)
832 * For section of type SHT_RELA/SHT_REL,
833 * ->sh_link contains section header index of associated
834 * symbol table. And ->sh_info contains section header
835 * index of section to which relocations apply.
837 if (sechdrs
[i
].sh_info
>= pi
->ehdr
->e_shnum
||
838 sechdrs
[i
].sh_link
>= pi
->ehdr
->e_shnum
)
841 section
= &sechdrs
[sechdrs
[i
].sh_info
];
842 symtab
= &sechdrs
[sechdrs
[i
].sh_link
];
844 if (!(section
->sh_flags
& SHF_ALLOC
))
848 * symtab->sh_link contain section header index of associated
851 if (symtab
->sh_link
>= pi
->ehdr
->e_shnum
)
852 /* Invalid section number? */
856 * Respective architecture needs to provide support for applying
857 * relocations of type SHT_RELA/SHT_REL.
859 if (sechdrs
[i
].sh_type
== SHT_RELA
)
860 ret
= arch_kexec_apply_relocations_add(pi
->ehdr
,
862 else if (sechdrs
[i
].sh_type
== SHT_REL
)
863 ret
= arch_kexec_apply_relocations(pi
->ehdr
,
872 /* Load relocatable purgatory object and relocate it appropriately */
873 int kexec_load_purgatory(struct kimage
*image
, unsigned long min
,
874 unsigned long max
, int top_down
,
875 unsigned long *load_addr
)
877 struct purgatory_info
*pi
= &image
->purgatory_info
;
880 if (kexec_purgatory_size
<= 0)
883 if (kexec_purgatory_size
< sizeof(Elf_Ehdr
))
886 pi
->ehdr
= (Elf_Ehdr
*)kexec_purgatory
;
888 if (memcmp(pi
->ehdr
->e_ident
, ELFMAG
, SELFMAG
) != 0
889 || pi
->ehdr
->e_type
!= ET_REL
890 || !elf_check_arch(pi
->ehdr
)
891 || pi
->ehdr
->e_shentsize
!= sizeof(Elf_Shdr
))
894 if (pi
->ehdr
->e_shoff
>= kexec_purgatory_size
895 || (pi
->ehdr
->e_shnum
* sizeof(Elf_Shdr
) >
896 kexec_purgatory_size
- pi
->ehdr
->e_shoff
))
899 ret
= __kexec_load_purgatory(image
, min
, max
, top_down
);
903 ret
= kexec_apply_relocations(image
);
907 *load_addr
= pi
->purgatory_load_addr
;
913 vfree(pi
->purgatory_buf
);
914 pi
->purgatory_buf
= NULL
;
918 static Elf_Sym
*kexec_purgatory_find_symbol(struct purgatory_info
*pi
,
927 if (!pi
->sechdrs
|| !pi
->ehdr
)
930 sechdrs
= pi
->sechdrs
;
933 for (i
= 0; i
< ehdr
->e_shnum
; i
++) {
934 if (sechdrs
[i
].sh_type
!= SHT_SYMTAB
)
937 if (sechdrs
[i
].sh_link
>= ehdr
->e_shnum
)
938 /* Invalid strtab section number */
940 strtab
= (char *)sechdrs
[sechdrs
[i
].sh_link
].sh_offset
;
941 syms
= (Elf_Sym
*)sechdrs
[i
].sh_offset
;
943 /* Go through symbols for a match */
944 for (k
= 0; k
< sechdrs
[i
].sh_size
/sizeof(Elf_Sym
); k
++) {
945 if (ELF_ST_BIND(syms
[k
].st_info
) != STB_GLOBAL
)
948 if (strcmp(strtab
+ syms
[k
].st_name
, name
) != 0)
951 if (syms
[k
].st_shndx
== SHN_UNDEF
||
952 syms
[k
].st_shndx
>= ehdr
->e_shnum
) {
953 pr_debug("Symbol: %s has bad section index %d.\n",
954 name
, syms
[k
].st_shndx
);
958 /* Found the symbol we are looking for */
966 void *kexec_purgatory_get_symbol_addr(struct kimage
*image
, const char *name
)
968 struct purgatory_info
*pi
= &image
->purgatory_info
;
972 sym
= kexec_purgatory_find_symbol(pi
, name
);
974 return ERR_PTR(-EINVAL
);
976 sechdr
= &pi
->sechdrs
[sym
->st_shndx
];
979 * Returns the address where symbol will finally be loaded after
980 * kexec_load_segment()
982 return (void *)(sechdr
->sh_addr
+ sym
->st_value
);
986 * Get or set value of a symbol. If "get_value" is true, symbol value is
987 * returned in buf otherwise symbol value is set based on value in buf.
989 int kexec_purgatory_get_set_symbol(struct kimage
*image
, const char *name
,
990 void *buf
, unsigned int size
, bool get_value
)
994 struct purgatory_info
*pi
= &image
->purgatory_info
;
997 sym
= kexec_purgatory_find_symbol(pi
, name
);
1001 if (sym
->st_size
!= size
) {
1002 pr_err("symbol %s size mismatch: expected %lu actual %u\n",
1003 name
, (unsigned long)sym
->st_size
, size
);
1007 sechdrs
= pi
->sechdrs
;
1009 if (sechdrs
[sym
->st_shndx
].sh_type
== SHT_NOBITS
) {
1010 pr_err("symbol %s is in a bss section. Cannot %s\n", name
,
1011 get_value
? "get" : "set");
1015 sym_buf
= (unsigned char *)sechdrs
[sym
->st_shndx
].sh_offset
+
1019 memcpy((void *)buf
, sym_buf
, size
);
1021 memcpy((void *)sym_buf
, buf
, size
);