/*

 */

/* Copyright  (c) 2002 Graz University of Technology. All rights reserved.

 *

 * Redistribution and use in  source and binary forms, with or without

 * modification, are permitted  provided that the following conditions are met:

 *

 * 1. Redistributions of  source code must retain the above copyright notice,

 *    this list of conditions and the following disclaimer.

 *

 * 2. Redistributions in  binary form must reproduce the above copyright notice,

 *    this list of conditions and the following disclaimer in the documentation

 *    and/or other materials provided with the distribution.

 *

 * 3. The end-user documentation included with the redistribution, if any, must

 *    include the following acknowledgment:

 *

 *    "This product includes software developed by IAIK of Graz University of

 *     Technology."

 *

 *    Alternately, this acknowledgment may appear in the software itself, if

 *    and wherever such third-party acknowledgments normally appear.

 *

 * 4. The names "Graz University of Technology" and "IAIK of Graz University of

 *    Technology" must not be used to endorse or promote products derived from

 *    this software without prior written permission.

 *

 * 5. Products derived from this software may not be called

 *    "IAIK PKCS Wrapper", nor may "IAIK" appear in their name, without prior

 *    written permission of Graz University of Technology.

 *

 *  THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESSED OR IMPLIED

 *  WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

 *  WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

 *  PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE LICENSOR BE

 *  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,

 *  OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

 *  PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,

 *  OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON

 *  ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,

 *  OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY

 *  OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

 *  POSSIBILITY  OF SUCH DAMAGE.

 */

#include "pkcs11wrapper.h"

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <assert.h>

#include "jlong.h"

#include "sun_security_pkcs11_wrapper_PKCS11.h"

#ifdef P11_ENABLE_C_DIGESTINIT

/*

 * Class:     sun_security_pkcs11_wrapper_PKCS11

 * Method:    C_DigestInit

 * Signature: (JLsun/security/pkcs11/wrapper/CK_MECHANISM;)V

 * Parametermapping:                    *PKCS11*

 * @param   jlong jSessionHandle        CK_SESSION_HANDLE hSession

 * @param   jobject jMechanism          CK_MECHANISM_PTR pMechanism

 */

JNIEXPORT void JNICALL Java_sun_security_pkcs11_wrapper_PKCS11_C_1DigestInit

    (JNIEnv *env, jobject obj, jlong jSessionHandle, jobject jMechanism)

{

    CK_SESSION_HANDLE ckSessionHandle;

    CK_MECHANISM ckMechanism;

    CK_RV rv;

    CK_FUNCTION_LIST_PTR ckpFunctions = getFunctionList(env, obj);

    if (ckpFunctions == NULL) { return; }

    ckSessionHandle = jLongToCKULong(jSessionHandle);

    jMechanismToCKMechanism(env, jMechanism, &ckMechanism);

    if ((*env)->ExceptionCheck(env)) { return; }

    rv = (*ckpFunctions->C_DigestInit)(ckSessionHandle, &ckMechanism);

    if (ckMechanism.pParameter != NULL_PTR) {

        free(ckMechanism.pParameter);

    }

    if (ckAssertReturnValueOK(env, rv) != CK_ASSERT_OK) { return; }

}

#endif

#ifdef P11_ENABLE_C_DIGEST

/*

 * Class:     sun_security_pkcs11_wrapper_PKCS11

 * Method:    C_Digest

 * Signature: (J[BII[BII)I

 * Parametermapping:                    *PKCS11*

 * @param   jlong jSessionHandle        CK_SESSION_HANDLE hSession

 * @param   jbyteArray jData            CK_BYTE_PTR pData

 *                                      CK_ULONG ulDataLen

 * @return  jbyteArray jDigest          CK_BYTE_PTR pDigest

 *                                      CK_ULONG_PTR pulDigestLen

 */

JNIEXPORT jint JNICALL Java_sun_security_pkcs11_wrapper_PKCS11_C_1DigestSingle

  (JNIEnv *env, jobject obj, jlong jSessionHandle, jobject jMechanism, jbyteArray jIn, jint jInOfs, jint jInLen, jbyteArray jDigest, jint jDigestOfs, jint jDigestLen)

{

    CK_SESSION_HANDLE ckSessionHandle;

    CK_RV rv;

    CK_BYTE_PTR bufP;

    CK_BYTE BUF[MAX_STACK_BUFFER_LEN];

    CK_BYTE DIGESTBUF[MAX_DIGEST_LEN];

    CK_ULONG ckDigestLength = min(MAX_DIGEST_LEN, jDigestLen);

    CK_MECHANISM ckMechanism;

    CK_FUNCTION_LIST_PTR ckpFunctions = getFunctionList(env, obj);

    if (ckpFunctions == NULL) { return 0; }

    ckSessionHandle = jLongToCKULong(jSessionHandle);

    jMechanismToCKMechanism(env, jMechanism, &ckMechanism);

    if ((*env)->ExceptionCheck(env)) { return 0; }

    rv = (*ckpFunctions->C_DigestInit)(ckSessionHandle, &ckMechanism);

    if (ckMechanism.pParameter != NULL_PTR) {

        free(ckMechanism.pParameter);

    }

    if (ckAssertReturnValueOK(env, rv) != CK_ASSERT_OK) { return 0; }

    if (jInLen <= MAX_STACK_BUFFER_LEN) {

        bufP = BUF;

    } else {

        /* always use single part op, even for large data */

        bufP = (CK_BYTE_PTR) malloc((size_t)jInLen);

        if (bufP == NULL) {

            throwOutOfMemoryError(env, 0);

            return 0;

        }

    }

    (*env)->GetByteArrayRegion(env, jIn, jInOfs, jInLen, (jbyte *)bufP);

    if ((*env)->ExceptionCheck(env)) {

        if (bufP != BUF) { free(bufP); }

        return 0;

    }

    rv = (*ckpFunctions->C_Digest)(ckSessionHandle, bufP, jInLen, DIGESTBUF, &ckDigestLength);

    if (ckAssertReturnValueOK(env, rv) == CK_ASSERT_OK) {

        (*env)->SetByteArrayRegion(env, jDigest, jDigestOfs, ckDigestLength, (jbyte *)DIGESTBUF);

    }

    if (bufP != BUF) { free(bufP); }

    return ckDigestLength;

}

#endif

#ifdef P11_ENABLE_C_DIGESTUPDATE

/*

 * Class:     sun_security_pkcs11_wrapper_PKCS11

 * Method:    C_DigestUpdate

 * Signature: (J[B)V

 * Parametermapping:                    *PKCS11*

 * @param   jlong jSessionHandle        CK_SESSION_HANDLE hSession

 * @param   jbyteArray jData            CK_BYTE_PTR pData

 *                                      CK_ULONG ulDataLen

 */

JNIEXPORT void JNICALL Java_sun_security_pkcs11_wrapper_PKCS11_C_1DigestUpdate

  (JNIEnv *env, jobject obj, jlong jSessionHandle, jlong directIn, jbyteArray jIn, jint jInOfs, jint jInLen)

{

    CK_SESSION_HANDLE ckSessionHandle;

    CK_RV rv;

    CK_BYTE_PTR bufP;

    CK_BYTE BUF[MAX_STACK_BUFFER_LEN];

    jsize bufLen;

    CK_FUNCTION_LIST_PTR ckpFunctions = getFunctionList(env, obj);

    if (ckpFunctions == NULL) { return; }

    ckSessionHandle = jLongToCKULong(jSessionHandle);

    if (directIn != 0) {

        rv = (*ckpFunctions->C_DigestUpdate)(ckSessionHandle, (CK_BYTE_PTR)jlong_to_ptr(directIn), jInLen);

        ckAssertReturnValueOK(env, rv);

        return;

    }

    if (jInLen <= MAX_STACK_BUFFER_LEN) {

        bufLen = MAX_STACK_BUFFER_LEN;

        bufP = BUF;

    } else {

        bufLen = min(MAX_HEAP_BUFFER_LEN, jInLen);

        bufP = (CK_BYTE_PTR) malloc((size_t)bufLen);

        if (bufP == NULL) {

            throwOutOfMemoryError(env, 0);

            return;

        }

    }

    while (jInLen > 0) {

        jsize chunkLen = min(bufLen, jInLen);

        (*env)->GetByteArrayRegion(env, jIn, jInOfs, chunkLen, (jbyte *)bufP);

        if ((*env)->ExceptionCheck(env)) {

            if (bufP != BUF) { free(bufP); }

            return;

        }

        rv = (*ckpFunctions->C_DigestUpdate)(ckSessionHandle, bufP, chunkLen);

        if (ckAssertReturnValueOK(env, rv) != CK_ASSERT_OK) {

            if (bufP != BUF) { free(bufP); }

            return;

        }

        jInOfs += chunkLen;

        jInLen -= chunkLen;

    }

    if (bufP != BUF) {

        free(bufP);

    }

}

#endif

#ifdef P11_ENABLE_C_DIGESTKEY

/*

 * Class:     sun_security_pkcs11_wrapper_PKCS11

 * Method:    C_DigestKey

 * Signature: (JJ)V

 * Parametermapping:                    *PKCS11*

 * @param   jlong jSessionHandle        CK_SESSION_HANDLE hSession

 * @param   jlong jKeyHandle            CK_OBJECT_HANDLE hKey

 */

JNIEXPORT void JNICALL Java_sun_security_pkcs11_wrapper_PKCS11_C_1DigestKey

    (JNIEnv *env, jobject obj, jlong jSessionHandle, jlong jKeyHandle)

{

    CK_SESSION_HANDLE ckSessionHandle;

    CK_ULONG ckKeyHandle;

    CK_RV rv;

    CK_FUNCTION_LIST_PTR ckpFunctions = getFunctionList(env, obj);

    if (ckpFunctions == NULL) { return; }

    ckSessionHandle = jLongToCKULong(jSessionHandle);

    ckKeyHandle = jLongToCKULong(jKeyHandle);

    rv = (*ckpFunctions->C_DigestKey)(ckSessionHandle, ckKeyHandle);

    if (ckAssertReturnValueOK(env, rv) != CK_ASSERT_OK) { return; }

}

#endif

#ifdef P11_ENABLE_C_DIGESTFINAL

/*

 * Class:     sun_security_pkcs11_wrapper_PKCS11

 * Method:    C_DigestFinal

 * Signature: (J[BII)I

 * Parametermapping:                    *PKCS11*

 * @param   jlong jSessionHandle        CK_SESSION_HANDLE hSession

 * @return  jbyteArray jDigest          CK_BYTE_PTR pDigest

 *                                      CK_ULONG_PTR pulDigestLen

 */

JNIEXPORT jint JNICALL Java_sun_security_pkcs11_wrapper_PKCS11_C_1DigestFinal

  (JNIEnv *env, jobject obj, jlong jSessionHandle, jbyteArray jDigest, jint jDigestOfs, jint jDigestLen)

{

    CK_SESSION_HANDLE ckSessionHandle;

    CK_RV rv;

    CK_BYTE BUF[MAX_DIGEST_LEN];

    CK_ULONG ckDigestLength = min(MAX_DIGEST_LEN, jDigestLen);

    CK_FUNCTION_LIST_PTR ckpFunctions = getFunctionList(env, obj);

    if (ckpFunctions == NULL) { return 0; }

    ckSessionHandle = jLongToCKULong(jSessionHandle);

    rv = (*ckpFunctions->C_DigestFinal)(ckSessionHandle, BUF, &ckDigestLength);

    if (ckAssertReturnValueOK(env, rv) == CK_ASSERT_OK) {

        (*env)->SetByteArrayRegion(env, jDigest, jDigestOfs, ckDigestLength, (jbyte *)BUF);

    }

    return ckDigestLength;

}

#endif

#ifdef P11_ENABLE_C_SEEDRANDOM

/*

 * Class:     sun_security_pkcs11_wrapper_PKCS11

 * Method:    C_SeedRandom

 * Signature: (J[B)V

 * Parametermapping:                    *PKCS11*

 * @param   jlong jSessionHandle        CK_SESSION_HANDLE hSession

 * @param   jbyteArray jSeed            CK_BYTE_PTR pSeed

 *                                      CK_ULONG ulSeedLen

 */

JNIEXPORT void JNICALL Java_sun_security_pkcs11_wrapper_PKCS11_C_1SeedRandom

    (JNIEnv *env, jobject obj, jlong jSessionHandle, jbyteArray jSeed)

{

    CK_SESSION_HANDLE ckSessionHandle;

    CK_BYTE_PTR ckpSeed = NULL_PTR;

    CK_ULONG ckSeedLength;

    CK_RV rv;

    CK_FUNCTION_LIST_PTR ckpFunctions = getFunctionList(env, obj);

    if (ckpFunctions == NULL) { return; }

    ckSessionHandle = jLongToCKULong(jSessionHandle);

    jByteArrayToCKByteArray(env, jSeed, &ckpSeed, &ckSeedLength);

    if ((*env)->ExceptionCheck(env)) { return; }

    rv = (*ckpFunctions->C_SeedRandom)(ckSessionHandle, ckpSeed, ckSeedLength);

    free(ckpSeed);

    if (ckAssertReturnValueOK(env, rv) != CK_ASSERT_OK) { return; }

}

#endif

#ifdef P11_ENABLE_C_GENERATERANDOM

/*

 * Class:     sun_security_pkcs11_wrapper_PKCS11

 * Method:    C_GenerateRandom

 * Signature: (J[B)V

 * Parametermapping:                    *PKCS11*

 * @param   jlong jSessionHandle        CK_SESSION_HANDLE hSession

 * @param   jbyteArray jRandomData      CK_BYTE_PTR pRandomData

 *                                      CK_ULONG ulRandomDataLen

 */

JNIEXPORT void JNICALL Java_sun_security_pkcs11_wrapper_PKCS11_C_1GenerateRandom

    (JNIEnv *env, jobject obj, jlong jSessionHandle, jbyteArray jRandomData)

{

    CK_SESSION_HANDLE ckSessionHandle;

    jbyte *jRandomBuffer;

    jlong jRandomBufferLength;

    CK_RV rv;

    CK_FUNCTION_LIST_PTR ckpFunctions = getFunctionList(env, obj);

    if (ckpFunctions == NULL) { return; }

    ckSessionHandle = jLongToCKULong(jSessionHandle);

    jRandomBufferLength = (*env)->GetArrayLength(env, jRandomData);

    jRandomBuffer = (*env)->GetByteArrayElements(env, jRandomData, NULL);

    if (jRandomBuffer == NULL) { return; }

    rv = (*ckpFunctions->C_GenerateRandom)(ckSessionHandle,

                                         (CK_BYTE_PTR) jRandomBuffer,

                                         jLongToCKULong(jRandomBufferLength));

    /* copy back generated bytes */

    (*env)->ReleaseByteArrayElements(env, jRandomData, jRandomBuffer, 0);

    if (ckAssertReturnValueOK(env, rv) != CK_ASSERT_OK) { return; }

}

#endif