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    
026    import org.apache.directory.server.kerberos.shared.messages.application.PrivateMessage;
027    import org.apache.directory.shared.asn1.der.ASN1OutputStream;
028    import org.apache.directory.shared.asn1.der.DERApplicationSpecific;
029    import org.apache.directory.shared.asn1.der.DERInteger;
030    import org.apache.directory.shared.asn1.der.DERSequence;
031    import org.apache.directory.shared.asn1.der.DERTaggedObject;
032    
033    
034    /**
035     * @author <a href="mailto:dev@directory.apache.org">Apache Directory Project</a>
036     * @version $Rev: 540371 $, $Date: 2007-05-22 02:00:43 +0200 (Tue, 22 May 2007) $
037     */
038    public class PrivateMessageEncoder
039    {
040        /**
041         * Encodes a {@link PrivateMessage} into a byte array.
042         *
043         * @param message
044         * @return The byte array.
045         * @throws IOException
046         */
047        public byte[] encode( PrivateMessage message ) throws IOException
048        {
049            ByteArrayOutputStream baos = new ByteArrayOutputStream();
050            ASN1OutputStream aos = new ASN1OutputStream( baos );
051    
052            DERSequence privateMessageSequence = encodePrivateMessageSequence( message );
053            aos
054                .writeObject( DERApplicationSpecific
055                    .valueOf( message.getMessageType().getOrdinal(), privateMessageSequence ) );
056            aos.close();
057    
058            return baos.toByteArray();
059        }
060    
061    
062        private DERSequence encodePrivateMessageSequence( PrivateMessage message )
063        {
064            DERSequence sequence = new DERSequence();
065    
066            sequence.add( new DERTaggedObject( 0, DERInteger.valueOf( message.getProtocolVersionNumber() ) ) );
067            sequence.add( new DERTaggedObject( 1, DERInteger.valueOf( message.getMessageType().getOrdinal() ) ) );
068            sequence.add( new DERTaggedObject( 3, EncryptedDataEncoder.encodeSequence( message.getEncryptedPart() ) ) );
069    
070            return sequence;
071        }
072    }