2002/10/03 23:37:07
[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 if (e instanceof JSObject)
129                     XWT.recursivePrintObject.call(Context.enter(), null, null, new Object[] { e });
130                 else Log.log(Platform.class, e);
131             }
132             return null;
133         }
134     }
135
136
137     // Authorization ///////////////////////////////////////////////////////////////////////////////////
138
139     public static class Authorization {
140
141         static public String authorization = null;
142         static public String authorization2 = null;
143         static public Semaphore waitingForUser = new Semaphore();
144
145         public static synchronized void getPassword(final String realm, final String style, final String proxyIP, String oldAuth) {
146
147             // this handles cases where multiple threads hit the proxy auth at the same time -- all but one will block on the
148             // synchronized keyword. If 'authorization' changed while the thread was blocked, it means that the user entered
149             // a password, so we should reattempt authorization.
150
151             if (authorization != oldAuth) return;
152             if (Log.on) Log.log(Authorization.class, "displaying proxy authorization dialog");
153             MessageQueue.add(new Message() {
154                     public void perform() {
155                         Box b = new Box("org.xwt.builtin.proxy_authorization", null);
156                         b.put("realm", realm);
157                         b.put("proxyIP", proxyIP);
158                     }
159                 });
160
161             waitingForUser.block();
162             if (Log.on) Log.log(Authorization.class, "got proxy authorization info; re-attempting connection");
163             
164         }
165     }
166
167
168     // ProxyAutoConfigRootScope ////////////////////////////////////////////////////////////////////
169     
170     public static class ProxyAutoConfigRootScope extends ScriptableObject {
171         
172         public String getClassName() { return "ProxyAutoConfigRootScope"; }
173         ProxyAutoConfigRootScope() { Context.enter().initStandardObjects(this); }
174         
175         public Object get(String name, Scriptable start) {
176             if (name.equals("isPlainHostName")) return isPlainHostName;
177             else if (name.equals("dnsDomainIs")) return dnsDomainIs;
178             else if (name.equals("localHostOrDomainIs")) return localHostOrDomainIs;
179             else if (name.equals("isResolvable")) return isResolvable;
180             else if (name.equals("isInNet")) return isInNet;
181             else if (name.equals("dnsResolve")) return dnsResolve;
182             else if (name.equals("myIpAddress")) return myIpAddress;
183             else if (name.equals("dnsDomainLevels")) return dnsDomainLevels;
184             else if (name.equals("shExpMatch")) return shExpMatch;
185             else if (name.equals("weekdayRange")) return weekdayRange;
186             else if (name.equals("dateRange")) return dateRange;
187             else if (name.equals("timeRange")) return timeRange;
188             else if (name.equals("ProxyConfig")) return ProxyConfig;
189             else return super.get(name, start);
190         }
191         
192         private static final JSObject proxyConfigBindings = new JSObject();
193         private static final JSObject ProxyConfig = new JSObject() {
194                 public Object get(String name, Scriptable start) {
195                     if (name.equals("bindings")) return proxyConfigBindings;
196                     return null;
197                 }
198             };
199         
200         private static abstract class JSFunction extends JSObject implements Function {
201             JSFunction() { setSeal(true); }
202             public Scriptable construct(Context cx, Scriptable scope, java.lang.Object[] args) { return null; }
203         }
204         
205         private static final JSFunction isPlainHostName = new JSFunction() {
206                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
207                     return (args[0].toString().indexOf('.') == -1) ? Boolean.TRUE : Boolean.FALSE;
208                 }
209             };
210         
211         private static final JSFunction dnsDomainIs = new JSFunction() {
212                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
213                     return (args[0].toString().endsWith(args[1].toString())) ? Boolean.TRUE : Boolean.FALSE;
214                 }
215             };
216         
217         private static final JSFunction localHostOrDomainIs = new JSFunction() {
218                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
219                     return (args[0].toString().equals(args[1].toString()) || 
220                             (args[0].toString().indexOf('.') == -1 && args[1].toString().startsWith(args[0].toString()))) ?
221                         Boolean.TRUE : Boolean.FALSE;
222                 }
223             };
224         
225         private static final JSFunction isResolvable = new JSFunction() {
226                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
227                     try {
228                         return (InetAddress.getByName(args[0].toString()) != null) ? Boolean.TRUE : Boolean.FALSE;
229                     } catch (UnknownHostException e) {
230                         return Boolean.FALSE;
231                     }
232                 }
233             };
234         
235         private static final JSFunction isInNet = new JSFunction() {
236                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
237                     if (args.length != 3) return Boolean.FALSE;
238                     try {
239                         byte[] host = InetAddress.getByName(args[0].toString()).getAddress();
240                         byte[] net = InetAddress.getByName(args[1].toString()).getAddress();
241                         byte[] mask = InetAddress.getByName(args[2].toString()).getAddress();
242                         return ((host[0] & mask[0]) == net[0] &&
243                                 (host[1] & mask[1]) == net[1] &&
244                                 (host[2] & mask[2]) == net[2] &&
245                                 (host[3] & mask[3]) == net[3]) ?
246                             Boolean.TRUE : Boolean.FALSE;
247                     } catch (Exception e) {
248                         throw new JavaScriptException("exception in isInNet(): " + e);
249                     }
250                 }
251             };
252         
253         private static final JSFunction dnsResolve = new JSFunction() {
254                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
255                     try {
256                         return InetAddress.getByName(args[0].toString()).getHostAddress();
257                     } catch (UnknownHostException e) {
258                         return null;
259                     }
260                 }
261             };
262         
263         private static final JSFunction myIpAddress = new JSFunction() {
264                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
265                     try {
266                         return InetAddress.getLocalHost().getHostAddress();
267                     } catch (UnknownHostException e) {
268                         if (Log.on) Log.log(this, "strange... host does not know its own address");
269                         return null;
270                     }
271                 }
272             };
273         
274         private static final JSFunction dnsDomainLevels = new JSFunction() {
275                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
276                     String s = args[0].toString();
277                     int i = 0;
278                     while((i = s.indexOf('.', i)) != -1) i++;
279                     return new Integer(i);
280                 }
281             };
282         
283         private static boolean match(String[] arr, String s, int index) {
284             if (index >= arr.length) return true;
285             for(int i=0; i<s.length(); i++) {
286                 String s2 = s.substring(i);
287                 if (s2.startsWith(arr[index]) && match(arr, s2.substring(arr[index].length()), index + 1)) return true;
288             }
289             return false;
290         }
291         
292         private static final JSFunction shExpMatch = new JSFunction() {
293                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
294                     StringTokenizer st = new StringTokenizer(args[1].toString(), "*", false);
295                     String[] arr = new String[st.countTokens()];
296                     String s = args[0].toString();
297                     for (int i=0; st.hasMoreTokens(); i++) arr[i] = st.nextToken();
298                     return match(arr, s, 0) ? Boolean.TRUE : Boolean.FALSE;
299                 }
300             };
301         
302         public static String[] days = { "SUN", "MON", "TUE", "WED", "THU", "FRI", "SAT" };
303         
304         private static final JSFunction weekdayRange = new JSFunction() {
305                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
306                     TimeZone tz = (args.length < 3 || args[2] == null || !args[2].equals("GMT")) ? TimeZone.getTimeZone("UTC") : TimeZone.getDefault();
307                     Calendar c = new GregorianCalendar();
308                     c.setTimeZone(tz);
309                     c.setTime(new Date());
310                     Date d = c.getTime();
311                     int day = d.getDay();
312                     
313                     String d1s = args[0].toString().toUpperCase();
314                     int d1 = 0, d2 = 0;
315                     for(int i=0; i<days.length; i++) if (days[i].equals(d1s)) d1 = i;
316                     
317                     if (args.length == 1)
318                         return d1 == day ? Boolean.TRUE : Boolean.FALSE;
319                     
320                     String d2s = args[1].toString().toUpperCase();
321                     for(int i=0; i<days.length; i++) if (days[i].equals(d2s)) d2 = i;
322                     
323                     return
324                         ((d1 <= d2 && day >= d1 && day <= d2) ||
325                          (d1 > d2 && (day >= d1 || day <= d2))) ?
326                         Boolean.TRUE : Boolean.FALSE;
327                 }
328             };
329         
330         private static final JSFunction dateRange = new JSFunction() {
331                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
332                     throw new JavaScriptException("XWT does not support dateRange() in PAC scripts");
333                 }
334             };
335         
336         private static final JSFunction timeRange = new JSFunction() {
337                 public Object call(Context cx, Scriptable thisObj, Scriptable ctorObj, Object[] args) throws JavaScriptException {
338                     throw new JavaScriptException("XWT does not support timeRange() in PAC scripts");
339                 }
340             };
341         
342     }
343     
344 }