001 /* 002 * Licensed to the Apache Software Foundation (ASF) under one 003 * or more contributor license agreements. See the NOTICE file 004 * distributed with this work for additional information 005 * regarding copyright ownership. The ASF licenses this file 006 * to you under the Apache License, Version 2.0 (the 007 * "License"); you may not use this file except in compliance 008 * with the License. You may obtain a copy of the License at 009 * 010 * http://www.apache.org/licenses/LICENSE-2.0 011 * 012 * Unless required by applicable law or agreed to in writing, 013 * software distributed under the License is distributed on an 014 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 015 * KIND, either express or implied. See the License for the 016 * specific language governing permissions and limitations 017 * under the License. 018 * 019 */ 020 package org.apache.directory.server.kerberos.shared.io.encoder; 021 022 023 import java.io.ByteArrayOutputStream; 024 import java.io.IOException; 025 import java.nio.ByteBuffer; 026 027 import org.apache.directory.server.kerberos.shared.messages.ErrorMessage; 028 import org.apache.directory.shared.asn1.der.ASN1OutputStream; 029 import org.apache.directory.shared.asn1.der.DERApplicationSpecific; 030 import org.apache.directory.shared.asn1.der.DERGeneralString; 031 import org.apache.directory.shared.asn1.der.DERInteger; 032 import org.apache.directory.shared.asn1.der.DEROctetString; 033 import org.apache.directory.shared.asn1.der.DERSequence; 034 import org.apache.directory.shared.asn1.der.DERTaggedObject; 035 036 037 /** 038 * @author <a href="mailto:dev@directory.apache.org">Apache Directory Project</a> 039 * @version $Rev: 540371 $, $Date: 2007-05-22 02:00:43 +0200 (Tue, 22 May 2007) $ 040 */ 041 public class ErrorMessageEncoder 042 { 043 /** 044 * Encodes an {@link ErrorMessage} into a {@link ByteBuffer}. 045 * 046 * @param message 047 * @param out 048 * @throws IOException 049 */ 050 public void encode( ErrorMessage message, ByteBuffer out ) throws IOException 051 { 052 ASN1OutputStream aos = new ASN1OutputStream( out ); 053 054 DERSequence errorReply = encodeErrorMessageSequence( message ); 055 aos.writeObject( DERApplicationSpecific.valueOf( message.getMessageType().getOrdinal(), errorReply ) ); 056 057 aos.close(); 058 } 059 060 061 /** 062 * Encodes an {@link ErrorMessage} into a byte array. 063 * 064 * @param message 065 * @return The byte array. 066 * @throws IOException 067 */ 068 public byte[] encode( ErrorMessage message ) throws IOException 069 { 070 ByteArrayOutputStream baos = new ByteArrayOutputStream(); 071 ASN1OutputStream aos = new ASN1OutputStream( baos ); 072 073 DERSequence errorReply = encodeErrorMessageSequence( message ); 074 aos.writeObject( DERApplicationSpecific.valueOf( message.getMessageType().getOrdinal(), errorReply ) ); 075 076 aos.close(); 077 078 return baos.toByteArray(); 079 } 080 081 082 private DERSequence encodeErrorMessageSequence( ErrorMessage message ) 083 { 084 DERSequence sequence = new DERSequence(); 085 086 sequence.add( new DERTaggedObject( 0, DERInteger.valueOf( message.getProtocolVersionNumber() ) ) ); 087 088 sequence.add( new DERTaggedObject( 1, DERInteger.valueOf( message.getMessageType().getOrdinal() ) ) ); 089 090 if ( message.getClientTime() != null ) 091 { 092 sequence.add( new DERTaggedObject( 2, KerberosTimeEncoder.encode( message.getClientTime() ) ) ); 093 } 094 095 if ( message.getClientMicroSecond() != null ) 096 { 097 sequence.add( new DERTaggedObject( 3, DERInteger.valueOf( message.getClientMicroSecond().intValue() ) ) ); 098 } 099 100 sequence.add( new DERTaggedObject( 4, KerberosTimeEncoder.encode( message.getServerTime() ) ) ); 101 102 sequence.add( new DERTaggedObject( 5, DERInteger.valueOf( message.getServerMicroSecond() ) ) ); 103 104 sequence.add( new DERTaggedObject( 6, DERInteger.valueOf( message.getErrorCode() ) ) ); 105 106 if ( message.getClientPrincipal() != null ) 107 { 108 sequence.add( new DERTaggedObject( 7, DERGeneralString.valueOf( message.getClientPrincipal().getRealm() 109 .toString() ) ) ); 110 } 111 112 if ( message.getClientPrincipal() != null ) 113 { 114 sequence.add( new DERTaggedObject( 8, PrincipalNameEncoder.encode( message.getClientPrincipal() ) ) ); 115 } 116 117 sequence.add( new DERTaggedObject( 9, DERGeneralString.valueOf( message.getServerPrincipal().getRealm() ) ) ); 118 119 sequence.add( new DERTaggedObject( 10, PrincipalNameEncoder.encode( message.getServerPrincipal() ) ) ); 120 121 if ( message.getExplanatoryText() != null ) 122 { 123 sequence.add( new DERTaggedObject( 11, DERGeneralString.valueOf( message.getExplanatoryText() ) ) ); 124 } 125 126 if ( message.getExplanatoryData() != null ) 127 { 128 sequence.add( new DERTaggedObject( 12, new DEROctetString( message.getExplanatoryData() ) ) ); 129 } 130 131 return sequence; 132 } 133 }