2008-11-01 17:32:06 -04:00
|
|
|
|
2009-12-14 21:50:53 -05:00
|
|
|
package com.fsck.k9.mail.internet;
|
2008-11-01 17:32:06 -04:00
|
|
|
|
2010-05-19 14:17:06 -04:00
|
|
|
import com.fsck.k9.helper.Utility;
|
2009-12-09 22:16:42 -05:00
|
|
|
|
2008-11-01 17:32:06 -04:00
|
|
|
import java.io.BufferedWriter;
|
|
|
|
import java.io.IOException;
|
|
|
|
import java.io.OutputStream;
|
|
|
|
import java.io.OutputStreamWriter;
|
2010-12-23 04:58:13 -05:00
|
|
|
import java.nio.charset.Charset;
|
2010-05-21 11:34:29 -04:00
|
|
|
import java.util.*;
|
2008-11-01 17:32:06 -04:00
|
|
|
|
2011-02-06 17:09:48 -05:00
|
|
|
public class MimeHeader {
|
2010-08-07 11:10:07 -04:00
|
|
|
private static final String[] EMPTY_STRING_ARRAY = new String[0];
|
|
|
|
|
2008-11-01 17:32:06 -04:00
|
|
|
/**
|
|
|
|
* Application specific header that contains Store specific information about an attachment.
|
|
|
|
* In IMAP this contains the IMAP BODYSTRUCTURE part id so that the ImapStore can later
|
|
|
|
* retrieve the attachment at will from the server.
|
|
|
|
* The info is recorded from this header on LocalStore.appendMessages and is put back
|
|
|
|
* into the MIME data by LocalStore.fetch.
|
|
|
|
*/
|
|
|
|
public static final String HEADER_ANDROID_ATTACHMENT_STORE_DATA = "X-Android-Attachment-StoreData";
|
|
|
|
|
|
|
|
public static final String HEADER_CONTENT_TYPE = "Content-Type";
|
|
|
|
public static final String HEADER_CONTENT_TRANSFER_ENCODING = "Content-Transfer-Encoding";
|
|
|
|
public static final String HEADER_CONTENT_DISPOSITION = "Content-Disposition";
|
2010-07-11 09:44:16 -04:00
|
|
|
public static final String HEADER_CONTENT_ID = "Content-ID";
|
2008-11-01 17:32:06 -04:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Fields that should be omitted when writing the header using writeTo()
|
|
|
|
*/
|
2011-02-06 17:09:48 -05:00
|
|
|
private static final String[] writeOmitFields = {
|
2008-11-01 17:32:06 -04:00
|
|
|
// HEADER_ANDROID_ATTACHMENT_DOWNLOADED,
|
|
|
|
// HEADER_ANDROID_ATTACHMENT_ID,
|
|
|
|
HEADER_ANDROID_ATTACHMENT_STORE_DATA
|
|
|
|
};
|
|
|
|
|
2014-10-04 06:45:45 -04:00
|
|
|
private List<Field> mFields = new ArrayList<Field>();
|
2010-12-23 04:58:13 -05:00
|
|
|
private String mCharset = null;
|
2008-11-01 17:32:06 -04:00
|
|
|
|
2011-02-06 17:09:48 -05:00
|
|
|
public void clear() {
|
2008-11-01 17:32:06 -04:00
|
|
|
mFields.clear();
|
|
|
|
}
|
|
|
|
|
2011-02-06 17:09:48 -05:00
|
|
|
public String getFirstHeader(String name) {
|
2008-11-01 17:32:06 -04:00
|
|
|
String[] header = getHeader(name);
|
2011-02-06 17:09:48 -05:00
|
|
|
if (header == null) {
|
2008-11-01 17:32:06 -04:00
|
|
|
return null;
|
|
|
|
}
|
|
|
|
return header[0];
|
|
|
|
}
|
|
|
|
|
2011-02-06 17:09:48 -05:00
|
|
|
public void addHeader(String name, String value) {
|
2014-11-13 16:40:58 -05:00
|
|
|
Field field = Field.newNameValueField(name, MimeUtility.foldAndEncode(value));
|
|
|
|
mFields.add(field);
|
|
|
|
}
|
|
|
|
|
|
|
|
void addRawHeader(String name, String raw) {
|
|
|
|
Field field = Field.newRawField(name, raw);
|
|
|
|
mFields.add(field);
|
2008-11-01 17:32:06 -04:00
|
|
|
}
|
|
|
|
|
2011-02-06 17:09:48 -05:00
|
|
|
public void setHeader(String name, String value) {
|
|
|
|
if (name == null || value == null) {
|
2008-11-01 17:32:06 -04:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
removeHeader(name);
|
|
|
|
addHeader(name, value);
|
|
|
|
}
|
|
|
|
|
2011-02-06 17:09:48 -05:00
|
|
|
public Set<String> getHeaderNames() {
|
2011-11-14 23:32:46 -05:00
|
|
|
Set<String> names = new LinkedHashSet<String>();
|
2011-02-06 17:09:48 -05:00
|
|
|
for (Field field : mFields) {
|
2014-11-13 16:40:58 -05:00
|
|
|
names.add(field.getName());
|
2009-06-08 23:11:35 -04:00
|
|
|
}
|
|
|
|
return names;
|
|
|
|
}
|
|
|
|
|
2011-02-06 17:09:48 -05:00
|
|
|
public String[] getHeader(String name) {
|
2014-10-04 06:45:45 -04:00
|
|
|
List<String> values = new ArrayList<String>();
|
2011-02-06 17:09:48 -05:00
|
|
|
for (Field field : mFields) {
|
2014-11-13 16:40:58 -05:00
|
|
|
if (field.getName().equalsIgnoreCase(name)) {
|
|
|
|
values.add(field.getValue());
|
2010-11-26 23:03:15 -05:00
|
|
|
}
|
|
|
|
}
|
2011-10-06 12:28:14 -04:00
|
|
|
if (values.isEmpty()) {
|
2008-11-01 17:32:06 -04:00
|
|
|
return null;
|
|
|
|
}
|
2010-08-07 11:10:07 -04:00
|
|
|
return values.toArray(EMPTY_STRING_ARRAY);
|
2008-11-01 17:32:06 -04:00
|
|
|
}
|
|
|
|
|
2011-02-06 17:09:48 -05:00
|
|
|
public void removeHeader(String name) {
|
2014-10-04 06:45:45 -04:00
|
|
|
List<Field> removeFields = new ArrayList<Field>();
|
2011-02-06 17:09:48 -05:00
|
|
|
for (Field field : mFields) {
|
2014-11-13 16:40:58 -05:00
|
|
|
if (field.getName().equalsIgnoreCase(name)) {
|
2010-11-26 23:03:15 -05:00
|
|
|
removeFields.add(field);
|
|
|
|
}
|
2008-11-01 17:32:06 -04:00
|
|
|
}
|
2010-11-26 23:03:15 -05:00
|
|
|
mFields.removeAll(removeFields);
|
2008-11-01 17:32:06 -04:00
|
|
|
}
|
|
|
|
|
2011-02-06 17:09:48 -05:00
|
|
|
public void writeTo(OutputStream out) throws IOException {
|
2008-11-01 17:32:06 -04:00
|
|
|
BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(out), 1024);
|
2011-02-06 17:09:48 -05:00
|
|
|
for (Field field : mFields) {
|
2014-11-13 16:40:58 -05:00
|
|
|
if (!Utility.arrayContains(writeOmitFields, field.getName())) {
|
|
|
|
if (field.hasRawData()) {
|
|
|
|
writer.write(field.getRaw());
|
|
|
|
} else {
|
|
|
|
writeNameValueField(writer, field);
|
2009-07-02 01:49:09 -04:00
|
|
|
}
|
2011-11-06 21:14:35 -05:00
|
|
|
writer.write("\r\n");
|
2008-11-01 17:32:06 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
writer.flush();
|
|
|
|
}
|
|
|
|
|
2014-11-13 16:40:58 -05:00
|
|
|
private void writeNameValueField(BufferedWriter writer, Field field) throws IOException {
|
|
|
|
String value = field.getValue();
|
|
|
|
|
|
|
|
if (hasToBeEncoded(value)) {
|
|
|
|
Charset charset = null;
|
|
|
|
|
|
|
|
if (mCharset != null) {
|
|
|
|
charset = Charset.forName(mCharset);
|
|
|
|
}
|
|
|
|
value = EncoderUtil.encodeEncodedWord(field.getValue(), charset);
|
|
|
|
}
|
|
|
|
|
|
|
|
writer.write(field.getName());
|
|
|
|
writer.write(": ");
|
|
|
|
writer.write(value);
|
|
|
|
}
|
|
|
|
|
2012-04-30 14:58:02 -04:00
|
|
|
// encode non printable characters except LF/CR/TAB codes.
|
2014-09-22 15:52:59 -04:00
|
|
|
private boolean hasToBeEncoded(String text) {
|
2011-02-06 17:09:48 -05:00
|
|
|
for (int i = 0; i < text.length(); i++) {
|
2009-07-02 01:49:09 -04:00
|
|
|
char c = text.charAt(i);
|
2012-07-06 08:23:21 -04:00
|
|
|
if ((c < 0x20 || 0x7e < c) && // non printable
|
|
|
|
(c != 0x0a && c != 0x0d && c != 0x09)) { // non LF/CR/TAB
|
|
|
|
return true;
|
2009-07-02 01:49:09 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2014-09-22 15:52:59 -04:00
|
|
|
private static class Field {
|
|
|
|
private final String name;
|
|
|
|
private final String value;
|
2014-11-13 16:40:58 -05:00
|
|
|
private final String raw;
|
|
|
|
|
|
|
|
public static Field newNameValueField(String name, String value) {
|
|
|
|
if (value == null) {
|
|
|
|
throw new NullPointerException("Argument 'value' cannot be null");
|
|
|
|
}
|
|
|
|
|
|
|
|
return new Field(name, value, null);
|
|
|
|
}
|
|
|
|
|
|
|
|
public static Field newRawField(String name, String raw) {
|
|
|
|
if (raw == null) {
|
|
|
|
throw new NullPointerException("Argument 'raw' cannot be null");
|
|
|
|
}
|
|
|
|
if (name != null && !raw.startsWith(name + ":")) {
|
|
|
|
throw new IllegalArgumentException("The value of 'raw' needs to start with the supplied field name " +
|
|
|
|
"followed by a colon");
|
|
|
|
}
|
|
|
|
|
|
|
|
return new Field(name, null, raw);
|
|
|
|
}
|
|
|
|
|
|
|
|
private Field(String name, String value, String raw) {
|
|
|
|
if (name == null) {
|
|
|
|
throw new NullPointerException("Argument 'name' cannot be null");
|
|
|
|
}
|
2008-11-01 17:32:06 -04:00
|
|
|
|
|
|
|
this.name = name;
|
|
|
|
this.value = value;
|
2014-11-13 16:40:58 -05:00
|
|
|
this.raw = raw;
|
|
|
|
}
|
|
|
|
|
|
|
|
public String getName() {
|
|
|
|
return name;
|
|
|
|
}
|
|
|
|
|
|
|
|
public String getValue() {
|
|
|
|
if (value != null) {
|
|
|
|
return value;
|
|
|
|
}
|
|
|
|
|
|
|
|
int delimiterIndex = raw.indexOf(':');
|
|
|
|
if (delimiterIndex == raw.length() - 1) {
|
|
|
|
return "";
|
|
|
|
}
|
|
|
|
|
|
|
|
return raw.substring(delimiterIndex + 1).trim();
|
|
|
|
}
|
|
|
|
|
|
|
|
public String getRaw() {
|
|
|
|
return raw;
|
|
|
|
}
|
|
|
|
|
|
|
|
public boolean hasRawData() {
|
|
|
|
return raw != null;
|
2008-11-01 17:32:06 -04:00
|
|
|
}
|
2010-08-18 10:13:37 -04:00
|
|
|
|
2010-11-03 23:11:34 -04:00
|
|
|
@Override
|
2011-02-06 17:09:48 -05:00
|
|
|
public String toString() {
|
2014-11-13 16:40:58 -05:00
|
|
|
return (hasRawData()) ? getRaw() : getName() + ": " + getValue();
|
2010-08-29 12:57:13 -04:00
|
|
|
}
|
2008-11-01 17:32:06 -04:00
|
|
|
}
|
2010-12-23 04:58:13 -05:00
|
|
|
|
2011-03-22 03:07:32 -04:00
|
|
|
public void setCharset(String charset) {
|
2010-12-23 04:58:13 -05:00
|
|
|
mCharset = charset;
|
|
|
|
}
|
2012-01-03 20:27:51 -05:00
|
|
|
|
2014-09-22 15:52:59 -04:00
|
|
|
@Override
|
2012-01-03 20:27:51 -05:00
|
|
|
public MimeHeader clone() {
|
|
|
|
MimeHeader header = new MimeHeader();
|
|
|
|
header.mCharset = mCharset;
|
|
|
|
|
|
|
|
header.mFields = new ArrayList<Field>(mFields);
|
|
|
|
|
|
|
|
return header;
|
|
|
|
}
|
2008-11-01 17:32:06 -04:00
|
|
|
}
|