wolfcrypt-jni/jni/jni_hmac.c

472 lines
12 KiB
C

/* jni_hmac.c
*
* Copyright (C) 2006-2025 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
*/
#include <stdint.h>
#ifdef WOLFSSL_USER_SETTINGS
#include <wolfssl/wolfcrypt/settings.h>
#elif !defined(__ANDROID__)
#include <wolfssl/options.h>
#endif
#include <wolfssl/wolfcrypt/hmac.h>
#include <com_wolfssl_wolfcrypt_Hmac.h>
#include <wolfcrypt_jni_NativeStruct.h>
#include <wolfcrypt_jni_error.h>
/* #define WOLFCRYPT_JNI_DEBUG_ON */
#include <wolfcrypt_jni_debug.h>
#if (HAVE_FIPS_VERSION >= 2)
#define MAX_DIGEST_SIZE WC_MAX_DIGEST_SIZE
#endif
#ifdef NO_OLD_WC_NAMES
#ifndef NO_MD5
#define MD5_DIGEST_SIZE WC_MD5_DIGEST_SIZE
#endif
#ifndef NO_SHA
#define SHA_DIGEST_SIZE WC_SHA_DIGEST_SIZE
#endif
#ifdef WOLFSSL_SHA224
#define SHA224_DIGEST_SIZE WC_SHA224_DIGEST_SIZE
#endif
#ifndef NO_SHA256
#define SHA256_DIGEST_SIZE WC_SHA256_DIGEST_SIZE
#endif
#ifdef WOLFSSL_SHA384
#define SHA384_DIGEST_SIZE WC_SHA384_DIGEST_SIZE
#endif
#ifdef WOLFSSL_SHA512
#define SHA512_DIGEST_SIZE WC_SHA512_DIGEST_SIZE
#endif
#define MAX_DIGEST_SIZE WC_MAX_DIGEST_SIZE
#endif
/* copy from cyassl/hmac.c */
static WC_INLINE int GetHashSizeByType(int type)
{
if (!(type == WC_MD5 || type == WC_SHA || type == WC_SHA224
|| type == WC_SHA256 || type == WC_SHA384 || type == WC_SHA512
|| type == WC_SHA3_224 || type == WC_SHA3_256 || type == WC_SHA3_384
|| type == WC_SHA3_512)) {
return BAD_FUNC_ARG;
}
switch (type) {
#ifndef NO_MD5
case WC_MD5:
return MD5_DIGEST_SIZE;
#endif
#ifndef NO_SHA
case WC_SHA:
return SHA_DIGEST_SIZE;
#endif
#ifdef WOLFSSL_SHA224
case WC_SHA224:
return SHA224_DIGEST_SIZE;
#endif
#ifndef NO_SHA256
case WC_SHA256:
return SHA256_DIGEST_SIZE;
#endif
#if defined(CYASSL_SHA384) || defined(WOLFSSL_SHA384)
case WC_SHA384:
return SHA384_DIGEST_SIZE;
#endif
#if defined(CYASSL_SHA512) || defined(WOLFSSL_SHA512)
case WC_SHA512:
return SHA512_DIGEST_SIZE;
#endif
#if defined(WOLFSSL_SHA3)
case WC_SHA3_224:
return WC_SHA3_224_DIGEST_SIZE;
case WC_SHA3_256:
return WC_SHA3_256_DIGEST_SIZE;
case WC_SHA3_384:
return WC_SHA3_384_DIGEST_SIZE;
case WC_SHA3_512:
return WC_SHA3_512_DIGEST_SIZE;
#endif
default:
return BAD_FUNC_ARG;
}
}
JNIEXPORT jlong JNICALL Java_com_wolfssl_wolfcrypt_Hmac_mallocNativeStruct
(JNIEnv* env, jobject this)
{
#ifndef NO_HMAC
Hmac* hmac = NULL;
hmac = (Hmac*) XMALLOC(sizeof(Hmac), NULL, DYNAMIC_TYPE_TMP_BUFFER);
if (hmac == NULL) {
throwOutOfMemoryException(env, "Failed to allocate Hmac object");
}
else {
XMEMSET(hmac, 0, sizeof(Hmac));
}
LogStr("new Hmac() = %p\n", hmac);
return (jlong)(uintptr_t)hmac;
#else
throwNotCompiledInException(env);
return (jlong)0;
#endif
}
JNIEXPORT void JNICALL Java_com_wolfssl_wolfcrypt_Hmac_wc_1HmacSetKey
(JNIEnv* env, jobject this, jint type, jbyteArray key_object)
{
#ifndef NO_HMAC
int ret = 0;
Hmac* hmac = NULL;
byte* key = NULL;
word32 keySz = 0;
hmac = (Hmac*) getNativeStruct(env, this);
if ((*env)->ExceptionOccurred(env)) {
/* getNativeStruct may throw exception, prevent throwing another */
return;
}
key = getByteArray(env, key_object);
keySz = getByteArrayLength(env, key_object);
ret = (!hmac || !key)
? BAD_FUNC_ARG
: wc_HmacSetKey(hmac, type, key, keySz);
if (ret != 0)
throwWolfCryptExceptionFromError(env, ret);
LogStr("HmacInit(hmac=%p) = %d\n", hmac, ret);
releaseByteArray(env, key_object, key, JNI_ABORT);
#else
throwNotCompiledInException(env);
#endif
}
JNIEXPORT void JNICALL Java_com_wolfssl_wolfcrypt_Hmac_wc_1HmacUpdate__B
(JNIEnv* env, jobject this, jbyte data)
{
#ifndef NO_HMAC
int ret = 0;
Hmac* hmac = (Hmac*) getNativeStruct(env, this);
if ((*env)->ExceptionOccurred(env)) {
/* getNativeStruct may throw exception, prevent throwing another */
return;
}
ret = (!hmac)
? BAD_FUNC_ARG
: wc_HmacUpdate(hmac, (const byte*)&data, 1);
if (ret != 0)
throwWolfCryptExceptionFromError(env, ret);
LogStr("wc_HmacUpdate(hmac=%p, data, 1) = %d\n", hmac, ret);
LogStr("data: %02x\n", data);
#else
throwNotCompiledInException(env);
#endif
}
JNIEXPORT void JNICALL Java_com_wolfssl_wolfcrypt_Hmac_wc_1HmacUpdate___3BII
(JNIEnv* env, jobject this, jbyteArray data_object, jint offset, jint length)
{
#ifndef NO_HMAC
int ret = 0;
Hmac* hmac = NULL;
byte* data = NULL;
hmac = (Hmac*) getNativeStruct(env, this);
if ((*env)->ExceptionOccurred(env)) {
/* getNativeStruct may throw exception, prevent throwing another */
return;
}
data = getByteArray(env, data_object);
ret = (!hmac || !data)
? BAD_FUNC_ARG
: wc_HmacUpdate(hmac, data + offset, length);
if (ret != 0)
throwWolfCryptExceptionFromError(env, ret);
LogStr("wc_HmacUpdate(hmac=%p, data, length) = %d\n", hmac, ret);
LogStr("data[%u]: [%p]\n", (word32)length, data + offset);
LogHex((byte*) data, offset, length);
releaseByteArray(env, data_object, data, JNI_ABORT);
#else
throwNotCompiledInException(env);
#endif
}
JNIEXPORT void JNICALL Java_com_wolfssl_wolfcrypt_Hmac_wc_1HmacUpdate__Ljava_nio_ByteBuffer_2II
(JNIEnv* env, jobject this, jobject data_object, jint offset, jint length)
{
#ifndef NO_HMAC
int ret = 0;
Hmac* hmac = NULL;
byte* data = NULL;
hmac = (Hmac*) getNativeStruct(env, this);
if ((*env)->ExceptionOccurred(env)) {
/* getNativeStruct may throw exception, prevent throwing another */
return;
}
data = getDirectBufferAddress(env, data_object);
ret = (!hmac || !data)
? BAD_FUNC_ARG
: wc_HmacUpdate(hmac, data + offset, length);
if (ret != 0)
throwWolfCryptExceptionFromError(env, ret);
LogStr("wc_HmacUpdate(hmac=%p, data, length) = %d\n", hmac, ret);
LogStr("data[%u]: [%p]\n", (word32)length, data + offset);
LogHex((byte*) data, offset, length);
#else
throwNotCompiledInException(env);
#endif
}
JNIEXPORT jbyteArray JNICALL Java_com_wolfssl_wolfcrypt_Hmac_wc_1HmacFinal
(JNIEnv* env, jobject this)
{
jbyteArray result = NULL;
#ifndef NO_HMAC
int ret = 0;
Hmac* hmac = NULL;
int hmacSz = 0;
byte tmp[MAX_DIGEST_SIZE];
hmac = (Hmac*) getNativeStruct(env, this);
if ((*env)->ExceptionOccurred(env)) {
/* getNativeStruct may throw exception, prevent throwing another */
return NULL;
}
hmacSz = GetHashSizeByType(hmac->macType);
if (hmacSz < 0) {
throwWolfCryptExceptionFromError(env, ret);
return result;
}
ret = (!hmac)
? BAD_FUNC_ARG
: wc_HmacFinal(hmac, tmp);
if (ret == 0) {
result = (*env)->NewByteArray(env, hmacSz);
if (result) {
(*env)->SetByteArrayRegion(env, result, 0, hmacSz,
(const jbyte*) tmp);
} else {
throwWolfCryptException(env, "Failed to allocate hmac");
}
} else {
throwWolfCryptExceptionFromError(env, ret);
}
LogStr("wc_HmacFinal(hmac=%p, result) = %d\n", hmac, ret);
LogStr("result[%u]: [%p]\n", (word32)hmacSz, tmp);
LogHex(tmp, 0, hmacSz);
#else
throwNotCompiledInException(env);
#endif
return result;
}
JNIEXPORT jint JNICALL Java_com_wolfssl_wolfcrypt_Hmac_wc_1HmacSizeByType
(JNIEnv* env, jobject this, jint type)
{
jint result = 0;
#ifndef NO_HMAC
int ret = GetHashSizeByType(type);
if (ret < 0)
throwWolfCryptExceptionFromError(env, ret);
else
result = ret;
LogStr("wc_HmacSizeByType(type=%d) = %d\n", type, ret);
#else
throwNotCompiledInException(env);
#endif
return result;
}
JNIEXPORT jint JNICALL Java_com_wolfssl_wolfcrypt_Hmac_getCodeMd5
(JNIEnv* env, jobject this)
{
#ifndef NO_MD5
jint result = WC_MD5;
LogStr("WC_MD5 = %d\n", result);
return result;
#else
/* not compiled in */
return (jint) -1;
#endif
}
JNIEXPORT jint JNICALL Java_com_wolfssl_wolfcrypt_Hmac_getCodeSha
(JNIEnv* env, jobject this)
{
#ifndef NO_SHA
jint result = WC_SHA;
LogStr("WC_SHA = %d\n", result);
return result;
#else
/* not compiled in */
return (jint) -1;
#endif
}
JNIEXPORT jint JNICALL Java_com_wolfssl_wolfcrypt_Hmac_getCodeSha224
(JNIEnv* env, jobject this)
{
#ifdef WOLFSSL_SHA224
jint result = WC_SHA224;
LogStr("WC_SHA224 = %d\n", result);
return result;
#else
/* not compiled in */
return (jint) -1;
#endif
}
JNIEXPORT jint JNICALL Java_com_wolfssl_wolfcrypt_Hmac_getCodeSha256
(JNIEnv* env, jobject this)
{
#ifndef NO_SHA256
jint result = WC_SHA256;
LogStr("WC_SHA256 = %d\n", result);
return result;
#else
/* not compiled in */
return (jint) -1;
#endif
}
JNIEXPORT jint JNICALL Java_com_wolfssl_wolfcrypt_Hmac_getCodeSha384
(JNIEnv* env, jobject this)
{
#ifdef WOLFSSL_SHA384
jint result = WC_SHA384;
LogStr("WC_SHA384 = %d\n", result);
return result;
#else
/* not compiled in */
return (jint) -1;
#endif
}
JNIEXPORT jint JNICALL Java_com_wolfssl_wolfcrypt_Hmac_getCodeSha512
(JNIEnv* env, jobject this)
{
#ifdef WOLFSSL_SHA512
jint result = WC_SHA512;
LogStr("WC_SHA512 = %d\n", result);
return result;
#else
/* not compiled in */
return (jint) -1;
#endif
}
JNIEXPORT jint JNICALL Java_com_wolfssl_wolfcrypt_Hmac_getCodeSha3_1224
(JNIEnv* env, jclass this)
{
#ifdef WOLFSSL_SHA3
jint result = WC_SHA3_224;
LogStr("WC_SHA3_224 = %d\n", result);
return result;
#else
/* not compiled in */
return (jint) -1;
#endif
}
JNIEXPORT jint JNICALL Java_com_wolfssl_wolfcrypt_Hmac_getCodeSha3_1256
(JNIEnv* env, jclass this)
{
#ifdef WOLFSSL_SHA3
jint result = WC_SHA3_256;
LogStr("WC_SHA3_256 = %d\n", result);
return result;
#else
/* not compiled in */
return (jint) -1;
#endif
}
JNIEXPORT jint JNICALL Java_com_wolfssl_wolfcrypt_Hmac_getCodeSha3_1384
(JNIEnv* env, jclass this)
{
#ifdef WOLFSSL_SHA3
jint result = WC_SHA3_384;
LogStr("WC_SHA3_384 = %d\n", result);
return result;
#else
/* not compiled in */
return (jint) -1;
#endif
}
JNIEXPORT jint JNICALL Java_com_wolfssl_wolfcrypt_Hmac_getCodeSha3_1512
(JNIEnv* env, jclass this)
{
#ifdef WOLFSSL_SHA3
jint result = WC_SHA3_512;
LogStr("WC_SHA3_512 = %d\n", result);
return result;
#else
/* not compiled in */
return (jint) -1;
#endif
}