// SPDX-License-Identifier: GPL-2.0

#include <linux/efi.h>
#include <linux/zstd.h>

#include <asm/efi.h>

#include "decompress_sources.h"
#include "efistub.h"

extern unsigned char _gzdata_start[], _gzdata_end[];
extern u32 __aligned(1) payload_size;

static size_t wksp_size;
static void *wksp;

efi_status_t efi_zboot_decompress_init(unsigned long *alloc_size)
{
        efi_status_t status;

        wksp_size = zstd_dctx_workspace_bound();
        status = efi_allocate_pages(wksp_size, (unsigned long *)&wksp, ULONG_MAX);
        if (status != EFI_SUCCESS)
                return status;

        *alloc_size = payload_size;
        return EFI_SUCCESS;
}

efi_status_t efi_zboot_decompress(u8 *out, unsigned long outlen)
{
        zstd_dctx *dctx = zstd_init_dctx(wksp, wksp_size);
        size_t ret;
        int retval;

        ret = zstd_decompress_dctx(dctx, out, outlen, _gzdata_start,
                                   _gzdata_end - _gzdata_start - 4);
        efi_free(wksp_size, (unsigned long)wksp);

        retval = zstd_get_error_code(ret);
        if (retval) {
                efi_err("ZSTD-decompression failed with status %d\n", retval);
                return EFI_LOAD_ERROR;
        }

        efi_cache_sync_image((unsigned long)out, outlen);

        return EFI_SUCCESS;
}