HTTPCLIENT-2293 - Implement RFC-compliant TRACE request interceptor (#486)
- Add RequestTraceInterceptor class to handle HTTP TRACE requests in compliance with RFC 7231, Section 4.3.8. - Throw ProtocolException for sensitive headers like 'Authorization' and 'Cookie' in TRACE requests. - Throw ProtocolException if TRACE request contains a body.
This commit is contained in:
parent
888232447b
commit
83e0339d27
|
@ -77,6 +77,7 @@ import org.apache.hc.client5.http.impl.routing.SystemDefaultRoutePlanner;
|
|||
import org.apache.hc.client5.http.io.HttpClientConnectionManager;
|
||||
import org.apache.hc.client5.http.protocol.RedirectStrategy;
|
||||
import org.apache.hc.client5.http.protocol.RequestAddCookies;
|
||||
import org.apache.hc.client5.http.protocol.RequestTraceInterceptor;
|
||||
import org.apache.hc.client5.http.protocol.RequestClientConnControl;
|
||||
import org.apache.hc.client5.http.protocol.RequestDefaultHeaders;
|
||||
import org.apache.hc.client5.http.protocol.RequestExpectContinue;
|
||||
|
@ -820,6 +821,8 @@ public class HttpClientBuilder {
|
|||
new RequestClientConnControl(),
|
||||
new RequestUserAgent(userAgentCopy),
|
||||
new RequestExpectContinue(),
|
||||
new RequestTraceInterceptor(),
|
||||
new RequestExpectContinue(),
|
||||
new RequestIfRange());
|
||||
if (!cookieManagementDisabled) {
|
||||
b.add(RequestAddCookies.INSTANCE);
|
||||
|
|
|
@ -0,0 +1,161 @@
|
|||
/*
|
||||
* ====================================================================
|
||||
* 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://www.apache.org/>.
|
||||
*
|
||||
*/
|
||||
|
||||
package org.apache.hc.client5.http.protocol;
|
||||
|
||||
import java.io.IOException;
|
||||
|
||||
import org.apache.hc.core5.annotation.Contract;
|
||||
import org.apache.hc.core5.annotation.ThreadingBehavior;
|
||||
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.HttpRequestInterceptor;
|
||||
import org.apache.hc.core5.http.Method;
|
||||
import org.apache.hc.core5.http.ProtocolException;
|
||||
import org.apache.hc.core5.http.protocol.HttpContext;
|
||||
import org.apache.hc.core5.util.Args;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
|
||||
/**
|
||||
* <h1>RequestTraceInterceptor</h1>
|
||||
*
|
||||
* <p>This class serves as an interceptor for HTTP TRACE requests, ensuring they adhere to specific security and protocol guidelines.</p>
|
||||
*
|
||||
* <p><strong>Responsibilities:</strong></p>
|
||||
* <ul>
|
||||
* <li>Validates TRACE requests by checking for sensitive headers such as {@code Authorization} and {@code Cookie}.</li>
|
||||
* <li>Ensures that TRACE requests do not contain a request body, throwing a {@link ProtocolException} if a body is present.</li>
|
||||
* </ul>
|
||||
*
|
||||
* <p><strong>Thread Safety:</strong> This class is stateless and therefore thread-safe, as indicated by its {@code ThreadingBehavior.STATELESS} annotation.</p>
|
||||
*
|
||||
* <p><strong>Interceptor Behavior:</strong></p>
|
||||
* <ul>
|
||||
* <li>If the HTTP method is TRACE, the interceptor throws a {@link ProtocolException} if any {@code Authorization} or {@code Cookie} headers are present to prevent sensitive data leakage.</li>
|
||||
* <li>If a TRACE request contains a body, a {@link ProtocolException} is thrown.</li>
|
||||
* </ul>
|
||||
*
|
||||
* @version 5.4
|
||||
* @see HttpRequestInterceptor
|
||||
* @see HttpException
|
||||
* @see IOException
|
||||
* @see ProtocolException
|
||||
* @see Method#TRACE
|
||||
* @see HttpHeaders#AUTHORIZATION
|
||||
* @see HttpHeaders#COOKIE
|
||||
*//**
|
||||
* <h1>RequestTraceInterceptor</h1>
|
||||
*
|
||||
* <p>This class serves as an interceptor for HTTP TRACE requests, ensuring they adhere to specific security and protocol guidelines.</p>
|
||||
*
|
||||
* <p><strong>Responsibilities:</strong></p>
|
||||
* <ul>
|
||||
* <li>Validates TRACE requests by checking for sensitive headers such as {@code Authorization} and {@code Cookie}.</li>
|
||||
* <li>Ensures that TRACE requests do not contain a request body, throwing a {@link ProtocolException} if a body is present.</li>
|
||||
* </ul>
|
||||
*
|
||||
* <p><strong>Thread Safety:</strong> This class is stateless and therefore thread-safe, as indicated by its {@code ThreadingBehavior.STATELESS} annotation.</p>
|
||||
*
|
||||
* <p><strong>Interceptor Behavior:</strong></p>
|
||||
* <ul>
|
||||
* <li>If the HTTP method is TRACE, the interceptor throws a {@link ProtocolException} if any {@code Authorization} or {@code Cookie} headers are present to prevent sensitive data leakage.</li>
|
||||
* <li>If a TRACE request contains a body, a {@link ProtocolException} is thrown.</li>
|
||||
* </ul>
|
||||
*
|
||||
* @version 5.4
|
||||
* @see HttpRequestInterceptor
|
||||
* @see HttpException
|
||||
* @see IOException
|
||||
* @see ProtocolException
|
||||
* @see Method#TRACE
|
||||
* @see HttpHeaders#AUTHORIZATION
|
||||
* @see HttpHeaders#COOKIE
|
||||
*/
|
||||
@Contract(threading = ThreadingBehavior.STATELESS)
|
||||
public class RequestTraceInterceptor implements HttpRequestInterceptor {
|
||||
|
||||
private static final Logger LOG = LoggerFactory.getLogger(RequestTraceInterceptor.class);
|
||||
|
||||
/**
|
||||
* Singleton instance of {@link RequestTraceInterceptor}.
|
||||
*/
|
||||
public static final HttpRequestInterceptor INSTANCE = new RequestTraceInterceptor();
|
||||
|
||||
/**
|
||||
* Default constructor.
|
||||
*/
|
||||
public RequestTraceInterceptor() {
|
||||
super();
|
||||
}
|
||||
|
||||
/**
|
||||
* Processes an incoming HTTP request. If the request is of type TRACE, it performs the following actions:
|
||||
* <ul>
|
||||
* <li>Throws a {@link ProtocolException} if the request contains an {@code Authorization} header to prevent sensitive data leakage.</li>
|
||||
* <li>Throws a {@link ProtocolException} if the request contains a {@code Cookie} header to prevent sensitive data leakage.</li>
|
||||
* <li>Throws a {@link ProtocolException} if the request contains a body.</li>
|
||||
* </ul>
|
||||
*
|
||||
* @param request The incoming HTTP request. Cannot be {@code null}.
|
||||
* @param entity Details of the request entity. Can be {@code null}.
|
||||
* @param context The HTTP context.
|
||||
* @throws HttpException If a protocol error occurs.
|
||||
* @throws IOException If an I/O error occurs.
|
||||
*/
|
||||
@Override
|
||||
public void process(final HttpRequest request, final EntityDetails entity, final HttpContext context)
|
||||
throws HttpException, IOException {
|
||||
|
||||
Args.notNull(request, "HTTP request");
|
||||
Args.notNull(context, "HTTP context");
|
||||
|
||||
// Check if the request method is TRACE
|
||||
if (Method.TRACE.isSame(request.getMethod())) {
|
||||
|
||||
// A client MUST NOT send content in a TRACE request.
|
||||
if (entity != null) {
|
||||
throw new ProtocolException("TRACE request MUST NOT contain a request body.");
|
||||
}
|
||||
|
||||
// Check for sensitive headers
|
||||
final Header authHeader = request.getHeader(HttpHeaders.AUTHORIZATION);
|
||||
if (authHeader != null) {
|
||||
throw new ProtocolException("TRACE request MUST NOT contain an Authorization header.");
|
||||
}
|
||||
|
||||
// Check for cookies
|
||||
final Header cookieHeader = request.getHeader(HttpHeaders.COOKIE);
|
||||
if (cookieHeader != null) {
|
||||
throw new ProtocolException("TRACE request MUST NOT contain a Cookie header.");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
|
@ -0,0 +1,95 @@
|
|||
/*
|
||||
* ====================================================================
|
||||
* 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://www.apache.org/>.
|
||||
*
|
||||
*/
|
||||
|
||||
package org.apache.hc.client5.http.protocol;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.assertNotNull;
|
||||
import static org.junit.jupiter.api.Assertions.assertNull;
|
||||
import static org.junit.jupiter.api.Assertions.assertThrows;
|
||||
|
||||
import java.io.IOException;
|
||||
|
||||
import org.apache.hc.core5.http.EntityDetails;
|
||||
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.ProtocolException;
|
||||
import org.apache.hc.core5.http.impl.BasicEntityDetails;
|
||||
import org.apache.hc.core5.http.message.BasicHttpRequest;
|
||||
import org.apache.hc.core5.http.protocol.BasicHttpContext;
|
||||
import org.apache.hc.core5.http.protocol.HttpContext;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
class TestRequestTraceInterceptor {
|
||||
|
||||
private RequestTraceInterceptor interceptor;
|
||||
private HttpRequest request;
|
||||
private HttpContext context;
|
||||
|
||||
@BeforeEach
|
||||
void setUp() {
|
||||
interceptor = new RequestTraceInterceptor();
|
||||
context = new BasicHttpContext();
|
||||
}
|
||||
|
||||
@Test
|
||||
void testTraceRequestWithoutSensitiveHeaders() throws HttpException, IOException {
|
||||
request = new BasicHttpRequest("TRACE", "/");
|
||||
interceptor.process(request, null, context);
|
||||
assertNull(request.getHeader(HttpHeaders.AUTHORIZATION));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testTraceRequestWithSensitiveHeaders() {
|
||||
request = new BasicHttpRequest("TRACE", "/");
|
||||
request.setHeader(HttpHeaders.AUTHORIZATION, "Bearer token");
|
||||
assertThrows(ProtocolException.class, () -> interceptor.process(request, null, context));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testTraceRequestWithBody() {
|
||||
request = new BasicHttpRequest("TRACE", "/");
|
||||
final EntityDetails entity = new BasicEntityDetails(10, null);
|
||||
assertThrows(ProtocolException.class, () -> interceptor.process(request, entity, context));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testNonTraceRequest() throws HttpException, IOException {
|
||||
request = new BasicHttpRequest("GET", "/");
|
||||
request.setHeader(HttpHeaders.AUTHORIZATION, "Bearer token");
|
||||
interceptor.process(request, null, context);
|
||||
assertNotNull(request.getHeader(HttpHeaders.AUTHORIZATION));
|
||||
}
|
||||
|
||||
@Test
|
||||
void testTraceRequestWithCookieHeader() {
|
||||
request = new BasicHttpRequest("TRACE", "/");
|
||||
request.setHeader(HttpHeaders.COOKIE, "someCookie=someValue");
|
||||
assertThrows(ProtocolException.class, () -> interceptor.process(request, null, context));
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue