1 /************************************************************************
2 * Copyright (c) 2000-2006 The Apache Software Foundation. *
3 * All rights reserved. *
4 * ------------------------------------------------------------------- *
5 * Licensed under the Apache License, Version 2.0 (the "License"); you *
6 * may not use this file except in compliance with the License. You *
7 * may obtain a copy of the License at: *
8 * *
9 * http://www.apache.org/licenses/LICENSE-2.0 *
10 * *
11 * Unless required by applicable law or agreed to in writing, software *
12 * distributed under the License is distributed on an "AS IS" BASIS, *
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or *
14 * implied. See the License for the specific language governing *
15 * permissions and limitations under the License. *
16 ***********************************************************************/
17
18 package org.apache.james.transport.matchers;
19
20 import org.apache.mailet.GenericMatcher;
21 import org.apache.mailet.Mail;
22
23 import javax.mail.Header;
24 import javax.mail.MessagingException;
25 import javax.mail.internet.MimeMessage;
26 import java.util.Collection;
27 import java.util.Enumeration;
28 import java.util.Locale;
29
30 /***
31 * Checks whether the message (entire message, not just content) is greater
32 * than a certain number of bytes. You can use 'k' and 'm' as optional postfixes.
33 * In other words, "1m" is the same as writing "1024k", which is the same as
34 * "1048576".
35 *
36 */
37 public class SizeGreaterThan extends GenericMatcher {
38
39 int cutoff = 0;
40
41 public void init() {
42 String amount = getCondition().trim().toLowerCase(Locale.US);
43 if (amount.endsWith("k")) {
44 amount = amount.substring(0, amount.length() - 1);
45 cutoff = Integer.parseInt(amount) * 1024;
46 } else if (amount.endsWith("m")) {
47 amount = amount.substring(0, amount.length() - 1);
48 cutoff = Integer.parseInt(amount) * 1024 * 1024;
49 } else {
50 cutoff = Integer.parseInt(amount);
51 }
52 }
53
54 public Collection match(Mail mail) throws MessagingException {
55 MimeMessage message = mail.getMessage();
56
57 int size = message.getSize();
58 Enumeration e = message.getAllHeaders();
59 while (e.hasMoreElements()) {
60 size += ((Header)e.nextElement()).toString().length();
61 }
62 if (size > cutoff) {
63 return mail.getRecipients();
64 } else {
65 return null;
66 }
67 }
68 }