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: 725712 $, $Date: 2008-12-11 17:32:04 +0200 (Thu, 11 Dec 2008) $
038 */
039 public class KerberosUdpEncoder 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 if ( message instanceof KdcReply )
050 {
051 replyEncoder.encode( ( KdcReply ) message, buf.buf() );
052 }
053 else
054 {
055 if ( message instanceof ErrorMessage )
056 {
057 errorEncoder.encode( ( ErrorMessage ) message, buf.buf() );
058 }
059 }
060
061 buf.flip();
062
063 out.write( buf );
064 }
065 }