1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * Copyright (C) 2010 "Wu Zhangjin" <wuzhangjin@gmail.com>
4  */
5 
6 #include <sys/types.h>
7 #include <sys/stat.h>
8 #include <errno.h>
9 #include <stdint.h>
10 #include <stdio.h>
11 #include <stdlib.h>
12 #include <linux/sizes.h>
13 
14 int main(int argc, char *argv[])
15 {
16 	unsigned long long vmlinux_size, vmlinux_load_addr, vmlinuz_load_addr;
17 	struct stat sb;
18 
19 	if (argc != 3) {
20 		fprintf(stderr, "Usage: %s <pathname> <vmlinux_load_addr>\n",
21 				argv[0]);
22 		return EXIT_FAILURE;
23 	}
24 
25 	if (stat(argv[1], &sb) == -1) {
26 		perror("stat");
27 		return EXIT_FAILURE;
28 	}
29 
30 	/* Convert hex characters to dec number */
31 	errno = 0;
32 	if (sscanf(argv[2], "%llx", &vmlinux_load_addr) != 1) {
33 		if (errno != 0)
34 			perror("sscanf");
35 		else
36 			fprintf(stderr, "No matching characters\n");
37 
38 		return EXIT_FAILURE;
39 	}
40 
41 	vmlinux_size = (uint64_t)sb.st_size;
42 	vmlinuz_load_addr = vmlinux_load_addr + vmlinux_size;
43 
44 	/*
45 	 * Align with 64KB: KEXEC needs load sections to be aligned to PAGE_SIZE,
46 	 * which may be as large as 64KB depending on the kernel configuration.
47 	 */
48 
49 	vmlinuz_load_addr += (SZ_64K - vmlinux_size % SZ_64K);
50 
51 	printf("0x%llx\n", vmlinuz_load_addr);
52 
53 	return EXIT_SUCCESS;
54 }
55