| /* |
| * 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.catalina.connector; |
| |
| import java.io.File; |
| import java.io.IOException; |
| import java.io.PrintWriter; |
| import java.util.HashMap; |
| import java.util.List; |
| import java.util.Map; |
| |
| import javax.servlet.ServletException; |
| import javax.servlet.http.HttpServlet; |
| import javax.servlet.http.HttpServletRequest; |
| import javax.servlet.http.HttpServletResponse; |
| import javax.servlet.http.HttpSession; |
| |
| import static org.junit.Assert.assertEquals; |
| import static org.junit.Assert.fail; |
| |
| import org.junit.Assert; |
| import org.junit.Test; |
| |
| import org.apache.catalina.Context; |
| import org.apache.catalina.startup.Tomcat; |
| import org.apache.catalina.startup.TomcatBaseTest; |
| import org.apache.tomcat.util.buf.ByteChunk; |
| |
| /** |
| * Test case for {@link Request}. |
| */ |
| public class TestResponse extends TomcatBaseTest { |
| |
| @Test |
| public void testBug49598() throws Exception { |
| // Setup Tomcat instance |
| Tomcat tomcat = getTomcatInstance(); |
| |
| // Must have a real docBase - just use temp |
| File docBase = new File(System.getProperty("java.io.tmpdir")); |
| Context ctx = tomcat.addContext("", docBase.getAbsolutePath()); |
| |
| Tomcat.addServlet(ctx, "servlet", new Bug49598Servlet()); |
| ctx.addServletMapping("/", "servlet"); |
| |
| tomcat.start(); |
| |
| Map<String,List<String>> headers = new HashMap<>(); |
| getUrl("http://localhost:" + getPort() + "/", new ByteChunk(), headers); |
| |
| // Check for headers without a name |
| for (Map.Entry<String,List<String>> header : headers.entrySet()) { |
| if (header.getKey() == null) { |
| // Expected if this is the response line |
| List<String> values = header.getValue(); |
| if (values.size() == 1 && |
| values.get(0).startsWith("HTTP/1.1")) { |
| continue; |
| } |
| fail("Null header name detected for value " + values); |
| } |
| } |
| |
| // Check for exactly one Set-Cookie header |
| int count = 0; |
| for (String headerName : headers.keySet()) { |
| if ("Set-Cookie".equals(headerName)) { |
| count ++; |
| } |
| } |
| assertEquals(1, count); |
| } |
| |
| private static final class Bug49598Servlet extends HttpServlet { |
| private static final long serialVersionUID = 1L; |
| |
| @Override |
| protected void doGet(HttpServletRequest req, HttpServletResponse resp) |
| throws ServletException, IOException { |
| HttpSession session = req.getSession(true); |
| session.invalidate(); |
| req.getSession(true); |
| } |
| |
| } |
| |
| |
| /** |
| * Tests an issue noticed during the investigation of BZ 52811. |
| */ |
| @Test |
| public void testCharset() throws Exception { |
| // Setup Tomcat instance |
| Tomcat tomcat = getTomcatInstance(); |
| |
| // Must have a real docBase - just use temp |
| File docBase = new File(System.getProperty("java.io.tmpdir")); |
| Context ctx = tomcat.addContext("", docBase.getAbsolutePath()); |
| |
| Tomcat.addServlet(ctx, "servlet", new CharsetServlet()); |
| ctx.addServletMapping("/", "servlet"); |
| |
| tomcat.start(); |
| |
| ByteChunk bc = getUrl("http://localhost:" + getPort() + "/"); |
| |
| assertEquals("OK", bc.toString()); |
| } |
| |
| private static final class CharsetServlet extends HttpServlet { |
| private static final long serialVersionUID = 1L; |
| |
| @Override |
| protected void doGet(HttpServletRequest req, HttpServletResponse resp) |
| throws ServletException, IOException { |
| PrintWriter pw = resp.getWriter(); |
| resp.setHeader("Content-Type", "text/plain;charset=UTF-8"); |
| |
| // Should be ISO-8859-1 because getWriter() was called before |
| // setHeader() |
| if (resp.getCharacterEncoding().equals("ISO-8859-1")) { |
| pw.print("OK"); |
| } else { |
| pw.print("FAIL: " + resp.getCharacterEncoding()); |
| } |
| } |
| |
| } |
| |
| |
| @Test |
| public void testBug52811() throws Exception { |
| // Setup Tomcat instance |
| Tomcat tomcat = getTomcatInstance(); |
| |
| // Must have a real docBase - just use temp |
| File docBase = new File(System.getProperty("java.io.tmpdir")); |
| Context ctx = tomcat.addContext("", docBase.getAbsolutePath()); |
| |
| Tomcat.addServlet(ctx, "servlet", new Bug52811Servlet()); |
| ctx.addServletMapping("/", "servlet"); |
| |
| tomcat.start(); |
| |
| ByteChunk bc = getUrl("http://localhost:" + getPort() + "/"); |
| |
| assertEquals("OK", bc.toString()); |
| } |
| |
| |
| @Test |
| public void testBug53062a() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.toAbsolute("./bar.html"); |
| |
| Assert.assertEquals("http://localhost:8080/level1/level2/bar.html", |
| result); |
| } |
| |
| |
| @Test |
| public void testBug53062b() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.toAbsolute("."); |
| |
| Assert.assertEquals("http://localhost:8080/level1/level2/", result); |
| } |
| |
| |
| @Test |
| public void testBug53062c() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.toAbsolute(".."); |
| |
| Assert.assertEquals("http://localhost:8080/level1/", result); |
| } |
| |
| |
| @Test |
| public void testBug53062d() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.toAbsolute(".././.."); |
| |
| Assert.assertEquals("http://localhost:8080/", result); |
| } |
| |
| |
| @Test(expected=IllegalArgumentException.class) |
| public void testBug53062e() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| resp.toAbsolute("../../.."); |
| } |
| |
| |
| @Test |
| public void testBug53062f() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.toAbsolute("bar.html"); |
| |
| Assert.assertEquals( |
| "http://localhost:8080/level1/level2/bar.html", result); |
| } |
| |
| |
| @Test |
| public void testBug53062g() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.toAbsolute("bar.html?x=/../"); |
| |
| Assert.assertEquals( |
| "http://localhost:8080/level1/level2/bar.html?x=/../", result); |
| } |
| |
| |
| @Test |
| public void testBug53062h() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.toAbsolute("bar.html?x=/../../"); |
| |
| Assert.assertEquals( |
| "http://localhost:8080/level1/level2/bar.html?x=/../../", |
| result); |
| } |
| |
| |
| @Test |
| public void testBug53062i() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.toAbsolute("./.?x=/../../"); |
| |
| Assert.assertEquals( |
| "http://localhost:8080/level1/level2/?x=/../../", result); |
| } |
| |
| |
| @Test |
| public void testBug53062j() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.toAbsolute("./..?x=/../../"); |
| |
| Assert.assertEquals("http://localhost:8080/level1/?x=/../../", result); |
| } |
| |
| |
| @Test |
| public void testBug53062k() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.toAbsolute("./..?x=/../.."); |
| |
| Assert.assertEquals( |
| "http://localhost:8080/level1/?x=/../..", |
| result); |
| } |
| |
| |
| @Test |
| public void testBug53062l() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.toAbsolute("bar.html#/../"); |
| |
| Assert.assertEquals( |
| "http://localhost:8080/level1/level2/bar.html#/../", result); |
| } |
| |
| |
| @Test |
| public void testBug53062m() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.toAbsolute("bar.html#/../../"); |
| |
| Assert.assertEquals( |
| "http://localhost:8080/level1/level2/bar.html#/../../", result); |
| } |
| |
| |
| @Test |
| public void testBug53062n() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.toAbsolute("./.#/../../"); |
| |
| Assert.assertEquals( |
| "http://localhost:8080/level1/level2/#/../../", result); |
| } |
| |
| |
| @Test |
| public void testBug53062o() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.toAbsolute("./..#/../../"); |
| |
| Assert.assertEquals("http://localhost:8080/level1/#/../../", result); |
| } |
| |
| |
| @Test |
| public void testBug53062p() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.toAbsolute("./..#/../.."); |
| |
| Assert.assertEquals("http://localhost:8080/level1/#/../..", result); |
| } |
| |
| |
| @Test |
| public void testBug53469a() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.encodeURL("../bar.html"); |
| |
| Assert.assertEquals("../bar.html", result); |
| } |
| |
| |
| @Test |
| public void testBug53469b() throws Exception { |
| Request req = new TesterRequest(); |
| Response resp = new Response(); |
| resp.setRequest(req); |
| |
| String result = resp.encodeURL("../../../../bar.html"); |
| |
| Assert.assertEquals("../../../../bar.html", result); |
| } |
| |
| |
| private static final class Bug52811Servlet extends HttpServlet { |
| private static final long serialVersionUID = 1L; |
| |
| @Override |
| protected void doGet(HttpServletRequest req, HttpServletResponse resp) |
| throws ServletException, IOException { |
| |
| resp.setContentType("multipart/related;" + |
| "boundary=1_4F50BD36_CDF8C28;" + |
| "Start=\"<31671603.smil>\";" + |
| "Type=\"application/smil;charset=UTF-8\""); |
| |
| // Should be ISO-8859-1 because the charset in the above is part |
| // of the Type parameter |
| PrintWriter pw = resp.getWriter(); |
| if (resp.getCharacterEncoding().equals("ISO-8859-1")) { |
| pw.print("OK"); |
| } else { |
| pw.print("FAIL: " + resp.getCharacterEncoding()); |
| } |
| } |
| |
| } |
| } |