// SPDX-License-Identifier: GPL-2.0

#include <linux/efi.h>
#include <asm/efi.h>
#include <asm/string.h>

#include "efistub.h"

#ifdef CONFIG_KASAN
#undef memcpy
#undef memmove
#undef memset
void *__memcpy(void *__dest, const void *__src, size_t __n) __alias(memcpy);
void *__memmove(void *__dest, const void *__src, size_t count) __alias(memmove);
void *__memset(void *s, int c, size_t count) __alias(memset);
#endif

static void *efistub_memmove(u8 *dst, const u8 *src, size_t len)
{
 if (src > dst || dst >= (src + len))
  for (size_t i = 0; i < len; i++)
   dst[i] = src[i];
 else
  for (ssize_t i = len - 1; i >= 0; i--)
   dst[i] = src[i];

 return dst;
}

static void *efistub_memset(void *dst, int c, size_t len)
{
 for (u8 *d = dst; len--; d++)
  *d = c;

 return dst;
}

void *memcpy(void *dst, const void *src, size_t len)
{
 if (efi_table_attr(efi_system_table, boottime) == NULL)
  return efistub_memmove(dst, src, len);

 efi_bs_call(copy_mem, dst, src, len);
 return dst;
}

extern void *memmove(void *dst, const void *src, size_t len) __alias(memcpy);

void *memset(void *dst, int c, size_t len)
{
 if (efi_table_attr(efi_system_table, boottime) == NULL)
  return efistub_memset(dst, c, len);

 efi_bs_call(set_mem, dst, len, c & U8_MAX);
 return dst;
}

/**
 * memcmp - Compare two areas of memory
 * @cs: One area of memory
 * @ct: Another area of memory
 * @count: The size of the area.
 */
#undef memcmp
int memcmp(const void *cs, const void *ct, size_t count)
{
 const unsigned char *su1, *su2;
 int res = 0;

 for (su1 = cs, su2 = ct; 0 < count; ++su1, ++su2, count--)
  if ((res = *su1 - *su2) != 0)
   break;
 return res;
}