1 |
| |
2 |
| |
3 |
| |
4 |
| |
5 |
| |
6 |
| |
7 |
| |
8 |
| |
9 |
| |
10 |
| |
11 |
| |
12 |
| |
13 |
| |
14 |
| |
15 |
| |
16 |
| |
17 |
| |
18 |
| package javax.mail.internet; |
19 |
| |
20 |
| import java.io.IOException; |
21 |
| import java.io.InputStream; |
22 |
| import java.io.OutputStream; |
23 |
| import java.util.ArrayList; |
24 |
| import java.util.Arrays; |
25 |
| import java.util.Collections; |
26 |
| import java.util.Enumeration; |
27 |
| import java.util.HashSet; |
28 |
| import java.util.Iterator; |
29 |
| import java.util.LinkedHashMap; |
30 |
| import java.util.List; |
31 |
| import java.util.Map; |
32 |
| import java.util.Set; |
33 |
| |
34 |
| import javax.mail.Address; |
35 |
| import javax.mail.Header; |
36 |
| import javax.mail.MessagingException; |
37 |
| |
38 |
| |
39 |
| |
40 |
| |
41 |
| |
42 |
| |
43 |
| public class InternetHeaders { |
44 |
| |
45 |
| protected List headers = new ArrayList(); |
46 |
| |
47 |
| private transient String lastHeaderName; |
48 |
| |
49 |
| |
50 |
| |
51 |
| |
52 |
47
| public InternetHeaders() {
|
53 |
| |
54 |
47
| addHeader("Return-path", null);
|
55 |
47
| addHeader("Received", null);
|
56 |
47
| addHeader("Message-ID", null);
|
57 |
47
| addHeader("Resent-Date", null);
|
58 |
47
| addHeader("Date", null);
|
59 |
47
| addHeader("Resent-From", null);
|
60 |
47
| addHeader("From", null);
|
61 |
47
| addHeader("Reply-To", null);
|
62 |
47
| addHeader("Sender", null);
|
63 |
47
| addHeader("To", null);
|
64 |
47
| addHeader("Subject", null);
|
65 |
47
| addHeader("Cc", null);
|
66 |
47
| addHeader("In-Reply-To", null);
|
67 |
47
| addHeader("Resent-Message-Id", null);
|
68 |
47
| addHeader("Errors-To", null);
|
69 |
47
| addHeader("MIME-Version", null);
|
70 |
47
| addHeader("Content-Type", null);
|
71 |
47
| addHeader("Content-Transfer-Encoding", null);
|
72 |
47
| addHeader("Content-MD5", null);
|
73 |
| |
74 |
47
| addHeader(":", null);
|
75 |
47
| addHeader("Content-Length", null);
|
76 |
47
| addHeader("Status", null);
|
77 |
| } |
78 |
| |
79 |
| |
80 |
| |
81 |
| |
82 |
| |
83 |
| |
84 |
| |
85 |
| |
86 |
| |
87 |
| |
88 |
9
| public InternetHeaders(InputStream in) throws MessagingException {
|
89 |
9
| load(in);
|
90 |
| } |
91 |
| |
92 |
| |
93 |
| |
94 |
| |
95 |
| |
96 |
| |
97 |
| |
98 |
| |
99 |
| |
100 |
| |
101 |
10
| public void load(InputStream in) throws MessagingException {
|
102 |
10
| try {
|
103 |
10
| StringBuffer name = new StringBuffer(32);
|
104 |
10
| StringBuffer value = new StringBuffer(128);
|
105 |
10
| done: while (true) {
|
106 |
44
| int c = in.read();
|
107 |
44
| char ch = (char) c;
|
108 |
44
| if (c == -1) {
|
109 |
0
| break;
|
110 |
44
| } else if (c == 13) {
|
111 |
| |
112 |
10
| in.read();
|
113 |
10
| break;
|
114 |
34
| } else if (Character.isWhitespace(ch)) {
|
115 |
| |
116 |
0
| do {
|
117 |
0
| c = in.read();
|
118 |
0
| if (c == -1) {
|
119 |
0
| break done;
|
120 |
| } |
121 |
0
| ch = (char) c;
|
122 |
0
| } while (Character.isWhitespace(ch));
|
123 |
| } else { |
124 |
| |
125 |
34
| if (name.length() > 0) {
|
126 |
24
| addHeader(name.toString().trim(), value.toString().trim());
|
127 |
| } |
128 |
34
| name.setLength(0);
|
129 |
34
| value.setLength(0);
|
130 |
34
| while (true) {
|
131 |
422
| name.append((char) c);
|
132 |
422
| c = in.read();
|
133 |
422
| if (c == -1) {
|
134 |
0
| break done;
|
135 |
422
| } else if (c == ':') {
|
136 |
34
| break;
|
137 |
| } |
138 |
| } |
139 |
34
| c = in.read();
|
140 |
34
| if (c == -1) {
|
141 |
0
| break done;
|
142 |
| } |
143 |
| } |
144 |
| |
145 |
34
| while (c != 13) {
|
146 |
806
| ch = (char) c;
|
147 |
806
| value.append(ch);
|
148 |
806
| c = in.read();
|
149 |
806
| if (c == -1) {
|
150 |
0
| break done;
|
151 |
| } |
152 |
| } |
153 |
| |
154 |
34
| c = in.read();
|
155 |
34
| if (c == -1) {
|
156 |
0
| break;
|
157 |
| } |
158 |
| } |
159 |
10
| if (name.length() > 0) {
|
160 |
10
| addHeader(name.toString().trim(), value.toString().trim());
|
161 |
| } |
162 |
| } catch (IOException e) { |
163 |
0
| throw new MessagingException("Error loading headers", e);
|
164 |
| } |
165 |
| } |
166 |
| |
167 |
| |
168 |
| |
169 |
| |
170 |
| |
171 |
| |
172 |
| |
173 |
| |
174 |
| |
175 |
210
| public String[] getHeader(String name) {
|
176 |
210
| List accumulator = new ArrayList();
|
177 |
| |
178 |
210
| for (int i = 0; i < headers.size(); i++) {
|
179 |
4286
| InternetHeader header = (InternetHeader)headers.get(i);
|
180 |
4286
| if (header.getName().equalsIgnoreCase(name) && header.getValue() != null) {
|
181 |
122
| accumulator.add(header.getValue());
|
182 |
| } |
183 |
| } |
184 |
| |
185 |
| |
186 |
210
| if (accumulator.isEmpty()) {
|
187 |
102
| return null;
|
188 |
| } |
189 |
| |
190 |
| |
191 |
108
| return (String[])accumulator.toArray(new String[accumulator.size()]);
|
192 |
| } |
193 |
| |
194 |
| |
195 |
| |
196 |
| |
197 |
| |
198 |
| |
199 |
| |
200 |
| |
201 |
| |
202 |
| |
203 |
| |
204 |
| |
205 |
86
| public String getHeader(String name, String delimiter) {
|
206 |
| |
207 |
86
| String[] matches = getHeader(name);
|
208 |
| |
209 |
| |
210 |
86
| if (matches == null) {
|
211 |
38
| return null;
|
212 |
| } |
213 |
| |
214 |
| |
215 |
48
| if (matches.length == 1 || delimiter == null) {
|
216 |
34
| return matches[0];
|
217 |
| } |
218 |
| |
219 |
| |
220 |
14
| StringBuffer result = new StringBuffer(matches[0]);
|
221 |
| |
222 |
14
| for (int i = 1; i < matches.length; i++) {
|
223 |
14
| result.append(delimiter);
|
224 |
14
| result.append(matches[i]);
|
225 |
| } |
226 |
| |
227 |
14
| return result.toString();
|
228 |
| } |
229 |
| |
230 |
| |
231 |
| |
232 |
| |
233 |
| |
234 |
| |
235 |
| |
236 |
| |
237 |
| |
238 |
| |
239 |
| |
240 |
85
| public void setHeader(String name, String value) {
|
241 |
| |
242 |
85
| for (int i = 0; i < headers.size(); i++) {
|
243 |
1328
| InternetHeader header = (InternetHeader)headers.get(i);
|
244 |
| |
245 |
1328
| if (name.equalsIgnoreCase(header.getName())) {
|
246 |
| |
247 |
71
| header.setValue(value);
|
248 |
| |
249 |
71
| removeHeaders(name, i + 1);
|
250 |
71
| return;
|
251 |
| } |
252 |
| } |
253 |
| |
254 |
| |
255 |
14
| addHeader(name, value);
|
256 |
| } |
257 |
| |
258 |
| |
259 |
| |
260 |
| |
261 |
| |
262 |
| |
263 |
| |
264 |
| |
265 |
| |
266 |
121
| private void removeHeaders(String name, int pos) {
|
267 |
| |
268 |
121
| for (int i = pos; i < headers.size(); i++) {
|
269 |
863
| InternetHeader header = (InternetHeader)headers.get(i);
|
270 |
| |
271 |
863
| if (name.equalsIgnoreCase(header.getName())) {
|
272 |
| |
273 |
10
| headers.remove(i);
|
274 |
10
| i--;
|
275 |
| } |
276 |
| } |
277 |
| } |
278 |
| |
279 |
| |
280 |
| |
281 |
| |
282 |
| |
283 |
| |
284 |
| |
285 |
| |
286 |
| |
287 |
| |
288 |
2138
| private int findHeader(String name) {
|
289 |
2138
| return findHeader(name, 0);
|
290 |
| } |
291 |
| |
292 |
| |
293 |
| |
294 |
| |
295 |
| |
296 |
| |
297 |
| |
298 |
| |
299 |
| |
300 |
| |
301 |
| |
302 |
| |
303 |
2152
| private int findHeader(String name, int start) {
|
304 |
2152
| for (int i = start; i < headers.size(); i++) {
|
305 |
23645
| InternetHeader header = (InternetHeader)headers.get(i);
|
306 |
| |
307 |
23645
| if (name.equalsIgnoreCase(header.getName())) {
|
308 |
176
| return i;
|
309 |
| } |
310 |
| } |
311 |
1976
| return -1;
|
312 |
| } |
313 |
| |
314 |
| |
315 |
| |
316 |
| |
317 |
| |
318 |
| |
319 |
| |
320 |
| |
321 |
| |
322 |
1099
| public void addHeader(String name, String value) {
|
323 |
1099
| InternetHeader newHeader = new InternetHeader(name, value);
|
324 |
| |
325 |
| |
326 |
| |
327 |
1099
| if (name.equalsIgnoreCase("Received") || name.equalsIgnoreCase("Return-Path")) {
|
328 |
| |
329 |
94
| int pos = findHeader(name);
|
330 |
| |
331 |
| |
332 |
94
| if (pos != -1) {
|
333 |
| |
334 |
| |
335 |
0
| InternetHeader oldHeader = (InternetHeader)headers.get(pos);
|
336 |
0
| if (oldHeader.getValue() == null) {
|
337 |
0
| oldHeader.setValue(value);
|
338 |
| } |
339 |
| else { |
340 |
0
| headers.add(pos, newHeader);
|
341 |
| } |
342 |
| } |
343 |
| else { |
344 |
| |
345 |
94
| headers.add(0, newHeader);
|
346 |
| } |
347 |
| } |
348 |
| |
349 |
| else { |
350 |
| |
351 |
1005
| int pos = findHeader(name);
|
352 |
| |
353 |
| |
354 |
1005
| if (pos != -1) {
|
355 |
16
| InternetHeader oldHeader = (InternetHeader)headers.get(pos);
|
356 |
| |
357 |
16
| if (oldHeader.getValue() == null) {
|
358 |
2
| oldHeader.setValue(value);
|
359 |
| } |
360 |
| else { |
361 |
| |
362 |
| |
363 |
| |
364 |
14
| int lastPos = findHeader(name, pos + 1);
|
365 |
| |
366 |
14
| while (lastPos != -1) {
|
367 |
0
| pos = lastPos;
|
368 |
0
| lastPos = findHeader(name, pos + 1);
|
369 |
| } |
370 |
| |
371 |
| |
372 |
14
| headers.add(pos + 1, newHeader);
|
373 |
| } |
374 |
| } |
375 |
| else { |
376 |
| |
377 |
989
| pos = findHeader(":");
|
378 |
989
| if (pos == -1) {
|
379 |
879
| pos = headers.size();
|
380 |
| } |
381 |
989
| headers.add(pos, newHeader);
|
382 |
| } |
383 |
| } |
384 |
| } |
385 |
| |
386 |
| |
387 |
| |
388 |
| |
389 |
| |
390 |
| |
391 |
| |
392 |
| |
393 |
50
| public void removeHeader(String name) {
|
394 |
| |
395 |
50
| int pos = findHeader(name);
|
396 |
| |
397 |
50
| if (pos != -1) {
|
398 |
50
| InternetHeader oldHeader = (InternetHeader)headers.get(pos);
|
399 |
| |
400 |
50
| oldHeader.setValue(null);
|
401 |
| |
402 |
50
| removeHeaders(name, pos + 1);
|
403 |
| } |
404 |
| } |
405 |
| |
406 |
| |
407 |
| |
408 |
| |
409 |
| |
410 |
| |
411 |
| |
412 |
0
| public Enumeration getAllHeaders() {
|
413 |
0
| List result = new ArrayList();
|
414 |
| |
415 |
0
| for (int i = 0; i < headers.size(); i++) {
|
416 |
0
| InternetHeader header = (InternetHeader)headers.get(i);
|
417 |
| |
418 |
0
| if (header.getValue() != null) {
|
419 |
0
| result.add(header);
|
420 |
| } |
421 |
| } |
422 |
| |
423 |
0
| return Collections.enumeration(result);
|
424 |
| } |
425 |
| |
426 |
| |
427 |
| |
428 |
| |
429 |
| |
430 |
| |
431 |
| |
432 |
| |
433 |
| |
434 |
| |
435 |
| |
436 |
| |
437 |
0
| private boolean matchHeader(String name, String[] names) {
|
438 |
0
| for (int i = 0; i < names.length; i++) {
|
439 |
0
| if (name.equalsIgnoreCase(names[i])) {
|
440 |
0
| return true;
|
441 |
| } |
442 |
| } |
443 |
0
| return false;
|
444 |
| } |
445 |
| |
446 |
| |
447 |
| |
448 |
| |
449 |
| |
450 |
0
| public Enumeration getMatchingHeaders(String[] names) {
|
451 |
0
| List result = new ArrayList();
|
452 |
| |
453 |
0
| for (int i = 0; i < headers.size(); i++) {
|
454 |
0
| InternetHeader header = (InternetHeader)headers.get(i);
|
455 |
| |
456 |
0
| if (header.getValue() != null) {
|
457 |
| |
458 |
0
| if (matchHeader(header.getName(), names)) {
|
459 |
0
| result.add(header);
|
460 |
| } |
461 |
| } |
462 |
| } |
463 |
0
| return Collections.enumeration(result);
|
464 |
| } |
465 |
| |
466 |
| |
467 |
| |
468 |
| |
469 |
| |
470 |
0
| public Enumeration getNonMatchingHeaders(String[] names) {
|
471 |
0
| List result = new ArrayList();
|
472 |
| |
473 |
0
| for (int i = 0; i < headers.size(); i++) {
|
474 |
0
| InternetHeader header = (InternetHeader)headers.get(i);
|
475 |
| |
476 |
0
| if (header.getValue() != null) {
|
477 |
| |
478 |
0
| if (!matchHeader(header.getName(), names)) {
|
479 |
0
| result.add(header);
|
480 |
| } |
481 |
| } |
482 |
| } |
483 |
0
| return Collections.enumeration(result);
|
484 |
| } |
485 |
| |
486 |
| |
487 |
| |
488 |
| |
489 |
| |
490 |
| |
491 |
| |
492 |
| |
493 |
| |
494 |
| |
495 |
| |
496 |
| |
497 |
0
| public void addHeaderLine(String line) {
|
498 |
| |
499 |
0
| if (line.length() == 0) {
|
500 |
0
| return;
|
501 |
| } |
502 |
| |
503 |
| |
504 |
0
| char ch = line.charAt(0);
|
505 |
| |
506 |
| |
507 |
0
| if (ch == ' ' || ch == '\t') {
|
508 |
0
| InternetHeader header = (InternetHeader)headers.get(headers.size() - 1);
|
509 |
0
| header.appendValue(line);
|
510 |
| } |
511 |
| else { |
512 |
| |
513 |
0
| headers.add(new InternetHeader(line));
|
514 |
| } |
515 |
| } |
516 |
| |
517 |
| |
518 |
| |
519 |
| |
520 |
| |
521 |
0
| public Enumeration getAllHeaderLines() {
|
522 |
0
| return new HeaderLineEnumeration(getAllHeaders());
|
523 |
| } |
524 |
| |
525 |
| |
526 |
| |
527 |
| |
528 |
0
| public Enumeration getMatchingHeaderLines(String[] names) {
|
529 |
0
| return new HeaderLineEnumeration(getMatchingHeaders(names));
|
530 |
| } |
531 |
| |
532 |
| |
533 |
| |
534 |
| |
535 |
0
| public Enumeration getNonMatchingHeaderLines(String[] names) {
|
536 |
0
| return new HeaderLineEnumeration(getNonMatchingHeaders(names));
|
537 |
| } |
538 |
| |
539 |
| |
540 |
| |
541 |
| |
542 |
| |
543 |
| |
544 |
| |
545 |
| |
546 |
| |
547 |
12
| void setHeader(String name, Address[] addresses) {
|
548 |
| |
549 |
12
| if (addresses.length == 0) {
|
550 |
0
| removeHeader(name);
|
551 |
| } |
552 |
| |
553 |
| |
554 |
12
| setHeader(name, addresses[0].toString());
|
555 |
| |
556 |
| |
557 |
12
| for (int i = 1; i < addresses.length; i++) {
|
558 |
5
| Address address = addresses[i];
|
559 |
5
| addHeader(name, address.toString());
|
560 |
| } |
561 |
| } |
562 |
| |
563 |
| |
564 |
16
| void writeTo(OutputStream out, String[] ignore) throws IOException {
|
565 |
16
| if (ignore == null) {
|
566 |
| |
567 |
16
| for (int i = 0; i < headers.size(); i++) {
|
568 |
361
| InternetHeader header = (InternetHeader)headers.get(i);
|
569 |
| |
570 |
361
| if (header.getValue() != null) {
|
571 |
44
| header.writeTo(out);
|
572 |
| } |
573 |
| } |
574 |
| } |
575 |
| else { |
576 |
| |
577 |
0
| for (int i = 0; i < headers.size(); i++) {
|
578 |
0
| InternetHeader header = (InternetHeader)headers.get(i);
|
579 |
| |
580 |
0
| if (header.getValue() != null) {
|
581 |
0
| if (matchHeader(header.getName(), ignore)) {
|
582 |
0
| header.writeTo(out);
|
583 |
| } |
584 |
| } |
585 |
| } |
586 |
| } |
587 |
| } |
588 |
| |
589 |
| protected static final class InternetHeader extends Header { |
590 |
| |
591 |
0
| public InternetHeader(String h) {
|
592 |
| |
593 |
0
| super("", "");
|
594 |
0
| int separator = h.indexOf(':');
|
595 |
| |
596 |
0
| if (separator == -1) {
|
597 |
0
| name = h.trim();
|
598 |
| } |
599 |
| else { |
600 |
0
| name = h.substring(0, separator);
|
601 |
| |
602 |
0
| separator++;
|
603 |
| |
604 |
0
| while (separator < h.length()) {
|
605 |
0
| char ch = h.charAt(separator);
|
606 |
0
| if (ch != ' ' && ch != '\t' && ch != '\r' && ch != '\n') {
|
607 |
0
| break;
|
608 |
| } |
609 |
0
| separator++;
|
610 |
| } |
611 |
| |
612 |
0
| value = h.substring(separator);
|
613 |
| } |
614 |
| } |
615 |
| |
616 |
1099
| public InternetHeader(String name, String value) {
|
617 |
1099
| super(name, value);
|
618 |
| } |
619 |
| |
620 |
| |
621 |
| |
622 |
| |
623 |
| |
624 |
| |
625 |
| |
626 |
123
| void setValue(String value) {
|
627 |
123
| this.value = value;
|
628 |
| } |
629 |
| |
630 |
| |
631 |
| |
632 |
| |
633 |
| |
634 |
| |
635 |
0
| void appendValue(String value) {
|
636 |
0
| if (this.value == null) {
|
637 |
0
| this.value = value;
|
638 |
| } |
639 |
| else { |
640 |
0
| this.value = this.value + "\r\n" + value;
|
641 |
| } |
642 |
| } |
643 |
| |
644 |
44
| void writeTo(OutputStream out) throws IOException {
|
645 |
44
| out.write(name.getBytes());
|
646 |
44
| out.write(':');
|
647 |
44
| out.write(' ');
|
648 |
44
| out.write(value.getBytes());
|
649 |
44
| out.write('\r');
|
650 |
44
| out.write('\n');
|
651 |
| } |
652 |
| } |
653 |
| |
654 |
| private static class HeaderLineEnumeration implements Enumeration { |
655 |
| private Enumeration headers; |
656 |
| |
657 |
0
| public HeaderLineEnumeration(Enumeration headers) {
|
658 |
0
| this.headers = headers;
|
659 |
| } |
660 |
| |
661 |
0
| public boolean hasMoreElements() {
|
662 |
0
| return headers.hasMoreElements();
|
663 |
| } |
664 |
| |
665 |
0
| public Object nextElement() {
|
666 |
0
| Header h = (Header) headers.nextElement();
|
667 |
0
| return h.getName() + ": " + h.getValue();
|
668 |
| } |
669 |
| } |
670 |
| } |