1 // Copyright 2000-2005 the Contributors, as shown in the revision logs.
2 // Licensed under the Apache Public Source License 2.0 ("the License").
3 // You may not use this file except in compliance with the License.
6 import static org.ibex.mail.MailException.*;
7 import org.ibex.crypto.*;
8 import org.ibex.util.*;
9 import org.ibex.mail.protocol.*;
12 import org.ibex.io.Fountain;
17 // FIXME: this is important: folded headers: can insert CRLF anywhere that whitespace appears (before the whitespace)
18 public abstract class Headers extends JS.Immutable implements Fountain {
20 public Headers set(String k, String v) {
21 Stream stream = getStream();
22 StringBuffer all = new StringBuffer();
25 for(String s = stream.readln(); s != null && !s.equals(""); s = stream.readln()) {
26 if (!Character.isSpace(s.charAt(0))) {
27 if (s.indexOf(':') == -1) throw new Malformed("Header line does not contain colon: " + s);
28 key = s.substring(0, s.indexOf(':')).toLowerCase();
30 if (key.toLowerCase().equals(k.toLowerCase())) {
31 if (v != null) { all.append(k + ": " + v + "\r\n"); lines++; v = null; }
40 all.append(k + ": " + v + "\r\n");
43 return new Original(new Stream(all.toString()));
47 public abstract String getString();
49 public abstract String get(String s);
50 public abstract java.util.Enumeration names();
52 public Headers set(String[] keyval) {
54 for(int i=0; i<keyval.length; i+=2)
55 ret = ret.set(keyval[i], keyval[i+1]);
58 public Headers remove(String key) { return set(key, null); /* FIXME */ }
60 public String getLowerCaseTrimmed(String s) {
62 return ret==null ? null : ret.toLowerCase();
65 public static class Original extends Headers {
66 private final Hash head = new Hash();
67 private final Hash headModified = new Hash();
69 public final boolean mime;
71 private StringFountain fountain;
73 public String getString() { return raw; }
74 public Stream getStream() { return fountain.getStream(); }
75 public int getLength() { return fountain.getLength(); }
76 public int getNumLines() { return fountain.getNumLines(); }
77 public Stream getStreamWithCRLF() { return new Stream(raw+"\r\n"); }
78 public JS get(JS s) throws JSExn { return JSU.S(get(JSU.toString(s).toLowerCase())); }
79 public java.util.Enumeration names() { return head.keys(); }
81 public String get(String s) { return (String)head.get(s.toLowerCase()); }
82 public Original(Stream stream) throws Malformed { this(stream, false); }
83 public Original(Stream stream, boolean assumeMime) throws Malformed {
84 StringBuffer all = new StringBuffer();
87 for(String s = stream.readln(); s != null && !s.equals(""); s = stream.readln()) {
91 if (Character.isSpace(s.charAt(0))) {
92 if (key == null) throw new Malformed("Message began with a blank line; no headers");
93 head.put(key, head.get(key) + " " + s.trim());
96 if (s.indexOf(':') == -1) throw new Malformed("Header line does not contain colon: " + s);
97 key = s.substring(0, s.indexOf(':')).toLowerCase();
98 for(int i=0; i<key.length(); i++)
99 if (key.charAt(i) < 33 || key.charAt(i) > 126)
100 throw new Malformed("Header key \""+key+"\" contains invalid character \"" + key.charAt(i) + "\"");
101 String val = s.substring(s.indexOf(':') + 1).trim();
102 if (get(key) != null) val = get(key) + " " + val; // just append it to the previous one;
105 this.raw = all.toString();
106 this.fountain = new Fountain.StringFountain(this.raw);
108 this.mime = assumeMime | (get("mime-version") != null && get("mime-version").trim().equals("1.0"));
110 java.util.Enumeration e = head.keys();
112 String k = (String)e.next();
113 String v = (String)head.get(k);
114 if (mime) k = Encode.RFC2047.decode(k);
116 if (mime) v = Encode.RFC2047.decode(v);
123 // Helpers //////////////////////////////////////////////////////////////////////////////
125 public static Stream skip(Stream stream) {
126 for(String s = stream.readln(); s!=null && s.length() > 0;) s = stream.readln();
130 public static String uncomment(String val) {
131 boolean inquotes = false;
132 for(int i=0; i<val.length(); i++) {
133 if (val.charAt(i) == '\"') inquotes = !inquotes;
134 if (val.charAt(i) == '(' && !inquotes)
135 val = val.substring(0, i) + val.substring(val.indexOf(i--, ')') + 1);