001    /****************************************************************
002     * Licensed to the Apache Software Foundation (ASF) under one   *
003     * or more contributor license agreements.  See the NOTICE file *
004     * distributed with this work for additional information        *
005     * regarding copyright ownership.  The ASF licenses this file   *
006     * to you under the Apache License, Version 2.0 (the            *
007     * "License"); you may not use this file except in compliance   *
008     * with the License.  You may obtain a copy of the License at   *
009     *                                                              *
010     *   http://www.apache.org/licenses/LICENSE-2.0                 *
011     *                                                              *
012     * Unless required by applicable law or agreed to in writing,   *
013     * software distributed under the License is distributed on an  *
014     * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY       *
015     * KIND, either express or implied.  See the License for the    *
016     * specific language governing permissions and limitations      *
017     * under the License.                                           *
018     ****************************************************************/
019    
020    
021    
022    package org.apache.james.protocols.smtp.core.fastfail;
023    
024    
025    import org.apache.james.protocols.smtp.SMTPRetCode;
026    import org.apache.james.protocols.smtp.SMTPSession;
027    import org.apache.james.protocols.smtp.dsn.DSNStatus;
028    import org.apache.james.protocols.smtp.hook.HookResult;
029    import org.apache.james.protocols.smtp.hook.HookReturnCode;
030    import org.apache.james.protocols.smtp.hook.RcptHook;
031    import org.apache.mailet.MailAddress;
032    
033    public class MaxRcptHandler implements RcptHook {
034    
035        private int maxRcpt = 0;
036    
037    
038        /**
039         * Set the max rcpt for wich should be accepted
040         * 
041         * @param maxRcpt
042         *            The max rcpt count
043         */
044        public void setMaxRcpt(int maxRcpt) {
045            this.maxRcpt = maxRcpt;
046        }
047       
048        /**
049         * @see org.apache.james.protocols.smtp.hook.RcptHook#doRcpt(org.apache.james.protocols.smtp.SMTPSession, org.apache.mailet.MailAddress, org.apache.mailet.MailAddress)
050         */
051        public HookResult doRcpt(SMTPSession session, MailAddress sender, MailAddress rcpt) {
052            if ((session.getRcptCount() + 1) > maxRcpt) {
053                session.getLogger().info("Maximum recipients of " + maxRcpt + " reached");
054                
055                return new HookResult(HookReturnCode.DENY, SMTPRetCode.SYSTEM_STORAGE_ERROR, DSNStatus.getStatus(DSNStatus.NETWORK, DSNStatus.DELIVERY_TOO_MANY_REC)
056                        + " Requested action not taken: max recipients reached");
057            } else {
058                return new HookResult(HookReturnCode.DECLINED);
059            }
060        }
061    }