1/*
2 * Copyright (c) 2003, 2012, Oracle and/or its affiliates. All rights reserved.
3 */
4
5/* Copyright  (c) 2002 Graz University of Technology. All rights reserved.
6 *
7 * Redistribution and use in  source and binary forms, with or without
8 * modification, are permitted  provided that the following conditions are met:
9 *
10 * 1. Redistributions of  source code must retain the above copyright notice,
11 *    this list of conditions and the following disclaimer.
12 *
13 * 2. Redistributions in  binary form must reproduce the above copyright notice,
14 *    this list of conditions and the following disclaimer in the documentation
15 *    and/or other materials provided with the distribution.
16 *
17 * 3. The end-user documentation included with the redistribution, if any, must
18 *    include the following acknowledgment:
19 *
20 *    "This product includes software developed by IAIK of Graz University of
21 *     Technology."
22 *
23 *    Alternately, this acknowledgment may appear in the software itself, if
24 *    and wherever such third-party acknowledgments normally appear.
25 *
26 * 4. The names "Graz University of Technology" and "IAIK of Graz University of
27 *    Technology" must not be used to endorse or promote products derived from
28 *    this software without prior written permission.
29 *
30 * 5. Products derived from this software may not be called
31 *    "IAIK PKCS Wrapper", nor may "IAIK" appear in their name, without prior
32 *    written permission of Graz University of Technology.
33 *
34 *  THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESSED OR IMPLIED
35 *  WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
36 *  WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
37 *  PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE LICENSOR BE
38 *  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
39 *  OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
40 *  PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
41 *  OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
42 *  ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
43 *  OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
44 *  OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
45 *  POSSIBILITY  OF SUCH DAMAGE.
46 */
47
48#include "pkcs11wrapper.h"
49
50#include <stdio.h>
51#include <stdlib.h>
52#include <string.h>
53#include <assert.h>
54#include "jlong.h"
55
56#include "sun_security_pkcs11_wrapper_PKCS11.h"
57
58#ifdef P11_ENABLE_C_DIGESTINIT
59/*
60 * Class:     sun_security_pkcs11_wrapper_PKCS11
61 * Method:    C_DigestInit
62 * Signature: (JLsun/security/pkcs11/wrapper/CK_MECHANISM;)V
63 * Parametermapping:                    *PKCS11*
64 * @param   jlong jSessionHandle        CK_SESSION_HANDLE hSession
65 * @param   jobject jMechanism          CK_MECHANISM_PTR pMechanism
66 */
67JNIEXPORT void JNICALL Java_sun_security_pkcs11_wrapper_PKCS11_C_1DigestInit
68    (JNIEnv *env, jobject obj, jlong jSessionHandle, jobject jMechanism)
69{
70    CK_SESSION_HANDLE ckSessionHandle;
71    CK_MECHANISM ckMechanism;
72    CK_RV rv;
73
74    CK_FUNCTION_LIST_PTR ckpFunctions = getFunctionList(env, obj);
75    if (ckpFunctions == NULL) { return; }
76
77    ckSessionHandle = jLongToCKULong(jSessionHandle);
78    jMechanismToCKMechanism(env, jMechanism, &ckMechanism);
79    if ((*env)->ExceptionCheck(env)) { return; }
80
81    rv = (*ckpFunctions->C_DigestInit)(ckSessionHandle, &ckMechanism);
82
83    if (ckMechanism.pParameter != NULL_PTR) {
84        free(ckMechanism.pParameter);
85    }
86
87    if (ckAssertReturnValueOK(env, rv) != CK_ASSERT_OK) { return; }
88}
89#endif
90
91#ifdef P11_ENABLE_C_DIGEST
92/*
93 * Class:     sun_security_pkcs11_wrapper_PKCS11
94 * Method:    C_Digest
95 * Signature: (J[BII[BII)I
96 * Parametermapping:                    *PKCS11*
97 * @param   jlong jSessionHandle        CK_SESSION_HANDLE hSession
98 * @param   jbyteArray jData            CK_BYTE_PTR pData
99 *                                      CK_ULONG ulDataLen
100 * @return  jbyteArray jDigest          CK_BYTE_PTR pDigest
101 *                                      CK_ULONG_PTR pulDigestLen
102 */
103JNIEXPORT jint JNICALL Java_sun_security_pkcs11_wrapper_PKCS11_C_1DigestSingle
104  (JNIEnv *env, jobject obj, jlong jSessionHandle, jobject jMechanism, jbyteArray jIn, jint jInOfs, jint jInLen, jbyteArray jDigest, jint jDigestOfs, jint jDigestLen)
105{
106    CK_SESSION_HANDLE ckSessionHandle;
107    CK_RV rv;
108    CK_BYTE_PTR bufP;
109    CK_BYTE BUF[MAX_STACK_BUFFER_LEN];
110    CK_BYTE DIGESTBUF[MAX_DIGEST_LEN];
111    CK_ULONG ckDigestLength = min(MAX_DIGEST_LEN, jDigestLen);
112    CK_MECHANISM ckMechanism;
113
114    CK_FUNCTION_LIST_PTR ckpFunctions = getFunctionList(env, obj);
115    if (ckpFunctions == NULL) { return 0; }
116
117    ckSessionHandle = jLongToCKULong(jSessionHandle);
118    jMechanismToCKMechanism(env, jMechanism, &ckMechanism);
119    if ((*env)->ExceptionCheck(env)) { return 0; }
120
121    rv = (*ckpFunctions->C_DigestInit)(ckSessionHandle, &ckMechanism);
122
123    if (ckMechanism.pParameter != NULL_PTR) {
124        free(ckMechanism.pParameter);
125    }
126
127    if (ckAssertReturnValueOK(env, rv) != CK_ASSERT_OK) { return 0; }
128
129    if (jInLen <= MAX_STACK_BUFFER_LEN) {
130        bufP = BUF;
131    } else {
132        /* always use single part op, even for large data */
133        bufP = (CK_BYTE_PTR) malloc((size_t)jInLen);
134        if (bufP == NULL) {
135            throwOutOfMemoryError(env, 0);
136            return 0;
137        }
138    }
139
140    (*env)->GetByteArrayRegion(env, jIn, jInOfs, jInLen, (jbyte *)bufP);
141    if ((*env)->ExceptionCheck(env)) {
142        if (bufP != BUF) { free(bufP); }
143        return 0;
144    }
145
146    rv = (*ckpFunctions->C_Digest)(ckSessionHandle, bufP, jInLen, DIGESTBUF, &ckDigestLength);
147    if (ckAssertReturnValueOK(env, rv) == CK_ASSERT_OK) {
148        (*env)->SetByteArrayRegion(env, jDigest, jDigestOfs, ckDigestLength, (jbyte *)DIGESTBUF);
149    }
150
151    if (bufP != BUF) { free(bufP); }
152
153    return ckDigestLength;
154}
155#endif
156
157#ifdef P11_ENABLE_C_DIGESTUPDATE
158/*
159 * Class:     sun_security_pkcs11_wrapper_PKCS11
160 * Method:    C_DigestUpdate
161 * Signature: (J[B)V
162 * Parametermapping:                    *PKCS11*
163 * @param   jlong jSessionHandle        CK_SESSION_HANDLE hSession
164 * @param   jbyteArray jData            CK_BYTE_PTR pData
165 *                                      CK_ULONG ulDataLen
166 */
167JNIEXPORT void JNICALL Java_sun_security_pkcs11_wrapper_PKCS11_C_1DigestUpdate
168  (JNIEnv *env, jobject obj, jlong jSessionHandle, jlong directIn, jbyteArray jIn, jint jInOfs, jint jInLen)
169{
170    CK_SESSION_HANDLE ckSessionHandle;
171    CK_RV rv;
172    CK_BYTE_PTR bufP;
173    CK_BYTE BUF[MAX_STACK_BUFFER_LEN];
174    jsize bufLen;
175
176    CK_FUNCTION_LIST_PTR ckpFunctions = getFunctionList(env, obj);
177    if (ckpFunctions == NULL) { return; }
178
179    ckSessionHandle = jLongToCKULong(jSessionHandle);
180
181    if (directIn != 0) {
182        rv = (*ckpFunctions->C_DigestUpdate)(ckSessionHandle, (CK_BYTE_PTR)jlong_to_ptr(directIn), jInLen);
183        ckAssertReturnValueOK(env, rv);
184        return;
185    }
186
187    if (jInLen <= MAX_STACK_BUFFER_LEN) {
188        bufLen = MAX_STACK_BUFFER_LEN;
189        bufP = BUF;
190    } else {
191        bufLen = min(MAX_HEAP_BUFFER_LEN, jInLen);
192        bufP = (CK_BYTE_PTR) malloc((size_t)bufLen);
193        if (bufP == NULL) {
194            throwOutOfMemoryError(env, 0);
195            return;
196        }
197    }
198
199    while (jInLen > 0) {
200        jsize chunkLen = min(bufLen, jInLen);
201        (*env)->GetByteArrayRegion(env, jIn, jInOfs, chunkLen, (jbyte *)bufP);
202        if ((*env)->ExceptionCheck(env)) {
203            if (bufP != BUF) { free(bufP); }
204            return;
205        }
206        rv = (*ckpFunctions->C_DigestUpdate)(ckSessionHandle, bufP, chunkLen);
207        if (ckAssertReturnValueOK(env, rv) != CK_ASSERT_OK) {
208            if (bufP != BUF) { free(bufP); }
209            return;
210        }
211        jInOfs += chunkLen;
212        jInLen -= chunkLen;
213    }
214
215    if (bufP != BUF) {
216        free(bufP);
217    }
218}
219#endif
220
221#ifdef P11_ENABLE_C_DIGESTKEY
222/*
223 * Class:     sun_security_pkcs11_wrapper_PKCS11
224 * Method:    C_DigestKey
225 * Signature: (JJ)V
226 * Parametermapping:                    *PKCS11*
227 * @param   jlong jSessionHandle        CK_SESSION_HANDLE hSession
228 * @param   jlong jKeyHandle            CK_OBJECT_HANDLE hKey
229 */
230JNIEXPORT void JNICALL Java_sun_security_pkcs11_wrapper_PKCS11_C_1DigestKey
231    (JNIEnv *env, jobject obj, jlong jSessionHandle, jlong jKeyHandle)
232{
233    CK_SESSION_HANDLE ckSessionHandle;
234    CK_ULONG ckKeyHandle;
235    CK_RV rv;
236
237    CK_FUNCTION_LIST_PTR ckpFunctions = getFunctionList(env, obj);
238    if (ckpFunctions == NULL) { return; }
239
240    ckSessionHandle = jLongToCKULong(jSessionHandle);
241    ckKeyHandle = jLongToCKULong(jKeyHandle);
242
243    rv = (*ckpFunctions->C_DigestKey)(ckSessionHandle, ckKeyHandle);
244    if (ckAssertReturnValueOK(env, rv) != CK_ASSERT_OK) { return; }
245}
246#endif
247
248#ifdef P11_ENABLE_C_DIGESTFINAL
249/*
250 * Class:     sun_security_pkcs11_wrapper_PKCS11
251 * Method:    C_DigestFinal
252 * Signature: (J[BII)I
253 * Parametermapping:                    *PKCS11*
254 * @param   jlong jSessionHandle        CK_SESSION_HANDLE hSession
255 * @return  jbyteArray jDigest          CK_BYTE_PTR pDigest
256 *                                      CK_ULONG_PTR pulDigestLen
257 */
258JNIEXPORT jint JNICALL Java_sun_security_pkcs11_wrapper_PKCS11_C_1DigestFinal
259  (JNIEnv *env, jobject obj, jlong jSessionHandle, jbyteArray jDigest, jint jDigestOfs, jint jDigestLen)
260{
261    CK_SESSION_HANDLE ckSessionHandle;
262    CK_RV rv;
263    CK_BYTE BUF[MAX_DIGEST_LEN];
264    CK_ULONG ckDigestLength = min(MAX_DIGEST_LEN, jDigestLen);
265
266    CK_FUNCTION_LIST_PTR ckpFunctions = getFunctionList(env, obj);
267    if (ckpFunctions == NULL) { return 0; }
268
269    ckSessionHandle = jLongToCKULong(jSessionHandle);
270
271    rv = (*ckpFunctions->C_DigestFinal)(ckSessionHandle, BUF, &ckDigestLength);
272    if (ckAssertReturnValueOK(env, rv) == CK_ASSERT_OK) {
273        (*env)->SetByteArrayRegion(env, jDigest, jDigestOfs, ckDigestLength, (jbyte *)BUF);
274    }
275    return ckDigestLength;
276}
277#endif
278
279#ifdef P11_ENABLE_C_SEEDRANDOM
280/*
281 * Class:     sun_security_pkcs11_wrapper_PKCS11
282 * Method:    C_SeedRandom
283 * Signature: (J[B)V
284 * Parametermapping:                    *PKCS11*
285 * @param   jlong jSessionHandle        CK_SESSION_HANDLE hSession
286 * @param   jbyteArray jSeed            CK_BYTE_PTR pSeed
287 *                                      CK_ULONG ulSeedLen
288 */
289JNIEXPORT void JNICALL Java_sun_security_pkcs11_wrapper_PKCS11_C_1SeedRandom
290    (JNIEnv *env, jobject obj, jlong jSessionHandle, jbyteArray jSeed)
291{
292    CK_SESSION_HANDLE ckSessionHandle;
293    CK_BYTE_PTR ckpSeed = NULL_PTR;
294    CK_ULONG ckSeedLength;
295    CK_RV rv;
296
297    CK_FUNCTION_LIST_PTR ckpFunctions = getFunctionList(env, obj);
298    if (ckpFunctions == NULL) { return; }
299
300    ckSessionHandle = jLongToCKULong(jSessionHandle);
301    jByteArrayToCKByteArray(env, jSeed, &ckpSeed, &ckSeedLength);
302    if ((*env)->ExceptionCheck(env)) { return; }
303
304    rv = (*ckpFunctions->C_SeedRandom)(ckSessionHandle, ckpSeed, ckSeedLength);
305
306    free(ckpSeed);
307
308    if (ckAssertReturnValueOK(env, rv) != CK_ASSERT_OK) { return; }
309}
310#endif
311
312#ifdef P11_ENABLE_C_GENERATERANDOM
313/*
314 * Class:     sun_security_pkcs11_wrapper_PKCS11
315 * Method:    C_GenerateRandom
316 * Signature: (J[B)V
317 * Parametermapping:                    *PKCS11*
318 * @param   jlong jSessionHandle        CK_SESSION_HANDLE hSession
319 * @param   jbyteArray jRandomData      CK_BYTE_PTR pRandomData
320 *                                      CK_ULONG ulRandomDataLen
321 */
322JNIEXPORT void JNICALL Java_sun_security_pkcs11_wrapper_PKCS11_C_1GenerateRandom
323    (JNIEnv *env, jobject obj, jlong jSessionHandle, jbyteArray jRandomData)
324{
325    CK_SESSION_HANDLE ckSessionHandle;
326    jbyte *jRandomBuffer;
327    jlong jRandomBufferLength;
328    CK_RV rv;
329
330    CK_FUNCTION_LIST_PTR ckpFunctions = getFunctionList(env, obj);
331    if (ckpFunctions == NULL) { return; }
332
333    ckSessionHandle = jLongToCKULong(jSessionHandle);
334
335    jRandomBufferLength = (*env)->GetArrayLength(env, jRandomData);
336    jRandomBuffer = (*env)->GetByteArrayElements(env, jRandomData, NULL);
337    if (jRandomBuffer == NULL) { return; }
338
339    rv = (*ckpFunctions->C_GenerateRandom)(ckSessionHandle,
340                                         (CK_BYTE_PTR) jRandomBuffer,
341                                         jLongToCKULong(jRandomBufferLength));
342
343    /* copy back generated bytes */
344    (*env)->ReleaseByteArrayElements(env, jRandomData, jRandomBuffer, 0);
345
346    if (ckAssertReturnValueOK(env, rv) != CK_ASSERT_OK) { return; }
347}
348#endif
349