| /**************************************************************** |
| * 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.james.mime4j.stream; |
| |
| import java.io.ByteArrayInputStream; |
| import java.io.IOException; |
| import java.util.Locale; |
| |
| import junit.framework.TestCase; |
| |
| import org.apache.commons.io.IOUtils; |
| import org.apache.james.mime4j.MimeException; |
| import org.apache.james.mime4j.io.BufferedLineReaderInputStream; |
| import org.apache.james.mime4j.io.LineNumberInputStream; |
| import org.apache.james.mime4j.io.MaxHeaderLengthLimitException; |
| import org.apache.james.mime4j.io.MaxHeaderLimitException; |
| import org.apache.james.mime4j.io.MaxLineLimitException; |
| |
| public class MimeEntityTest extends TestCase { |
| |
| public void testSimpleEntity() throws Exception { |
| String message = |
| "To: Road Runner <runner@example.org>\r\n" + |
| "From: Wile E. Cayote <wile@example.org>\r\n" + |
| "Date: Tue, 12 Feb 2008 17:34:09 +0000 (GMT)\r\n" + |
| "Subject: Mail\r\n" + |
| "Content-Type: text/plain\r\n" + |
| "\r\n" + |
| "a very important message"; |
| byte[] raw = message.getBytes("US-ASCII"); |
| ByteArrayInputStream instream = new ByteArrayInputStream(raw); |
| LineNumberInputStream lineInput = new LineNumberInputStream(instream); |
| BufferedLineReaderInputStream rawstream = new BufferedLineReaderInputStream(lineInput, 12); |
| |
| MimeEntity entity = new MimeEntity( |
| lineInput, |
| rawstream, new DefaultBodyDescriptor()); |
| |
| |
| assertEquals(EntityState.T_START_MESSAGE, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_START_HEADER, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("To", entity.getField().getName()); |
| assertEquals("Road Runner <runner@example.org>", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("From", entity.getField().getName()); |
| assertEquals("Wile E. Cayote <wile@example.org>", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("Date", entity.getField().getName()); |
| assertEquals("Tue, 12 Feb 2008 17:34:09 +0000 (GMT)", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("Subject", entity.getField().getName()); |
| assertEquals("Mail", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("Content-Type", entity.getField().getName()); |
| assertEquals("text/plain", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_END_HEADER, entity.getState()); |
| try { |
| entity.getField().getName(); |
| fail("IllegalStateException should have been thrown"); |
| } catch (IllegalStateException expected) { |
| } |
| try { |
| entity.getField().getBody(); |
| fail("IllegalStateException should have been thrown"); |
| } catch (IllegalStateException expected) { |
| } |
| |
| entity.advance(); |
| assertEquals(EntityState.T_BODY, entity.getState()); |
| assertEquals("a very important message", IOUtils.toString(entity.getContentStream())); |
| entity.advance(); |
| assertEquals(EntityState.T_END_MESSAGE, entity.getState()); |
| try { |
| entity.getContentStream(); |
| fail("IllegalStateException should have been thrown"); |
| } catch (IllegalStateException expected) { |
| } |
| entity.advance(); |
| assertEquals(EntityState.T_END_OF_STREAM, entity.getState()); |
| try { |
| entity.advance(); |
| fail("IllegalStateException should have been thrown"); |
| } catch (IllegalStateException expected) { |
| } |
| } |
| |
| public void testObsoleteSyntaxEntity() throws Exception { |
| String message = |
| "To : Road Runner <runner@example.org>\r\n" + |
| "From : Wile E. Cayote <wile@example.org>\r\n" + |
| "Date :Tue, 12 Feb 2008 17:34:09 +0000 (GMT)\r\n" + |
| "Subject :Mail\r\n" + |
| " \r\n" + |
| " with a folded subject \r\n" + |
| "Content-Type: text/plain\r\n" + |
| "\r\n" + |
| "a very important message"; |
| byte[] raw = message.getBytes("US-ASCII"); |
| ByteArrayInputStream instream = new ByteArrayInputStream(raw); |
| LineNumberInputStream lineInput = new LineNumberInputStream(instream); |
| BufferedLineReaderInputStream rawstream = new BufferedLineReaderInputStream(lineInput, 12); |
| |
| MimeEntity entity = new MimeEntity( |
| lineInput, |
| rawstream, new DefaultBodyDescriptor()); |
| |
| |
| assertEquals(EntityState.T_START_MESSAGE, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_START_HEADER, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("To", entity.getField().getName()); |
| assertEquals("Road Runner <runner@example.org>", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("From", entity.getField().getName()); |
| assertEquals("Wile E. Cayote <wile@example.org>", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("Date", entity.getField().getName()); |
| assertEquals("Tue, 12 Feb 2008 17:34:09 +0000 (GMT)", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("Subject", entity.getField().getName()); |
| assertEquals("Mail with a folded subject ", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("Content-Type", entity.getField().getName()); |
| assertEquals("text/plain", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_END_HEADER, entity.getState()); |
| try { |
| entity.getField().getName(); |
| fail("IllegalStateException should have been thrown"); |
| } catch (IllegalStateException expected) { |
| } |
| try { |
| entity.getField().getBody(); |
| fail("IllegalStateException should have been thrown"); |
| } catch (IllegalStateException expected) { |
| } |
| |
| entity.advance(); |
| assertEquals(EntityState.T_BODY, entity.getState()); |
| assertEquals("a very important message", IOUtils.toString(entity.getContentStream())); |
| entity.advance(); |
| assertEquals(EntityState.T_END_MESSAGE, entity.getState()); |
| try { |
| entity.getContentStream(); |
| fail("IllegalStateException should have been thrown"); |
| } catch (IllegalStateException expected) { |
| } |
| entity.advance(); |
| assertEquals(EntityState.T_END_OF_STREAM, entity.getState()); |
| try { |
| entity.advance(); |
| fail("IllegalStateException should have been thrown"); |
| } catch (IllegalStateException expected) { |
| } |
| } |
| |
| public void testMultipartEntity() throws Exception { |
| String message = |
| "To: Road Runner <runner@example.org>\r\n" + |
| "From: Wile E. Cayote <wile@example.org>\r\n" + |
| "Date: Tue, 12 Feb 2008 17:34:09 +0000 (GMT)\r\n" + |
| "Subject: Mail\r\n" + |
| "Content-Type: multipart/mixed;boundary=1729\r\n" + |
| "\r\n" + |
| "Hello!\r\n" + |
| "--1729\r\n" + |
| "Content-Type: text/plain; charset=US-ASCII\r\n" + |
| "\r\n" + |
| "blah blah blah\r\n" + |
| "--1729\r\n" + |
| "Content-Type: text/plain; charset=US-ASCII\r\n" + |
| "\r\n" + |
| "yada yada yada\r\n" + |
| "--1729--\r\n" + |
| "Goodbye!"; |
| byte[] raw = message.getBytes("US-ASCII"); |
| ByteArrayInputStream instream = new ByteArrayInputStream(raw); |
| LineNumberInputStream lineInput = new LineNumberInputStream(instream); |
| BufferedLineReaderInputStream rawstream = new BufferedLineReaderInputStream(lineInput, 24); |
| |
| MimeEntity entity = new MimeEntity( |
| lineInput, |
| rawstream, new DefaultBodyDescriptor()); |
| |
| assertEquals(EntityState.T_START_MESSAGE, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_START_HEADER, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("To", entity.getField().getName()); |
| assertEquals("Road Runner <runner@example.org>", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("From", entity.getField().getName()); |
| assertEquals("Wile E. Cayote <wile@example.org>", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("Date", entity.getField().getName()); |
| assertEquals("Tue, 12 Feb 2008 17:34:09 +0000 (GMT)", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("Subject", entity.getField().getName()); |
| assertEquals("Mail", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("Content-Type", entity.getField().getName()); |
| assertEquals("multipart/mixed;boundary=1729", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_END_HEADER, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_START_MULTIPART, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_PREAMBLE, entity.getState()); |
| assertEquals("Hello!", IOUtils.toString(entity.getContentStream())); |
| |
| EntityStateMachine p1 = entity.advance(); |
| assertNotNull(p1); |
| |
| assertEquals(EntityState.T_START_BODYPART, p1.getState()); |
| p1.advance(); |
| assertEquals(EntityState.T_START_HEADER, p1.getState()); |
| p1.advance(); |
| assertEquals(EntityState.T_FIELD, p1.getState()); |
| assertEquals("Content-Type", p1.getField().getName()); |
| assertEquals("text/plain; charset=US-ASCII", p1.getField().getBody()); |
| p1.advance(); |
| assertEquals(EntityState.T_END_HEADER, p1.getState()); |
| p1.advance(); |
| assertEquals(EntityState.T_BODY, p1.getState()); |
| assertEquals("blah blah blah", IOUtils.toString(p1.getContentStream())); |
| p1.advance(); |
| assertEquals(EntityState.T_END_BODYPART, p1.getState()); |
| p1.advance(); |
| assertEquals(EntityState.T_END_OF_STREAM, p1.getState()); |
| |
| EntityStateMachine p2 = entity.advance(); |
| assertNotNull(p2); |
| |
| assertEquals(EntityState.T_START_BODYPART, p2.getState()); |
| p2.advance(); |
| assertEquals(EntityState.T_START_HEADER, p2.getState()); |
| p2.advance(); |
| assertEquals(EntityState.T_FIELD, p2.getState()); |
| assertEquals("Content-Type", p2.getField().getName()); |
| assertEquals("text/plain; charset=US-ASCII", p2.getField().getBody()); |
| p2.advance(); |
| assertEquals(EntityState.T_END_HEADER, p2.getState()); |
| p2.advance(); |
| assertEquals(EntityState.T_BODY, p2.getState()); |
| assertEquals("yada yada yada", IOUtils.toString(p2.getContentStream())); |
| p2.advance(); |
| assertEquals(EntityState.T_END_BODYPART, p2.getState()); |
| p2.advance(); |
| assertEquals(EntityState.T_END_OF_STREAM, p2.getState()); |
| |
| entity.advance(); |
| assertEquals(EntityState.T_EPILOGUE, entity.getState()); |
| assertEquals("Goodbye!", IOUtils.toString(entity.getContentStream())); |
| entity.advance(); |
| assertEquals(EntityState.T_END_MULTIPART, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_END_MESSAGE, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_END_OF_STREAM, entity.getState()); |
| } |
| |
| public void testRawEntity() throws Exception { |
| String message = |
| "To: Road Runner <runner@example.org>\r\n" + |
| "From: Wile E. Cayote <wile@example.org>\r\n" + |
| "Date: Tue, 12 Feb 2008 17:34:09 +0000 (GMT)\r\n" + |
| "Subject: Mail\r\n" + |
| "Content-Type: multipart/mixed;boundary=1729\r\n" + |
| "\r\n" + |
| "Hello!\r\n" + |
| "--1729\r\n" + |
| "Content-Type: text/plain; charset=US-ASCII\r\n" + |
| "\r\n" + |
| "blah blah blah\r\n" + |
| "--1729\r\n" + |
| "Content-Type: text/plain; charset=US-ASCII\r\n" + |
| "\r\n" + |
| "yada yada yada\r\n" + |
| "--1729--\r\n" + |
| "Goodbye!"; |
| byte[] raw = message.getBytes("US-ASCII"); |
| ByteArrayInputStream instream = new ByteArrayInputStream(raw); |
| LineNumberInputStream lineInput = new LineNumberInputStream(instream); |
| BufferedLineReaderInputStream rawstream = new BufferedLineReaderInputStream(lineInput, 24); |
| |
| MimeEntity entity = new MimeEntity( |
| lineInput, |
| rawstream, new DefaultBodyDescriptor()); |
| |
| entity.setRecursionMode(RecursionMode.M_RAW); |
| |
| assertEquals(EntityState.T_START_MESSAGE, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_START_HEADER, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("To", entity.getField().getName()); |
| assertEquals("Road Runner <runner@example.org>", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("From", entity.getField().getName()); |
| assertEquals("Wile E. Cayote <wile@example.org>", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("Date", entity.getField().getName()); |
| assertEquals("Tue, 12 Feb 2008 17:34:09 +0000 (GMT)", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("Subject", entity.getField().getName()); |
| assertEquals("Mail", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("Content-Type", entity.getField().getName()); |
| assertEquals("multipart/mixed;boundary=1729", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_END_HEADER, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_START_MULTIPART, entity.getState()); |
| |
| entity.advance(); |
| assertEquals(EntityState.T_PREAMBLE, entity.getState()); |
| assertEquals("Hello!", IOUtils.toString(entity.getContentStream())); |
| |
| EntityStateMachine p1 = entity.advance(); |
| assertNotNull(p1); |
| |
| assertEquals(EntityState.T_RAW_ENTITY, p1.getState()); |
| assertNull(p1.getBodyDescriptor()); |
| assertNull(p1.getField()); |
| assertEquals( |
| "Content-Type: text/plain; charset=US-ASCII\r\n" + |
| "\r\n" + |
| "blah blah blah", IOUtils.toString(p1.getContentStream())); |
| p1.advance(); |
| assertEquals(EntityState.T_END_OF_STREAM, p1.getState()); |
| |
| EntityStateMachine p2 = entity.advance(); |
| assertNotNull(p2); |
| |
| assertEquals(EntityState.T_RAW_ENTITY, p2.getState()); |
| assertNull(p2.getBodyDescriptor()); |
| assertNull(p2.getField()); |
| assertEquals( |
| "Content-Type: text/plain; charset=US-ASCII\r\n" + |
| "\r\n" + |
| "yada yada yada", IOUtils.toString(p2.getContentStream())); |
| p2.advance(); |
| assertEquals(EntityState.T_END_OF_STREAM, p2.getState()); |
| |
| entity.advance(); |
| assertEquals(EntityState.T_EPILOGUE, entity.getState()); |
| assertEquals("Goodbye!", IOUtils.toString(entity.getContentStream())); |
| entity.advance(); |
| assertEquals(EntityState.T_END_MULTIPART, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_END_MESSAGE, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_END_OF_STREAM, entity.getState()); |
| } |
| |
| public void testMaxLineLimitCheck() throws Exception { |
| MimeEntityConfig config = new MimeEntityConfig(); |
| config.setMaxLineLen(50); |
| |
| String message = |
| "To: Road Runner <runner@example.org>\r\n" + |
| "From: Wile E. Cayote <wile@example.org>\r\n" + |
| "Date: Tue, 12 Feb 2008 17:34:09 +0000 (GMT)\r\n" + |
| "Subject: Mail\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "Content-Type: text/plain\r\n" + |
| "\r\n" + |
| "a very important message"; |
| byte[] raw = message.getBytes("US-ASCII"); |
| ByteArrayInputStream instream = new ByteArrayInputStream(raw); |
| LineNumberInputStream lineInput = new LineNumberInputStream(instream); |
| BufferedLineReaderInputStream rawstream = new BufferedLineReaderInputStream(lineInput, 12, config.getMaxLineLen()); |
| |
| MimeEntity entity = new MimeEntity( |
| lineInput, |
| rawstream, |
| config, |
| new DefaultBodyDescriptor()); |
| |
| assertEquals(EntityState.T_START_MESSAGE, entity.getState()); |
| entity.advance(); // advances to T_START_HEADER |
| assertEquals(EntityState.T_START_HEADER, entity.getState()); |
| entity.advance(); // reads To: into field buffer, From: into line buffer |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| entity.advance(); // reads Date: into line buffer |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| entity.advance(); // reads Subject: into line buffer |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| try { |
| entity.advance(); // reads DoS: into line buffer |
| fail("MimeException caused by MaxLineLimitException should have been thrown"); |
| } catch (MimeException expected) { |
| assertTrue(expected.getCause() instanceof MaxLineLimitException); |
| } |
| } |
| |
| public void testMaxHeaderLimitCheckFoldedLines() throws Exception { |
| String message = |
| "To: Road Runner <runner@example.org>\r\n" + |
| "From: Wile E. Cayote <wile@example.org>\r\n" + |
| "Date: Tue, 12 Feb 2008 17:34:09 +0000 (GMT)\r\n" + |
| "Subject: Mail\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| " xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| " xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| " xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| " xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| " xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| " xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| " xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| " xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| " xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| " xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| " xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "Content-Type: text/plain\r\n" + |
| "\r\n" + |
| "a very important message"; |
| byte[] raw = message.getBytes("US-ASCII"); |
| ByteArrayInputStream instream = new ByteArrayInputStream(raw); |
| LineNumberInputStream lineInput = new LineNumberInputStream(instream); |
| BufferedLineReaderInputStream rawstream = new BufferedLineReaderInputStream(lineInput, 12); |
| |
| MimeEntityConfig config = new MimeEntityConfig(); |
| config.setMaxLineLen(100); |
| config.setMaxHeaderLen(200); |
| MimeEntity entity = new MimeEntity( |
| lineInput, |
| rawstream, |
| config, |
| new DefaultBodyDescriptor()); |
| |
| assertEquals(EntityState.T_START_MESSAGE, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_START_HEADER, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| try { |
| entity.advance(); |
| fail("MimeException caused by MaxLineLimitException should have been thrown"); |
| } catch (MaxHeaderLengthLimitException expected) { |
| } |
| } |
| |
| public void testMaxHeaderLengthMayExceedMaxLineLength() throws Exception { |
| MimeEntityConfig config = new MimeEntityConfig(); |
| config.setMaxLineLen(50); |
| config.setMaxHeaderLen(130); |
| |
| String message = |
| "To: Road Runner <runner@example.org>\r\n" + |
| "From: Wile E. Cayote <wile@example.org>\r\n" + |
| "Date: Tue, 12 Feb 2008 17:34:09 +0000 (GMT)\r\n" + |
| "Subject: Mail\r\n" + |
| "X-LongHeader: xxxxxxxxxxxxxxxxxxxxxxx\r\n" + |
| " xxxxxxxxxxxxxxxxxxxxxxx\r\n" + |
| " xxxxxxxxxxxxxxxxxxxxxxx\r\n" + |
| " xxxxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "Content-Type: text/plain\r\n" + |
| "\r\n" + |
| "a very important message"; |
| byte[] raw = message.getBytes("US-ASCII"); |
| ByteArrayInputStream instream = new ByteArrayInputStream(raw); |
| LineNumberInputStream lineInput = new LineNumberInputStream(instream); |
| BufferedLineReaderInputStream rawstream = new BufferedLineReaderInputStream(lineInput, 12, config.getMaxLineLen()); |
| |
| MimeEntity entity = new MimeEntity( |
| lineInput, |
| rawstream, |
| config, |
| new DefaultBodyDescriptor()); |
| |
| assertEquals(EntityState.T_START_MESSAGE, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_START_HEADER, entity.getState()); |
| for (int i = 0; i < 6; i++) { |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| } |
| entity.advance(); |
| assertEquals(EntityState.T_END_HEADER, entity.getState()); |
| } |
| |
| public void testMaxHeaderCount() throws Exception { |
| String message = |
| "To: Road Runner <runner@example.org>\r\n" + |
| "From: Wile E. Cayote <wile@example.org>\r\n" + |
| "Date: Tue, 12 Feb 2008 17:34:09 +0000 (GMT)\r\n" + |
| "Subject: Mail\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "DoS: xxxxxxxxxxxxxxxxxxxxx\r\n" + |
| "Content-Type: text/plain\r\n" + |
| "\r\n" + |
| "a very important message"; |
| byte[] raw = message.getBytes("US-ASCII"); |
| ByteArrayInputStream instream = new ByteArrayInputStream(raw); |
| LineNumberInputStream lineInput = new LineNumberInputStream(instream); |
| BufferedLineReaderInputStream rawstream = new BufferedLineReaderInputStream(lineInput, 12); |
| |
| MimeEntityConfig config = new MimeEntityConfig(); |
| config.setMaxHeaderCount(20); |
| MimeEntity entity = new MimeEntity( |
| lineInput, |
| rawstream, |
| config, |
| new DefaultBodyDescriptor()); |
| |
| assertEquals(EntityState.T_START_MESSAGE, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_START_HEADER, entity.getState()); |
| |
| for (int i = 0; i < 20; i++) { |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| } |
| try { |
| entity.advance(); |
| fail("MaxHeaderLimitException should have been thrown"); |
| } catch (MaxHeaderLimitException expected) { |
| } |
| } |
| |
| public void testMaxContentLimitCheck() throws Exception { |
| String message = |
| "To: Road Runner <runner@example.org>\r\n" + |
| "From: Wile E. Cayote <wile@example.org>\r\n" + |
| "Date: Tue, 12 Feb 2008 17:34:09 +0000 (GMT)\r\n" + |
| "Subject: Mail\r\n" + |
| "Content-Type: text/plain\r\n" + |
| "\r\n" + |
| "DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS\r\n" + |
| "DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS\r\n" + |
| "DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS\r\n" + |
| "DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS\r\n" + |
| "DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS\r\n" + |
| "DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS\r\n" + |
| "DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS\r\n" + |
| "DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS\r\n" + |
| "DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS\r\n" + |
| "DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS DoS\r\n"; |
| byte[] raw = message.getBytes("US-ASCII"); |
| ByteArrayInputStream instream = new ByteArrayInputStream(raw); |
| LineNumberInputStream lineInput = new LineNumberInputStream(instream); |
| BufferedLineReaderInputStream rawstream = new BufferedLineReaderInputStream(lineInput, 12); |
| |
| MimeEntityConfig config = new MimeEntityConfig(); |
| config.setMaxContentLen(100); |
| MimeEntity entity = new MimeEntity( |
| lineInput, |
| rawstream, |
| config, |
| new DefaultBodyDescriptor()); |
| |
| assertEquals(EntityState.T_START_MESSAGE, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_START_HEADER, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_END_HEADER, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_BODY, entity.getState()); |
| try { |
| IOUtils.toByteArray(entity.getContentStream()); |
| fail("IOException should have been thrown"); |
| } catch (IOException expected) { |
| } |
| } |
| |
| public void testSkipFields() throws Exception { |
| String message = |
| "To: Road Runner <runner@example.org>\r\n" + |
| "From: Wile E. Cayote <wile@example.org>\r\n" + |
| "Date: Tue, 12 Feb 2008 17:34:09 +0000 (GMT)\r\n" + |
| "Subject: Mail\r\n" + |
| "Content-Type: text/plain\r\n" + |
| "\r\n" + |
| "a very important message"; |
| byte[] raw = message.getBytes("US-ASCII"); |
| ByteArrayInputStream instream = new ByteArrayInputStream(raw); |
| LineNumberInputStream lineInput = new LineNumberInputStream(instream); |
| BufferedLineReaderInputStream rawstream = new BufferedLineReaderInputStream(lineInput, 12); |
| |
| DefaultFieldBuilder fieldBuilder = new DefaultFieldBuilder(-1) { |
| |
| @Override |
| public RawField build() throws MimeException { |
| RawField raw = super.build(); |
| String name = raw.getName().toLowerCase(Locale.US); |
| if (name.equals("content-type") || name.equals("subject")) { |
| return raw; |
| } else { |
| return null; |
| } |
| } |
| |
| }; |
| |
| MimeEntity entity = new MimeEntity( |
| lineInput, |
| rawstream, fieldBuilder, new DefaultBodyDescriptor()); |
| |
| |
| assertEquals(EntityState.T_START_MESSAGE, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_START_HEADER, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("Subject", entity.getField().getName()); |
| assertEquals("Mail", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_FIELD, entity.getState()); |
| assertEquals("Content-Type", entity.getField().getName()); |
| assertEquals("text/plain", entity.getField().getBody()); |
| entity.advance(); |
| assertEquals(EntityState.T_END_HEADER, entity.getState()); |
| entity.advance(); |
| assertEquals(EntityState.T_BODY, entity.getState()); |
| assertEquals("a very important message", IOUtils.toString(entity.getContentStream())); |
| entity.advance(); |
| assertEquals(EntityState.T_END_MESSAGE, entity.getState()); |
| } |
| |
| } |