// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright (C) 2010 "Wu Zhangjin" <wuzhangjin@gmail.com>
 */

#include <sys/types.h>
#include <sys/stat.h>
#include <errno.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <linux/sizes.h>

int main(int argc, char *argv[])
{
        unsigned long long vmlinux_size, vmlinux_load_addr, vmlinuz_load_addr;
        struct stat sb;

        if (argc != 3) {
                fprintf(stderr, "Usage: %s <pathname> <vmlinux_load_addr>\n",
                                argv[0]);
                return EXIT_FAILURE;
        }

        if (stat(argv[1], &sb) == -1) {
                perror("stat");
                return EXIT_FAILURE;
        }

        /* Convert hex characters to dec number */
        errno = 0;
        if (sscanf(argv[2], "%llx", &vmlinux_load_addr) != 1) {
                if (errno != 0)
                        perror("sscanf");
                else
                        fprintf(stderr, "No matching characters\n");

                return EXIT_FAILURE;
        }

        vmlinux_size = (uint64_t)sb.st_size;
        vmlinuz_load_addr = vmlinux_load_addr + vmlinux_size;

        /*
         * Align with 64KB: KEXEC needs load sections to be aligned to PAGE_SIZE,
         * which may be as large as 64KB depending on the kernel configuration.
         */

        vmlinuz_load_addr += (SZ_64K - vmlinux_size % SZ_64K);

        printf("0x%llx\n", vmlinuz_load_addr);

        return EXIT_SUCCESS;
}