/*
* 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.
*/
package org.apache.coyote.http2;
import java.io.IOException;
import java.nio.ByteBuffer;
import jakarta.servlet.ServletException;
import jakarta.servlet.http.HttpServlet;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import org.junit.Assert;
import org.junit.Test;
import org.apache.catalina.Context;
import org.apache.catalina.startup.Tomcat;
import trailers.ResponseTrailers;
public class TestStream extends Http2TestBase {
/*
* https://bz.apache.org/bugzilla/show_bug.cgi?id=61120
*/
@Test
public void testPathParam() throws Exception {
enableHttp2();
Tomcat tomcat = getTomcatInstance();
Context ctxt = getProgrammaticRootContext();
Tomcat.addServlet(ctxt, "simple", new SimpleServlet());
ctxt.addServletMappingDecoded("/simple", "simple");
Tomcat.addServlet(ctxt, "pathparam", new PathParam());
ctxt.addServletMappingDecoded("/pathparam", "pathparam");
tomcat.start();
openClientConnection();
doHttpUpgrade();
sendClientPreface();
validateHttp2InitialResponse();
byte[] frameHeader = new byte[9];
ByteBuffer headersPayload = ByteBuffer.allocate(128);
buildGetRequest(frameHeader, headersPayload, null, 3, "/pathparam;jsessionid=" + PathParam.EXPECTED_SESSION_ID);
writeFrame(frameHeader, headersPayload);
readSimpleGetResponse();
Assert.assertEquals("3-HeadersStart " + "3-Header-[:status]-[200] " +
"3-Header-[content-type]-[text/plain;charset=UTF-8] " + "3-Header-[content-length]-[2] " +
"3-Header-[date]-[Wed, 11 Nov 2015 19:18:42 GMT] " + "3-HeadersEnd " + "3-Body-2 " +
"3-EndOfStream ", output.getTrace());
}
@Test
public void testResponseTrailerFields() throws Exception {
enableHttp2();
Tomcat tomcat = getTomcatInstance();
Context ctxt = getProgrammaticRootContext();
Tomcat.addServlet(ctxt, "simple", new SimpleServlet());
ctxt.addServletMappingDecoded("/simple", "simple");
Tomcat.addServlet(ctxt, "trailers", new ResponseTrailers());
ctxt.addServletMappingDecoded("/trailers", "trailers");
tomcat.start();
openClientConnection();
doHttpUpgrade();
sendClientPreface();
validateHttp2InitialResponse();
byte[] frameHeader = new byte[9];
ByteBuffer headersPayload = ByteBuffer.allocate(128);
buildGetRequest(frameHeader, headersPayload, null, 3, "/trailers");
writeFrame(frameHeader, headersPayload);
// Headers
parser.readFrame();
// Body
parser.readFrame();
// Trailers
parser.readFrame();
Assert.assertEquals(
"3-HeadersStart " + "3-Header-[:status]-[200] " +
"3-Header-[content-type]-[text/plain;charset=UTF-8] " + "3-Header-[content-length]-[44] " +
"3-Header-[date]-[Wed, 11 Nov 2015 19:18:42 GMT] " + "3-HeadersEnd " + "3-Body-44 " +
"3-HeadersStart " + "3-Header-[x-trailer-2]-[Trailer value two] " +
"3-Header-[x-trailer-1]-[Trailer value one] " + "3-HeadersEnd " + "3-EndOfStream ",
output.getTrace());
}
private static final class PathParam extends HttpServlet {
private static final long serialVersionUID = 1L;
public static final String EXPECTED_SESSION_ID = "0123456789ABCDEF";
@Override
protected void doGet(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
response.setContentType("text/plain");
response.setCharacterEncoding("UTF-8");
if (EXPECTED_SESSION_ID.equals(request.getRequestedSessionId())) {
response.getWriter().write("OK");
} else {
response.getWriter().write("FAIL");
}
}
}
}