jetty-http2/http2-server/src/test/java/org/eclipse/jetty/http2/server/HTTP2CServerTest.java - gerrit/jetty/org.eclipse.jetty.project - Git at Google

 //
 //  ========================================================================
 //  Copyright (c) 1995-2015 Mort Bay Consulting Pty. Ltd.
 //  ------------------------------------------------------------------------
 //  All rights reserved. This program and the accompanying materials
 //  are made available under the terms of the Eclipse Public License v1.0
 //  and Apache License v2.0 which accompanies this distribution.
 //
 //      The Eclipse Public License is available at
 //      http://www.eclipse.org/legal/epl-v10.html
 //
 //      The Apache License v2.0 is available at
 //      http://www.opensource.org/licenses/apache2.0.php
 //
 //  You may elect to redistribute this code under either of these licenses.
 //  ========================================================================
 //

 package org.eclipse.jetty.http2.server;

 import java.io.InputStream;
 import java.io.OutputStream;
 import java.net.Socket;
 import java.nio.ByteBuffer;
 import java.nio.charset.StandardCharsets;
 import java.util.HashMap;
 import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicReference;

 import org.eclipse.jetty.http.HostPortHttpField;
 import org.eclipse.jetty.http.HttpFields;
 import org.eclipse.jetty.http.HttpScheme;
 import org.eclipse.jetty.http.HttpVersion;
 import org.eclipse.jetty.http.MetaData;
 import org.eclipse.jetty.http2.frames.DataFrame;
 import org.eclipse.jetty.http2.frames.HeadersFrame;
 import org.eclipse.jetty.http2.frames.PrefaceFrame;
 import org.eclipse.jetty.http2.frames.SettingsFrame;
 import org.eclipse.jetty.http2.generator.Generator;
 import org.eclipse.jetty.http2.parser.Parser;
 import org.eclipse.jetty.io.ByteBufferPool;
 import org.eclipse.jetty.io.MappedByteBufferPool;
 import org.eclipse.jetty.server.NetworkConnector;
 import org.eclipse.jetty.util.BufferUtil;
 import org.eclipse.jetty.util.IO;
 import org.eclipse.jetty.util.Utf8StringBuilder;
 import org.junit.After;
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;

 import static org.hamcrest.Matchers.containsString;
 import static org.junit.Assert.assertThat;
 import static org.junit.Assert.assertTrue;

 public class HTTP2CServerTest extends AbstractServerTest
 {
     private HTTP2CServer _server;
     private int _port;

     @Before
     public void before() throws Exception
     {
         _server = new HTTP2CServer(0);
         _server.start();
         _port = ((NetworkConnector)_server.getConnectors()[0]).getLocalPort();
     }

     @After
     public void after() throws Exception
     {
         _server.stop();
     }

     @Test
     public void testHTTP_1_0_Simple() throws Exception
     {
         try (Socket client = new Socket("localhost", _port))
         {
             client.getOutputStream().write("GET / HTTP/1.0\r\n\r\n".getBytes(StandardCharsets.ISO_8859_1));
             client.getOutputStream().flush();

             String response = IO.toString(client.getInputStream());

             assertThat(response, containsString("HTTP/1.1 200 OK"));
             assertThat(response, containsString("Hello from Jetty using HTTP/1.0"));
         }
     }

     @Test
     public void testHTTP_1_1_Simple() throws Exception
     {
         try (Socket client = new Socket("localhost", _port))
         {
             client.getOutputStream().write("GET /one HTTP/1.1\r\nHost: localhost\r\n\r\n".getBytes(StandardCharsets.ISO_8859_1));
             client.getOutputStream().write("GET /two HTTP/1.1\r\nHost: localhost\r\nConnection: close\r\n\r\n".getBytes(StandardCharsets.ISO_8859_1));
             client.getOutputStream().flush();

             String response = IO.toString(client.getInputStream());

             assertThat(response, containsString("HTTP/1.1 200 OK"));
             assertThat(response, containsString("Hello from Jetty using HTTP/1.1"));
             assertThat(response, containsString("uri=/one"));
             assertThat(response, containsString("uri=/two"));
         }
     }

     @Test
     public void testHTTP_1_1_Upgrade() throws Exception
     {
         try (Socket client = new Socket("localhost", _port))
         {
             OutputStream output = client.getOutputStream();
             output.write(("" +
                     "GET /one HTTP/1.1\r\n" +
                     "Host: localhost\r\n" +
                     "Connection: something, else, upgrade, HTTP2-Settings\r\n" +
                     "Upgrade: h2c\r\n" +
                     "HTTP2-Settings: \r\n" +
                     "\r\n").getBytes(StandardCharsets.ISO_8859_1));
             output.flush();

             InputStream input = client.getInputStream();
             Utf8StringBuilder upgrade = new Utf8StringBuilder();
             int crlfs = 0;
             while (true)
             {
                 int read = input.read();
                 if (read == '\r' || read == '\n')
                     ++crlfs;
                 else
                     crlfs = 0;
                 upgrade.append((byte)read);
                 if (crlfs == 4)
                     break;
             }

             assertTrue(upgrade.toString().startsWith("HTTP/1.1 101 "));

             byteBufferPool = new MappedByteBufferPool();
             generator = new Generator(byteBufferPool);

             final AtomicReference<HeadersFrame> headersRef = new AtomicReference<>();
             final AtomicReference<DataFrame> dataRef = new AtomicReference<>();
             final AtomicReference<CountDownLatch> latchRef = new AtomicReference<>(new CountDownLatch(2));
             Parser parser = new Parser(byteBufferPool, new Parser.Listener.Adapter()
             {
                 @Override
                 public void onHeaders(HeadersFrame frame)
                 {
                     headersRef.set(frame);
                     latchRef.get().countDown();
                 }

                 @Override
                 public void onData(DataFrame frame)
                 {
                     dataRef.set(frame);
                     latchRef.get().countDown();
                 }
             }, 4096, 8192);

             parseResponse(client, parser);

             Assert.assertTrue(latchRef.get().await(5, TimeUnit.SECONDS));

             HeadersFrame response = headersRef.get();
             Assert.assertNotNull(response);
             MetaData.Response responseMetaData = (MetaData.Response)response.getMetaData();
             Assert.assertEquals(200, responseMetaData.getStatus());

             DataFrame responseData = dataRef.get();
             Assert.assertNotNull(responseData);

             String content = BufferUtil.toString(responseData.getData());

             // The upgrade request is seen as HTTP/1.1.
             assertThat(content, containsString("Hello from Jetty using HTTP/1.1"));
             assertThat(content, containsString("uri=/one"));

             // Send a HTTP/2 request.
             headersRef.set(null);
             dataRef.set(null);
             latchRef.set(new CountDownLatch(2));
             ByteBufferPool.Lease lease = new ByteBufferPool.Lease(byteBufferPool);
             generator.control(lease, new PrefaceFrame());
             generator.control(lease, new SettingsFrame(new HashMap<>(), false));
             MetaData.Request metaData = new MetaData.Request("GET", HttpScheme.HTTP, new HostPortHttpField("localhost:" + _port), "/two", HttpVersion.HTTP_2, new HttpFields());
             generator.control(lease, new HeadersFrame(3, metaData, null, true));
             for (ByteBuffer buffer : lease.getByteBuffers())
                 output.write(BufferUtil.toArray(buffer));
             output.flush();

             parseResponse(client, parser);

             Assert.assertTrue(latchRef.get().await(5, TimeUnit.SECONDS));

             response = headersRef.get();
             Assert.assertNotNull(response);
             responseMetaData = (MetaData.Response)response.getMetaData();
             Assert.assertEquals(200, responseMetaData.getStatus());

             responseData = dataRef.get();
             Assert.assertNotNull(responseData);

             content = BufferUtil.toString(responseData.getData());

             assertThat(content, containsString("Hello from Jetty using HTTP/2.0"));
             assertThat(content, containsString("uri=/two"));
         }
     }

     @Test
     public void testHTTP_2_0_Direct() throws Exception
     {
         final CountDownLatch latch = new CountDownLatch(3);

         byteBufferPool = new MappedByteBufferPool();
         generator = new Generator(byteBufferPool);

         ByteBufferPool.Lease lease = new ByteBufferPool.Lease(byteBufferPool);
         generator.control(lease, new PrefaceFrame());
         generator.control(lease, new SettingsFrame(new HashMap<>(), false));
         MetaData.Request metaData = new MetaData.Request("GET", HttpScheme.HTTP, new HostPortHttpField("localhost:" + _port), "/test", HttpVersion.HTTP_2, new HttpFields());
         generator.control(lease, new HeadersFrame(1, metaData, null, true));

         try (Socket client = new Socket("localhost", _port))
         {
             OutputStream output = client.getOutputStream();
             for (ByteBuffer buffer : lease.getByteBuffers())
             {
                 output.write(BufferUtil.toArray(buffer));
             }

             final AtomicReference<HeadersFrame> headersRef = new AtomicReference<>();
             final AtomicReference<DataFrame> dataRef = new AtomicReference<>();
             Parser parser = new Parser(byteBufferPool, new Parser.Listener.Adapter()
             {
                 @Override
                 public void onSettings(SettingsFrame frame)
                 {
                     latch.countDown();
                 }

                 @Override
                 public void onHeaders(HeadersFrame frame)
                 {
                     headersRef.set(frame);
                     latch.countDown();
                 }

                 @Override
                 public void onData(DataFrame frame)
                 {
                     dataRef.set(frame);
                     latch.countDown();
                 }
             }, 4096, 8192);

             parseResponse(client, parser);

             Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));

             HeadersFrame response = headersRef.get();
             Assert.assertNotNull(response);
             MetaData.Response responseMetaData = (MetaData.Response)response.getMetaData();
             Assert.assertEquals(200, responseMetaData.getStatus());

             DataFrame responseData = dataRef.get();
             Assert.assertNotNull(responseData);

             String s = BufferUtil.toString(responseData.getData());

             assertThat(s, containsString("Hello from Jetty using HTTP/2.0"));
             assertThat(s, containsString("uri=/test"));
         }
     }
 }
	//
	// ========================================================================
	// Copyright (c) 1995-2015 Mort Bay Consulting Pty. Ltd.
	// ------------------------------------------------------------------------
	// All rights reserved. This program and the accompanying materials
	// are made available under the terms of the Eclipse Public License v1.0
	// and Apache License v2.0 which accompanies this distribution.
	//
	// The Eclipse Public License is available at
	// http://www.eclipse.org/legal/epl-v10.html
	//
	// The Apache License v2.0 is available at
	// http://www.opensource.org/licenses/apache2.0.php
	//
	// You may elect to redistribute this code under either of these licenses.
	// ========================================================================
	//

	package org.eclipse.jetty.http2.server;

	import java.io.InputStream;
	import java.io.OutputStream;
	import java.net.Socket;
	import java.nio.ByteBuffer;
	import java.nio.charset.StandardCharsets;
	import java.util.HashMap;
	import java.util.concurrent.CountDownLatch;
	import java.util.concurrent.TimeUnit;
	import java.util.concurrent.atomic.AtomicReference;

	import org.eclipse.jetty.http.HostPortHttpField;
	import org.eclipse.jetty.http.HttpFields;
	import org.eclipse.jetty.http.HttpScheme;
	import org.eclipse.jetty.http.HttpVersion;
	import org.eclipse.jetty.http.MetaData;
	import org.eclipse.jetty.http2.frames.DataFrame;
	import org.eclipse.jetty.http2.frames.HeadersFrame;
	import org.eclipse.jetty.http2.frames.PrefaceFrame;
	import org.eclipse.jetty.http2.frames.SettingsFrame;
	import org.eclipse.jetty.http2.generator.Generator;
	import org.eclipse.jetty.http2.parser.Parser;
	import org.eclipse.jetty.io.ByteBufferPool;
	import org.eclipse.jetty.io.MappedByteBufferPool;
	import org.eclipse.jetty.server.NetworkConnector;
	import org.eclipse.jetty.util.BufferUtil;
	import org.eclipse.jetty.util.IO;
	import org.eclipse.jetty.util.Utf8StringBuilder;
	import org.junit.After;
	import org.junit.Assert;
	import org.junit.Before;
	import org.junit.Test;

	import static org.hamcrest.Matchers.containsString;
	import static org.junit.Assert.assertThat;
	import static org.junit.Assert.assertTrue;

	public class HTTP2CServerTest extends AbstractServerTest
	{
	private HTTP2CServer _server;
	private int _port;

	@Before
	public void before() throws Exception
	{
	_server = new HTTP2CServer(0);
	_server.start();
	_port = ((NetworkConnector)_server.getConnectors()[0]).getLocalPort();
	}

	@After
	public void after() throws Exception
	{
	_server.stop();
	}

	@Test
	public void testHTTP_1_0_Simple() throws Exception
	{
	try (Socket client = new Socket("localhost", _port))
	{
	client.getOutputStream().write("GET / HTTP/1.0\r\n\r\n".getBytes(StandardCharsets.ISO_8859_1));
	client.getOutputStream().flush();

	String response = IO.toString(client.getInputStream());

	assertThat(response, containsString("HTTP/1.1 200 OK"));
	assertThat(response, containsString("Hello from Jetty using HTTP/1.0"));
	}
	}

	@Test
	public void testHTTP_1_1_Simple() throws Exception
	{
	try (Socket client = new Socket("localhost", _port))
	{
	client.getOutputStream().write("GET /one HTTP/1.1\r\nHost: localhost\r\n\r\n".getBytes(StandardCharsets.ISO_8859_1));
	client.getOutputStream().write("GET /two HTTP/1.1\r\nHost: localhost\r\nConnection: close\r\n\r\n".getBytes(StandardCharsets.ISO_8859_1));
	client.getOutputStream().flush();

	String response = IO.toString(client.getInputStream());

	assertThat(response, containsString("HTTP/1.1 200 OK"));
	assertThat(response, containsString("Hello from Jetty using HTTP/1.1"));
	assertThat(response, containsString("uri=/one"));
	assertThat(response, containsString("uri=/two"));
	}
	}

	@Test
	public void testHTTP_1_1_Upgrade() throws Exception
	{
	try (Socket client = new Socket("localhost", _port))
	{
	OutputStream output = client.getOutputStream();
	output.write(("" +
	"GET /one HTTP/1.1\r\n" +
	"Host: localhost\r\n" +
	"Connection: something, else, upgrade, HTTP2-Settings\r\n" +
	"Upgrade: h2c\r\n" +
	"HTTP2-Settings: \r\n" +
	"\r\n").getBytes(StandardCharsets.ISO_8859_1));
	output.flush();

	InputStream input = client.getInputStream();
	Utf8StringBuilder upgrade = new Utf8StringBuilder();
	int crlfs = 0;
	while (true)
	{
	int read = input.read();
	if (read == '\r' \|\| read == '\n')
	++crlfs;
	else
	crlfs = 0;
	upgrade.append((byte)read);
	if (crlfs == 4)
	break;
	}

	assertTrue(upgrade.toString().startsWith("HTTP/1.1 101 "));

	byteBufferPool = new MappedByteBufferPool();
	generator = new Generator(byteBufferPool);

	final AtomicReference<HeadersFrame> headersRef = new AtomicReference<>();
	final AtomicReference<DataFrame> dataRef = new AtomicReference<>();
	final AtomicReference<CountDownLatch> latchRef = new AtomicReference<>(new CountDownLatch(2));
	Parser parser = new Parser(byteBufferPool, new Parser.Listener.Adapter()
	{
	@Override
	public void onHeaders(HeadersFrame frame)
	{
	headersRef.set(frame);
	latchRef.get().countDown();
	}

	@Override
	public void onData(DataFrame frame)
	{
	dataRef.set(frame);
	latchRef.get().countDown();
	}
	}, 4096, 8192);

	parseResponse(client, parser);

	Assert.assertTrue(latchRef.get().await(5, TimeUnit.SECONDS));

	HeadersFrame response = headersRef.get();
	Assert.assertNotNull(response);
	MetaData.Response responseMetaData = (MetaData.Response)response.getMetaData();
	Assert.assertEquals(200, responseMetaData.getStatus());

	DataFrame responseData = dataRef.get();
	Assert.assertNotNull(responseData);

	String content = BufferUtil.toString(responseData.getData());

	// The upgrade request is seen as HTTP/1.1.
	assertThat(content, containsString("Hello from Jetty using HTTP/1.1"));
	assertThat(content, containsString("uri=/one"));

	// Send a HTTP/2 request.
	headersRef.set(null);
	dataRef.set(null);
	latchRef.set(new CountDownLatch(2));
	ByteBufferPool.Lease lease = new ByteBufferPool.Lease(byteBufferPool);
	generator.control(lease, new PrefaceFrame());
	generator.control(lease, new SettingsFrame(new HashMap<>(), false));
	MetaData.Request metaData = new MetaData.Request("GET", HttpScheme.HTTP, new HostPortHttpField("localhost:" + _port), "/two", HttpVersion.HTTP_2, new HttpFields());
	generator.control(lease, new HeadersFrame(3, metaData, null, true));
	for (ByteBuffer buffer : lease.getByteBuffers())
	output.write(BufferUtil.toArray(buffer));
	output.flush();

	parseResponse(client, parser);

	Assert.assertTrue(latchRef.get().await(5, TimeUnit.SECONDS));

	response = headersRef.get();
	Assert.assertNotNull(response);
	responseMetaData = (MetaData.Response)response.getMetaData();
	Assert.assertEquals(200, responseMetaData.getStatus());

	responseData = dataRef.get();
	Assert.assertNotNull(responseData);

	content = BufferUtil.toString(responseData.getData());

	assertThat(content, containsString("Hello from Jetty using HTTP/2.0"));
	assertThat(content, containsString("uri=/two"));
	}
	}

	@Test
	public void testHTTP_2_0_Direct() throws Exception
	{
	final CountDownLatch latch = new CountDownLatch(3);

	byteBufferPool = new MappedByteBufferPool();
	generator = new Generator(byteBufferPool);

	ByteBufferPool.Lease lease = new ByteBufferPool.Lease(byteBufferPool);
	generator.control(lease, new PrefaceFrame());
	generator.control(lease, new SettingsFrame(new HashMap<>(), false));
	MetaData.Request metaData = new MetaData.Request("GET", HttpScheme.HTTP, new HostPortHttpField("localhost:" + _port), "/test", HttpVersion.HTTP_2, new HttpFields());
	generator.control(lease, new HeadersFrame(1, metaData, null, true));

	try (Socket client = new Socket("localhost", _port))
	{
	OutputStream output = client.getOutputStream();
	for (ByteBuffer buffer : lease.getByteBuffers())
	{
	output.write(BufferUtil.toArray(buffer));
	}

	final AtomicReference<HeadersFrame> headersRef = new AtomicReference<>();
	final AtomicReference<DataFrame> dataRef = new AtomicReference<>();
	Parser parser = new Parser(byteBufferPool, new Parser.Listener.Adapter()
	{
	@Override
	public void onSettings(SettingsFrame frame)
	{
	latch.countDown();
	}

	@Override
	public void onHeaders(HeadersFrame frame)
	{
	headersRef.set(frame);
	latch.countDown();
	}

	@Override
	public void onData(DataFrame frame)
	{
	dataRef.set(frame);
	latch.countDown();
	}
	}, 4096, 8192);

	parseResponse(client, parser);

	Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));

	HeadersFrame response = headersRef.get();
	Assert.assertNotNull(response);
	MetaData.Response responseMetaData = (MetaData.Response)response.getMetaData();
	Assert.assertEquals(200, responseMetaData.getStatus());

	DataFrame responseData = dataRef.get();
	Assert.assertNotNull(responseData);

	String s = BufferUtil.toString(responseData.getData());

	assertThat(s, containsString("Hello from Jetty using HTTP/2.0"));
	assertThat(s, containsString("uri=/test"));
	}
	}
	}