/****************************************************************
* 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 FallbackBodyDescriptorBuilder());
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 FallbackBodyDescriptorBuilder());
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 FallbackBodyDescriptorBuilder());
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 FallbackBodyDescriptorBuilder());
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 {
MimeConfig config = new MimeConfig();
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 FallbackBodyDescriptorBuilder());
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);
MimeConfig config = new MimeConfig();
config.setMaxLineLen(100);
config.setMaxHeaderLen(200);
MimeEntity entity = new MimeEntity(
lineInput,
rawstream,
config,
new FallbackBodyDescriptorBuilder());
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 {
MimeConfig config = new MimeConfig();
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 FallbackBodyDescriptorBuilder());
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);
MimeConfig config = new MimeConfig();
config.setMaxHeaderCount(20);
MimeEntity entity = new MimeEntity(
lineInput,
rawstream,
config,
new FallbackBodyDescriptorBuilder());
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);
MimeConfig config = new MimeConfig();
config.setMaxContentLen(100);
MimeEntity entity = new MimeEntity(
lineInput,
rawstream,
config,
new FallbackBodyDescriptorBuilder());
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 FallbackBodyDescriptorBuilder());
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());
}
}