/* hash.h
 *
 * Copyright (C) 2006-2017 wolfSSL Inc.
 *
 * This file is part of wolfSSL.
 *
 * wolfSSL is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * wolfSSL is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA
 */


#ifndef WOLF_CRYPT_HASH_H
# define WOLF_CRYPT_HASH_H

# include "types.h"

# ifndef NO_MD5
#  include "md5.h"
# endif
# ifndef NO_SHA
#  include "sha.h"
# endif
# if defined(WOLFSSL_SHA224) || !defined(NO_SHA256)
#  include "sha256.h"
# endif
# if defined(WOLFSSL_SHA384) || defined(WOLFSSL_SHA512)
#  include "sha512.h"
# endif

# ifdef __cplusplus
extern "C" {
# endif

/* Hash types */
enum wc_HashType {
    WC_HASH_TYPE_NONE    = 0,
    WC_HASH_TYPE_MD2     = 1,
    WC_HASH_TYPE_MD4     = 2,
    WC_HASH_TYPE_MD5     = 3,
    WC_HASH_TYPE_SHA     = 4, /* SHA-1 (not old SHA-0) */
    WC_HASH_TYPE_SHA224  = 9,
    WC_HASH_TYPE_SHA256  = 5,
    WC_HASH_TYPE_SHA384  = 6,
    WC_HASH_TYPE_SHA512  = 7,
    WC_HASH_TYPE_MD5_SHA = 8,
};

typedef union {
    # ifndef NO_MD5
    wc_Md5    md5;
    # endif
    # ifndef NO_SHA
    wc_Sha    sha;
    # endif
    # ifdef WOLFSSL_SHA224
    wc_Sha224 sha224;
    # endif
    # ifndef NO_SHA256
    wc_Sha256 sha256;
    # endif
    # ifdef WOLFSSL_SHA384
    wc_Sha384 sha384;
    # endif
    # ifdef WOLFSSL_SHA512
    wc_Sha512 sha512;
    # endif
} wc_HashAlg;

/* Find largest possible digest size
 * Note if this gets up to the size of 80 or over check smallstack build */
# if defined(WOLFSSL_SHA512)
#  define WC_MAX_DIGEST_SIZE WC_SHA512_DIGEST_SIZE
# elif defined(WOLFSSL_SHA384)
#  define WC_MAX_DIGEST_SIZE WC_SHA384_DIGEST_SIZE
# elif !defined(NO_SHA256)
#  define WC_MAX_DIGEST_SIZE WC_SHA256_DIGEST_SIZE
# elif defined(WOLFSSL_SHA224)
#  define WC_MAX_DIGEST_SIZE WC_SHA224_DIGEST_SIZE
# elif !defined(NO_SHA)
#  define WC_MAX_DIGEST_SIZE WC_SHA_DIGEST_SIZE
# elif !defined(NO_MD5)
#  define WC_MAX_DIGEST_SIZE WC_MD5_DIGEST_SIZE
# else // if defined(WOLFSSL_SHA512)
#  define WC_MAX_DIGEST_SIZE 64 /* default to max size of 64 */
# endif // if defined(WOLFSSL_SHA512)

# if !defined(NO_ASN) || !defined(NO_DH) || defined(HAVE_ECC)
WOLFSSL_API int
wc_HashGetOID(enum wc_HashType hash_type);
# endif

WOLFSSL_API int
wc_HashGetDigestSize(enum wc_HashType hash_type);
WOLFSSL_API int
wc_Hash(enum wc_HashType hash_type,
  const byte * data, word32 data_len,
  byte * hash, word32 hash_len);

/* generic hash operation wrappers */
WOLFSSL_API int
wc_HashInit(wc_HashAlg * hash, enum wc_HashType type);
WOLFSSL_API int
wc_HashUpdate(wc_HashAlg * hash, enum wc_HashType type,
  const byte * data, word32 dataSz);
WOLFSSL_API int
wc_HashFinal(wc_HashAlg * hash, enum wc_HashType type,
  byte * out);


# ifndef NO_MD5
#  include "md5.h"
WOLFSSL_API int
wc_Md5Hash(const byte * data, word32 len, byte * hash);
# endif

# ifndef NO_SHA
#  include "sha.h"
WOLFSSL_API int wc_ShaHash(const byte *, word32, byte *);
# endif

# ifndef NO_SHA256
#  include "sha256.h"
WOLFSSL_API int wc_Sha256Hash(const byte *, word32, byte *);

#  if defined(WOLFSSL_SHA224)
WOLFSSL_API int wc_Sha224Hash(const byte *, word32, byte *);
#  endif /* defined(WOLFSSL_SHA224) */
# endif

# ifdef WOLFSSL_SHA512
#  include <wolfssl/wolfcrypt/sha512.h>
WOLFSSL_API int wc_Sha512Hash(const byte *, word32, byte *);

#  if defined(WOLFSSL_SHA384)
WOLFSSL_API int wc_Sha384Hash(const byte *, word32, byte *);
#  endif /* defined(WOLFSSL_SHA384) */
# endif /* WOLFSSL_SHA512 */

# ifdef __cplusplus
} /* extern "C" */
# endif

#endif /* WOLF_CRYPT_HASH_H */
