Code cleanup.
This commit is contained in:
parent
f9e3c535d6
commit
ad8bdde4f3
|
@ -60,9 +60,9 @@ import org.junit.Test;
|
||||||
|
|
||||||
public class SslConnectionFactoryTest
|
public class SslConnectionFactoryTest
|
||||||
{
|
{
|
||||||
Server _server;
|
private Server _server;
|
||||||
ServerConnector _connector;
|
private ServerConnector _connector;
|
||||||
int _port;
|
private int _port;
|
||||||
|
|
||||||
@Before
|
@Before
|
||||||
public void before() throws Exception
|
public void before() throws Exception
|
||||||
|
@ -70,9 +70,7 @@ public class SslConnectionFactoryTest
|
||||||
String keystorePath = "src/test/resources/keystore";
|
String keystorePath = "src/test/resources/keystore";
|
||||||
File keystoreFile = new File(keystorePath);
|
File keystoreFile = new File(keystorePath);
|
||||||
if (!keystoreFile.exists())
|
if (!keystoreFile.exists())
|
||||||
{
|
|
||||||
throw new FileNotFoundException(keystoreFile.getAbsolutePath());
|
throw new FileNotFoundException(keystoreFile.getAbsolutePath());
|
||||||
}
|
|
||||||
|
|
||||||
_server = new Server();
|
_server = new Server();
|
||||||
|
|
||||||
|
@ -129,21 +127,10 @@ public class SslConnectionFactoryTest
|
||||||
@Test
|
@Test
|
||||||
public void testSNIConnect() throws Exception
|
public void testSNIConnect() throws Exception
|
||||||
{
|
{
|
||||||
String response;
|
String response = getResponse("localhost", "localhost", "jetty.eclipse.org");
|
||||||
|
|
||||||
response= getResponse("localhost","localhost","jetty.eclipse.org");
|
|
||||||
Assert.assertThat(response, Matchers.containsString("host=localhost"));
|
Assert.assertThat(response, Matchers.containsString("host=localhost"));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
private String getResponse(String host,String cn) throws Exception
|
|
||||||
{
|
|
||||||
String response = getResponse(host,host,cn);
|
|
||||||
Assert.assertThat(response,Matchers.startsWith("HTTP/1.1 200 OK"));
|
|
||||||
Assert.assertThat(response,Matchers.containsString("url=/ctx/path"));
|
|
||||||
return response;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testBadHandshake() throws Exception
|
public void testBadHandshake() throws Exception
|
||||||
{
|
{
|
||||||
|
@ -156,43 +143,6 @@ public class SslConnectionFactoryTest
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private String getResponse(String sniHost,String reqHost, String cn) throws Exception
|
|
||||||
{
|
|
||||||
SslContextFactory clientContextFactory = new SslContextFactory(true);
|
|
||||||
clientContextFactory.start();
|
|
||||||
SSLSocketFactory factory = clientContextFactory.getSslContext().getSocketFactory();
|
|
||||||
|
|
||||||
SSLSocket sslSocket = (SSLSocket)factory.createSocket("127.0.0.1", _port);
|
|
||||||
|
|
||||||
if (cn!=null)
|
|
||||||
{
|
|
||||||
SNIHostName serverName = new SNIHostName(sniHost);
|
|
||||||
List<SNIServerName> serverNames = new ArrayList<>();
|
|
||||||
serverNames.add(serverName);
|
|
||||||
|
|
||||||
SSLParameters params = sslSocket.getSSLParameters();
|
|
||||||
params.setServerNames(serverNames);
|
|
||||||
sslSocket.setSSLParameters(params);
|
|
||||||
}
|
|
||||||
sslSocket.startHandshake();
|
|
||||||
|
|
||||||
|
|
||||||
if (cn!=null)
|
|
||||||
{
|
|
||||||
X509Certificate cert = ((X509Certificate)sslSocket.getSession().getPeerCertificates()[0]);
|
|
||||||
|
|
||||||
Assert.assertThat(cert.getSubjectX500Principal().getName("CANONICAL"), Matchers.startsWith("cn="+cn));
|
|
||||||
}
|
|
||||||
|
|
||||||
sslSocket.getOutputStream().write(("GET /ctx/path HTTP/1.0\r\nHost: "+reqHost+":"+_port+"\r\n\r\n").getBytes(StandardCharsets.ISO_8859_1));
|
|
||||||
String response = IO.toString(sslSocket.getInputStream());
|
|
||||||
|
|
||||||
sslSocket.close();
|
|
||||||
clientContextFactory.stop();
|
|
||||||
return response;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testSocketCustomization() throws Exception
|
public void testSocketCustomization() throws Exception
|
||||||
{
|
{
|
||||||
|
@ -235,4 +185,45 @@ public class SslConnectionFactoryTest
|
||||||
Assert.assertEquals(0, history.size());
|
Assert.assertEquals(0, history.size());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private String getResponse(String host, String cn) throws Exception
|
||||||
|
{
|
||||||
|
String response = getResponse(host, host, cn);
|
||||||
|
Assert.assertThat(response, Matchers.startsWith("HTTP/1.1 200 OK"));
|
||||||
|
Assert.assertThat(response, Matchers.containsString("url=/ctx/path"));
|
||||||
|
return response;
|
||||||
|
}
|
||||||
|
|
||||||
|
private String getResponse(String sniHost, String reqHost, String cn) throws Exception
|
||||||
|
{
|
||||||
|
SslContextFactory clientContextFactory = new SslContextFactory(true);
|
||||||
|
clientContextFactory.start();
|
||||||
|
SSLSocketFactory factory = clientContextFactory.getSslContext().getSocketFactory();
|
||||||
|
|
||||||
|
SSLSocket sslSocket = (SSLSocket)factory.createSocket("127.0.0.1", _port);
|
||||||
|
|
||||||
|
if (cn != null)
|
||||||
|
{
|
||||||
|
SNIHostName serverName = new SNIHostName(sniHost);
|
||||||
|
List<SNIServerName> serverNames = new ArrayList<>();
|
||||||
|
serverNames.add(serverName);
|
||||||
|
|
||||||
|
SSLParameters params = sslSocket.getSSLParameters();
|
||||||
|
params.setServerNames(serverNames);
|
||||||
|
sslSocket.setSSLParameters(params);
|
||||||
|
}
|
||||||
|
sslSocket.startHandshake();
|
||||||
|
|
||||||
|
if (cn != null)
|
||||||
|
{
|
||||||
|
X509Certificate cert = ((X509Certificate)sslSocket.getSession().getPeerCertificates()[0]);
|
||||||
|
Assert.assertThat(cert.getSubjectX500Principal().getName("CANONICAL"), Matchers.startsWith("cn=" + cn));
|
||||||
|
}
|
||||||
|
|
||||||
|
sslSocket.getOutputStream().write(("GET /ctx/path HTTP/1.0\r\nHost: " + reqHost + ":" + _port + "\r\n\r\n").getBytes(StandardCharsets.ISO_8859_1));
|
||||||
|
String response = IO.toString(sslSocket.getInputStream());
|
||||||
|
|
||||||
|
sslSocket.close();
|
||||||
|
clientContextFactory.stop();
|
||||||
|
return response;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue