--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/src/jdk.crypto.pkcs11/share/classes/sun/security/pkcs11/P11ECDHKeyAgreement.java Sun Aug 17 15:54:13 2014 +0100
@@ -0,0 +1,214 @@
+/*
+ * Copyright (c) 2006, 2007, Oracle and/or its affiliates. All rights reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation. Oracle designates this
+ * particular file as subject to the "Classpath" exception as provided
+ * by Oracle in the LICENSE file that accompanied this code.
+ *
+ * This code 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
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
+ * or visit www.oracle.com if you need additional information or have any
+ * questions.
+ */
+
+package sun.security.pkcs11;
+
+import java.security.*;
+import java.security.interfaces.ECPublicKey;
+import java.security.spec.AlgorithmParameterSpec;
+
+import javax.crypto.*;
+
+import static sun.security.pkcs11.TemplateManager.*;
+import sun.security.pkcs11.wrapper.*;
+import static sun.security.pkcs11.wrapper.PKCS11Constants.*;
+
+/**
+ * KeyAgreement implementation for ECDH.
+ *
+ * @author Andreas Sterbenz
+ * @since 1.6
+ */
+final class P11ECDHKeyAgreement extends KeyAgreementSpi {
+
+ // token instance
+ private final Token token;
+
+ // algorithm name
+ private final String algorithm;
+
+ // mechanism id
+ private final long mechanism;
+
+ // private key, if initialized
+ private P11Key privateKey;
+
+ // encoded public point, non-null between doPhase() and generateSecret() only
+ private byte[] publicValue;
+
+ // length of the secret to be derived
+ private int secretLen;
+
+ P11ECDHKeyAgreement(Token token, String algorithm, long mechanism) {
+ super();
+ this.token = token;
+ this.algorithm = algorithm;
+ this.mechanism = mechanism;
+ }
+
+ // see JCE spec
+ protected void engineInit(Key key, SecureRandom random)
+ throws InvalidKeyException {
+ if (key instanceof PrivateKey == false) {
+ throw new InvalidKeyException
+ ("Key must be instance of PrivateKey");
+ }
+ privateKey = P11KeyFactory.convertKey(token, key, "EC");
+ publicValue = null;
+ }
+
+ // see JCE spec
+ protected void engineInit(Key key, AlgorithmParameterSpec params,
+ SecureRandom random) throws InvalidKeyException,
+ InvalidAlgorithmParameterException {
+ if (params != null) {
+ throw new InvalidAlgorithmParameterException
+ ("Parameters not supported");
+ }
+ engineInit(key, random);
+ }
+
+ // see JCE spec
+ protected Key engineDoPhase(Key key, boolean lastPhase)
+ throws InvalidKeyException, IllegalStateException {
+ if (privateKey == null) {
+ throw new IllegalStateException("Not initialized");
+ }
+ if (publicValue != null) {
+ throw new IllegalStateException("Phase already executed");
+ }
+ if (lastPhase == false) {
+ throw new IllegalStateException
+ ("Only two party agreement supported, lastPhase must be true");
+ }
+ if (key instanceof ECPublicKey == false) {
+ throw new InvalidKeyException
+ ("Key must be a PublicKey with algorithm EC");
+ }
+ ECPublicKey ecKey = (ECPublicKey)key;
+ int keyLenBits = ecKey.getParams().getCurve().getField().getFieldSize();
+ secretLen = (keyLenBits + 7) >> 3;
+ publicValue = P11ECKeyFactory.getEncodedPublicValue(ecKey);
+ return null;
+ }
+
+ // see JCE spec
+ protected byte[] engineGenerateSecret() throws IllegalStateException {
+ if ((privateKey == null) || (publicValue == null)) {
+ throw new IllegalStateException("Not initialized correctly");
+ }
+ Session session = null;
+ try {
+ session = token.getOpSession();
+ CK_ATTRIBUTE[] attributes = new CK_ATTRIBUTE[] {
+ new CK_ATTRIBUTE(CKA_CLASS, CKO_SECRET_KEY),
+ new CK_ATTRIBUTE(CKA_KEY_TYPE, CKK_GENERIC_SECRET),
+ };
+ CK_ECDH1_DERIVE_PARAMS ckParams =
+ new CK_ECDH1_DERIVE_PARAMS(CKD_NULL, null, publicValue);
+ attributes = token.getAttributes
+ (O_GENERATE, CKO_SECRET_KEY, CKK_GENERIC_SECRET, attributes);
+ long keyID = token.p11.C_DeriveKey(session.id(),
+ new CK_MECHANISM(mechanism, ckParams), privateKey.keyID,
+ attributes);
+ attributes = new CK_ATTRIBUTE[] {
+ new CK_ATTRIBUTE(CKA_VALUE)
+ };
+ token.p11.C_GetAttributeValue(session.id(), keyID, attributes);
+ byte[] secret = attributes[0].getByteArray();
+ token.p11.C_DestroyObject(session.id(), keyID);
+ return secret;
+ } catch (PKCS11Exception e) {
+ throw new ProviderException("Could not derive key", e);
+ } finally {
+ publicValue = null;
+ token.releaseSession(session);
+ }
+ }
+
+ // see JCE spec
+ protected int engineGenerateSecret(byte[] sharedSecret, int
+ offset) throws IllegalStateException, ShortBufferException {
+ if (offset + secretLen > sharedSecret.length) {
+ throw new ShortBufferException("Need " + secretLen
+ + " bytes, only " + (sharedSecret.length - offset) + " available");
+ }
+ byte[] secret = engineGenerateSecret();
+ System.arraycopy(secret, 0, sharedSecret, offset, secret.length);
+ return secret.length;
+ }
+
+ // see JCE spec
+ protected SecretKey engineGenerateSecret(String algorithm)
+ throws IllegalStateException, NoSuchAlgorithmException,
+ InvalidKeyException {
+ if (algorithm == null) {
+ throw new NoSuchAlgorithmException("Algorithm must not be null");
+ }
+ if (algorithm.equals("TlsPremasterSecret") == false) {
+ throw new NoSuchAlgorithmException
+ ("Only supported for algorithm TlsPremasterSecret");
+ }
+ return nativeGenerateSecret(algorithm);
+ }
+
+ private SecretKey nativeGenerateSecret(String algorithm)
+ throws IllegalStateException, NoSuchAlgorithmException,
+ InvalidKeyException {
+ if ((privateKey == null) || (publicValue == null)) {
+ throw new IllegalStateException("Not initialized correctly");
+ }
+ long keyType = CKK_GENERIC_SECRET;
+ Session session = null;
+ try {
+ session = token.getObjSession();
+ CK_ATTRIBUTE[] attributes = new CK_ATTRIBUTE[] {
+ new CK_ATTRIBUTE(CKA_CLASS, CKO_SECRET_KEY),
+ new CK_ATTRIBUTE(CKA_KEY_TYPE, keyType),
+ };
+ CK_ECDH1_DERIVE_PARAMS ckParams =
+ new CK_ECDH1_DERIVE_PARAMS(CKD_NULL, null, publicValue);
+ attributes = token.getAttributes
+ (O_GENERATE, CKO_SECRET_KEY, keyType, attributes);
+ long keyID = token.p11.C_DeriveKey(session.id(),
+ new CK_MECHANISM(mechanism, ckParams), privateKey.keyID,
+ attributes);
+ CK_ATTRIBUTE[] lenAttributes = new CK_ATTRIBUTE[] {
+ new CK_ATTRIBUTE(CKA_VALUE_LEN),
+ };
+ token.p11.C_GetAttributeValue(session.id(), keyID, lenAttributes);
+ int keyLen = (int)lenAttributes[0].getLong();
+ SecretKey key = P11Key.secretKey
+ (session, keyID, algorithm, keyLen << 3, attributes);
+ return key;
+ } catch (PKCS11Exception e) {
+ throw new InvalidKeyException("Could not derive key", e);
+ } finally {
+ publicValue = null;
+ token.releaseSession(session);
+ }
+ }
+
+}