| /* |
| * 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.mina.http; |
| |
| import static org.junit.Assert.assertEquals; |
| import static org.junit.Assert.assertNotEquals; |
| import static org.junit.Assert.assertTrue; |
| |
| import java.nio.charset.CharacterCodingException; |
| import java.nio.charset.Charset; |
| import java.nio.charset.CharsetEncoder; |
| |
| import org.apache.mina.core.buffer.IoBuffer; |
| import org.apache.mina.core.filterchain.IoFilter.NextFilter; |
| import org.apache.mina.core.session.DummySession; |
| import org.apache.mina.core.session.IoSession; |
| import org.apache.mina.filter.codec.AbstractProtocolDecoderOutput; |
| import org.apache.mina.filter.codec.ProtocolDecoder; |
| import org.apache.mina.http.api.HttpEndOfContent; |
| import org.apache.mina.http.api.HttpRequest; |
| import org.junit.Test; |
| |
| public class HttpServerDecoderTest { |
| private static final String DECODER_STATE_ATT = "http.ds"; |
| |
| private static final CharsetEncoder encoder = Charset.forName("US-ASCII").newEncoder(); //$NON-NLS-1$ |
| |
| private static final ProtocolDecoder decoder = new HttpServerDecoder(); |
| |
| /* |
| * Use a single session for all requests in order to test state management better |
| */ |
| private static IoSession session = new DummySession(); |
| |
| /** |
| * Build an IO buffer containing a simple minimal HTTP request. |
| * |
| * @param method the HTTP method |
| * @param body the option body |
| * @return the built IO buffer |
| * @throws CharacterCodingException if encoding fails |
| */ |
| protected static IoBuffer getRequestBuffer(String method, String body) throws CharacterCodingException { |
| IoBuffer buffer = IoBuffer.allocate(0).setAutoExpand(true); |
| buffer.putString(method + " / HTTP/1.1\r\nHost: dummy\r\n", encoder); |
| |
| if (body != null) { |
| buffer.putString("Content-Length: " + body.length() + "\r\n\r\n", encoder); |
| buffer.putString(body, encoder); |
| } else { |
| buffer.putString("\r\n", encoder); |
| } |
| |
| buffer.rewind(); |
| |
| return buffer; |
| } |
| |
| protected static IoBuffer getRequestBuffer(String method) throws CharacterCodingException { |
| return getRequestBuffer(method, null); |
| } |
| |
| /** |
| * Execute an HTPP request and return the queue of messages. |
| * |
| * @param method the HTTP method |
| * @param body the optional body |
| * @return the protocol output and its queue of messages |
| * @throws Exception if error occurs (encoding,...) |
| */ |
| protected static AbstractProtocolDecoderOutput executeRequest(String method, String body) throws Exception { |
| AbstractProtocolDecoderOutput out = new AbstractProtocolDecoderOutput() { |
| public void flush(NextFilter nextFilter, IoSession session) { |
| } |
| }; |
| |
| IoBuffer buffer = getRequestBuffer(method, body); //$NON-NLS-1$ |
| |
| while (buffer.hasRemaining()) { |
| decoder.decode(session, buffer, out); |
| } |
| |
| return out; |
| } |
| |
| @Test |
| public void testGetRequestWithoutBody() throws Exception { |
| AbstractProtocolDecoderOutput out = executeRequest("GET", null); |
| assertEquals(2, out.getMessageQueue().size()); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpRequest); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| } |
| |
| @Test |
| public void testGetRequestBody() throws Exception { |
| AbstractProtocolDecoderOutput out = executeRequest("GET", "body"); |
| assertEquals(3, out.getMessageQueue().size()); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpRequest); |
| assertTrue(out.getMessageQueue().poll() instanceof IoBuffer); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| } |
| |
| @Test |
| public void testPutRequestWithoutBody() throws Exception { |
| AbstractProtocolDecoderOutput out = executeRequest("PUT", null); |
| assertEquals(2, out.getMessageQueue().size()); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpRequest); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| } |
| |
| @Test |
| public void testPutRequestBody() throws Exception { |
| AbstractProtocolDecoderOutput out = executeRequest("PUT", "body"); |
| assertEquals(3, out.getMessageQueue().size()); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpRequest); |
| assertTrue(out.getMessageQueue().poll() instanceof IoBuffer); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| } |
| |
| @Test |
| public void testPostRequestWithoutBody() throws Exception { |
| AbstractProtocolDecoderOutput out = executeRequest("POST", null); |
| assertEquals(2, out.getMessageQueue().size()); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpRequest); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| } |
| |
| @Test |
| public void testPostRequestBody() throws Exception { |
| AbstractProtocolDecoderOutput out = executeRequest("POST", "body"); |
| assertEquals(3, out.getMessageQueue().size()); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpRequest); |
| assertTrue(out.getMessageQueue().poll() instanceof IoBuffer); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| } |
| |
| @Test |
| public void testDeleteRequestWithoutBody() throws Exception { |
| AbstractProtocolDecoderOutput out = executeRequest("DELETE", null); |
| assertEquals(2, out.getMessageQueue().size()); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpRequest); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| } |
| |
| @Test |
| public void testDeleteRequestBody() throws Exception { |
| AbstractProtocolDecoderOutput out = executeRequest("DELETE", "body"); |
| assertEquals(3, out.getMessageQueue().size()); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpRequest); |
| assertTrue(out.getMessageQueue().poll() instanceof IoBuffer); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| } |
| |
| @Test |
| public void testDIRMINA965NoContent() throws Exception { |
| AbstractProtocolDecoderOutput out = new AbstractProtocolDecoderOutput() { |
| public void flush(NextFilter nextFilter, IoSession session) { |
| } |
| }; |
| IoBuffer buffer = IoBuffer.allocate(0).setAutoExpand(true); |
| buffer.putString("GET / HTTP/1.1\r\nHost: ", encoder); |
| buffer.rewind(); |
| while (buffer.hasRemaining()) { |
| decoder.decode(session, buffer, out); |
| } |
| buffer = IoBuffer.allocate(0).setAutoExpand(true); |
| buffer.putString("dummy\r\n\r\n", encoder); |
| buffer.rewind(); |
| while (buffer.hasRemaining()) { |
| decoder.decode(session, buffer, out); |
| } |
| assertEquals(2, out.getMessageQueue().size()); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpRequest); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| } |
| |
| @Test |
| public void testDIRMINA965WithContent() throws Exception { |
| AbstractProtocolDecoderOutput out = new AbstractProtocolDecoderOutput() { |
| public void flush(NextFilter nextFilter, IoSession session) { |
| } |
| }; |
| IoBuffer buffer = IoBuffer.allocate(0).setAutoExpand(true); |
| buffer.putString("GET / HTTP/1.1\r\nHost: ", encoder); |
| buffer.rewind(); |
| while (buffer.hasRemaining()) { |
| decoder.decode(session, buffer, out); |
| } |
| buffer = IoBuffer.allocate(0).setAutoExpand(true); |
| buffer.putString("dummy\r\nContent-Length: 1\r\n\r\nA", encoder); |
| buffer.rewind(); |
| while (buffer.hasRemaining()) { |
| decoder.decode(session, buffer, out); |
| } |
| assertEquals(3, out.getMessageQueue().size()); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpRequest); |
| assertTrue(out.getMessageQueue().poll() instanceof IoBuffer); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| } |
| @Test |
| public void testDIRMINA965WithContentOnTwoChunks() throws Exception { |
| AbstractProtocolDecoderOutput out = new AbstractProtocolDecoderOutput() { |
| public void flush(NextFilter nextFilter, IoSession session) { |
| } |
| }; |
| IoBuffer buffer = IoBuffer.allocate(0).setAutoExpand(true); |
| buffer.putString("GET / HTTP/1.1\r\nHost: ", encoder); |
| buffer.rewind(); |
| while (buffer.hasRemaining()) { |
| decoder.decode(session, buffer, out); |
| } |
| buffer = IoBuffer.allocate(0).setAutoExpand(true); |
| buffer.putString("dummy\r\nContent-Length: 2\r\n\r\nA", encoder); |
| buffer.rewind(); |
| while (buffer.hasRemaining()) { |
| decoder.decode(session, buffer, out); |
| } |
| buffer = IoBuffer.allocate(0).setAutoExpand(true); |
| buffer.putString("B", encoder); |
| buffer.rewind(); |
| while (buffer.hasRemaining()) { |
| decoder.decode(session, buffer, out); |
| } |
| assertEquals(4, out.getMessageQueue().size()); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpRequest); |
| assertTrue(out.getMessageQueue().poll() instanceof IoBuffer); |
| assertTrue(out.getMessageQueue().poll() instanceof IoBuffer); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| } |
| |
| @Test |
| public void testDIRMINA1035HeadersWithColons() throws Exception { |
| AbstractProtocolDecoderOutput out = new AbstractProtocolDecoderOutput() { |
| public void flush(NextFilter nextFilter, IoSession session) { |
| } |
| }; |
| |
| IoBuffer buffer = IoBuffer.allocate(0).setAutoExpand(true); |
| buffer.putString("GET / HTTP/1.0\r\nHost: localhost\r\n", encoder); |
| buffer.putString("SomeHeaderA: Value-A\r\n", encoder); |
| buffer.putString("SomeHeaderB: Value-B:Has:Some:Colons\r\n", encoder); |
| buffer.putString("SomeHeaderC: Value-C\r\n", encoder); |
| buffer.putString("SomeHeaderD:\r\n\r\n", encoder); |
| buffer.rewind(); |
| while (buffer.hasRemaining()) { |
| decoder.decode(session, buffer, out); |
| } |
| assertEquals(2, out.getMessageQueue().size()); |
| HttpRequest request = (HttpRequest) out.getMessageQueue().poll(); |
| assertEquals("Value-A", request.getHeader("SomeHeaderA".toLowerCase())); |
| assertEquals("Value-B:Has:Some:Colons", request.getHeader("SomeHeaderB".toLowerCase())); |
| assertEquals("Value-C", request.getHeader("SomeHeaderC".toLowerCase())); |
| assertEquals("", request.getHeader("SomeHeaderD".toLowerCase())); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| } |
| |
| @Test |
| public void verifyThatHeaderWithoutLeadingSpaceIsSupported() throws Exception { |
| AbstractProtocolDecoderOutput out = new AbstractProtocolDecoderOutput() { |
| public void flush(NextFilter nextFilter, IoSession session) { |
| } |
| }; |
| IoBuffer buffer = IoBuffer.allocate(0).setAutoExpand(true); |
| buffer.putString("GET / HTTP/1.0\r\nHost:localhost\r\n\r\n", encoder); |
| buffer.rewind(); |
| while (buffer.hasRemaining()) { |
| decoder.decode(session, buffer, out); |
| } |
| assertEquals(2, out.getMessageQueue().size()); |
| HttpRequest request = (HttpRequest) out.getMessageQueue().poll(); |
| assertEquals("localhost", request.getHeader("host")); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| } |
| |
| @Test |
| public void verifyThatLeadingSpacesAreRemovedFromHeader() throws Exception { |
| AbstractProtocolDecoderOutput out = new AbstractProtocolDecoderOutput() { |
| public void flush(NextFilter nextFilter, IoSession session) { |
| } |
| }; |
| IoBuffer buffer = IoBuffer.allocate(0).setAutoExpand(true); |
| buffer.putString("GET / HTTP/1.0\r\nHost: localhost\r\n\r\n", encoder); |
| buffer.rewind(); |
| while (buffer.hasRemaining()) { |
| decoder.decode(session, buffer, out); |
| } |
| assertEquals(2, out.getMessageQueue().size()); |
| HttpRequest request = (HttpRequest) out.getMessageQueue().poll(); |
| assertEquals("localhost", request.getHeader("host")); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| } |
| |
| @Test |
| public void verifyThatTrailingSpacesAreRemovedFromHeader() throws Exception { |
| AbstractProtocolDecoderOutput out = new AbstractProtocolDecoderOutput() { |
| public void flush(NextFilter nextFilter, IoSession session) { |
| } |
| }; |
| IoBuffer buffer = IoBuffer.allocate(0).setAutoExpand(true); |
| buffer.putString("GET / HTTP/1.0\r\nHost:localhost \r\n\r\n", encoder); |
| buffer.rewind(); |
| while (buffer.hasRemaining()) { |
| decoder.decode(session, buffer, out); |
| } |
| assertEquals(2, out.getMessageQueue().size()); |
| HttpRequest request = (HttpRequest) out.getMessageQueue().poll(); |
| assertEquals("localhost", request.getHeader("host")); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| } |
| |
| @Test |
| public void dosOnRequestWithAdditionalData() throws Exception { |
| AbstractProtocolDecoderOutput out = new AbstractProtocolDecoderOutput() { |
| public void flush(NextFilter nextFilter, IoSession session) { |
| } |
| }; |
| |
| IoBuffer buffer = IoBuffer.allocate(0).setAutoExpand(true); |
| buffer.putString("GET / HTTP/1.0\r\nHost:localhost \r\n\r\ndummy", encoder); |
| buffer.rewind(); |
| int prevBufferPosition = buffer.position(); |
| |
| while (buffer.hasRemaining()) { |
| decoder.decode(session, buffer, out); |
| assertNotEquals("Buffer at new position", prevBufferPosition, buffer.position()); |
| prevBufferPosition = buffer.position(); |
| } |
| |
| assertEquals(2, out.getMessageQueue().size()); |
| HttpRequest request = (HttpRequest) out.getMessageQueue().poll(); |
| assertEquals("localhost", request.getHeader("host")); |
| assertTrue(out.getMessageQueue().poll() instanceof HttpEndOfContent); |
| session.removeAttribute(DECODER_STATE_ATT); // This test leaves session in HEAD state, crashing following test |
| } |
| } |