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   *    http://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.codec.actions.searchResultReference;
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.i18n.I18n;
27  import org.apache.directory.api.ldap.codec.api.LdapMessageContainer;
28  import org.apache.directory.api.ldap.codec.decorators.SearchResultReferenceDecorator;
29  import org.apache.directory.api.ldap.model.exception.LdapURLEncodingException;
30  import org.apache.directory.api.ldap.model.message.Referral;
31  import org.apache.directory.api.ldap.model.message.ReferralImpl;
32  import org.apache.directory.api.ldap.model.message.SearchResultReference;
33  import org.apache.directory.api.ldap.model.url.LdapUrl;
34  import org.apache.directory.api.util.Strings;
35  import org.slf4j.Logger;
36  import org.slf4j.LoggerFactory;
37  
38  
39  /**
40   * The action used to store a reference into a searchResultReference
41   * <pre>
42   * LdapMessage ::= ... SearchResultReference ...
43   * SearchResultReference ::= [APPLICATION 19] SEQUENCE OF LDAPURL
44   * </pre>
45   * @author <a href="mailto:dev@directory.apache.org">Apache Directory Project</a>
46   */
47  public class StoreReference extends GrammarAction<LdapMessageContainer<SearchResultReferenceDecorator>>
48  {
49      /** The logger */
50      private static final Logger LOG = LoggerFactory.getLogger( StoreReference.class );
51  
52      /** Speedup for logs */
53      private static final boolean IS_DEBUG = LOG.isDebugEnabled();
54  
55  
56      /**
57       * Instantiates a new store reference action.
58       */
59      public StoreReference()
60      {
61          super( "Store a reference" );
62      }
63  
64  
65      /**
66       * {@inheritDoc}
67       */
68      public void action( LdapMessageContainer<SearchResultReferenceDecorator> container ) throws DecoderException
69      {
70          SearchResultReference searchResultReference = container.getMessage();
71  
72          // Get the Value and store it in the BindRequest
73          TLV tlv = container.getCurrentTLV();
74  
75          // Get the referral, or create it if not existing
76          Referral referral = searchResultReference.getReferral();
77  
78          if ( referral == null )
79          {
80              referral = new ReferralImpl();
81              searchResultReference.setReferral( referral );
82          }
83  
84          // We have to handle the special case of a 0 length list of referrals
85          LdapUrl url = LdapUrl.EMPTY_URL;
86  
87          if ( tlv.getLength() == 0 )
88          {
89              referral.addLdapUrl( "" );
90          }
91          else
92          {
93              String urlStr = Strings.utf8ToString( tlv.getValue().getData() );
94  
95              try
96              {
97                  url = new LdapUrl( urlStr );
98                  referral.addLdapUrl( urlStr );
99              }
100             catch ( LdapURLEncodingException luee )
101             {
102                 LOG.error( I18n.err( I18n.ERR_04021, urlStr, luee.getMessage() ) );
103                 throw new DecoderException( I18n.err( I18n.ERR_04016, luee.getMessage() ), luee );
104             }
105         }
106 
107         if ( IS_DEBUG )
108         {
109             LOG.debug( "Search reference URL found : {}", url );
110         }
111 
112         // We can have an END transition
113         container.setGrammarEndAllowed( true );
114     }
115 }