// 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;
}