Home | History | Annotate | Download | only in ssl
      1 /*
      2  *  Licensed to the Apache Software Foundation (ASF) under one or more
      3  *  contributor license agreements.  See the NOTICE file distributed with
      4  *  this work for additional information regarding copyright ownership.
      5  *  The ASF licenses this file to You under the Apache License, Version 2.0
      6  *  (the "License"); you may not use this file except in compliance with
      7  *  the License.  You may obtain a copy of the License at
      8  *
      9  *     http://www.apache.org/licenses/LICENSE-2.0
     10  *
     11  *  Unless required by applicable law or agreed to in writing, software
     12  *  distributed under the License is distributed on an "AS IS" BASIS,
     13  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     14  *  See the License for the specific language governing permissions and
     15  *  limitations under the License.
     16  */
     17 
     18 package tests.api.javax.net.ssl;
     19 
     20 import java.io.ByteArrayInputStream;
     21 import java.io.InputStream;
     22 import java.io.OutputStream;
     23 import java.net.InetAddress;
     24 import java.net.InetSocketAddress;
     25 import java.security.KeyStore;
     26 import java.security.Principal;
     27 import java.security.cert.Certificate;
     28 import java.security.cert.X509Certificate;
     29 import javax.net.ssl.KeyManager;
     30 import javax.net.ssl.KeyManagerFactory;
     31 import javax.net.ssl.SSLContext;
     32 import javax.net.ssl.SSLServerSocket;
     33 import javax.net.ssl.SSLSession;
     34 import javax.net.ssl.SSLSessionBindingEvent;
     35 import javax.net.ssl.SSLSessionBindingListener;
     36 import javax.net.ssl.SSLSocket;
     37 import javax.net.ssl.TrustManager;
     38 import junit.framework.TestCase;
     39 import org.apache.harmony.luni.util.Base64;
     40 import tests.api.javax.net.ssl.HandshakeCompletedEventTest.MyHandshakeListener;
     41 import tests.api.javax.net.ssl.HandshakeCompletedEventTest.TestTrustManager;
     42 import tests.support.Support_PortManager;
     43 
     44 /**
     45  * Tests for SSLSession class
     46  *
     47  */
     48 public class SSLSessionTest extends TestCase {
     49 
     50     // set to true if on Android, false if on RI
     51     boolean useBKS = true;
     52 
     53     /**
     54      * javax.net.ssl.SSLSession#getPeerHost()
     55      * javax.net.ssl.SSLSession#getPeerPort()
     56      */
     57     public void test_getPeerHost() throws Exception {
     58         SSLSession s = clientSession;
     59         assertEquals(InetAddress.getLocalHost().getHostName(), s.getPeerHost());
     60         assertEquals(port, s.getPeerPort());
     61     }
     62 
     63     /**
     64      * javax.net.ssl.SSLSession#invalidate()
     65      * javax.net.ssl.SSLSession#isValid()
     66      */
     67     public void test_invalidate() {
     68         SSLSession s = clientSession;
     69         assertTrue(s.isValid());
     70         s.invalidate();
     71         assertFalse(s.isValid());
     72     }
     73 
     74     /**
     75      * javax.net.ssl.SSLSession#getPeerPrincipal()
     76      */
     77     public void test_getPeerPrincipal() throws Exception {
     78         Principal p1 = clientSession.getPeerPrincipal();
     79         KeyStore store = server.getStore();
     80         X509Certificate cert = (X509Certificate)store.getCertificate("mykey");
     81         Principal p2 = cert.getSubjectX500Principal();
     82         assertEquals(p1, p2);
     83     }
     84 
     85     /**
     86      * javax.net.ssl.SSLSession#getApplicationBufferSize()
     87      */
     88     public void test_getApplicationBufferSize() {
     89         assertTrue(clientSession.getApplicationBufferSize() > 0);
     90     }
     91 
     92     /**
     93      * javax.net.ssl.SSLSession#getCipherSuite()
     94      */
     95     public void test_getCipherSuite() {
     96         assertEquals(CIPHER_SUITE, clientSession.getCipherSuite());
     97     }
     98 
     99     /**
    100      * javax.net.ssl.SSLSession#getCreationTime()
    101      */
    102     public void test_getCreationTime() {
    103         // check if creation time was in the last 10 seconds
    104         long currentTime = System.currentTimeMillis();
    105         long sessionTime = clientSession.getCreationTime();
    106         long diff = currentTime - sessionTime;
    107         assertTrue("diff between " + currentTime + " and " + sessionTime + " should be < 10000",
    108                    diff < 10000);
    109     }
    110 
    111     /**
    112      * javax.net.ssl.SSLSession#getId()
    113      */
    114     public void test_getId() {
    115         byte[] id = clientSession.getId();
    116         SSLSession sess = clientSslContext.getClientSessionContext().getSession(id);
    117         assertNotNull("Could not find session for id " + id, sess);
    118         assertEquals(clientSession, sess);
    119     }
    120 
    121     /**
    122      * javax.net.ssl.SSLSession#getLastAccessedTime()
    123      */
    124     public void test_getLastAccessedTime() {
    125         // check if last access time was in the last 10 seconds
    126         long currentTime = System.currentTimeMillis();
    127         long sessionTime = clientSession.getLastAccessedTime();
    128         long diff = currentTime - sessionTime;
    129         assertTrue("diff between " + currentTime + " and " + sessionTime + " should be < 10000",
    130                    diff < 10000);
    131         assertTrue ("diff should be < 10000 but is " + diff, diff < 10000);
    132     }
    133 
    134     /**
    135      * javax.net.ssl.SSLSession#getLocalCertificates()
    136      */
    137     public void test_getLocalCertificates() throws Exception {
    138         KeyStore store = client.getStore();
    139         Certificate cert = store.getCertificate("mykey");
    140         Certificate[] certs = clientSession.getLocalCertificates();
    141         assertEquals(cert, certs[0]);
    142     }
    143 
    144     /**
    145      * javax.net.ssl.SSLSession#getLocalPrincipal()
    146      */
    147     public void test_getLocalPrincipal() throws Exception {
    148         Principal p1 = clientSession.getLocalPrincipal();
    149         KeyStore store = client.getStore();
    150         X509Certificate cert = (X509Certificate)store.getCertificate("mykey");
    151         Principal p2 = cert.getSubjectX500Principal();
    152         assertEquals(p1, p2);
    153     }
    154 
    155     /**
    156      * javax.net.ssl.SSLSession#getPacketBufferSize()
    157      */
    158     public void test_getPacketBufferSize() {
    159         assertTrue(clientSession.getPacketBufferSize() > 0);
    160     }
    161 
    162     /**
    163      * javax.net.ssl.SSLSession#getPeerCertificates()
    164      */
    165     public void test_getPeerCertificates() throws Exception {
    166         Certificate[] res = clientSession.getPeerCertificates();
    167         assertTrue(res.length > 0);
    168     }
    169 
    170     /**
    171      * javax.net.ssl.SSLSession#getPeerCertificateChain()
    172      */
    173     public void test_getPeerCertificateChain() throws Exception {
    174         javax.security.cert.X509Certificate[] res = clientSession.getPeerCertificateChain();
    175         assertTrue(res.length > 0);
    176     }
    177 
    178     /**
    179      * javax.net.ssl.SSLSession#getProtocol()
    180      */
    181     public void test_getProtocol() {
    182         assertEquals("TLSv1", clientSession.getProtocol());
    183     }
    184 
    185     /**
    186      * javax.net.ssl.SSLSession#getSessionContext()
    187      */
    188     public void test_getSessionContext() {
    189         assertEquals(clientSession.getSessionContext(),
    190                      clientSslContext.getClientSessionContext());
    191     }
    192 
    193     /**
    194      * javax.net.ssl.SSLSession#putValue(String name, Object value)
    195      * javax.net.ssl.SSLSession#removeValue(String name)
    196      * javax.net.ssl.SSLSession#getValueNames()
    197      */
    198     public void test_putValue() {
    199         SSLSession s = clientSession;
    200         mySSLSessionBindingListener sbl = new mySSLSessionBindingListener();
    201         assertNotNull(s.getValueNames());
    202         assertEquals(0, s.getValueNames().length);
    203         s.putValue("Name_01", sbl);
    204         s.putValue("Name_02", sbl);
    205         s.putValue("Name_03", sbl);
    206         assertEquals(3, s.getValueNames().length);
    207         s.removeValue("Name_01");
    208         assertEquals(2, s.getValueNames().length);
    209 
    210         try {
    211             s.putValue(null, null);
    212             fail("IllegalArgumentException wasn't thrown");
    213         } catch (IllegalArgumentException expected) {
    214             // expected
    215         }
    216         try {
    217             s.putValue("ABC", null);
    218             fail("IllegalArgumentException wasn't thrown");
    219         } catch (IllegalArgumentException expected) {
    220             // expected
    221         }
    222         try {
    223             s.putValue(null, sbl);
    224             fail("IllegalArgumentException wasn't thrown");
    225         } catch (IllegalArgumentException expected) {
    226             // expected
    227         }
    228 
    229         try {
    230             s.removeValue(null);
    231             fail("IllegalArgumentException wasn't thrown");
    232         } catch (IllegalArgumentException expected) {
    233             // expected
    234         }
    235     }
    236 
    237     /**
    238      * javax.net.ssl.SSLSession#getValue(String name)
    239      */
    240     public void test_getValue() {
    241         SSLSession s = clientSession;
    242         mySSLSessionBindingListener sbl = new mySSLSessionBindingListener();
    243 
    244         try {
    245             s.getValue(null);
    246             fail("IllegalArgumentException wasn't thrown");
    247         } catch (IllegalArgumentException expected) {
    248             // expected
    249         }
    250 
    251         s.putValue("Name", sbl);
    252         Object obj = s.getValue("Name");
    253         assertTrue(obj instanceof SSLSessionBindingListener);
    254     }
    255 
    256     Thread serverThread, clientThread;
    257     TestServer server;
    258     TestClient client;
    259 
    260     @Override
    261     protected void setUp() {
    262         port = Support_PortManager.getNextPort();
    263         String serverKeys = (useBKS ? SERVER_KEYS_BKS : SERVER_KEYS_JKS);
    264         String clientKeys = (useBKS ? CLIENT_KEYS_BKS : CLIENT_KEYS_JKS);
    265         server = new TestServer(true,
    266                 TestServer.CLIENT_AUTH_WANTED, serverKeys);
    267         client = new TestClient(true, clientKeys);
    268 
    269         serverThread = new Thread(server);
    270         clientThread = new Thread(client);
    271 
    272         serverThread.start();
    273         try {
    274             Thread.currentThread().sleep(1000);
    275             clientThread.start();
    276         } catch (InterruptedException e) {
    277             fail("Could not create server or cient " + e.getMessage());
    278         }
    279         while (clientSession == null
    280                 && server.exception == null
    281                 && client.exception == null) {
    282             try {
    283                 Thread.currentThread().sleep(500);
    284             } catch (InterruptedException e) {
    285                 fail("couldn't create session");
    286             }
    287         }
    288         if (server.exception != null) {
    289             server.exception.printStackTrace();
    290         }
    291         assertNull("server thread has a pending exception: " + server.exception,
    292                 server.exception);
    293         if (client.exception != null) {
    294             client.exception.printStackTrace();
    295         }
    296         assertNull("client thread has a pending exception: " + client.exception,
    297                 client.exception);
    298         assertNotNull("Could not initialize session", clientSession);
    299     }
    300 
    301     @Override
    302     protected void tearDown() {
    303         notFinished = false;
    304         try {
    305             serverThread.join();
    306         } catch (InterruptedException e) {
    307             throw new RuntimeException(e);
    308         }
    309         try {
    310             clientThread.join();
    311         } catch (InterruptedException e) {
    312             throw new RuntimeException(e);
    313         }
    314 
    315         // The server must have completed without an exception.
    316         if (server.getException() != null) {
    317             throw new RuntimeException(server.getException());
    318         }
    319 
    320         // The client must have completed without an exception.
    321         if (client.getException() != null) {
    322             throw new RuntimeException(client.getException());
    323         }
    324     }
    325 
    326     public class mySSLSessionBindingListener implements
    327             SSLSessionBindingListener {
    328         mySSLSessionBindingListener() {
    329         }
    330         public void valueBound(SSLSessionBindingEvent event) {}
    331         public void valueUnbound(SSLSessionBindingEvent event) {}
    332     }
    333 
    334     /**
    335      * Defines the keystore contents for the server, BKS version. Holds just a
    336      * single self-generated key. The subject name is "Test Server".
    337      */
    338     private static final String SERVER_KEYS_BKS =
    339             "AAAAAQAAABQDkebzoP1XwqyWKRCJEpn/t8dqIQAABDkEAAVteWtleQAAARpYl20nAAAAAQAFWC41"
    340             + "MDkAAAJNMIICSTCCAbKgAwIBAgIESEfU1jANBgkqhkiG9w0BAQUFADBpMQswCQYDVQQGEwJVUzET"
    341             + "MBEGA1UECBMKQ2FsaWZvcm5pYTEMMAoGA1UEBxMDTVRWMQ8wDQYDVQQKEwZHb29nbGUxEDAOBgNV"
    342             + "BAsTB0FuZHJvaWQxFDASBgNVBAMTC1Rlc3QgU2VydmVyMB4XDTA4MDYwNTExNTgxNFoXDTA4MDkw"
    343             + "MzExNTgxNFowaTELMAkGA1UEBhMCVVMxEzARBgNVBAgTCkNhbGlmb3JuaWExDDAKBgNVBAcTA01U"
    344             + "VjEPMA0GA1UEChMGR29vZ2xlMRAwDgYDVQQLEwdBbmRyb2lkMRQwEgYDVQQDEwtUZXN0IFNlcnZl"
    345             + "cjCBnzANBgkqhkiG9w0BAQEFAAOBjQAwgYkCgYEA0LIdKaIr9/vsTq8BZlA3R+NFWRaH4lGsTAQy"
    346             + "DPMF9ZqEDOaL6DJuu0colSBBBQ85hQTPa9m9nyJoN3pEi1hgamqOvQIWcXBk+SOpUGRZZFXwniJV"
    347             + "zDKU5nE9MYgn2B9AoiH3CSuMz6HRqgVaqtppIe1jhukMc/kHVJvlKRNy9XMCAwEAATANBgkqhkiG"
    348             + "9w0BAQUFAAOBgQC7yBmJ9O/eWDGtSH9BH0R3dh2NdST3W9hNZ8hIa8U8klhNHbUCSSktZmZkvbPU"
    349             + "hse5LI3dh6RyNDuqDrbYwcqzKbFJaq/jX9kCoeb3vgbQElMRX8D2ID1vRjxwlALFISrtaN4VpWzV"
    350             + "yeoHPW4xldeZmoVtjn8zXNzQhLuBqX2MmAAAAqwAAAAUvkUScfw9yCSmALruURNmtBai7kQAAAZx"
    351             + "4Jmijxs/l8EBaleaUru6EOPioWkUAEVWCxjM/TxbGHOi2VMsQWqRr/DZ3wsDmtQgw3QTrUK666sR"
    352             + "MBnbqdnyCyvM1J2V1xxLXPUeRBmR2CXorYGF9Dye7NkgVdfA+9g9L/0Au6Ugn+2Cj5leoIgkgApN"
    353             + "vuEcZegFlNOUPVEs3SlBgUF1BY6OBM0UBHTPwGGxFBBcetcuMRbUnu65vyDG0pslT59qpaR0TMVs"
    354             + "P+tcheEzhyjbfM32/vwhnL9dBEgM8qMt0sqF6itNOQU/F4WGkK2Cm2v4CYEyKYw325fEhzTXosck"
    355             + "MhbqmcyLab8EPceWF3dweoUT76+jEZx8lV2dapR+CmczQI43tV9btsd1xiBbBHAKvymm9Ep9bPzM"
    356             + "J0MQi+OtURL9Lxke/70/MRueqbPeUlOaGvANTmXQD2OnW7PISwJ9lpeLfTG0LcqkoqkbtLKQLYHI"
    357             + "rQfV5j0j+wmvmpMxzjN3uvNajLa4zQ8l0Eok9SFaRr2RL0gN8Q2JegfOL4pUiHPsh64WWya2NB7f"
    358             + "V+1s65eA5ospXYsShRjo046QhGTmymwXXzdzuxu8IlnTEont6P4+J+GsWk6cldGbl20hctuUKzyx"
    359             + "OptjEPOKejV60iDCYGmHbCWAzQ8h5MILV82IclzNViZmzAapeeCnexhpXhWTs+xDEYSKEiG/camt"
    360             + "bhmZc3BcyVJrW23PktSfpBQ6D8ZxoMfF0L7V2GQMaUg+3r7ucrx82kpqotjv0xHghNIm95aBr1Qw"
    361             + "1gaEjsC/0wGmmBDg1dTDH+F1p9TInzr3EFuYD0YiQ7YlAHq3cPuyGoLXJ5dXYuSBfhDXJSeddUkl"
    362             + "k1ufZyOOcskeInQge7jzaRfmKg3U94r+spMEvb0AzDQVOKvjjo1ivxMSgFRZaDb/4qw=";
    363 
    364     /**
    365      * Defines the keystore contents for the client, BKS version. Holds just a
    366      * single self-generated key. The subject name is "Test Client".
    367      */
    368     private static final String CLIENT_KEYS_BKS =
    369             "AAAAAQAAABT4Rka6fxbFps98Y5k2VilmbibNkQAABfQEAAVteWtleQAAARpYl+POAAAAAQAFWC41"
    370             + "MDkAAAJNMIICSTCCAbKgAwIBAgIESEfU9TANBgkqhkiG9w0BAQUFADBpMQswCQYDVQQGEwJVUzET"
    371             + "MBEGA1UECBMKQ2FsaWZvcm5pYTEMMAoGA1UEBxMDTVRWMQ8wDQYDVQQKEwZHb29nbGUxEDAOBgNV"
    372             + "BAsTB0FuZHJvaWQxFDASBgNVBAMTC1Rlc3QgQ2xpZW50MB4XDTA4MDYwNTExNTg0NVoXDTA4MDkw"
    373             + "MzExNTg0NVowaTELMAkGA1UEBhMCVVMxEzARBgNVBAgTCkNhbGlmb3JuaWExDDAKBgNVBAcTA01U"
    374             + "VjEPMA0GA1UEChMGR29vZ2xlMRAwDgYDVQQLEwdBbmRyb2lkMRQwEgYDVQQDEwtUZXN0IENsaWVu"
    375             + "dDCBnzANBgkqhkiG9w0BAQEFAAOBjQAwgYkCgYEApUvmWsQDHPpbDKK13Yez2/q54tTOmRml/qva"
    376             + "2K6dZjkjSTW0iRuk7ztaVEvdJpfVIDv1oBsCI51ttyLHROy1epjF+GoL74mJb7fkcd0VOoSOTjtD"
    377             + "+3GgZkHPAm5YmUYxiJXqxKKJJqMCTIW46eJaA2nAep9QIwZ14/NFAs4ObV8CAwEAATANBgkqhkiG"
    378             + "9w0BAQUFAAOBgQCJrCr3hZQFDlLIfsSKI1/w+BLvyf4fubOid0pBxfklR8KBNPTiqjSmu7pd/C/F"
    379             + "1FR8CdZUDoPflZHCOU+fj5r5KUC1HyigY/tEUvlforBpfB0uCF+tXW4DbUfOWhfMtLV4nCOJOOZg"
    380             + "awfZLJWBJouLKOp427vDftxTSB+Ks8YjlgAAAqwAAAAU+NH6TtrzjyDdCXm5B6Vo7xX5G4YAAAZx"
    381             + "EAUkcZtmykn7YdaYxC1jRFJ+GEJpC8nZVg83QClVuCSIS8a5f8Hl44Bk4oepOZsPzhtz3RdVzDVi"
    382             + "RFfoyZFsrk9F5bDTVJ6sQbb/1nfJkLhZFXokka0vND5AXMSoD5Bj1Fqem3cK7fSUyqKvFoRKC3XD"
    383             + "FQvhqoam29F1rbl8FaYdPvhhZo8TfZQYUyUKwW+RbR44M5iHPx+ykieMe/C/4bcM3z8cwIbYI1aO"
    384             + "gjQKS2MK9bs17xaDzeAh4sBKrskFGrDe+2dgvrSKdoakJhLTNTBSG6m+rzqMSCeQpafLKMSjTSSz"
    385             + "+KoQ9bLyax8cbvViGGju0SlVhquloZmKOfHr8TukIoV64h3uCGFOVFtQjCYDOq6NbfRvMh14UVF5"
    386             + "zgDIGczoD9dMoULWxBmniGSntoNgZM+QP6Id7DBasZGKfrHIAw3lHBqcvB5smemSu7F4itRoa3D8"
    387             + "N7hhUEKAc+xA+8NKmXfiCBoHfPHTwDvt4IR7gWjeP3Xv5vitcKQ/MAfO5RwfzkYCXQ3FfjfzmsE1"
    388             + "1IfLRDiBj+lhQSulhRVStKI88Che3M4JUNGKllrc0nt1pWa1vgzmUhhC4LSdm6trTHgyJnB6OcS9"
    389             + "t2furYjK88j1AuB4921oxMxRm8c4Crq8Pyuf+n3YKi8Pl2BzBtw++0gj0ODlgwut8SrVj66/nvIB"
    390             + "jN3kLVahR8nZrEFF6vTTmyXi761pzq9yOVqI57wJGx8o3Ygox1p+pWUPl1hQR7rrhUbgK/Q5wno9"
    391             + "uJk07h3IZnNxE+/IKgeMTP/H4+jmyT4mhsexJ2BFHeiKF1KT/FMcJdSi+ZK5yoNVcYuY8aZbx0Ef"
    392             + "lHorCXAmLFB0W6Cz4KPP01nD9YBB4olxiK1t7m0AU9zscdivNiuUaB5OIEr+JuZ6dNw=";
    393 
    394     private static final String CIPHER_SUITE = "SSL_RSA_WITH_RC4_128_MD5";
    395     /**
    396      * Defines the keystore contents for the server, JKS version. Holds just a
    397      * single self-generated key. The subject name is "Test Server".
    398      */
    399     private static final String SERVER_KEYS_JKS =
    400             "/u3+7QAAAAIAAAABAAAAAQAFbXlrZXkAAAEaWFfBeAAAArowggK2MA4GCisGAQQBKgIRAQEFAASC"
    401             + "AqI2kp5XjnF8YZkhcF92YsJNQkvsmH7zqMM87j23zSoV4DwyE3XeC/gZWq1ToScIhoqZkzlbWcu4"
    402             + "T/Zfc/DrfGk/rKbBL1uWKGZ8fMtlZk8KoAhxZk1JSyJvdkyKxqmzUbxk1OFMlN2VJNu97FPVH+du"
    403             + "dvjTvmpdoM81INWBW/1fZJeQeDvn4mMbbe0IxgpiLnI9WSevlaDP/sm1X3iO9yEyzHLL+M5Erspo"
    404             + "Cwa558fOu5DdsICMXhvDQxjWFKFhPHnKtGe+VvwkG9/bAaDgx3kfhk0w5zvdnkKb+8Ed9ylNRzdk"
    405             + "ocAa/mxlMTOsTvDKXjjsBupNPIIj7OP4GNnZaxkJjSs98pEO67op1GX2qhy6FSOPNuq8k/65HzUc"
    406             + "PYn6voEeh6vm02U/sjEnzRevQ2+2wXoAdp0EwtQ/DlMe+NvcwPGWKuMgX4A4L93DZGb04N2VmAU3"
    407             + "YLOtZwTO0LbuWrcCM/q99G/7LcczkxIVrO2I/rh8RXVczlf9QzcrFObFv4ATuspWJ8xG7DhsMbnk"
    408             + "rT94Pq6TogYeoz8o8ZMykesAqN6mt/9+ToIemmXv+e+KU1hI5oLwWMnUG6dXM6hIvrULY6o+QCPH"
    409             + "172YQJMa+68HAeS+itBTAF4Clm/bLn6reHCGGU6vNdwU0lYldpiOj9cB3t+u2UuLo6tiFWjLf5Zs"
    410             + "EQJETd4g/EK9nHxJn0GAKrWnTw7pEHQJ08elzUuy04C/jEEG+4QXU1InzS4o/kR0Sqz2WTGDoSoq"
    411             + "ewuPRU5bzQs/b9daq3mXrnPtRBL6HfSDAdpTK76iHqLCGdqx3avHjVSBm4zFvEuYBCev+3iKOBmg"
    412             + "yh7eQRTjz4UOWfy85omMBr7lK8PtfVBDzOXpasxS0uBgdUyBDX4tO6k9jZ8a1kmQRQAAAAEABVgu"
    413             + "NTA5AAACSDCCAkQwggGtAgRIR8SKMA0GCSqGSIb3DQEBBAUAMGkxCzAJBgNVBAYTAlVTMRMwEQYD"
    414             + "VQQIEwpDYWxpZm9ybmlhMQwwCgYDVQQHEwNNVFYxDzANBgNVBAoTBkdvb2dsZTEQMA4GA1UECxMH"
    415             + "QW5kcm9pZDEUMBIGA1UEAxMLVGVzdCBTZXJ2ZXIwHhcNMDgwNjA1MTA0ODQyWhcNMDgwOTAzMTA0"
    416             + "ODQyWjBpMQswCQYDVQQGEwJVUzETMBEGA1UECBMKQ2FsaWZvcm5pYTEMMAoGA1UEBxMDTVRWMQ8w"
    417             + "DQYDVQQKEwZHb29nbGUxEDAOBgNVBAsTB0FuZHJvaWQxFDASBgNVBAMTC1Rlc3QgU2VydmVyMIGf"
    418             + "MA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCwoC6chqCI84rj1PrXuJgbiit4EV909zR6N0jNlYfg"
    419             + "itwB39bP39wH03rFm8T59b3mbSptnGmCIpLZn25KPPFsYD3JJ+wFlmiUdEP9H05flfwtFQJnw9uT"
    420             + "3rRIdYVMPcQ3RoZzwAMliGr882I2thIDbA6xjGU/1nRIdvk0LtxH3QIDAQABMA0GCSqGSIb3DQEB"
    421             + "BAUAA4GBAJn+6YgUlY18Ie+0+Vt8oEi81DNi/bfPrAUAh63fhhBikx/3R9dl3wh09Z6p7cIdNxjW"
    422             + "n2ll+cRW9eqF7z75F0Omm0C7/KAEPjukVbszmzeU5VqzkpSt0j84YWi+TfcHRrfvhLbrlmGITVpY"
    423             + "ol5pHLDyqGmDs53pgwipWqsn/nEXEBgj3EoqPeqHbDf7YaP8h/5BSt0=";
    424 
    425     /**
    426      * Defines the keystore contents for the client, JKS version. Holds just a
    427      * single self-generated key. The subject name is "Test Client".
    428      */
    429     private static final String CLIENT_KEYS_JKS =
    430             "/u3+7QAAAAIAAAABAAAAAQAFbXlrZXkAAAEaWFhyMAAAArkwggK1MA4GCisGAQQBKgIRAQEFAASC"
    431             + "AqGVSfXolBStZy4nnRNn4fAr+S7kfU2BS23wwW8uB2Ru3GvtLzlK9q08Gvq/LNqBafjyFTVL5FV5"
    432             + "SED/8YomO5a98GpskSeRvytCiTBLJdgGhws5TOGekgIAcBROPGIyOtJPQ0HfOQs+BqgzGDHzHQhw"
    433             + "u/8Tm6yQwiP+W/1I9B1QnaEztZA3mhTyMMJsmsFTYroGgAog885D5Cmzd8sYGfxec3R6I+xcmBAY"
    434             + "eibR5kGpWwt1R+qMvRrtBqh5r6WSKhCBNax+SJVbtUNRiKyjKccdJg6fGqIWWeivwYTy0OhjA6b4"
    435             + "NiZ/ZZs5pxFGWUj/Rlp0RYy8fCF6aw5/5s4Bf4MI6dPSqMG8Hf7sJR91GbcELyzPdM0h5lNavgit"
    436             + "QPEzKeuDrGxhY1frJThBsNsS0gxeu+OgfJPEb/H4lpYX5IvuIGbWKcxoO9zq4/fimIZkdA8A+3eY"
    437             + "mfDaowvy65NBVQPJSxaOyFhLHfeLqOeCsVENAea02vA7andZHTZehvcrqyKtm+z8ncHGRC2H9H8O"
    438             + "jKwKHfxxrYY/jMAKLl00+PBb3kspO+BHI2EcQnQuMw/zr83OR9Meq4TJ0TMuNkApZELAeFckIBbS"
    439             + "rBr8NNjAIfjuCTuKHhsTFWiHfk9ZIzigxXagfeDRiyVc6khOuF/bGorj23N2o7Rf3uLoU6PyXWi4"
    440             + "uhctR1aL6NzxDoK2PbYCeA9hxbDv8emaVPIzlVwpPK3Ruvv9mkjcOhZ74J8bPK2fQmbplbOljcZi"
    441             + "tZijOfzcO/11JrwhuJZRA6wanTqHoujgChV9EukVrmbWGGAcewFnAsSbFXIik7/+QznXaDIt5NgL"
    442             + "H/Bcz4Z/fdV7Ae1eUaxKXdPbI//4J+8liVT/d8awjW2tldIaDlmGMR3aoc830+3mAAAAAQAFWC41"
    443             + "MDkAAAJIMIICRDCCAa0CBEhHxLgwDQYJKoZIhvcNAQEEBQAwaTELMAkGA1UEBhMCVVMxEzARBgNV"
    444             + "BAgTCkNhbGlmb3JuaWExDDAKBgNVBAcTA01UVjEPMA0GA1UEChMGR29vZ2xlMRAwDgYDVQQLEwdB"
    445             + "bmRyb2lkMRQwEgYDVQQDEwtUZXN0IENsaWVudDAeFw0wODA2MDUxMDQ5MjhaFw0wODA5MDMxMDQ5"
    446             + "MjhaMGkxCzAJBgNVBAYTAlVTMRMwEQYDVQQIEwpDYWxpZm9ybmlhMQwwCgYDVQQHEwNNVFYxDzAN"
    447             + "BgNVBAoTBkdvb2dsZTEQMA4GA1UECxMHQW5kcm9pZDEUMBIGA1UEAxMLVGVzdCBDbGllbnQwgZ8w"
    448             + "DQYJKoZIhvcNAQEBBQADgY0AMIGJAoGBAIK3Q+KiFbmCGg422TAo4gggdhMH6FJhiuz8DxRyeMKR"
    449             + "UAfP4MK0wtc8N42waZ6OKvxpBFUy0BRfBsX0GD4Ku99yu9/tavSigTraeJtwV3WWRRjIqk7L3wX5"
    450             + "cmgS2KSD43Y0rNUKrko26lnt9N4qiYRBSj+tcAN3Lx9+ptqk1LApAgMBAAEwDQYJKoZIhvcNAQEE"
    451             + "BQADgYEANb7Q1GVSuy1RPJ0FmiXoMYCCtvlRLkmJphwxovK0cAQK12Vll+yAzBhHiQHy/RA11mng"
    452             + "wYudC7u3P8X/tBT8GR1Yk7QW3KgFyPafp3lQBBCraSsfrjKj+dCLig1uBLUr4f68W8VFWZWWTHqp"
    453             + "NMGpCX6qmjbkJQLVK/Yfo1ePaUexPSOX0G9m8+DoV3iyNw6at01NRw==";
    454 
    455 
    456     int port;
    457     SSLSocket serverSocket;
    458     MyHandshakeListener listener;
    459     String host = "localhost";
    460     boolean notFinished = true;
    461     SSLSession clientSession = null;
    462     SSLContext clientSslContext = null;
    463     String testData = "PING";
    464 
    465     private String PASSWORD = "android";
    466 
    467     /**
    468      * Implements a test SSL socket server. It waits for a connection on a given
    469      * port, requests client authentication (if specified), reads from the socket,
    470      * and writes to the socket.
    471      */
    472     class TestServer implements Runnable {
    473 
    474         public static final int CLIENT_AUTH_NONE = 0;
    475 
    476         public static final int CLIENT_AUTH_WANTED = 1;
    477 
    478         public static final int CLIENT_AUTH_NEEDED = 2;
    479 
    480         private TestTrustManager trustManager;
    481 
    482         private Exception exception;
    483 
    484         String keys;
    485 
    486         private int clientAuth;
    487 
    488         private boolean provideKeys;
    489 
    490         private KeyStore store;
    491 
    492         public TestServer(boolean provideKeys, int clientAuth, String keys) {
    493             this.keys = keys;
    494             this.clientAuth = clientAuth;
    495             this.provideKeys = provideKeys;
    496 
    497             trustManager = new TestTrustManager();
    498         }
    499 
    500         public void run() {
    501             try {
    502                 store = provideKeys ? getKeyStore(keys) : null;
    503                 KeyManager[] keyManagers = store != null ? getKeyManagers(store) : null;
    504                 TrustManager[] trustManagers = new TrustManager[] { trustManager };
    505 
    506                 SSLContext sslContext = SSLContext.getInstance("TLS");
    507                 sslContext.init(keyManagers, trustManagers, null);
    508 
    509                 SSLServerSocket serverSocket = (SSLServerSocket)sslContext
    510                         .getServerSocketFactory().createServerSocket();
    511 
    512                 if (clientAuth == CLIENT_AUTH_WANTED) {
    513                     serverSocket.setWantClientAuth(true);
    514                 } else if (clientAuth == CLIENT_AUTH_NEEDED) {
    515                     serverSocket.setNeedClientAuth(true);
    516                 } else {
    517                     serverSocket.setWantClientAuth(false);
    518                 }
    519 
    520                 serverSocket.bind(new InetSocketAddress(port));
    521 
    522                 SSLSocket clientSocket = (SSLSocket)serverSocket.accept();
    523 
    524                 InputStream istream = clientSocket.getInputStream();
    525                 byte[] buffer = new byte[1024];
    526                 istream.read(buffer);
    527 
    528                 OutputStream ostream = clientSocket.getOutputStream();
    529                 ostream.write(testData.getBytes());
    530                 ostream.flush();
    531 
    532                 while (notFinished) {
    533                     Thread.currentThread().sleep(500);
    534                 }
    535 
    536                 clientSocket.close();
    537                 serverSocket.close();
    538 
    539             } catch (Exception ex) {
    540                 exception = ex;
    541             }
    542         }
    543 
    544         public Exception getException() {
    545             return exception;
    546         }
    547 
    548         public javax.security.cert.X509Certificate[] getChain() {
    549             return trustManager.getChain();
    550         }
    551 
    552         public KeyStore getStore() {
    553             return store;
    554         }
    555 
    556     }
    557 
    558     /**
    559      * Implements a test SSL socket client. It opens a connection to localhost on
    560      * a given port, writes to the socket, and reads from the socket.
    561      */
    562     class TestClient implements Runnable {
    563 
    564         private TestTrustManager trustManager;
    565 
    566         private Exception exception;
    567 
    568         private String keys;
    569 
    570         private boolean provideKeys;
    571 
    572         private KeyStore store;
    573 
    574         public TestClient(boolean provideKeys, String keys) {
    575             this.keys = keys;
    576             this.provideKeys = provideKeys;
    577 
    578             trustManager = new TestTrustManager();
    579         }
    580 
    581         public void run() {
    582             try {
    583                 store = provideKeys ? getKeyStore(keys) : null;
    584                 KeyManager[] keyManagers = store != null ? getKeyManagers(store) : null;
    585                 TrustManager[] trustManagers = new TrustManager[] { trustManager };
    586 
    587                 clientSslContext = SSLContext.getInstance("TLS");
    588                 clientSslContext.init(keyManagers, trustManagers, null);
    589 
    590                 SSLSocket socket = (SSLSocket)clientSslContext.getSocketFactory().createSocket();
    591 
    592                 socket.connect(new InetSocketAddress(port));
    593                 OutputStream ostream = socket.getOutputStream();
    594                 ostream.write(testData.getBytes());
    595                 ostream.flush();
    596 
    597                 InputStream istream = socket.getInputStream();
    598                 byte[] buffer = new byte[1024];
    599                 istream.read(buffer);
    600 
    601                 clientSession = socket.getSession();
    602                 while (notFinished) {
    603                     Thread.currentThread().sleep(500);
    604                 }
    605                 socket.close();
    606 
    607             } catch (Exception ex) {
    608                 exception = ex;
    609             }
    610         }
    611 
    612         public Exception getException() {
    613             return exception;
    614         }
    615 
    616         public javax.security.cert.X509Certificate[] getChain() {
    617             return trustManager.getChain();
    618         }
    619 
    620         public KeyStore getStore() {
    621             return store;
    622         }
    623     }
    624 
    625     /**
    626      * Loads a keystore from a base64-encoded String. Returns the KeyManager[]
    627      * for the result.
    628      */
    629     private KeyStore getKeyStore(String keys) throws Exception {
    630         byte[] bytes = Base64.decode(keys.getBytes());
    631         InputStream inputStream = new ByteArrayInputStream(bytes);
    632 
    633         KeyStore keyStore = KeyStore.getInstance(KeyStore.getDefaultType());
    634         keyStore.load(inputStream, PASSWORD.toCharArray());
    635         inputStream.close();
    636         return keyStore;
    637     }
    638 
    639     /**
    640      * Loads a keystore from a base64-encoded String. Returns the KeyManager[]
    641      * for the result.
    642      */
    643     private KeyManager[] getKeyManagers(KeyStore keyStore) throws Exception {
    644         String algorithm = KeyManagerFactory.getDefaultAlgorithm();
    645         KeyManagerFactory keyManagerFactory = KeyManagerFactory.getInstance(algorithm);
    646         keyManagerFactory.init(keyStore, PASSWORD.toCharArray());
    647 
    648         return keyManagerFactory.getKeyManagers();
    649     }
    650 }
    651