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 }