From 6bbaa8feafb820d6cefe516efe0ac7df3bfb07ae Mon Sep 17 00:00:00 2001 From: Michael Gorovoy Date: Wed, 16 Jun 2010 22:55:43 +0000 Subject: 288194 Add blacklist/whitelist to ProxyServlet and ProxyHandler git-svn-id: svn+ssh://dev.eclipse.org/svnroot/rt/org.eclipse.jetty/jetty/trunk@2020 7e9141cc-0065-0410-87d8-b60c137991c4 --- .../main/java/org/eclipse/jetty/util/HostMap.java | 101 +++++++++++++++++++++ 1 file changed, 101 insertions(+) create mode 100644 jetty-util/src/main/java/org/eclipse/jetty/util/HostMap.java (limited to 'jetty-util') diff --git a/jetty-util/src/main/java/org/eclipse/jetty/util/HostMap.java b/jetty-util/src/main/java/org/eclipse/jetty/util/HostMap.java new file mode 100644 index 0000000000..768171561e --- /dev/null +++ b/jetty-util/src/main/java/org/eclipse/jetty/util/HostMap.java @@ -0,0 +1,101 @@ +// ======================================================================== +// Copyright (c) 2010 Mort Bay Consulting Pty. Ltd. +// ------------------------------------------------------------------------ +// All rights reserved. This program and the accompanying materials +// are made available under the terms of the Eclipse Public License v1.0 +// and Apache License v2.0 which accompanies this distribution. +// The Eclipse Public License is available at +// http://www.eclipse.org/legal/epl-v10.html +// The Apache License v2.0 is available at +// http://www.opensource.org/licenses/apache2.0.php +// You may elect to redistribute this code under either of these licenses. +// ======================================================================== + + +package org.eclipse.jetty.util; + +import java.util.HashMap; +import java.util.HashSet; +import java.util.Map; + +/* ------------------------------------------------------------ */ +/** + */ +public class HostMap extends HashMap +{ + /* --------------------------------------------------------------- */ + /** Construct empty HostMap. + */ + public HostMap() + { + super(11); + } + + /* --------------------------------------------------------------- */ + /** Construct empty HostMap. + * + * @param capacity initial capacity + */ + public HostMap(int capacity) + { + super (capacity); + } + + /* ------------------------------------------------------------ */ + /** + * @see java.util.HashMap#put(java.lang.Object, java.lang.Object) + */ + @Override + public TYPE put(String host, TYPE object) + throws IllegalArgumentException + { + return super.put(host, object); + } + + /* ------------------------------------------------------------ */ + /** + * @see java.util.HashMap#get(java.lang.Object) + */ + @Override + public TYPE get(Object key) + { + return super.get(key); + } + + /* ------------------------------------------------------------ */ + /** + * Retrieve a lazy list of map entries associated with specified + * hostname by taking into account the domain suffix matches. + * + * @param host hostname + * @return lazy list of map entries + */ + public Object getLazyMatches(String host) + { + if (host == null) + return LazyList.getList(super.entrySet()); + + int idx = 0; + String domain = host.trim(); + HashSet domains = new HashSet(); + do { + domains.add(domain); + if ((idx = domain.indexOf('.')) > 0) + { + domain = domain.substring(idx+1); + } + } while (idx > 0); + + Object entries = null; + for(Map.Entry entry: super.entrySet()) + { + if (domains.contains(entry.getKey())) + { + entries = LazyList.add(entries,entry); + } + } + + return entries; + } + +} -- cgit v1.2.3