src/java.base/share/classes/sun/security/ssl/PskKeyExchangeModesExtension.java
branchJDK-8145252-TLS13-branch
changeset 56542 56aaa6cb3693
child 56603 f103e0c2be1e
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/java.base/share/classes/sun/security/ssl/PskKeyExchangeModesExtension.java	Fri May 11 15:53:12 2018 -0700
@@ -0,0 +1,174 @@
+/*
+ * Copyright (c) 2018, 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.ssl;
+
+import java.io.IOException;
+import java.nio.ByteBuffer;
+import java.text.MessageFormat;
+import java.util.*;
+import sun.security.ssl.SSLExtension.ExtensionConsumer;
+
+import sun.security.ssl.SSLExtension.SSLExtensionSpec;
+import sun.security.ssl.SSLHandshake.HandshakeMessage;
+
+/**
+ * Pack of the "psk_key_exchange_modes" extensions.
+ */
+final class PskKeyExchangeModesExtension {
+    static final HandshakeProducer chNetworkProducer =
+            new PskKeyExchangeModesProducer();
+    static final ExtensionConsumer chOnLoadConsumer =
+            new PskKeyExchangeModesConsumer();
+
+    enum PskKeyExchangeMode {
+        PSK_KE(0),
+        PSK_DHE_KE(1);
+
+        private final int v;
+
+        PskKeyExchangeMode(int v) {
+            this.v = v;
+        }
+
+        static PskKeyExchangeMode ofInt(int v) {
+            for(PskKeyExchangeMode mode : values()) {
+                if (mode.v == v) {
+                    return mode;
+                }
+            }
+
+            return null;
+        }
+    }
+
+    static final class PskKeyExchangeModesSpec implements SSLExtensionSpec {
+
+
+        final List<PskKeyExchangeMode> modes;
+
+        PskKeyExchangeModesSpec(List<PskKeyExchangeMode> modes) {
+            this.modes = modes;
+        }
+
+        PskKeyExchangeModesSpec(ByteBuffer m) throws IOException {
+
+            modes = new ArrayList<>();
+            int modesEncodedLength = Record.getInt8(m);
+            int modesReadLength = 0;
+            while (modesReadLength < modesEncodedLength) {
+                int mode = Record.getInt8(m);
+                // TODO: handle incorrect values
+                modes.add(PskKeyExchangeMode.ofInt(mode));
+                modesReadLength += 1;
+            }
+        }
+
+        byte[] getEncoded() throws IOException {
+
+            int encodedLength = modes.size() + 1;
+            byte[] buffer = new byte[encodedLength];
+            ByteBuffer m = ByteBuffer.wrap(buffer);
+            Record.putInt8(m, modes.size());
+            for(PskKeyExchangeMode curMode : modes) {
+                Record.putInt8(m, curMode.v);
+            }
+
+            return buffer;
+        }
+
+        @Override
+        public String toString() {
+            MessageFormat messageFormat = new MessageFormat(
+            "\"PskKeyExchangeModes\": '{'\n" +
+            "  \"ke_modes\"      : \"{0}\",\n" +
+            "'}'",
+            Locale.ENGLISH);
+
+            Object[] messageFields = {
+            Utilities.indent(modesString()),
+            };
+
+            return messageFormat.format(messageFields);
+        }
+
+        String modesString() {
+            StringBuilder result = new StringBuilder();
+            for(PskKeyExchangeMode curMode : modes) {
+                result.append(curMode.toString() + "\n");
+            }
+
+            return result.toString();
+        }
+    }
+
+
+    private static final
+            class PskKeyExchangeModesConsumer implements ExtensionConsumer {
+        // Prevent instantiation of this class.
+        private PskKeyExchangeModesConsumer() {
+            // blank
+        }
+
+        @Override
+        public void consume(ConnectionContext context,
+            HandshakeMessage message, ByteBuffer buffer) throws IOException {
+
+            ServerHandshakeContext shc =
+                    (ServerHandshakeContext) message.handshakeContext;
+
+            PskKeyExchangeModesSpec modes = new PskKeyExchangeModesSpec(buffer);
+            if (SSLLogger.isOn && SSLLogger.isOn("ssl,handshake")) {
+                SSLLogger.fine(
+                "Received PskKeyExchangeModes extension: ", modes);
+            }
+
+            shc.pskKeyExchangeModes = modes.modes;
+        }
+
+    }
+
+    private static final
+            class PskKeyExchangeModesProducer implements HandshakeProducer {
+
+        static final List<PskKeyExchangeMode> MODES =
+            List.of(PskKeyExchangeMode.PSK_DHE_KE);
+        static final PskKeyExchangeModesSpec MODES_MSG =
+            new PskKeyExchangeModesSpec(MODES);
+
+        // Prevent instantiation of this class.
+        private PskKeyExchangeModesProducer() {
+            // blank
+        }
+
+        @Override
+        public byte[] produce(ConnectionContext context,
+                HandshakeMessage message) throws IOException {
+
+            return MODES_MSG.getEncoded();
+        }
+
+    }
+}
+