2002/08/07 04:44:53
[org.ibex.core.git] / src / org / xwt / Proxy.java
1 // Copyright 2002 Adam Megacz, see the COPYING file for licensing [GPL]
2 package org.xwt;
3
4 import java.net.*;
5 import java.io.*;
6 import java.util.*;
7 import org.xwt.util.*;
8 import org.mozilla.javascript.*;
9 import org.bouncycastle.util.encoders.Base64;
10
11 /** encapsulates most of the proxy logic; some is shared in HTTP.java */
12 public class Proxy {
13     
14     public Proxy() { }
15     
16     /** the HTTP Proxy host to use */
17     public String httpProxyHost = null;
18     
19     /** the HTTP Proxy port to use */
20     public int httpProxyPort = -1;
21     
22     /** if a seperate proxy should be used for HTTPS, this is the hostname; otherwise, httpProxyHost is used */
23     public String httpsProxyHost = null;
24     
25     /** if a seperate proxy should be used for HTTPS, this is the port */
26     public int httpsProxyPort = -1;
27     
28     /** the SOCKS Proxy Host to use */
29     public String socksProxyHost = null;
30     
31     /** the SOCKS Proxy Port to use */
32     public int socksProxyPort = -1;
33     
34     /** hosts to be excluded from proxy use; wildcards permitted */
35     public String[] excluded = null;
36     
37     /** the PAC script */
38     public Function proxyAutoConfigFunction = null;
39     
40     public static Proxy detectProxyViaManual() {
41         Proxy ret = new Proxy();
42         
43         ret.httpProxyHost = Platform.getEnv("http_proxy");
44         if (ret.httpProxyHost != null) {
45             if (ret.httpProxyHost.startsWith("http://")) ret.httpProxyHost = ret.httpProxyHost.substring(7);
46             if (ret.httpProxyHost.endsWith("/")) ret.httpProxyHost = ret.httpProxyHost.substring(0, ret.httpProxyHost.length() - 1);
47             if (ret.httpProxyHost.indexOf(':') != -1) {
48                 ret.httpProxyPort = Integer.parseInt(ret.httpProxyHost.substring(ret.httpProxyHost.indexOf(':') + 1));
49                 ret.httpProxyHost = ret.httpProxyHost.substring(0, ret.httpProxyHost.indexOf(':'));
50             } else {
51                 ret.httpProxyPort = 80;
52             }
53         }
54         
55         ret.httpsProxyHost = Platform.getEnv("https_proxy");
56         if (ret.httpsProxyHost != null) {
57             if (ret.httpsProxyHost.startsWith("https://")) ret.httpsProxyHost = ret.httpsProxyHost.substring(7);
58             if (ret.httpsProxyHost.endsWith("/")) ret.httpsProxyHost = ret.httpsProxyHost.substring(0, ret.httpsProxyHost.length() - 1);
59             if (ret.httpsProxyHost.indexOf(':') != -1) {
60                 ret.httpsProxyPort = Integer.parseInt(ret.httpsProxyHost.substring(ret.httpsProxyHost.indexOf(':') + 1));
61                 ret.httpsProxyHost = ret.httpsProxyHost.substring(0, ret.httpsProxyHost.indexOf(':'));
62             } else {
63                 ret.httpsProxyPort = 80;
64             }
65         }
66         
67         ret.socksProxyHost = Platform.getEnv("socks_proxy");
68         if (ret.socksProxyHost != null) {
69             if (ret.socksProxyHost.startsWith("socks://")) ret.socksProxyHost = ret.socksProxyHost.substring(7);
70             if (ret.socksProxyHost.endsWith("/")) ret.socksProxyHost = ret.socksProxyHost.substring(0, ret.socksProxyHost.length() - 1);
71             if (ret.socksProxyHost.indexOf(':') != -1) {
72                 ret.socksProxyPort = Integer.parseInt(ret.socksProxyHost.substring(ret.socksProxyHost.indexOf(':') + 1));
73                 ret.socksProxyHost = ret.socksProxyHost.substring(0, ret.socksProxyHost.indexOf(':'));
74             } else {
75                 ret.socksProxyPort = 80;
76             }
77         }
78         
79         String noproxy = Platform.getEnv("no_proxy");
80         if (noproxy != null) {
81             StringTokenizer st = new StringTokenizer(noproxy, ",");
82             ret.excluded = new String[st.countTokens()];
83             for(int i=0; st.hasMoreTokens(); i++) ret.excluded[i] = st.nextToken();
84         }
85         
86         if (ret.httpProxyHost == null && ret.socksProxyHost == null) return null;
87         return ret;
88     }
89     
90     public static Scriptable proxyAutoConfigRootScope = new ProxyAutoConfigRootScope();
91     
92     public static Function getProxyAutoConfigFunction(String url) {
93         try { 
94             Context cx = Context.enter();
95             cx.setOptimizationLevel(-1);
96             BufferedReader br = new BufferedReader(new InputStreamReader(new HTTP(url, true).GET()));
97             String s = null;
98             String script = "";
99             while((s = br.readLine()) != null) script += s + "\n";
100             if (Log.on) Log.log(Proxy.class, "successfully retrieved WPAD PAC:");
101             if (Log.on) Log.log(Proxy.class, script);
102             
103             // MS CARP hack
104             Vector carpHosts = new Vector();
105             for(int i=0; i<script.length(); i++)
106                 if (script.regionMatches(i, "new Node(", 0, 9)) {
107                     String host = script.substring(i + 10, script.indexOf('\"', i + 11));
108                     if (Log.on) Log.log(Proxy.class, "Detected MS Proxy Server CARP Script, Host=" + host);
109                     carpHosts.addElement(host);
110                 }
111             if (carpHosts.size() > 0) {
112                 script = "function FindProxyForURL(url, host) {\nreturn \"";
113                 for(int i=0; i<carpHosts.size(); i++)
114                     script += "PROXY " + carpHosts.elementAt(i) + "; ";
115                 script += "\";\n}";
116                 if (Log.on) Log.log(Proxy.class, "DeCARPed PAC script:");
117                 if (Log.on) Log.log(Proxy.class, script);
118             }
119             
120             Script scr = cx.compileReader(proxyAutoConfigRootScope, new StringReader(script), "PAC script at " + url, 0, null);
121             scr.exec(cx, proxyAutoConfigRootScope);
122             return (Function)proxyAutoConfigRootScope.get("FindProxyForURL", null);
123         } catch (Exception e) {
124             if (Log.on) {
125                 Log.log(Platform.class, "WPAD detection failed due to:");
126                 if (e instanceof EcmaError) Log.log(HTTP.class, ((EcmaError)e).getMessage() + " at " +
127                                                     ((EcmaError)e).getSourceName() + ":" + ((EcmaError)e).getLineNumber());
128                 else Log.log(Platform.class, e);
129             }
130             return null;
131         }
132     }
133
134
135     // Authorization ///////////////////////////////////////////////////////////////////////////////////
136
137     public static class Authorization {
138
139         static public String authorization = null;
140         static public String authorization2 = null;
141         static public Semaphore waitingForUser = new Semaphore();
142
143         public static synchronized void getPassword(final String realm, final String style, final String proxyIP, String oldAuth) {
144
145             // this handles cases where multiple threads hit the proxy auth at the same time -- all but one will block on the
146             // synchronized keyword. If 'authorization' changed while the thread was blocked, it means that the user entered
147             // a password, so we should reattempt authorization.
148
149             if (authorization != oldAuth) return;
150             if (Log.on) Log.log(Authorization.class, "displaying proxy authorization dialog");
151             MessageQueue.add(new Message() {
152                     public void perform() {
153                         Box b = new Box("org.xwt.builtin.proxy_authorization", null);
154                         b.put("realm", realm);
155                         b.put("proxyIP", proxyIP);
156                     }
157                 });
158
159             waitingForUser.block();
160             if (Log.on) Log.log(Authorization.class, "got proxy authorization info; re-attempting connection");
161             
162         }
163     }
164
165
166     // ProxyAutoConfigRootScope ////////////////////////////////////////////////////////////////////
167     
168     public static class ProxyAutoConfigRootScope extends ScriptableObject {
169         
170         public String getClassName() { return "ProxyAutoConfigRootScope"; }
171         ProxyAutoConfigRootScope() { Context.enter().initStandardObjects(this); }
172         
173         public Object get(String name, Scriptable start) {
174             if (name.equals("isPlainHostName")) return isPlainHostName;
175             else if (name.equals("dnsDomainIs")) return dnsDomainIs;
176             else if (name.equals("localHostOrDomainIs")) return localHostOrDomainIs;
177             else if (name.equals("isResolvable")) return isResolvable;
178             else if (name.equals("isInNet")) return isInNet;
179             else if (name.equals("dnsResolve")) return dnsResolve;
180             else if (name.equals("myIpAddress")) return myIpAddress;
181             else if (name.equals("dnsDomainLevels")) return dnsDomainLevels;
182             else if (name.equals("shExpMatch")) return shExpMatch;
183             else if (name.equals("weekdayRange")) return weekdayRange;
184             else if (name.equals("dateRange")) return dateRange;
185             else if (name.equals("timeRange")) return timeRange;
186             else if (name.equals("ProxyConfig")) return ProxyConfig;
187             else return super.get(name, start);
188         }
189         
190         private static final JSObject proxyConfigBindings = new JSObject();
191         private static final JSObject ProxyConfig = new JSObject() {
192                 public Object get(String name, Scriptable start) {
193                     if (name.equals("bindings")) return proxyConfigBindings;
194                     return null;
195                 }
196             };
197         
198         private static abstract class JSFunction extends JSObject implements Function {
199             JSFunction() { setSeal(true); }
200             public Scriptable construct(Context cx, Scriptable scope, java.lang.Object[] args) { return null; }
201         }
202         
203         private static final JSFunction isPlainHostName = new JSFunction() {
204                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
205                     return (args[0].toString().indexOf('.') == -1) ? Boolean.TRUE : Boolean.FALSE;
206                 }
207             };
208         
209         private static final JSFunction dnsDomainIs = new JSFunction() {
210                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
211                     return (args[0].toString().endsWith(args[1].toString())) ? Boolean.TRUE : Boolean.FALSE;
212                 }
213             };
214         
215         private static final JSFunction localHostOrDomainIs = new JSFunction() {
216                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
217                     return (args[0].toString().equals(args[1].toString()) || 
218                             (args[0].toString().indexOf('.') == -1 && args[1].toString().startsWith(args[0].toString()))) ?
219                         Boolean.TRUE : Boolean.FALSE;
220                 }
221             };
222         
223         private static final JSFunction isResolvable = new JSFunction() {
224                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
225                     try {
226                         return (InetAddress.getByName(args[0].toString()) != null) ? Boolean.TRUE : Boolean.FALSE;
227                     } catch (UnknownHostException e) {
228                         return Boolean.FALSE;
229                     }
230                 }
231             };
232         
233         private static final JSFunction isInNet = new JSFunction() {
234                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
235                     if (args.length != 3) return Boolean.FALSE;
236                     try {
237                         byte[] host = InetAddress.getByName(args[0].toString()).getAddress();
238                         byte[] net = InetAddress.getByName(args[1].toString()).getAddress();
239                         byte[] mask = InetAddress.getByName(args[2].toString()).getAddress();
240                         return ((host[0] & mask[0]) == net[0] &&
241                                 (host[1] & mask[1]) == net[1] &&
242                                 (host[2] & mask[2]) == net[2] &&
243                                 (host[3] & mask[3]) == net[3]) ?
244                             Boolean.TRUE : Boolean.FALSE;
245                     } catch (Exception e) {
246                         throw new JavaScriptException("exception in isInNet(): " + e);
247                     }
248                 }
249             };
250         
251         private static final JSFunction dnsResolve = new JSFunction() {
252                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
253                     try {
254                         return InetAddress.getByName(args[0].toString()).getHostAddress();
255                     } catch (UnknownHostException e) {
256                         return null;
257                     }
258                 }
259             };
260         
261         private static final JSFunction myIpAddress = new JSFunction() {
262                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
263                     try {
264                         return InetAddress.getLocalHost().getHostAddress();
265                     } catch (UnknownHostException e) {
266                         if (Log.on) Log.log(this, "strange... host does not know its own address");
267                         return null;
268                     }
269                 }
270             };
271         
272         private static final JSFunction dnsDomainLevels = new JSFunction() {
273                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
274                     String s = args[0].toString();
275                     int i = 0;
276                     while((i = s.indexOf('.', i)) != -1) i++;
277                     return new Integer(i);
278                 }
279             };
280         
281         private static boolean match(String[] arr, String s, int index) {
282             if (index == arr.length) return true;
283             for(int i=0; i<s.length(); i++) {
284                 String s2 = s.substring(i);
285                 if (s2.startsWith(arr[index]) && match(arr, s.substring(arr[index].length()), index + 1)) return true;
286             }
287             return false;
288         }
289         
290         private static final JSFunction shExpMatch = new JSFunction() {
291                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
292                     StringTokenizer st = new StringTokenizer(args[1].toString(), "*", false);
293                     String[] arr = new String[st.countTokens()];
294                     String s = args[0].toString();
295                     for (int i=0; st.hasMoreTokens(); i++) arr[i] = st.nextToken();
296                     return match(arr, s, 0) ? Boolean.TRUE : Boolean.FALSE;
297                 }
298             };
299         
300         public static String[] days = { "SUN", "MON", "TUE", "WED", "THU", "FRI", "SAT" };
301         
302         private static final JSFunction weekdayRange = new JSFunction() {
303                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
304                     TimeZone tz = (args.length < 3 || args[2] == null || !args[2].equals("GMT")) ? TimeZone.getTimeZone("UTC") : TimeZone.getDefault();
305                     Calendar c = new GregorianCalendar();
306                     c.setTimeZone(tz);
307                     c.setTime(new Date());
308                     Date d = c.getTime();
309                     int day = d.getDay();
310                     
311                     String d1s = args[0].toString().toUpperCase();
312                     int d1 = 0, d2 = 0;
313                     for(int i=0; i<days.length; i++) if (days[i].equals(d1s)) d1 = i;
314                     
315                     if (args.length == 1)
316                         return d1 == day ? Boolean.TRUE : Boolean.FALSE;
317                     
318                     String d2s = args[1].toString().toUpperCase();
319                     for(int i=0; i<days.length; i++) if (days[i].equals(d2s)) d2 = i;
320                     
321                     return
322                         ((d1 <= d2 && day >= d1 && day <= d2) ||
323                          (d1 > d2 && (day >= d1 || day <= d2))) ?
324                         Boolean.TRUE : Boolean.FALSE;
325                 }
326             };
327         
328         private static final JSFunction dateRange = new JSFunction() {
329                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
330                     throw new JavaScriptException("XWT does not support dateRange() in PAC scripts");
331                 }
332             };
333         
334         private static final JSFunction timeRange = new JSFunction() {
335                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
336                     throw new JavaScriptException("XWT does not support timeRange() in PAC scripts");
337                 }
338             };
339         
340     }
341     
342 }