svn merge -c 1587608 FIXES: HADOOP-10498. Add support for proxy server. (daryn)

git-svn-id: https://svn.apache.org/repos/asf/hadoop/common/branches/branch-2@1587609 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Daryn Sharp 2014-04-15 15:27:04 +00:00
parent 7c89cbfcd2
commit aa2f2da767
3 changed files with 34 additions and 1 deletions

View File

@ -8,6 +8,8 @@ Release 2.5.0 - UNRELEASED
NEW FEATURES NEW FEATURES
HADOOP-10498. Add support for proxy server. (daryn)
IMPROVEMENTS IMPROVEMENTS
HADOOP-10451. Remove unused field and imports from SaslRpcServer. HADOOP-10451. Remove unused field and imports from SaslRpcServer.

View File

@ -19,9 +19,11 @@
package org.apache.hadoop.security.authorize; package org.apache.hadoop.security.authorize;
import java.net.InetAddress; import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.UnknownHostException; import java.net.UnknownHostException;
import java.util.Collection; import java.util.Collection;
import java.util.HashMap; import java.util.HashMap;
import java.util.HashSet;
import java.util.Map; import java.util.Map;
import java.util.Map.Entry; import java.util.Map.Entry;
@ -39,12 +41,16 @@ public class ProxyUsers {
public static final String CONF_GROUPS = ".groups"; public static final String CONF_GROUPS = ".groups";
public static final String CONF_HADOOP_PROXYUSER = "hadoop.proxyuser."; public static final String CONF_HADOOP_PROXYUSER = "hadoop.proxyuser.";
public static final String CONF_HADOOP_PROXYUSER_RE = "hadoop\\.proxyuser\\."; public static final String CONF_HADOOP_PROXYUSER_RE = "hadoop\\.proxyuser\\.";
public static final String CONF_HADOOP_PROXYSERVERS = "hadoop.proxyservers";
private static boolean init = false; private static boolean init = false;
// list of groups and hosts per proxyuser // list of groups and hosts per proxyuser
private static Map<String, Collection<String>> proxyGroups = private static Map<String, Collection<String>> proxyGroups =
new HashMap<String, Collection<String>>(); new HashMap<String, Collection<String>>();
private static Map<String, Collection<String>> proxyHosts = private static Map<String, Collection<String>> proxyHosts =
new HashMap<String, Collection<String>>(); new HashMap<String, Collection<String>>();
private static Collection<String> proxyServers =
new HashSet<String>();
/** /**
* reread the conf and get new values for "hadoop.proxyuser.*.groups/hosts" * reread the conf and get new values for "hadoop.proxyuser.*.groups/hosts"
@ -60,9 +66,10 @@ public class ProxyUsers {
*/ */
public static synchronized void refreshSuperUserGroupsConfiguration(Configuration conf) { public static synchronized void refreshSuperUserGroupsConfiguration(Configuration conf) {
// remove alle existing stuff // remove all existing stuff
proxyGroups.clear(); proxyGroups.clear();
proxyHosts.clear(); proxyHosts.clear();
proxyServers.clear();
// get all the new keys for groups // get all the new keys for groups
String regex = CONF_HADOOP_PROXYUSER_RE+"[^.]*\\"+CONF_GROUPS; String regex = CONF_HADOOP_PROXYUSER_RE+"[^.]*\\"+CONF_GROUPS;
@ -80,9 +87,23 @@ public class ProxyUsers {
StringUtils.getTrimmedStringCollection(entry.getValue())); StringUtils.getTrimmedStringCollection(entry.getValue()));
} }
// trusted proxy servers such as http proxies
for (String host : conf.getTrimmedStrings(CONF_HADOOP_PROXYSERVERS)) {
InetSocketAddress addr = new InetSocketAddress(host, 0);
if (!addr.isUnresolved()) {
proxyServers.add(addr.getAddress().getHostAddress());
}
}
init = true; init = true;
} }
public static synchronized boolean isProxyServer(String remoteAddr) {
if(!init) {
refreshSuperUserGroupsConfiguration();
}
return proxyServers.contains(remoteAddr);
}
/** /**
* Returns configuration key for effective user groups allowed for a superuser * Returns configuration key for effective user groups allowed for a superuser
* *

View File

@ -169,6 +169,16 @@ public class TestProxyUsers {
assertEquals (1,hosts.size()); assertEquals (1,hosts.size());
} }
@Test
public void testProxyServer() {
Configuration conf = new Configuration();
assertFalse(ProxyUsers.isProxyServer("1.1.1.1"));
conf.set(ProxyUsers.CONF_HADOOP_PROXYSERVERS, "2.2.2.2, 3.3.3.3");
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
assertFalse(ProxyUsers.isProxyServer("1.1.1.1"));
assertTrue(ProxyUsers.isProxyServer("2.2.2.2"));
assertTrue(ProxyUsers.isProxyServer("3.3.3.3"));
}
private void assertNotAuthorized(UserGroupInformation proxyUgi, String host) { private void assertNotAuthorized(UserGroupInformation proxyUgi, String host) {
try { try {