wolfssljni/native/com_wolfssl_WolfSSL.c

368 lines
9.8 KiB
C

/* com_wolfssl_WolfSSL.c
*
* Copyright (C) 2006-2013 wolfSSL Inc.
*
* This file is part of CyaSSL.
*
* CyaSSL 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.
*
* CyaSSL 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
*/
#include <stdio.h>
#include <cyassl/ssl.h>
#include <cyassl/error.h>
#include <cyassl/ctaocrypt/logging.h>
#include <cyassl/ctaocrypt/hmac.h>
#include "com_wolfssl_globals.h"
#include "com_wolfssl_WolfSSL.h"
/* global object refs for logging callbacks */
static jobject g_loggingCbIfaceObj;
/* custom native fn prototypes */
void NativeLoggingCallback(const int logLevel, const char *const logMessage);
/* called when native library is loaded */
jint JNI_OnLoad(JavaVM* vm, void* reserved)
{
/* store JavaVM */
g_vm = vm;
return JNI_VERSION_1_6;
}
JNIEXPORT jint JNICALL Java_com_wolfssl_WolfSSL_init
(JNIEnv* jenv, jobject jcl)
{
return (jint)CyaSSL_Init();
}
/* used in unit tests */
JNIEXPORT void JNICALL Java_com_wolfssl_WolfSSL_nativeFree
(JNIEnv* jenv, jobject jcl, jlong ptr)
{
if((void*)ptr)
free((void*)ptr);
}
JNIEXPORT jlong JNICALL Java_com_wolfssl_WolfSSL_SSLv3_1ServerMethod
(JNIEnv* jenv, jclass jcl)
{
return (jlong)CyaSSLv3_server_method();
}
JNIEXPORT jlong JNICALL Java_com_wolfssl_WolfSSL_SSLv3_1ClientMethod
(JNIEnv* jenv, jclass jcl)
{
return (jlong)CyaSSLv3_client_method();
}
JNIEXPORT jlong JNICALL Java_com_wolfssl_WolfSSL_TLSv1_1ServerMethod
(JNIEnv* jenv, jclass jcl)
{
return (jlong)CyaTLSv1_server_method();
}
JNIEXPORT jlong JNICALL Java_com_wolfssl_WolfSSL_TLSv1_1ClientMethod
(JNIEnv* jenv, jclass jcl)
{
return (jlong)CyaTLSv1_client_method();
}
JNIEXPORT jlong JNICALL Java_com_wolfssl_WolfSSL_TLSv1_11_1ServerMethod
(JNIEnv* jenv, jclass jcl)
{
return (jlong)CyaTLSv1_1_server_method();
}
JNIEXPORT jlong JNICALL Java_com_wolfssl_WolfSSL_TLSv1_11_1ClientMethod
(JNIEnv* jenv, jclass jcl)
{
return (jlong)CyaTLSv1_1_client_method();
}
JNIEXPORT jlong JNICALL Java_com_wolfssl_WolfSSL_TLSv1_12_1ServerMethod
(JNIEnv* jenv, jclass jcl)
{
return (jlong)CyaTLSv1_2_server_method();
}
JNIEXPORT jlong JNICALL Java_com_wolfssl_WolfSSL_TLSv1_12_1ClientMethod(
JNIEnv* jenv, jclass jcl)
{
return (jlong)CyaTLSv1_2_client_method();
}
JNIEXPORT jlong JNICALL Java_com_wolfssl_WolfSSL_DTLSv1_1ClientMethod
(JNIEnv* jenv, jclass jcl)
{
return (jlong)CyaDTLSv1_client_method();
}
JNIEXPORT jlong JNICALL Java_com_wolfssl_WolfSSL_DTLSv1_1ServerMethod
(JNIEnv* jenv, jclass jcl)
{
return (jlong)CyaDTLSv1_server_method();
}
JNIEXPORT jlong JNICALL Java_com_wolfssl_WolfSSL_DTLSv1_12_1ClientMethod
(JNIEnv* jenv, jclass jcl)
{
return (jlong)CyaDTLSv1_2_client_method();
}
JNIEXPORT jlong JNICALL Java_com_wolfssl_WolfSSL_DTLSv1_12_1ServerMethod
(JNIEnv* jenv, jclass jcl)
{
return (jlong)CyaDTLSv1_2_server_method();
}
JNIEXPORT jlong JNICALL Java_com_wolfssl_WolfSSL_SSLv23_1ServerMethod
(JNIEnv* jenv, jclass jcl)
{
return (jlong)CyaSSLv23_server_method();
}
JNIEXPORT jlong JNICALL Java_com_wolfssl_WolfSSL_SSLv23_1ClientMethod
(JNIEnv* jenv, jclass jcl)
{
return (jlong)CyaSSLv23_client_method();
}
JNIEXPORT jstring JNICALL Java_com_wolfssl_WolfSSL_getErrorString
(JNIEnv* jenv, jclass jcl, jlong errNumber)
{
char buffer[80];
jstring retString;
CyaSSL_ERR_error_string(errNumber, buffer);
retString = (*jenv)->NewStringUTF(jenv, buffer);
return retString;
}
JNIEXPORT jint JNICALL Java_com_wolfssl_WolfSSL_cleanup
(JNIEnv* jenv, jclass jcl)
{
return CyaSSL_Cleanup();
}
JNIEXPORT jint JNICALL Java_com_wolfssl_WolfSSL_debuggingON
(JNIEnv* jenv, jclass jcl)
{
return CyaSSL_Debugging_ON();
}
JNIEXPORT void JNICALL Java_com_wolfssl_WolfSSL_debuggingOFF
(JNIEnv* jenv, jclass jcl)
{
return CyaSSL_Debugging_OFF();
}
JNIEXPORT jint JNICALL Java_com_wolfssl_WolfSSL_setLoggingCb
(JNIEnv* jenv, jclass jcl, jobject callback)
{
int ret = 0;
if (!jenv || !callback) {
return BAD_FUNC_ARG;
}
/* store Java logging callback Interface object */
g_loggingCbIfaceObj = (*jenv)->NewGlobalRef(jenv, callback);
if (!g_loggingCbIfaceObj) {
printf("error storing global logging callback interface\n");
return SSL_FAILURE;
}
ret = CyaSSL_SetLoggingCb(NativeLoggingCallback);
return ret;
}
void NativeLoggingCallback(const int logLevel, const char *const logMessage)
{
JNIEnv* jenv;
jint vmret = 0;
jclass excClass;
jmethodID logMethod;
jobjectRefType refcheck;
/* get JNIEnv from JavaVM */
vmret = (int)((*g_vm)->GetEnv(g_vm, (void**) &jenv, JNI_VERSION_1_6));
if (vmret == JNI_EDETACHED) {
#ifdef __ANDROID__
vmret = (*g_vm)->AttachCurrentThread(g_vm, &jenv, NULL);
#else
vmret = (*g_vm)->AttachCurrentThread(g_vm, (void**) &jenv, NULL);
#endif
if (vmret) {
printf("Failed to attach JNIEnv to thread\n");
}
} else if (vmret != JNI_OK) {
printf("Unable to get JNIEnv from JavaVM\n");
}
/* find exception class */
excClass = (*jenv)->FindClass(jenv, "java/lang/Exception");
if ((*jenv)->ExceptionOccurred(jenv)) {
(*jenv)->ExceptionDescribe(jenv);
(*jenv)->ExceptionClear(jenv);
return;
}
/* check if our stored object reference is valid */
refcheck = (*jenv)->GetObjectRefType(jenv, g_loggingCbIfaceObj);
if (refcheck == 2) {
/* lookup WolfSSLLoggingCallback class from global object ref */
jclass logClass = (*jenv)->GetObjectClass(jenv, g_loggingCbIfaceObj);
if (!logClass) {
if ((*jenv)->ExceptionOccurred(jenv)) {
(*jenv)->ExceptionDescribe(jenv);
(*jenv)->ExceptionClear(jenv);
}
(*jenv)->ThrowNew(jenv, excClass,
"Can't get native WolfSSLLoggingCallback class reference");
return;
}
logMethod = (*jenv)->GetMethodID(jenv, logClass,
"loggingCallback",
"(ILjava/lang/String;)V");
if (logMethod == 0) {
if ((*jenv)->ExceptionOccurred(jenv)) {
(*jenv)->ExceptionDescribe(jenv);
(*jenv)->ExceptionClear(jenv);
}
(*jenv)->ThrowNew(jenv, excClass,
"Error getting loggingCallback method from JNI");
return;
}
/* create jstring from char* */
jstring logMsg = (*jenv)->NewStringUTF(jenv, logMessage);
(*jenv)->CallVoidMethod(jenv, g_loggingCbIfaceObj, logMethod,
logLevel, logMsg);
if ((*jenv)->ExceptionOccurred(jenv)) {
(*jenv)->ExceptionDescribe(jenv);
(*jenv)->ExceptionClear(jenv);
(*jenv)->ThrowNew(jenv, excClass,
"Error calling logging callback from JNI");
return;
}
} else {
if ((*jenv)->ExceptionOccurred(jenv)) {
(*jenv)->ExceptionDescribe(jenv);
(*jenv)->ExceptionClear(jenv);
}
(*jenv)->ThrowNew(jenv, excClass,
"Object reference invalid in NativeLoggingCallback");
}
}
JNIEXPORT jint JNICALL Java_com_wolfssl_WolfSSL_memsaveSessionCache
(JNIEnv* jenv, jclass jcl, jbyteArray mem, jint sz)
{
int ret;
int cacheSz;
char memBuf[sz];
if (!jenv || !mem || (sz <= 0))
return BAD_FUNC_ARG;
ret = CyaSSL_memsave_session_cache(memBuf, sz);
/* how much data do we need to write? */
cacheSz = CyaSSL_get_session_cache_memsize();
/* set jbyteArray for return */
if (cacheSz >= 0) {
(*jenv)->SetByteArrayRegion(jenv, mem, 0, cacheSz, (jbyte*)memBuf);
if ((*jenv)->ExceptionOccurred(jenv)) {
(*jenv)->ExceptionDescribe(jenv);
(*jenv)->ExceptionClear(jenv);
return SSL_FAILURE;
}
}
return ret;
}
JNIEXPORT jint JNICALL Java_com_wolfssl_WolfSSL_memrestoreSessionCache
(JNIEnv* jenv, jclass jcl, jbyteArray mem, jint sz)
{
int ret;
char memBuf[sz];
if (!jenv || !mem || (sz <= 0))
return BAD_FUNC_ARG;
(*jenv)->GetByteArrayRegion(jenv, mem, 0, sz, (jbyte*)memBuf);
if ((*jenv)->ExceptionOccurred(jenv)) {
(*jenv)->ExceptionDescribe(jenv);
(*jenv)->ExceptionClear(jenv);
return SSL_FAILURE;
}
ret = CyaSSL_memrestore_session_cache(memBuf, sz);
return ret;
}
JNIEXPORT jint JNICALL Java_com_wolfssl_WolfSSL_getSessionCacheMemsize
(JNIEnv* jenv, jclass jcl)
{
return CyaSSL_get_session_cache_memsize();
}
JNIEXPORT jbyteArray JNICALL Java_com_wolfssl_WolfSSL_x509_1getDer
(JNIEnv* jenv, jclass jcl, jlong x509)
{
int* outSz = NULL;
const unsigned char* derCert;
jbyteArray out = NULL;
if (!jenv || !x509)
return NULL;
derCert = CyaSSL_X509_get_der((CYASSL_X509*)x509, outSz);
if (outSz >= 0) {
(*jenv)->SetByteArrayRegion(jenv, out, 0, *outSz, (jbyte*)derCert);
if ((*jenv)->ExceptionOccurred(jenv)) {
(*jenv)->ExceptionDescribe(jenv);
(*jenv)->ExceptionClear(jenv);
return NULL;
}
return out;
} else {
return NULL;
}
}
JNIEXPORT jint JNICALL Java_com_wolfssl_WolfSSL_getHmacMaxSize
(JNIEnv* jenv, jclass jcl)
{
return MAX_DIGEST_SIZE;
}