blob: 176eaeba8d4ce3995a250c917d0a5df6a9ad7b07 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
package com.btr.proxy.selector.whitelist;
import java.net.URI;
import com.btr.proxy.util.UriFilter;
/*****************************************************************************
* Tests if a host name of a given URI matches some criteria.
*
* @author Bernd Rosstauscher (proxyvole@rosstauscher.de) Copyright 2009
****************************************************************************/
public class HostnameFilter implements UriFilter {
private static final String PROTOCOL_ENDING = "://";
public enum Mode {BEGINS_WITH, ENDS_WITH, REGEX}
private String matchTo;
private String protocolFilter;
private Mode mode;
/*************************************************************************
* Constructor
* @param mode the filter mode.
* @param matchTo the match criteria.
************************************************************************/
public HostnameFilter(Mode mode, String matchTo) {
super();
this.mode = mode;
this.matchTo = matchTo.toLowerCase();
extractProtocolFilter();
}
/*************************************************************************
* Extracts the protocol if one is given to initialize the protocol matcher.
************************************************************************/
private void extractProtocolFilter() {
int protocolIndex = this.matchTo.indexOf(PROTOCOL_ENDING);
if (protocolIndex != -1) {
this.protocolFilter = this.matchTo.substring(0, protocolIndex);
this.matchTo = this.matchTo.substring(protocolIndex+PROTOCOL_ENDING.length());
}
}
/*************************************************************************
* accept
* @see com.btr.proxy.util.UriFilter#accept(java.net.URI)
************************************************************************/
public boolean accept(URI uri) {
if (uri == null || uri.getAuthority() == null) {
return false;
}
if (!isProtocolMatching(uri)) {
return false;
}
String host = uri.getAuthority();
// Strip away port.
int index = host.indexOf(':');
if (index != -1) {
host = host.substring(0, index);
}
switch (this.mode) {
case BEGINS_WITH :
return host.toLowerCase().startsWith(this.matchTo);
case ENDS_WITH :
return host.toLowerCase().endsWith(this.matchTo);
case REGEX :
return host.toLowerCase().matches(this.matchTo);
}
return false;
}
/*************************************************************************
* Applies the protocol filter if available to see if we have a match.
* @param uri to test for a correct protocol.
* @return true if passed else false.
************************************************************************/
private boolean isProtocolMatching(URI uri) {
return this.protocolFilter == null
|| uri.getScheme() == null
|| uri.getScheme().equalsIgnoreCase(this.protocolFilter);
}
}
|