/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*
*/
package org.apache.directory.api.ldap.codec;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import java.nio.ByteBuffer;
import org.apache.directory.api.asn1.DecoderException;
import org.apache.directory.api.asn1.EncoderException;
import org.apache.directory.api.asn1.ber.Asn1Container;
import org.apache.directory.api.asn1.ber.Asn1Decoder;
import org.apache.directory.api.ldap.codec.api.LdapMessageContainer;
import org.apache.directory.api.ldap.codec.api.MessageDecorator;
import org.apache.directory.api.ldap.codec.decorators.UnbindRequestDecorator;
import org.apache.directory.api.ldap.codec.osgi.AbstractCodecServiceTest;
import org.apache.directory.api.ldap.model.message.Message;
import org.apache.directory.api.ldap.model.message.UnbindRequest;
import org.apache.directory.api.ldap.model.message.UnbindRequestImpl;
import org.apache.directory.api.util.Strings;
import org.junit.Test;
import org.junit.runner.RunWith;
import com.mycila.junit.concurrent.Concurrency;
import com.mycila.junit.concurrent.ConcurrentJunitRunner;
/**
* A global Ldap Decoder test
*
* @author <a href="mailto:dev@directory.apache.org">Apache Directory Project</a>
*/
@RunWith(ConcurrentJunitRunner.class)
@Concurrency()
public class LdapMessageTest extends AbstractCodecServiceTest
{
/**
* Test the decoding of null length messageId
*/
@Test
public void testDecodeMessageLengthNull()
{
Asn1Decoder ldapDecoder = new Asn1Decoder();
ByteBuffer stream = ByteBuffer.allocate( 0x02 );
stream.put( new byte[]
{ 0x30, 0x00, // LDAPMessage ::=SEQUENCE {
} );
stream.flip();
// Allocate a LdapMessage Container
Asn1Container ldapMessageContainer = new LdapMessageContainer<MessageDecorator<? extends Message>>( codec );
// Decode a BindRequest PDU
try
{
ldapDecoder.decode( stream, ldapMessageContainer );
}
catch ( DecoderException de )
{
assertTrue( true );
return;
}
fail( "We should not reach this point !" );
}
/**
* Test the decoding of null length messageId
*/
@Test
public void testDecodeMessageIdLengthNull()
{
Asn1Decoder ldapDecoder = new Asn1Decoder();
ByteBuffer stream = ByteBuffer.allocate( 0x04 );
stream.put( new byte[]
{ 0x30, 0x02, // LDAPMessage ::=SEQUENCE {
0x02,
0x00 // messageID MessageID
} );
stream.flip();
// Allocate a LdapMessage Container
Asn1Container ldapMessageContainer = new LdapMessageContainer<MessageDecorator<? extends Message>>( codec );
// Decode a BindRequest PDU
try
{
ldapDecoder.decode( stream, ldapMessageContainer );
}
catch ( DecoderException de )
{
assertTrue( true );
return;
}
fail( "We should not reach this point !" );
}
/**
* Test the decoding of null length messageId
*/
@Test
public void testDecodeMessageIdMinusOne()
{
Asn1Decoder ldapDecoder = new Asn1Decoder();
ByteBuffer stream = ByteBuffer.allocate( 0x05 );
stream.put( new byte[]
{ 0x30, 0x03, // LDAPMessage ::=SEQUENCE {
0x02,
0x01,
( byte ) 0xff // messageID MessageID = -1
} );
stream.flip();
// Allocate a LdapMessage Container
Asn1Container ldapMessageContainer = new LdapMessageContainer<MessageDecorator<? extends Message>>( codec );
// Decode a BindRequest PDU
try
{
ldapDecoder.decode( stream, ldapMessageContainer );
}
catch ( DecoderException de )
{
assertTrue( true );
return;
}
fail( "We should not reach this point !" );
}
/**
* Test the decoding of messageId which value is -1
*/
@Test
public void testDecodeMessageIdMaxInt()
{
Asn1Decoder ldapDecoder = new Asn1Decoder();
ByteBuffer stream = ByteBuffer.allocate( 0x08 );
stream.put( new byte[]
{ 0x30, 0x06, // LDAPMessage ::=SEQUENCE {
// messageID MessageID = -1
0x02,
0x04,
( byte ) 0x7f,
( byte ) 0xff,
( byte ) 0xff,
( byte ) 0xff } );
stream.flip();
// Allocate a LdapMessage Container
Asn1Container ldapMessageContainer = new LdapMessageContainer<MessageDecorator<? extends Message>>( codec );
// Decode a BindRequest PDU
try
{
ldapDecoder.decode( stream, ldapMessageContainer );
}
catch ( DecoderException de )
{
assertTrue( true );
return;
}
fail( "We should not reach this point !" );
}
/**
* Test the decoding of a message with a wrong protocol operation
*/
@Test
public void testDecodeWrongProtocolOpMaxInt()
{
byte[] buffer = new byte[]
{ 0x30, 0x05, // LDAPMessage ::=SEQUENCE {
0x02,
0x01,
0x01, // messageID MessageID = 1
0x42,
0x00 // ProtocolOp
};
Asn1Decoder ldapDecoder = new Asn1Decoder();
ByteBuffer stream = ByteBuffer.allocate( 0x07 );
for ( int i = 0; i < 256; i++ )
{
buffer[5] = ( byte ) i;
stream.put( buffer );
stream.flip();
// Allocate a LdapMessage Container
Asn1Container ldapMessageContainer = new LdapMessageContainer<MessageDecorator<? extends Message>>( codec );
// Decode a BindRequest PDU
try
{
ldapDecoder.decode( stream, ldapMessageContainer );
}
catch ( DecoderException de )
{
switch ( i )
{
case 0x42:
case 0x4A:
case 0x50: // AbandonRequest
case 0x60:
case 0x61:
case 0x63:
case 0x64:
case 0x65:
case 0x66:
case 0x67:
case 0x68:
case 0x69:
case 0x6B:
case 0x6C:
case 0x6D:
case 0x6E:
case 0x6F:
case 0x73:
case 0x77:
case 0x78:
assertTrue( true );
break;
default:
String res = de.getMessage();
if ( res.startsWith( "ERR_00001_BAD_TRANSITION_FROM_STATE" )
|| res.startsWith( "Universal tag " )
|| res.startsWith( "ERR_00010_TRUNCATED_PDU Truncated PDU" ) )
{
assertTrue( true );
}
else
{
fail( "Bad exception : " + res );
return;
}
break;
}
}
stream.clear();
}
assertTrue( true );
}
/**
* Test the decoding of a LdapMessage with a large MessageId
*/
@Test
public void testDecodeUnBindRequestNoControls()
{
Asn1Decoder ldapDecoder = new Asn1Decoder();
ByteBuffer stream = ByteBuffer.allocate( 0x08 );
stream.put( new byte[]
{ 0x30, 0x06, // LDAPMessage ::=SEQUENCE {
0x02,
0x02,
0x01,
( byte ) 0xF4, // messageID MessageID (500)
0x42,
0x00, // CHOICE { ..., unbindRequest UnbindRequest,...
// UnbindRequest ::= [APPLICATION 2] NULL
} );
String decodedPdu = Strings.dumpBytes( stream.array() );
stream.flip();
// Allocate a BindRequest Container
LdapMessageContainer<UnbindRequestDecorator> container =
new LdapMessageContainer<UnbindRequestDecorator>( codec );
try
{
ldapDecoder.decode( stream, container );
}
catch ( DecoderException de )
{
de.printStackTrace();
fail( de.getMessage() );
}
Message message = container.getMessage();
assertEquals( 500, message.getMessageId() );
// Check the length
UnbindRequest internalUnbindRequest = new UnbindRequestImpl();
internalUnbindRequest.setMessageId( message.getMessageId() );
try
{
ByteBuffer bb = encoder.encodeMessage( internalUnbindRequest );
// Check the length
assertEquals( 0x08, bb.limit() );
String encodedPdu = Strings.dumpBytes( bb.array() );
assertEquals( encodedPdu, decodedPdu );
}
catch ( EncoderException ee )
{
ee.printStackTrace();
fail( ee.getMessage() );
}
}
}