1 /* 2 * Copyright (c) 2019, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 /* 25 * @test 26 * @bug 8226374 27 * @library /javax/net/ssl/templates 28 * @summary Restrict signature algorithms and named groups 29 * @run main/othervm RestrictSignatureScheme 30 */ 31 import java.io.ByteArrayInputStream; 32 import java.security.KeyFactory; 33 import java.security.KeyStore; 34 import java.security.PrivateKey; 35 import java.security.Security; 36 import java.security.cert.Certificate; 37 import java.security.cert.CertificateFactory; 38 import java.security.spec.PKCS8EncodedKeySpec; 39 import java.util.Arrays; 40 import java.util.Base64; 41 import javax.net.ssl.KeyManagerFactory; 42 import javax.net.ssl.SSLContext; 43 import javax.net.ssl.SSLSocket; 44 import javax.net.ssl.SSLServerSocket; 45 import javax.net.ssl.TrustManagerFactory; 46 import javax.net.ssl.SSLException; 47 48 public class RestrictSignatureScheme extends SSLSocketTemplate { 49 50 private static volatile int index; 51 private static final String[][][] protocols = { 52 {{"TLSv1.3"}, {"TLSv1.3"}}, 53 {{"TLSv1.3", "TLSv1.2"}, {"TLSv1.2"}}, 54 {{"TLSv1.3", "TLSv1.2"}, {"TLSv1.2"}}, 55 {{"TLSv1.2"}, {"TLSv1.3", "TLSv1.2"}}, 56 {{"TLSv1.2"}, {"TLSv1.2"}} 57 }; 58 59 private final SSLContext context; 60 RestrictSignatureScheme() throws Exception { 61 this.context = createSSLContext(); 62 } 63 64 @Override 65 protected SSLContext createClientSSLContext() throws Exception { 66 return context; 67 } 68 69 @Override 70 protected SSLContext createServerSSLContext() throws Exception { 71 return context; 72 } 73 74 // Servers are configured before clients, increment test case after. 75 @Override 76 protected void configureClientSocket(SSLSocket socket) { 77 String[] ps = protocols[index][0]; 78 79 System.out.print("Setting client protocol(s): "); 80 Arrays.stream(ps).forEachOrdered(System.out::print); 81 System.out.println(); 82 83 socket.setEnabledProtocols(ps); 84 socket.setEnabledCipherSuites(new String[] { 85 "TLS_AES_128_GCM_SHA256", 86 "TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA256"}); 87 } 88 89 @Override 90 protected void configureServerSocket(SSLServerSocket serverSocket) { 91 String[] ps = protocols[index][1]; 92 93 System.out.print("Setting server protocol(s): "); 94 Arrays.stream(ps).forEachOrdered(System.out::print); 95 System.out.println(); 96 97 serverSocket.setEnabledProtocols(ps); 98 serverSocket.setEnabledCipherSuites(new String[] { 99 "TLS_AES_128_GCM_SHA256", 100 "TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA256"}); 101 } 102 103 /* 104 * Run the test case. 105 */ 106 public static void main(String[] args) throws Exception { 107 Security.setProperty("jdk.tls.disabledAlgorithms", "RSASSA-PSS"); 108 109 for (index = 0; index < protocols.length; index++) { 110 try { 111 (new RestrictSignatureScheme()).run(); 112 } catch (SSLException | IllegalStateException ssle) { 113 // The named group should be restricted. 114 continue; 115 } 116 117 throw new Exception("The test case should be disabled"); 118 } 119 } 120 121 122 private static final String trustedCertStr = 123 /** 124 * Signature Algorithm: rsassaPss 125 * Issuer: CN = localhost 126 * Validity Not Before: Jun 6 07:11:00 2018 GMT 127 * Not After : Jun 1 07:11:00 2038 GMT 128 * Subject: CN = localhost 129 * Public Key Algorithm: rsassaPss 130 */ 131 "-----BEGIN CERTIFICATE-----\n" 132 + "MIIDZjCCAh2gAwIBAgIUHxwPs3eAgJ057nJwiLgWZWeNqdgwPgYJKoZIhvcNAQEK\n" 133 + "MDGgDTALBglghkgBZQMEAgGhGjAYBgkqhkiG9w0BAQgwCwYJYIZIAWUDBAIBogQC\n" 134 + "AgDeMBQxEjAQBgNVBAMMCWxvY2FsaG9zdDAeFw0xODA2MDYwNzExMDBaFw0zODA2\n" 135 + "MDEwNzExMDBaMBQxEjAQBgNVBAMMCWxvY2FsaG9zdDCCASAwCwYJKoZIhvcNAQEK\n" 136 + "A4IBDwAwggEKAoIBAQCl8r4Qrg27BYUO/1Va2Ix8QPGzN/lvzmKvP5Ff26ovNW4v\n" 137 + "RUx68HzAhhiWtcl+PwLSbJqJreEkTlle7PnRAypby3fO7ZAK0Y3YiHquaBg7d+7Y\n" 138 + "FhhHwv8gG0lZcyA0BkXFJHqdq76qar0xHC6DVezXm0K3mcceymGtFR9BzWmAj+7D\n" 139 + "YsSwvtTQ7WNoQmf0cdDMSM71IwaTwIwvT2wzX1vv5hcdDyXdr64WFqWSA9sNJ2K6\n" 140 + "arxaaU1klwKSgDokF6njafWQ4UxdR67d5W1MYoiioDs2Yy3utsMpO2OUzZVBZNdT\n" 141 + "gkr1jsJhIurpz/5K51lwJIRQBezEFSb+60AFVoMJAgMBAAGjUDBOMB0GA1UdDgQW\n" 142 + "BBQfFit5ilWJmZgCX4QY0HsaI9iIDDAfBgNVHSMEGDAWgBQfFit5ilWJmZgCX4QY\n" 143 + "0HsaI9iIDDAMBgNVHRMEBTADAQH/MD4GCSqGSIb3DQEBCjAxoA0wCwYJYIZIAWUD\n" 144 + "BAIBoRowGAYJKoZIhvcNAQEIMAsGCWCGSAFlAwQCAaIEAgIA3gOCAQEAa4yUQ3gh\n" 145 + "d1YWPdEa1sv2hdkhtenw6m5yxbmaQl2+nIKSpk4RfpXC7K1EYwBF8TdfFbD8hGGh\n" 146 + "5n81BT0/dn1R9SRGCv7KTxx4lfQt31frlsw/tVciwyXQtcUZ6DqfnLP0/aRVLNgx\n" 147 + "zaP542JUHFYLTC3EGz2zUgv70ZUTlIsPG3/p8YO1iXdnYGQyzOuQPUBpI7nS7UtR\n" 148 + "Ug8VE9ACpBxxI3qChMahFZGHlXCCSjSmxpQa6UO4SQl8q5tPNnqdzWwvAW8qkCy4\n" 149 + "6barRQ4sMcGayhHh/uSTx7bcl0FMJpcI1ygbw7/Pc03zKtw0gMTBMns7q4yXjb/u\n" 150 + "ef47nW0t+LRAAg==\n" 151 + "-----END CERTIFICATE-----\n"; 152 153 private static final String keyCertStr = trustedCertStr; 154 155 private static final String privateKey = 156 "MIIEuwIBADALBgkqhkiG9w0BAQoEggSnMIIEowIBAAKCAQEApfK+EK4NuwWFDv9V\n" 157 + "WtiMfEDxszf5b85irz+RX9uqLzVuL0VMevB8wIYYlrXJfj8C0myaia3hJE5ZXuz5\n" 158 + "0QMqW8t3zu2QCtGN2Ih6rmgYO3fu2BYYR8L/IBtJWXMgNAZFxSR6nau+qmq9MRwu\n" 159 + "g1Xs15tCt5nHHsphrRUfQc1pgI/uw2LEsL7U0O1jaEJn9HHQzEjO9SMGk8CML09s\n" 160 + "M19b7+YXHQ8l3a+uFhalkgPbDSdiumq8WmlNZJcCkoA6JBep42n1kOFMXUeu3eVt\n" 161 + "TGKIoqA7NmMt7rbDKTtjlM2VQWTXU4JK9Y7CYSLq6c/+SudZcCSEUAXsxBUm/utA\n" 162 + "BVaDCQIDAQABAoIBAAc4vRS0vlw5LUUtz2UYr2Ro3xvRf8Vh0eGWfpkRUiKjzJu6\n" 163 + "BE4FUSh/rWpBlvcrfs/xcfgz3OxbjIAZB/YUkS9Vd21F4VLXM7kMl2onlYZg/b/h\n" 164 + "lkTpM3kONu7xl6Er9LVTlRJveuinpHwSoeONRbVMSGb9BjFM1VtW4/lVGxZBG05D\n" 165 + "y9i/o4vCZqULn9cAumOwicKuCyTcS58XcMJ+puSPfRA71PYLxqFkASAoJsUwCXpo\n" 166 + "gs39lLsIFgrfO8mBO1ux/SE+QaRc+9XqFSHHKD1XqF/9zSYBgWjE910EcpdYEdZx\n" 167 + "GEkwea7Fn4brO5OpIrHY/45naqbUOBzv6gufMAECgYEAz7PHCdcrQvmOb8EiNbQH\n" 168 + "uvSimwObWJFeN1ykp6mfRbSnkXw7p8+M4Tc8HFi8QLpoq63Ev2AwoaQCQvHbFC2Y\n" 169 + "1Cz0EkC0aOp+tZP7U2AUBdkcDesZAJQTad0zV6KesyIUXdxZXDG8JJ1XSNWfTJV4\n" 170 + "QD+BjLZ0jiAyCIfVYvWQqYkCgYEAzIln1nKTixLMPr5CldSmR7ZarEtPJU+hHwVg\n" 171 + "dV/Lc6d2Yy9JgunOXRo4BXB1TEo8JFbK3HBQH6tS8li4qDr7WK5wyYfh8qb4WZyu\n" 172 + "lc562f2WVYntcN8/Ojb+Vyrt7lk9sq/8KoVHxEAWd6mqL9VTPYuAu1Vw9fTGIZfB\n" 173 + "lDeELYECgYAvdzU4UXzofGGJtohb332YwwlaBZP9xJLUcg6K5l+orWVSASMc8XiP\n" 174 + "i3DoRXsYC8GZ4kdBOPlEJ1gA9oaLcPQpIPDSLwlLpLM6Scw4vI822uvnXl/DWxOo\n" 175 + "sM1n7Jj59QLUhGPDhvYpI+/rjC4wcUQe4qR3hMbUKBVnD6u7RsU9iQKBgQCQ17VK\n" 176 + "7bSCRfuRaxaoGADww7gOTv5rQ6qr1xjpxb7D1hFGR9Rc+smCsPB/GZZXQjK44SWj\n" 177 + "WX3ED4Ubzaxmpe4cbNu+O5XMSmWQwB36RFBHUwdE5/nXdqDFzu/qNqJrqZLBmVKP\n" 178 + "ofaiiWffsaytVvotmT6+atElvAMbAua42V+nAQKBgHtIn3mYMHLriYGhQzpkFEA2\n" 179 + "8YcAMlKppueOMAKVy8nLu2r3MidmLAhMiKJQKG45I3Yg0/t/25tXLiOPJlwrOebh\n" 180 + "xQqUBI/JUOIpGAEnr48jhOXnCS+i+z294G5U/RgjXrlR4bCPvrtCmwzWwe0h79w2\n" 181 + "Q2hO5ZTW6UD9CVA85whf"; 182 183 private static SSLContext createSSLContext() throws Exception { 184 // Generate certificate from cert string 185 CertificateFactory cf = CertificateFactory.getInstance("X.509"); 186 187 // Create a key store 188 KeyStore ts = KeyStore.getInstance("PKCS12"); 189 KeyStore ks = KeyStore.getInstance("PKCS12"); 190 ts.load(null, null); 191 ks.load(null, null); 192 char passphrase[] = "passphrase".toCharArray(); 193 194 // Import the trusted cert 195 ts.setCertificateEntry("trusted-cert-RSASSA-PSS", 196 cf.generateCertificate(new ByteArrayInputStream( 197 trustedCertStr.getBytes()))); 198 199 boolean hasKeyMaterials = keyCertStr != null && privateKey != null; 200 if (hasKeyMaterials) { 201 202 // Generate the private key. 203 PKCS8EncodedKeySpec priKeySpec = new PKCS8EncodedKeySpec( 204 Base64.getMimeDecoder().decode(privateKey)); 205 KeyFactory kf = KeyFactory.getInstance("RSASSA-PSS"); 206 PrivateKey priKey = kf.generatePrivate(priKeySpec); 207 208 // Generate certificate chain 209 Certificate keyCert = cf.generateCertificate( 210 new ByteArrayInputStream(keyCertStr.getBytes())); 211 Certificate[] chain = new Certificate[]{keyCert}; 212 213 // Import the key entry. 214 ks.setKeyEntry("cert-RSASSA-PSS", priKey, passphrase, chain); 215 } 216 217 // Create SSL context 218 TrustManagerFactory tmf = TrustManagerFactory.getInstance("PKIX"); 219 tmf.init(ts); 220 221 SSLContext context = SSLContext.getInstance("TLS"); 222 if (hasKeyMaterials) { 223 KeyManagerFactory kmf = KeyManagerFactory.getInstance("NewSunX509"); 224 kmf.init(ks, passphrase); 225 context.init(kmf.getKeyManagers(), tmf.getTrustManagers(), null); 226 } else { 227 context.init(null, tmf.getTrustManagers(), null); 228 } 229 230 return context; 231 } 232 }