#ifndef SPX_FORS_H
#define SPX_FORS_H

#include <Library/UefiLib.h>
#include <Library/BaseLib.h>
#include <Library/UefiBootServicesTableLib.h>
#include <Library/BaseMemoryLib.h>
#include <Uefi.h>

#include "params.h"
#include "context.h"

/**
 * Signs a message m, deriving the secret key from sk_seed and the FTS address.
 * Assumes m contains at least SPX_FORS_HEIGHT * SPX_FORS_TREES bits.
 */
#define fors_sign SPX_NAMESPACE(fors_sign)
VOID fors_sign(UINT8 *sig, UINT8 *pk,
               const UINT8 *m,
               const spx_ctx* ctx,
               const UINT32 fors_addr[8]);

/**
 * Derives the FORS public key from a signature.
 * This can be used for verification by comparing to a known public key, or to
 * subsequently verify a signature on the derived public key. The latter is the
 * typical use-case when used as an FTS below an OTS in a hypertree.
 * Assumes m contains at least SPX_FORS_HEIGHT * SPX_FORS_TREES bits.
 */
#define fors_pk_from_sig SPX_NAMESPACE(fors_pk_from_sig)
VOID fors_pk_from_sig(UINT8 *pk,
                      const UINT8 *sig, const UINT8 *m,
                      const spx_ctx* ctx,
                      const UINT32 fors_addr[8]);

#endif
