blob: fd72a799fa5d8a82f2d24dd0ba416a8e840d193e [file] [log] [blame]
/*
* Copyright (C) 2010 Google Inc.
*
* 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 benchmarks.regression;
import com.google.caliper.Param;
import com.google.caliper.SimpleBenchmark;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.InetAddress;
import java.net.Socket;
import java.net.URL;
import javax.net.SocketFactory;
import javax.net.ssl.SSLContext;
public class SSLSocketBenchmark extends SimpleBenchmark {
private static final int BUFFER_SIZE = 8192;
final byte[] buffer = new byte[BUFFER_SIZE];
@Param private WebSite webSite;
public enum WebSite {
DOCS("https://docs.google.com"),
MAIL("https://mail.google.com"),
SITES("https://sites.google.com"),
WWW("https://www.google.com");
final InetAddress host;
final int port;
final byte[] request;
WebSite(String uri) {
try {
URL url = new URL(uri);
this.host = InetAddress.getByName(url.getHost());
int p = url.getPort();
String portString;
if (p == -1) {
this.port = 443;
portString = "";
} else {
this.port = p;
portString = ":" + port;
}
this.request = ("GET " + uri + " HTTP/1.0\r\n"
+ "Host: " + host + portString + "\r\n"
+"\r\n").getBytes();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
}
@Param private Implementation implementation;
public enum Implementation { OPENSSL, HARMONY };
private SocketFactory sf;
@Override protected void setUp() throws Exception {
SSLContext sslContext;
switch (implementation) {
case OPENSSL:
sslContext = SSLContext.getInstance("SSL", "AndroidOpenSSL");
break;
case HARMONY:
sslContext = SSLContext.getInstance("SSL", "HarmonyJSSE");
break;
default:
throw new RuntimeException(implementation.toString());
}
sslContext.init(null, null, null);
this.sf = sslContext.getSocketFactory();
}
public void time(int reps) throws Exception {
for (int i = 0; i < reps; ++i) {
Socket s = sf.createSocket(webSite.host, webSite.port);
OutputStream out = s.getOutputStream();
out.write(webSite.request);
InputStream in = s.getInputStream();
while (true) {
int n = in.read(buffer);
if (n == -1) {
break;
}
}
in.close();
}
}
}