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, software 013 * distributed under the License is distributed on an "AS IS" BASIS, 014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 015 * See the License for the specific language governing permissions and 016 * limitations under the License. 017 */ 018package org.apache.hadoop.hbase.security; 019 020import javax.security.sasl.SaslException; 021import org.apache.yetus.audience.InterfaceAudience; 022 023import org.apache.hbase.thirdparty.io.netty.buffer.ByteBuf; 024import org.apache.hbase.thirdparty.io.netty.channel.ChannelHandlerContext; 025import org.apache.hbase.thirdparty.io.netty.handler.codec.MessageToByteEncoder; 026 027/** 028 * wrap sasl messages. 029 */ 030@InterfaceAudience.Private 031public class SaslWrapHandler extends MessageToByteEncoder<ByteBuf> { 032 033 public interface Wrapper { 034 byte[] wrap(byte[] outgoing, int offset, int len) throws SaslException; 035 } 036 037 private final Wrapper wrapper; 038 039 public SaslWrapHandler(Wrapper wrapper) { 040 this.wrapper = wrapper; 041 } 042 043 @Override 044 protected void encode(ChannelHandlerContext ctx, ByteBuf msg, ByteBuf out) throws Exception { 045 byte[] bytes = new byte[msg.readableBytes()]; 046 msg.readBytes(bytes); 047 byte[] wrapperBytes = wrapper.wrap(bytes, 0, bytes.length); 048 out.ensureWritable(4 + wrapperBytes.length); 049 out.writeInt(wrapperBytes.length); 050 out.writeBytes(wrapperBytes); 051 } 052}