use of org.apache.directory.api.ldap.codec.decorators.SearchResultReferenceDecorator in project directory-ldap-api by apache.
the class SearchResultReferenceTest method testDecodeSearchResultReferenceSuccess.
/**
* Test the decoding of a SearchResultReference
*/
@Test
public void testDecodeSearchResultReferenceSuccess() {
Asn1Decoder ldapDecoder = new Asn1Decoder();
ByteBuffer stream = ByteBuffer.allocate(0x3d8);
String[] ldapUrls = new String[] { "ldap:///", "ldap://directory.apache.org:80/", "ldap://d-a.org:80/", "ldap://1.2.3.4/", "ldap://1.2.3.4:80/", "ldap://1.1.1.100000.a/", "ldap://directory.apache.org:389/dc=example,dc=org/", "ldap://directory.apache.org:389/dc=example", "ldap://directory.apache.org:389/dc=example%202,dc=org", "ldap://directory.apache.org:389/dc=example,dc=org?ou", "ldap://directory.apache.org:389/dc=example,dc=org?ou,objectclass,dc", "ldap://directory.apache.org:389/dc=example,dc=org?ou,dc,ou", "ldap:///o=University%20of%20Michigan,c=US", "ldap://ldap.itd.umich.edu/o=University%20of%20Michigan,c=US", "ldap://ldap.itd.umich.edu/o=University%20of%20Michigan,c=US?postalAddress", "ldap://host.com:6666/o=University%20of%20Michigan,c=US??sub?(cn=Babs%20Jensen)", "ldap://ldap.itd.umich.edu/c=GB?objectClass?one", "ldap://ldap.question.com/o=Question%3f,c=US?mail", "ldap://ldap.netscape.com/o=Babsco,c=US???(int=%5c00%5c00%5c00%5c04)", "ldap:///??sub??bindname=cn=Manager%2co=Foo", "ldap:///??sub??!bindname=cn=Manager%2co=Foo" };
stream.put(new byte[] { // LDAPMessage
0x30, // LDAPMessage
(byte) 0x82, // LDAPMessage
0x03, // LDAPMessage
(byte) 0xd4, // ::=SEQUENCE {
0x02, 0x01, // messageID MessageID
0x01, 0x73, (byte) 0x82, 0x03, // CHOICE { ...,
(byte) 0xcd // searchResEntry
// SearchResultEntry,
// ...
// SearchResultReference ::= [APPLICATION 19] SEQUENCE OF LDAPURL
});
for (int i = 0; i < ldapUrls.length; i++) {
stream.put((byte) 0x04);
stream.put((byte) Strings.getBytesUtf8(ldapUrls[i]).length);
byte[] bytes = Strings.getBytesUtf8(ldapUrls[i]);
for (int j = 0; j < bytes.length; j++) {
stream.put(bytes[j]);
}
}
String decodedPdu = Strings.dumpBytes(stream.array());
stream.flip();
// Allocate a BindRequest Container
LdapMessageContainer<SearchResultReferenceDecorator> ldapMessageContainer = new LdapMessageContainer<SearchResultReferenceDecorator>(codec);
try {
ldapDecoder.decode(stream, ldapMessageContainer);
} catch (DecoderException de) {
de.printStackTrace();
fail(de.getMessage());
}
SearchResultReference searchResultReference = ldapMessageContainer.getMessage();
assertEquals(1, searchResultReference.getMessageId());
Set<String> ldapUrlsSet = new HashSet<String>();
for (int i = 0; i < ldapUrls.length; i++) {
ldapUrlsSet.add(ldapUrls[i]);
}
Referral referral = searchResultReference.getReferral();
assertNotNull(referral);
for (String ldapUrl : referral.getLdapUrls()) {
if (ldapUrlsSet.contains(ldapUrl)) {
ldapUrlsSet.remove(ldapUrl);
} else {
fail(ldapUrl.toString() + " is not present");
}
}
assertTrue(ldapUrlsSet.size() == 0);
// Check the encoding
try {
ByteBuffer bb = encoder.encodeMessage(searchResultReference);
// Check the length
assertEquals(0x3D8, bb.limit());
String encodedPdu = Strings.dumpBytes(bb.array());
assertEquals(encodedPdu, decodedPdu);
} catch (EncoderException ee) {
ee.printStackTrace();
fail(ee.getMessage());
}
}
use of org.apache.directory.api.ldap.codec.decorators.SearchResultReferenceDecorator in project directory-ldap-api by apache.
the class SearchResultReferenceTest method testDecodeSearchResultReferenceOneReference.
/**
* Test the decoding of a SearchResultReference with one reference
*/
@Test
public void testDecodeSearchResultReferenceOneReference() {
Asn1Decoder ldapDecoder = new Asn1Decoder();
ByteBuffer stream = ByteBuffer.allocate(0x11);
stream.put(new byte[] { // LDAPMessage ::=SEQUENCE {
0x30, // LDAPMessage ::=SEQUENCE {
0x0F, 0x02, 0x01, // messageID MessageID
0x01, 0x73, // CHOICE { ..., searchResEntry SearchResultEntry,
0x0A, // ...
0x04, 0x08, 'l', 'd', 'a', 'p', ':', '/', '/', // SearchResultReference
'/' // ::=
// [APPLICATION
// 19]
// SEQUENCE
// OF
// LDAPURL
});
String decodedPdu = Strings.dumpBytes(stream.array());
stream.flip();
// Allocate a BindRequest Container
LdapMessageContainer<SearchResultReferenceDecorator> ldapMessageContainer = new LdapMessageContainer<SearchResultReferenceDecorator>(codec);
try {
ldapDecoder.decode(stream, ldapMessageContainer);
} catch (DecoderException de) {
de.printStackTrace();
fail(de.getMessage());
}
SearchResultReference searchResultReference = ldapMessageContainer.getMessage();
assertEquals(1, searchResultReference.getMessageId());
Referral referral = searchResultReference.getReferral();
assertNotNull(referral);
for (String ldapUrl : referral.getLdapUrls()) {
assertEquals("ldap:///", ldapUrl);
}
// Check the encoding
try {
ByteBuffer bb = encoder.encodeMessage(searchResultReference);
// Check the length
assertEquals(0x11, bb.limit());
String encodedPdu = Strings.dumpBytes(bb.array());
assertEquals(encodedPdu, decodedPdu);
} catch (EncoderException ee) {
ee.printStackTrace();
fail(ee.getMessage());
}
}
use of org.apache.directory.api.ldap.codec.decorators.SearchResultReferenceDecorator in project directory-ldap-api by apache.
the class SearchResultReferenceTest method testDecodeSearchResultReferenceSuccessWithControls.
/**
* Test the decoding of a SearchResultReference with controls
*/
@Test
public void testDecodeSearchResultReferenceSuccessWithControls() {
Asn1Decoder ldapDecoder = new Asn1Decoder();
ByteBuffer stream = ByteBuffer.allocate(0x3F5);
String[] ldapUrls = new String[] { "ldap:///", "ldap://directory.apache.org:80/", "ldap://d-a.org:80/", "ldap://1.2.3.4/", "ldap://1.2.3.4:80/", "ldap://1.1.1.100000.a/", "ldap://directory.apache.org:389/dc=example,dc=org/", "ldap://directory.apache.org:389/dc=example", "ldap://directory.apache.org:389/dc=example%202,dc=org", "ldap://directory.apache.org:389/dc=example,dc=org?ou", "ldap://directory.apache.org:389/dc=example,dc=org?ou,objectclass,dc", "ldap://directory.apache.org:389/dc=example,dc=org?ou,dc,ou", "ldap:///o=University%20of%20Michigan,c=US", "ldap://ldap.itd.umich.edu/o=University%20of%20Michigan,c=US", "ldap://ldap.itd.umich.edu/o=University%20of%20Michigan,c=US?postalAddress", "ldap://host.com:6666/o=University%20of%20Michigan,c=US??sub?(cn=Babs%20Jensen)", "ldap://ldap.itd.umich.edu/c=GB?objectClass?one", "ldap://ldap.question.com/o=Question%3f,c=US?mail", "ldap://ldap.netscape.com/o=Babsco,c=US???(int=%5c00%5c00%5c00%5c04)", "ldap:///??sub??bindname=cn=Manager%2co=Foo", "ldap:///??sub??!bindname=cn=Manager%2co=Foo" };
stream.put(new byte[] { // LDAPMessage
0x30, // LDAPMessage
(byte) 0x82, // LDAPMessage
0x03, // LDAPMessage
(byte) 0xF1, // ::=SEQUENCE {
0x02, 0x01, // messageID MessageID
0x01, 0x73, (byte) 0x82, 0x03, // CHOICE { ...,
(byte) 0xcd // searchResEntry
// SearchResultEntry,
// ...
// SearchResultReference ::= [APPLICATION 19] SEQUENCE OF LDAPURL
});
for (int i = 0; i < ldapUrls.length; i++) {
stream.put((byte) 0x04);
stream.put((byte) Strings.getBytesUtf8(ldapUrls[i]).length);
byte[] bytes = Strings.getBytesUtf8(ldapUrls[i]);
for (int j = 0; j < bytes.length; j++) {
stream.put(bytes[j]);
}
}
byte[] controlBytes = new byte[] { (byte) 0xA0, // A control
0x1B, 0x30, 0x19, 0x04, 0x17, 0x32, 0x2E, 0x31, 0x36, 0x2E, 0x38, 0x34, 0x30, 0x2E, 0x31, 0x2E, 0x31, 0x31, 0x33, 0x37, 0x33, 0x30, 0x2E, 0x33, 0x2E, 0x34, 0x2E, 0x32 };
for (int i = 0; i < controlBytes.length; i++) {
stream.put(controlBytes[i]);
}
String decodedPdu = Strings.dumpBytes(stream.array());
stream.flip();
// Allocate a BindRequest Container
LdapMessageContainer<SearchResultReferenceDecorator> ldapMessageContainer = new LdapMessageContainer<SearchResultReferenceDecorator>(codec);
try {
ldapMessageContainer.clean();
ldapDecoder.decode(stream, ldapMessageContainer);
} catch (DecoderException de) {
de.printStackTrace();
fail(de.getMessage());
}
stream.flip();
SearchResultReference searchResultReference = ldapMessageContainer.getMessage();
assertEquals(1, searchResultReference.getMessageId());
Set<String> ldapUrlsSet = new HashSet<String>();
for (int i = 0; i < ldapUrls.length; i++) {
ldapUrlsSet.add(ldapUrls[i]);
}
Referral referral = searchResultReference.getReferral();
assertNotNull(referral);
for (String ldapUrl : referral.getLdapUrls()) {
if (ldapUrlsSet.contains(ldapUrl)) {
ldapUrlsSet.remove(ldapUrl);
} else {
fail(ldapUrl.toString() + " is not present");
}
}
assertTrue(ldapUrlsSet.size() == 0);
// Check the Control
Map<String, Control> controls = searchResultReference.getControls();
assertEquals(1, controls.size());
@SuppressWarnings("unchecked") CodecControl<Control> control = (org.apache.directory.api.ldap.codec.api.CodecControl<Control>) controls.get("2.16.840.1.113730.3.4.2");
assertEquals("2.16.840.1.113730.3.4.2", control.getOid());
assertEquals("", Strings.dumpBytes((byte[]) control.getValue()));
// Check the encoding
try {
ByteBuffer bb = encoder.encodeMessage(searchResultReference);
// Check the length
assertEquals(0x3F5, bb.limit());
String encodedPdu = Strings.dumpBytes(bb.array());
assertEquals(encodedPdu, decodedPdu);
} catch (EncoderException ee) {
ee.printStackTrace();
fail(ee.getMessage());
}
}
use of org.apache.directory.api.ldap.codec.decorators.SearchResultReferenceDecorator in project directory-ldap-api by apache.
the class SearchResultReferenceTest method testDecodeSearchResultReferenceNoReference.
/**
* Test the decoding of a SearchResultReference with no reference
*/
@Test
public void testDecodeSearchResultReferenceNoReference() {
Asn1Decoder ldapDecoder = new Asn1Decoder();
ByteBuffer stream = ByteBuffer.allocate(0x07);
stream.put(new byte[] { // LDAPMessage ::=SEQUENCE {
0x30, // LDAPMessage ::=SEQUENCE {
0x05, 0x02, 0x01, // messageID MessageID
0x01, 0x73, // CHOICE { ..., searchResEntry SearchResultEntry,
0x00 // ...
// SearchResultReference ::= [APPLICATION 19] SEQUENCE OF LDAPURL
});
stream.flip();
// Allocate a LdapMessage Container
LdapMessageContainer<SearchResultReferenceDecorator> ldapMessageContainer = new LdapMessageContainer<SearchResultReferenceDecorator>(codec);
// Decode a SearchResultReference message
try {
ldapDecoder.decode(stream, ldapMessageContainer);
} catch (DecoderException de) {
assertTrue(true);
return;
}
fail("We should not reach this point");
}
use of org.apache.directory.api.ldap.codec.decorators.SearchResultReferenceDecorator in project directory-ldap-api by apache.
the class MessageDecorator method getDecorator.
/**
* Gets the decorator associated with a given message
*
* @param codec The LdapApiService to use
* @param decoratedMessage The message to decorate
* @return The decorator instance
*/
public static MessageDecorator<? extends Message> getDecorator(LdapApiService codec, Message decoratedMessage) {
if (decoratedMessage instanceof MessageDecorator) {
return (MessageDecorator<?>) decoratedMessage;
}
MessageDecorator<?> decorator;
switch(decoratedMessage.getType()) {
case ABANDON_REQUEST:
decorator = new AbandonRequestDecorator(codec, (AbandonRequest) decoratedMessage);
break;
case ADD_REQUEST:
decorator = new AddRequestDecorator(codec, (AddRequest) decoratedMessage);
break;
case ADD_RESPONSE:
decorator = new AddResponseDecorator(codec, (AddResponse) decoratedMessage);
break;
case BIND_REQUEST:
decorator = new BindRequestDecorator(codec, (BindRequest) decoratedMessage);
break;
case BIND_RESPONSE:
decorator = new BindResponseDecorator(codec, (BindResponse) decoratedMessage);
break;
case COMPARE_REQUEST:
decorator = new CompareRequestDecorator(codec, (CompareRequest) decoratedMessage);
break;
case COMPARE_RESPONSE:
decorator = new CompareResponseDecorator(codec, (CompareResponse) decoratedMessage);
break;
case DEL_REQUEST:
decorator = new DeleteRequestDecorator(codec, (DeleteRequest) decoratedMessage);
break;
case DEL_RESPONSE:
decorator = new DeleteResponseDecorator(codec, (DeleteResponse) decoratedMessage);
break;
case EXTENDED_REQUEST:
decorator = codec.decorate((ExtendedRequest) decoratedMessage);
break;
case EXTENDED_RESPONSE:
decorator = codec.decorate((ExtendedResponse) decoratedMessage);
break;
case INTERMEDIATE_RESPONSE:
decorator = new IntermediateResponseDecorator(codec, (IntermediateResponse) decoratedMessage);
break;
case MODIFY_REQUEST:
decorator = new ModifyRequestDecorator(codec, (ModifyRequest) decoratedMessage);
break;
case MODIFY_RESPONSE:
decorator = new ModifyResponseDecorator(codec, (ModifyResponse) decoratedMessage);
break;
case MODIFYDN_REQUEST:
decorator = new ModifyDnRequestDecorator(codec, (ModifyDnRequest) decoratedMessage);
break;
case MODIFYDN_RESPONSE:
decorator = new ModifyDnResponseDecorator(codec, (ModifyDnResponse) decoratedMessage);
break;
case SEARCH_REQUEST:
decorator = new SearchRequestDecorator(codec, (SearchRequest) decoratedMessage);
break;
case SEARCH_RESULT_DONE:
decorator = new SearchResultDoneDecorator(codec, (SearchResultDone) decoratedMessage);
break;
case SEARCH_RESULT_ENTRY:
decorator = new SearchResultEntryDecorator(codec, (SearchResultEntry) decoratedMessage);
break;
case SEARCH_RESULT_REFERENCE:
decorator = new SearchResultReferenceDecorator(codec, (SearchResultReference) decoratedMessage);
break;
case UNBIND_REQUEST:
decorator = new UnbindRequestDecorator(codec, (UnbindRequest) decoratedMessage);
break;
default:
return null;
}
Map<String, Control> controls = decoratedMessage.getControls();
if (controls != null) {
for (Control control : controls.values()) {
decorator.addControl(control);
}
}
return decorator;
}
Aggregations