0001
0002
0003
0004
0005
0006
0007
0008
0009
0010
0011
0012
0013
0014
0015
0016
0017
0018
0019 package org.apache.hive.service.auth;
0020
0021 import org.apache.hive.service.cli.thrift.TCLIService;
0022 import org.apache.hive.service.cli.thrift.TCLIService.Iface;
0023 import org.apache.thrift.TException;
0024 import org.apache.thrift.protocol.TProtocol;
0025 import org.apache.thrift.transport.TSaslClientTransport;
0026 import org.apache.thrift.transport.TSaslServerTransport;
0027 import org.apache.thrift.transport.TSocket;
0028 import org.apache.thrift.transport.TTransport;
0029 import org.slf4j.Logger;
0030 import org.slf4j.LoggerFactory;
0031
0032
0033
0034
0035
0036
0037
0038
0039
0040 public class TSetIpAddressProcessor<I extends Iface> extends TCLIService.Processor<Iface> {
0041
0042 private static final Logger LOGGER = LoggerFactory.getLogger(TSetIpAddressProcessor.class.getName());
0043
0044 public TSetIpAddressProcessor(Iface iface) {
0045 super(iface);
0046 }
0047
0048 @Override
0049 public boolean process(final TProtocol in, final TProtocol out) throws TException {
0050 setIpAddress(in);
0051 setUserName(in);
0052 try {
0053 return super.process(in, out);
0054 } finally {
0055 THREAD_LOCAL_USER_NAME.remove();
0056 THREAD_LOCAL_IP_ADDRESS.remove();
0057 }
0058 }
0059
0060 private void setUserName(final TProtocol in) {
0061 TTransport transport = in.getTransport();
0062 if (transport instanceof TSaslServerTransport) {
0063 String userName = ((TSaslServerTransport) transport).getSaslServer().getAuthorizationID();
0064 THREAD_LOCAL_USER_NAME.set(userName);
0065 }
0066 }
0067
0068 protected void setIpAddress(final TProtocol in) {
0069 TTransport transport = in.getTransport();
0070 TSocket tSocket = getUnderlyingSocketFromTransport(transport);
0071 if (tSocket == null) {
0072 LOGGER.warn("Unknown Transport, cannot determine ipAddress");
0073 } else {
0074 THREAD_LOCAL_IP_ADDRESS.set(tSocket.getSocket().getInetAddress().getHostAddress());
0075 }
0076 }
0077
0078 private TSocket getUnderlyingSocketFromTransport(TTransport transport) {
0079 while (transport != null) {
0080 if (transport instanceof TSaslServerTransport) {
0081 transport = ((TSaslServerTransport) transport).getUnderlyingTransport();
0082 }
0083 if (transport instanceof TSaslClientTransport) {
0084 transport = ((TSaslClientTransport) transport).getUnderlyingTransport();
0085 }
0086 if (transport instanceof TSocket) {
0087 return (TSocket) transport;
0088 }
0089 }
0090 return null;
0091 }
0092
0093 private static final ThreadLocal<String> THREAD_LOCAL_IP_ADDRESS = new ThreadLocal<String>() {
0094 @Override
0095 protected synchronized String initialValue() {
0096 return null;
0097 }
0098 };
0099
0100 private static final ThreadLocal<String> THREAD_LOCAL_USER_NAME = new ThreadLocal<String>() {
0101 @Override
0102 protected synchronized String initialValue() {
0103 return null;
0104 }
0105 };
0106
0107 public static String getUserIpAddress() {
0108 return THREAD_LOCAL_IP_ADDRESS.get();
0109 }
0110
0111 public static String getUserName() {
0112 return THREAD_LOCAL_USER_NAME.get();
0113 }
0114 }