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    package org.apache.directory.shared.ldap.codec.actions;
021    
022    
023    import org.apache.directory.shared.asn1.ber.IAsn1Container;
024    import org.apache.directory.shared.asn1.ber.grammar.GrammarAction;
025    import org.apache.directory.shared.asn1.ber.tlv.TLV;
026    import org.apache.directory.shared.asn1.ber.tlv.Value;
027    import org.apache.directory.shared.asn1.codec.DecoderException;
028    import org.apache.directory.shared.ldap.codec.LdapMessageCodec;
029    import org.apache.directory.shared.ldap.codec.LdapMessageContainer;
030    import org.apache.directory.shared.ldap.codec.controls.CodecControl;
031    import org.apache.directory.shared.ldap.codec.controls.ControlDecoder;
032    import org.apache.directory.shared.ldap.message.control.Control;
033    import org.apache.directory.shared.ldap.util.StringTools;
034    import org.slf4j.Logger;
035    import org.slf4j.LoggerFactory;
036    
037    
038    /**
039     * The action used to set the value of a control. This is an extension point
040     * where different controls can be plugged in (at least eventually). For now we
041     * hard code controls.
042     * 
043     * @author <a href="mailto:dev@directory.apache.org">Apache Directory Project</a>
044     * @version $Rev$, $Date$, 
045     */
046    public class ControlValueAction extends GrammarAction
047    {
048        /** The logger */
049        private static final Logger log = LoggerFactory.getLogger( ControlValueAction.class );
050    
051        /** Speedup for logs */
052        private static final boolean IS_DEBUG = log.isDebugEnabled();
053    
054    
055        public void action( IAsn1Container container ) throws DecoderException
056        {
057            LdapMessageContainer ldapMessageContainer = ( LdapMessageContainer ) container;
058            TLV tlv = ldapMessageContainer.getCurrentTLV();
059            LdapMessageCodec message = ldapMessageContainer.getLdapMessage();
060    
061            // Get the current control
062            Control control = message.getCurrentControl();
063            Value value = tlv.getValue();
064    
065            ControlDecoder decoder = ((CodecControl)control).getDecoder();
066    
067            // Store the value - have to handle the special case of a 0 length value
068            if ( tlv.getLength() == 0 )
069            {
070                control.setValue( StringTools.EMPTY_BYTES );
071            }
072            else
073            {
074                if ( decoder == null )
075                {
076                    // No decoder : store the raw value
077                    control.setValue( value.getData() );
078                }
079                else
080                {
081                    decoder.decode( value.getData(), control );
082                }
083            }
084    
085            // We can have an END transition
086            ldapMessageContainer.grammarEndAllowed( true );
087    
088            if ( IS_DEBUG )
089            {
090                log.debug( "Control value : " + StringTools.dumpBytes( ( byte[] ) control.getValue() ) );
091            }
092        }
093    }