aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat (limited to 'jetty-server/src/test/java/org/eclipse/jetty/server/util/SimpleHttpParser.java')
-rw-r--r--jetty-server/src/test/java/org/eclipse/jetty/server/util/SimpleHttpParser.java158
1 files changed, 158 insertions, 0 deletions
diff --git a/jetty-server/src/test/java/org/eclipse/jetty/server/util/SimpleHttpParser.java b/jetty-server/src/test/java/org/eclipse/jetty/server/util/SimpleHttpParser.java
new file mode 100644
index 0000000..a2154ad
--- /dev/null
+++ b/jetty-server/src/test/java/org/eclipse/jetty/server/util/SimpleHttpParser.java
@@ -0,0 +1,158 @@
+//========================================================================
+//Copyright 2012 Mort Bay Consulting Pty. Ltd.
+//------------------------------------------------------------------------
+//All rights reserved. This program and the accompanying materials
+//are made available under the terms of the Eclipse Public License v1.0
+//and Apache License v2.0 which accompanies this distribution.
+//The Eclipse Public License is available at
+//http://www.eclipse.org/legal/epl-v10.html
+//The Apache License v2.0 is available at
+//http://www.opensource.org/licenses/apache2.0.php
+//You may elect to redistribute this code under either of these licenses.
+//========================================================================
+package org.eclipse.jetty.server.util;
+
+import java.io.BufferedReader;
+import java.io.EOFException;
+import java.io.IOException;
+import java.net.SocketTimeoutException;
+import java.util.LinkedHashMap;
+import java.util.Map;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import org.eclipse.jetty.server.TestHttpResponse;
+
+import static org.hamcrest.Matchers.is;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertTrue;
+
+public class SimpleHttpParser
+{
+ final String httpVersion;
+
+ public SimpleHttpParser(String httpVersion)
+ {
+ this.httpVersion = httpVersion;
+ }
+
+ public TestHttpResponse readResponse(BufferedReader reader) throws IOException
+ {
+ // Simplified parser for HTTP responses
+ String line = reader.readLine();
+ if (line == null)
+ throw new EOFException();
+ Matcher responseLine = Pattern.compile("HTTP/1.1" + "\\s+(\\d+)").matcher(line);
+ assertThat("http version is 1.1", responseLine.lookingAt(), is(true));
+ String code = responseLine.group(1);
+
+ Map<String, String> headers = new LinkedHashMap<>();
+ while ((line = reader.readLine()) != null)
+ {
+ if (line.trim().length() == 0)
+ break;
+
+ parseHeader(line, headers);
+ }
+
+ StringBuilder body;
+ if (headers.containsKey("content-length"))
+ {
+ body = parseContentLengthDelimitedBody(reader, headers);
+ }
+ else if ("chunked".equals(headers.get("transfer-encoding")))
+ {
+ body = parseChunkedBody(reader);
+ }
+ else
+ {
+ body = parseEOFDelimitedBody(reader, headers);
+ }
+
+ return new TestHttpResponse(code, headers, body.toString().trim());
+ }
+
+ private void parseHeader(String line, Map<String, String> headers)
+ {
+ Matcher header = Pattern.compile("([^:]+):\\s*(.*)").matcher(line);
+ assertTrue(header.lookingAt());
+ String headerName = header.group(1);
+ String headerValue = header.group(2);
+ headers.put(headerName.toLowerCase(), headerValue.toLowerCase());
+ }
+
+ private StringBuilder parseContentLengthDelimitedBody(BufferedReader reader, Map<String, String> headers) throws IOException
+ {
+ StringBuilder body;
+ int readLen = 0;
+ int length = Integer.parseInt(headers.get("content-length"));
+ body = new StringBuilder(length);
+ try
+ {
+ //TODO: UTF-8 reader from joakim
+ for (int i = 0; i < length; ++i)
+ {
+ char c = (char)reader.read();
+ body.append(c);
+ readLen++;
+ }
+
+ }
+ catch (SocketTimeoutException e)
+ {
+ System.err.printf("Read %,d bytes (out of an expected %,d bytes)%n", readLen, length);
+ throw e;
+ }
+ return body;
+ }
+
+ private StringBuilder parseChunkedBody(BufferedReader reader) throws IOException
+ {
+ StringBuilder body;
+ String line;
+ body = new StringBuilder(64 * 1024);
+ while ((line = reader.readLine()) != null)
+ {
+ if ("0".equals(line))
+ {
+ line = reader.readLine();
+ assertThat("There's no more content after as 0 indicated the final chunk", line, is(""));
+ break;
+ }
+
+ int length = Integer.parseInt(line, 16);
+ //TODO: UTF-8 reader from joakim
+ for (int i = 0; i < length; ++i)
+ {
+ char c = (char)reader.read();
+ body.append(c);
+ }
+ reader.readLine();
+ // assertThat("chunk is followed by an empty line", line, is("")); //TODO: is this right? - NO. Don't
+ // think you can really do chunks with read line generally, but maybe for this test is OK.
+ }
+ return body;
+ }
+
+ private StringBuilder parseEOFDelimitedBody(BufferedReader reader, Map<String, String> headers) throws IOException
+ {
+ StringBuilder body;
+ if ("HTTP/1.1".equals(httpVersion))
+ assertThat("if no content-length or transfer-encoding header is set, " +
+ "connection: close header must be set", headers.get("connection"),
+ is("close"));
+
+ // read until EOF
+ body = new StringBuilder();
+ while (true)
+ {
+ //TODO: UTF-8 reader from joakim
+ int read = reader.read();
+ if (read == -1)
+ break;
+ char c = (char)read;
+ body.append(c);
+ }
+ return body;
+ }
+}