1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.apache.directory.api.ldap.codec.actions.controls;
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.LdapMessageContainer;
29 import org.apache.directory.api.ldap.codec.api.MessageDecorator;
30 import org.apache.directory.api.ldap.model.message.Control;
31 import org.apache.directory.api.ldap.model.message.Message;
32 import org.apache.directory.api.util.Strings;
33 import org.slf4j.Logger;
34 import org.slf4j.LoggerFactory;
35
36
37
38
39
40
41
42
43
44
45
46
47 public class AddControl extends GrammarAction<LdapMessageContainer<MessageDecorator<? extends Message>>>
48 {
49
50 private static final Logger LOG = LoggerFactory.getLogger( AddControl.class );
51
52
53 private static final boolean IS_DEBUG = LOG.isDebugEnabled();
54
55
56
57
58
59 public AddControl()
60 {
61 super( "Add a new control" );
62 }
63
64
65
66
67
68 public void action( LdapMessageContainer<MessageDecorator<? extends Message>> container ) throws DecoderException
69 {
70 TLV tlv = container.getCurrentTLV();
71
72
73
74 if ( tlv.getLength() == 0 )
75 {
76 String msg = I18n.err( I18n.ERR_04097 );
77 LOG.error( msg );
78
79
80 throw new DecoderException( msg );
81 }
82
83 byte[] value = tlv.getValue().getData();
84 String oidValue = Strings.asciiBytesToString( value );
85
86
87 if ( !Oid.isOid( oidValue ) )
88 {
89 LOG.error( I18n.err( I18n.ERR_04098, Strings.dumpBytes( value ) ) );
90
91
92 throw new DecoderException( I18n.err( I18n.ERR_04099, oidValue ) );
93 }
94
95 Message message = container.getMessage();
96
97 Control control = container.getLdapCodecService().newControl( oidValue );
98
99 message.addControl( control );
100
101
102 container.setGrammarEndAllowed( true );
103
104 if ( IS_DEBUG )
105 {
106 LOG.debug( "Control OID : " + oidValue );
107 }
108 }
109 }