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.protocol;
021
022
023 import java.io.IOException;
024
025 import org.apache.directory.server.kerberos.shared.io.encoder.ErrorMessageEncoder;
026 import org.apache.directory.server.kerberos.shared.io.encoder.KdcReplyEncoder;
027 import org.apache.directory.server.kerberos.shared.messages.ErrorMessage;
028 import org.apache.directory.server.kerberos.shared.messages.KdcReply;
029 import org.apache.mina.core.buffer.IoBuffer;
030 import org.apache.mina.core.session.IoSession;
031 import org.apache.mina.filter.codec.ProtocolEncoderAdapter;
032 import org.apache.mina.filter.codec.ProtocolEncoderOutput;
033
034
035 /**
036 * @author <a href="mailto:dev@directory.apache.org">Apache Directory Project</a>
037 * @version $Rev$, $Date$
038 */
039 public class KerberosTcpEncoder extends ProtocolEncoderAdapter
040 {
041 private KdcReplyEncoder replyEncoder = new KdcReplyEncoder();
042 private ErrorMessageEncoder errorEncoder = new ErrorMessageEncoder();
043
044
045 public void encode( IoSession session, Object message, ProtocolEncoderOutput out ) throws IOException
046 {
047 IoBuffer buf = IoBuffer.allocate( 1024 );
048
049 // make space for int length
050 buf.putInt( 0 );
051
052 if ( message instanceof KdcReply )
053 {
054 replyEncoder.encode( ( KdcReply ) message, buf.buf() );
055 }
056 else
057 {
058 if ( message instanceof ErrorMessage )
059 {
060 errorEncoder.encode( ( ErrorMessage ) message, buf.buf() );
061 }
062 }
063
064 // mark position
065 int pos = buf.position();
066
067 // length is the data minus 4 bytes for the pre-pended length
068 int recordLength = buf.position() - 4;
069
070 // write the length
071 buf.rewind();
072 buf.putInt( recordLength );
073
074 // set the position back before flipping the buffer
075 buf.position( pos );
076 buf.flip();
077
078 out.write( buf );
079 }
080 }