Java tutorial
/* * Copyright (C) 2009 The Android Open Source Project * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package android.core; import android.test.AndroidTestCase; import android.os.Debug; import org.apache.harmony.xnet.provider.jsse.FileClientSessionCache; import org.apache.harmony.xnet.provider.jsse.OpenSSLContextImpl; import org.apache.harmony.xnet.provider.jsse.SSLClientSessionCache; import org.apache.http.conn.scheme.SchemeRegistry; import org.apache.http.conn.scheme.Scheme; import org.apache.http.conn.ClientConnectionManager; import org.apache.http.conn.ssl.SSLSocketFactory; import org.apache.http.impl.conn.SingleClientConnManager; import org.apache.http.impl.client.DefaultHttpClient; import org.apache.http.client.methods.HttpGet; import org.apache.http.client.ResponseHandler; import org.apache.http.client.ClientProtocolException; import org.apache.http.HttpResponse; import javax.net.ssl.SSLSession; import javax.net.ssl.SSLSessionContext; import java.io.File; import java.io.IOException; import java.lang.reflect.Method; import java.lang.reflect.InvocationTargetException; import java.security.cert.Certificate; import java.security.Principal; import java.security.KeyManagementException; import java.util.Arrays; public class SSLPerformanceTest extends AndroidTestCase { static final byte[] SESSION_DATA = new byte[6000]; static { for (int i = 0; i < SESSION_DATA.length; i++) { SESSION_DATA[i] = (byte) i; } } static final File dataDir = new File("/data/data/android.core/"); static final File filesDir = new File(dataDir, "files"); static final File dbDir = new File(dataDir, "databases"); static final String CACHE_DIR = SSLPerformanceTest.class.getName() + "/cache"; static final int ITERATIONS = 10; public void testCreateNewEmptyDatabase() { deleteDatabase(); Stopwatch stopwatch = new Stopwatch(); DatabaseSessionCache cache = new DatabaseSessionCache(getContext()); cache.getSessionData("crazybob.org", 443); stopwatch.stop(); } public void testCreateNewEmptyDirectory() throws IOException { deleteDirectory(); Stopwatch stopwatch = new Stopwatch(); SSLClientSessionCache cache = FileClientSessionCache.usingDirectory(getCacheDirectory()); cache.getSessionData("crazybob.org", 443); stopwatch.stop(); } public void testOpenDatabaseWith10Sessions() { deleteDatabase(); DatabaseSessionCache cache = new DatabaseSessionCache(getContext()); putSessionsIn(cache); closeDatabase(); System.err .println("Size of ssl_sessions.db w/ 10 sessions: " + new File(dbDir, "ssl_sessions.db").length()); Stopwatch stopwatch = new Stopwatch(); cache = new DatabaseSessionCache(getContext()); cache.getSessionData("crazybob.org", 443); stopwatch.stop(); } public void testOpenDirectoryWith10Sessions() throws IOException { deleteDirectory(); SSLClientSessionCache cache = FileClientSessionCache.usingDirectory(getCacheDirectory()); putSessionsIn(cache); closeDirectoryCache(); Stopwatch stopwatch = new Stopwatch(); cache = FileClientSessionCache.usingDirectory(getCacheDirectory()); cache.getSessionData("crazybob.org", 443); stopwatch.stop(); } public void testGetSessionFromDatabase() { deleteDatabase(); DatabaseSessionCache cache = new DatabaseSessionCache(getContext()); cache.putSessionData(new FakeSession("foo"), SESSION_DATA); closeDatabase(); cache = new DatabaseSessionCache(getContext()); cache.getSessionData("crazybob.org", 443); Stopwatch stopwatch = new Stopwatch(); byte[] sessionData = cache.getSessionData("foo", 443); stopwatch.stop(); assertTrue(Arrays.equals(SESSION_DATA, sessionData)); } public void testGetSessionFromDirectory() throws IOException { deleteDirectory(); SSLClientSessionCache cache = FileClientSessionCache.usingDirectory(getCacheDirectory()); cache.putSessionData(new FakeSession("foo"), SESSION_DATA); closeDirectoryCache(); cache = FileClientSessionCache.usingDirectory(getCacheDirectory()); cache.getSessionData("crazybob.org", 443); Stopwatch stopwatch = new Stopwatch(); byte[] sessionData = cache.getSessionData("foo", 443); stopwatch.stop(); assertTrue(Arrays.equals(SESSION_DATA, sessionData)); } public void testPutSessionIntoDatabase() { deleteDatabase(); DatabaseSessionCache cache = new DatabaseSessionCache(getContext()); cache.getSessionData("crazybob.org", 443); Stopwatch stopwatch = new Stopwatch(); cache.putSessionData(new FakeSession("foo"), SESSION_DATA); stopwatch.stop(); } public void testPutSessionIntoDirectory() throws IOException { deleteDirectory(); SSLClientSessionCache cache = FileClientSessionCache.usingDirectory(getCacheDirectory()); cache.getSessionData("crazybob.org", 443); Stopwatch stopwatch = new Stopwatch(); cache.putSessionData(new FakeSession("foo"), SESSION_DATA); stopwatch.stop(); } public void testEngineInit() throws IOException, KeyManagementException { Stopwatch stopwatch = new Stopwatch(); new OpenSSLContextImpl().engineInit(null, null, null); stopwatch.stop(); } public void testWebRequestWithoutCache() throws IOException, KeyManagementException { OpenSSLContextImpl sslContext = new OpenSSLContextImpl(); sslContext.engineInit(null, null, null); Stopwatch stopwatch = new Stopwatch(); getVerisignDotCom(sslContext); stopwatch.stop(); } public void testWebRequestWithFileCache() throws IOException, KeyManagementException { deleteDirectory(); OpenSSLContextImpl sslContext = new OpenSSLContextImpl(); sslContext.engineInit(null, null, null); sslContext.engineGetClientSessionContext() .setPersistentCache(FileClientSessionCache.usingDirectory(getCacheDirectory())); // Make sure www.google.com is in the cache. getVerisignDotCom(sslContext); // Re-initialize so we hit the file cache. sslContext.engineInit(null, null, null); sslContext.engineGetClientSessionContext() .setPersistentCache(FileClientSessionCache.usingDirectory(getCacheDirectory())); Stopwatch stopwatch = new Stopwatch(); getVerisignDotCom(sslContext); stopwatch.stop(); } public void testWebRequestWithInMemoryCache() throws IOException, KeyManagementException { deleteDirectory(); OpenSSLContextImpl sslContext = new OpenSSLContextImpl(); sslContext.engineInit(null, null, null); // Make sure www.google.com is in the cache. getVerisignDotCom(sslContext); Stopwatch stopwatch = new Stopwatch(); getVerisignDotCom(sslContext); stopwatch.stop(); } private void getVerisignDotCom(OpenSSLContextImpl sslContext) throws IOException { SchemeRegistry schemeRegistry = new SchemeRegistry(); schemeRegistry .register(new Scheme("https", new SSLSocketFactory(sslContext.engineGetSocketFactory()), 443)); ClientConnectionManager manager = new SingleClientConnManager(null, schemeRegistry); new DefaultHttpClient(manager, null).execute(new HttpGet("https://www.verisign.com"), new ResponseHandler<Object>() { public Object handleResponse(HttpResponse response) throws ClientProtocolException, IOException { return null; } }); } private void putSessionsIn(SSLClientSessionCache cache) { for (int i = 0; i < 10; i++) { cache.putSessionData(new FakeSession("host" + i), SESSION_DATA); } } private void deleteDatabase() { closeDatabase(); if (!new File(dbDir, "ssl_sessions.db").delete()) { System.err.println("Failed to delete database."); } } private void closeDatabase() { if (DatabaseSessionCache.sDefaultDatabaseHelper != null) { DatabaseSessionCache.sDefaultDatabaseHelper.close(); } DatabaseSessionCache.sDefaultDatabaseHelper = null; DatabaseSessionCache.sHookInitializationDone = false; DatabaseSessionCache.mNeedsCacheLoad = true; } private void deleteDirectory() { closeDirectoryCache(); File dir = getCacheDirectory(); if (!dir.exists()) { return; } for (File file : dir.listFiles()) { file.delete(); } if (!dir.delete()) { System.err.println("Failed to delete directory."); } } private void closeDirectoryCache() { try { Method reset = FileClientSessionCache.class.getDeclaredMethod("reset"); reset.setAccessible(true); reset.invoke(null); } catch (NoSuchMethodException e) { throw new RuntimeException(e); } catch (IllegalAccessException e) { throw new RuntimeException(e); } catch (InvocationTargetException e) { throw new RuntimeException(e); } } private File getCacheDirectory() { return new File(getContext().getFilesDir(), CACHE_DIR); } class Stopwatch { { Debug.startAllocCounting(); } long start = System.nanoTime(); void stop() { long elapsed = (System.nanoTime() - start) / 1000; Debug.stopAllocCounting(); System.err.println(getName() + ": " + elapsed + "us, " + Debug.getThreadAllocCount() + " allocations, " + Debug.getThreadAllocSize() + " bytes"); } } } class FakeSession implements SSLSession { final String host; FakeSession(String host) { this.host = host; } public int getApplicationBufferSize() { throw new UnsupportedOperationException(); } public String getCipherSuite() { throw new UnsupportedOperationException(); } public long getCreationTime() { throw new UnsupportedOperationException(); } public byte[] getId() { return host.getBytes(); } public long getLastAccessedTime() { throw new UnsupportedOperationException(); } public Certificate[] getLocalCertificates() { throw new UnsupportedOperationException(); } public Principal getLocalPrincipal() { throw new UnsupportedOperationException(); } public int getPacketBufferSize() { throw new UnsupportedOperationException(); } public javax.security.cert.X509Certificate[] getPeerCertificateChain() { throw new UnsupportedOperationException(); } public Certificate[] getPeerCertificates() { throw new UnsupportedOperationException(); } public String getPeerHost() { return host; } public int getPeerPort() { return 443; } public Principal getPeerPrincipal() { throw new UnsupportedOperationException(); } public String getProtocol() { throw new UnsupportedOperationException(); } public SSLSessionContext getSessionContext() { throw new UnsupportedOperationException(); } public Object getValue(String name) { throw new UnsupportedOperationException(); } public String[] getValueNames() { throw new UnsupportedOperationException(); } public void invalidate() { throw new UnsupportedOperationException(); } public boolean isValid() { throw new UnsupportedOperationException(); } public void putValue(String name, Object value) { throw new UnsupportedOperationException(); } public void removeValue(String name) { throw new UnsupportedOperationException(); } }