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 *    https://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 */
020package org.apache.directory.api.ldap.codec.actions.ldapResult;
021
022
023import org.apache.directory.api.asn1.DecoderException;
024import org.apache.directory.api.asn1.ber.grammar.GrammarAction;
025import org.apache.directory.api.asn1.ber.tlv.BerValue;
026import org.apache.directory.api.asn1.ber.tlv.IntegerDecoder;
027import org.apache.directory.api.asn1.ber.tlv.IntegerDecoderException;
028import org.apache.directory.api.asn1.ber.tlv.TLV;
029import org.apache.directory.api.i18n.I18n;
030import org.apache.directory.api.ldap.codec.api.LdapMessageContainer;
031import org.apache.directory.api.ldap.model.message.LdapResult;
032import org.apache.directory.api.ldap.model.message.LdapResultImpl;
033import org.apache.directory.api.ldap.model.message.Message;
034import org.apache.directory.api.ldap.model.message.ResultCodeEnum;
035import org.apache.directory.api.ldap.model.message.ResultResponse;
036import org.apache.directory.api.util.Strings;
037import org.slf4j.Logger;
038import org.slf4j.LoggerFactory;
039
040
041/**
042 * The action used to set the LdapResult result code.
043 * <pre>
044 * LDAPResult ::= SEQUENCE {
045 *     resultCode ENUMERATED {
046 *         ...
047 * </pre>
048 * @author <a href="mailto:dev@directory.apache.org">Apache Directory Project</a>
049 */
050public class StoreResultCode extends GrammarAction<LdapMessageContainer<Message>>
051{
052    /** The logger */
053    private static final Logger LOG = LoggerFactory.getLogger( StoreResultCode.class );
054
055    /**
056     * Instantiates a new result code action.
057     */
058    public StoreResultCode()
059    {
060        super( "Store resultCode" );
061    }
062
063
064    /**
065     * {@inheritDoc}
066     */
067    @Override
068    public void action( LdapMessageContainer<Message> container ) throws DecoderException
069    {
070        // The current TLV should be a integer
071        // We get it and store it in MessageId
072        TLV tlv = container.getCurrentTLV();
073
074        BerValue value = tlv.getValue();
075        ResultCodeEnum resultCode = ResultCodeEnum.SUCCESS;
076
077        try
078        {
079            resultCode = ResultCodeEnum.getResultCode( IntegerDecoder.parse( value, 0,
080                ResultCodeEnum.E_SYNC_REFRESH_REQUIRED
081                    .getResultCode() ) );
082        }
083        catch ( IntegerDecoderException ide )
084        {
085            LOG.error( I18n.err( I18n.ERR_05108_INVALID_RESULT_CODE, Strings.dumpBytes( value.getData() ), ide.getMessage() ) );
086
087            throw new DecoderException( ide.getMessage(), ide );
088        }
089
090        if ( LOG.isDebugEnabled() )
091        {
092            LOG.debug( I18n.msg( I18n.MSG_05109_RESULT_CODE_IS, resultCode ) );
093        }
094
095        ResultResponse response = ( ResultResponse ) container.getMessage();
096        
097        LdapResult ldapResult;
098        
099        if ( response == null ) 
100        {
101            ldapResult = new LdapResultImpl();
102        }
103        else
104        {
105            ldapResult = response.getLdapResult();
106        }
107
108        container.setLdapResult( ldapResult );
109        ldapResult.setResultCode( resultCode );
110    }
111}