message) {
clientEndpoint.releaseAndReuse();
final HttpResponse response = message.getHead();
final String body = message.getBody();
System.out.println(requestUri + "->" + response.getCode());
System.out.println(body);
latch.countDown();
}
@Override
public void failed(final Exception ex) {
clientEndpoint.releaseAndDiscard();
System.out.println(requestUri + "->" + ex);
latch.countDown();
}
@Override
public void cancelled() {
clientEndpoint.releaseAndDiscard();
System.out.println(requestUri + " cancelled");
latch.countDown();
}
});
}
latch.await();
System.out.println("Shutting down I/O reactor");
requester.initiateShutdown();
}
}
httpcore5-h2/src/test/java/org/apache/hc/core5/http2/examples/H2GreetingServer.java 0100664 0000000 0000000 00000017604 14157612023 027021 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.examples;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.util.Date;
import java.util.List;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Future;
import org.apache.hc.core5.function.Supplier;
import org.apache.hc.core5.http.ContentType;
import org.apache.hc.core5.http.EndpointDetails;
import org.apache.hc.core5.http.EntityDetails;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.HttpException;
import org.apache.hc.core5.http.HttpHeaders;
import org.apache.hc.core5.http.HttpRequest;
import org.apache.hc.core5.http.HttpResponse;
import org.apache.hc.core5.http.Message;
import org.apache.hc.core5.http.NameValuePair;
import org.apache.hc.core5.http.URIScheme;
import org.apache.hc.core5.http.impl.bootstrap.HttpAsyncServer;
import org.apache.hc.core5.http.message.BasicHttpResponse;
import org.apache.hc.core5.http.nio.AsyncEntityConsumer;
import org.apache.hc.core5.http.nio.AsyncRequestConsumer;
import org.apache.hc.core5.http.nio.AsyncServerExchangeHandler;
import org.apache.hc.core5.http.nio.AsyncServerRequestHandler;
import org.apache.hc.core5.http.nio.entity.AsyncEntityProducers;
import org.apache.hc.core5.http.nio.entity.NoopEntityConsumer;
import org.apache.hc.core5.http.nio.entity.StringAsyncEntityConsumer;
import org.apache.hc.core5.http.nio.support.AbstractServerExchangeHandler;
import org.apache.hc.core5.http.nio.support.BasicRequestConsumer;
import org.apache.hc.core5.http.nio.support.BasicResponseProducer;
import org.apache.hc.core5.http.protocol.HttpContext;
import org.apache.hc.core5.http.protocol.HttpCoreContext;
import org.apache.hc.core5.http2.HttpVersionPolicy;
import org.apache.hc.core5.http2.config.H2Config;
import org.apache.hc.core5.http2.impl.nio.bootstrap.H2ServerBootstrap;
import org.apache.hc.core5.io.CloseMode;
import org.apache.hc.core5.net.WWWFormCodec;
import org.apache.hc.core5.reactor.IOReactorConfig;
import org.apache.hc.core5.reactor.ListenerEndpoint;
import org.apache.hc.core5.util.TimeValue;
/**
* Example HTTP2 server that reads an entity body and responds back with a greeting.
*
*
* {@code
* $ curl -id name=bob localhost:8080
* HTTP/1.1 200 OK
* Date: Sat, 25 May 2019 03:44:49 GMT
* Server: Apache-HttpCore/5.0-beta8-SNAPSHOT (Java/1.8.0_202)
* Transfer-Encoding: chunked
* Content-Type: text/plain; charset=ISO-8859-1
*
* Hello bob
* }
*
* This examples uses a {@link AbstractServerExchangeHandler} for the basic request / response processing cycle.
*/
public class H2GreetingServer {
public static void main(final String[] args) throws ExecutionException, InterruptedException {
int port = 8080;
if (args.length >= 1) {
port = Integer.parseInt(args[0]);
}
final HttpAsyncServer server = H2ServerBootstrap.bootstrap()
.setH2Config(H2Config.DEFAULT)
.setIOReactorConfig(IOReactorConfig.DEFAULT)
.setVersionPolicy(HttpVersionPolicy.NEGOTIATE) // fallback to HTTP/1 as needed
// wildcard path matcher:
.register("*", new Supplier() {
@Override
public AsyncServerExchangeHandler get() {
return new CustomServerExchangeHandler();
}
})
.create();
Runtime.getRuntime().addShutdownHook(new Thread(new Runnable() {
@Override
public void run() {
System.out.println("HTTP server shutting down");
server.close(CloseMode.GRACEFUL);
}
}));
server.start();
final Future future = server.listen(new InetSocketAddress(port), URIScheme.HTTP);
final ListenerEndpoint listenerEndpoint = future.get();
System.out.println("Listening on " + listenerEndpoint.getAddress());
server.awaitShutdown(TimeValue.ofDays(Long.MAX_VALUE));
}
static class CustomServerExchangeHandler extends AbstractServerExchangeHandler> {
@Override
protected AsyncRequestConsumer> supplyConsumer(
final HttpRequest request,
final EntityDetails entityDetails,
final HttpContext context) {
// if there's no body don't try to parse entity:
AsyncEntityConsumer entityConsumer = new NoopEntityConsumer();
if (entityDetails != null) {
entityConsumer = new StringAsyncEntityConsumer();
}
//noinspection unchecked
return new BasicRequestConsumer<>(entityConsumer);
}
@Override
protected void handle(final Message requestMessage,
final AsyncServerRequestHandler.ResponseTrigger responseTrigger,
final HttpContext context) throws HttpException, IOException {
final HttpCoreContext coreContext = HttpCoreContext.adapt(context);
final EndpointDetails endpoint = coreContext.getEndpointDetails();
final HttpRequest req = requestMessage.getHead();
final String httpEntity = requestMessage.getBody();
// generic success response:
final HttpResponse resp = new BasicHttpResponse(200);
// recording the request
System.out.println(String.format("[%s] %s %s %s", new Date(),
endpoint.getRemoteAddress(),
req.getMethod(),
req.getPath()));
// Request without an entity - GET/HEAD/DELETE
if (httpEntity == null) {
responseTrigger.submitResponse(
new BasicResponseProducer(resp), context);
return;
}
// Request with an entity - POST/PUT
final Header cth = req.getHeader(HttpHeaders.CONTENT_TYPE);
final ContentType contentType = cth != null ? ContentType.parse(cth.getValue()) : null;
String name = "stranger";
if (contentType != null && contentType.isSameMimeType(ContentType.APPLICATION_FORM_URLENCODED)) {
// decoding the form entity into key/value pairs:
final List args = WWWFormCodec.parse(httpEntity, contentType.getCharset());
if (!args.isEmpty()) {
name = args.get(0).getValue();
}
}
// composing greeting:
final String greeting = String.format("Hello %s\n", name);
responseTrigger.submitResponse(
new BasicResponseProducer(resp, AsyncEntityProducers.create(greeting)), context);
}
}
}
httpcore5-h2/src/test/java/org/apache/hc/core5/http2/examples/H2FullDuplexServerExample.java 0100664 0000000 0000000 00000026023 14157612023 030650 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.examples;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.SocketException;
import java.nio.ByteBuffer;
import java.util.List;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import org.apache.hc.core5.function.Supplier;
import org.apache.hc.core5.http.EntityDetails;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.HttpConnection;
import org.apache.hc.core5.http.HttpException;
import org.apache.hc.core5.http.HttpRequest;
import org.apache.hc.core5.http.HttpResponse;
import org.apache.hc.core5.http.HttpStatus;
import org.apache.hc.core5.http.URIScheme;
import org.apache.hc.core5.http.impl.bootstrap.HttpAsyncServer;
import org.apache.hc.core5.http.message.BasicHttpResponse;
import org.apache.hc.core5.http.nio.AsyncServerExchangeHandler;
import org.apache.hc.core5.http.nio.CapacityChannel;
import org.apache.hc.core5.http.nio.DataStreamChannel;
import org.apache.hc.core5.http.nio.ResponseChannel;
import org.apache.hc.core5.http.protocol.HttpContext;
import org.apache.hc.core5.http2.HttpVersionPolicy;
import org.apache.hc.core5.http2.config.H2Config;
import org.apache.hc.core5.http2.frame.RawFrame;
import org.apache.hc.core5.http2.impl.nio.H2StreamListener;
import org.apache.hc.core5.http2.impl.nio.bootstrap.H2ServerBootstrap;
import org.apache.hc.core5.io.CloseMode;
import org.apache.hc.core5.reactor.IOReactorConfig;
import org.apache.hc.core5.reactor.ListenerEndpoint;
import org.apache.hc.core5.util.TimeValue;
/**
* Example of full-duplex, streaming HTTP message exchanges with an asynchronous embedded HTTP/2 server.
*/
public class H2FullDuplexServerExample {
public static void main(final String[] args) throws Exception {
int port = 8080;
if (args.length >= 1) {
port = Integer.parseInt(args[0]);
}
final IOReactorConfig config = IOReactorConfig.custom()
.setSoTimeout(15, TimeUnit.SECONDS)
.setTcpNoDelay(true)
.build();
final H2Config h2Config = H2Config.custom()
.setPushEnabled(true)
.setMaxConcurrentStreams(100)
.build();
final HttpAsyncServer server = H2ServerBootstrap.bootstrap()
.setIOReactorConfig(config)
.setH2Config(h2Config)
.setVersionPolicy(HttpVersionPolicy.FORCE_HTTP_2)
.setStreamListener(new H2StreamListener() {
@Override
public void onHeaderInput(final HttpConnection connection, final int streamId, final List headers) {
for (int i = 0; i < headers.size(); i++) {
System.out.println(connection.getRemoteAddress() + " (" + streamId + ") << " + headers.get(i));
}
}
@Override
public void onHeaderOutput(final HttpConnection connection, final int streamId, final List headers) {
for (int i = 0; i < headers.size(); i++) {
System.out.println(connection.getRemoteAddress() + " (" + streamId + ") >> " + headers.get(i));
}
}
@Override
public void onFrameInput(final HttpConnection connection, final int streamId, final RawFrame frame) {
}
@Override
public void onFrameOutput(final HttpConnection connection, final int streamId, final RawFrame frame) {
}
@Override
public void onInputFlowControl(final HttpConnection connection, final int streamId, final int delta, final int actualSize) {
}
@Override
public void onOutputFlowControl(final HttpConnection connection, final int streamId, final int delta, final int actualSize) {
}
})
.register("/echo", new Supplier() {
@Override
public AsyncServerExchangeHandler get() {
return new AsyncServerExchangeHandler() {
ByteBuffer buffer = ByteBuffer.allocate(2048);
CapacityChannel inputCapacityChannel;
DataStreamChannel outputDataChannel;
boolean endStream;
private void ensureCapacity(final int chunk) {
if (buffer.remaining() < chunk) {
final ByteBuffer oldBuffer = buffer;
oldBuffer.flip();
buffer = ByteBuffer.allocate(oldBuffer.remaining() + (chunk > 2048 ? chunk : 2048));
buffer.put(oldBuffer);
}
}
@Override
public void handleRequest(
final HttpRequest request,
final EntityDetails entityDetails,
final ResponseChannel responseChannel,
final HttpContext context) throws HttpException, IOException {
final HttpResponse response = new BasicHttpResponse(HttpStatus.SC_OK);
responseChannel.sendResponse(response, entityDetails, context);
}
@Override
public void consume(final ByteBuffer src) throws IOException {
if (buffer.position() == 0) {
if (outputDataChannel != null) {
outputDataChannel.write(src);
}
}
if (src.hasRemaining()) {
ensureCapacity(src.remaining());
buffer.put(src);
if (outputDataChannel != null) {
outputDataChannel.requestOutput();
}
}
}
@Override
public void updateCapacity(final CapacityChannel capacityChannel) throws IOException {
if (buffer.hasRemaining()) {
capacityChannel.update(buffer.remaining());
inputCapacityChannel = null;
} else {
inputCapacityChannel = capacityChannel;
}
}
@Override
public void streamEnd(final List trailers) throws IOException {
endStream = true;
if (buffer.position() == 0) {
if (outputDataChannel != null) {
outputDataChannel.endStream();
}
} else {
if (outputDataChannel != null) {
outputDataChannel.requestOutput();
}
}
}
@Override
public int available() {
return buffer.position();
}
@Override
public void produce(final DataStreamChannel channel) throws IOException {
outputDataChannel = channel;
buffer.flip();
if (buffer.hasRemaining()) {
channel.write(buffer);
}
buffer.compact();
if (buffer.position() == 0 && endStream) {
channel.endStream();
}
final CapacityChannel capacityChannel = inputCapacityChannel;
if (capacityChannel != null && buffer.hasRemaining()) {
capacityChannel.update(buffer.remaining());
}
}
@Override
public void failed(final Exception cause) {
if (!(cause instanceof SocketException)) {
cause.printStackTrace(System.out);
}
}
@Override
public void releaseResources() {
}
};
}
})
.create();
Runtime.getRuntime().addShutdownHook(new Thread() {
@Override
public void run() {
System.out.println("HTTP server shutting down");
server.close(CloseMode.GRACEFUL);
}
});
server.start();
final Future future = server.listen(new InetSocketAddress(port), URIScheme.HTTP);
final ListenerEndpoint listenerEndpoint = future.get();
System.out.print("Listening on " + listenerEndpoint.getAddress());
server.awaitShutdown(TimeValue.ofDays(Long.MAX_VALUE));
}
}
httpcore5-h2/src/test/java/org/apache/hc/core5/http2/examples/H2FileServerExample.java 0100664 0000000 0000000 00000023512 14157612023 027443 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.examples;
import java.io.File;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.List;
import java.util.Locale;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import org.apache.hc.core5.http.ContentType;
import org.apache.hc.core5.http.EndpointDetails;
import org.apache.hc.core5.http.EntityDetails;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.HttpConnection;
import org.apache.hc.core5.http.HttpException;
import org.apache.hc.core5.http.HttpRequest;
import org.apache.hc.core5.http.HttpStatus;
import org.apache.hc.core5.http.Message;
import org.apache.hc.core5.http.ProtocolException;
import org.apache.hc.core5.http.URIScheme;
import org.apache.hc.core5.http.impl.bootstrap.HttpAsyncServer;
import org.apache.hc.core5.http.nio.AsyncRequestConsumer;
import org.apache.hc.core5.http.nio.AsyncServerRequestHandler;
import org.apache.hc.core5.http.nio.entity.AsyncEntityProducers;
import org.apache.hc.core5.http.nio.entity.NoopEntityConsumer;
import org.apache.hc.core5.http.nio.support.AsyncResponseBuilder;
import org.apache.hc.core5.http.nio.support.BasicRequestConsumer;
import org.apache.hc.core5.http.protocol.HttpContext;
import org.apache.hc.core5.http.protocol.HttpCoreContext;
import org.apache.hc.core5.http2.HttpVersionPolicy;
import org.apache.hc.core5.http2.frame.RawFrame;
import org.apache.hc.core5.http2.impl.nio.H2StreamListener;
import org.apache.hc.core5.http2.impl.nio.bootstrap.H2ServerBootstrap;
import org.apache.hc.core5.io.CloseMode;
import org.apache.hc.core5.reactor.IOReactorConfig;
import org.apache.hc.core5.reactor.ListenerEndpoint;
import org.apache.hc.core5.util.TimeValue;
/**
* Example of asynchronous embedded HTTP/2 file server.
*/
public class H2FileServerExample {
public static void main(final String[] args) throws Exception {
if (args.length < 1) {
System.err.println("Please specify document root directory");
System.exit(1);
}
// Document root directory
final File docRoot = new File(args[0]);
int port = 8080;
if (args.length >= 2) {
port = Integer.parseInt(args[1]);
}
final IOReactorConfig config = IOReactorConfig.custom()
.setSoTimeout(15, TimeUnit.SECONDS)
.setTcpNoDelay(true)
.build();
final HttpAsyncServer server = H2ServerBootstrap.bootstrap()
.setIOReactorConfig(config)
.setVersionPolicy(HttpVersionPolicy.FORCE_HTTP_2)
.setStreamListener(new H2StreamListener() {
@Override
public void onHeaderInput(final HttpConnection connection, final int streamId, final List headers) {
for (int i = 0; i < headers.size(); i++) {
System.out.println(connection.getRemoteAddress() + " (" + streamId + ") << " + headers.get(i));
}
}
@Override
public void onHeaderOutput(final HttpConnection connection, final int streamId, final List headers) {
for (int i = 0; i < headers.size(); i++) {
System.out.println(connection.getRemoteAddress() + " (" + streamId + ") >> " + headers.get(i));
}
}
@Override
public void onFrameInput(final HttpConnection connection, final int streamId, final RawFrame frame) {
}
@Override
public void onFrameOutput(final HttpConnection connection, final int streamId, final RawFrame frame) {
}
@Override
public void onInputFlowControl(final HttpConnection connection, final int streamId, final int delta, final int actualSize) {
}
@Override
public void onOutputFlowControl(final HttpConnection connection, final int streamId, final int delta, final int actualSize) {
}
})
.register("*", new AsyncServerRequestHandler>() {
@Override
public AsyncRequestConsumer> prepare(
final HttpRequest request,
final EntityDetails entityDetails,
final HttpContext context) throws HttpException {
return new BasicRequestConsumer<>(entityDetails != null ? new NoopEntityConsumer() : null);
}
@Override
public void handle(
final Message message,
final ResponseTrigger responseTrigger,
final HttpContext context) throws HttpException, IOException {
final HttpRequest request = message.getHead();
final URI requestUri;
try {
requestUri = request.getUri();
} catch (final URISyntaxException ex) {
throw new ProtocolException(ex.getMessage(), ex);
}
final String path = requestUri.getPath();
final File file = new File(docRoot, path);
if (!file.exists()) {
System.out.println("File " + file.getPath() + " not found");
responseTrigger.submitResponse(
AsyncResponseBuilder.create(HttpStatus.SC_NOT_FOUND)
.setEntity("File" + file.getPath() +
" not found
", ContentType.TEXT_HTML)
.build(),
context);
} else if (!file.canRead() || file.isDirectory()) {
System.out.println("Cannot read file " + file.getPath());
responseTrigger.submitResponse(
AsyncResponseBuilder.create(HttpStatus.SC_FORBIDDEN)
.setEntity("Access denied
", ContentType.TEXT_HTML)
.build(),
context);
} else {
final ContentType contentType;
final String filename = file.getName().toLowerCase(Locale.ROOT);
if (filename.endsWith(".txt")) {
contentType = ContentType.TEXT_PLAIN;
} else if (filename.endsWith(".html") || filename.endsWith(".htm")) {
contentType = ContentType.TEXT_HTML;
} else if (filename.endsWith(".xml")) {
contentType = ContentType.TEXT_XML;
} else {
contentType = ContentType.DEFAULT_BINARY;
}
final HttpCoreContext coreContext = HttpCoreContext.adapt(context);
final EndpointDetails endpoint = coreContext.getEndpointDetails();
System.out.println(endpoint + ": serving file " + file.getPath());
responseTrigger.submitResponse(
AsyncResponseBuilder.create(HttpStatus.SC_OK)
.setEntity(AsyncEntityProducers.create(file, contentType))
.build(),
context);
}
}
})
.create();
Runtime.getRuntime().addShutdownHook(new Thread() {
@Override
public void run() {
System.out.println("HTTP server shutting down");
server.close(CloseMode.GRACEFUL);
}
});
server.start();
final Future future = server.listen(new InetSocketAddress(port), URIScheme.HTTP);
final ListenerEndpoint listenerEndpoint = future.get();
System.out.print("Listening on " + listenerEndpoint.getAddress());
server.awaitShutdown(TimeValue.ofDays(Long.MAX_VALUE));
}
}
httpcore5-h2/src/test/java/org/apache/hc/core5/http2/examples/H2TlsAlpnRequestExecutionExample.java 0100664 0000000 0000000 00000017552 14157612023 032216 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.examples;
import java.util.List;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.Future;
import javax.net.ssl.SSLEngine;
import javax.net.ssl.SSLException;
import org.apache.hc.core5.concurrent.FutureCallback;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.HttpConnection;
import org.apache.hc.core5.http.HttpHost;
import org.apache.hc.core5.http.HttpResponse;
import org.apache.hc.core5.http.Message;
import org.apache.hc.core5.http.impl.bootstrap.HttpAsyncRequester;
import org.apache.hc.core5.http.nio.AsyncClientEndpoint;
import org.apache.hc.core5.http.nio.entity.StringAsyncEntityConsumer;
import org.apache.hc.core5.http.nio.support.AsyncRequestBuilder;
import org.apache.hc.core5.http.nio.support.BasicResponseConsumer;
import org.apache.hc.core5.http2.config.H2Config;
import org.apache.hc.core5.http2.frame.RawFrame;
import org.apache.hc.core5.http2.impl.nio.H2StreamListener;
import org.apache.hc.core5.http2.impl.nio.bootstrap.H2RequesterBootstrap;
import org.apache.hc.core5.http2.ssl.H2ClientTlsStrategy;
import org.apache.hc.core5.io.CloseMode;
import org.apache.hc.core5.net.NamedEndpoint;
import org.apache.hc.core5.reactor.ssl.SSLSessionVerifier;
import org.apache.hc.core5.reactor.ssl.TlsDetails;
import org.apache.hc.core5.ssl.SSLContexts;
import org.apache.hc.core5.util.Timeout;
/**
* This example demonstrates how to execute HTTP/2 requests over TLS connections.
*
* It requires Java runtime with ALPN protocol support (such as Oracle JRE 9 or newer).
*/
public class H2TlsAlpnRequestExecutionExample {
public final static void main(final String[] args) throws Exception {
// Create and start requester
final H2Config h2Config = H2Config.custom()
.setPushEnabled(false)
.build();
final HttpAsyncRequester requester = H2RequesterBootstrap.bootstrap()
.setH2Config(h2Config)
.setTlsStrategy(new H2ClientTlsStrategy(SSLContexts.createSystemDefault(), new SSLSessionVerifier() {
@Override
public TlsDetails verify(final NamedEndpoint endpoint, final SSLEngine sslEngine) throws SSLException {
// IMPORTANT uncomment the following line when running Java 9 or older
// in order to avoid the illegal reflective access operation warning
// ====
// return new TlsDetails(sslEngine.getSession(), sslEngine.getApplicationProtocol());
// ====
return null;
}
}))
.setStreamListener(new H2StreamListener() {
@Override
public void onHeaderInput(final HttpConnection connection, final int streamId, final List headers) {
for (int i = 0; i < headers.size(); i++) {
System.out.println(connection.getRemoteAddress() + " (" + streamId + ") << " + headers.get(i));
}
}
@Override
public void onHeaderOutput(final HttpConnection connection, final int streamId, final List headers) {
for (int i = 0; i < headers.size(); i++) {
System.out.println(connection.getRemoteAddress() + " (" + streamId + ") >> " + headers.get(i));
}
}
@Override
public void onFrameInput(final HttpConnection connection, final int streamId, final RawFrame frame) {
}
@Override
public void onFrameOutput(final HttpConnection connection, final int streamId, final RawFrame frame) {
}
@Override
public void onInputFlowControl(final HttpConnection connection, final int streamId, final int delta, final int actualSize) {
}
@Override
public void onOutputFlowControl(final HttpConnection connection, final int streamId, final int delta, final int actualSize) {
}
})
.create();
Runtime.getRuntime().addShutdownHook(new Thread() {
@Override
public void run() {
System.out.println("HTTP requester shutting down");
requester.close(CloseMode.GRACEFUL);
}
});
requester.start();
final HttpHost target = new HttpHost("https", "nghttp2.org", 443);
final String[] requestUris = new String[] {"/httpbin/ip", "/httpbin/user-agent", "/httpbin/headers"};
final CountDownLatch latch = new CountDownLatch(requestUris.length);
for (final String requestUri: requestUris) {
final Future future = requester.connect(target, Timeout.ofSeconds(5));
final AsyncClientEndpoint clientEndpoint = future.get();
clientEndpoint.execute(
AsyncRequestBuilder.get()
.setHttpHost(target)
.setPath(requestUri)
.build(),
new BasicResponseConsumer<>(new StringAsyncEntityConsumer()),
new FutureCallback>() {
@Override
public void completed(final Message message) {
clientEndpoint.releaseAndReuse();
final HttpResponse response = message.getHead();
final String body = message.getBody();
System.out.println(requestUri + "->" + response.getCode() + " " + response.getVersion());
System.out.println(body);
latch.countDown();
}
@Override
public void failed(final Exception ex) {
clientEndpoint.releaseAndDiscard();
System.out.println(requestUri + "->" + ex);
latch.countDown();
}
@Override
public void cancelled() {
clientEndpoint.releaseAndDiscard();
System.out.println(requestUri + " cancelled");
latch.countDown();
}
});
}
latch.await();
System.out.println("Shutting down I/O reactor");
requester.initiateShutdown();
}
}
httpcore5-h2/src/test/java/org/apache/hc/core5/http2/impl/ 0040755 0000000 0000000 00000000000 14157612023 022145 5 ustar 00 0000000 0000000 httpcore5-h2/src/test/java/org/apache/hc/core5/http2/impl/TestDefaultH2ResponseConverter.java 0100664 0000000 0000000 00000020331 14157612023 031033 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl;
import java.util.Arrays;
import java.util.List;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.HttpException;
import org.apache.hc.core5.http.HttpResponse;
import org.apache.hc.core5.http.message.BasicHeader;
import org.apache.hc.core5.http.message.BasicHttpResponse;
import org.junit.Assert;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.ExpectedException;
public class TestDefaultH2ResponseConverter {
@Rule
public ExpectedException thrown = ExpectedException.none();
@Test
public void testConvertFromFieldsBasic() throws Exception {
final List headers = Arrays.asList(
new BasicHeader(":status", "200"),
new BasicHeader("location", "http://www.example.com/"),
new BasicHeader("custom123", "value"));
final DefaultH2ResponseConverter converter = new DefaultH2ResponseConverter();
final HttpResponse response = converter.convert(headers);
Assert.assertNotNull(response );
Assert.assertEquals(200, response .getCode());
final Header[] allHeaders = response.getHeaders();
Assert.assertEquals(2, allHeaders.length);
Assert.assertEquals("location", allHeaders[0].getName());
Assert.assertEquals("http://www.example.com/", allHeaders[0].getValue());
Assert.assertEquals("custom123", allHeaders[1].getName());
Assert.assertEquals("value", allHeaders[1].getValue());
}
@Test
public void testConvertFromFieldsUpperCaseHeaderName() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Header name ':Status' is invalid (header name contains uppercase characters)");
final List headers = Arrays.asList(
new BasicHeader(":Status", "200"),
new BasicHeader("location", "http://www.example.com/"),
new BasicHeader("custom123", "value"));
final DefaultH2ResponseConverter converter = new DefaultH2ResponseConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsInvalidStatusCode() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Invalid response status: boom");
final List headers = Arrays.asList(
new BasicHeader(":status", "boom"),
new BasicHeader("location", "http://www.example.com/"),
new BasicHeader("custom123", "value"));
final DefaultH2ResponseConverter converter = new DefaultH2ResponseConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsConnectionHeader() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Header 'connection: keep-alive' is illegal for HTTP/2 messages");
final List headers = Arrays.asList(
new BasicHeader(":status", "200"),
new BasicHeader("location", "http://www.example.com/"),
new BasicHeader("connection", "keep-alive"));
final DefaultH2ResponseConverter converter = new DefaultH2ResponseConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsMissingStatus() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Mandatory response header ':status' not found");
final List headers = Arrays.asList(
new BasicHeader("location", "http://www.example.com/"),
new BasicHeader("custom", "value"));
final DefaultH2ResponseConverter converter = new DefaultH2ResponseConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsUnknownPseudoHeader() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Unsupported response header ':custom'");
final List headers = Arrays.asList(
new BasicHeader(":status", "200"),
new BasicHeader(":custom", "200"),
new BasicHeader("location", "http://www.example.com/"),
new BasicHeader("custom1", "value"));
final DefaultH2ResponseConverter converter = new DefaultH2ResponseConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsMultipleStatus() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Multiple ':status' response headers are illegal");
final List headers = Arrays.asList(
new BasicHeader(":status", "200"),
new BasicHeader(":status", "200"),
new BasicHeader("location", "http://www.example.com/"),
new BasicHeader("custom1", "value"));
final DefaultH2ResponseConverter converter = new DefaultH2ResponseConverter();
converter.convert(headers);
}
@Test
public void testConvertFromMessageBasic() throws Exception {
final HttpResponse response = new BasicHttpResponse(200);
response.addHeader("custom123", "Value");
final DefaultH2ResponseConverter converter = new DefaultH2ResponseConverter();
final List headers = converter.convert(response);
Assert.assertNotNull(headers);
Assert.assertEquals(2, headers.size());
final Header header1 = headers.get(0);
Assert.assertEquals(":status", header1.getName());
Assert.assertEquals("200", header1.getValue());
final Header header2 = headers.get(1);
Assert.assertEquals("custom123", header2.getName());
Assert.assertEquals("Value", header2.getValue());
}
@Test
public void testConvertFromMessageInvalidStatus() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Response status 99 is invalid");
final HttpResponse response = new BasicHttpResponse(99);
response.addHeader("Custom123", "Value");
final DefaultH2ResponseConverter converter = new DefaultH2ResponseConverter();
converter.convert(response);
}
@Test
public void testConvertFromMessageConnectionHeader() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Header 'Connection: Keep-Alive' is illegal for HTTP/2 messages");
final HttpResponse response = new BasicHttpResponse(200);
response.addHeader("Connection", "Keep-Alive");
final DefaultH2ResponseConverter converter = new DefaultH2ResponseConverter();
converter.convert(response);
}
@Test
public void testConvertFromMessageInvalidHeader() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Header name ':custom' is invalid");
final HttpResponse response = new BasicHttpResponse(200);
response.addHeader(":custom", "stuff");
final DefaultH2ResponseConverter converter = new DefaultH2ResponseConverter();
converter.convert(response);
}
}
httpcore5-h2/src/test/java/org/apache/hc/core5/http2/impl/TestDefaultH2RequestConverter.java 0100664 0000000 0000000 00000041271 14157612023 030673 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl;
import java.util.Arrays;
import java.util.List;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.HttpException;
import org.apache.hc.core5.http.HttpHost;
import org.apache.hc.core5.http.HttpRequest;
import org.apache.hc.core5.http.message.BasicHeader;
import org.apache.hc.core5.http.message.BasicHttpRequest;
import org.apache.hc.core5.net.URIAuthority;
import org.junit.Assert;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.ExpectedException;
public class TestDefaultH2RequestConverter {
@Rule
public ExpectedException thrown = ExpectedException.none();
@Test
public void testConvertFromFieldsBasic() throws Exception {
final List headers = Arrays.asList(
new BasicHeader(":method", "GET"),
new BasicHeader(":scheme", "http"),
new BasicHeader(":authority", "www.example.com"),
new BasicHeader(":path", "/"),
new BasicHeader("custom123", "value"));
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
final HttpRequest request = converter.convert(headers);
Assert.assertNotNull(request);
Assert.assertEquals("GET", request.getMethod());
Assert.assertEquals("http", request.getScheme());
Assert.assertEquals(new URIAuthority("www.example.com"), request.getAuthority());
Assert.assertEquals("/", request.getPath());
final Header[] allHeaders = request.getHeaders();
Assert.assertEquals(1, allHeaders.length);
Assert.assertEquals("custom123", allHeaders[0].getName());
Assert.assertEquals("value", allHeaders[0].getValue());
}
@Test
public void testConvertFromFieldsUpperCaseHeaderName() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Header name ':Path' is invalid (header name contains uppercase characters)");
final List headers = Arrays.asList(
new BasicHeader(":method", "GET"),
new BasicHeader(":scheme", "http"),
new BasicHeader(":authority", "www.example.com"),
new BasicHeader(":Path", "/"),
new BasicHeader("custom", "value"));
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsConnectionHeader() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Header 'connection: keep-alive' is illegal for HTTP/2 messages");
final List headers = Arrays.asList(
new BasicHeader(":method", "GET"),
new BasicHeader(":scheme", "http"),
new BasicHeader(":authority", "www.example.com"),
new BasicHeader(":path", "/"),
new BasicHeader("connection", "keep-alive"));
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsPseudoHeaderSequence() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Invalid sequence of headers (pseudo-headers must precede message headers)");
final List headers = Arrays.asList(
new BasicHeader(":method", "GET"),
new BasicHeader(":scheme", "http"),
new BasicHeader("custom", "value"),
new BasicHeader(":authority", "www.example.com"),
new BasicHeader(":path", "/"));
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsMissingMethod() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Mandatory request header ':method' not found");
final List headers = Arrays.asList(
new BasicHeader(":scheme", "http"),
new BasicHeader(":authority", "www.example.com"),
new BasicHeader(":path", "/"),
new BasicHeader("custom", "value"));
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsMissingScheme() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Mandatory request header ':scheme' not found");
final List headers = Arrays.asList(
new BasicHeader(":method", "GET"),
new BasicHeader(":authority", "www.example.com"),
new BasicHeader(":path", "/"),
new BasicHeader("custom", "value"));
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsMissingPath() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Mandatory request header ':path' not found");
final List headers = Arrays.asList(
new BasicHeader(":method", "GET"),
new BasicHeader(":scheme", "http"),
new BasicHeader(":authority", "www.example.com"),
new BasicHeader("custom", "value"));
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsUnknownPseudoHeader() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Unsupported request header ':custom'");
final List headers = Arrays.asList(
new BasicHeader(":method", "GET"),
new BasicHeader(":scheme", "http"),
new BasicHeader(":authority", "www.example.com"),
new BasicHeader(":path", "/"),
new BasicHeader(":custom", "value"));
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsMultipleMethod() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Multiple ':method' request headers are illegal");
final List headers = Arrays.asList(
new BasicHeader(":method", "GET"),
new BasicHeader(":method", "GET"),
new BasicHeader(":scheme", "http"),
new BasicHeader(":authority", "www.example.com"),
new BasicHeader(":path", "/"),
new BasicHeader("custom", "value"));
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsMultipleScheme() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Multiple ':scheme' request headers are illegal");
final List headers = Arrays.asList(
new BasicHeader(":method", "GET"),
new BasicHeader(":scheme", "http"),
new BasicHeader(":scheme", "https"),
new BasicHeader(":authority", "www.example.com"),
new BasicHeader(":path", "/"),
new BasicHeader("custom", "value"));
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsMultiplePath() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Multiple ':path' request headers are illegal");
final List headers = Arrays.asList(
new BasicHeader(":method", "GET"),
new BasicHeader(":scheme", "https"),
new BasicHeader(":authority", "www.example.com"),
new BasicHeader(":path", "/"),
new BasicHeader(":path", "/"),
new BasicHeader("custom", "value"));
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsConnect() throws Exception {
final List headers = Arrays.asList(
new BasicHeader(":method", "CONNECT"),
new BasicHeader(":authority", "www.example.com"),
new BasicHeader("custom", "value"));
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsConnectMissingAuthority() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Header ':authority' is mandatory for CONNECT request");
final List headers = Arrays.asList(
new BasicHeader(":method", "CONNECT"),
new BasicHeader("custom", "value"));
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsConnectPresentScheme() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Header ':scheme' must not be set for CONNECT request");
final List headers = Arrays.asList(
new BasicHeader(":method", "CONNECT"),
new BasicHeader(":scheme", "http"),
new BasicHeader(":authority", "www.example.com"),
new BasicHeader("custom", "value"));
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(headers);
}
@Test
public void testConvertFromFieldsConnectPresentPath() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Header ':path' must not be set for CONNECT request");
final List headers = Arrays.asList(
new BasicHeader(":method", "CONNECT"),
new BasicHeader(":authority", "www.example.com"),
new BasicHeader(":path", "/"),
new BasicHeader("custom", "value"));
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(headers);
}
@Test
public void testConvertFromMessageBasic() throws Exception {
final HttpRequest request = new BasicHttpRequest("GET", new HttpHost("host"), "/");
request.addHeader("custom123", "Value");
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
final List headers = converter.convert(request);
Assert.assertNotNull(headers);
Assert.assertEquals(5, headers.size());
final Header header1 = headers.get(0);
Assert.assertEquals(":method", header1.getName());
Assert.assertEquals("GET", header1.getValue());
final Header header2 = headers.get(1);
Assert.assertEquals(":scheme", header2.getName());
Assert.assertEquals("http", header2.getValue());
final Header header3 = headers.get(2);
Assert.assertEquals(":authority", header3.getName());
Assert.assertEquals("host", header3.getValue());
final Header header4 = headers.get(3);
Assert.assertEquals(":path", header4.getName());
Assert.assertEquals("/", header4.getValue());
final Header header5 = headers.get(4);
Assert.assertEquals("custom123", header5.getName());
Assert.assertEquals("Value", header5.getValue());
}
@Test
public void testConvertFromMessageMissingScheme() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Request scheme is not set");
final HttpRequest request = new BasicHttpRequest("GET", new HttpHost("host"), "/");
request.addHeader("Custom123", "Value");
request.setScheme(null);
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(request);
}
@Test
public void testConvertFromMessageMissingPath() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Request path is not set");
final HttpRequest request = new BasicHttpRequest("GET", new HttpHost("host"), "/");
request.addHeader("Custom123", "Value");
request.setPath(null);
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(request);
}
@Test
public void testConvertFromMessageConnect() throws Exception {
final HttpRequest request = new BasicHttpRequest("CONNECT", new HttpHost("host:80"), null);
request.addHeader("custom123", "Value");
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
final List headers = converter.convert(request);
Assert.assertNotNull(headers);
Assert.assertEquals(3, headers.size());
final Header header1 = headers.get(0);
Assert.assertEquals(":method", header1.getName());
Assert.assertEquals("CONNECT", header1.getValue());
final Header header2 = headers.get(1);
Assert.assertEquals(":authority", header2.getName());
Assert.assertEquals("host:80", header2.getValue());
final Header header3 = headers.get(2);
Assert.assertEquals("custom123", header3.getName());
Assert.assertEquals("Value", header3.getValue());
}
@Test
public void testConvertFromMessageConnectMissingAuthority() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("CONNECT request authority is not set");
final HttpRequest request = new BasicHttpRequest("CONNECT", null, null);
request.addHeader("Custom123", "Value");
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(request);
}
@Test
public void testConvertFromMessageConnectWithPath() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("CONNECT request path must be null");
final HttpRequest request = new BasicHttpRequest("CONNECT", "/");
request.setAuthority(new URIAuthority("host"));
request.addHeader("Custom123", "Value");
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(request);
}
@Test
public void testConvertFromMessageConnectionHeader() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Header 'Connection: Keep-Alive' is illegal for HTTP/2 messages");
final HttpRequest request = new BasicHttpRequest("GET", new HttpHost("host"), "/");
request.addHeader("Connection", "Keep-Alive");
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(request);
}
@Test
public void testConvertFromMessageInvalidHeader() throws Exception {
thrown.expect(HttpException.class);
thrown.expectMessage("Header name ':custom' is invalid");
final HttpRequest request = new BasicHttpRequest("GET", new HttpHost("host"), "/");
request.addHeader(":custom", "stuff");
final DefaultH2RequestConverter converter = new DefaultH2RequestConverter();
converter.convert(request);
}
}
httpcore5-h2/src/test/java/org/apache/hc/core5/http2/impl/io/ 0040755 0000000 0000000 00000000000 14157612023 022554 5 ustar 00 0000000 0000000 httpcore5-h2/src/test/java/org/apache/hc/core5/http2/impl/io/TestFrameInOutBuffers.java 0100664 0000000 0000000 00000025717 14157612023 027620 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.io;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.nio.ByteBuffer;
import org.apache.hc.core5.http.ConnectionClosedException;
import org.apache.hc.core5.http2.H2ConnectionException;
import org.apache.hc.core5.http2.H2CorruptFrameException;
import org.apache.hc.core5.http2.frame.FrameConsts;
import org.apache.hc.core5.http2.frame.FrameFlag;
import org.apache.hc.core5.http2.frame.FrameType;
import org.apache.hc.core5.http2.frame.RawFrame;
import org.apache.hc.core5.http2.impl.BasicH2TransportMetrics;
import org.junit.Assert;
import org.junit.Test;
public class TestFrameInOutBuffers {
@Test
public void testReadWriteFrame() throws Exception {
final ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
final FrameOutputBuffer outbuffer = new FrameOutputBuffer(16 * 1024);
final RawFrame frame = new RawFrame(FrameType.DATA.getValue(), 0, 1,
ByteBuffer.wrap(new byte[]{1,2,3,4,5}));
outbuffer.write(frame, outputStream);
final FrameInputBuffer inBuffer = new FrameInputBuffer(16 * 1024);
final byte[] bytes = outputStream.toByteArray();
final ByteArrayInputStream inputStream = new ByteArrayInputStream(bytes);
Assert.assertEquals(FrameConsts.HEAD_LEN + 5, bytes.length);
Assert.assertEquals(1, outbuffer.getMetrics().getFramesTransferred());
Assert.assertEquals(bytes.length, outbuffer.getMetrics().getBytesTransferred());
final RawFrame frame2 = inBuffer.read(inputStream);
Assert.assertEquals(FrameType.DATA.getValue(), frame2.getType());
Assert.assertEquals(0, frame2.getFlags());
Assert.assertEquals(1L, frame2.getStreamId());
final ByteBuffer payload2 = frame2.getPayloadContent();
Assert.assertNotNull(payload2);
Assert.assertEquals(5, payload2.remaining());
Assert.assertEquals(1, payload2.get());
Assert.assertEquals(2, payload2.get());
Assert.assertEquals(3, payload2.get());
Assert.assertEquals(4, payload2.get());
Assert.assertEquals(5, payload2.get());
Assert.assertEquals(-1, inputStream.read());
Assert.assertEquals(1, inBuffer.getMetrics().getFramesTransferred());
Assert.assertEquals(bytes.length, inBuffer.getMetrics().getBytesTransferred());
}
@Test
public void testReadFrameMultiple() throws Exception {
final FrameInputBuffer inBuffer = new FrameInputBuffer(16 * 1024);
final ByteArrayInputStream inputStream = new ByteArrayInputStream(
new byte[] {
0,0,10,0,8,0,0,0,8,4,0,1,2,3,4,0,0,0,0,
0,0,10,0,9,0,0,0,8,4,5,6,7,8,9,0,0,0,0
});
final RawFrame frame1 = inBuffer.read(inputStream);
Assert.assertEquals(FrameType.DATA, FrameType.valueOf(frame1.getType()));
Assert.assertEquals(8, frame1.getFlags());
Assert.assertEquals(8, frame1.getStreamId());
final ByteBuffer payload1 = frame1.getPayloadContent();
Assert.assertNotNull(payload1);
Assert.assertEquals(5, payload1.remaining());
Assert.assertEquals(0, payload1.get());
Assert.assertEquals(1, payload1.get());
Assert.assertEquals(2, payload1.get());
Assert.assertEquals(3, payload1.get());
Assert.assertEquals(4, payload1.get());
final RawFrame frame2 = inBuffer.read(inputStream);
Assert.assertEquals(FrameType.DATA, FrameType.valueOf(frame2.getType()));
Assert.assertEquals(FrameFlag.of(FrameFlag.END_STREAM, FrameFlag.PADDED), frame2.getFlags());
Assert.assertEquals(8, frame2.getStreamId());
final ByteBuffer payload2 = frame2.getPayloadContent();
Assert.assertNotNull(payload2);
Assert.assertEquals(5, payload2.remaining());
Assert.assertEquals(5, payload2.get());
Assert.assertEquals(6, payload2.get());
Assert.assertEquals(7, payload2.get());
Assert.assertEquals(8, payload2.get());
Assert.assertEquals(9, payload2.get());
Assert.assertEquals(-1, inputStream.read());
}
@Test
public void testReadFrameMultipleSmallBuffer() throws Exception {
final FrameInputBuffer inBuffer = new FrameInputBuffer(new BasicH2TransportMetrics(), 20, 10);
final ByteArrayInputStream inputStream = new ByteArrayInputStream(
new byte[] {
0,0,10,0,8,0,0,0,8,4,1,1,1,1,1,0,0,0,0,
0,0,5,0,0,0,0,0,8,2,2,2,2,2,
0,0,10,0,9,0,0,0,8,4,3,3,3,3,3,0,0,0,0
});
final RawFrame frame1 = inBuffer.read(inputStream);
Assert.assertEquals(FrameType.DATA, FrameType.valueOf(frame1.getType()));
Assert.assertEquals(8, frame1.getFlags());
Assert.assertEquals(8, frame1.getStreamId());
final ByteBuffer payload1 = frame1.getPayloadContent();
Assert.assertNotNull(payload1);
Assert.assertEquals(5, payload1.remaining());
Assert.assertEquals(1, payload1.get());
Assert.assertEquals(1, payload1.get());
Assert.assertEquals(1, payload1.get());
Assert.assertEquals(1, payload1.get());
Assert.assertEquals(1, payload1.get());
final RawFrame frame2 = inBuffer.read(inputStream);
Assert.assertEquals(FrameType.DATA, FrameType.valueOf(frame2.getType()));
Assert.assertEquals(0, frame2.getFlags());
Assert.assertEquals(8, frame2.getStreamId());
final ByteBuffer payload2 = frame2.getPayloadContent();
Assert.assertNotNull(payload2);
Assert.assertEquals(5, payload2.remaining());
Assert.assertEquals(2, payload2.get());
Assert.assertEquals(2, payload2.get());
Assert.assertEquals(2, payload2.get());
Assert.assertEquals(2, payload2.get());
Assert.assertEquals(2, payload2.get());
final RawFrame frame3 = inBuffer.read(inputStream);
Assert.assertEquals(FrameType.DATA, FrameType.valueOf(frame3.getType()));
Assert.assertEquals(FrameFlag.of(FrameFlag.END_STREAM, FrameFlag.PADDED), frame3.getFlags());
Assert.assertEquals(8, frame3.getStreamId());
final ByteBuffer payload3 = frame3.getPayloadContent();
Assert.assertNotNull(payload3);
Assert.assertEquals(5, payload3.remaining());
Assert.assertEquals(3, payload3.get());
Assert.assertEquals(3, payload3.get());
Assert.assertEquals(3, payload3.get());
Assert.assertEquals(3, payload3.get());
Assert.assertEquals(3, payload3.get());
Assert.assertEquals(-1, inputStream.read());
}
@Test
public void testReadFramePartialReads() throws Exception {
final FrameInputBuffer inBuffer = new FrameInputBuffer(16 * 1024);
final MultiByteArrayInputStream inputStream = new MultiByteArrayInputStream(
new byte[] {0,0},
new byte[] {10,0,9,0},
new byte[] {0,0,8},
new byte[] {4},
new byte[] {1,2,3,4},
new byte[] {5,0},
new byte[] {0,0,0});
final RawFrame frame = inBuffer.read(inputStream);
Assert.assertEquals(FrameType.DATA, FrameType.valueOf(frame.getType()));
Assert.assertEquals(FrameFlag.of(FrameFlag.END_STREAM, FrameFlag.PADDED), frame.getFlags());
Assert.assertEquals(8, frame.getStreamId());
final ByteBuffer payload = frame.getPayloadContent();
Assert.assertNotNull(payload);
Assert.assertEquals(5, payload.remaining());
Assert.assertEquals(1, payload.get());
Assert.assertEquals(2, payload.get());
Assert.assertEquals(3, payload.get());
Assert.assertEquals(4, payload.get());
Assert.assertEquals(5, payload.get());
Assert.assertEquals(-1, inputStream.read());
}
@Test
public void testReadEmptyFrame() throws Exception {
final FrameInputBuffer inBuffer = new FrameInputBuffer(16 * 1024);
final ByteArrayInputStream inputStream = new ByteArrayInputStream(new byte[] {0,0,0,0,0,0,0,0,0});
final RawFrame frame = inBuffer.read(inputStream);
Assert.assertEquals(FrameType.DATA, FrameType.valueOf(frame.getType()));
Assert.assertEquals(0, frame.getFlags());
Assert.assertEquals(0, frame.getStreamId());
final ByteBuffer payload = frame.getPayloadContent();
Assert.assertNull(payload);
}
@Test(expected = ConnectionClosedException.class)
public void testReadFrameConnectionClosed() throws Exception {
final FrameInputBuffer inBuffer = new FrameInputBuffer(16 * 1024);
final ByteArrayInputStream inputStream = new ByteArrayInputStream(new byte[] {});
inBuffer.read(inputStream);
}
@Test(expected = H2CorruptFrameException.class)
public void testReadFrameCorruptFrame() throws Exception {
final FrameInputBuffer inBuffer = new FrameInputBuffer(16 * 1024);
final ByteArrayInputStream inputStream = new ByteArrayInputStream(new byte[] {0,0});
inBuffer.read(inputStream);
}
@Test(expected = H2ConnectionException.class)
public void testWriteFrameExceedingLimit() throws Exception {
final ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
final FrameOutputBuffer outbuffer = new FrameOutputBuffer(1024);
final RawFrame frame = new RawFrame(FrameType.DATA.getValue(), 0, 1,
ByteBuffer.wrap(new byte[2048]));
outbuffer.write(frame, outputStream);
}
@Test(expected = H2ConnectionException.class)
public void testReadFrameExceedingLimit() throws Exception {
final FrameInputBuffer inBuffer = new FrameInputBuffer(16 * 1024);
final ByteArrayInputStream inputStream = new ByteArrayInputStream(
new byte[] {0,-128,-128,0,0,0,0,0,1});
inBuffer.read(inputStream);
}
}
httpcore5-h2/src/test/java/org/apache/hc/core5/http2/impl/io/MultiByteArrayInputStream.java 0100644 0000000 0000000 00000006536 13072725455 030550 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.io;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayDeque;
import java.util.Queue;
class MultiByteArrayInputStream extends InputStream {
private final Queue bufs;
private byte[] current;
private int pos;
public MultiByteArrayInputStream(final byte[]... bufs) {
super();
this.bufs = new ArrayDeque<>();
for (final byte[] buf: bufs) {
if (buf.length > 0) {
this.bufs.add(buf);
}
}
}
private void advance() {
if (this.current != null) {
if (this.pos >= this.current.length) {
this.current = null;
}
}
if (this.current == null) {
this.current = this.bufs.poll();
this.pos = 0;
}
}
@Override
public int read() throws IOException {
advance();
if (this.current == null) {
return -1;
}
return this.current[this.pos++];
}
@Override
public int read(final byte b[], final int off, final int len) throws IOException {
if (b == null) {
throw new NullPointerException();
} else if ((off < 0) || (off > b.length) || (len < 0) ||
((off + len) > b.length) || ((off + len) < 0)) {
throw new IndexOutOfBoundsException();
}
advance();
if (this.current == null) {
return -1;
}
final int chunk = Math.min(this.current.length - this.pos, len);
if (chunk <= 0) {
return 0;
}
System.arraycopy(this.current, this.pos, b, off, chunk);
this.pos += chunk;
return chunk;
}
@Override
public long skip(final long n) {
advance();
final int chunk = Math.min(this.current.length - this.pos, (int) n);
if (chunk <= 0) {
return 0;
}
this.pos += chunk;
return chunk;
}
@Override
public int available() {
advance();
return this.current != null ? this.current.length - this.pos : 0;
}
@Override
public boolean markSupported() {
return false;
}
}
httpcore5-h2/src/test/java/org/apache/hc/core5/http2/impl/nio/ 0040755 0000000 0000000 00000000000 14157612023 022732 5 ustar 00 0000000 0000000 httpcore5-h2/src/test/java/org/apache/hc/core5/http2/impl/nio/TestFrameInOutBuffers.java 0100664 0000000 0000000 00000031142 14157612023 027763 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import java.nio.ByteBuffer;
import org.apache.hc.core5.http.ConnectionClosedException;
import org.apache.hc.core5.http2.H2ConnectionException;
import org.apache.hc.core5.http2.H2CorruptFrameException;
import org.apache.hc.core5.http2.ReadableByteChannelMock;
import org.apache.hc.core5.http2.WritableByteChannelMock;
import org.apache.hc.core5.http2.frame.FrameConsts;
import org.apache.hc.core5.http2.frame.FrameFlag;
import org.apache.hc.core5.http2.frame.FrameType;
import org.apache.hc.core5.http2.frame.RawFrame;
import org.apache.hc.core5.http2.impl.BasicH2TransportMetrics;
import org.junit.Assert;
import org.junit.Test;
public class TestFrameInOutBuffers {
@Test
public void testReadWriteFrame() throws Exception {
final WritableByteChannelMock writableChannel = new WritableByteChannelMock(1024);
final FrameOutputBuffer outbuffer = new FrameOutputBuffer(16 * 1024);
final RawFrame frame = new RawFrame(FrameType.DATA.getValue(), 0, 1,
ByteBuffer.wrap(new byte[]{1,2,3,4,5}));
outbuffer.write(frame, writableChannel);
final FrameInputBuffer inBuffer = new FrameInputBuffer(16 * 1024);
final byte[] bytes = writableChannel.toByteArray();
Assert.assertEquals(FrameConsts.HEAD_LEN + 5, bytes.length);
Assert.assertEquals(1, outbuffer.getMetrics().getFramesTransferred());
Assert.assertEquals(bytes.length, outbuffer.getMetrics().getBytesTransferred());
final ReadableByteChannelMock readableChannel = new ReadableByteChannelMock(bytes);
final RawFrame frame2 = inBuffer.read(readableChannel);
Assert.assertEquals(FrameType.DATA.getValue(), frame2.getType());
Assert.assertEquals(0, frame2.getFlags());
Assert.assertEquals(1L, frame2.getStreamId());
final ByteBuffer payload2 = frame2.getPayloadContent();
Assert.assertNotNull(payload2);
Assert.assertEquals(5, payload2.remaining());
Assert.assertEquals(1, payload2.get());
Assert.assertEquals(2, payload2.get());
Assert.assertEquals(3, payload2.get());
Assert.assertEquals(4, payload2.get());
Assert.assertEquals(5, payload2.get());
Assert.assertEquals(-1, readableChannel.read(ByteBuffer.allocate(1024)));
Assert.assertEquals(1, inBuffer.getMetrics().getFramesTransferred());
Assert.assertEquals(bytes.length, inBuffer.getMetrics().getBytesTransferred());
}
@Test
public void testPartialFrameWrite() throws Exception {
final WritableByteChannelMock writableChannel = new WritableByteChannelMock(1024, FrameConsts.HEAD_LEN + 10);
final FrameOutputBuffer outbuffer = new FrameOutputBuffer(16 * 1024);
final RawFrame frame = new RawFrame(FrameType.DATA.getValue(), FrameFlag.END_STREAM.getValue(), 5,
ByteBuffer.wrap(new byte[]{'0','1','2','3','4','5','6','7','8','9','a','b','c','d','e','f'}));
outbuffer.write(frame, writableChannel);
Assert.assertArrayEquals(new byte[] {0,0,16,0,1,0,0,0,5,48,49,50,51,52,53,54,55,56,57},
writableChannel.toByteArray());
Assert.assertFalse(outbuffer.isEmpty());
outbuffer.flush(writableChannel);
Assert.assertArrayEquals(new byte[] {0,0,16,0,1,0,0,0,5,48,49,50,51,52,53,54,55,56,57},
writableChannel.toByteArray());
writableChannel.flush();
outbuffer.flush(writableChannel);
Assert.assertArrayEquals(new byte[] {0,0,16,0,1,0,0,0,5,48,49,50,51,52,53,54,55,56,57,97,98,99,100,101,102},
writableChannel.toByteArray());
}
@Test
public void testReadFrameMultiple() throws Exception {
final FrameInputBuffer inBuffer = new FrameInputBuffer(16 * 1024);
final ReadableByteChannelMock readableChannel = new ReadableByteChannelMock(
new byte[] {
0,0,10,0,8,0,0,0,8,4,0,1,2,3,4,0,0,0,0,
0,0,10,0,9,0,0,0,8,4,5,6,7,8,9,0,0,0,0
});
final RawFrame frame1 = inBuffer.read(readableChannel);
Assert.assertEquals(FrameType.DATA, FrameType.valueOf(frame1.getType()));
Assert.assertEquals(8, frame1.getFlags());
Assert.assertEquals(8, frame1.getStreamId());
final ByteBuffer payload1 = frame1.getPayloadContent();
Assert.assertNotNull(payload1);
Assert.assertEquals(5, payload1.remaining());
Assert.assertEquals(0, payload1.get());
Assert.assertEquals(1, payload1.get());
Assert.assertEquals(2, payload1.get());
Assert.assertEquals(3, payload1.get());
Assert.assertEquals(4, payload1.get());
final RawFrame frame2 = inBuffer.read(readableChannel);
Assert.assertEquals(FrameType.DATA, FrameType.valueOf(frame2.getType()));
Assert.assertEquals(FrameFlag.of(FrameFlag.END_STREAM, FrameFlag.PADDED), frame2.getFlags());
Assert.assertEquals(8, frame2.getStreamId());
final ByteBuffer payload2 = frame2.getPayloadContent();
Assert.assertNotNull(payload2);
Assert.assertEquals(5, payload2.remaining());
Assert.assertEquals(5, payload2.get());
Assert.assertEquals(6, payload2.get());
Assert.assertEquals(7, payload2.get());
Assert.assertEquals(8, payload2.get());
Assert.assertEquals(9, payload2.get());
Assert.assertEquals(-1, readableChannel.read(ByteBuffer.allocate(1024)));
}
@Test
public void testReadFrameMultipleSmallBuffer() throws Exception {
final FrameInputBuffer inBuffer = new FrameInputBuffer(new BasicH2TransportMetrics(), 20, 10);
final ReadableByteChannelMock readableChannel = new ReadableByteChannelMock(
new byte[] {
0,0,10,0,8,0,0,0,8,4,1,1,1,1,1,0,0,0,0,
0,0,5,0,0,0,0,0,8,2,2,2,2,2,
0,0,10,0,9,0,0,0,8,4,3,3,3,3,3,0,0,0,0
});
final RawFrame frame1 = inBuffer.read(readableChannel);
Assert.assertEquals(FrameType.DATA, FrameType.valueOf(frame1.getType()));
Assert.assertEquals(8, frame1.getFlags());
Assert.assertEquals(8, frame1.getStreamId());
final ByteBuffer payload1 = frame1.getPayloadContent();
Assert.assertNotNull(payload1);
Assert.assertEquals(5, payload1.remaining());
Assert.assertEquals(1, payload1.get());
Assert.assertEquals(1, payload1.get());
Assert.assertEquals(1, payload1.get());
Assert.assertEquals(1, payload1.get());
Assert.assertEquals(1, payload1.get());
final RawFrame frame2 = inBuffer.read(readableChannel);
Assert.assertEquals(FrameType.DATA, FrameType.valueOf(frame2.getType()));
Assert.assertEquals(0, frame2.getFlags());
Assert.assertEquals(8, frame2.getStreamId());
final ByteBuffer payload2 = frame2.getPayloadContent();
Assert.assertNotNull(payload2);
Assert.assertEquals(5, payload2.remaining());
Assert.assertEquals(2, payload2.get());
Assert.assertEquals(2, payload2.get());
Assert.assertEquals(2, payload2.get());
Assert.assertEquals(2, payload2.get());
Assert.assertEquals(2, payload2.get());
final RawFrame frame3 = inBuffer.read(readableChannel);
Assert.assertEquals(FrameType.DATA, FrameType.valueOf(frame3.getType()));
Assert.assertEquals(FrameFlag.of(FrameFlag.END_STREAM, FrameFlag.PADDED), frame3.getFlags());
Assert.assertEquals(8, frame3.getStreamId());
final ByteBuffer payload3 = frame3.getPayloadContent();
Assert.assertNotNull(payload3);
Assert.assertEquals(5, payload3.remaining());
Assert.assertEquals(3, payload3.get());
Assert.assertEquals(3, payload3.get());
Assert.assertEquals(3, payload3.get());
Assert.assertEquals(3, payload3.get());
Assert.assertEquals(3, payload3.get());
Assert.assertEquals(-1, readableChannel.read(ByteBuffer.allocate(1024)));
}
@Test
public void testReadFramePartialReads() throws Exception {
final FrameInputBuffer inBuffer = new FrameInputBuffer(16 * 1024);
final ReadableByteChannelMock readableChannel = new ReadableByteChannelMock(
new byte[] {0,0},
new byte[] {10,0,9,0},
new byte[] {0,0,8},
new byte[] {4},
new byte[] {1,2,3,4},
new byte[] {5,0},
new byte[] {0,0,0});
final RawFrame frame = inBuffer.read(readableChannel);
Assert.assertEquals(FrameType.DATA, FrameType.valueOf(frame.getType()));
Assert.assertEquals(FrameFlag.of(FrameFlag.END_STREAM, FrameFlag.PADDED), frame.getFlags());
Assert.assertEquals(8, frame.getStreamId());
final ByteBuffer payload = frame.getPayloadContent();
Assert.assertNotNull(payload);
Assert.assertEquals(5, payload.remaining());
Assert.assertEquals(1, payload.get());
Assert.assertEquals(2, payload.get());
Assert.assertEquals(3, payload.get());
Assert.assertEquals(4, payload.get());
Assert.assertEquals(5, payload.get());
Assert.assertEquals(-1, readableChannel.read(ByteBuffer.allocate(1024)));
}
@Test
public void testReadEmptyFrame() throws Exception {
final FrameInputBuffer inBuffer = new FrameInputBuffer(16 * 1024);
final ReadableByteChannelMock readableChannel = new ReadableByteChannelMock(
new byte[] {0,0,0,0,0,0,0,0,0});
final RawFrame frame = inBuffer.read(readableChannel);
Assert.assertEquals(FrameType.DATA, FrameType.valueOf(frame.getType()));
Assert.assertEquals(0, frame.getFlags());
Assert.assertEquals(0, frame.getStreamId());
final ByteBuffer payload = frame.getPayloadContent();
Assert.assertNull(payload);
}
@Test(expected = ConnectionClosedException.class)
public void testReadFrameConnectionClosed() throws Exception {
final FrameInputBuffer inBuffer = new FrameInputBuffer(16 * 1024);
final ReadableByteChannelMock readableChannel = new ReadableByteChannelMock(new byte[] {});
Assert.assertEquals(null, inBuffer.read(readableChannel));
inBuffer.read(readableChannel);
}
@Test(expected = H2CorruptFrameException.class)
public void testReadFrameCorruptFrame() throws Exception {
final FrameInputBuffer inBuffer = new FrameInputBuffer(16 * 1024);
final ReadableByteChannelMock readableChannel = new ReadableByteChannelMock(new byte[] {0,0});
Assert.assertEquals(null, inBuffer.read(readableChannel));
inBuffer.read(readableChannel);
}
@Test(expected = H2ConnectionException.class)
public void testWriteFrameExceedingLimit() throws Exception {
final WritableByteChannelMock writableChannel = new WritableByteChannelMock(1024);
final FrameOutputBuffer outbuffer = new FrameOutputBuffer(1024);
final RawFrame frame = new RawFrame(FrameType.DATA.getValue(), 0, 1,
ByteBuffer.wrap(new byte[2048]));
outbuffer.write(frame, writableChannel);
}
@Test(expected = H2ConnectionException.class)
public void testReadFrameExceedingLimit() throws Exception {
final FrameInputBuffer inBuffer = new FrameInputBuffer(16 * 1024);
final ReadableByteChannelMock readableChannel = new ReadableByteChannelMock(
new byte[] {0,-128,-128,0,0,0,0,0,1});
Assert.assertEquals(null, inBuffer.read(readableChannel));
inBuffer.read(readableChannel);
}
}
httpcore5-h2/src/test/java/org/apache/hc/core5/http2/ReadableByteChannelMock.java 0100644 0000000 0000000 00000005324 12767505342 026510 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.ClosedChannelException;
import java.nio.channels.ReadableByteChannel;
public class ReadableByteChannelMock implements ReadableByteChannel {
private final byte[][] chunks;
private int chunkCount = 0;
private ByteBuffer currentChunk;
private boolean eof = false;
private boolean closed = false;
public ReadableByteChannelMock(final byte[]... chunks) {
super();
this.chunks = chunks;
}
private void prepareChunk() {
if (this.currentChunk == null || !this.currentChunk.hasRemaining()) {
if (this.chunkCount < this.chunks.length) {
final byte[] bytes = this.chunks[this.chunkCount];
this.chunkCount++;
this.currentChunk = ByteBuffer.wrap(bytes);
} else {
this.eof = true;
}
}
}
@Override
public int read(final ByteBuffer dst) throws IOException {
if (this.closed) {
throw new ClosedChannelException();
}
prepareChunk();
if (this.eof) {
return -1;
}
int i = 0;
while (dst.hasRemaining() && this.currentChunk.hasRemaining()) {
dst.put(this.currentChunk.get());
i++;
}
return i;
}
@Override
public void close() throws IOException {
this.closed = true;
}
@Override
public boolean isOpen() {
return !this.closed && !this.eof;
}
}
httpcore5-h2/src/main/ 0040755 0000000 0000000 00000000000 12735734031 013554 5 ustar 00 0000000 0000000 httpcore5-h2/src/main/java/ 0040755 0000000 0000000 00000000000 12735734031 014475 5 ustar 00 0000000 0000000 httpcore5-h2/src/main/java/org/ 0040755 0000000 0000000 00000000000 12735734031 015264 5 ustar 00 0000000 0000000 httpcore5-h2/src/main/java/org/apache/ 0040755 0000000 0000000 00000000000 12735734031 016505 5 ustar 00 0000000 0000000 httpcore5-h2/src/main/java/org/apache/hc/ 0040755 0000000 0000000 00000000000 12735734031 017077 5 ustar 00 0000000 0000000 httpcore5-h2/src/main/java/org/apache/hc/core5/ 0040755 0000000 0000000 00000000000 12773734502 020121 5 ustar 00 0000000 0000000 httpcore5-h2/src/main/java/org/apache/hc/core5/http2/ 0040755 0000000 0000000 00000000000 14157612023 021151 5 ustar 00 0000000 0000000 httpcore5-h2/src/main/java/org/apache/hc/core5/http2/H2PseudoResponseHeaders.java 0100644 0000000 0000000 00000002612 13341740003 026450 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2;
/**
* Response pseudo HTTP headers defined by the HTTP/2 specification.
*
* @since 5.0
*/
public final class H2PseudoResponseHeaders {
public static final String STATUS = ":status";
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/ssl/ 0040755 0000000 0000000 00000000000 14157612023 021752 5 ustar 00 0000000 0000000 httpcore5-h2/src/main/java/org/apache/hc/core5/http2/ssl/H2ClientTlsStrategy.java 0100664 0000000 0000000 00000007622 14157612023 026441 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.ssl;
import java.net.SocketAddress;
import javax.net.ssl.SSLContext;
import org.apache.hc.core5.http.HttpHost;
import org.apache.hc.core5.http.URIScheme;
import org.apache.hc.core5.http.nio.ssl.TlsStrategy;
import org.apache.hc.core5.reactor.ssl.SSLBufferMode;
import org.apache.hc.core5.reactor.ssl.SSLSessionInitializer;
import org.apache.hc.core5.reactor.ssl.SSLSessionVerifier;
import org.apache.hc.core5.reactor.ssl.TransportSecurityLayer;
import org.apache.hc.core5.ssl.SSLContexts;
import org.apache.hc.core5.util.Args;
import org.apache.hc.core5.util.Timeout;
/**
* Basic client-side implementation of {@link TlsStrategy} that upgrades to TLS for all endpoints
* with {@code HTTPS} scheme.
*
* @since 5.0
*/
public class H2ClientTlsStrategy implements TlsStrategy {
private final SSLContext sslContext;
private final SSLBufferMode sslBufferMode;
private final SSLSessionInitializer initializer;
private final SSLSessionVerifier verifier;
public H2ClientTlsStrategy(
final SSLContext sslContext,
final SSLBufferMode sslBufferMode,
final SSLSessionInitializer initializer,
final SSLSessionVerifier verifier) {
this.sslContext = Args.notNull(sslContext, "SSL context");
this.sslBufferMode = sslBufferMode;
this.initializer = initializer;
this.verifier = verifier;
}
public H2ClientTlsStrategy(
final SSLContext sslContext,
final SSLSessionInitializer initializer,
final SSLSessionVerifier verifier) {
this(sslContext, null, initializer, verifier);
}
public H2ClientTlsStrategy(
final SSLContext sslContext,
final SSLSessionVerifier verifier) {
this(sslContext, null, null, verifier);
}
public H2ClientTlsStrategy(final SSLContext sslContext) {
this(sslContext, null, null, null);
}
public H2ClientTlsStrategy() {
this(SSLContexts.createSystemDefault());
}
@Override
public boolean upgrade(
final TransportSecurityLayer tlsSession,
final HttpHost host,
final SocketAddress localAddress,
final SocketAddress remoteAddress,
final Object attachment,
final Timeout handshakeTimeout) {
final String scheme = host != null ? host.getSchemeName() : null;
if (URIScheme.HTTPS.same(scheme)) {
tlsSession.startTls(
sslContext,
host,
sslBufferMode,
H2TlsSupport.enforceRequirements(attachment, initializer),
verifier,
handshakeTimeout);
return true;
}
return false;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/ssl/ConscryptSupport.java 0100664 0000000 0000000 00000007227 14157612023 026205 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.ssl;
import javax.net.ssl.SSLEngine;
import javax.net.ssl.SSLException;
import javax.net.ssl.SSLParameters;
import org.apache.hc.core5.http.ssl.TLS;
import org.apache.hc.core5.http.ssl.TlsCiphers;
import org.apache.hc.core5.net.NamedEndpoint;
import org.apache.hc.core5.reactor.ssl.SSLSessionInitializer;
import org.apache.hc.core5.reactor.ssl.SSLSessionVerifier;
import org.apache.hc.core5.reactor.ssl.TlsDetails;
import org.conscrypt.Conscrypt;
/**
* Conscrypt TLS support methods
*
* @since 5.0
*/
public final class ConscryptSupport {
public static SSLSessionInitializer initialize(
final Object attachment,
final SSLSessionInitializer initializer) {
return new SSLSessionInitializer() {
@Override
public void initialize(final NamedEndpoint endpoint, final SSLEngine sslEngine) {
final SSLParameters sslParameters = sslEngine.getSSLParameters();
sslParameters.setProtocols(TLS.excludeWeak(sslParameters.getProtocols()));
sslParameters.setCipherSuites(TlsCiphers.excludeH2Blacklisted(sslParameters.getCipherSuites()));
H2TlsSupport.setEnableRetransmissions(sslParameters, false);
final String[] appProtocols = H2TlsSupport.selectApplicationProtocols(attachment);
if (Conscrypt.isConscrypt(sslEngine)) {
sslEngine.setSSLParameters(sslParameters);
Conscrypt.setApplicationProtocols(sslEngine, appProtocols);
} else {
H2TlsSupport.setApplicationProtocols(sslParameters, appProtocols);
sslEngine.setSSLParameters(sslParameters);
}
if (initializer != null) {
initializer.initialize(endpoint, sslEngine);
}
}
};
}
public static SSLSessionVerifier verify(final SSLSessionVerifier verifier) {
return new SSLSessionVerifier() {
@Override
public TlsDetails verify(final NamedEndpoint endpoint, final SSLEngine sslEngine) throws SSLException {
TlsDetails tlsDetails = verifier != null ? verifier.verify(endpoint, sslEngine) : null;
if (tlsDetails == null && Conscrypt.isConscrypt(sslEngine)) {
tlsDetails = new TlsDetails(sslEngine.getSession(), Conscrypt.getApplicationProtocol(sslEngine));
}
return tlsDetails;
}
};
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/ssl/H2ServerTlsStrategy.java 0100664 0000000 0000000 00000014416 14157612023 026470 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.ssl;
import java.net.SocketAddress;
import javax.net.ssl.SSLContext;
import org.apache.hc.core5.http.HttpHost;
import org.apache.hc.core5.http.nio.ssl.FixedPortStrategy;
import org.apache.hc.core5.http.nio.ssl.SecurePortStrategy;
import org.apache.hc.core5.http.nio.ssl.TlsStrategy;
import org.apache.hc.core5.reactor.ssl.SSLBufferMode;
import org.apache.hc.core5.reactor.ssl.SSLSessionInitializer;
import org.apache.hc.core5.reactor.ssl.SSLSessionVerifier;
import org.apache.hc.core5.reactor.ssl.TransportSecurityLayer;
import org.apache.hc.core5.ssl.SSLContexts;
import org.apache.hc.core5.util.Args;
import org.apache.hc.core5.util.Timeout;
/**
* Basic side-side implementation of {@link TlsStrategy} that upgrades to TLS for endpoints
* with the specified local ports.
*
* @since 5.0
*/
public class H2ServerTlsStrategy implements TlsStrategy {
private final SSLContext sslContext;
@SuppressWarnings("deprecation")
private final SecurePortStrategy securePortStrategy;
private final SSLBufferMode sslBufferMode;
private final SSLSessionInitializer initializer;
private final SSLSessionVerifier verifier;
/**
* @deprecated Use {@link H2ServerTlsStrategy#H2ServerTlsStrategy(SSLContext, SSLBufferMode, SSLSessionInitializer, SSLSessionVerifier)}
*/
@Deprecated
public H2ServerTlsStrategy(
final SSLContext sslContext,
final SecurePortStrategy securePortStrategy,
final SSLBufferMode sslBufferMode,
final SSLSessionInitializer initializer,
final SSLSessionVerifier verifier) {
this.sslContext = Args.notNull(sslContext, "SSL context");
this.securePortStrategy = securePortStrategy;
this.sslBufferMode = sslBufferMode;
this.initializer = initializer;
this.verifier = verifier;
}
/**
* @deprecated Use {@link H2ServerTlsStrategy#H2ServerTlsStrategy(SSLContext, SSLSessionInitializer, SSLSessionVerifier)}
*/
@Deprecated
public H2ServerTlsStrategy(
final SSLContext sslContext,
final SecurePortStrategy securePortStrategy,
final SSLSessionInitializer initializer,
final SSLSessionVerifier verifier) {
this(sslContext, securePortStrategy, null, initializer, verifier);
}
/**
* @deprecated Use {@link H2ServerTlsStrategy#H2ServerTlsStrategy(SSLContext, SSLSessionVerifier)}
*/
@Deprecated
public H2ServerTlsStrategy(
final SSLContext sslContext,
final SecurePortStrategy securePortStrategy,
final SSLSessionVerifier verifier) {
this(sslContext, securePortStrategy, null, null, verifier);
}
/**
* @deprecated Use {@link H2ServerTlsStrategy#H2ServerTlsStrategy(SSLContext)}
*/
@Deprecated
public H2ServerTlsStrategy(final SSLContext sslContext, final SecurePortStrategy securePortStrategy) {
this(sslContext, securePortStrategy, null, null, null);
}
/**
* @deprecated Use {@link H2ServerTlsStrategy#H2ServerTlsStrategy()}
*/
@Deprecated
public H2ServerTlsStrategy(final int... securePorts) {
this(SSLContexts.createSystemDefault(), new FixedPortStrategy(securePorts));
}
public H2ServerTlsStrategy(
final SSLContext sslContext,
final SSLBufferMode sslBufferMode,
final SSLSessionInitializer initializer,
final SSLSessionVerifier verifier) {
this.sslContext = Args.notNull(sslContext, "SSL context");
this.sslBufferMode = sslBufferMode;
this.initializer = initializer;
this.verifier = verifier;
this.securePortStrategy = null;
}
public H2ServerTlsStrategy(
final SSLContext sslContext,
final SSLSessionInitializer initializer,
final SSLSessionVerifier verifier) {
this(sslContext, (SSLBufferMode) null, initializer, verifier);
}
public H2ServerTlsStrategy(final SSLContext sslContext, final SSLSessionVerifier verifier) {
this(sslContext, (SSLBufferMode) null, null, verifier);
}
public H2ServerTlsStrategy(final SSLContext sslContext) {
this(sslContext, (SSLBufferMode) null, null, null);
}
public H2ServerTlsStrategy() {
this(SSLContexts.createSystemDefault());
}
private boolean isApplicable(final SocketAddress localAddress) {
return securePortStrategy == null || securePortStrategy.isSecure(localAddress);
}
@Override
public boolean upgrade(
final TransportSecurityLayer tlsSession,
final HttpHost host,
final SocketAddress localAddress,
final SocketAddress remoteAddress,
final Object attachment,
final Timeout handshakeTimeout) {
if (isApplicable(localAddress)) {
tlsSession.startTls(
sslContext,
host,
sslBufferMode,
H2TlsSupport.enforceRequirements(attachment, initializer),
verifier,
handshakeTimeout);
return true;
}
return false;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/ssl/package-info.java 0100644 0000000 0000000 00000002371 13341740003 025133 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
/**
* HTTP/2 specific TLS protocol support.
*/
package org.apache.hc.core5.http2.ssl;
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/ssl/ConscryptClientTlsStrategy.java 0100664 0000000 0000000 00000007502 14157612023 030151 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.ssl;
import java.net.SocketAddress;
import javax.net.ssl.SSLContext;
import org.apache.hc.core5.http.HttpHost;
import org.apache.hc.core5.http.URIScheme;
import org.apache.hc.core5.http.nio.ssl.TlsStrategy;
import org.apache.hc.core5.reactor.ssl.SSLBufferMode;
import org.apache.hc.core5.reactor.ssl.SSLSessionInitializer;
import org.apache.hc.core5.reactor.ssl.SSLSessionVerifier;
import org.apache.hc.core5.reactor.ssl.TransportSecurityLayer;
import org.apache.hc.core5.util.Args;
import org.apache.hc.core5.util.Timeout;
/**
* Basic client-side implementation of {@link TlsStrategy} that upgrades to TLS for all endpoints
* with {@code HTTPS} scheme.
*
* @since 5.0
*/
public class ConscryptClientTlsStrategy implements TlsStrategy {
private final SSLContext sslContext;
private final SSLBufferMode sslBufferMode;
private final SSLSessionInitializer initializer;
private final SSLSessionVerifier verifier;
public ConscryptClientTlsStrategy(
final SSLContext sslContext,
final SSLBufferMode sslBufferMode,
final SSLSessionInitializer initializer,
final SSLSessionVerifier verifier) {
this.sslContext = Args.notNull(sslContext, "SSL context");
this.sslBufferMode = sslBufferMode;
this.initializer = initializer;
this.verifier = verifier;
}
public ConscryptClientTlsStrategy(
final SSLContext sslContext,
final SSLSessionInitializer initializer,
final SSLSessionVerifier verifier) {
this(sslContext, null, initializer, verifier);
}
public ConscryptClientTlsStrategy(
final SSLContext sslContext,
final SSLSessionVerifier verifier) {
this(sslContext, null, null, verifier);
}
public ConscryptClientTlsStrategy(final SSLContext sslContext) {
this(sslContext, null, null, null);
}
@Override
public boolean upgrade(
final TransportSecurityLayer tlsSession,
final HttpHost host,
final SocketAddress localAddress,
final SocketAddress remoteAddress,
final Object attachment,
final Timeout handshakeTimeout) {
final String scheme = host != null ? host.getSchemeName() : null;
if (URIScheme.HTTPS.same(scheme)) {
tlsSession.startTls(
sslContext,
host,
sslBufferMode,
ConscryptSupport.initialize(attachment, initializer),
ConscryptSupport.verify(verifier),
handshakeTimeout);
return true;
}
return false;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/ssl/ConscryptServerTlsStrategy.java 0100664 0000000 0000000 00000014467 14157612023 030211 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.ssl;
import java.net.SocketAddress;
import javax.net.ssl.SSLContext;
import org.apache.hc.core5.http.HttpHost;
import org.apache.hc.core5.http.nio.ssl.FixedPortStrategy;
import org.apache.hc.core5.http.nio.ssl.SecurePortStrategy;
import org.apache.hc.core5.http.nio.ssl.TlsStrategy;
import org.apache.hc.core5.reactor.ssl.SSLBufferMode;
import org.apache.hc.core5.reactor.ssl.SSLSessionInitializer;
import org.apache.hc.core5.reactor.ssl.SSLSessionVerifier;
import org.apache.hc.core5.reactor.ssl.TransportSecurityLayer;
import org.apache.hc.core5.util.Args;
import org.apache.hc.core5.util.Timeout;
/**
* Basic side-side implementation of {@link TlsStrategy} that upgrades to TLS for endpoints
* with the specified local ports.
*
* @since 5.0
*/
public class ConscryptServerTlsStrategy implements TlsStrategy {
private final SSLContext sslContext;
@SuppressWarnings("deprecation")
private final SecurePortStrategy securePortStrategy;
private final SSLBufferMode sslBufferMode;
private final SSLSessionInitializer initializer;
private final SSLSessionVerifier verifier;
/**
* @deprecated Use {@link ConscryptServerTlsStrategy#ConscryptServerTlsStrategy(SSLContext, SSLBufferMode, SSLSessionInitializer, SSLSessionVerifier)}
*/
@Deprecated
public ConscryptServerTlsStrategy(
final SSLContext sslContext,
final SecurePortStrategy securePortStrategy,
final SSLBufferMode sslBufferMode,
final SSLSessionInitializer initializer,
final SSLSessionVerifier verifier) {
this.sslContext = Args.notNull(sslContext, "SSL context");
this.securePortStrategy = securePortStrategy;
this.sslBufferMode = sslBufferMode;
this.initializer = initializer;
this.verifier = verifier;
}
/**
* @deprecated Use {@link ConscryptServerTlsStrategy#ConscryptServerTlsStrategy(SSLContext, SSLSessionInitializer, SSLSessionVerifier)}
*/
@Deprecated
public ConscryptServerTlsStrategy(
final SSLContext sslContext,
final SecurePortStrategy securePortStrategy,
final SSLSessionInitializer initializer,
final SSLSessionVerifier verifier) {
this(sslContext, securePortStrategy, null, initializer, verifier);
}
/**
* @deprecated Use {@link ConscryptServerTlsStrategy#ConscryptServerTlsStrategy(SSLContext, SSLSessionVerifier)}
*/
@Deprecated
public ConscryptServerTlsStrategy(
final SSLContext sslContext,
final SecurePortStrategy securePortStrategy,
final SSLSessionVerifier verifier) {
this(sslContext, securePortStrategy, null, null, verifier);
}
/**
* @deprecated Use {@link ConscryptServerTlsStrategy#ConscryptServerTlsStrategy(SSLContext)}
*/
@Deprecated
public ConscryptServerTlsStrategy(final SSLContext sslContext, final SecurePortStrategy securePortStrategy) {
this(sslContext, securePortStrategy, null, null, null);
}
/**
* @deprecated Use {@link ConscryptServerTlsStrategy#ConscryptServerTlsStrategy(SSLContext)}
*/
@Deprecated
public ConscryptServerTlsStrategy(final SSLContext sslContext, final int... securePorts) {
this(sslContext, new FixedPortStrategy(securePorts));
}
public ConscryptServerTlsStrategy(
final SSLContext sslContext,
final SSLBufferMode sslBufferMode,
final SSLSessionInitializer initializer,
final SSLSessionVerifier verifier) {
this.sslContext = Args.notNull(sslContext, "SSL context");
this.sslBufferMode = sslBufferMode;
this.initializer = initializer;
this.verifier = verifier;
this.securePortStrategy = null;
}
public ConscryptServerTlsStrategy(
final SSLContext sslContext,
final SSLSessionInitializer initializer,
final SSLSessionVerifier verifier) {
this(sslContext, (SSLBufferMode) null, initializer, verifier);
}
public ConscryptServerTlsStrategy(final SSLContext sslContext, final SSLSessionVerifier verifier) {
this(sslContext, (SSLBufferMode) null, null, verifier);
}
public ConscryptServerTlsStrategy(final SSLContext sslContext) {
this(sslContext, (SSLBufferMode) null, null, null);
}
private boolean isApplicable(final SocketAddress localAddress) {
return securePortStrategy == null || securePortStrategy.isSecure(localAddress);
}
@Override
public boolean upgrade(
final TransportSecurityLayer tlsSession,
final HttpHost host,
final SocketAddress localAddress,
final SocketAddress remoteAddress,
final Object attachment,
final Timeout handshakeTimeout) {
if (isApplicable(localAddress)) {
tlsSession.startTls(
sslContext,
host,
sslBufferMode,
ConscryptSupport.initialize(attachment, initializer),
ConscryptSupport.verify(verifier),
handshakeTimeout);
return true;
}
return false;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/ssl/H2TlsSupport.java 0100664 0000000 0000000 00000007205 14157612023 025151 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.ssl;
import javax.net.ssl.SSLEngine;
import javax.net.ssl.SSLParameters;
import org.apache.hc.core5.http.ssl.TLS;
import org.apache.hc.core5.http.ssl.TlsCiphers;
import org.apache.hc.core5.http2.HttpVersionPolicy;
import org.apache.hc.core5.net.NamedEndpoint;
import org.apache.hc.core5.reactor.ssl.SSLSessionInitializer;
import org.apache.hc.core5.util.ReflectionUtils;
/**
* HTTP/2 TLS support methods
*
* @since 5.0
*/
public final class H2TlsSupport {
public static void setEnableRetransmissions(final SSLParameters sslParameters, final boolean value) {
ReflectionUtils.callSetter(sslParameters, "EnableRetransmissions", Boolean.TYPE, value);
}
public static void setApplicationProtocols(final SSLParameters sslParameters, final String[] values) {
ReflectionUtils.callSetter(sslParameters, "ApplicationProtocols", String[].class, values);
}
public static String[] selectApplicationProtocols(final Object attachment) {
final HttpVersionPolicy versionPolicy = attachment instanceof HttpVersionPolicy ?
(HttpVersionPolicy) attachment : HttpVersionPolicy.NEGOTIATE;
switch (versionPolicy) {
case FORCE_HTTP_1:
return new String[] { ApplicationProtocol.HTTP_1_1.id };
case FORCE_HTTP_2:
return new String[] { ApplicationProtocol.HTTP_2.id };
default:
return new String[] { ApplicationProtocol.HTTP_2.id, ApplicationProtocol.HTTP_1_1.id };
}
}
public static SSLSessionInitializer enforceRequirements(
final Object attachment,
final SSLSessionInitializer initializer) {
return new SSLSessionInitializer() {
@Override
public void initialize(final NamedEndpoint endpoint, final SSLEngine sslEngine) {
final SSLParameters sslParameters = sslEngine.getSSLParameters();
sslParameters.setProtocols(TLS.excludeWeak(sslParameters.getProtocols()));
sslParameters.setCipherSuites(TlsCiphers.excludeH2Blacklisted(sslParameters.getCipherSuites()));
setEnableRetransmissions(sslParameters, false);
setApplicationProtocols(sslParameters, selectApplicationProtocols(attachment));
sslEngine.setSSLParameters(sslParameters);
if (initializer != null) {
initializer.initialize(endpoint, sslEngine);
}
}
};
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/ssl/ApplicationProtocol.java 0100644 0000000 0000000 00000003000 13574403762 026602 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.ssl;
/**
* Supported application protocols.
*
* @since 5.0
*/
public enum ApplicationProtocol {
HTTP_2("h2"), HTTP_1_1("http/1.1");
public final String id;
ApplicationProtocol(final String id) {
this.id = id;
}
@Override
public String toString() {
return id;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/HttpVersionPolicy.java 0100644 0000000 0000000 00000002514 13341740003 025452 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2;
/**
* HTTP protocol version policy.
*
* @since 5.0
*/
public enum HttpVersionPolicy {
FORCE_HTTP_1, FORCE_HTTP_2, NEGOTIATE
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/H2MessageConverter.java 0100644 0000000 0000000 00000004371 13341740003 025456 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2;
import java.util.List;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.HttpException;
import org.apache.hc.core5.http.HttpMessage;
/**
* Abstract message converter intended to convert from a list of HTTP/2 headers to object
* representing an HTTP message and from an object representing an HTTP message to a list
* of HTTP/2 headers.
*
* @param represents {@link HttpMessage}
*
* @since 5.0
*/
public interface H2MessageConverter {
/**
* Converts given list of HTTP/2 headers to a {@link HttpMessage}.
*
* @param headers list of HTTP/2 headers
* @return HTTP message
* @throws HttpException in case of HTTP protocol violation
*/
T convert(List headers) throws HttpException;
/**
* Converts given {@link HttpMessage} to a list of HTTP/2 headers.
*
* @param message HTTP message
* @return list of HTTP/2 headers
* @throws HttpException in case of HTTP protocol violation
*/
List convert(T message) throws HttpException;
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/H2ConnectionException.java 0100664 0000000 0000000 00000003640 14114632321 026162 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2;
import java.io.IOException;
import org.apache.hc.core5.util.Args;
/**
* Signals fatal HTTP/2 protocol violation that renders the actual
* HTTP/2 connection unreliable.
*
* @since 5.0
*/
public class H2ConnectionException extends IOException {
private static final long serialVersionUID = -2014204317155428658L;
private final int code;
public H2ConnectionException(final H2Error error, final String message) {
super(message);
Args.notNull(error, "H2 Error code");
this.code = error.getCode();
}
public H2ConnectionException(final int code, final String message) {
super(message);
this.code = code;
}
public int getCode() {
return code;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/config/ 0040755 0000000 0000000 00000000000 14157612023 022416 5 ustar 00 0000000 0000000 httpcore5-h2/src/main/java/org/apache/hc/core5/http2/config/H2Param.java 0100644 0000000 0000000 00000004305 13537747351 024530 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.config;
/**
* HTTP/2 protocol parameters.
*
* @since 5.0
*/
public enum H2Param {
HEADER_TABLE_SIZE (0x1),
ENABLE_PUSH (0x2),
MAX_CONCURRENT_STREAMS (0x3),
INITIAL_WINDOW_SIZE (0x4),
MAX_FRAME_SIZE (0x5),
MAX_HEADER_LIST_SIZE (0x6);
int code;
H2Param(final int code) {
this.code = code;
}
public int getCode() {
return code;
}
private static final H2Param[] LOOKUP_TABLE = new H2Param[6];
static {
for (final H2Param param: H2Param.values()) {
LOOKUP_TABLE[param.code - 1] = param;
}
}
public static H2Param valueOf(final int code) {
if (code < 1 || code > LOOKUP_TABLE.length) {
return null;
}
return LOOKUP_TABLE[code - 1];
}
public static String toString(final int code) {
if (code < 1 || code > LOOKUP_TABLE.length) {
return Integer.toString(code);
}
return LOOKUP_TABLE[code - 1].name();
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/config/package-info.java 0100644 0000000 0000000 00000002366 13341740003 025603 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
/**
* Core HTTP/2 configuration APIs.
*/
package org.apache.hc.core5.http2.config;
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/config/H2Config.java 0100664 0000000 0000000 00000017334 14157612023 024667 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.config;
import org.apache.hc.core5.annotation.Contract;
import org.apache.hc.core5.annotation.ThreadingBehavior;
import org.apache.hc.core5.http2.frame.FrameConsts;
import org.apache.hc.core5.util.Args;
/**
* HTTP/2 protocol configuration.
*
* @since 5.0
*/
@Contract(threading = ThreadingBehavior.IMMUTABLE)
public class H2Config {
public static final H2Config DEFAULT = custom().build();
public static final H2Config INIT = initial().build();
private final int headerTableSize;
private final boolean pushEnabled;
private final int maxConcurrentStreams;
private final int initialWindowSize;
private final int maxFrameSize;
private final int maxHeaderListSize;
private final boolean compressionEnabled;
H2Config(final int headerTableSize, final boolean pushEnabled, final int maxConcurrentStreams,
final int initialWindowSize, final int maxFrameSize, final int maxHeaderListSize,
final boolean compressionEnabled) {
super();
this.headerTableSize = headerTableSize;
this.pushEnabled = pushEnabled;
this.maxConcurrentStreams = maxConcurrentStreams;
this.initialWindowSize = initialWindowSize;
this.maxFrameSize = maxFrameSize;
this.maxHeaderListSize = maxHeaderListSize;
this.compressionEnabled = compressionEnabled;
}
public int getHeaderTableSize() {
return headerTableSize;
}
public boolean isPushEnabled() {
return pushEnabled;
}
public int getMaxConcurrentStreams() {
return maxConcurrentStreams;
}
public int getInitialWindowSize() {
return initialWindowSize;
}
public int getMaxFrameSize() {
return maxFrameSize;
}
public int getMaxHeaderListSize() {
return maxHeaderListSize;
}
public boolean isCompressionEnabled() {
return compressionEnabled;
}
@Override
public String toString() {
final StringBuilder builder = new StringBuilder();
builder.append("[headerTableSize=").append(this.headerTableSize)
.append(", pushEnabled=").append(this.pushEnabled)
.append(", maxConcurrentStreams=").append(this.maxConcurrentStreams)
.append(", initialWindowSize=").append(this.initialWindowSize)
.append(", maxFrameSize=").append(this.maxFrameSize)
.append(", maxHeaderListSize=").append(this.maxHeaderListSize)
.append(", compressionEnabled=").append(this.compressionEnabled)
.append("]");
return builder.toString();
}
public static H2Config.Builder custom() {
return new Builder();
}
private static final int INIT_HEADER_TABLE_SIZE = 4096;
private static final boolean INIT_ENABLE_PUSH = true;
private static final int INIT_MAX_FRAME_SIZE = FrameConsts.MIN_FRAME_SIZE;
private static final int INIT_WINDOW_SIZE = 65535;
public static H2Config.Builder initial() {
return new Builder()
.setHeaderTableSize(INIT_HEADER_TABLE_SIZE)
.setPushEnabled(INIT_ENABLE_PUSH)
.setMaxConcurrentStreams(Integer.MAX_VALUE) // no limit
.setMaxFrameSize(INIT_MAX_FRAME_SIZE)
.setInitialWindowSize(INIT_WINDOW_SIZE)
.setMaxHeaderListSize(Integer.MAX_VALUE); // unlimited
}
public static H2Config.Builder copy(final H2Config config) {
Args.notNull(config, "Connection config");
return new Builder()
.setHeaderTableSize(config.getHeaderTableSize())
.setPushEnabled(config.isPushEnabled())
.setMaxConcurrentStreams(config.getMaxConcurrentStreams())
.setInitialWindowSize(config.getInitialWindowSize())
.setMaxFrameSize(config.getMaxFrameSize())
.setMaxHeaderListSize(config.getMaxHeaderListSize())
.setCompressionEnabled(config.isCompressionEnabled());
}
public static class Builder {
private int headerTableSize;
private boolean pushEnabled;
private int maxConcurrentStreams;
private int initialWindowSize;
private int maxFrameSize;
private int maxHeaderListSize;
private boolean compressionEnabled;
Builder() {
this.headerTableSize = INIT_HEADER_TABLE_SIZE * 2;
this.pushEnabled = INIT_ENABLE_PUSH;
this.maxConcurrentStreams = 250;
this.initialWindowSize = INIT_WINDOW_SIZE;
this.maxFrameSize = FrameConsts.MIN_FRAME_SIZE * 4;
this.maxHeaderListSize = FrameConsts.MAX_FRAME_SIZE;
this.compressionEnabled = true;
}
public Builder setHeaderTableSize(final int headerTableSize) {
Args.notNegative(headerTableSize, "Header table size");
this.headerTableSize = headerTableSize;
return this;
}
public Builder setPushEnabled(final boolean pushEnabled) {
this.pushEnabled = pushEnabled;
return this;
}
public Builder setMaxConcurrentStreams(final int maxConcurrentStreams) {
Args.positive(maxConcurrentStreams, "Max concurrent streams");
this.maxConcurrentStreams = maxConcurrentStreams;
return this;
}
public Builder setInitialWindowSize(final int initialWindowSize) {
Args.positive(initialWindowSize, "Initial window size");
this.initialWindowSize = initialWindowSize;
return this;
}
public Builder setMaxFrameSize(final int maxFrameSize) {
this.maxFrameSize = Args.checkRange(maxFrameSize, FrameConsts.MIN_FRAME_SIZE, FrameConsts.MAX_FRAME_SIZE,
"Invalid max frame size");
return this;
}
public Builder setMaxHeaderListSize(final int maxHeaderListSize) {
Args.positive(maxHeaderListSize, "Max header list size");
this.maxHeaderListSize = maxHeaderListSize;
return this;
}
public Builder setCompressionEnabled(final boolean compressionEnabled) {
this.compressionEnabled = compressionEnabled;
return this;
}
public H2Config build() {
return new H2Config(
headerTableSize,
pushEnabled,
maxConcurrentStreams,
initialWindowSize,
maxFrameSize,
maxHeaderListSize,
compressionEnabled);
}
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/config/H2Setting.java 0100644 0000000 0000000 00000003654 13341740003 025067 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.config;
import org.apache.hc.core5.util.Args;
/**
* HTTP/2 protocol settings.
*
* @since 5.0
*/
public final class H2Setting {
private final H2Param param;
private final int value;
public H2Setting(final H2Param param, final int value) {
Args.notNull(param, "Setting parameter");
Args.notNegative(value, "Setting value must be a non-negative value");
this.param = param;
this.value = value;
}
public int getCode() {
return param.code;
}
public int getValue() {
return value;
}
@Override
public String toString() {
final StringBuilder sb = new StringBuilder().append(param).append(": ").append(value);
return sb.toString();
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/frame/ 0040755 0000000 0000000 00000000000 14114632321 022237 5 ustar 00 0000000 0000000 httpcore5-h2/src/main/java/org/apache/hc/core5/http2/frame/FrameConsts.java 0100644 0000000 0000000 00000003150 13341740003 025320 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.frame;
/**
* Values used by HTTP/2 protocol.
*
* @since 5.0
*/
public final class FrameConsts {
private FrameConsts() {
// Do not allow utility class to be instantiated.
}
public final static int HEAD_LEN = 9;
public final static int MAX_PADDING = 255;
public final static int MIN_FRAME_SIZE = 16384; // 2 ^ 14
public final static int MAX_FRAME_SIZE = 16777215; // 2 ^ 24 - 1;
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/frame/Frame.java 0100644 0000000 0000000 00000004527 13341740003 024137 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.frame;
/**
* Abstract HTTP/2 data frame.
*
* @since 5.0
*
* @param frame payload representation.
*/
public abstract class Frame {
private final int type;
private final int flags;
private final int streamId;
public Frame(final int type, final int flags, final int streamId) {
this.type = type;
this.flags = flags;
this.streamId = streamId;
}
public boolean isType(final FrameType type) {
return getType() == type.value;
}
public boolean isFlagSet(final FrameFlag flag) {
return (getFlags() & flag.value) != 0;
}
public int getType() {
return type;
}
public int getFlags() {
return flags;
}
public int getStreamId() {
return streamId;
}
public abstract T getPayload();
@Override
public String toString() {
final StringBuilder sb = new StringBuilder("[");
sb.append("type=").append(type);
sb.append(", flags=").append(flags);
sb.append(", streamId=").append(streamId);
sb.append(", payoad=").append(getPayload());
sb.append(']');
return sb.toString();
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/frame/StreamIdGenerator.java 0100664 0000000 0000000 00000004213 14114632321 026460 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.frame;
/**
* HTTP/2 stream ID generator.
*
* @since 5.0
*/
public interface StreamIdGenerator {
int generate(int previousMax);
boolean isSameSide(int streamId);
StreamIdGenerator ODD = new StreamIdGenerator() {
@Override
public int generate(final int previousMax) {
int i = previousMax + 1;
if (i % 2 == 0) {
i++;
}
return i;
}
@Override
public boolean isSameSide(final int streamId) {
return (streamId & 1) == 1;
}
};
StreamIdGenerator EVEN = new StreamIdGenerator() {
@Override
public int generate(final int previousMax) {
int i = previousMax + 1;
if (i % 2 == 1) {
i++;
}
return i;
}
@Override
public boolean isSameSide(final int streamId) {
return (streamId & 1) == 0;
}
};
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/frame/FrameType.java 0100644 0000000 0000000 00000004436 13341740003 025000 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.frame;
/**
* Standard HTTP/2 frame types.
*
* @since 5.0
*/
public enum FrameType {
DATA (0x00),
HEADERS (0x01),
PRIORITY (0x02),
RST_STREAM (0x03),
SETTINGS (0x04),
PUSH_PROMISE (0x05),
PING (0x06),
GOAWAY (0x07),
WINDOW_UPDATE (0x08),
CONTINUATION (0x09);
int value;
FrameType(final int value) {
this.value = value;
}
public int getValue() {
return value;
}
private static final FrameType[] LOOKUP_TABLE = new FrameType[10];
static {
for (final FrameType frameType: FrameType.values()) {
LOOKUP_TABLE[frameType.value] = frameType;
}
}
public static FrameType valueOf(final int value) {
if (value < 0 || value >= LOOKUP_TABLE.length) {
return null;
}
return LOOKUP_TABLE[value];
}
public static String toString(final int value) {
if (value < 0 || value >= LOOKUP_TABLE.length) {
return Integer.toString(value);
}
return LOOKUP_TABLE[value].name();
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/frame/FrameFlag.java 0100664 0000000 0000000 00000003364 14114632321 024733 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.frame;
/**
* Standard HTTP/2 frame flags.
*
* @since 5.0
*/
public enum FrameFlag {
END_STREAM (0x01),
ACK (0x01),
END_HEADERS (0x04),
PADDED (0x08),
PRIORITY (0x20);
final int value;
FrameFlag(final int value) {
this.value = value;
}
public int getValue() {
return value;
}
public static int of(final FrameFlag... flags) {
int value = 0;
for (final FrameFlag flag: flags) {
value |= flag.value;
}
return value;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/frame/FramePrinter.java 0100644 0000000 0000000 00000022460 13756257112 025515 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.frame;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.charset.StandardCharsets;
import java.util.Objects;
import org.apache.hc.core5.annotation.Internal;
import org.apache.hc.core5.http.Chars;
import org.apache.hc.core5.http2.H2Error;
import org.apache.hc.core5.http2.config.H2Param;
@Internal
public final class FramePrinter {
public void printFrameInfo(final RawFrame frame, final Appendable appendable) throws IOException {
appendable.append("stream ").append(Integer.toString(frame.getStreamId())).append(" frame: ");
final FrameType type = FrameType.valueOf(frame.getType());
appendable.append(Objects.toString(type))
.append(" (0x").append(Integer.toHexString(frame.getType())).append("); flags: ");
final int flags = frame.getFlags();
if (flags > 0) {
switch (type) {
case SETTINGS:
case PING:
if ((flags & FrameFlag.ACK.value) > 0) {
appendable.append(FrameFlag.ACK.name()).append(" ");
}
break;
case DATA:
if ((flags & FrameFlag.END_STREAM.value) > 0) {
appendable.append(FrameFlag.END_STREAM.name()).append(" ");
}
if ((flags & FrameFlag.PADDED.value) > 0) {
appendable.append(FrameFlag.PADDED.name()).append(" ");
}
break;
case HEADERS:
if ((flags & FrameFlag.END_STREAM.value) > 0) {
appendable.append(FrameFlag.END_STREAM.name()).append(" ");
}
if ((flags & FrameFlag.END_HEADERS.value) > 0) {
appendable.append(FrameFlag.END_HEADERS.name()).append(" ");
}
if ((flags & FrameFlag.PADDED.value) > 0) {
appendable.append(FrameFlag.PADDED.name()).append(" ");
}
if ((flags & FrameFlag.PRIORITY.value) > 0) {
appendable.append(FrameFlag.PRIORITY.name()).append(" ");
}
break;
case PUSH_PROMISE:
if ((flags & FrameFlag.END_HEADERS.value) > 0) {
appendable.append(FrameFlag.END_HEADERS.name()).append(" ");
}
if ((flags & FrameFlag.PADDED.value) > 0) {
appendable.append(FrameFlag.PADDED.name()).append(" ");
}
break;
case CONTINUATION:
if ((flags & FrameFlag.END_HEADERS.value) > 0) {
appendable.append(FrameFlag.END_HEADERS.name()).append(" ");
}
}
}
appendable.append("(0x").append(Integer.toHexString(flags)).append("); length: ").append(Integer.toString(frame.getLength()));
}
public void printPayload(final RawFrame frame, final Appendable appendable) throws IOException {
final FrameType type = FrameType.valueOf(frame.getType());
final ByteBuffer buf = frame.getPayloadContent();
if (buf != null) {
switch (type) {
case SETTINGS:
if ((buf.remaining() % 6) == 0) {
while (buf.hasRemaining()) {
final int code = buf.getShort();
final H2Param param = H2Param.valueOf(code);
final int value = buf.getInt();
if (param != null) {
appendable.append(param.name());
} else {
appendable.append("0x").append(Integer.toHexString(code));
}
appendable.append(": ").append(Integer.toString(value)).append("\r\n");
}
} else {
appendable.append("Invalid\r\n");
}
break;
case RST_STREAM:
if (buf.remaining() == 4) {
appendable.append("Code ");
final int code = buf.getInt();
final H2Error error = H2Error.getByCode(code);
if (error != null) {
appendable.append(error.name());
} else {
appendable.append("0x").append(Integer.toHexString(code));
}
appendable.append("\r\n");
} else {
appendable.append("Invalid\r\n");
}
break;
case GOAWAY:
if (buf.remaining() >= 8) {
final int lastStream = buf.getInt();
appendable.append("Last stream ").append(Integer.toString(lastStream)).append("\r\n");
appendable.append("Code ");
final int code2 = buf.getInt();
final H2Error error2 = H2Error.getByCode(code2);
if (error2 != null) {
appendable.append(error2.name());
} else {
appendable.append("0x").append(Integer.toHexString(code2));
}
appendable.append("\r\n");
final byte[] tmp = new byte[buf.remaining()];
buf.get(tmp);
appendable.append(new String(tmp, StandardCharsets.US_ASCII));
appendable.append("\r\n");
} else {
appendable.append("Invalid\r\n");
}
break;
case WINDOW_UPDATE:
if (buf.remaining() == 4) {
final int increment = buf.getInt();
appendable.append("Increment ").append(Integer.toString(increment)).append("\r\n");
} else {
appendable.append("Invalid\r\n");
}
break;
case PUSH_PROMISE:
if (buf.remaining() > 4) {
final int streamId = buf.getInt();
appendable.append("Promised stream ").append(Integer.toString(streamId)).append("\r\n");
printData(buf, appendable);
} else {
appendable.append("Invalid\r\n");
}
break;
default:
printData(frame.getPayload(), appendable);
}
}
}
public void printData(final ByteBuffer data, final Appendable appendable) throws IOException {
final ByteBuffer buf = data.duplicate();
final byte[] line = new byte[16];
while (buf.hasRemaining()) {
final int chunk = Math.min(buf.remaining(), line.length);
buf.get(line, 0, chunk);
for (int i = 0; i < chunk; i++) {
final char ch = (char) line[i];
if (ch > Chars.SP && ch <= Chars.DEL) {
appendable.append(ch);
} else if (Character.isWhitespace(ch)) {
appendable.append(' ');
} else {
appendable.append('.');
}
}
for (int i = chunk; i < 17; i++) {
appendable.append(' ');
}
for (int i = 0; i < chunk; i++) {
appendable.append(' ');
final int b = line[i] & 0xff;
final String s = Integer.toHexString(b);
if (s.length() == 1) {
appendable.append("0");
}
appendable.append(s);
}
appendable.append("\r\n");
}
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/frame/package-info.java 0100644 0000000 0000000 00000002362 13341740003 025424 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
/**
* Core HTTP/2 data frame APIs.
*/
package org.apache.hc.core5.http2.frame;
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/frame/DefaultFrameFactory.java 0100644 0000000 0000000 00000005551 13341740003 026772 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.frame;
import java.nio.ByteBuffer;
import org.apache.hc.core5.util.Args;
/**
* Default {@link FrameFactory} implementation.
*
* @since 5.0
*/
public class DefaultFrameFactory extends FrameFactory {
public static final FrameFactory INSTANCE = new DefaultFrameFactory();
@Override
public RawFrame createHeaders(final int streamId, final ByteBuffer payload, final boolean endHeaders, final boolean endStream) {
Args.positive(streamId, "Stream id");
final int flags = (endHeaders ? FrameFlag.END_HEADERS.value : 0) | (endStream ? FrameFlag.END_STREAM.value : 0);
return new RawFrame(FrameType.HEADERS.getValue(), flags, streamId, payload);
}
@Override
public RawFrame createContinuation(final int streamId, final ByteBuffer payload, final boolean endHeaders) {
Args.positive(streamId, "Stream id");
final int flags = (endHeaders ? FrameFlag.END_HEADERS.value : 0);
return new RawFrame(FrameType.CONTINUATION.getValue(), flags, streamId, payload);
}
@Override
public RawFrame createPushPromise(final int streamId, final ByteBuffer payload, final boolean endHeaders) {
Args.positive(streamId, "Stream id");
final int flags = (endHeaders ? FrameFlag.END_HEADERS.value : 0);
return new RawFrame(FrameType.PUSH_PROMISE.getValue(), flags, streamId, payload);
}
@Override
public RawFrame createData(final int streamId, final ByteBuffer payload, final boolean endStream) {
Args.positive(streamId, "Stream id");
final int flags = (endStream ? FrameFlag.END_STREAM.value : 0);
return new RawFrame(FrameType.DATA.getValue(), flags, streamId, payload);
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/frame/FrameFactory.java 0100644 0000000 0000000 00000011075 13551612520 025470 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.frame;
import java.nio.ByteBuffer;
import java.nio.charset.StandardCharsets;
import org.apache.hc.core5.http2.H2Error;
import org.apache.hc.core5.http2.config.H2Setting;
import org.apache.hc.core5.util.Args;
/**
* Abstract {@link RawFrame} factory that supports standard
* HTTP/2 {@link FrameType}s.
*
* @since 5.0
*/
public abstract class FrameFactory {
public RawFrame createSettings(final H2Setting... settings) {
final ByteBuffer payload = ByteBuffer.allocate(settings.length * 12);
for (final H2Setting setting: settings) {
payload.putShort((short) setting.getCode());
payload.putInt(setting.getValue());
}
payload.flip();
return new RawFrame(FrameType.SETTINGS.getValue(), 0, 0, payload);
}
public RawFrame createSettingsAck() {
return new RawFrame(FrameType.SETTINGS.getValue(), FrameFlag.ACK.getValue(), 0, null);
}
public RawFrame createResetStream(final int streamId, final H2Error error) {
Args.notNull(error, "Error");
return createResetStream(streamId, error.getCode());
}
public RawFrame createResetStream(final int streamId, final int code) {
Args.positive(streamId, "Stream id");
final ByteBuffer payload = ByteBuffer.allocate(4);
payload.putInt(code);
payload.flip();
return new RawFrame(FrameType.RST_STREAM.getValue(), 0, streamId, payload);
}
public RawFrame createPing(final ByteBuffer opaqueData) {
Args.notNull(opaqueData, "Opaque data");
Args.check(opaqueData.remaining() == 8, "Opaque data length must be equal 8");
return new RawFrame(FrameType.PING.getValue(), 0, 0, opaqueData);
}
public RawFrame createPingAck(final ByteBuffer opaqueData) {
Args.notNull(opaqueData, "Opaque data");
Args.check(opaqueData.remaining() == 8, "Opaque data length must be equal 8");
return new RawFrame(FrameType.PING.getValue(), FrameFlag.ACK.value, 0, opaqueData);
}
public RawFrame createGoAway(final int lastStream, final H2Error error, final String message) {
Args.notNegative(lastStream, "Last stream id");
final byte[] debugData = message != null ? message.getBytes(StandardCharsets.US_ASCII) : null;
final ByteBuffer payload = ByteBuffer.allocate(8 + (debugData != null ? debugData.length : 0));
payload.putInt(lastStream);
payload.putInt(error.getCode());
if (debugData != null) {
payload.put(debugData);
}
payload.flip();
return new RawFrame(FrameType.GOAWAY.getValue(), 0, 0, payload);
}
public abstract RawFrame createHeaders(int streamId, ByteBuffer payload, boolean endHeaders, boolean endStream);
public abstract RawFrame createContinuation(int streamId, ByteBuffer payload, boolean endHeaders);
public abstract RawFrame createPushPromise(int streamId, ByteBuffer payload, boolean endHeaders);
public abstract RawFrame createData(int streamId, ByteBuffer payload, boolean endStream);
public RawFrame createWindowUpdate(final int streamId, final int increment) {
Args.notNegative(streamId, "Stream id");
Args.positive(increment, "Increment");
final ByteBuffer payload = ByteBuffer.allocate(4);
payload.putInt(increment);
payload.flip();
return new RawFrame(FrameType.WINDOW_UPDATE.getValue(), 0, streamId, payload);
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/frame/RawFrame.java 0100644 0000000 0000000 00000004717 13331401103 024604 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.frame;
import java.nio.ByteBuffer;
/**
* HTTP/2 stream frame.
*
* @since 5.0
*/
public final class RawFrame extends Frame {
private final ByteBuffer payload;
private final int len;
public RawFrame(final int type, final int flags, final int streamId, final ByteBuffer payload) {
super(type, flags, streamId);
this.payload = payload;
this.len = payload != null ? payload.remaining() : 0;
}
public boolean isPadded() {
return isFlagSet(FrameFlag.PADDED);
}
public int getLength() {
return len;
}
public ByteBuffer getPayloadContent() {
if (payload != null) {
if (isPadded()) {
final ByteBuffer dup = payload.duplicate();
if (dup.remaining() == 0) {
return null;
}
final int padding = dup.get() & 0xff;
if (padding > dup.remaining()) {
return null;
}
dup.limit(dup.limit() - padding);
return dup;
}
return payload.duplicate();
}
return null;
}
@Override
public ByteBuffer getPayload() {
return payload != null ? payload.duplicate() : null;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/ 0040755 0000000 0000000 00000000000 14157612023 022237 5 ustar 00 0000000 0000000 httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/OutboundDynamicTable.java 0100664 0000000 0000000 00000010353 14157612023 027157 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.util.Asserts;
final class OutboundDynamicTable {
private final StaticTable staticTable;
private final FifoLinkedList headers;
private final Map> mapByName;
private int maxSize;
private int currentSize;
OutboundDynamicTable(final StaticTable staticTable) {
this.staticTable = staticTable;
this.headers = new FifoLinkedList();
this.mapByName = new HashMap<>();
this.maxSize = Integer.MAX_VALUE;
this.currentSize = 0;
}
OutboundDynamicTable() {
this(StaticTable.INSTANCE);
}
public int getMaxSize() {
return maxSize;
}
public void setMaxSize(final int maxSize) {
this.maxSize = maxSize;
evict();
}
public int getCurrentSize() {
return currentSize;
}
int staticLength() {
return staticTable.length();
}
int dynamicLength() {
return headers.size();
}
Header getDynamicEntry(final int index) {
return headers.get(index);
}
public int length() {
return staticTable.length() + headers.size();
}
public Header getHeader(final int index) {
if (index < 1 || index > length()) {
throw new IndexOutOfBoundsException();
}
return index <= staticTable.length()
? staticTable.get(index)
: headers.get(index - staticTable.length() - 1);
}
public void add(final HPackHeader header) {
final int entrySize = header.getTotalSize();
if (entrySize > this.maxSize) {
clear();
this.mapByName.clear();
return;
}
final String key = header.getName();
final FifoLinkedList.InternalNode node = headers.addFirst(header);
LinkedList nodes = mapByName.get(key);
if (nodes == null) {
nodes = new LinkedList<>();
mapByName.put(key, nodes);
}
nodes.addFirst(node);
currentSize += entrySize;
evict();
}
private void clear() {
currentSize = 0;
headers.clear();
}
public List getByName(final String key) {
return this.mapByName.get(key);
}
private void evict() {
while (currentSize > maxSize) {
final FifoLinkedList.InternalNode node = headers.removeLast();
if (node != null) {
final HPackHeader header = node.getHeader();
currentSize -= header.getTotalSize();
final String key = header.getName();
final LinkedList nodes = mapByName.get(key);
if (nodes != null) {
nodes.remove(node);
}
} else {
Asserts.check(currentSize == 0, "Current table size must be zero");
break;
}
}
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/HuffmanEncoder.java 0100644 0000000 0000000 00000005733 12767505342 026005 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
import java.nio.ByteBuffer;
import org.apache.hc.core5.util.ByteArrayBuffer;
/**
* This Huffman codec implementation has been derived from Twitter HPack project
* (https://github.com/twitter/hpack)
*/
final class HuffmanEncoder {
private final int[] codes;
private final byte[] lengths;
HuffmanEncoder(final int[] codes, final byte[] lengths) {
this.codes = codes;
this.lengths = lengths;
}
void encode(final ByteArrayBuffer out, final ByteBuffer src) {
long current = 0;
int n = 0;
while (src.hasRemaining()) {
final int b = src.get() & 0xFF;
final int code = codes[b];
final int nbits = lengths[b];
current <<= nbits;
current |= code;
n += nbits;
while (n >= 8) {
n -= 8;
out.append((int)(current >> n));
}
}
if (n > 0) {
current <<= (8 - n);
current |= (0xFF >>> n); // this should be EOS symbol
out.append((int) current);
}
}
void encode(final ByteArrayBuffer out, final CharSequence src, final int off, final int len) {
long current = 0;
int n = 0;
for (int i = 0; i < len; i++) {
final int b = src.charAt(off + i) & 0xFF;
final int code = codes[b];
final int nbits = lengths[b];
current <<= nbits;
current |= code;
n += nbits;
while (n >= 8) {
n -= 8;
out.append((int)(current >> n));
}
}
if (n > 0) {
current <<= (8 - n);
current |= (0xFF >>> n); // this should be EOS symbol
out.append((int) current);
}
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/FifoBuffer.java 0100644 0000000 0000000 00000006136 13331401103 025106 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
import org.apache.hc.core5.http.Header;
final class FifoBuffer {
private HPackHeader[] array;
private int head;
private int tail;
FifoBuffer(final int initialCapacity) {
this.array = new HPackHeader[initialCapacity];
this.head = 0;
this.tail = 0;
}
private void expand() {
int newcapacity = (array.length + 1) << 1;
if (newcapacity < 0) {
newcapacity = Integer.MAX_VALUE;
}
final Header[] oldArray = array;
final int len = oldArray.length;
final HPackHeader[] newArray = new HPackHeader[newcapacity];
System.arraycopy(oldArray, head, newArray, 0, len - head);
System.arraycopy(oldArray, 0, newArray, len - head, head);
array = newArray;
head = len;
tail = 0;
}
public void clear() {
head = 0;
tail = 0;
}
public void addFirst(final HPackHeader header) {
array[head++] = header;
if (head == array.length) {
head = 0;
}
if (head == tail) {
expand();
}
}
public HPackHeader get(final int index) {
int i = head - index - 1;
if (i < 0) {
i = array.length + i;
}
return array[i];
}
public HPackHeader getFirst() {
return array[head > 0 ? head - 1 : array.length - 1];
}
public HPackHeader getLast() {
return array[tail];
}
public HPackHeader removeLast() {
final HPackHeader header = array[tail];
if (header != null) {
array[tail++] = null;
if (tail == array.length) {
tail = 0;
}
}
return header;
}
public int capacity() {
return array.length;
}
public int size() {
int i = head - tail;
if (i < 0) {
i = array.length + i;
}
return i;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/HPackException.java 0100644 0000000 0000000 00000003145 13341740003 025741 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
import org.apache.hc.core5.http.HttpException;
/**
* Signals HPACK protocol violation.
*
* @since 5.0
*/
public class HPackException extends HttpException {
private static final long serialVersionUID = 1L;
public HPackException(final String message) {
super(message);
}
public HPackException(final String message, final Exception cause) {
super(message, cause);
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/HuffmanDecoder.java 0100644 0000000 0000000 00000010326 12735734300 025756 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
import java.nio.ByteBuffer;
import org.apache.hc.core5.util.ByteArrayBuffer;
/**
* This Huffman codec implementation has been derived from Twitter HPack project
* (https://github.com/twitter/hpack)
*/
final class HuffmanDecoder {
private final HuffmanNode root;
HuffmanDecoder(final int[] codes, final byte[] lengths) {
root = buildTree(codes, lengths);
}
void decode(final ByteArrayBuffer out, final ByteBuffer src) throws HPackException {
HuffmanNode node = this.root;
int current = 0;
int bits = 0;
while (src.hasRemaining()) {
final int b = src.get() & 0xFF;
current = (current << 8) | b;
bits += 8;
while (bits >= 8) {
final int c = (current >>> (bits - 8)) & 0xFF;
node = node.getChild(c);
bits -= node.getBits();
if (node.isTerminal()) {
if (node.getSymbol() == Huffman.EOS) {
throw new HPackException("EOS decoded");
}
out.append(node.getSymbol());
node = root;
}
}
}
while (bits > 0) {
final int c = (current << (8 - bits)) & 0xFF;
node = node.getChild(c);
if (node.isTerminal() && node.getBits() <= bits) {
bits -= node.getBits();
out.append(node.getSymbol());
node = this.root;
} else {
break;
}
}
// Section 5.2. String Literal Representation
// Padding not corresponding to the most significant bits of the code
// for the EOS symbol (0xFF) MUST be treated as a decoding error.
final int mask = (1 << bits) - 1;
if ((current & mask) != mask) {
throw new HPackException("Invalid padding");
}
}
private static HuffmanNode buildTree(final int[] codes, final byte[] lengths) {
final HuffmanNode root = new HuffmanNode();
for (int symbol = 0; symbol < codes.length; symbol++) {
final int code = codes[symbol];
int length = lengths[symbol];
HuffmanNode current = root;
while (length > 8) {
if (current.isTerminal()) {
throw new IllegalStateException("Invalid Huffman code: prefix not unique");
}
length -= 8;
final int i = (code >>> length) & 0xFF;
if (!current.hasChild(i)) {
current.setChild(i, new HuffmanNode());
}
current = current.getChild(i);
}
final HuffmanNode terminal = new HuffmanNode(symbol, length);
final int shift = 8 - length;
final int start = (code << shift) & 0xFF;
final int end = 1 << shift;
for (int i = start; i < start + end; i++) {
current.setChild(i, terminal);
}
}
return root;
}
} httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/InboundDynamicTable.java 0100664 0000000 0000000 00000006477 14157612023 026772 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.util.Asserts;
final class InboundDynamicTable {
private final StaticTable staticTable;
private final FifoBuffer headers;
private int maxSize;
private int currentSize;
InboundDynamicTable(final StaticTable staticTable) {
this.staticTable = staticTable;
this.headers = new FifoBuffer(256);
this.maxSize = Integer.MAX_VALUE;
this.currentSize = 0;
}
InboundDynamicTable() {
this(StaticTable.INSTANCE);
}
public int getMaxSize() {
return maxSize;
}
public void setMaxSize(final int maxSize) {
this.maxSize = maxSize;
evict();
}
public int getCurrentSize() {
return currentSize;
}
int staticLength() {
return staticTable.length();
}
int dynamicLength() {
return headers.size();
}
Header getDynamicEntry(final int index) {
return headers.get(index);
}
public int length() {
return staticTable.length() + headers.size();
}
public HPackHeader getHeader(final int index) {
if (index < 1 || index > length()) {
throw new IndexOutOfBoundsException();
}
return index <= staticTable.length()
? staticTable.get(index)
: headers.get(index - staticTable.length() - 1);
}
public void add(final HPackHeader header) {
final int entrySize = header.getTotalSize();
if (entrySize > this.maxSize) {
clear();
return;
}
headers.addFirst(header);
currentSize += entrySize;
evict();
}
private void clear() {
currentSize = 0;
headers.clear();
}
private void evict() {
while (currentSize > maxSize) {
final HPackHeader header = headers.removeLast();
if (header != null) {
currentSize -= header.getTotalSize();
} else {
Asserts.check(currentSize == 0, "Current table size must be zero");
break;
}
}
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/FifoLinkedList.java 0100664 0000000 0000000 00000010322 14157612023 025745 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
import org.apache.hc.core5.http.Header;
final class FifoLinkedList {
private final InternalNode master;
private int length;
FifoLinkedList() {
this.master = new InternalNode(null);
this.master.previous = this.master;
this.master.next = this.master;
}
public Header get(final int index) {
if (index < 0 || index >= length) {
throw new IndexOutOfBoundsException();
}
InternalNode current = master.next;
int n = 0;
while (current != master) {
if (index == n) {
return current.header;
}
current = current.next;
n++;
}
return null;
}
public int getIndex(final InternalNode node) {
final int seqNum = node.seqNum;
if (seqNum < 1) {
return -1;
}
return length - (seqNum - master.previous.seqNum) - 1;
}
public Header getFirst() {
return master.next.header;
}
public Header getLast() {
return master.previous.header;
}
public int size() {
return length;
}
public InternalNode addFirst(final HPackHeader header) {
final InternalNode newNode = new InternalNode(header);
final InternalNode oldNode = master.next;
master.next = newNode;
newNode.previous = master;
newNode.next = oldNode;
oldNode.previous = newNode;
newNode.seqNum = oldNode.seqNum + 1;
length++;
return newNode;
}
public InternalNode removeLast() {
final InternalNode last = master.previous;
if (last.header != null) {
final InternalNode lastButOne = last.previous;
master.previous = lastButOne;
lastButOne.next = master;
last.previous = null;
last.next = null;
last.seqNum = 0;
length--;
return last;
}
master.seqNum = 0;
return null;
}
public void clear() {
master.previous = master;
master.next = master;
master.seqNum = 0;
length = 0;
}
class InternalNode implements HPackEntry {
private final HPackHeader header;
private InternalNode previous;
private InternalNode next;
private int seqNum;
InternalNode(final HPackHeader header) {
this.header = header;
}
@Override
public HPackHeader getHeader() {
return header;
}
@Override
public int getIndex() {
return StaticTable.INSTANCE.length() + FifoLinkedList.this.getIndex(this) + 1;
}
@Override
public String toString() {
return "[" +
(header != null ? header.toString() : "master") +
"; seqNum=" + seqNum +
"; previous=" + (previous != null ? previous.header : null) +
"; next=" + (next != null ? next.header : null) +
']';
}
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/HPackEntry.java 0100644 0000000 0000000 00000002430 12735734300 025111 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
interface HPackEntry {
int getIndex();
HPackHeader getHeader();
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/HPackHeader.java 0100644 0000000 0000000 00000005715 13341740003 025200 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
import org.apache.hc.core5.http.Header;
/**
* Internal HPack header representation that also contains binary length of
* header name and header value.
*/
final class HPackHeader implements Header {
static private final int ENTRY_SIZE_OVERHEAD = 32;
private final String name;
private final int nameLen;
private final String value;
private final int valueLen;
private final boolean sensitive;
HPackHeader(final String name, final int nameLen, final String value, final int valueLen, final boolean sensitive) {
this.name = name;
this.nameLen = nameLen;
this.value = value;
this.valueLen = valueLen;
this.sensitive = sensitive;
}
HPackHeader(final String name, final String value, final boolean sensitive) {
this(name, name.length(), value, value.length(), sensitive);
}
HPackHeader(final String name, final String value) {
this(name, value, false);
}
HPackHeader(final Header header) {
this(header.getName(), header.getValue(), header.isSensitive());
}
@Override
public String getName() {
return name;
}
public int getNameLen() {
return nameLen;
}
@Override
public String getValue() {
return value;
}
public int getValueLen() {
return valueLen;
}
@Override
public boolean isSensitive() {
return sensitive;
}
public int getTotalSize() {
return nameLen + valueLen + ENTRY_SIZE_OVERHEAD;
}
@Override
public String toString() {
final StringBuilder buf = new StringBuilder();
buf.append(this.name).append(": ");
if (this.value != null) {
buf.append(this.value);
}
return buf.toString();
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/Huffman.java 0100644 0000000 0000000 00000020206 13341740003 024455 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
import org.apache.hc.core5.annotation.Internal;
/**
* This Huffman codec implementation has been derived from Twitter HPack project
* (https://github.com/twitter/hpack)
*/
@Internal
public final class Huffman {
private Huffman() {
// Do not allow utility class to be instantiated.
}
static final int[] CODES = {
0x1ff8,
0x7fffd8,
0xfffffe2,
0xfffffe3,
0xfffffe4,
0xfffffe5,
0xfffffe6,
0xfffffe7,
0xfffffe8,
0xffffea,
0x3ffffffc,
0xfffffe9,
0xfffffea,
0x3ffffffd,
0xfffffeb,
0xfffffec,
0xfffffed,
0xfffffee,
0xfffffef,
0xffffff0,
0xffffff1,
0xffffff2,
0x3ffffffe,
0xffffff3,
0xffffff4,
0xffffff5,
0xffffff6,
0xffffff7,
0xffffff8,
0xffffff9,
0xffffffa,
0xffffffb,
0x14,
0x3f8,
0x3f9,
0xffa,
0x1ff9,
0x15,
0xf8,
0x7fa,
0x3fa,
0x3fb,
0xf9,
0x7fb,
0xfa,
0x16,
0x17,
0x18,
0x0,
0x1,
0x2,
0x19,
0x1a,
0x1b,
0x1c,
0x1d,
0x1e,
0x1f,
0x5c,
0xfb,
0x7ffc,
0x20,
0xffb,
0x3fc,
0x1ffa,
0x21,
0x5d,
0x5e,
0x5f,
0x60,
0x61,
0x62,
0x63,
0x64,
0x65,
0x66,
0x67,
0x68,
0x69,
0x6a,
0x6b,
0x6c,
0x6d,
0x6e,
0x6f,
0x70,
0x71,
0x72,
0xfc,
0x73,
0xfd,
0x1ffb,
0x7fff0,
0x1ffc,
0x3ffc,
0x22,
0x7ffd,
0x3,
0x23,
0x4,
0x24,
0x5,
0x25,
0x26,
0x27,
0x6,
0x74,
0x75,
0x28,
0x29,
0x2a,
0x7,
0x2b,
0x76,
0x2c,
0x8,
0x9,
0x2d,
0x77,
0x78,
0x79,
0x7a,
0x7b,
0x7ffe,
0x7fc,
0x3ffd,
0x1ffd,
0xffffffc,
0xfffe6,
0x3fffd2,
0xfffe7,
0xfffe8,
0x3fffd3,
0x3fffd4,
0x3fffd5,
0x7fffd9,
0x3fffd6,
0x7fffda,
0x7fffdb,
0x7fffdc,
0x7fffdd,
0x7fffde,
0xffffeb,
0x7fffdf,
0xffffec,
0xffffed,
0x3fffd7,
0x7fffe0,
0xffffee,
0x7fffe1,
0x7fffe2,
0x7fffe3,
0x7fffe4,
0x1fffdc,
0x3fffd8,
0x7fffe5,
0x3fffd9,
0x7fffe6,
0x7fffe7,
0xffffef,
0x3fffda,
0x1fffdd,
0xfffe9,
0x3fffdb,
0x3fffdc,
0x7fffe8,
0x7fffe9,
0x1fffde,
0x7fffea,
0x3fffdd,
0x3fffde,
0xfffff0,
0x1fffdf,
0x3fffdf,
0x7fffeb,
0x7fffec,
0x1fffe0,
0x1fffe1,
0x3fffe0,
0x1fffe2,
0x7fffed,
0x3fffe1,
0x7fffee,
0x7fffef,
0xfffea,
0x3fffe2,
0x3fffe3,
0x3fffe4,
0x7ffff0,
0x3fffe5,
0x3fffe6,
0x7ffff1,
0x3ffffe0,
0x3ffffe1,
0xfffeb,
0x7fff1,
0x3fffe7,
0x7ffff2,
0x3fffe8,
0x1ffffec,
0x3ffffe2,
0x3ffffe3,
0x3ffffe4,
0x7ffffde,
0x7ffffdf,
0x3ffffe5,
0xfffff1,
0x1ffffed,
0x7fff2,
0x1fffe3,
0x3ffffe6,
0x7ffffe0,
0x7ffffe1,
0x3ffffe7,
0x7ffffe2,
0xfffff2,
0x1fffe4,
0x1fffe5,
0x3ffffe8,
0x3ffffe9,
0xffffffd,
0x7ffffe3,
0x7ffffe4,
0x7ffffe5,
0xfffec,
0xfffff3,
0xfffed,
0x1fffe6,
0x3fffe9,
0x1fffe7,
0x1fffe8,
0x7ffff3,
0x3fffea,
0x3fffeb,
0x1ffffee,
0x1ffffef,
0xfffff4,
0xfffff5,
0x3ffffea,
0x7ffff4,
0x3ffffeb,
0x7ffffe6,
0x3ffffec,
0x3ffffed,
0x7ffffe7,
0x7ffffe8,
0x7ffffe9,
0x7ffffea,
0x7ffffeb,
0xffffffe,
0x7ffffec,
0x7ffffed,
0x7ffffee,
0x7ffffef,
0x7fffff0,
0x3ffffee,
0x3fffffff
};
static final byte[] LENGTHS = {
13, 23, 28, 28, 28, 28, 28, 28, 28, 24, 30, 28, 28, 30, 28, 28,
28, 28, 28, 28, 28, 28, 30, 28, 28, 28, 28, 28, 28, 28, 28, 28,
6, 10, 10, 12, 13, 6, 8, 11, 10, 10, 8, 11, 8, 6, 6, 6,
5, 5, 5, 6, 6, 6, 6, 6, 6, 6, 7, 8, 15, 6, 12, 10,
13, 6, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
7, 7, 7, 7, 7, 7, 7, 7, 8, 7, 8, 13, 19, 13, 14, 6,
15, 5, 6, 5, 6, 5, 6, 6, 6, 5, 7, 7, 6, 6, 6, 5,
6, 7, 6, 5, 5, 6, 7, 7, 7, 7, 7, 15, 11, 14, 13, 28,
20, 22, 20, 20, 22, 22, 22, 23, 22, 23, 23, 23, 23, 23, 24, 23,
24, 24, 22, 23, 24, 23, 23, 23, 23, 21, 22, 23, 22, 23, 23, 24,
22, 21, 20, 22, 22, 23, 23, 21, 23, 22, 22, 24, 21, 22, 23, 23,
21, 21, 22, 21, 23, 22, 23, 23, 20, 22, 22, 22, 23, 22, 22, 23,
26, 26, 20, 19, 22, 23, 22, 25, 26, 26, 26, 27, 27, 26, 24, 25,
19, 21, 26, 27, 27, 26, 27, 24, 21, 21, 26, 26, 28, 27, 27, 27,
20, 24, 20, 21, 22, 21, 21, 23, 22, 22, 25, 25, 24, 24, 26, 23,
26, 27, 26, 26, 27, 27, 27, 27, 27, 28, 27, 27, 27, 27, 27, 26,
30
};
static final int EOS = 256;
static final HuffmanEncoder ENCODER = new HuffmanEncoder(CODES, LENGTHS);
static final HuffmanDecoder DECODER = new HuffmanDecoder(CODES, LENGTHS);
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/HuffmanNode.java 0100644 0000000 0000000 00000005145 12735734300 025301 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
import java.util.Arrays;
import org.apache.hc.core5.util.Asserts;
/**
* This Huffman codec implementation has been derived from Twitter HPack project
* (https://github.com/twitter/hpack)
*/
final class HuffmanNode {
private final int symbol;
private final int bits;
private final HuffmanNode[] children;
HuffmanNode() {
this.symbol = 0;
this.bits = 8;
this.children = new HuffmanNode[256];
}
HuffmanNode(final int symbol, final int bits) {
this.symbol = symbol;
this.bits = bits;
this.children = null;
}
public int getBits() {
return this.bits;
}
public int getSymbol() {
return this.symbol;
}
public boolean hasChild(final int index) {
return this.children != null && this.children[index] != null;
}
public HuffmanNode getChild(final int index) {
return this.children != null ? this.children[index] : null;
}
void setChild(final int index, final HuffmanNode child) {
Asserts.notNull(this.children, "Children nodes");
this.children[index] = child;
}
public boolean isTerminal() {
return this.children == null;
}
@Override
public String toString() {
return "[" +
"symbol=" + symbol +
", bits=" + bits +
", children=" + Arrays.toString(children) +
']';
}
} httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/package-info.java 0100644 0000000 0000000 00000002556 13537747351 025451 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
/**
* HTTP/2 HPACK APIs.
*
* This Huffman codec implementation has been derived from Twitter HPack project
* (https://github.com/twitter/hpack)
*
*/
package org.apache.hc.core5.http2.hpack;
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/HPackRepresentation.java 0100644 0000000 0000000 00000002426 12735734300 027017 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
enum HPackRepresentation { WITH_INDEXING, WITHOUT_INDEXING, NEVER_INDEXED } httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/StaticTable.java 0100644 0000000 0000000 00000014436 13620207336 025307 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
import java.util.List;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.CopyOnWriteArrayList;
import org.apache.hc.core5.http2.H2PseudoRequestHeaders;
import org.apache.hc.core5.http2.H2PseudoResponseHeaders;
final class StaticTable {
static final HPackHeader[] STANDARD_HEADERS = {
new HPackHeader(H2PseudoRequestHeaders.AUTHORITY, ""),
new HPackHeader(H2PseudoRequestHeaders.METHOD, "GET"),
new HPackHeader(H2PseudoRequestHeaders.METHOD, "POST"),
new HPackHeader(H2PseudoRequestHeaders.PATH, "/"),
new HPackHeader(H2PseudoRequestHeaders.PATH, "/index.html"),
new HPackHeader(H2PseudoRequestHeaders.SCHEME, "http"),
new HPackHeader(H2PseudoRequestHeaders.SCHEME, "https"),
new HPackHeader(H2PseudoResponseHeaders.STATUS, "200"),
new HPackHeader(H2PseudoResponseHeaders.STATUS, "204"),
new HPackHeader(H2PseudoResponseHeaders.STATUS, "206"),
new HPackHeader(H2PseudoResponseHeaders.STATUS, "304"),
new HPackHeader(H2PseudoResponseHeaders.STATUS, "400"),
new HPackHeader(H2PseudoResponseHeaders.STATUS, "404"),
new HPackHeader(H2PseudoResponseHeaders.STATUS, "500"),
new HPackHeader("accept-charset", ""),
new HPackHeader("accept-encoding", "gzip, deflate"),
new HPackHeader("accept-language", ""),
new HPackHeader("accept-ranges", ""),
new HPackHeader("accept", ""),
new HPackHeader("access-control-allow-origin", ""),
new HPackHeader("age", ""),
new HPackHeader("allow", ""),
new HPackHeader("authorization", ""),
new HPackHeader("cache-control", ""),
new HPackHeader("content-disposition", ""),
new HPackHeader("content-encoding", ""),
new HPackHeader("content-language", ""),
new HPackHeader("content-length", ""),
new HPackHeader("content-location", ""),
new HPackHeader("content-range", ""),
new HPackHeader("content-type", ""),
new HPackHeader("cookie", ""),
new HPackHeader("date", ""),
new HPackHeader("etag", ""),
new HPackHeader("expect", ""),
new HPackHeader("expires", ""),
new HPackHeader("from", ""),
new HPackHeader("host", ""),
new HPackHeader("if-match", ""),
new HPackHeader("if-modified-since", ""),
new HPackHeader("if-none-match", ""),
new HPackHeader("if-range", ""),
new HPackHeader("if-unmodified-since", ""),
new HPackHeader("last-modified", ""),
new HPackHeader("link", ""),
new HPackHeader("location", ""),
new HPackHeader("max-forwards", ""),
new HPackHeader("proxy-authenticate", ""),
new HPackHeader("proxy-authorization", ""),
new HPackHeader("range", ""),
new HPackHeader("referer", ""),
new HPackHeader("refresh", ""),
new HPackHeader("retry-after", ""),
new HPackHeader("server", ""),
new HPackHeader("set-cookie", ""),
new HPackHeader("strict-transport-security", ""),
new HPackHeader("transfer-encoding", ""),
new HPackHeader("user-agent", ""),
new HPackHeader("vary", ""),
new HPackHeader("via", ""),
new HPackHeader("www-authenticate", "")
};
final static StaticTable INSTANCE = new StaticTable(STANDARD_HEADERS);
private final HPackHeader[] headers;
private final ConcurrentMap> mapByName;
StaticTable(final HPackHeader... headers) {
this.headers = headers;
this.mapByName = new ConcurrentHashMap<>();
for (int i = 0; i < headers.length; i++) {
final HPackHeader header = headers[i];
final String key = header.getName();
CopyOnWriteArrayList entries = this.mapByName.get(key);
if (entries == null) {
entries = new CopyOnWriteArrayList<>(new HPackEntry[] { new InternalEntry(header, i) });
this.mapByName.put(key, entries);
} else {
entries.add(new InternalEntry(header, i));
}
}
}
public int length() {
return this.headers.length;
}
public HPackHeader get(final int index) {
return this.headers[index - 1];
}
public List getByName(final String key) {
return this.mapByName.get(key);
}
static class InternalEntry implements HPackEntry {
private final HPackHeader header;
private final int index;
InternalEntry(final HPackHeader header, final int index) {
this.header = header;
this.index = index;
}
@Override
public int getIndex() {
return index + 1;
}
@Override
public HPackHeader getHeader() {
return header;
}
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/HeaderListConstraintException.java 0100664 0000000 0000000 00000003126 14157612023 031053 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
/**
* Signals a header list constraint violation.
*
* @since 5.0
*/
public class HeaderListConstraintException extends HPackException {
/**
* Creates a HeaderListConstraintException with the specified detail message.
*
* @param message The exception detail message
*/
public HeaderListConstraintException(final String message) {
super(message);
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/HPackDecoder.java 0100664 0000000 0000000 00000027334 14157602333 025371 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
import java.nio.ByteBuffer;
import java.nio.CharBuffer;
import java.nio.charset.CharacterCodingException;
import java.nio.charset.Charset;
import java.nio.charset.CharsetDecoder;
import java.nio.charset.CoderResult;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;
import org.apache.hc.core5.annotation.Internal;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.message.BasicHeader;
import org.apache.hc.core5.util.Args;
import org.apache.hc.core5.util.ByteArrayBuffer;
/**
* HPACK decoder.
*
* @since 5.0
*/
@Internal
public final class HPackDecoder {
private static final String UNEXPECTED_EOS = "Unexpected end of HPACK data";
private static final String MAX_LIMIT_EXCEEDED = "Max integer exceeded";
private final InboundDynamicTable dynamicTable;
private final ByteArrayBuffer contentBuf;
private final CharsetDecoder charsetDecoder;
private CharBuffer tmpBuf;
private int maxTableSize;
private int maxListSize;
HPackDecoder(final InboundDynamicTable dynamicTable, final CharsetDecoder charsetDecoder) {
this.dynamicTable = dynamicTable != null ? dynamicTable : new InboundDynamicTable();
this.contentBuf = new ByteArrayBuffer(256);
this.charsetDecoder = charsetDecoder;
this.maxTableSize = dynamicTable != null ? dynamicTable.getMaxSize() : Integer.MAX_VALUE;
this.maxListSize = Integer.MAX_VALUE;
}
HPackDecoder(final InboundDynamicTable dynamicTable, final Charset charset) {
this(dynamicTable, charset != null && !StandardCharsets.US_ASCII.equals(charset) ? charset.newDecoder() : null);
}
public HPackDecoder(final Charset charset) {
this(new InboundDynamicTable(), charset);
}
public HPackDecoder(final CharsetDecoder charsetDecoder) {
this(new InboundDynamicTable(), charsetDecoder);
}
static int readByte(final ByteBuffer src) throws HPackException {
if (!src.hasRemaining()) {
throw new HPackException(UNEXPECTED_EOS);
}
return src.get() & 0xff;
}
static int peekByte(final ByteBuffer src) throws HPackException {
if (!src.hasRemaining()) {
throw new HPackException(UNEXPECTED_EOS);
}
final int pos = src.position();
final int b = src.get() & 0xff;
src.position(pos);
return b;
}
static int decodeInt(final ByteBuffer src, final int n) throws HPackException {
final int nbits = 0xff >>> (8 - n);
int value = readByte(src) & nbits;
if (value < nbits) {
return value;
}
int m = 0;
while (m < 32) {
final int b = readByte(src);
if ((b & 0x80) != 0) {
value += (b & 0x7f) << m;
m += 7;
} else {
if (m == 28 && (b & 0xf8) != 0) {
break;
}
value += b << m;
return value;
}
}
throw new HPackException(MAX_LIMIT_EXCEEDED);
}
static void decodePlainString(final ByteArrayBuffer buffer, final ByteBuffer src) throws HPackException {
final int strLen = decodeInt(src, 7);
final int remaining = src.remaining();
if (strLen > remaining) {
throw new HPackException(UNEXPECTED_EOS);
}
final int originalLimit = src.limit();
src.limit(originalLimit - (remaining - strLen));
buffer.append(src);
src.limit(originalLimit);
}
static void decodeHuffman(final ByteArrayBuffer buffer, final ByteBuffer src) throws HPackException {
final int strLen = decodeInt(src, 7);
if (strLen > src.remaining()) {
throw new HPackException(UNEXPECTED_EOS);
}
final int limit = src.limit();
src.limit(src.position() + strLen);
Huffman.DECODER.decode(buffer, src);
src.limit(limit);
}
void decodeString(final ByteArrayBuffer buffer, final ByteBuffer src) throws HPackException {
final int firstByte = peekByte(src);
if ((firstByte & 0x80) == 0x80) {
decodeHuffman(buffer, src);
} else {
decodePlainString(buffer, src);
}
}
private void clearState() {
if (this.tmpBuf != null) {
this.tmpBuf.clear();
}
if (this.charsetDecoder != null) {
this.charsetDecoder.reset();
}
this.contentBuf.clear();
}
private void expandCapacity(final int capacity) {
final CharBuffer previous = this.tmpBuf;
this.tmpBuf = CharBuffer.allocate(capacity);
previous.flip();
this.tmpBuf.put(previous);
}
private void ensureCapacity(final int extra) {
if (this.tmpBuf == null) {
this.tmpBuf = CharBuffer.allocate(Math.max(256, extra));
}
final int requiredCapacity = this.tmpBuf.remaining() + extra;
if (requiredCapacity > this.tmpBuf.capacity()) {
expandCapacity(requiredCapacity);
}
}
int decodeString(final ByteBuffer src, final StringBuilder buf) throws HPackException, CharacterCodingException {
clearState();
decodeString(this.contentBuf, src);
final int binaryLen = this.contentBuf.length();
if (binaryLen == 0) {
return 0;
}
if (this.charsetDecoder == null) {
buf.ensureCapacity(binaryLen);
for (int i = 0; i < binaryLen; i++) {
buf.append((char) (this.contentBuf.byteAt(i) & 0xff));
}
} else {
final ByteBuffer in = ByteBuffer.wrap(this.contentBuf.array(), 0, binaryLen);
while (in.hasRemaining()) {
ensureCapacity(in.remaining());
final CoderResult result = this.charsetDecoder.decode(in, this.tmpBuf, true);
if (result.isError()) {
result.throwException();
}
}
ensureCapacity(8);
final CoderResult result = this.charsetDecoder.flush(this.tmpBuf);
if (result.isError()) {
result.throwException();
}
this.tmpBuf.flip();
buf.append(this.tmpBuf);
}
return binaryLen;
}
HPackHeader decodeLiteralHeader(
final ByteBuffer src,
final HPackRepresentation representation) throws HPackException, CharacterCodingException {
final int n = representation == HPackRepresentation.WITH_INDEXING ? 6 : 4;
final int index = decodeInt(src, n);
final String name;
final int nameLen;
if (index == 0) {
final StringBuilder buf = new StringBuilder();
nameLen = decodeString(src, buf);
name = buf.toString();
} else {
final HPackHeader existing = this.dynamicTable.getHeader(index);
if (existing == null) {
throw new HPackException("Invalid header index");
}
name = existing.getName();
nameLen = existing.getNameLen();
}
final StringBuilder buf = new StringBuilder();
final int valueLen = decodeString(src, buf);
final String value = buf.toString();
final HPackHeader header = new HPackHeader(name, nameLen, value, valueLen, representation == HPackRepresentation.NEVER_INDEXED);
if (representation == HPackRepresentation.WITH_INDEXING) {
this.dynamicTable.add(header);
}
return header;
}
HPackHeader decodeIndexedHeader(final ByteBuffer src) throws HPackException {
final int index = decodeInt(src, 7);
final HPackHeader existing = this.dynamicTable.getHeader(index);
if (existing == null) {
throw new HPackException("Invalid header index");
}
return existing;
}
public Header decodeHeader(final ByteBuffer src) throws HPackException {
final HPackHeader header = decodeHPackHeader(src);
return header != null ? new BasicHeader(header.getName(), header.getValue(), header.isSensitive()) : null;
}
HPackHeader decodeHPackHeader(final ByteBuffer src) throws HPackException {
try {
while (src.hasRemaining()) {
final int b = peekByte(src);
if ((b & 0x80) == 0x80) {
return decodeIndexedHeader(src);
} else if ((b & 0xc0) == 0x40) {
return decodeLiteralHeader(src, HPackRepresentation.WITH_INDEXING);
} else if ((b & 0xf0) == 0x00) {
return decodeLiteralHeader(src, HPackRepresentation.WITHOUT_INDEXING);
} else if ((b & 0xf0) == 0x10) {
return decodeLiteralHeader(src, HPackRepresentation.NEVER_INDEXED);
} else if ((b & 0xe0) == 0x20) {
final int maxSize = decodeInt(src, 5);
this.dynamicTable.setMaxSize(Math.min(this.maxTableSize, maxSize));
} else {
throw new HPackException("Unexpected header first byte: 0x" + Integer.toHexString(b));
}
}
return null;
} catch (final CharacterCodingException ex) {
throw new HPackException(ex.getMessage(), ex);
}
}
public List decodeHeaders(final ByteBuffer src) throws HPackException {
final boolean enforceSizeLimit = maxListSize < Integer.MAX_VALUE;
int listSize = 0;
final List list = new ArrayList<>();
while (src.hasRemaining()) {
final HPackHeader header = decodeHPackHeader(src);
if (header == null) {
break;
}
if (enforceSizeLimit) {
listSize += header.getTotalSize();
if (listSize >= maxListSize) {
throw new HeaderListConstraintException("Maximum header list size exceeded");
}
}
list.add(new BasicHeader(header.getName(), header.getValue(), header.isSensitive()));
}
return list;
}
public int getMaxTableSize() {
return this.maxTableSize;
}
public void setMaxTableSize(final int maxTableSize) {
Args.notNegative(maxTableSize, "Max table size");
this.maxTableSize = maxTableSize;
this.dynamicTable.setMaxSize(maxTableSize);
}
public int getMaxListSize() {
return maxListSize;
}
public void setMaxListSize(final int maxListSize) {
Args.notNegative(maxListSize, "Max list size");
this.maxListSize = maxListSize;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/hpack/HPackEncoder.java 0100664 0000000 0000000 00000031053 14157602333 025374 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.hpack;
import java.nio.ByteBuffer;
import java.nio.CharBuffer;
import java.nio.charset.CharacterCodingException;
import java.nio.charset.Charset;
import java.nio.charset.CharsetEncoder;
import java.nio.charset.CoderResult;
import java.nio.charset.StandardCharsets;
import java.util.List;
import org.apache.hc.core5.annotation.Internal;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.util.Args;
import org.apache.hc.core5.util.ByteArrayBuffer;
import org.apache.hc.core5.util.LangUtils;
/**
* HPACK encoder.
*
* @since 5.0
*/
@Internal
public final class HPackEncoder {
private final OutboundDynamicTable dynamicTable;
private final ByteArrayBuffer huffmanBuf;
private final CharsetEncoder charsetEncoder;
private ByteBuffer tmpBuf;
private int maxTableSize;
HPackEncoder(final OutboundDynamicTable dynamicTable, final CharsetEncoder charsetEncoder) {
this.dynamicTable = dynamicTable != null ? dynamicTable : new OutboundDynamicTable();
this.huffmanBuf = new ByteArrayBuffer(128);
this.charsetEncoder = charsetEncoder;
}
HPackEncoder(final OutboundDynamicTable dynamicTable, final Charset charset) {
this(dynamicTable, charset != null && !StandardCharsets.US_ASCII.equals(charset) ? charset.newEncoder() : null);
}
public HPackEncoder(final Charset charset) {
this(new OutboundDynamicTable(), charset);
}
public HPackEncoder(final CharsetEncoder charsetEncoder) {
this(new OutboundDynamicTable(), charsetEncoder);
}
static void encodeInt(final ByteArrayBuffer dst, final int n, final int i, final int mask) {
final int nbits = 0xFF >>> (8 - n);
int value = i;
if (value < nbits) {
dst.append(i | mask);
} else {
dst.append(nbits | mask);
value -= nbits;
while (value >= 0x80) {
dst.append((value & 0x7F) | 0x80);
value >>>= 7;
}
dst.append(value);
}
}
static void encodeHuffman(final ByteArrayBuffer dst, final ByteBuffer src) {
Huffman.ENCODER.encode(dst, src);
}
void encodeString(final ByteArrayBuffer dst, final ByteBuffer src, final boolean huffman) {
final int strLen = src.remaining();
if (huffman) {
this.huffmanBuf.clear();
this.huffmanBuf.ensureCapacity(strLen);
Huffman.ENCODER.encode(this.huffmanBuf, src);
dst.ensureCapacity(this.huffmanBuf.length() + 8);
encodeInt(dst, 7, this.huffmanBuf.length(), 0x80);
dst.append(this.huffmanBuf.array(), 0, this.huffmanBuf.length());
} else {
dst.ensureCapacity(strLen + 8);
encodeInt(dst, 7, strLen, 0x0);
dst.append(src);
}
}
private void clearState() {
if (this.tmpBuf != null) {
this.tmpBuf.clear();
}
if (this.charsetEncoder != null) {
this.charsetEncoder.reset();
}
}
private void expandCapacity(final int capacity) {
final ByteBuffer previous = this.tmpBuf;
this.tmpBuf = ByteBuffer.allocate(capacity);
previous.flip();
this.tmpBuf.put(previous);
}
private void ensureCapacity(final int extra) {
if (this.tmpBuf == null) {
this.tmpBuf = ByteBuffer.allocate(Math.max(256, extra));
}
final int requiredCapacity = this.tmpBuf.remaining() + extra;
if (requiredCapacity > this.tmpBuf.capacity()) {
expandCapacity(requiredCapacity);
}
}
int encodeString(
final ByteArrayBuffer dst,
final CharSequence charSequence, final int off, final int len,
final boolean huffman) throws CharacterCodingException {
clearState();
if (this.charsetEncoder == null) {
if (huffman) {
this.huffmanBuf.clear();
this.huffmanBuf.ensureCapacity(len);
Huffman.ENCODER.encode(this.huffmanBuf, charSequence, off, len);
dst.ensureCapacity(this.huffmanBuf.length() + 8);
encodeInt(dst, 7, this.huffmanBuf.length(), 0x80);
dst.append(this.huffmanBuf.array(), 0, this.huffmanBuf.length());
} else {
dst.ensureCapacity(len + 8);
encodeInt(dst, 7, len, 0x0);
for (int i = 0; i < len; i++) {
dst.append(charSequence.charAt(off + i));
}
}
return len;
}
if (charSequence.length() > 0) {
final CharBuffer in = CharBuffer.wrap(charSequence, off, len);
while (in.hasRemaining()) {
ensureCapacity((int) (in.remaining() * this.charsetEncoder.averageBytesPerChar()) + 8);
final CoderResult result = this.charsetEncoder.encode(in, this.tmpBuf, true);
if (result.isError()) {
result.throwException();
}
}
ensureCapacity(8);
final CoderResult result = this.charsetEncoder.flush(this.tmpBuf);
if (result.isError()) {
result.throwException();
}
}
this.tmpBuf.flip();
final int binaryLen = this.tmpBuf.remaining();
encodeString(dst, this.tmpBuf, huffman);
return binaryLen;
}
int encodeString(final ByteArrayBuffer dst, final String s, final boolean huffman) throws CharacterCodingException {
return encodeString(dst, s, 0, s.length(), huffman);
}
void encodeLiteralHeader(
final ByteArrayBuffer dst, final HPackEntry existing, final Header header,
final HPackRepresentation representation, final boolean useHuffman) throws CharacterCodingException {
encodeLiteralHeader(dst, existing, header.getName(), header.getValue(), header.isSensitive(), representation, useHuffman);
}
void encodeLiteralHeader(
final ByteArrayBuffer dst, final HPackEntry existing, final String key, final String value, final boolean sensitive,
final HPackRepresentation representation, final boolean useHuffman) throws CharacterCodingException {
final int n;
final int mask;
switch (representation) {
case WITH_INDEXING:
mask = 0x40;
n = 6;
break;
case WITHOUT_INDEXING:
mask = 0x00;
n = 4;
break;
case NEVER_INDEXED:
mask = 0x10;
n = 4;
break;
default:
throw new IllegalStateException("Unexpected value: " + representation);
}
final int index = existing != null ? existing.getIndex() : 0;
final int nameLen;
if (index <= 0) {
encodeInt(dst, n, 0, mask);
nameLen = encodeString(dst, key, useHuffman);
} else {
encodeInt(dst, n, index, mask);
nameLen = existing.getHeader().getNameLen();
}
final int valueLen = encodeString(dst, value != null ? value : "", useHuffman);
if (representation == HPackRepresentation.WITH_INDEXING) {
dynamicTable.add(new HPackHeader(key, nameLen, value, valueLen, sensitive));
}
}
void encodeIndex(final ByteArrayBuffer dst, final int index) {
encodeInt(dst, 7, index, 0x80);
}
private int findFullMatch(final List entries, final String value) {
if (entries == null || entries.isEmpty()) {
return 0;
}
for (int i = 0; i < entries.size(); i++) {
final HPackEntry entry = entries.get(i);
if (LangUtils.equals(value, entry.getHeader().getValue())) {
return entry.getIndex();
}
}
return 0;
}
void encodeHeader(
final ByteArrayBuffer dst, final Header header,
final boolean noIndexing, final boolean useHuffman) throws CharacterCodingException {
encodeHeader(dst, header.getName(), header.getValue(), header.isSensitive(), noIndexing, useHuffman);
}
void encodeHeader(
final ByteArrayBuffer dst, final String name, final String value, final boolean sensitive,
final boolean noIndexing, final boolean useHuffman) throws CharacterCodingException {
final HPackRepresentation representation;
if (sensitive) {
representation = HPackRepresentation.NEVER_INDEXED;
} else if (noIndexing) {
representation = HPackRepresentation.WITHOUT_INDEXING;
} else {
representation = HPackRepresentation.WITH_INDEXING;
}
final List staticEntries = StaticTable.INSTANCE.getByName(name);
if (representation == HPackRepresentation.WITH_INDEXING) {
// Try to find full match and encode as as index
final int staticIndex = findFullMatch(staticEntries, value);
if (staticIndex > 0) {
encodeIndex(dst, staticIndex);
return;
}
final List dynamicEntries = dynamicTable.getByName(name);
final int dynamicIndex = findFullMatch(dynamicEntries, value);
if (dynamicIndex > 0) {
encodeIndex(dst, dynamicIndex);
return;
}
}
// Encode as literal
HPackEntry existing = null;
if (staticEntries != null && !staticEntries.isEmpty()) {
existing = staticEntries.get(0);
} else {
final List dynamicEntries = dynamicTable.getByName(name);
if (dynamicEntries != null && !dynamicEntries.isEmpty()) {
existing = dynamicEntries.get(0);
}
}
encodeLiteralHeader(dst, existing, name, value, sensitive, representation, useHuffman);
}
void encodeHeaders(
final ByteArrayBuffer dst, final List headers,
final boolean noIndexing, final boolean useHuffman) throws CharacterCodingException {
for (int i = 0; i < headers.size(); i++) {
encodeHeader(dst, headers.get(i), noIndexing, useHuffman);
}
}
public void encodeHeader(
final ByteArrayBuffer dst, final Header header) throws CharacterCodingException {
Args.notNull(dst, "ByteArrayBuffer");
Args.notNull(header, "Header");
encodeHeader(dst, header.getName(), header.getValue(), header.isSensitive());
}
public void encodeHeader(
final ByteArrayBuffer dst, final String name, final String value, final boolean sensitive) throws CharacterCodingException {
Args.notNull(dst, "ByteArrayBuffer");
Args.notEmpty(name, "Header name");
encodeHeader(dst, name, value, sensitive, false, true);
}
public void encodeHeaders(
final ByteArrayBuffer dst, final List headers, final boolean useHuffman) throws CharacterCodingException {
Args.notNull(dst, "ByteArrayBuffer");
Args.notEmpty(headers, "Header list");
encodeHeaders(dst, headers, false, useHuffman);
}
public int getMaxTableSize() {
return this.maxTableSize;
}
public void setMaxTableSize(final int maxTableSize) {
Args.notNegative(maxTableSize, "Max table size");
this.maxTableSize = maxTableSize;
this.dynamicTable.setMaxSize(maxTableSize);
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/H2PseudoRequestHeaders.java 0100644 0000000 0000000 00000003053 13341740003 026302 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2;
/**
* Request pseudo HTTP headers defined by the HTTP/2 specification.
*
* @since 5.0
*/
public final class H2PseudoRequestHeaders {
public static final String METHOD = ":method";
public static final String SCHEME = ":scheme";
public static final String AUTHORITY = ":authority";
public static final String PATH = ":path";
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/package-info.java 0100644 0000000 0000000 00000002365 13341740003 024335 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
/**
* Core HTTP/2 transport component APIs.
*/
package org.apache.hc.core5.http2;
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/ 0040755 0000000 0000000 00000000000 14157612023 022112 5 ustar 00 0000000 0000000 httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/DefaultH2ResponseConverter.java 0100664 0000000 0000000 00000012677 14157612023 030156 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Locale;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.HttpException;
import org.apache.hc.core5.http.HttpHeaders;
import org.apache.hc.core5.http.HttpResponse;
import org.apache.hc.core5.http.HttpVersion;
import org.apache.hc.core5.http.ProtocolException;
import org.apache.hc.core5.http.message.BasicHeader;
import org.apache.hc.core5.http.message.BasicHttpResponse;
import org.apache.hc.core5.http2.H2MessageConverter;
import org.apache.hc.core5.http2.H2PseudoResponseHeaders;
/**
* HTTP/2 response converter.
*
* @since 5.0
*/
public class DefaultH2ResponseConverter implements H2MessageConverter {
public final static DefaultH2ResponseConverter INSTANCE = new DefaultH2ResponseConverter();
@Override
public HttpResponse convert(final List headers) throws HttpException {
String statusText = null;
final List messageHeaders = new ArrayList<>();
for (int i = 0; i < headers.size(); i++) {
final Header header = headers.get(i);
final String name = header.getName();
final String value = header.getValue();
for (int n = 0; n < name.length(); n++) {
final char ch = name.charAt(n);
if (Character.isAlphabetic(ch) && !Character.isLowerCase(ch)) {
throw new ProtocolException("Header name '%s' is invalid (header name contains uppercase characters)", name);
}
}
if (name.startsWith(":")) {
if (!messageHeaders.isEmpty()) {
throw new ProtocolException("Invalid sequence of headers (pseudo-headers must precede message headers)");
}
if (name.equals(H2PseudoResponseHeaders.STATUS)) {
if (statusText != null) {
throw new ProtocolException("Multiple '%s' response headers are illegal", name);
}
statusText = value;
} else {
throw new ProtocolException("Unsupported response header '%s'", name);
}
} else {
if (name.equalsIgnoreCase(HttpHeaders.CONNECTION)) {
throw new ProtocolException("Header '%s: %s' is illegal for HTTP/2 messages", header.getName(), header.getValue());
}
messageHeaders.add(header);
}
}
if (statusText == null) {
throw new ProtocolException("Mandatory response header '%s' not found", H2PseudoResponseHeaders.STATUS);
}
final int statusCode;
try {
statusCode = Integer.parseInt(statusText);
} catch (final NumberFormatException ex) {
throw new ProtocolException("Invalid response status: " + statusText);
}
final HttpResponse response = new BasicHttpResponse(statusCode, null);
response.setVersion(HttpVersion.HTTP_2);
for (int i = 0; i < messageHeaders.size(); i++) {
response.addHeader(messageHeaders.get(i));
}
return response;
}
@Override
public List convert(final HttpResponse message) throws HttpException {
final int code = message.getCode();
if (code < 100 || code >= 600) {
throw new ProtocolException("Response status %s is invalid", code);
}
final List headers = new ArrayList<>();
headers.add(new BasicHeader(H2PseudoResponseHeaders.STATUS, Integer.toString(code), false));
for (final Iterator it = message.headerIterator(); it.hasNext(); ) {
final Header header = it.next();
final String name = header.getName();
final String value = header.getValue();
if (name.startsWith(":")) {
throw new ProtocolException("Header name '%s' is invalid", name);
}
if (name.equalsIgnoreCase(HttpHeaders.CONNECTION)) {
throw new ProtocolException("Header '%s: %s' is illegal for HTTP/2 messages", name, value);
}
headers.add(new BasicHeader(name.toLowerCase(Locale.ROOT), value));
}
return headers;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/BasicH2TransportMetrics.java 0100644 0000000 0000000 00000003610 13012072252 027422 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl;
import java.util.concurrent.atomic.AtomicLong;
import org.apache.hc.core5.http.impl.BasicHttpTransportMetrics;
import org.apache.hc.core5.http2.H2TransportMetrics;
/**
* Default implementation of {@link H2TransportMetrics}.
*
* @since 5.0
*/
public class BasicH2TransportMetrics extends BasicHttpTransportMetrics implements H2TransportMetrics {
private final AtomicLong framesTransferred;
public BasicH2TransportMetrics() {
this.framesTransferred = new AtomicLong(0);
}
@Override
public long getFramesTransferred() {
return framesTransferred.get();
}
public void incrementFramesTransferred() {
framesTransferred.incrementAndGet();
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/package-info.java 0100644 0000000 0000000 00000002401 13341740003 025265 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
/**
* Default implementations of core HTTP/2 APIs.
*/
package org.apache.hc.core5.http2.impl;
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/H2Processors.java 0100644 0000000 0000000 00000007330 13511606054 025311 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl;
import org.apache.hc.core5.http.impl.HttpProcessors;
import org.apache.hc.core5.http.protocol.HttpProcessor;
import org.apache.hc.core5.http.protocol.HttpProcessorBuilder;
import org.apache.hc.core5.http.protocol.RequestExpectContinue;
import org.apache.hc.core5.http.protocol.RequestUserAgent;
import org.apache.hc.core5.http.protocol.ResponseDate;
import org.apache.hc.core5.http.protocol.ResponseServer;
import org.apache.hc.core5.http2.protocol.H2RequestConnControl;
import org.apache.hc.core5.http2.protocol.H2RequestContent;
import org.apache.hc.core5.http2.protocol.H2RequestTargetHost;
import org.apache.hc.core5.http2.protocol.H2RequestValidateHost;
import org.apache.hc.core5.http2.protocol.H2ResponseConnControl;
import org.apache.hc.core5.http2.protocol.H2ResponseContent;
import org.apache.hc.core5.util.TextUtils;
import org.apache.hc.core5.util.VersionInfo;
/**
* @since 5.0
*/
public final class H2Processors {
private final static String SOFTWARE = "Apache-HttpCore";
public static HttpProcessorBuilder customServer(final String serverInfo) {
return HttpProcessorBuilder.create()
.addAll(
new ResponseDate(),
new ResponseServer(!TextUtils.isBlank(serverInfo) ? serverInfo :
VersionInfo.getSoftwareInfo(SOFTWARE, "org.apache.hc.core5", H2Processors.class)),
new H2ResponseContent(),
new H2ResponseConnControl())
.addAll(
new H2RequestValidateHost());
}
public static HttpProcessor server(final String serverInfo) {
return customServer(serverInfo).build();
}
public static HttpProcessor server() {
return customServer(null).build();
}
public static HttpProcessorBuilder customClient(final String agentInfo) {
return HttpProcessorBuilder.create()
.addAll(
new H2RequestContent(),
new H2RequestTargetHost(),
new H2RequestConnControl(),
new RequestUserAgent(!TextUtils.isBlank(agentInfo) ? agentInfo :
VersionInfo.getSoftwareInfo(SOFTWARE, "org.apache.hc.core5", HttpProcessors.class)),
new RequestExpectContinue());
}
public static HttpProcessor client(final String agentInfo) {
return customClient(agentInfo).build();
}
public static HttpProcessor client() {
return customClient(null).build();
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/io/ 0040755 0000000 0000000 00000000000 14114632321 022515 5 ustar 00 0000000 0000000 httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/io/FrameOutputBuffer.java 0100664 0000000 0000000 00000010311 14114632321 026760 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.io;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.ByteBuffer;
import org.apache.hc.core5.http2.H2ConnectionException;
import org.apache.hc.core5.http2.H2Error;
import org.apache.hc.core5.http2.H2TransportMetrics;
import org.apache.hc.core5.http2.frame.FrameConsts;
import org.apache.hc.core5.http2.frame.FrameFlag;
import org.apache.hc.core5.http2.frame.RawFrame;
import org.apache.hc.core5.http2.impl.BasicH2TransportMetrics;
import org.apache.hc.core5.util.Args;
/**
* Frame output buffer for HTTP/2 blocking connections.
*
* @since 5.0
*/
public final class FrameOutputBuffer {
private final BasicH2TransportMetrics metrics;
private final int maxFramePayloadSize;
private final byte[] buffer;
public FrameOutputBuffer(final BasicH2TransportMetrics metrics, final int maxFramePayloadSize) {
super();
Args.notNull(metrics, "HTTP2 transport metrics");
Args.positive(maxFramePayloadSize, "Maximum payload size");
this.metrics = metrics;
this.maxFramePayloadSize = maxFramePayloadSize;
this.buffer = new byte[FrameConsts.HEAD_LEN + maxFramePayloadSize + FrameConsts.MAX_PADDING + 1];
}
public FrameOutputBuffer(final int maxFramePayloadSize) {
this(new BasicH2TransportMetrics(), maxFramePayloadSize);
}
public void write(final RawFrame frame, final OutputStream outStream) throws IOException {
if (frame == null) {
return;
}
final int type = frame.getType();
final long streamId = frame.getStreamId();
final int flags = frame.getFlags();
final ByteBuffer payload = frame.getPayload();
final int payloadLen = payload != null ? payload.remaining() : 0;
if (payload != null && payload.remaining() > maxFramePayloadSize) {
throw new H2ConnectionException(H2Error.FRAME_SIZE_ERROR, "Frame size exceeds maximum");
}
buffer[0] = (byte) (payloadLen >> 16 & 0xff);
buffer[1] = (byte) (payloadLen >> 8 & 0xff);
buffer[2] = (byte) (payloadLen & 0xff);
buffer[3] = (byte) (type & 0xff);
buffer[4] = (byte) (flags & 0xff);
buffer[5] = (byte) (streamId >> 24 & 0xff);
buffer[6] = (byte) (streamId >> 16 & 0xff);
buffer[7] = (byte) (streamId >> 8 & 0xff);
buffer[8] = (byte) (streamId & 0xff);
int frameLen = FrameConsts.HEAD_LEN;
int padding = 0;
if ((flags & FrameFlag.PADDED.getValue()) > 0) {
padding = 16;
buffer[9] = (byte) (padding & 0xff);
frameLen++;
}
if (payload != null) {
payload.get(buffer, frameLen, payload.remaining());
frameLen += payloadLen;
}
for (int i = 0; i < padding; i++) {
buffer[frameLen++] = 0;
}
outStream.write(buffer, 0, frameLen);
metrics.incrementFramesTransferred();
metrics.incrementBytesTransferred(frameLen);
}
public H2TransportMetrics getMetrics() {
return metrics;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/io/FrameInputBuffer.java 0100664 0000000 0000000 00000012303 14114632321 026562 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.io;
import java.io.IOException;
import java.io.InputStream;
import java.nio.ByteBuffer;
import org.apache.hc.core5.http.ConnectionClosedException;
import org.apache.hc.core5.http2.H2ConnectionException;
import org.apache.hc.core5.http2.H2CorruptFrameException;
import org.apache.hc.core5.http2.H2Error;
import org.apache.hc.core5.http2.H2TransportMetrics;
import org.apache.hc.core5.http2.frame.FrameConsts;
import org.apache.hc.core5.http2.frame.FrameFlag;
import org.apache.hc.core5.http2.frame.RawFrame;
import org.apache.hc.core5.http2.impl.BasicH2TransportMetrics;
import org.apache.hc.core5.util.Args;
/**
* Frame input buffer for HTTP/2 blocking connections.
*
* @since 5.0
*/
public final class FrameInputBuffer {
private final BasicH2TransportMetrics metrics;
private final int maxFramePayloadSize;
private final byte[] buffer;
private int off;
private int dataLen;
FrameInputBuffer(final BasicH2TransportMetrics metrics, final int bufferLen, final int maxFramePayloadSize) {
Args.notNull(metrics, "HTTP2 transport metrics");
Args.positive(maxFramePayloadSize, "Maximum payload size");
this.metrics = metrics;
this.maxFramePayloadSize = maxFramePayloadSize;
this.buffer = new byte[bufferLen];
this.dataLen = 0;
}
public FrameInputBuffer(final BasicH2TransportMetrics metrics, final int maxFramePayloadSize) {
this(metrics, FrameConsts.HEAD_LEN + maxFramePayloadSize, maxFramePayloadSize);
}
public FrameInputBuffer(final int maxFramePayloadSize) {
this(new BasicH2TransportMetrics(), maxFramePayloadSize);
}
boolean hasData() {
return this.dataLen > 0;
}
void fillBuffer(final InputStream inStream, final int requiredLen) throws IOException {
while (dataLen < requiredLen) {
if (off > 0) {
System.arraycopy(buffer, off, buffer, 0, dataLen);
off = 0;
}
final int bytesRead = inStream.read(buffer, off + dataLen, buffer.length - dataLen);
if (bytesRead == -1) {
if (dataLen > 0) {
throw new H2CorruptFrameException("Corrupt or incomplete HTTP2 frame");
}
throw new ConnectionClosedException();
}
dataLen += bytesRead;
this.metrics.incrementBytesTransferred(bytesRead);
}
}
public RawFrame read(final InputStream inStream) throws IOException {
fillBuffer(inStream, FrameConsts.HEAD_LEN);
final int payloadOff = FrameConsts.HEAD_LEN;
final int payloadLen = (buffer[off] & 0xff) << 16 | (buffer[off + 1] & 0xff) << 8 | (buffer[off + 2] & 0xff);
final int type = buffer[off + 3] & 0xff;
final int flags = buffer[off + 4] & 0xff;
final int streamId = Math.abs(buffer[off + 5] & 0xff) << 24 | (buffer[off + 6] & 0xff << 16) | (buffer[off + 7] & 0xff) << 8 | (buffer[off + 8] & 0xff);
if (payloadLen > maxFramePayloadSize) {
throw new H2ConnectionException(H2Error.FRAME_SIZE_ERROR, "Frame size exceeds maximum");
}
final int frameLen = payloadOff + payloadLen;
fillBuffer(inStream, frameLen);
if ((flags & FrameFlag.PADDED.getValue()) > 0) {
if (payloadLen == 0) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Inconsistent padding");
}
final int padding = buffer[off + FrameConsts.HEAD_LEN] & 0xff;
if (payloadLen < padding + 1) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Inconsistent padding");
}
}
final ByteBuffer payload = payloadLen > 0 ? ByteBuffer.wrap(buffer, off + payloadOff, payloadLen) : null;
final RawFrame frame = new RawFrame(type, flags, streamId, payload);
off += frameLen;
dataLen -= frameLen;
this.metrics.incrementFramesTransferred();
return frame;
}
public H2TransportMetrics getMetrics() {
return metrics;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/io/FrameOutputStream.java 0100664 0000000 0000000 00000006440 14114632321 027012 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.io;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.ByteBuffer;
import org.apache.hc.core5.util.Args;
/**
* @since 5.0
*/
abstract class FrameOutputStream extends OutputStream {
private final OutputStream outputStream;
private final byte[] cache;
private int cachePosition;
public FrameOutputStream(final int minChunkSize, final OutputStream outputStream) {
super();
this.outputStream = Args.notNull(outputStream, "Output stream");
this.cache = new byte[minChunkSize];
}
protected abstract void write(final ByteBuffer src, final boolean endStream, final OutputStream outputStream) throws IOException;
private void flushCache(final boolean endStream) throws IOException {
if (this.cachePosition > 0) {
write(ByteBuffer.wrap(this.cache, 0, this.cachePosition), endStream, this.outputStream);
this.cachePosition = 0;
}
}
@Override
public void write(final int b) throws IOException {
this.cache[this.cachePosition] = (byte) b;
this.cachePosition++;
if (this.cachePosition == this.cache.length) {
flushCache(false);
}
}
@Override
public void write(final byte[] b) throws IOException {
write(b, 0, b.length);
}
@Override
public void write(final byte[] src, final int off, final int len) throws IOException {
if (len >= this.cache.length - this.cachePosition) {
flushCache(false);
write(ByteBuffer.wrap(src, off, len), false, this.outputStream);
} else {
System.arraycopy(src, off, cache, this.cachePosition, len);
this.cachePosition += len;
}
}
@Override
public void flush() throws IOException {
flushCache(false);
this.outputStream.flush();
}
@Override
public void close() throws IOException {
if (this.cachePosition > 0) {
flushCache(true);
} else {
write(null, true, this.outputStream);
}
flushCache(true);
this.outputStream.flush();
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/ 0040755 0000000 0000000 00000000000 14157612023 022677 5 ustar 00 0000000 0000000 ././@LongLink 0100644 0000000 0000000 00000000146 14114632321 011631 L ustar 0000000 0000000 httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/H2OnlyServerHttpProtocolNegotiator.java httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/H2OnlyServerHttpProtocolNegotiator.jav0100664 0000000 0000000 00000007744 14114632321 032330 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import java.io.IOException;
import java.nio.ByteBuffer;
import org.apache.hc.core5.annotation.Internal;
import org.apache.hc.core5.concurrent.FutureCallback;
import org.apache.hc.core5.http.ConnectionClosedException;
import org.apache.hc.core5.http.impl.nio.BufferedData;
import org.apache.hc.core5.reactor.IOSession;
import org.apache.hc.core5.reactor.ProtocolIOSession;
import org.apache.hc.core5.util.Args;
/**
* I/O event handler for events fired by {@link ProtocolIOSession} that implements
* server side of the HTTP/2 protocol negotiation handshake.
*
* @since 5.1
*/
@Internal
public class H2OnlyServerHttpProtocolNegotiator extends ProtocolNegotiatorBase {
final static byte[] PREFACE = ClientHttpProtocolNegotiator.PREFACE;
private final ServerH2StreamMultiplexerFactory http2StreamHandlerFactory;
private final BufferedData inBuf;
public H2OnlyServerHttpProtocolNegotiator(
final ProtocolIOSession ioSession,
final ServerH2StreamMultiplexerFactory http2StreamHandlerFactory) {
this(ioSession, http2StreamHandlerFactory, null);
}
public H2OnlyServerHttpProtocolNegotiator(
final ProtocolIOSession ioSession,
final ServerH2StreamMultiplexerFactory http2StreamHandlerFactory,
final FutureCallback resultCallback) {
super(ioSession, resultCallback);
this.http2StreamHandlerFactory = Args.notNull(http2StreamHandlerFactory, "HTTP/2 stream handler factory");
this.inBuf = BufferedData.allocate(1024);
}
@Override
public void connected(final IOSession session) throws IOException {
}
@Override
public void inputReady(final IOSession session, final ByteBuffer src) throws IOException {
if (src != null) {
inBuf.put(src);
}
boolean endOfStream = false;
if (inBuf.length() < PREFACE.length) {
final int bytesRead = inBuf.readFrom(session);
if (bytesRead == -1) {
endOfStream = true;
}
}
final ByteBuffer data = inBuf.data();
if (data.remaining() >= PREFACE.length) {
for (int i = 0; i < PREFACE.length; i++) {
if (data.get() != PREFACE[i]) {
throw new ProtocolNegotiationException("Unexpected HTTP/2 preface");
}
}
startProtocol(new ServerH2IOEventHandler(http2StreamHandlerFactory.create(ioSession)), data.hasRemaining() ? data : null);
} else {
if (endOfStream) {
throw new ConnectionClosedException();
}
}
}
@Override
public void outputReady(final IOSession session) throws IOException {
}
@Override
public String toString() {
return getClass().getName();
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/AbstractH2IOEventHandler.java 0100644 0000000 0000000 00000010645 13543352600 030232 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import java.io.IOException;
import java.net.SocketAddress;
import java.nio.ByteBuffer;
import javax.net.ssl.SSLSession;
import org.apache.hc.core5.http.EndpointDetails;
import org.apache.hc.core5.http.HttpException;
import org.apache.hc.core5.http.ProtocolVersion;
import org.apache.hc.core5.http.impl.nio.HttpConnectionEventHandler;
import org.apache.hc.core5.io.CloseMode;
import org.apache.hc.core5.reactor.IOSession;
import org.apache.hc.core5.util.Args;
import org.apache.hc.core5.util.Timeout;
class AbstractH2IOEventHandler implements HttpConnectionEventHandler {
final AbstractH2StreamMultiplexer streamMultiplexer;
AbstractH2IOEventHandler(final AbstractH2StreamMultiplexer streamMultiplexer) {
this.streamMultiplexer = Args.notNull(streamMultiplexer, "Stream multiplexer");
}
@Override
public void connected(final IOSession session) throws IOException {
try {
streamMultiplexer.onConnect();
} catch (final HttpException ex) {
streamMultiplexer.onException(ex);
}
}
@Override
public void inputReady(final IOSession session, final ByteBuffer src) throws IOException {
try {
streamMultiplexer.onInput(src);
} catch (final HttpException ex) {
streamMultiplexer.onException(ex);
}
}
@Override
public void outputReady(final IOSession session) throws IOException {
try {
streamMultiplexer.onOutput();
} catch (final HttpException ex) {
streamMultiplexer.onException(ex);
}
}
@Override
public void timeout(final IOSession session, final Timeout timeout) throws IOException {
try {
streamMultiplexer.onTimeout(timeout);
} catch (final HttpException ex) {
streamMultiplexer.onException(ex);
}
}
@Override
public void exception(final IOSession session, final Exception cause) {
streamMultiplexer.onException(cause);
}
@Override
public void disconnected(final IOSession session) {
streamMultiplexer.onDisconnect();
}
@Override
public void close() throws IOException {
streamMultiplexer.close();
}
@Override
public void close(final CloseMode closeMode) {
streamMultiplexer.close(closeMode);
}
@Override
public boolean isOpen() {
return streamMultiplexer.isOpen();
}
@Override
public void setSocketTimeout(final Timeout timeout) {
streamMultiplexer.setSocketTimeout(timeout);
}
@Override
public SSLSession getSSLSession() {
return streamMultiplexer.getSSLSession();
}
@Override
public EndpointDetails getEndpointDetails() {
return streamMultiplexer.getEndpointDetails();
}
@Override
public Timeout getSocketTimeout() {
return streamMultiplexer.getSocketTimeout();
}
@Override
public ProtocolVersion getProtocolVersion() {
return streamMultiplexer.getProtocolVersion();
}
@Override
public SocketAddress getRemoteAddress() {
return streamMultiplexer.getRemoteAddress();
}
@Override
public SocketAddress getLocalAddress() {
return streamMultiplexer.getLocalAddress();
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/H2OnlyClientProtocolNegotiator.java 0100664 0000000 0000000 00000012650 14157612023 031575 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.util.concurrent.atomic.AtomicBoolean;
import org.apache.hc.core5.annotation.Internal;
import org.apache.hc.core5.concurrent.FutureCallback;
import org.apache.hc.core5.http2.ssl.ApplicationProtocol;
import org.apache.hc.core5.reactor.IOSession;
import org.apache.hc.core5.reactor.ProtocolIOSession;
import org.apache.hc.core5.reactor.ssl.TlsDetails;
import org.apache.hc.core5.util.Args;
import org.apache.hc.core5.util.TextUtils;
/**
* I/O event handler for events fired by {@link ProtocolIOSession} that implements
* client side of the HTTP/2 protocol negotiation handshake always forcing the choice
* of HTTP/2.
*
* @since 5.0
*/
@Internal
public class H2OnlyClientProtocolNegotiator extends ProtocolNegotiatorBase {
private final ClientH2StreamMultiplexerFactory http2StreamHandlerFactory;
private final boolean strictALPNHandshake;
private final AtomicBoolean initialized;
private volatile ByteBuffer preface;
public H2OnlyClientProtocolNegotiator(
final ProtocolIOSession ioSession,
final ClientH2StreamMultiplexerFactory http2StreamHandlerFactory,
final boolean strictALPNHandshake) {
this(ioSession, http2StreamHandlerFactory, strictALPNHandshake, null);
}
/**
* @since 5.1
*/
public H2OnlyClientProtocolNegotiator(
final ProtocolIOSession ioSession,
final ClientH2StreamMultiplexerFactory http2StreamHandlerFactory,
final boolean strictALPNHandshake,
final FutureCallback resultCallback) {
super(ioSession, resultCallback);
this.http2StreamHandlerFactory = Args.notNull(http2StreamHandlerFactory, "HTTP/2 stream handler factory");
this.strictALPNHandshake = strictALPNHandshake;
this.initialized = new AtomicBoolean();
}
private void initialize() throws IOException {
final TlsDetails tlsDetails = ioSession.getTlsDetails();
if (tlsDetails != null) {
final String applicationProtocol = tlsDetails.getApplicationProtocol();
if (TextUtils.isEmpty(applicationProtocol)) {
if (strictALPNHandshake) {
throw new ProtocolNegotiationException("ALPN: missing application protocol");
}
} else {
if (!ApplicationProtocol.HTTP_2.id.equals(applicationProtocol)) {
throw new ProtocolNegotiationException("ALPN: unexpected application protocol '" + applicationProtocol + "'");
}
}
}
this.preface = ByteBuffer.wrap(ClientHttpProtocolNegotiator.PREFACE);
ioSession.setEvent(SelectionKey.OP_WRITE);
}
private void writeOutPreface(final IOSession session) throws IOException {
if (preface.hasRemaining()) {
session.write(preface);
}
if (!preface.hasRemaining()) {
session.clearEvent(SelectionKey.OP_WRITE);
final ClientH2StreamMultiplexer streamMultiplexer = http2StreamHandlerFactory.create(ioSession);
startProtocol(new ClientH2IOEventHandler(streamMultiplexer), null);
preface = null;
}
}
@Override
public void connected(final IOSession session) throws IOException {
if (initialized.compareAndSet(false, true)) {
initialize();
}
}
@Override
public void outputReady(final IOSession session) throws IOException {
if (initialized.compareAndSet(false, true)) {
initialize();
}
if (preface != null) {
writeOutPreface(session);
} else {
throw new ProtocolNegotiationException("Unexpected output");
}
}
@Override
public void inputReady(final IOSession session, final ByteBuffer src) throws IOException {
if (src != null) {
throw new ProtocolNegotiationException("Unexpected input");
}
if (preface != null) {
writeOutPreface(session);
} else {
throw new ProtocolNegotiationException("Unexpected input");
}
}
@Override
public String toString() {
return getClass().getName();
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/ClientPushH2StreamHandler.java 0100644 0000000 0000000 00000017713 13602614500 030466 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.util.List;
import java.util.concurrent.atomic.AtomicBoolean;
import org.apache.hc.core5.http.EntityDetails;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.HttpException;
import org.apache.hc.core5.http.HttpRequest;
import org.apache.hc.core5.http.HttpResponse;
import org.apache.hc.core5.http.HttpVersion;
import org.apache.hc.core5.http.ProtocolException;
import org.apache.hc.core5.http.impl.BasicHttpConnectionMetrics;
import org.apache.hc.core5.http.impl.IncomingEntityDetails;
import org.apache.hc.core5.http.impl.nio.MessageState;
import org.apache.hc.core5.http.nio.AsyncPushConsumer;
import org.apache.hc.core5.http.nio.HandlerFactory;
import org.apache.hc.core5.http.protocol.HttpCoreContext;
import org.apache.hc.core5.http.protocol.HttpProcessor;
import org.apache.hc.core5.http2.H2ConnectionException;
import org.apache.hc.core5.http2.H2Error;
import org.apache.hc.core5.http2.H2StreamResetException;
import org.apache.hc.core5.http2.impl.DefaultH2RequestConverter;
import org.apache.hc.core5.http2.impl.DefaultH2ResponseConverter;
import org.apache.hc.core5.util.Asserts;
class ClientPushH2StreamHandler implements H2StreamHandler {
private final H2StreamChannel internalOutputChannel;
private final HttpProcessor httpProcessor;
private final BasicHttpConnectionMetrics connMetrics;
private final HandlerFactory pushHandlerFactory;
private final HttpCoreContext context;
private final AtomicBoolean failed;
private final AtomicBoolean done;
private volatile HttpRequest request;
private volatile AsyncPushConsumer exchangeHandler;
private volatile MessageState requestState;
private volatile MessageState responseState;
ClientPushH2StreamHandler(
final H2StreamChannel outputChannel,
final HttpProcessor httpProcessor,
final BasicHttpConnectionMetrics connMetrics,
final HandlerFactory pushHandlerFactory,
final HttpCoreContext context) {
this.internalOutputChannel = outputChannel;
this.httpProcessor = httpProcessor;
this.connMetrics = connMetrics;
this.pushHandlerFactory = pushHandlerFactory;
this.context = context;
this.failed = new AtomicBoolean(false);
this.done = new AtomicBoolean(false);
this.requestState = MessageState.HEADERS;
this.responseState = MessageState.HEADERS;
}
@Override
public HandlerFactory getPushHandlerFactory() {
return pushHandlerFactory;
}
@Override
public boolean isOutputReady() {
return false;
}
@Override
public void produceOutput() throws HttpException, IOException {
}
@Override
public void consumePromise(final List headers) throws HttpException, IOException {
if (requestState == MessageState.HEADERS) {
request = DefaultH2RequestConverter.INSTANCE.convert(headers);
final AsyncPushConsumer handler;
try {
handler = pushHandlerFactory != null ? pushHandlerFactory.create(request, context) : null;
} catch (final ProtocolException ex) {
throw new H2StreamResetException(H2Error.PROTOCOL_ERROR, ex.getMessage());
}
if (handler == null) {
throw new H2StreamResetException(H2Error.REFUSED_STREAM, "Stream refused");
}
exchangeHandler = handler;
context.setProtocolVersion(HttpVersion.HTTP_2);
context.setAttribute(HttpCoreContext.HTTP_REQUEST, request);
httpProcessor.process(request, null, context);
connMetrics.incrementRequestCount();
this.requestState = MessageState.COMPLETE;
} else {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Unexpected promise");
}
}
@Override
public void consumeHeader(final List headers, final boolean endStream) throws HttpException, IOException {
if (responseState == MessageState.HEADERS) {
Asserts.notNull(request, "Request");
Asserts.notNull(exchangeHandler, "Exchange handler");
final HttpResponse response = DefaultH2ResponseConverter.INSTANCE.convert(headers);
final EntityDetails entityDetails = endStream ? null : new IncomingEntityDetails(request, -1);
context.setAttribute(HttpCoreContext.HTTP_RESPONSE, response);
httpProcessor.process(response, entityDetails, context);
connMetrics.incrementResponseCount();
exchangeHandler.consumePromise(request, response, entityDetails, context);
if (endStream) {
responseState = MessageState.COMPLETE;
exchangeHandler.streamEnd(null);
} else {
responseState = MessageState.BODY;
}
} else {
throw new ProtocolException("Unexpected message headers");
}
}
@Override
public void updateInputCapacity() throws IOException {
Asserts.notNull(exchangeHandler, "Exchange handler");
exchangeHandler.updateCapacity(internalOutputChannel);
}
@Override
public void consumeData(final ByteBuffer src, final boolean endStream) throws HttpException, IOException {
if (responseState != MessageState.BODY) {
throw new ProtocolException("Unexpected message data");
}
Asserts.notNull(exchangeHandler, "Exchange handler");
if (src != null) {
exchangeHandler.consume(src);
}
if (endStream) {
responseState = MessageState.COMPLETE;
exchangeHandler.streamEnd(null);
}
}
public boolean isDone() {
return responseState == MessageState.COMPLETE;
}
@Override
public void failed(final Exception cause) {
try {
if (failed.compareAndSet(false, true)) {
if (exchangeHandler != null) {
exchangeHandler.failed(cause);
}
}
} finally {
releaseResources();
}
}
@Override
public void handle(final HttpException ex, final boolean endStream) throws HttpException {
throw ex;
}
@Override
public void releaseResources() {
if (done.compareAndSet(false, true)) {
responseState = MessageState.COMPLETE;
requestState = MessageState.COMPLETE;
if (exchangeHandler != null) {
exchangeHandler.releaseResources();
}
}
}
@Override
public String toString() {
return "[" +
"requestState=" + requestState +
", responseState=" + responseState +
']';
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/ClientHttpProtocolNegotiator.java 0100664 0000000 0000000 00000015525 14157612023 031405 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.util.concurrent.atomic.AtomicBoolean;
import org.apache.hc.core5.annotation.Internal;
import org.apache.hc.core5.concurrent.FutureCallback;
import org.apache.hc.core5.http.impl.nio.BufferedData;
import org.apache.hc.core5.http.impl.nio.ClientHttp1IOEventHandler;
import org.apache.hc.core5.http.impl.nio.ClientHttp1StreamDuplexerFactory;
import org.apache.hc.core5.http2.HttpVersionPolicy;
import org.apache.hc.core5.http2.ssl.ApplicationProtocol;
import org.apache.hc.core5.reactor.IOSession;
import org.apache.hc.core5.reactor.ProtocolIOSession;
import org.apache.hc.core5.reactor.ssl.TlsDetails;
import org.apache.hc.core5.util.Args;
/**
* I/O event handler for events fired by {@link ProtocolIOSession} that implements
* client side of the HTTP/2 protocol negotiation handshake
* based on {@link HttpVersionPolicy} configuration.
*
* @since 5.0
*/
@Internal
public class ClientHttpProtocolNegotiator extends ProtocolNegotiatorBase {
// PRI * HTTP/2.0\r\n\r\nSM\r\n\r\n
final static byte[] PREFACE = new byte[] {
0x50, 0x52, 0x49, 0x20, 0x2a, 0x20, 0x48, 0x54, 0x54, 0x50,
0x2f, 0x32, 0x2e, 0x30, 0x0d, 0x0a, 0x0d, 0x0a, 0x53, 0x4d,
0x0d, 0x0a, 0x0d, 0x0a};
private final ClientHttp1StreamDuplexerFactory http1StreamHandlerFactory;
private final ClientH2StreamMultiplexerFactory http2StreamHandlerFactory;
private final HttpVersionPolicy versionPolicy;
private final AtomicBoolean initialized;
private volatile ByteBuffer preface;
private volatile BufferedData inBuf;
public ClientHttpProtocolNegotiator(
final ProtocolIOSession ioSession,
final ClientHttp1StreamDuplexerFactory http1StreamHandlerFactory,
final ClientH2StreamMultiplexerFactory http2StreamHandlerFactory,
final HttpVersionPolicy versionPolicy) {
this(ioSession, http1StreamHandlerFactory, http2StreamHandlerFactory, versionPolicy, null);
}
/**
* @since 5.1
*/
public ClientHttpProtocolNegotiator(
final ProtocolIOSession ioSession,
final ClientHttp1StreamDuplexerFactory http1StreamHandlerFactory,
final ClientH2StreamMultiplexerFactory http2StreamHandlerFactory,
final HttpVersionPolicy versionPolicy,
final FutureCallback resultCallback) {
super(ioSession, resultCallback);
this.http1StreamHandlerFactory = Args.notNull(http1StreamHandlerFactory, "HTTP/1.1 stream handler factory");
this.http2StreamHandlerFactory = Args.notNull(http2StreamHandlerFactory, "HTTP/2 stream handler factory");
this.versionPolicy = versionPolicy != null ? versionPolicy : HttpVersionPolicy.NEGOTIATE;
this.initialized = new AtomicBoolean();
}
private void startHttp1() throws IOException {
final ByteBuffer data = inBuf != null ? inBuf.data() : null;
startProtocol(new ClientHttp1IOEventHandler(http1StreamHandlerFactory.create(ioSession)), data);
if (inBuf != null) {
inBuf.clear();
}
}
private void startHttp2() throws IOException {
final ByteBuffer data = inBuf != null ? inBuf.data() : null;
startProtocol(new ClientH2IOEventHandler(http2StreamHandlerFactory.create(ioSession)), data);
if (inBuf != null) {
inBuf.clear();
}
}
private void initialize() throws IOException {
switch (versionPolicy) {
case NEGOTIATE:
final TlsDetails tlsDetails = ioSession.getTlsDetails();
if (tlsDetails != null) {
if (ApplicationProtocol.HTTP_2.id.equals(tlsDetails.getApplicationProtocol())) {
// Proceed with the H2 preface
preface = ByteBuffer.wrap(PREFACE);
}
}
break;
case FORCE_HTTP_2:
preface = ByteBuffer.wrap(PREFACE);
break;
}
if (preface == null) {
startHttp1();
} else {
ioSession.setEvent(SelectionKey.OP_WRITE);
}
}
private void writeOutPreface(final IOSession session) throws IOException {
if (preface.hasRemaining()) {
session.write(preface);
}
if (!preface.hasRemaining()) {
session.clearEvent(SelectionKey.OP_WRITE);
startHttp2();
preface = null;
}
}
@Override
public void connected(final IOSession session) throws IOException {
if (initialized.compareAndSet(false, true)) {
initialize();
}
if (preface != null) {
writeOutPreface(session);
}
}
@Override
public void inputReady(final IOSession session, final ByteBuffer src) throws IOException {
if (src != null) {
if (inBuf == null) {
inBuf = BufferedData.allocate(src.remaining());
}
inBuf.put(src);
}
if (preface != null) {
writeOutPreface(session);
} else {
throw new ProtocolNegotiationException("Unexpected input");
}
}
@Override
public void outputReady(final IOSession session) throws IOException {
if (initialized.compareAndSet(false, true)) {
initialize();
}
if (preface != null) {
writeOutPreface(session);
} else {
throw new ProtocolNegotiationException("Unexpected output");
}
}
@Override
public String toString() {
return getClass().getName() + "/" + versionPolicy;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/AbstractH2StreamMultiplexer.java 0100664 0000000 0000000 00000212745 14157612023 031120 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import java.io.IOException;
import java.net.SocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.charset.CharacterCodingException;
import java.util.Deque;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Queue;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentLinkedDeque;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.atomic.AtomicInteger;
import javax.net.ssl.SSLSession;
import org.apache.hc.core5.concurrent.Cancellable;
import org.apache.hc.core5.concurrent.CancellableDependency;
import org.apache.hc.core5.http.ConnectionClosedException;
import org.apache.hc.core5.http.EndpointDetails;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.HttpConnection;
import org.apache.hc.core5.http.HttpException;
import org.apache.hc.core5.http.HttpStreamResetException;
import org.apache.hc.core5.http.HttpVersion;
import org.apache.hc.core5.http.ProtocolException;
import org.apache.hc.core5.http.ProtocolVersion;
import org.apache.hc.core5.http.RequestNotExecutedException;
import org.apache.hc.core5.http.config.CharCodingConfig;
import org.apache.hc.core5.http.impl.BasicEndpointDetails;
import org.apache.hc.core5.http.impl.BasicHttpConnectionMetrics;
import org.apache.hc.core5.http.impl.CharCodingSupport;
import org.apache.hc.core5.http.nio.AsyncPushConsumer;
import org.apache.hc.core5.http.nio.AsyncPushProducer;
import org.apache.hc.core5.http.nio.HandlerFactory;
import org.apache.hc.core5.http.nio.command.ExecutableCommand;
import org.apache.hc.core5.http.nio.command.ShutdownCommand;
import org.apache.hc.core5.http.protocol.HttpCoreContext;
import org.apache.hc.core5.http.protocol.HttpProcessor;
import org.apache.hc.core5.http2.H2ConnectionException;
import org.apache.hc.core5.http2.H2Error;
import org.apache.hc.core5.http2.H2StreamResetException;
import org.apache.hc.core5.http2.config.H2Config;
import org.apache.hc.core5.http2.config.H2Param;
import org.apache.hc.core5.http2.config.H2Setting;
import org.apache.hc.core5.http2.frame.FrameFactory;
import org.apache.hc.core5.http2.frame.FrameFlag;
import org.apache.hc.core5.http2.frame.FrameType;
import org.apache.hc.core5.http2.frame.RawFrame;
import org.apache.hc.core5.http2.frame.StreamIdGenerator;
import org.apache.hc.core5.http2.hpack.HPackDecoder;
import org.apache.hc.core5.http2.hpack.HPackEncoder;
import org.apache.hc.core5.http2.impl.BasicH2TransportMetrics;
import org.apache.hc.core5.http2.nio.AsyncPingHandler;
import org.apache.hc.core5.http2.nio.command.PingCommand;
import org.apache.hc.core5.io.CloseMode;
import org.apache.hc.core5.reactor.Command;
import org.apache.hc.core5.reactor.ProtocolIOSession;
import org.apache.hc.core5.reactor.ssl.TlsDetails;
import org.apache.hc.core5.util.Args;
import org.apache.hc.core5.util.ByteArrayBuffer;
import org.apache.hc.core5.util.Identifiable;
import org.apache.hc.core5.util.Timeout;
abstract class AbstractH2StreamMultiplexer implements Identifiable, HttpConnection {
private static final long LINGER_TIME = 1000; // 1 second
private static final long CONNECTION_WINDOW_LOW_MARK = 10 * 1024 * 1024; // 10 MiB
enum ConnectionHandshake { READY, ACTIVE, GRACEFUL_SHUTDOWN, SHUTDOWN}
enum SettingsHandshake { READY, TRANSMITTED, ACKED }
private final ProtocolIOSession ioSession;
private final FrameFactory frameFactory;
private final StreamIdGenerator idGenerator;
private final HttpProcessor httpProcessor;
private final H2Config localConfig;
private final BasicH2TransportMetrics inputMetrics;
private final BasicH2TransportMetrics outputMetrics;
private final BasicHttpConnectionMetrics connMetrics;
private final FrameInputBuffer inputBuffer;
private final FrameOutputBuffer outputBuffer;
private final Deque outputQueue;
private final HPackEncoder hPackEncoder;
private final HPackDecoder hPackDecoder;
private final Map streamMap;
private final Queue pingHandlers;
private final AtomicInteger connInputWindow;
private final AtomicInteger connOutputWindow;
private final AtomicInteger outputRequests;
private final AtomicInteger lastStreamId;
private final H2StreamListener streamListener;
private ConnectionHandshake connState = ConnectionHandshake.READY;
private SettingsHandshake localSettingState = SettingsHandshake.READY;
private SettingsHandshake remoteSettingState = SettingsHandshake.READY;
private int initInputWinSize;
private int initOutputWinSize;
private int lowMark;
private volatile H2Config remoteConfig;
private Continuation continuation;
private int processedRemoteStreamId;
private EndpointDetails endpointDetails;
private boolean goAwayReceived;
AbstractH2StreamMultiplexer(
final ProtocolIOSession ioSession,
final FrameFactory frameFactory,
final StreamIdGenerator idGenerator,
final HttpProcessor httpProcessor,
final CharCodingConfig charCodingConfig,
final H2Config h2Config,
final H2StreamListener streamListener) {
this.ioSession = Args.notNull(ioSession, "IO session");
this.frameFactory = Args.notNull(frameFactory, "Frame factory");
this.idGenerator = Args.notNull(idGenerator, "Stream id generator");
this.httpProcessor = Args.notNull(httpProcessor, "HTTP processor");
this.localConfig = h2Config != null ? h2Config : H2Config.DEFAULT;
this.inputMetrics = new BasicH2TransportMetrics();
this.outputMetrics = new BasicH2TransportMetrics();
this.connMetrics = new BasicHttpConnectionMetrics(this.inputMetrics, this.outputMetrics);
this.inputBuffer = new FrameInputBuffer(this.inputMetrics, this.localConfig.getMaxFrameSize());
this.outputBuffer = new FrameOutputBuffer(this.outputMetrics, this.localConfig.getMaxFrameSize());
this.outputQueue = new ConcurrentLinkedDeque<>();
this.pingHandlers = new ConcurrentLinkedQueue<>();
this.outputRequests = new AtomicInteger(0);
this.lastStreamId = new AtomicInteger(0);
this.hPackEncoder = new HPackEncoder(CharCodingSupport.createEncoder(charCodingConfig));
this.hPackDecoder = new HPackDecoder(CharCodingSupport.createDecoder(charCodingConfig));
this.streamMap = new ConcurrentHashMap<>();
this.remoteConfig = H2Config.INIT;
this.connInputWindow = new AtomicInteger(H2Config.INIT.getInitialWindowSize());
this.connOutputWindow = new AtomicInteger(H2Config.INIT.getInitialWindowSize());
this.initInputWinSize = H2Config.INIT.getInitialWindowSize();
this.initOutputWinSize = H2Config.INIT.getInitialWindowSize();
this.hPackDecoder.setMaxTableSize(H2Config.INIT.getHeaderTableSize());
this.hPackEncoder.setMaxTableSize(H2Config.INIT.getHeaderTableSize());
this.hPackDecoder.setMaxListSize(H2Config.INIT.getMaxHeaderListSize());
this.lowMark = H2Config.INIT.getInitialWindowSize() / 2;
this.streamListener = streamListener;
}
@Override
public String getId() {
return ioSession.getId();
}
abstract void acceptHeaderFrame() throws H2ConnectionException;
abstract void acceptPushRequest() throws H2ConnectionException;
abstract void acceptPushFrame() throws H2ConnectionException;
abstract H2StreamHandler createRemotelyInitiatedStream(
H2StreamChannel channel,
HttpProcessor httpProcessor,
BasicHttpConnectionMetrics connMetrics,
HandlerFactory pushHandlerFactory) throws IOException;
abstract H2StreamHandler createLocallyInitiatedStream(
ExecutableCommand command,
H2StreamChannel channel,
HttpProcessor httpProcessor,
BasicHttpConnectionMetrics connMetrics) throws IOException;
private int updateWindow(final AtomicInteger window, final int delta) throws ArithmeticException {
for (;;) {
final int current = window.get();
long newValue = (long) current + delta;
//TODO: work-around for what looks like a bug in Ngnix (1.11)
// Tolerate if the update window exceeded by one
if (newValue == 0x80000000L) {
newValue = Integer.MAX_VALUE;
}
//TODO: needs to be removed
if (Math.abs(newValue) > 0x7fffffffL) {
throw new ArithmeticException("Update causes flow control window to exceed " + Integer.MAX_VALUE);
}
if (window.compareAndSet(current, (int) newValue)) {
return (int) newValue;
}
}
}
private int updateInputWindow(
final int streamId, final AtomicInteger window, final int delta) throws ArithmeticException {
final int newSize = updateWindow(window, delta);
if (streamListener != null) {
streamListener.onInputFlowControl(this, streamId, delta, newSize);
}
return newSize;
}
private int updateOutputWindow(
final int streamId, final AtomicInteger window, final int delta) throws ArithmeticException {
final int newSize = updateWindow(window, delta);
if (streamListener != null) {
streamListener.onOutputFlowControl(this, streamId, delta, newSize);
}
return newSize;
}
private void commitFrameInternal(final RawFrame frame) throws IOException {
if (outputBuffer.isEmpty() && outputQueue.isEmpty()) {
if (streamListener != null) {
streamListener.onFrameOutput(this, frame.getStreamId(), frame);
}
outputBuffer.write(frame, ioSession);
} else {
outputQueue.addLast(frame);
}
ioSession.setEvent(SelectionKey.OP_WRITE);
}
private void commitFrame(final RawFrame frame) throws IOException {
Args.notNull(frame, "Frame");
ioSession.getLock().lock();
try {
commitFrameInternal(frame);
} finally {
ioSession.getLock().unlock();
}
}
private void commitHeaders(
final int streamId, final List headers, final boolean endStream) throws IOException {
if (streamListener != null) {
streamListener.onHeaderOutput(this, streamId, headers);
}
final ByteArrayBuffer buf = new ByteArrayBuffer(512);
hPackEncoder.encodeHeaders(buf, headers, localConfig.isCompressionEnabled());
int off = 0;
int remaining = buf.length();
boolean continuation = false;
while (remaining > 0) {
final int chunk = Math.min(remoteConfig.getMaxFrameSize(), remaining);
final ByteBuffer payload = ByteBuffer.wrap(buf.array(), off, chunk);
remaining -= chunk;
off += chunk;
final boolean endHeaders = remaining == 0;
final RawFrame frame;
if (!continuation) {
frame = frameFactory.createHeaders(streamId, payload, endHeaders, endStream);
continuation = true;
} else {
frame = frameFactory.createContinuation(streamId, payload, endHeaders);
}
commitFrameInternal(frame);
}
}
private void commitPushPromise(
final int streamId, final int promisedStreamId, final List headers) throws IOException {
if (headers == null || headers.isEmpty()) {
throw new H2ConnectionException(H2Error.INTERNAL_ERROR, "Message headers are missing");
}
if (streamListener != null) {
streamListener.onHeaderOutput(this, streamId, headers);
}
final ByteArrayBuffer buf = new ByteArrayBuffer(512);
buf.append((byte)(promisedStreamId >> 24));
buf.append((byte)(promisedStreamId >> 16));
buf.append((byte)(promisedStreamId >> 8));
buf.append((byte)(promisedStreamId));
hPackEncoder.encodeHeaders(buf, headers, localConfig.isCompressionEnabled());
int off = 0;
int remaining = buf.length();
boolean continuation = false;
while (remaining > 0) {
final int chunk = Math.min(remoteConfig.getMaxFrameSize(), remaining);
final ByteBuffer payload = ByteBuffer.wrap(buf.array(), off, chunk);
remaining -= chunk;
off += chunk;
final boolean endHeaders = remaining == 0;
final RawFrame frame;
if (!continuation) {
frame = frameFactory.createPushPromise(streamId, payload, endHeaders);
continuation = true;
} else {
frame = frameFactory.createContinuation(streamId, payload, endHeaders);
}
commitFrameInternal(frame);
}
}
private void streamDataFrame(
final int streamId,
final AtomicInteger streamOutputWindow,
final ByteBuffer payload,
final int chunk) throws IOException {
final RawFrame dataFrame = frameFactory.createData(streamId, payload, false);
if (streamListener != null) {
streamListener.onFrameOutput(this, streamId, dataFrame);
}
updateOutputWindow(0, connOutputWindow, -chunk);
updateOutputWindow(streamId, streamOutputWindow, -chunk);
outputBuffer.write(dataFrame, ioSession);
}
private int streamData(
final int streamId, final AtomicInteger streamOutputWindow, final ByteBuffer payload) throws IOException {
if (outputBuffer.isEmpty() && outputQueue.isEmpty()) {
final int capacity = Math.min(connOutputWindow.get(), streamOutputWindow.get());
if (capacity <= 0) {
return 0;
}
final int maxPayloadSize = Math.min(capacity, remoteConfig.getMaxFrameSize());
final int chunk;
if (payload.remaining() <= maxPayloadSize) {
chunk = payload.remaining();
streamDataFrame(streamId, streamOutputWindow, payload, chunk);
} else {
chunk = maxPayloadSize;
final int originalLimit = payload.limit();
try {
payload.limit(payload.position() + chunk);
streamDataFrame(streamId, streamOutputWindow, payload, chunk);
} finally {
payload.limit(originalLimit);
}
}
payload.position(payload.position() + chunk);
ioSession.setEvent(SelectionKey.OP_WRITE);
return chunk;
}
return 0;
}
private void incrementInputCapacity(
final int streamId, final AtomicInteger inputWindow, final int inputCapacity) throws IOException {
if (inputCapacity > 0) {
final int streamWinSize = inputWindow.get();
final int remainingCapacity = Integer.MAX_VALUE - streamWinSize;
final int chunk = Math.min(inputCapacity, remainingCapacity);
if (chunk != 0) {
final RawFrame windowUpdateFrame = frameFactory.createWindowUpdate(streamId, chunk);
commitFrame(windowUpdateFrame);
updateInputWindow(streamId, inputWindow, chunk);
}
}
}
private void requestSessionOutput() {
outputRequests.incrementAndGet();
ioSession.setEvent(SelectionKey.OP_WRITE);
}
private void updateLastStreamId(final int streamId) {
final int currentId = lastStreamId.get();
if (streamId > currentId) {
lastStreamId.compareAndSet(currentId, streamId);
}
}
private int generateStreamId() {
for (;;) {
final int currentId = lastStreamId.get();
final int newStreamId = idGenerator.generate(currentId);
if (lastStreamId.compareAndSet(currentId, newStreamId)) {
return newStreamId;
}
}
}
public final void onConnect() throws HttpException, IOException {
connState = ConnectionHandshake.ACTIVE;
final RawFrame settingsFrame = frameFactory.createSettings(
new H2Setting(H2Param.HEADER_TABLE_SIZE, localConfig.getHeaderTableSize()),
new H2Setting(H2Param.ENABLE_PUSH, localConfig.isPushEnabled() ? 1 : 0),
new H2Setting(H2Param.MAX_CONCURRENT_STREAMS, localConfig.getMaxConcurrentStreams()),
new H2Setting(H2Param.INITIAL_WINDOW_SIZE, localConfig.getInitialWindowSize()),
new H2Setting(H2Param.MAX_FRAME_SIZE, localConfig.getMaxFrameSize()),
new H2Setting(H2Param.MAX_HEADER_LIST_SIZE, localConfig.getMaxHeaderListSize()));
commitFrame(settingsFrame);
localSettingState = SettingsHandshake.TRANSMITTED;
maximizeConnWindow(connInputWindow.get());
if (streamListener != null) {
final int initInputWindow = connInputWindow.get();
streamListener.onInputFlowControl(this, 0, initInputWindow, initInputWindow);
final int initOutputWindow = connOutputWindow.get();
streamListener.onOutputFlowControl(this, 0, initOutputWindow, initOutputWindow);
}
}
public final void onInput(final ByteBuffer src) throws HttpException, IOException {
if (connState == ConnectionHandshake.SHUTDOWN) {
ioSession.clearEvent(SelectionKey.OP_READ);
} else {
if (src != null) {
inputBuffer.put(src);
}
RawFrame frame;
while ((frame = inputBuffer.read(ioSession)) != null) {
if (streamListener != null) {
streamListener.onFrameInput(this, frame.getStreamId(), frame);
}
consumeFrame(frame);
}
}
}
public final void onOutput() throws HttpException, IOException {
ioSession.getLock().lock();
try {
if (!outputBuffer.isEmpty()) {
outputBuffer.flush(ioSession);
}
while (outputBuffer.isEmpty()) {
final RawFrame frame = outputQueue.poll();
if (frame != null) {
if (streamListener != null) {
streamListener.onFrameOutput(this, frame.getStreamId(), frame);
}
outputBuffer.write(frame, ioSession);
} else {
break;
}
}
} finally {
ioSession.getLock().unlock();
}
if (connState.compareTo(ConnectionHandshake.SHUTDOWN) < 0) {
if (connOutputWindow.get() > 0 && remoteSettingState == SettingsHandshake.ACKED) {
produceOutput();
}
final int pendingOutputRequests = outputRequests.get();
boolean outputPending = false;
if (!streamMap.isEmpty() && connOutputWindow.get() > 0) {
for (final Iterator> it = streamMap.entrySet().iterator(); it.hasNext(); ) {
final Map.Entry entry = it.next();
final H2Stream stream = entry.getValue();
if (!stream.isLocalClosed()
&& stream.getOutputWindow().get() > 0
&& stream.isOutputReady()) {
outputPending = true;
break;
}
}
}
ioSession.getLock().lock();
try {
if (!outputPending && outputBuffer.isEmpty() && outputQueue.isEmpty()
&& outputRequests.compareAndSet(pendingOutputRequests, 0)) {
ioSession.clearEvent(SelectionKey.OP_WRITE);
} else {
outputRequests.addAndGet(-pendingOutputRequests);
}
} finally {
ioSession.getLock().unlock();
}
}
if (connState.compareTo(ConnectionHandshake.ACTIVE) <= 0 && remoteSettingState == SettingsHandshake.ACKED) {
processPendingCommands();
}
if (connState.compareTo(ConnectionHandshake.GRACEFUL_SHUTDOWN) == 0) {
int liveStreams = 0;
for (final Iterator> it = streamMap.entrySet().iterator(); it.hasNext(); ) {
final Map.Entry entry = it.next();
final H2Stream stream = entry.getValue();
if (stream.isLocalClosed() && stream.isRemoteClosed()) {
stream.releaseResources();
it.remove();
} else {
if (idGenerator.isSameSide(stream.getId()) || stream.getId() <= processedRemoteStreamId) {
liveStreams++;
}
}
}
if (liveStreams == 0) {
connState = ConnectionHandshake.SHUTDOWN;
}
}
if (connState.compareTo(ConnectionHandshake.SHUTDOWN) >= 0) {
if (!streamMap.isEmpty()) {
for (final H2Stream stream : streamMap.values()) {
stream.releaseResources();
}
streamMap.clear();
}
ioSession.getLock().lock();
try {
if (outputBuffer.isEmpty() && outputQueue.isEmpty()) {
ioSession.close();
}
} finally {
ioSession.getLock().unlock();
}
}
}
public final void onTimeout(final Timeout timeout) throws HttpException, IOException {
connState = ConnectionHandshake.SHUTDOWN;
final RawFrame goAway;
if (localSettingState != SettingsHandshake.ACKED) {
goAway = frameFactory.createGoAway(processedRemoteStreamId, H2Error.SETTINGS_TIMEOUT,
"Setting timeout (" + timeout + ")");
} else {
goAway = frameFactory.createGoAway(processedRemoteStreamId, H2Error.NO_ERROR,
"Timeout due to inactivity (" + timeout + ")");
}
commitFrame(goAway);
for (final Iterator> it = streamMap.entrySet().iterator(); it.hasNext(); ) {
final Map.Entry entry = it.next();
final H2Stream stream = entry.getValue();
stream.reset(new H2StreamResetException(H2Error.NO_ERROR, "Timeout due to inactivity (" + timeout + ")"));
}
streamMap.clear();
}
public final void onDisconnect() {
for (;;) {
final AsyncPingHandler pingHandler = pingHandlers.poll();
if (pingHandler != null) {
pingHandler.cancel();
} else {
break;
}
}
for (final Iterator> it = streamMap.entrySet().iterator(); it.hasNext(); ) {
final Map.Entry entry = it.next();
final H2Stream stream = entry.getValue();
stream.cancel();
}
for (;;) {
final Command command = ioSession.poll();
if (command != null) {
if (command instanceof ExecutableCommand) {
((ExecutableCommand) command).failed(new ConnectionClosedException());
} else {
command.cancel();
}
} else {
break;
}
}
}
private void processPendingCommands() throws IOException, HttpException {
while (streamMap.size() < remoteConfig.getMaxConcurrentStreams()) {
final Command command = ioSession.poll();
if (command == null) {
break;
}
if (command instanceof ShutdownCommand) {
final ShutdownCommand shutdownCommand = (ShutdownCommand) command;
if (shutdownCommand.getType() == CloseMode.IMMEDIATE) {
for (final Iterator> it = streamMap.entrySet().iterator(); it.hasNext(); ) {
final Map.Entry entry = it.next();
final H2Stream stream = entry.getValue();
stream.cancel();
}
streamMap.clear();
connState = ConnectionHandshake.SHUTDOWN;
} else {
if (connState.compareTo(ConnectionHandshake.ACTIVE) <= 0) {
final RawFrame goAway = frameFactory.createGoAway(processedRemoteStreamId, H2Error.NO_ERROR, "Graceful shutdown");
commitFrame(goAway);
connState = streamMap.isEmpty() ? ConnectionHandshake.SHUTDOWN : ConnectionHandshake.GRACEFUL_SHUTDOWN;
}
}
break;
} else if (command instanceof PingCommand) {
final PingCommand pingCommand = (PingCommand) command;
final AsyncPingHandler handler = pingCommand.getHandler();
pingHandlers.add(handler);
final RawFrame ping = frameFactory.createPing(handler.getData());
commitFrame(ping);
} else if (command instanceof ExecutableCommand) {
final int streamId = generateStreamId();
final H2StreamChannelImpl channel = new H2StreamChannelImpl(
streamId, true, initInputWinSize, initOutputWinSize);
final ExecutableCommand executableCommand = (ExecutableCommand) command;
final H2StreamHandler streamHandler = createLocallyInitiatedStream(
executableCommand, channel, httpProcessor, connMetrics);
final H2Stream stream = new H2Stream(channel, streamHandler, false);
streamMap.put(streamId, stream);
if (streamListener != null) {
final int initInputWindow = stream.getInputWindow().get();
streamListener.onInputFlowControl(this, streamId, initInputWindow, initInputWindow);
final int initOutputWindow = stream.getOutputWindow().get();
streamListener.onOutputFlowControl(this, streamId, initOutputWindow, initOutputWindow);
}
if (stream.isOutputReady()) {
stream.produceOutput();
}
final CancellableDependency cancellableDependency = executableCommand.getCancellableDependency();
if (cancellableDependency != null) {
cancellableDependency.setDependency(new Cancellable() {
@Override
public boolean cancel() {
return stream.abort();
}
});
}
if (!outputQueue.isEmpty()) {
return;
}
}
}
}
public final void onException(final Exception cause) {
try {
for (;;) {
final AsyncPingHandler pingHandler = pingHandlers.poll();
if (pingHandler != null) {
pingHandler.failed(cause);
} else {
break;
}
}
for (;;) {
final Command command = ioSession.poll();
if (command != null) {
if (command instanceof ExecutableCommand) {
((ExecutableCommand) command).failed(new ConnectionClosedException());
} else {
command.cancel();
}
} else {
break;
}
}
for (final Iterator> it = streamMap.entrySet().iterator(); it.hasNext(); ) {
final Map.Entry entry = it.next();
final H2Stream stream = entry.getValue();
stream.reset(cause);
}
streamMap.clear();
if (!(cause instanceof ConnectionClosedException)) {
if (connState.compareTo(ConnectionHandshake.GRACEFUL_SHUTDOWN) <= 0) {
final H2Error errorCode;
if (cause instanceof H2ConnectionException) {
errorCode = H2Error.getByCode(((H2ConnectionException) cause).getCode());
} else if (cause instanceof ProtocolException){
errorCode = H2Error.PROTOCOL_ERROR;
} else {
errorCode = H2Error.INTERNAL_ERROR;
}
final RawFrame goAway = frameFactory.createGoAway(processedRemoteStreamId, errorCode, cause.getMessage());
commitFrame(goAway);
}
}
} catch (final IOException ignore) {
} finally {
connState = ConnectionHandshake.SHUTDOWN;
final CloseMode closeMode;
if (cause instanceof ConnectionClosedException) {
closeMode = CloseMode.GRACEFUL;
} else if (cause instanceof IOException) {
closeMode = CloseMode.IMMEDIATE;
} else {
closeMode = CloseMode.GRACEFUL;
}
ioSession.close(closeMode);
}
}
private H2Stream getValidStream(final int streamId) throws H2ConnectionException {
if (streamId == 0) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Illegal stream id: " + streamId);
}
final H2Stream stream = streamMap.get(streamId);
if (stream == null) {
if (streamId <= lastStreamId.get()) {
throw new H2ConnectionException(H2Error.STREAM_CLOSED, "Stream closed");
} else {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Unexpected stream id: " + streamId);
}
}
return stream;
}
private void consumeFrame(final RawFrame frame) throws HttpException, IOException {
final FrameType frameType = FrameType.valueOf(frame.getType());
final int streamId = frame.getStreamId();
if (continuation != null && frameType != FrameType.CONTINUATION) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "CONTINUATION frame expected");
}
switch (frameType) {
case DATA: {
final H2Stream stream = getValidStream(streamId);
try {
consumeDataFrame(frame, stream);
} catch (final H2StreamResetException ex) {
stream.localReset(ex);
} catch (final HttpStreamResetException ex) {
stream.localReset(ex, ex.getCause() != null ? H2Error.INTERNAL_ERROR : H2Error.CANCEL);
}
if (stream.isTerminated()) {
streamMap.remove(streamId);
stream.releaseResources();
requestSessionOutput();
}
}
break;
case HEADERS: {
if (streamId == 0) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Illegal stream id: " + streamId);
}
H2Stream stream = streamMap.get(streamId);
if (stream == null) {
acceptHeaderFrame();
if (idGenerator.isSameSide(streamId)) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Illegal stream id: " + streamId);
}
if (goAwayReceived ) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "GOAWAY received");
}
updateLastStreamId(streamId);
final H2StreamChannelImpl channel = new H2StreamChannelImpl(
streamId, false, initInputWinSize, initOutputWinSize);
final H2StreamHandler streamHandler;
if (connState.compareTo(ConnectionHandshake.ACTIVE) <= 0) {
streamHandler = createRemotelyInitiatedStream(channel, httpProcessor, connMetrics, null);
} else {
streamHandler = NoopH2StreamHandler.INSTANCE;
channel.setLocalEndStream();
}
stream = new H2Stream(channel, streamHandler, true);
if (stream.isOutputReady()) {
stream.produceOutput();
}
streamMap.put(streamId, stream);
}
try {
consumeHeaderFrame(frame, stream);
if (stream.isOutputReady()) {
stream.produceOutput();
}
} catch (final H2StreamResetException ex) {
stream.localReset(ex);
} catch (final HttpStreamResetException ex) {
stream.localReset(ex, ex.getCause() != null ? H2Error.INTERNAL_ERROR : H2Error.CANCEL);
} catch (final HttpException ex) {
stream.handle(ex);
}
if (stream.isTerminated()) {
streamMap.remove(streamId);
stream.releaseResources();
requestSessionOutput();
}
}
break;
case CONTINUATION: {
if (continuation == null) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Unexpected CONTINUATION frame");
}
if (streamId != continuation.streamId) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Unexpected CONTINUATION stream id: " + streamId);
}
final H2Stream stream = getValidStream(streamId);
try {
consumeContinuationFrame(frame, stream);
} catch (final H2StreamResetException ex) {
stream.localReset(ex);
} catch (final HttpStreamResetException ex) {
stream.localReset(ex, ex.getCause() != null ? H2Error.INTERNAL_ERROR : H2Error.CANCEL);
}
if (stream.isTerminated()) {
streamMap.remove(streamId);
stream.releaseResources();
requestSessionOutput();
}
}
break;
case WINDOW_UPDATE: {
final ByteBuffer payload = frame.getPayload();
if (payload == null || payload.remaining() != 4) {
throw new H2ConnectionException(H2Error.FRAME_SIZE_ERROR, "Invalid WINDOW_UPDATE frame payload");
}
final int delta = payload.getInt();
if (delta <= 0) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Invalid WINDOW_UPDATE delta");
}
if (streamId == 0) {
try {
updateOutputWindow(0, connOutputWindow, delta);
} catch (final ArithmeticException ex) {
throw new H2ConnectionException(H2Error.FLOW_CONTROL_ERROR, ex.getMessage());
}
} else {
final H2Stream stream = streamMap.get(streamId);
if (stream != null) {
try {
updateOutputWindow(streamId, stream.getOutputWindow(), delta);
} catch (final ArithmeticException ex) {
throw new H2ConnectionException(H2Error.FLOW_CONTROL_ERROR, ex.getMessage());
}
}
}
ioSession.setEvent(SelectionKey.OP_WRITE);
}
break;
case RST_STREAM: {
if (streamId == 0) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Illegal stream id: " + streamId);
}
final H2Stream stream = streamMap.get(streamId);
if (stream == null) {
if (streamId > lastStreamId.get()) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Unexpected stream id: " + streamId);
}
} else {
final ByteBuffer payload = frame.getPayload();
if (payload == null || payload.remaining() != 4) {
throw new H2ConnectionException(H2Error.FRAME_SIZE_ERROR, "Invalid RST_STREAM frame payload");
}
final int errorCode = payload.getInt();
stream.reset(new H2StreamResetException(errorCode, "Stream reset (" + errorCode + ")"));
streamMap.remove(streamId);
stream.releaseResources();
requestSessionOutput();
}
}
break;
case PING: {
if (streamId != 0) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Illegal stream id");
}
final ByteBuffer ping = frame.getPayloadContent();
if (ping == null || ping.remaining() != 8) {
throw new H2ConnectionException(H2Error.FRAME_SIZE_ERROR, "Invalid PING frame payload");
}
if (frame.isFlagSet(FrameFlag.ACK)) {
final AsyncPingHandler pingHandler = pingHandlers.poll();
if (pingHandler != null) {
pingHandler.consumeResponse(ping);
}
} else {
final ByteBuffer pong = ByteBuffer.allocate(ping.remaining());
pong.put(ping);
pong.flip();
final RawFrame response = frameFactory.createPingAck(pong);
commitFrame(response);
}
}
break;
case SETTINGS: {
if (streamId != 0) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Illegal stream id");
}
if (frame.isFlagSet(FrameFlag.ACK)) {
if (localSettingState == SettingsHandshake.TRANSMITTED) {
localSettingState = SettingsHandshake.ACKED;
ioSession.setEvent(SelectionKey.OP_WRITE);
applyLocalSettings();
}
} else {
final ByteBuffer payload = frame.getPayload();
if (payload != null) {
if ((payload.remaining() % 6) != 0) {
throw new H2ConnectionException(H2Error.FRAME_SIZE_ERROR, "Invalid SETTINGS payload");
}
consumeSettingsFrame(payload);
remoteSettingState = SettingsHandshake.TRANSMITTED;
}
// Send ACK
final RawFrame response = frameFactory.createSettingsAck();
commitFrame(response);
remoteSettingState = SettingsHandshake.ACKED;
}
}
break;
case PRIORITY:
// Stream priority not supported
break;
case PUSH_PROMISE: {
acceptPushFrame();
if (goAwayReceived ) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "GOAWAY received");
}
if (!localConfig.isPushEnabled()) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Push is disabled");
}
final H2Stream stream = getValidStream(streamId);
if (stream.isRemoteClosed()) {
stream.localReset(new H2StreamResetException(H2Error.STREAM_CLOSED, "Stream closed"));
break;
}
final ByteBuffer payload = frame.getPayloadContent();
if (payload == null || payload.remaining() < 4) {
throw new H2ConnectionException(H2Error.FRAME_SIZE_ERROR, "Invalid PUSH_PROMISE payload");
}
final int promisedStreamId = payload.getInt();
if (promisedStreamId == 0 || idGenerator.isSameSide(promisedStreamId)) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Illegal promised stream id: " + promisedStreamId);
}
if (streamMap.get(promisedStreamId) != null) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Unexpected promised stream id: " + promisedStreamId);
}
updateLastStreamId(promisedStreamId);
final H2StreamChannelImpl channel = new H2StreamChannelImpl(
promisedStreamId, false, initInputWinSize, initOutputWinSize);
final H2StreamHandler streamHandler;
if (connState.compareTo(ConnectionHandshake.ACTIVE) <= 0) {
streamHandler = createRemotelyInitiatedStream(channel, httpProcessor, connMetrics,
stream.getPushHandlerFactory());
} else {
streamHandler = NoopH2StreamHandler.INSTANCE;
channel.setLocalEndStream();
}
final H2Stream promisedStream = new H2Stream(channel, streamHandler, true);
streamMap.put(promisedStreamId, promisedStream);
try {
consumePushPromiseFrame(frame, payload, promisedStream);
} catch (final H2StreamResetException ex) {
promisedStream.localReset(ex);
} catch (final HttpStreamResetException ex) {
promisedStream.localReset(ex, ex.getCause() != null ? H2Error.INTERNAL_ERROR : H2Error.NO_ERROR);
}
}
break;
case GOAWAY: {
if (streamId != 0) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Illegal stream id");
}
final ByteBuffer payload = frame.getPayload();
if (payload == null || payload.remaining() < 8) {
throw new H2ConnectionException(H2Error.FRAME_SIZE_ERROR, "Invalid GOAWAY payload");
}
final int processedLocalStreamId = payload.getInt();
final int errorCode = payload.getInt();
goAwayReceived = true;
if (errorCode == H2Error.NO_ERROR.getCode()) {
if (connState.compareTo(ConnectionHandshake.ACTIVE) <= 0) {
for (final Iterator> it = streamMap.entrySet().iterator(); it.hasNext(); ) {
final Map.Entry entry = it.next();
final int activeStreamId = entry.getKey();
if (!idGenerator.isSameSide(activeStreamId) && activeStreamId > processedLocalStreamId) {
final H2Stream stream = entry.getValue();
stream.cancel();
it.remove();
}
}
}
connState = streamMap.isEmpty() ? ConnectionHandshake.SHUTDOWN : ConnectionHandshake.GRACEFUL_SHUTDOWN;
} else {
for (final Iterator> it = streamMap.entrySet().iterator(); it.hasNext(); ) {
final Map.Entry entry = it.next();
final H2Stream stream = entry.getValue();
stream.reset(new H2StreamResetException(errorCode, "Connection terminated by the peer (" + errorCode + ")"));
}
streamMap.clear();
connState = ConnectionHandshake.SHUTDOWN;
}
}
ioSession.setEvent(SelectionKey.OP_WRITE);
break;
}
}
private void consumeDataFrame(final RawFrame frame, final H2Stream stream) throws HttpException, IOException {
final int streamId = stream.getId();
final ByteBuffer payload = frame.getPayloadContent();
if (payload != null) {
final int frameLength = frame.getLength();
final int streamWinSize = updateInputWindow(streamId, stream.getInputWindow(), -frameLength);
if (streamWinSize < lowMark && !stream.isRemoteClosed()) {
stream.produceInputCapacityUpdate();
}
final int connWinSize = updateInputWindow(0, connInputWindow, -frameLength);
if (connWinSize < CONNECTION_WINDOW_LOW_MARK) {
maximizeConnWindow(connWinSize);
}
}
if (stream.isRemoteClosed()) {
throw new H2StreamResetException(H2Error.STREAM_CLOSED, "Stream already closed");
}
if (frame.isFlagSet(FrameFlag.END_STREAM)) {
stream.setRemoteEndStream();
}
if (stream.isLocalReset()) {
return;
}
stream.consumeData(payload);
}
private void maximizeConnWindow(final int connWinSize) throws IOException {
final int delta = Integer.MAX_VALUE - connWinSize;
if (delta > 0) {
final RawFrame windowUpdateFrame = frameFactory.createWindowUpdate(0, delta);
commitFrame(windowUpdateFrame);
updateInputWindow(0, connInputWindow, delta);
}
}
private void consumePushPromiseFrame(final RawFrame frame, final ByteBuffer payload, final H2Stream promisedStream) throws HttpException, IOException {
final int promisedStreamId = promisedStream.getId();
if (!frame.isFlagSet(FrameFlag.END_HEADERS)) {
continuation = new Continuation(promisedStreamId, frame.getType(), true);
}
if (continuation == null) {
final List headers = hPackDecoder.decodeHeaders(payload);
if (promisedStreamId > processedRemoteStreamId) {
processedRemoteStreamId = promisedStreamId;
}
if (streamListener != null) {
streamListener.onHeaderInput(this, promisedStreamId, headers);
}
promisedStream.consumePromise(headers);
} else {
continuation.copyPayload(payload);
}
}
List decodeHeaders(final ByteBuffer payload) throws HttpException {
return hPackDecoder.decodeHeaders(payload);
}
private void consumeHeaderFrame(final RawFrame frame, final H2Stream stream) throws HttpException, IOException {
final int streamId = stream.getId();
if (!frame.isFlagSet(FrameFlag.END_HEADERS)) {
continuation = new Continuation(streamId, frame.getType(), frame.isFlagSet(FrameFlag.END_STREAM));
}
final ByteBuffer payload = frame.getPayloadContent();
if (frame.isFlagSet(FrameFlag.PRIORITY)) {
// Priority not supported
payload.getInt();
payload.get();
}
if (continuation == null) {
final List headers = decodeHeaders(payload);
if (stream.isRemoteInitiated() && streamId > processedRemoteStreamId) {
processedRemoteStreamId = streamId;
}
if (streamListener != null) {
streamListener.onHeaderInput(this, streamId, headers);
}
if (stream.isRemoteClosed()) {
throw new H2StreamResetException(H2Error.STREAM_CLOSED, "Stream already closed");
}
if (stream.isLocalReset()) {
return;
}
if (frame.isFlagSet(FrameFlag.END_STREAM)) {
stream.setRemoteEndStream();
}
stream.consumeHeader(headers);
} else {
continuation.copyPayload(payload);
}
}
private void consumeContinuationFrame(final RawFrame frame, final H2Stream stream) throws HttpException, IOException {
final int streamId = frame.getStreamId();
final ByteBuffer payload = frame.getPayload();
continuation.copyPayload(payload);
if (frame.isFlagSet(FrameFlag.END_HEADERS)) {
final List headers = decodeHeaders(continuation.getContent());
if (stream.isRemoteInitiated() && streamId > processedRemoteStreamId) {
processedRemoteStreamId = streamId;
}
if (streamListener != null) {
streamListener.onHeaderInput(this, streamId, headers);
}
if (stream.isRemoteClosed()) {
throw new H2StreamResetException(H2Error.STREAM_CLOSED, "Stream already closed");
}
if (stream.isLocalReset()) {
return;
}
if (continuation.endStream) {
stream.setRemoteEndStream();
}
if (continuation.type == FrameType.PUSH_PROMISE.getValue()) {
stream.consumePromise(headers);
} else {
stream.consumeHeader(headers);
}
continuation = null;
}
}
private void consumeSettingsFrame(final ByteBuffer payload) throws HttpException, IOException {
final H2Config.Builder configBuilder = H2Config.initial();
while (payload.hasRemaining()) {
final int code = payload.getShort();
final int value = payload.getInt();
final H2Param param = H2Param.valueOf(code);
if (param != null) {
switch (param) {
case HEADER_TABLE_SIZE:
try {
configBuilder.setHeaderTableSize(value);
} catch (final IllegalArgumentException ex) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, ex.getMessage());
}
break;
case MAX_CONCURRENT_STREAMS:
try {
configBuilder.setMaxConcurrentStreams(value);
} catch (final IllegalArgumentException ex) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, ex.getMessage());
}
break;
case ENABLE_PUSH:
configBuilder.setPushEnabled(value == 1);
break;
case INITIAL_WINDOW_SIZE:
try {
configBuilder.setInitialWindowSize(value);
} catch (final IllegalArgumentException ex) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, ex.getMessage());
}
break;
case MAX_FRAME_SIZE:
try {
configBuilder.setMaxFrameSize(value);
} catch (final IllegalArgumentException ex) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, ex.getMessage());
}
break;
case MAX_HEADER_LIST_SIZE:
try {
configBuilder.setMaxHeaderListSize(value);
} catch (final IllegalArgumentException ex) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, ex.getMessage());
}
break;
}
}
}
applyRemoteSettings(configBuilder.build());
}
private void produceOutput() throws HttpException, IOException {
for (final Iterator> it = streamMap.entrySet().iterator(); it.hasNext(); ) {
final Map.Entry entry = it.next();
final H2Stream stream = entry.getValue();
if (!stream.isLocalClosed() && stream.getOutputWindow().get() > 0) {
stream.produceOutput();
}
if (stream.isTerminated()) {
it.remove();
stream.releaseResources();
requestSessionOutput();
}
if (!outputQueue.isEmpty()) {
break;
}
}
}
private void applyRemoteSettings(final H2Config config) throws H2ConnectionException {
remoteConfig = config;
hPackEncoder.setMaxTableSize(remoteConfig.getHeaderTableSize());
final int delta = remoteConfig.getInitialWindowSize() - initOutputWinSize;
initOutputWinSize = remoteConfig.getInitialWindowSize();
if (delta != 0) {
if (!streamMap.isEmpty()) {
for (final Iterator> it = streamMap.entrySet().iterator(); it.hasNext(); ) {
final Map.Entry entry = it.next();
final H2Stream stream = entry.getValue();
try {
updateOutputWindow(stream.getId(), stream.getOutputWindow(), delta);
} catch (final ArithmeticException ex) {
throw new H2ConnectionException(H2Error.FLOW_CONTROL_ERROR, ex.getMessage());
}
}
}
}
}
private void applyLocalSettings() throws H2ConnectionException {
hPackDecoder.setMaxTableSize(localConfig.getHeaderTableSize());
hPackDecoder.setMaxListSize(localConfig.getMaxHeaderListSize());
final int delta = localConfig.getInitialWindowSize() - initInputWinSize;
initInputWinSize = localConfig.getInitialWindowSize();
if (delta != 0 && !streamMap.isEmpty()) {
for (final Iterator> it = streamMap.entrySet().iterator(); it.hasNext(); ) {
final Map.Entry entry = it.next();
final H2Stream stream = entry.getValue();
try {
updateInputWindow(stream.getId(), stream.getInputWindow(), delta);
} catch (final ArithmeticException ex) {
throw new H2ConnectionException(H2Error.FLOW_CONTROL_ERROR, ex.getMessage());
}
}
}
lowMark = initInputWinSize / 2;
}
@Override
public void close() throws IOException {
ioSession.enqueue(ShutdownCommand.GRACEFUL, Command.Priority.IMMEDIATE);
}
@Override
public void close(final CloseMode closeMode) {
ioSession.close(closeMode);
}
@Override
public boolean isOpen() {
return connState == ConnectionHandshake.ACTIVE;
}
@Override
public void setSocketTimeout(final Timeout timeout) {
ioSession.setSocketTimeout(timeout);
}
@Override
public SSLSession getSSLSession() {
final TlsDetails tlsDetails = ioSession.getTlsDetails();
return tlsDetails != null ? tlsDetails.getSSLSession() : null;
}
@Override
public EndpointDetails getEndpointDetails() {
if (endpointDetails == null) {
endpointDetails = new BasicEndpointDetails(
ioSession.getRemoteAddress(),
ioSession.getLocalAddress(),
connMetrics,
ioSession.getSocketTimeout());
}
return endpointDetails;
}
@Override
public Timeout getSocketTimeout() {
return ioSession.getSocketTimeout();
}
@Override
public ProtocolVersion getProtocolVersion() {
return HttpVersion.HTTP_2;
}
@Override
public SocketAddress getRemoteAddress() {
return ioSession.getRemoteAddress();
}
@Override
public SocketAddress getLocalAddress() {
return ioSession.getLocalAddress();
}
void appendState(final StringBuilder buf) {
buf.append("connState=").append(connState)
.append(", connInputWindow=").append(connInputWindow)
.append(", connOutputWindow=").append(connOutputWindow)
.append(", outputQueue=").append(outputQueue.size())
.append(", streamMap=").append(streamMap.size())
.append(", processedRemoteStreamId=").append(processedRemoteStreamId);
}
private static class Continuation {
final int streamId;
final int type;
final boolean endStream;
final ByteArrayBuffer headerBuffer;
private Continuation(final int streamId, final int type, final boolean endStream) {
this.streamId = streamId;
this.type = type;
this.endStream = endStream;
this.headerBuffer = new ByteArrayBuffer(1024);
}
void copyPayload(final ByteBuffer payload) {
if (payload == null) {
return;
}
headerBuffer.ensureCapacity(payload.remaining());
payload.get(headerBuffer.array(), headerBuffer.length(), payload.remaining());
}
ByteBuffer getContent() {
return ByteBuffer.wrap(headerBuffer.array(), 0, headerBuffer.length());
}
}
private class H2StreamChannelImpl implements H2StreamChannel {
private final int id;
private final AtomicInteger inputWindow;
private final AtomicInteger outputWindow;
private volatile boolean idle;
private volatile boolean remoteEndStream;
private volatile boolean localEndStream;
private volatile long deadline;
H2StreamChannelImpl(final int id, final boolean idle, final int initialInputWindowSize, final int initialOutputWindowSize) {
this.id = id;
this.idle = idle;
this.inputWindow = new AtomicInteger(initialInputWindowSize);
this.outputWindow = new AtomicInteger(initialOutputWindowSize);
}
int getId() {
return id;
}
AtomicInteger getOutputWindow() {
return outputWindow;
}
AtomicInteger getInputWindow() {
return inputWindow;
}
@Override
public void submit(final List headers, final boolean endStream) throws IOException {
ioSession.getLock().lock();
try {
if (headers == null || headers.isEmpty()) {
throw new H2ConnectionException(H2Error.INTERNAL_ERROR, "Message headers are missing");
}
if (localEndStream) {
return;
}
idle = false;
commitHeaders(id, headers, endStream);
if (endStream) {
localEndStream = true;
}
} finally {
ioSession.getLock().unlock();
}
}
@Override
public void push(final List headers, final AsyncPushProducer pushProducer) throws HttpException, IOException {
acceptPushRequest();
final int promisedStreamId = generateStreamId();
final H2StreamChannelImpl channel = new H2StreamChannelImpl(
promisedStreamId,
true,
localConfig.getInitialWindowSize(),
remoteConfig.getInitialWindowSize());
final HttpCoreContext context = HttpCoreContext.create();
context.setAttribute(HttpCoreContext.SSL_SESSION, getSSLSession());
context.setAttribute(HttpCoreContext.CONNECTION_ENDPOINT, getEndpointDetails());
final H2StreamHandler streamHandler = new ServerPushH2StreamHandler(
channel, httpProcessor, connMetrics, pushProducer, context);
final H2Stream stream = new H2Stream(channel, streamHandler, false);
streamMap.put(promisedStreamId, stream);
ioSession.getLock().lock();
try {
if (localEndStream) {
stream.releaseResources();
return;
}
commitPushPromise(id, promisedStreamId, headers);
idle = false;
} finally {
ioSession.getLock().unlock();
}
}
@Override
public void update(final int increment) throws IOException {
if (remoteEndStream) {
return;
}
incrementInputCapacity(0, connInputWindow, increment);
incrementInputCapacity(id, inputWindow, increment);
}
@Override
public int write(final ByteBuffer payload) throws IOException {
ioSession.getLock().lock();
try {
if (localEndStream) {
return 0;
}
return streamData(id, outputWindow, payload);
} finally {
ioSession.getLock().unlock();
}
}
@Override
public void endStream(final List trailers) throws IOException {
ioSession.getLock().lock();
try {
if (localEndStream) {
return;
}
localEndStream = true;
if (trailers != null && !trailers.isEmpty()) {
commitHeaders(id, trailers, true);
} else {
final RawFrame frame = frameFactory.createData(id, null, true);
commitFrameInternal(frame);
}
} finally {
ioSession.getLock().unlock();
}
}
@Override
public void endStream() throws IOException {
endStream(null);
}
@Override
public void requestOutput() {
requestSessionOutput();
}
boolean isRemoteClosed() {
return remoteEndStream;
}
void setRemoteEndStream() {
remoteEndStream = true;
}
boolean isLocalClosed() {
return localEndStream;
}
void setLocalEndStream() {
localEndStream = true;
}
boolean isLocalReset() {
return deadline > 0;
}
boolean isResetDeadline() {
final long l = deadline;
return l > 0 && l < System.currentTimeMillis();
}
boolean localReset(final int code) throws IOException {
ioSession.getLock().lock();
try {
if (localEndStream) {
return false;
}
localEndStream = true;
deadline = System.currentTimeMillis() + LINGER_TIME;
if (!idle) {
final RawFrame resetStream = frameFactory.createResetStream(id, code);
commitFrameInternal(resetStream);
return true;
}
return false;
} finally {
ioSession.getLock().unlock();
}
}
boolean localReset(final H2Error error) throws IOException {
return localReset(error!= null ? error.getCode() : H2Error.INTERNAL_ERROR.getCode());
}
@Override
public boolean cancel() {
try {
return localReset(H2Error.CANCEL);
} catch (final IOException ignore) {
return false;
}
}
void appendState(final StringBuilder buf) {
buf.append("id=").append(id)
.append(", connState=").append(connState)
.append(", inputWindow=").append(inputWindow)
.append(", outputWindow=").append(outputWindow)
.append(", localEndStream=").append(localEndStream)
.append(", idle=").append(idle);
}
@Override
public String toString() {
final StringBuilder buf = new StringBuilder();
buf.append("[");
appendState(buf);
buf.append("]");
return buf.toString();
}
}
static class H2Stream {
private final H2StreamChannelImpl channel;
private final H2StreamHandler handler;
private final boolean remoteInitiated;
private H2Stream(
final H2StreamChannelImpl channel,
final H2StreamHandler handler,
final boolean remoteInitiated) {
this.channel = channel;
this.handler = handler;
this.remoteInitiated = remoteInitiated;
}
int getId() {
return channel.getId();
}
boolean isRemoteInitiated() {
return remoteInitiated;
}
AtomicInteger getOutputWindow() {
return channel.getOutputWindow();
}
AtomicInteger getInputWindow() {
return channel.getInputWindow();
}
boolean isTerminated() {
return channel.isLocalClosed() && (channel.isRemoteClosed() || channel.isResetDeadline());
}
boolean isRemoteClosed() {
return channel.isRemoteClosed();
}
boolean isLocalClosed() {
return channel.isLocalClosed();
}
boolean isLocalReset() {
return channel.isLocalReset();
}
void setRemoteEndStream() {
channel.setRemoteEndStream();
}
void consumePromise(final List headers) throws HttpException, IOException {
try {
handler.consumePromise(headers);
channel.setLocalEndStream();
} catch (final ProtocolException ex) {
localReset(ex, H2Error.PROTOCOL_ERROR);
}
}
void consumeHeader(final List headers) throws HttpException, IOException {
try {
handler.consumeHeader(headers, channel.isRemoteClosed());
} catch (final ProtocolException ex) {
localReset(ex, H2Error.PROTOCOL_ERROR);
}
}
void consumeData(final ByteBuffer src) throws HttpException, IOException {
try {
handler.consumeData(src, channel.isRemoteClosed());
} catch (final CharacterCodingException ex) {
localReset(ex, H2Error.INTERNAL_ERROR);
} catch (final ProtocolException ex) {
localReset(ex, H2Error.PROTOCOL_ERROR);
}
}
boolean isOutputReady() {
return handler.isOutputReady();
}
void produceOutput() throws HttpException, IOException {
try {
handler.produceOutput();
} catch (final ProtocolException ex) {
localReset(ex, H2Error.PROTOCOL_ERROR);
}
}
void produceInputCapacityUpdate() throws IOException {
handler.updateInputCapacity();
}
void reset(final Exception cause) {
channel.setRemoteEndStream();
channel.setLocalEndStream();
handler.failed(cause);
}
void localReset(final Exception cause, final int code) throws IOException {
channel.localReset(code);
handler.failed(cause);
}
void localReset(final Exception cause, final H2Error error) throws IOException {
localReset(cause, error != null ? error.getCode() : H2Error.INTERNAL_ERROR.getCode());
}
void localReset(final H2StreamResetException ex) throws IOException {
localReset(ex, ex.getCode());
}
void handle(final HttpException ex) throws IOException, HttpException {
handler.handle(ex, channel.isRemoteClosed());
}
HandlerFactory getPushHandlerFactory() {
return handler.getPushHandlerFactory();
}
void cancel() {
reset(new RequestNotExecutedException());
}
boolean abort() {
final boolean cancelled = channel.cancel();
handler.failed(new RequestNotExecutedException());
return cancelled;
}
void releaseResources() {
handler.releaseResources();
}
void appendState(final StringBuilder buf) {
buf.append("channel=[");
channel.appendState(buf);
buf.append("]");
}
@Override
public String toString() {
final StringBuilder buf = new StringBuilder();
buf.append("[");
appendState(buf);
buf.append("]");
return buf.toString();
}
}
}
././@LongLink 0100644 0000000 0000000 00000000147 14157612023 011636 L ustar 0000000 0000000 httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/ServerHttpProtocolNegotiatorFactory.java httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/ServerHttpProtocolNegotiatorFactory.ja0100664 0000000 0000000 00000010400 14157612023 032421 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import org.apache.hc.core5.annotation.Contract;
import org.apache.hc.core5.annotation.Internal;
import org.apache.hc.core5.annotation.ThreadingBehavior;
import org.apache.hc.core5.http.URIScheme;
import org.apache.hc.core5.http.impl.nio.ServerHttp1StreamDuplexerFactory;
import org.apache.hc.core5.http.nio.ssl.TlsStrategy;
import org.apache.hc.core5.http2.HttpVersionPolicy;
import org.apache.hc.core5.reactor.EndpointParameters;
import org.apache.hc.core5.reactor.IOEventHandlerFactory;
import org.apache.hc.core5.reactor.ProtocolIOSession;
import org.apache.hc.core5.util.Args;
import org.apache.hc.core5.util.Asserts;
import org.apache.hc.core5.util.Timeout;
/**
* {@link ServerHttpProtocolNegotiator} factory.
*
* @since 5.0
*/
@Contract(threading = ThreadingBehavior.IMMUTABLE_CONDITIONAL)
@Internal
public class ServerHttpProtocolNegotiatorFactory implements IOEventHandlerFactory {
private final ServerHttp1StreamDuplexerFactory http1StreamDuplexerFactory;
private final ServerH2StreamMultiplexerFactory http2StreamMultiplexerFactory;
private final HttpVersionPolicy versionPolicy;
private final TlsStrategy tlsStrategy;
private final Timeout handshakeTimeout;
public ServerHttpProtocolNegotiatorFactory(
final ServerHttp1StreamDuplexerFactory http1StreamDuplexerFactory,
final ServerH2StreamMultiplexerFactory http2StreamMultiplexerFactory,
final HttpVersionPolicy versionPolicy,
final TlsStrategy tlsStrategy,
final Timeout handshakeTimeout) {
this.http1StreamDuplexerFactory = Args.notNull(http1StreamDuplexerFactory, "HTTP/1.1 stream handler factory");
this.http2StreamMultiplexerFactory = Args.notNull(http2StreamMultiplexerFactory, "HTTP/2 stream handler factory");
this.versionPolicy = versionPolicy != null ? versionPolicy : HttpVersionPolicy.NEGOTIATE;
this.tlsStrategy = tlsStrategy;
this.handshakeTimeout = handshakeTimeout;
}
@Override
public ServerHttpProtocolNegotiator createHandler(final ProtocolIOSession ioSession, final Object attachment) {
HttpVersionPolicy endpointPolicy = versionPolicy;
if (attachment instanceof EndpointParameters) {
final EndpointParameters params = (EndpointParameters) attachment;
if (URIScheme.HTTPS.same(params.getScheme())) {
Asserts.notNull(tlsStrategy, "TLS strategy");
tlsStrategy.upgrade(
ioSession,
null,
ioSession.getLocalAddress(),
ioSession.getRemoteAddress(),
params.getAttachment(),
handshakeTimeout);
}
if (params.getAttachment() instanceof HttpVersionPolicy) {
endpointPolicy = (HttpVersionPolicy) params.getAttachment();
}
}
return new ServerHttpProtocolNegotiator(
ioSession,
http1StreamDuplexerFactory,
http2StreamMultiplexerFactory,
endpointPolicy);
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/ProtocolNegotiationException.java 0100664 0000000 0000000 00000003172 14157612023 031425 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import java.io.IOException;
/**
* Signals a protocol error in HTTP protocol negotiation.
*
* @since 5.1
*/
public class ProtocolNegotiationException extends IOException {
/**
* Creates a MessageConstraintException with the specified detail message.
*
* @param message The exception detail message
*/
public ProtocolNegotiationException(final String message) {
super(message);
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/H2StreamListener.java 0100644 0000000 0000000 00000004323 13511606054 026674 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import java.util.List;
import org.apache.hc.core5.annotation.Contract;
import org.apache.hc.core5.annotation.Internal;
import org.apache.hc.core5.annotation.ThreadingBehavior;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.HttpConnection;
import org.apache.hc.core5.http2.frame.RawFrame;
/**
* HTTP/2 stream event listener.
*
* @since 5.0
*/
@Contract(threading = ThreadingBehavior.STATELESS)
@Internal
public interface H2StreamListener {
void onHeaderInput(HttpConnection connection, int streamId, List headers);
void onHeaderOutput(HttpConnection connection, int streamId, List headers);
void onFrameInput(HttpConnection connection, int streamId, RawFrame frame);
void onFrameOutput(HttpConnection connection, int streamId, RawFrame frame);
void onInputFlowControl(HttpConnection connection, int streamId, int delta, int actualSize);
void onOutputFlowControl(HttpConnection connection, int streamId, int delta, int actualSize);
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/FrameOutputBuffer.java 0100664 0000000 0000000 00000010176 14114632321 027147 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.GatheringByteChannel;
import java.nio.channels.WritableByteChannel;
import org.apache.hc.core5.http2.H2ConnectionException;
import org.apache.hc.core5.http2.H2Error;
import org.apache.hc.core5.http2.H2TransportMetrics;
import org.apache.hc.core5.http2.frame.FrameConsts;
import org.apache.hc.core5.http2.frame.RawFrame;
import org.apache.hc.core5.http2.impl.BasicH2TransportMetrics;
import org.apache.hc.core5.util.Args;
/**
* Frame output buffer for HTTP/2 non-blocking connections.
*
* @since 5.0
*/
public final class FrameOutputBuffer {
private final BasicH2TransportMetrics metrics;
private final int maxFramePayloadSize;
private final ByteBuffer buffer;
public FrameOutputBuffer(final BasicH2TransportMetrics metrics, final int maxFramePayloadSize) {
Args.notNull(metrics, "HTTP2 transport metrics");
Args.positive(maxFramePayloadSize, "Maximum payload size");
this.metrics = metrics;
this.maxFramePayloadSize = maxFramePayloadSize;
this.buffer = ByteBuffer.allocate(FrameConsts.HEAD_LEN + maxFramePayloadSize);
}
public FrameOutputBuffer(final int maxFramePayloadSize) {
this(new BasicH2TransportMetrics(), maxFramePayloadSize);
}
public void write(final RawFrame frame, final WritableByteChannel channel) throws IOException {
Args.notNull(frame, "Frame");
final ByteBuffer payload = frame.getPayload();
if (payload != null && payload.remaining() > maxFramePayloadSize) {
throw new H2ConnectionException(H2Error.FRAME_SIZE_ERROR, "Frame size exceeds maximum");
}
buffer.putInt((payload != null ? payload.remaining() << 8 : 0) | (frame.getType() & 0xff));
buffer.put((byte) (frame.getFlags() & 0xff));
buffer.putInt(frame.getStreamId());
if (payload != null) {
if (channel instanceof GatheringByteChannel) {
buffer.flip();
((GatheringByteChannel) channel).write(new ByteBuffer[]{buffer, payload});
buffer.compact();
if (payload.hasRemaining()) {
buffer.put(payload);
}
} else {
buffer.put(payload);
}
}
flush(channel);
metrics.incrementFramesTransferred();
}
public void flush(final WritableByteChannel channel) throws IOException {
if (buffer.position() > 0) {
buffer.flip();
try {
final int bytesWritten = channel.write(buffer);
if (bytesWritten > 0) {
metrics.incrementBytesTransferred(bytesWritten);
}
} finally {
buffer.compact();
}
}
}
public boolean isEmpty() {
return buffer.position() == 0;
}
public H2TransportMetrics getMetrics() {
return metrics;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/H2StreamHandler.java 0100644 0000000 0000000 00000004263 13602614500 026463 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.util.List;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.HttpException;
import org.apache.hc.core5.http.nio.AsyncPushConsumer;
import org.apache.hc.core5.http.nio.HandlerFactory;
import org.apache.hc.core5.http.nio.ResourceHolder;
interface H2StreamHandler extends ResourceHolder {
boolean isOutputReady();
void produceOutput() throws HttpException, IOException;
void consumePromise(List headers) throws HttpException, IOException;
void consumeHeader(List headers, boolean endStream) throws HttpException, IOException;
void updateInputCapacity() throws IOException;
void consumeData(ByteBuffer src, boolean endStream) throws HttpException, IOException;
HandlerFactory getPushHandlerFactory();
void failed(Exception cause);
void handle(final HttpException ex, final boolean endStream) throws HttpException, IOException;
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/FrameInputBuffer.java 0100664 0000000 0000000 00000014336 14114632321 026750 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.ReadableByteChannel;
import org.apache.hc.core5.http.ConnectionClosedException;
import org.apache.hc.core5.http2.H2ConnectionException;
import org.apache.hc.core5.http2.H2CorruptFrameException;
import org.apache.hc.core5.http2.H2Error;
import org.apache.hc.core5.http2.H2TransportMetrics;
import org.apache.hc.core5.http2.frame.FrameConsts;
import org.apache.hc.core5.http2.frame.FrameFlag;
import org.apache.hc.core5.http2.frame.RawFrame;
import org.apache.hc.core5.http2.impl.BasicH2TransportMetrics;
import org.apache.hc.core5.util.Args;
/**
* Frame input buffer for HTTP/2 non-blocking connections.
*
* @since 5.0
*/
public final class FrameInputBuffer {
enum State { HEAD_EXPECTED, PAYLOAD_EXPECTED }
private final BasicH2TransportMetrics metrics;
private final int maxFramePayloadSize;
private final byte[] bytes;
private final ByteBuffer buffer;
private State state;
private int payloadLen;
private int type;
private int flags;
private int streamId;
FrameInputBuffer(final BasicH2TransportMetrics metrics, final int bufferLen, final int maxFramePayloadSize) {
Args.notNull(metrics, "HTTP2 transport metrics");
Args.positive(maxFramePayloadSize, "Maximum payload size");
this.metrics = metrics;
this.maxFramePayloadSize = Math.max(maxFramePayloadSize, FrameConsts.MIN_FRAME_SIZE);
this.bytes = new byte[bufferLen];
this.buffer = ByteBuffer.wrap(bytes);
this.buffer.flip();
this.state = State.HEAD_EXPECTED;
}
public FrameInputBuffer(final BasicH2TransportMetrics metrics, final int maxFramePayloadSize) {
this(metrics, FrameConsts.HEAD_LEN + maxFramePayloadSize, maxFramePayloadSize);
}
public FrameInputBuffer(final int maxFramePayloadSize) {
this(new BasicH2TransportMetrics(), maxFramePayloadSize);
}
public void put(final ByteBuffer src) {
if (buffer.hasRemaining()) {
buffer.compact();
} else {
buffer.clear();
}
buffer.put(src);
buffer.flip();
}
public RawFrame read(final ReadableByteChannel channel) throws IOException {
for (;;) {
switch (state) {
case HEAD_EXPECTED:
if (buffer.remaining() >= FrameConsts.HEAD_LEN) {
final int lengthAndType = buffer.getInt();
payloadLen = lengthAndType >> 8;
if (payloadLen > maxFramePayloadSize) {
throw new H2ConnectionException(H2Error.FRAME_SIZE_ERROR, "Frame size exceeds maximum");
}
type = lengthAndType & 0xff;
flags = buffer.get();
streamId = Math.abs(buffer.getInt());
state = State.PAYLOAD_EXPECTED;
} else {
break;
}
case PAYLOAD_EXPECTED:
if (buffer.remaining() >= payloadLen) {
if ((flags & FrameFlag.PADDED.getValue()) > 0) {
if (payloadLen == 0) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Inconsistent padding");
}
buffer.mark();
final int padding = buffer.get();
if (payloadLen < padding + 1) {
throw new H2ConnectionException(H2Error.PROTOCOL_ERROR, "Inconsistent padding");
}
buffer.reset();
}
final ByteBuffer payload = payloadLen > 0 ? ByteBuffer.wrap(bytes, buffer.position(), payloadLen) : null;
buffer.position(buffer.position() + payloadLen);
state = State.HEAD_EXPECTED;
metrics.incrementFramesTransferred();
return new RawFrame(type, flags, streamId, payload);
}
}
if (buffer.hasRemaining()) {
buffer.compact();
} else {
buffer.clear();
}
final int bytesRead = channel.read(buffer);
buffer.flip();
if (bytesRead > 0) {
metrics.incrementBytesTransferred(bytesRead);
}
if (bytesRead == 0) {
break;
} else if (bytesRead < 0) {
if (state != State.HEAD_EXPECTED || buffer.hasRemaining()) {
throw new H2CorruptFrameException("Corrupt or incomplete HTTP2 frame");
} else {
throw new ConnectionClosedException();
}
}
}
return null;
}
public void reset() {
buffer.compact();
state = State.HEAD_EXPECTED;
}
public H2TransportMetrics getMetrics() {
return metrics;
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/ProtocolNegotiatorBase.java 0100664 0000000 0000000 00000013276 14156462653 030215 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import java.io.IOException;
import java.net.SocketAddress;
import java.nio.ByteBuffer;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicReference;
import javax.net.ssl.SSLSession;
import org.apache.hc.core5.concurrent.FutureCallback;
import org.apache.hc.core5.http.ConnectionClosedException;
import org.apache.hc.core5.http.EndpointDetails;
import org.apache.hc.core5.http.ProtocolVersion;
import org.apache.hc.core5.http.impl.nio.HttpConnectionEventHandler;
import org.apache.hc.core5.http.nio.command.CommandSupport;
import org.apache.hc.core5.io.CloseMode;
import org.apache.hc.core5.io.SocketTimeoutExceptionFactory;
import org.apache.hc.core5.reactor.IOSession;
import org.apache.hc.core5.reactor.ProtocolIOSession;
import org.apache.hc.core5.reactor.ssl.TlsDetails;
import org.apache.hc.core5.util.Args;
import org.apache.hc.core5.util.Timeout;
abstract class ProtocolNegotiatorBase implements HttpConnectionEventHandler {
final ProtocolIOSession ioSession;
private final AtomicReference protocolHandlerRef;
private final FutureCallback resultCallback;
private final AtomicBoolean completed;
ProtocolNegotiatorBase(
final ProtocolIOSession ioSession,
final FutureCallback resultCallback) {
this.ioSession = Args.notNull(ioSession, "I/O session");
this.protocolHandlerRef = new AtomicReference<>();
this.resultCallback = resultCallback;
this.completed = new AtomicBoolean();
}
void startProtocol(final HttpConnectionEventHandler protocolHandler, final ByteBuffer data) throws IOException {
protocolHandlerRef.set(protocolHandler);
ioSession.upgrade(protocolHandler);
protocolHandler.connected(ioSession);
if (data != null && data.hasRemaining()) {
protocolHandler.inputReady(ioSession, data);
}
if (completed.compareAndSet(false, true) && resultCallback != null) {
resultCallback.completed(ioSession);
}
}
@Override
public void timeout(final IOSession session, final Timeout timeout) {
exception(session, SocketTimeoutExceptionFactory.create(timeout));
}
@Override
public void exception(final IOSession session, final Exception cause) {
final HttpConnectionEventHandler protocolHandler = protocolHandlerRef.get();
try {
session.close(CloseMode.IMMEDIATE);
if (protocolHandler != null) {
protocolHandler.exception(session, cause);
} else {
CommandSupport.failCommands(session, cause);
}
} catch (final Exception ex) {
if (completed.compareAndSet(false, true) && resultCallback != null) {
resultCallback.failed(ex);
}
}
}
@Override
public void disconnected(final IOSession session) {
final HttpConnectionEventHandler protocolHandler = protocolHandlerRef.getAndSet(null);
try {
if (protocolHandler != null) {
protocolHandler.disconnected(ioSession);
} else {
CommandSupport.cancelCommands(session);
}
} finally {
if (completed.compareAndSet(false, true) && resultCallback != null) {
resultCallback.failed(new ConnectionClosedException());
}
}
}
@Override
public SSLSession getSSLSession() {
final TlsDetails tlsDetails = ioSession.getTlsDetails();
return tlsDetails != null ? tlsDetails.getSSLSession() : null;
}
@Override
public EndpointDetails getEndpointDetails() {
return null;
}
@Override
public void setSocketTimeout(final Timeout timeout) {
ioSession.setSocketTimeout(timeout);
}
@Override
public Timeout getSocketTimeout() {
return ioSession.getSocketTimeout();
}
@Override
public ProtocolVersion getProtocolVersion() {
return null;
}
@Override
public SocketAddress getRemoteAddress() {
return ioSession.getRemoteAddress();
}
@Override
public SocketAddress getLocalAddress() {
return ioSession.getLocalAddress();
}
@Override
public boolean isOpen() {
return ioSession.isOpen();
}
@Override
public void close() throws IOException {
ioSession.close();
}
@Override
public void close(final CloseMode closeMode) {
ioSession.close(closeMode);
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/ServerPushH2StreamHandler.java 0100644 0000000 0000000 00000023350 13602614500 030510 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.util.List;
import java.util.concurrent.atomic.AtomicBoolean;
import org.apache.hc.core5.http.EntityDetails;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.HttpException;
import org.apache.hc.core5.http.HttpRequest;
import org.apache.hc.core5.http.HttpResponse;
import org.apache.hc.core5.http.HttpStatus;
import org.apache.hc.core5.http.HttpVersion;
import org.apache.hc.core5.http.ProtocolException;
import org.apache.hc.core5.http.impl.BasicHttpConnectionMetrics;
import org.apache.hc.core5.http.impl.nio.MessageState;
import org.apache.hc.core5.http.nio.AsyncPushConsumer;
import org.apache.hc.core5.http.nio.AsyncPushProducer;
import org.apache.hc.core5.http.nio.DataStreamChannel;
import org.apache.hc.core5.http.nio.HandlerFactory;
import org.apache.hc.core5.http.nio.ResponseChannel;
import org.apache.hc.core5.http.protocol.HttpContext;
import org.apache.hc.core5.http.protocol.HttpCoreContext;
import org.apache.hc.core5.http.protocol.HttpProcessor;
import org.apache.hc.core5.http2.H2ConnectionException;
import org.apache.hc.core5.http2.H2Error;
import org.apache.hc.core5.http2.impl.DefaultH2RequestConverter;
import org.apache.hc.core5.http2.impl.DefaultH2ResponseConverter;
class ServerPushH2StreamHandler implements H2StreamHandler {
private final H2StreamChannel outputChannel;
private final DataStreamChannel dataChannel;
private final HttpProcessor httpProcessor;
private final BasicHttpConnectionMetrics connMetrics;
private final AsyncPushProducer pushProducer;
private final HttpCoreContext context;
private final AtomicBoolean responseCommitted;
private final AtomicBoolean failed;
private final AtomicBoolean done;
private volatile MessageState requestState;
private volatile MessageState responseState;
ServerPushH2StreamHandler(
final H2StreamChannel outputChannel,
final HttpProcessor httpProcessor,
final BasicHttpConnectionMetrics connMetrics,
final AsyncPushProducer pushProducer,
final HttpCoreContext context) {
this.outputChannel = outputChannel;
this.dataChannel = new DataStreamChannel() {
@Override
public void requestOutput() {
outputChannel.requestOutput();
}
@Override
public int write(final ByteBuffer src) throws IOException {
return outputChannel.write(src);
}
@Override
public void endStream(final List trailers) throws IOException {
outputChannel.endStream(trailers);
responseState = MessageState.COMPLETE;
}
@Override
public void endStream() throws IOException {
outputChannel.endStream();
responseState = MessageState.COMPLETE;
}
};
this.httpProcessor = httpProcessor;
this.connMetrics = connMetrics;
this.pushProducer = pushProducer;
this.context = context;
this.responseCommitted = new AtomicBoolean(false);
this.failed = new AtomicBoolean(false);
this.done = new AtomicBoolean(false);
this.requestState = MessageState.COMPLETE;
this.responseState = MessageState.IDLE;
}
@Override
public HandlerFactory getPushHandlerFactory() {
return null;
}
@Override
public void consumePromise(final List headers) throws HttpException, IOException {
throw new ProtocolException("Unexpected message promise");
}
@Override
public void consumeHeader(final List requestHeaders, final boolean requestEndStream) throws HttpException, IOException {
throw new ProtocolException("Unexpected message headers");
}
@Override
public void updateInputCapacity() throws IOException {
}
@Override
public void consumeData(final ByteBuffer src, final boolean endStream) throws HttpException, IOException {
throw new ProtocolException("Unexpected message data");
}
@Override
public boolean isOutputReady() {
switch (responseState) {
case IDLE:
return true;
case BODY:
return pushProducer.available() > 0;
default:
return false;
}
}
private void commitInformation(final HttpResponse response) throws IOException, HttpException {
if (responseCommitted.get()) {
throw new H2ConnectionException(H2Error.INTERNAL_ERROR, "Response already committed");
}
final int status = response.getCode();
if (status < HttpStatus.SC_INFORMATIONAL || status >= HttpStatus.SC_SUCCESS) {
throw new HttpException("Invalid intermediate response: " + status);
}
final List responseHeaders = DefaultH2ResponseConverter.INSTANCE.convert(response);
outputChannel.submit(responseHeaders, false);
}
private void commitResponse(
final HttpResponse response,
final EntityDetails responseEntityDetails) throws HttpException, IOException {
if (responseCommitted.compareAndSet(false, true)) {
context.setProtocolVersion(HttpVersion.HTTP_2);
context.setAttribute(HttpCoreContext.HTTP_RESPONSE, response);
httpProcessor.process(response, responseEntityDetails, context);
final List headers = DefaultH2ResponseConverter.INSTANCE.convert(response);
outputChannel.submit(headers, responseEntityDetails == null);
connMetrics.incrementResponseCount();
if (responseEntityDetails == null) {
responseState = MessageState.COMPLETE;
} else {
responseState = MessageState.BODY;
pushProducer.produce(outputChannel);
}
}
}
private void commitPromise(
final HttpRequest promise,
final AsyncPushProducer pushProducer) throws HttpException, IOException {
context.setProtocolVersion(HttpVersion.HTTP_2);
context.setAttribute(HttpCoreContext.HTTP_REQUEST, promise);
httpProcessor.process(promise, null, context);
final List headers = DefaultH2RequestConverter.INSTANCE.convert(promise);
outputChannel.push(headers, pushProducer);
connMetrics.incrementRequestCount();
}
@Override
public void produceOutput() throws HttpException, IOException {
switch (responseState) {
case IDLE:
responseState = MessageState.HEADERS;
pushProducer.produceResponse(new ResponseChannel() {
@Override
public void sendInformation(final HttpResponse response, final HttpContext httpContext) throws HttpException, IOException {
commitInformation(response);
}
@Override
public void sendResponse(
final HttpResponse response, final EntityDetails entityDetails, final HttpContext httpContext) throws HttpException, IOException {
commitResponse(response, entityDetails);
}
@Override
public void pushPromise(
final HttpRequest promise, final AsyncPushProducer pushProducer, final HttpContext httpContext) throws HttpException, IOException {
commitPromise(promise, pushProducer);
}
}, context);
break;
case BODY:
pushProducer.produce(dataChannel);
break;
}
}
@Override
public void handle(final HttpException ex, final boolean endStream) throws HttpException, IOException {
throw ex;
}
@Override
public void failed(final Exception cause) {
try {
if (failed.compareAndSet(false, true)) {
pushProducer.failed(cause);
}
} finally {
releaseResources();
}
}
@Override
public void releaseResources() {
if (done.compareAndSet(false, true)) {
requestState = MessageState.COMPLETE;
responseState = MessageState.COMPLETE;
pushProducer.releaseResources();
}
}
@Override
public String toString() {
return "[" +
"requestState=" + requestState +
", responseState=" + responseState +
']';
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/ServerH2StreamMultiplexerFactory.java 0100644 0000000 0000000 00000006476 13511606054 032153 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import org.apache.hc.core5.annotation.Contract;
import org.apache.hc.core5.annotation.Internal;
import org.apache.hc.core5.annotation.ThreadingBehavior;
import org.apache.hc.core5.http.config.CharCodingConfig;
import org.apache.hc.core5.http.nio.AsyncServerExchangeHandler;
import org.apache.hc.core5.http.nio.HandlerFactory;
import org.apache.hc.core5.http.protocol.HttpProcessor;
import org.apache.hc.core5.http2.config.H2Config;
import org.apache.hc.core5.http2.frame.DefaultFrameFactory;
import org.apache.hc.core5.reactor.ProtocolIOSession;
import org.apache.hc.core5.util.Args;
/**
* {@link ServerH2StreamMultiplexer} factory.
*
* @since 5.0
*/
@Contract(threading = ThreadingBehavior.IMMUTABLE_CONDITIONAL)
@Internal
public final class ServerH2StreamMultiplexerFactory {
private final HttpProcessor httpProcessor;
private final HandlerFactory exchangeHandlerFactory;
private final H2Config h2Config;
private final CharCodingConfig charCodingConfig;
private final H2StreamListener streamListener;
public ServerH2StreamMultiplexerFactory(
final HttpProcessor httpProcessor,
final HandlerFactory exchangeHandlerFactory,
final H2Config h2Config,
final CharCodingConfig charCodingConfig,
final H2StreamListener streamListener) {
this.httpProcessor = Args.notNull(httpProcessor, "HTTP processor");
this.exchangeHandlerFactory = Args.notNull(exchangeHandlerFactory, "Exchange handler factory");
this.h2Config = h2Config != null ? h2Config : H2Config.DEFAULT;
this.charCodingConfig = charCodingConfig != null ? charCodingConfig : CharCodingConfig.DEFAULT;
this.streamListener = streamListener;
}
public ServerH2StreamMultiplexer create(final ProtocolIOSession ioSession) {
return new ServerH2StreamMultiplexer(
ioSession,
DefaultFrameFactory.INSTANCE,
httpProcessor,
exchangeHandlerFactory,
charCodingConfig,
h2Config,
streamListener);
}
}
httpcore5-h2/src/main/java/org/apache/hc/core5/http2/impl/nio/ServerH2StreamHandler.java 0100644 0000000 0000000 00000034417 13602614500 027656 0 ustar 00 0000000 0000000 /*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package org.apache.hc.core5.http2.impl.nio;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.util.List;
import java.util.concurrent.atomic.AtomicBoolean;
import org.apache.hc.core5.http.EntityDetails;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.HttpException;
import org.apache.hc.core5.http.HttpRequest;
import org.apache.hc.core5.http.HttpResponse;
import org.apache.hc.core5.http.HttpStatus;
import org.apache.hc.core5.http.HttpVersion;
import org.apache.hc.core5.http.Method;
import org.apache.hc.core5.http.ProtocolException;
import org.apache.hc.core5.http.impl.BasicHttpConnectionMetrics;
import org.apache.hc.core5.http.impl.IncomingEntityDetails;
import org.apache.hc.core5.http.impl.ServerSupport;
import org.apache.hc.core5.http.impl.nio.MessageState;
import org.apache.hc.core5.http.nio.AsyncPushConsumer;
import org.apache.hc.core5.http.nio.AsyncPushProducer;
import org.apache.hc.core5.http.nio.AsyncResponseProducer;
import org.apache.hc.core5.http.nio.AsyncServerExchangeHandler;
import org.apache.hc.core5.http.nio.DataStreamChannel;
import org.apache.hc.core5.http.nio.HandlerFactory;
import org.apache.hc.core5.http.nio.ResponseChannel;
import org.apache.hc.core5.http.nio.support.BasicResponseProducer;
import org.apache.hc.core5.http.nio.support.ImmediateResponseExchangeHandler;
import org.apache.hc.core5.http.protocol.HttpContext;
import org.apache.hc.core5.http.protocol.HttpCoreContext;
import org.apache.hc.core5.http.protocol.HttpProcessor;
import org.apache.hc.core5.http2.H2ConnectionException;
import org.apache.hc.core5.http2.H2Error;
import org.apache.hc.core5.http2.H2StreamResetException;
import org.apache.hc.core5.http2.impl.DefaultH2RequestConverter;
import org.apache.hc.core5.http2.impl.DefaultH2ResponseConverter;
import org.apache.hc.core5.util.Asserts;
class ServerH2StreamHandler implements H2StreamHandler {
private final H2StreamChannel outputChannel;
private final DataStreamChannel dataChannel;
private final ResponseChannel responseChannel;
private final HttpProcessor httpProcessor;
private final BasicHttpConnectionMetrics connMetrics;
private final HandlerFactory exchangeHandlerFactory;
private final HttpCoreContext context;
private final AtomicBoolean responseCommitted;
private final AtomicBoolean failed;
private final AtomicBoolean done;
private volatile AsyncServerExchangeHandler exchangeHandler;
private volatile HttpRequest receivedRequest;
private volatile MessageState requestState;
private volatile MessageState responseState;
ServerH2StreamHandler(
final H2StreamChannel outputChannel,
final HttpProcessor httpProcessor,
final BasicHttpConnectionMetrics connMetrics,
final HandlerFactory exchangeHandlerFactory,
final HttpCoreContext context) {
this.outputChannel = outputChannel;
this.dataChannel = new DataStreamChannel() {
@Override
public void requestOutput() {
outputChannel.requestOutput();
}
@Override
public int write(final ByteBuffer src) throws IOException {
return outputChannel.write(src);
}
@Override
public void endStream(final List trailers) throws IOException {
outputChannel.endStream(trailers);
responseState = MessageState.COMPLETE;
}
@Override
public void endStream() throws IOException {
outputChannel.endStream();
responseState = MessageState.COMPLETE;
}
};
this.responseChannel = new ResponseChannel() {
@Override
public void sendInformation(final HttpResponse response, final HttpContext httpContext) throws HttpException, IOException {
commitInformation(response);
}
@Override
public void sendResponse(
final HttpResponse response, final EntityDetails responseEntityDetails, final HttpContext httpContext) throws HttpException, IOException {
ServerSupport.validateResponse(response, responseEntityDetails);
commitResponse(response, responseEntityDetails);
}
@Override
public void pushPromise(
final HttpRequest promise, final AsyncPushProducer pushProducer, final HttpContext httpContext) throws HttpException, IOException {
commitPromise(promise, pushProducer);
}
};
this.httpProcessor = httpProcessor;
this.connMetrics = connMetrics;
this.exchangeHandlerFactory = exchangeHandlerFactory;
this.context = context;
this.responseCommitted = new AtomicBoolean(false);
this.failed = new AtomicBoolean(false);
this.done = new AtomicBoolean(false);
this.requestState = MessageState.HEADERS;
this.responseState = MessageState.IDLE;
}
@Override
public HandlerFactory