View Javadoc
1   /*
2    *  Licensed to the Apache Software Foundation (ASF) under one
3    *  or more contributor license agreements.  See the NOTICE file
4    *  distributed with this work for additional information
5    *  regarding copyright ownership.  The ASF licenses this file
6    *  to you under the Apache License, Version 2.0 (the
7    *  "License"); you may not use this file except in compliance
8    *  with the License.  You may obtain a copy of the License at
9    *
10   *    https://www.apache.org/licenses/LICENSE-2.0
11   *
12   *  Unless required by applicable law or agreed to in writing,
13   *  software distributed under the License is distributed on an
14   *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15   *  KIND, either express or implied.  See the License for the
16   *  specific language governing permissions and limitations
17   *  under the License.
18   *
19   */
20  package org.apache.directory.api.ldap.extras.extended.ads_impl.endTransaction.controls.actions;
21  
22  
23  import org.apache.directory.api.asn1.DecoderException;
24  import org.apache.directory.api.asn1.ber.grammar.GrammarAction;
25  import org.apache.directory.api.asn1.ber.tlv.TLV;
26  import org.apache.directory.api.asn1.util.Oid;
27  import org.apache.directory.api.i18n.I18n;
28  import org.apache.directory.api.ldap.codec.api.ControlFactory;
29  import org.apache.directory.api.ldap.extras.extended.ads_impl.endTransaction.controls.ControlsContainer;
30  import org.apache.directory.api.ldap.model.message.Control;
31  import org.apache.directory.api.util.Strings;
32  import org.slf4j.Logger;
33  import org.slf4j.LoggerFactory;
34  
35  
36  /**
37   * The action used add a new control. We store its OID.
38   * <pre>
39   * Control ::= SEQUENCE {
40   *     controlType             LDAPOID,
41   *     ...
42   * </pre>
43   *
44   * @author <a href="mailto:dev@directory.apache.org">Apache Directory Project</a>
45   */
46  public class AddControl extends GrammarAction<ControlsContainer>
47  {
48      /** The logger */
49      private static final Logger LOG = LoggerFactory.getLogger( AddControl.class );
50  
51      /**
52       * Instantiates a new AddControl action.
53       */
54      public AddControl()
55      {
56          super( "Add a new control" );
57      }
58  
59  
60      /**
61       * {@inheritDoc}
62       */
63      @Override
64      public void action( ControlsContainer container ) throws DecoderException
65      {
66          TLV tlv = container.getCurrentTLV();
67  
68          // Store the type
69          // We have to handle the special case of a 0 length OID
70          if ( tlv.getLength() == 0 )
71          {
72              String msg = I18n.err( I18n.ERR_08214_NULL_OID );
73              LOG.error( msg );
74  
75              // This will generate a PROTOCOL_ERROR
76              throw new DecoderException( msg );
77          }
78  
79          byte[] value = tlv.getValue().getData();
80          String oidValue = Strings.asciiBytesToString( value );
81  
82          // The OID is encoded as a String, not an Object Id
83          if ( !Oid.isOid( oidValue ) )
84          {
85              String msg = I18n.err( I18n.ERR_08215_INVALID_CONTROL_OID, oidValue );
86              LOG.error( msg );
87  
88              // This will generate a PROTOCOL_ERROR
89              throw new DecoderException( msg );
90          }
91  
92          ControlFactory<?> factory = container.getLdapCodecService().getResponseControlFactories().get( oidValue );
93          container.setFactory( factory );
94          Control control = factory.newControl();
95  
96          container.setCurrentControl( control );
97          container.addControl( control );
98  
99          // We can have an END transition
100         container.setGrammarEndAllowed( true );
101 
102         if ( LOG.isDebugEnabled() )
103         {
104             LOG.debug( I18n.msg( I18n.MSG_08201_CONTROL_OID, oidValue ) );
105         }
106     }
107 }