initial checkin
[org.ibex.nanogoat.git] / src / org / ibex / HTTP.java
1 // Copyright 2004 Adam Megacz, see the COPYING file for licensing [GPL]
2 package org.ibex;
3
4 import java.net.*;
5 import java.io.*;
6 import java.util.*;
7 import org.ibex.js.*;
8 import org.ibex.util.*;
9 import org.bouncycastle.util.encoders.Base64;
10 import org.bouncycastle.crypto.digests.*;
11
12 /**
13  *  This object encapsulates a *single* HTTP connection. Multiple requests may be pipelined over a connection (thread-safe),
14  *  although any IOException encountered in a request will invalidate all later requests.
15  */
16 public class HTTP {
17
18
19     // Public Methods ////////////////////////////////////////////////////////////////////////////////////////
20
21     public HTTP(String url) { this(url, false); }
22     public HTTP(String url, boolean skipResolveCheck) { originalUrl = url; this.skipResolveCheck = skipResolveCheck; }
23
24     /** Performs an HTTP GET request */
25     public InputStream GET() throws IOException { return makeRequest(null, null); }
26
27     /** Performs an HTTP POST request; content is additional headers, blank line, and body */
28     public InputStream POST(String contentType, String content) throws IOException { return makeRequest(contentType, content); }
29
30     public static class HTTPException extends IOException { public HTTPException(String s) { super(s); } }
31
32     public static HTTP stdio = new HTTP("stdio:");
33
34
35     // Statics ///////////////////////////////////////////////////////////////////////////////////////////////
36
37     static Hash resolvedHosts = new Hash();            ///< cache for resolveAndCheckIfFirewalled()
38     private static Hash authCache = new Hash();        ///< cache of userInfo strings, keyed on originalUrl
39
40
41     // Instance Data ///////////////////////////////////////////////////////////////////////////////////////////////
42
43     final String originalUrl;              ///< the URL as passed to the original constructor; this is never changed
44     String url = null;                     ///< the URL to connect to; this is munged when the url is parsed */
45     String host = null;                    ///< the host to connect to
46     int port = -1;                         ///< the port to connect on
47     boolean ssl = false;                   ///< true if SSL (HTTPS) should be used
48     String path = null;                    ///< the path (URI) to retrieve on the server
49     Socket sock = null;                    ///< the socket
50     InputStream in = null;                 ///< the socket's inputstream
51     String userInfo = null;                ///< the username and password portions of the URL
52     boolean firstRequest = true;           ///< true iff this is the first request to be made on this socket
53     boolean skipResolveCheck = false;      ///< allowed to skip the resolve check when downloading PAC script
54     boolean proxied = false;               ///< true iff we're using a proxy
55
56     /** this is null if the current request is the first request on
57      *  this HTTP connection; otherwise it is a Semaphore which will be
58      *  released once the request ahead of us has recieved its response
59      */
60     Semaphore okToRecieve = null;
61
62     /**
63      *  This method isn't synchronized; however, only one thread can be in the inner synchronized block at a time, and the rest of
64      *  the method is protected by in-order one-at-a-time semaphore lock-steps
65      */
66     private InputStream makeRequest(String contentType, String content) throws IOException {
67
68         // Step 1: send the request and establish a semaphore to stop any requests that pipeline after us
69         Semaphore blockOn = null;
70         Semaphore releaseMe = null;
71         synchronized(this) {
72             try {
73                 connect();
74                 sendRequest(contentType, content);
75             } catch (IOException e) {
76                 reset();
77                 throw e;
78             }
79             blockOn = okToRecieve;
80             releaseMe = okToRecieve = new Semaphore();
81         }
82         
83         // Step 2: wait for requests ahead of us to complete, then read the reply off the stream
84         boolean doRelease = true;
85         try {
86             if (blockOn != null) blockOn.block();
87             
88             // previous call wrecked the socket connection, but we already sent our request, so we can't just retry --
89             // this could cause the server to receive the request twice, which could be bad (think of the case where the
90             // server call causes Amazon.com to ship you an item with one-click purchasing).
91             if (in == null)
92                 throw new HTTPException("a previous pipelined call messed up the socket");
93             
94             Hashtable h = in == null ? null : parseHeaders(in);
95             if (h == null) {
96                 if (firstRequest) throw new HTTPException("server closed the socket with no response");
97                 // sometimes the server chooses to close the stream between requests
98                 reset();
99                 releaseMe.release();
100                 return makeRequest(contentType, content);
101             }
102
103             String reply = h.get("STATUSLINE").toString();
104             
105             if (reply.startsWith("407") || reply.startsWith("401")) {
106                 
107                 if (reply.startsWith("407")) doProxyAuth(h, content == null ? "GET" : "POST");
108                 else doWebAuth(h, content == null ? "GET" : "POST");
109                 
110                 if (h.get("HTTP").equals("1.0") && h.get("content-length") == null) {
111                     if (Log.on) Log.info(this, "proxy returned an HTTP/1.0 reply with no content-length...");
112                     reset();
113                 } else {
114                     int cl = h.get("content-length") == null ? -1 : Integer.parseInt(h.get("content-length").toString());
115                     new HTTPInputStream(in, cl, releaseMe).close();
116                 }
117                 releaseMe.release();
118                 return makeRequest(contentType, content);
119                 
120             } else if (reply.startsWith("2")) {
121                 if (h.get("HTTP").equals("1.0") && h.get("content-length") == null)
122                     throw new HTTPException("Ibex does not support HTTP/1.0 servers which fail to return the Content-Length header");
123                 int cl = h.get("content-length") == null ? -1 : Integer.parseInt(h.get("content-length").toString());
124                 InputStream ret = new HTTPInputStream(in, cl, releaseMe);
125                 if ("gzip".equals(h.get("content-encoding"))) ret = new java.util.zip.GZIPInputStream(ret);
126                 doRelease = false;
127                 return ret;
128                 
129             } else {
130                 throw new HTTPException("HTTP Error: " + reply);
131                 
132             }
133             
134         } catch (IOException e) { reset(); throw e;
135         } finally { if (doRelease) releaseMe.release();
136         }
137     }
138
139
140     // Safeguarded DNS Resolver ///////////////////////////////////////////////////////////////////////////
141
142     /**
143      *  resolves the hostname and returns it as a string in the form "x.y.z.w"
144      *  @throws HTTPException if the host falls within a firewalled netblock
145      */
146     private void resolveAndCheckIfFirewalled(String host) throws HTTPException {
147
148         // cached
149         if (resolvedHosts.get(host) != null) return;
150
151         // if all scripts are trustworthy (local FS), continue
152         if (Main.originAddr == null) return;
153
154         // resolve using DNS
155         try {
156             InetAddress addr = InetAddress.getByName(host);
157             byte[] quadbyte = addr.getAddress();
158             if ((quadbyte[0] == 10 ||
159                  (quadbyte[0] == 192 && quadbyte[1] == 168) ||
160                  (quadbyte[0] == 172 && (quadbyte[1] & 0xF0) == 16)) && !addr.equals(Main.originAddr))
161                 throw new HTTPException("security violation: " + host + " [" + addr.getHostAddress() +
162                                         "] is in a firewalled netblock");
163             return;
164         } catch (UnknownHostException uhe) { }
165
166         if (Platform.detectProxy() == null)
167             throw new HTTPException("could not resolve hostname \"" + host + "\" and no proxy configured");
168     }
169
170
171     // Methods to attempt socket creation /////////////////////////////////////////////////////////////////
172
173     private Socket getSocket(String host, int port, boolean ssl, boolean negotiate) throws IOException {
174         Socket ret = ssl ? new SSL(host, port, negotiate) : new Socket(java.net.InetAddress.getByName(host), port);
175         ret.setTcpNoDelay(true);
176         return ret;
177     }
178
179     /** Attempts a direct connection */
180     private Socket attemptDirect() {
181         try {
182             if (Log.verbose) Log.info(this, "attempting to create unproxied socket to " +
183                                      host + ":" + port + (ssl ? " [ssl]" : ""));
184             return getSocket(host, port, ssl, true);
185         } catch (IOException e) {
186             if (Log.on) Log.info(this, "exception in attemptDirect(): " + e);
187             return null;
188         }
189     }
190
191     /** Attempts to use an HTTP proxy, employing the CONNECT method if HTTPS is requested */
192     private Socket attemptHttpProxy(String proxyHost, int proxyPort) {
193         try {
194             if (Log.verbose) Log.info(this, "attempting to create HTTP proxied socket using proxy " + proxyHost + ":" + proxyPort);
195             Socket sock = getSocket(proxyHost, proxyPort, ssl, false);
196
197             if (!ssl) {
198                 if (!path.startsWith("http://")) path = "http://" + host + ":" + port + path;
199                 return sock;
200             }
201
202             PrintWriter pw = new PrintWriter(new OutputStreamWriter(sock.getOutputStream()));
203             BufferedReader br = new BufferedReader(new InputStreamReader(sock.getInputStream()));
204             pw.print("CONNECT " + host + ":" + port + " HTTP/1.1\r\n\r\n");
205             pw.flush();
206             String s = br.readLine();
207             if (s.charAt(9) != '2') throw new HTTPException("proxy refused CONNECT method: \"" + s + "\"");
208             while (br.readLine().length() > 0) { };
209             ((SSL)sock).negotiate();
210             return sock;
211
212         } catch (IOException e) {
213             if (Log.on) Log.info(this, "exception in attemptHttpProxy(): " + e);
214             return null;
215         }
216     }
217
218     /**
219      *  Implements SOCKSv4 with v4a DNS extension
220      *  @see http://www.socks.nec.com/protocol/socks4.protocol
221      *  @see http://www.socks.nec.com/protocol/socks4a.protocol
222      */
223     private Socket attemptSocksProxy(String proxyHost, int proxyPort) {
224
225         // even if host is already a "x.y.z.w" string, we use this to parse it into bytes
226         InetAddress addr = null;
227         try { addr = InetAddress.getByName(host); } catch (Exception e) { }
228
229         if (Log.verbose) Log.info(this, "attempting to create SOCKSv4" + (addr == null ? "" : "a") +
230                                  " proxied socket using proxy " + proxyHost + ":" + proxyPort);
231
232         try {
233             Socket sock = getSocket(proxyHost, proxyPort, ssl, false);
234             
235             DataOutputStream dos = new DataOutputStream(sock.getOutputStream());
236             dos.writeByte(0x04);                         // SOCKSv4(a)
237             dos.writeByte(0x01);                         // CONNECT
238             dos.writeShort(port & 0xffff);               // port
239             if (addr == null) dos.writeInt(0x00000001);  // bogus IP
240             else dos.write(addr.getAddress());           // actual IP
241             dos.writeByte(0x00);                         // no userid
242             if (addr == null) {
243                 PrintWriter pw = new PrintWriter(new OutputStreamWriter(dos));
244                 pw.print(host);
245                 pw.flush();
246                 dos.writeByte(0x00);                     // hostname null terminator
247             }
248             dos.flush();
249
250             DataInputStream dis = new DataInputStream(sock.getInputStream());
251             dis.readByte();                              // reply version
252             byte success = dis.readByte();               // success/fail
253             dis.skip(6);                                 // ip/port
254             
255             if ((int)(success & 0xff) == 90) {
256                 if (ssl) ((SSL)sock).negotiate();
257                 return sock;
258             }
259             if (Log.on) Log.info(this, "SOCKS server denied access, code " + (success & 0xff));
260             return null;
261
262         } catch (IOException e) {
263             if (Log.on) Log.info(this, "exception in attemptSocksProxy(): " + e);
264             return null;
265         }
266     }
267
268     /** executes the PAC script and dispatches a call to one of the other attempt methods based on the result */
269     private Socket attemptPAC(org.ibex.js.JS pacFunc) {
270         if (Log.verbose) Log.info(this, "evaluating PAC script");
271         String pac = null;
272         try {
273             Object obj = pacFunc.call(url, host, null, null, 2);
274             if (Log.verbose) Log.info(this, "  PAC script returned \"" + obj + "\"");
275             pac = obj.toString();
276         } catch (Throwable e) {
277             if (Log.on) Log.info(this, "PAC script threw exception " + e);
278             return null;
279         }
280
281         StringTokenizer st = new StringTokenizer(pac, ";", false);
282         while (st.hasMoreTokens()) {
283             String token = st.nextToken().trim();
284             if (Log.verbose) Log.info(this, "  trying \"" + token + "\"...");
285             try {
286                 Socket ret = null;
287                 if (token.startsWith("DIRECT"))
288                     ret = attemptDirect();
289                 else if (token.startsWith("PROXY"))
290                     ret = attemptHttpProxy(token.substring(token.indexOf(' ') + 1, token.indexOf(':')),
291                                            Integer.parseInt(token.substring(token.indexOf(':') + 1)));
292                 else if (token.startsWith("SOCKS"))
293                     ret = attemptSocksProxy(token.substring(token.indexOf(' ') + 1, token.indexOf(':')),
294                                             Integer.parseInt(token.substring(token.indexOf(':') + 1)));
295                 if (ret != null) return ret;
296             } catch (Throwable e) {
297                 if (Log.on) Log.info(this, "attempt at \"" + token + "\" failed due to " + e + "; trying next token");
298             }
299         }
300         if (Log.on) Log.info(this, "all PAC results exhausted");
301         return null;
302     }
303
304
305     // Everything Else ////////////////////////////////////////////////////////////////////////////
306
307     private synchronized void connect() throws IOException {
308         if (originalUrl.equals("stdio:")) { in = new BufferedInputStream(System.in); return; }
309         if (sock != null) {
310             if (in == null) in = new BufferedInputStream(sock.getInputStream());
311             return;
312         }
313         // grab the userinfo; gcj doesn't have java.net.URL.getUserInfo()
314         String url = originalUrl;
315         userInfo = url.substring(url.indexOf("://") + 3);
316         userInfo = userInfo.indexOf('/') == -1 ? userInfo : userInfo.substring(0, userInfo.indexOf('/'));
317         if (userInfo.indexOf('@') != -1) {
318             userInfo = userInfo.substring(0, userInfo.indexOf('@'));
319             url = url.substring(0, url.indexOf("://") + 3) + url.substring(url.indexOf('@') + 1);
320         } else {
321             userInfo = null;
322         }
323
324         if (url.startsWith("https:")) {
325             ssl = true;
326         } else if (!url.startsWith("http:")) {
327             throw new IOException("HTTP only supports http/https urls");
328         }
329         if (url.indexOf("://") == -1) throw new IOException("URLs must contain a ://");
330         String temphost = url.substring(url.indexOf("://") + 1);
331         path = temphost.substring(temphost.indexOf('/'));
332         temphost = temphost.substring(0, temphost.indexOf('/'));
333         if (temphost.indexOf(':') != -1) {
334             port = Integer.parseInt(temphost.substring(temphost.indexOf(':')+1));
335             temphost = temphost.substring(0, temphost.indexOf(':'));
336         } else {
337             port = ssl ? 443 : 80;
338         }
339         if (!skipResolveCheck) resolveAndCheckIfFirewalled(temphost);
340         host = temphost;
341         if (Log.verbose) Log.info(this, "creating HTTP object for connection to " + host + ":" + port);
342
343         Proxy pi = Platform.detectProxy();
344         OUTER: do {
345             if (pi != null) {
346                 for(int i=0; i<pi.excluded.length; i++) if (host.equals(pi.excluded[i])) break OUTER;
347                 if (sock == null && pi.proxyAutoConfigFunction != null) sock = attemptPAC(pi.proxyAutoConfigFunction);
348                 if (sock == null && ssl && pi.httpsProxyHost != null) sock = attemptHttpProxy(pi.httpsProxyHost,pi.httpsProxyPort);
349                 if (sock == null && pi.httpProxyHost != null) sock = attemptHttpProxy(pi.httpProxyHost, pi.httpProxyPort);
350                 if (sock == null && pi.socksProxyHost != null) sock = attemptSocksProxy(pi.socksProxyHost, pi.socksProxyPort);
351             }
352         } while (false);
353         proxied = sock != null;
354         if (sock == null) sock = attemptDirect();
355         if (sock == null) throw new HTTPException("unable to contact host " + host);
356         if (in == null) in = new BufferedInputStream(sock.getInputStream());
357     }
358
359     private void sendRequest(String contentType, String content) throws IOException {
360         PrintWriter pw = new PrintWriter(new OutputStreamWriter(originalUrl.equals("stdio:") ?
361                                                                 System.out : sock.getOutputStream()));
362         if (content != null) {
363             pw.print("POST " + path + " HTTP/1.0\r\n"); // FIXME chunked encoding
364             int contentLength = content.substring(0, 2).equals("\r\n") ?
365                 content.length() - 2 :
366                 (content.length() - content.indexOf("\r\n\r\n") - 4);
367             pw.print("Content-Length: " + contentLength + "\r\n");
368             if (contentType != null) pw.print("Content-Type: " + contentType + "\r\n");
369         } else {
370             pw.print("GET " + path + " HTTP/1.1\r\n");
371         }
372         
373         pw.print("User-Agent: Ibex\r\n");
374         pw.print("Accept-encoding: gzip\r\n");
375         pw.print("Host: " + (host + (port == 80 ? "" : (":" + port))) + "\r\n");
376         if (proxied) pw.print("X-RequestOrigin: " + Main.originHost + "\r\n");
377
378         if (Proxy.Authorization.authorization != null) pw.print("Proxy-Authorization: "+Proxy.Authorization.authorization2+"\r\n");
379         if (authCache.get(originalUrl) != null) pw.print("Authorization: " + authCache.get(originalUrl) + "\r\n");
380
381         pw.print(content == null ? "\r\n" : content);
382         pw.print("\r\n");
383         pw.flush();
384     }
385
386     private void doWebAuth(Hashtable h0, String method) throws IOException {
387         if (userInfo == null) throw new HTTPException("web server demanded username/password, but none were supplied");
388         Hashtable h = parseAuthenticationChallenge(h0.get("www-authenticate").toString());
389         
390         if (h.get("AUTHTYPE").equals("Basic")) {
391             if (authCache.get(originalUrl) != null) throw new HTTPException("username/password rejected");
392             authCache.put(originalUrl, "Basic " + new String(Base64.encode(userInfo.getBytes("UTF8"))));
393             
394         } else if (h.get("AUTHTYPE").equals("Digest")) {
395             if (authCache.get(originalUrl) != null && !"true".equals(h.get("stale")))
396                 throw new HTTPException("username/password rejected");
397             String path2 = path;
398             if (path2.startsWith("http://") || path2.startsWith("https://")) {
399                 path2 = path2.substring(path2.indexOf("://") + 3);
400                 path2 = path2.substring(path2.indexOf('/'));
401             }
402             String A1 = userInfo.substring(0, userInfo.indexOf(':')) + ":" + h.get("realm") + ":" +
403                 userInfo.substring(userInfo.indexOf(':') + 1);
404             String A2 = method + ":" + path2;
405             authCache.put(originalUrl,
406                           "Digest " +
407                           "username=\"" + userInfo.substring(0, userInfo.indexOf(':')) + "\", " +
408                           "realm=\"" + h.get("realm") + "\", " +
409                           "nonce=\"" + h.get("nonce") + "\", " +
410                           "uri=\"" + path2 + "\", " +
411                           (h.get("opaque") == null ? "" : ("opaque=\"" + h.get("opaque") + "\", ")) + 
412                           "response=\"" + H(H(A1) + ":" + h.get("nonce") + ":" + H(A2)) + "\", " +
413                           "algorithm=MD5"
414                           );
415             
416         } else {
417             throw new HTTPException("unknown authentication type: " + h.get("AUTHTYPE"));
418         }
419     }
420
421     private void doProxyAuth(Hashtable h0, String method) throws IOException {
422         if (Log.on) Log.info(this, "Proxy AuthChallenge: " + h0.get("proxy-authenticate"));
423         Hashtable h = parseAuthenticationChallenge(h0.get("proxy-authenticate").toString());
424         String style = h.get("AUTHTYPE").toString();
425         String realm = (String)h.get("realm");
426
427         if (style.equals("NTLM") && Proxy.Authorization.authorization2 == null) {
428             Log.info(this, "Proxy identified itself as NTLM, sending Type 1 packet");
429             Proxy.Authorization.authorization2 = "NTLM " + Base64.encode(Proxy.NTLM.type1);
430             return;
431         }
432
433         if (!realm.equals("Digest") || Proxy.Authorization.authorization2 == null || !"true".equals(h.get("stale")))
434             Proxy.Authorization.getPassword(realm, style, sock.getInetAddress().getHostAddress(),
435                                             Proxy.Authorization.authorization);
436
437         if (style.equals("Basic")) {
438             Proxy.Authorization.authorization2 =
439                 "Basic " + new String(Base64.encode(Proxy.Authorization.authorization.getBytes("UTF8")));
440             
441         } else if (style.equals("Digest")) {
442             String A1 = Proxy.Authorization.authorization.substring(0, userInfo.indexOf(':')) + ":" + h.get("realm") + ":" +
443                 Proxy.Authorization.authorization.substring(Proxy.Authorization.authorization.indexOf(':') + 1);
444             String A2 = method + ":" + path;
445             Proxy.Authorization.authorization2 = 
446                 "Digest " +
447                 "username=\"" + Proxy.Authorization.authorization.substring(0, Proxy.Authorization.authorization.indexOf(':')) +
448                 "\", " +
449                 "realm=\"" + h.get("realm") + "\", " +
450                 "nonce=\"" + h.get("nonce") + "\", " +
451                 "uri=\"" + path + "\", " +
452                 (h.get("opaque") == null ? "" : ("opaque=\"" + h.get("opaque") + "\", ")) + 
453                 "response=\"" + H(H(A1) + ":" + h.get("nonce") + ":" + H(A2)) + "\", " +
454                 "algorithm=MD5";
455
456         } else if (style.equals("NTLM")) {
457             Log.info(this, "Proxy identified itself as NTLM, got Type 2 packet");
458             byte[] type2 = Base64.decode(((String)h0.get("proxy-authenticate")).substring(5).trim());
459             for(int i=0; i<type2.length; i += 4) {
460                 String log = "";
461                 if (i<type2.length) log += Integer.toString(type2[i] & 0xff, 16) + " ";
462                 if (i+1<type2.length) log += Integer.toString(type2[i+1] & 0xff, 16) + " ";
463                 if (i+2<type2.length) log += Integer.toString(type2[i+2] & 0xff, 16) + " ";
464                 if (i+3<type2.length) log += Integer.toString(type2[i+3] & 0xff, 16) + " ";
465                 Log.info(this, log);
466             }
467             // FEATURE: need to keep the connection open between type1 and type3
468             // FEATURE: finish this
469             //byte[] type3 = Proxy.NTLM.getResponse(
470             //Proxy.Authorization.authorization2 = "NTLM " + Base64.encode(type3));
471         }            
472     }
473
474
475     // HTTPInputStream ///////////////////////////////////////////////////////////////////////////////////
476
477     /** An input stream that represents a subset of a longer input stream. Supports HTTP chunking as well */
478     public class HTTPInputStream extends FilterInputStream implements KnownLength {
479
480         private int length = 0;              ///< if chunking, numbytes left in this subset; else the remainder of the chunk
481         private Semaphore releaseMe = null;  ///< this semaphore will be released when the stream is closed
482         boolean chunkedDone = false;         ///< indicates that we have encountered the zero-length terminator chunk
483         boolean firstChunk = true;           ///< if we're on the first chunk, we don't pre-read a CRLF
484         private int contentLength = 0;       ///< the length of the entire content body; -1 if chunked
485
486         HTTPInputStream(InputStream in, int length, Semaphore releaseMe) throws IOException {
487             super(in);
488             this.releaseMe = releaseMe;
489             this.contentLength = length;
490             this.length = length == -1 ? 0 : length;
491         }
492
493         public int getLength() { return contentLength; }
494         public boolean markSupported() { return false; }
495         public int read(byte[] b) throws IOException { return read(b, 0, b.length); }
496         public long skip(long n) throws IOException { return read(null, -1, (int)n); }
497         public int available() throws IOException {
498             if (contentLength == -1) return java.lang.Math.min(super.available(), length);
499             return super.available();
500         }
501
502         public int read() throws IOException {
503             byte[] b = new byte[1];
504             int ret = read(b, 0, 1);
505             return ret == -1 ? -1 : b[0] & 0xff;
506         }
507
508         private void readChunk() throws IOException {
509             if (chunkedDone) return;
510             if (!firstChunk) super.skip(2); // CRLF
511             firstChunk = false;
512             String chunkLen = "";
513             while(true) {
514                 int i = super.read();
515                 if (i == -1) throw new HTTPException("encountered end of stream while reading chunk length");
516
517                 // FEATURE: handle chunking extensions
518                 if (i == '\r') {
519                     super.read();    // LF
520                     break;
521                 } else {
522                     chunkLen += (char)i;
523                 }
524             }
525             length = Integer.parseInt(chunkLen.trim(), 16);
526             if (length == 0) chunkedDone = true;
527         }
528
529         public int read(byte[] b, int off, int len) throws IOException {
530             boolean good = false;
531             try {
532                 if (length == 0 && contentLength == -1) {
533                     readChunk();
534                     if (chunkedDone) { good = true; return -1; }
535                 } else {
536                     if (length == 0) { good = true; return -1; }
537                 }
538                 if (len > length) len = length;
539                 int ret = b == null ? (int)super.skip(len) : super.read(b, off, len);
540                 if (ret >= 0) {
541                     length -= ret;
542                     good = true;
543                 }
544                 return ret;
545             } finally {
546                 if (!good) reset();
547             }
548         }
549
550         public void close() throws IOException {
551             if (contentLength == -1) {
552                 while(!chunkedDone) {
553                     if (length != 0) skip(length);
554                     readChunk();
555                 }
556                 skip(2);
557             } else {
558                 if (length != 0) skip(length);
559             }
560             if (releaseMe != null) releaseMe.release();
561         }
562     }
563
564     void reset() {
565         firstRequest = true;
566         in = null;
567         sock = null;
568     }
569
570
571     // Misc Helpers ///////////////////////////////////////////////////////////////////////////////////
572
573     /** reads a set of HTTP headers off of the input stream, returning null if the stream is already at its end */
574     private Hashtable parseHeaders(InputStream in) throws IOException {
575         Hashtable ret = new Hashtable();
576
577         // we can't use a BufferedReader directly on the input stream, since it will buffer past the end of the headers
578         byte[] buf = new byte[4096];
579         int buflen = 0;
580         while(true) {
581             int read = in.read();
582             if (read == -1 && buflen == 0) return null;
583             if (read == -1) throw new HTTPException("stream closed while reading headers");
584             buf[buflen++] = (byte)read;
585             if (buflen >= 4 && buf[buflen - 4] == '\r' && buf[buflen - 3] == '\n' &&
586                 buf[buflen - 2] == '\r' && buf[buflen - 1] == '\n')
587                 break;
588             if (buflen >=2 && buf[buflen - 1] == '\n' && buf[buflen - 2] == '\n')
589                 break;  // nice for people using stdio
590             if (buflen == buf.length) {
591                 byte[] newbuf = new byte[buf.length * 2];
592                 System.arraycopy(buf, 0, newbuf, 0, buflen);
593                 buf = newbuf;
594             }
595         }
596
597         BufferedReader br = new BufferedReader(new InputStreamReader(new ByteArrayInputStream(buf, 0, buflen)));
598         String s = br.readLine();
599         if (!s.startsWith("HTTP/")) throw new HTTPException("Expected reply to start with \"HTTP/\", got: " + s);
600         ret.put("STATUSLINE", s.substring(s.indexOf(' ') + 1));
601         ret.put("HTTP", s.substring(5, s.indexOf(' ')));
602
603         while((s = br.readLine()) != null && s.length() > 0) {
604             String front = s.substring(0, s.indexOf(':')).toLowerCase();
605             String back = s.substring(s.indexOf(':') + 1).trim();
606             // ugly hack: we never replace a Digest-auth with a Basic-auth (proxy + www)
607             if (front.endsWith("-authenticate") && ret.get(front) != null && !back.equals("Digest")) continue;
608             ret.put(front, back);
609         }
610         return ret;
611     }
612
613     private Hashtable parseAuthenticationChallenge(String s) {
614         Hashtable ret = new Hashtable();
615
616         s = s.trim();
617         ret.put("AUTHTYPE", s.substring(0, s.indexOf(' ')));
618         s = s.substring(s.indexOf(' ')).trim();
619
620         while (s.length() > 0) {
621             String val = null;
622             String key = s.substring(0, s.indexOf('='));
623             s = s.substring(s.indexOf('=') + 1);
624             if (s.charAt(0) == '\"') {
625                 s = s.substring(1);
626                 val = s.substring(0, s.indexOf('\"'));
627                 s = s.substring(s.indexOf('\"') + 1);
628             } else {
629                 val = s.indexOf(',') == -1 ? s : s.substring(0, s.indexOf(','));
630                 s = s.indexOf(',') == -1 ? "" : s.substring(s.indexOf(',') + 1);
631             }
632             if (s.length() > 0 && s.charAt(0) == ',') s = s.substring(1);
633             s = s.trim();
634             ret.put(key, val);
635         }
636         return ret;
637     }
638
639     private String H(String s) throws IOException {
640         byte[] b = s.getBytes("UTF8");
641         MD5Digest md5 = new MD5Digest();
642         md5.update(b, 0, b.length);
643         byte[] out = new byte[md5.getDigestSize()];
644         md5.doFinal(out, 0);
645         String ret = "";
646         for(int i=0; i<out.length; i++) {
647             ret += "0123456789abcdef".charAt((out[i] & 0xf0) >> 4);
648             ret += "0123456789abcdef".charAt(out[i] & 0x0f);
649         }
650         return ret;
651     }
652
653
654     // Proxy ///////////////////////////////////////////////////////////
655
656     /** encapsulates most of the proxy logic; some is shared in HTTP.java */
657     public static class Proxy {
658         
659         public String httpProxyHost = null;                  ///< the HTTP Proxy host to use
660         public int httpProxyPort = -1;                       ///< the HTTP Proxy port to use
661         public String httpsProxyHost = null;                 ///< seperate proxy for HTTPS
662         public int httpsProxyPort = -1;
663         public String socksProxyHost = null;                 ///< the SOCKS Proxy Host to use
664         public int socksProxyPort = -1;                      ///< the SOCKS Proxy Port to use
665         public String[] excluded = new String[] { };         ///< hosts to be excluded from proxy use; wildcards permitted
666         public JS proxyAutoConfigFunction = null;  ///< the PAC script
667     
668         public static Proxy detectProxyViaManual() {
669             Proxy ret = new Proxy();
670         
671             ret.httpProxyHost = Platform.getEnv("http_proxy");
672             if (ret.httpProxyHost != null) {
673                 if (ret.httpProxyHost.startsWith("http://")) ret.httpProxyHost = ret.httpProxyHost.substring(7);
674                 if (ret.httpProxyHost.endsWith("/"))
675                     ret.httpProxyHost = ret.httpProxyHost.substring(0, ret.httpProxyHost.length() - 1);
676                 if (ret.httpProxyHost.indexOf(':') != -1) {
677                     ret.httpProxyPort = Integer.parseInt(ret.httpProxyHost.substring(ret.httpProxyHost.indexOf(':') + 1));
678                     ret.httpProxyHost = ret.httpProxyHost.substring(0, ret.httpProxyHost.indexOf(':'));
679                 } else {
680                     ret.httpProxyPort = 80;
681                 }
682             }
683         
684             ret.httpsProxyHost = Platform.getEnv("https_proxy");
685             if (ret.httpsProxyHost != null) {
686                 if (ret.httpsProxyHost.startsWith("https://")) ret.httpsProxyHost = ret.httpsProxyHost.substring(7);
687                 if (ret.httpsProxyHost.endsWith("/"))
688                     ret.httpsProxyHost = ret.httpsProxyHost.substring(0, ret.httpsProxyHost.length() - 1);
689                 if (ret.httpsProxyHost.indexOf(':') != -1) {
690                     ret.httpsProxyPort = Integer.parseInt(ret.httpsProxyHost.substring(ret.httpsProxyHost.indexOf(':') + 1));
691                     ret.httpsProxyHost = ret.httpsProxyHost.substring(0, ret.httpsProxyHost.indexOf(':'));
692                 } else {
693                     ret.httpsProxyPort = 80;
694                 }
695             }
696         
697             ret.socksProxyHost = Platform.getEnv("socks_proxy");
698             if (ret.socksProxyHost != null) {
699                 if (ret.socksProxyHost.startsWith("socks://")) ret.socksProxyHost = ret.socksProxyHost.substring(7);
700                 if (ret.socksProxyHost.endsWith("/"))
701                     ret.socksProxyHost = ret.socksProxyHost.substring(0, ret.socksProxyHost.length() - 1);
702                 if (ret.socksProxyHost.indexOf(':') != -1) {
703                     ret.socksProxyPort = Integer.parseInt(ret.socksProxyHost.substring(ret.socksProxyHost.indexOf(':') + 1));
704                     ret.socksProxyHost = ret.socksProxyHost.substring(0, ret.socksProxyHost.indexOf(':'));
705                 } else {
706                     ret.socksProxyPort = 80;
707                 }
708             }
709         
710             String noproxy = Platform.getEnv("no_proxy");
711             if (noproxy != null) {
712                 StringTokenizer st = new StringTokenizer(noproxy, ",");
713                 ret.excluded = new String[st.countTokens()];
714                 for(int i=0; st.hasMoreTokens(); i++) ret.excluded[i] = st.nextToken();
715             }
716         
717             if (ret.httpProxyHost == null && ret.socksProxyHost == null) return null;
718             return ret;
719         }
720     
721         public static JSScope proxyAutoConfigRootScope = new ProxyAutoConfigRootScope();
722         public static JS getProxyAutoConfigFunction(String url) {
723             try { 
724                 BufferedReader br = new BufferedReader(new InputStreamReader(new HTTP(url, true).GET()));
725                 String s = null;
726                 String script = "";
727                 while((s = br.readLine()) != null) script += s + "\n";
728                 if (Log.on) Log.info(Proxy.class, "successfully retrieved WPAD PAC:");
729                 if (Log.on) Log.info(Proxy.class, script);
730             
731                 // MS CARP hack
732                 Vector carpHosts = new Vector();
733                 for(int i=0; i<script.length(); i++)
734                     if (script.regionMatches(i, "new Node(", 0, 9)) {
735                         String host = script.substring(i + 10, script.indexOf('\"', i + 11));
736                         if (Log.on) Log.info(Proxy.class, "Detected MS Proxy Server CARP Script, Host=" + host);
737                         carpHosts.addElement(host);
738                     }
739                 if (carpHosts.size() > 0) {
740                     script = "function FindProxyForURL(url, host) {\nreturn \"";
741                     for(int i=0; i<carpHosts.size(); i++)
742                         script += "PROXY " + carpHosts.elementAt(i) + "; ";
743                     script += "\";\n}";
744                     if (Log.on) Log.info(Proxy.class, "DeCARPed PAC script:");
745                     if (Log.on) Log.info(Proxy.class, script);
746                 }
747
748                 JS scr = JS.fromReader("PAC script at " + url, 0, new StringReader(script));
749                 JS.cloneWithNewParentScope(scr, proxyAutoConfigRootScope).call(null, null, null, null, 0);
750                 return (JS)proxyAutoConfigRootScope.get("FindProxyForURL");
751             } catch (Exception e) {
752                 if (Log.on) {
753                     Log.info(Platform.class, "WPAD detection failed due to:");
754                     if (e instanceof JSExn) {
755                         try {
756                             org.ibex.js.JSArray arr = new org.ibex.js.JSArray();
757                             arr.addElement(((JSExn)e).getObject());
758                         } catch (Exception e2) {
759                             Log.info(Platform.class, e);
760                         }
761                     }
762                     else Log.info(Platform.class, e);
763                 }
764                 return null;
765             }
766         }
767
768
769         // Authorization ///////////////////////////////////////////////////////////////////////////////////
770
771         public static class Authorization {
772
773             static public String authorization = null;
774             static public String authorization2 = null;
775             static public Semaphore waitingForUser = new Semaphore();
776
777             public static synchronized void getPassword(final String realm, final String style,
778                                                         final String proxyIP, String oldAuth) throws IOException {
779
780                 // this handles cases where multiple threads hit the proxy auth at the same time -- all but one will block on the
781                 // synchronized keyword. If 'authorization' changed while the thread was blocked, it means that the user entered
782                 // a password, so we should reattempt authorization.
783
784                 if (authorization != oldAuth) return;
785                 if (Log.on) Log.info(Authorization.class, "displaying proxy authorization dialog");
786                 Scheduler.add(new Scheduler.Task() {
787                         public void perform() throws IOException, JSExn {
788                             Box b = new Box();
789                             Template t = null;
790                             // FIXME
791                             //Template.buildTemplate("org/ibex/builtin/proxy_authorization.ibex", Stream.getInputStream((JS)Main.builtin.get("org/ibex/builtin/proxy_authorization.ibex")), new Ibex(null));
792                             t.apply(b);
793                             b.put("realm", realm);
794                             b.put("proxyIP", proxyIP);
795                         }
796                     });
797
798                 waitingForUser.block();
799                 if (Log.on) Log.info(Authorization.class, "got proxy authorization info; re-attempting connection");
800             }
801         }
802
803
804         // ProxyAutoConfigRootJSScope ////////////////////////////////////////////////////////////////////
805
806         public static class ProxyAutoConfigRootScope extends JSScope.Global {
807
808             public ProxyAutoConfigRootScope() { super(); }
809         
810             public Object get(Object name) throws JSExn {
811                 //#switch(name)
812                 case "isPlainHostName": return METHOD;
813                 case "dnsDomainIs": return METHOD;
814                 case "localHostOrDomainIs": return METHOD;
815                 case "isResolvable": return METHOD;
816                 case "isInNet": return METHOD;
817                 case "dnsResolve": return METHOD;
818                 case "myIpAddress": return METHOD;
819                 case "dnsDomainLevels": return METHOD;
820                 case "shExpMatch": return METHOD;
821                 case "weekdayRange": return METHOD;
822                 case "dateRange": return METHOD;
823                 case "timeRange": return METHOD;
824                 case "ProxyConfig": return ProxyConfig;
825                 //#end
826                 return super.get(name);
827             }
828         
829             private static final JS proxyConfigBindings = new JS();
830             private static final JS ProxyConfig = new JS() {
831                     public Object get(Object name) {
832                         if (name.equals("bindings")) return proxyConfigBindings;
833                         return null;
834                     }
835                 };
836
837             public Object callMethod(Object method, Object a0, Object a1, Object a2, Object[] rest, int nargs) throws JSExn {
838                 //#switch(method)
839                 case "isPlainHostName": return (a0.toString().indexOf('.') == -1) ? Boolean.TRUE : Boolean.FALSE;
840                 case "dnsDomainIs": return (a0.toString().endsWith(a1.toString())) ? Boolean.TRUE : Boolean.FALSE;
841                 case "localHostOrDomainIs":
842                     return (a0.equals(a1) || (a0.toString().indexOf('.') == -1 && a1.toString().startsWith(a0.toString()))) ? T:F;
843                 case "isResolvable": try {
844                     return (InetAddress.getByName(a0.toString()) != null) ? Boolean.TRUE : Boolean.FALSE;
845                 } catch (UnknownHostException e) { return F; }
846                 case "isInNet":
847                     if (nargs != 3) return Boolean.FALSE;
848                     try {
849                         byte[] host = InetAddress.getByName(a0.toString()).getAddress();
850                         byte[] net = InetAddress.getByName(a1.toString()).getAddress();
851                         byte[] mask = InetAddress.getByName(a2.toString()).getAddress();
852                         return ((host[0] & mask[0]) == net[0] &&
853                                 (host[1] & mask[1]) == net[1] &&
854                                 (host[2] & mask[2]) == net[2] &&
855                                 (host[3] & mask[3]) == net[3]) ?
856                             Boolean.TRUE : Boolean.FALSE;
857                     } catch (Exception e) {
858                         throw new JSExn("exception in isInNet(): " + e);
859                     }
860                 case "dnsResolve":
861                     try {
862                         return InetAddress.getByName(a0.toString()).getHostAddress();
863                     } catch (UnknownHostException e) {
864                         return null;
865                     }
866                 case "myIpAddress":
867                     try {
868                         return InetAddress.getLocalHost().getHostAddress();
869                     } catch (UnknownHostException e) {
870                         if (Log.on) Log.info(this, "strange... host does not know its own address");
871                         return null;
872                     }
873                 case "dnsDomainLevels":
874                     String s = a0.toString();
875                     int i = 0;
876                     while((i = s.indexOf('.', i)) != -1) i++;
877                     return new Integer(i);
878                 case "shExpMatch":
879                     StringTokenizer st = new StringTokenizer(a1.toString(), "*", false);
880                     String[] arr = new String[st.countTokens()];
881                     String s = a0.toString();
882                     for (int i=0; st.hasMoreTokens(); i++) arr[i] = st.nextToken();
883                     return match(arr, s, 0) ? Boolean.TRUE : Boolean.FALSE;
884                 case "weekdayRange":
885                     TimeZone tz = (nargs < 3 || a2 == null || !a2.equals("GMT")) ?
886                         TimeZone.getTimeZone("UTC") : TimeZone.getDefault();
887                     Calendar c = new GregorianCalendar();
888                     c.setTimeZone(tz);
889                     c.setTime(new java.util.Date());
890                     java.util.Date d = c.getTime();
891                     int day = d.getDay();
892                     String d1s = a0.toString().toUpperCase();
893                     int d1 = 0, d2 = 0;
894                     for(int i=0; i<days.length; i++) if (days[i].equals(d1s)) d1 = i;
895                     
896                     if (nargs == 1)
897                         return d1 == day ? Boolean.TRUE : Boolean.FALSE;
898                     
899                     String d2s = a1.toString().toUpperCase();
900                     for(int i=0; i<days.length; i++) if (days[i].equals(d2s)) d2 = i;
901                     
902                     return ((d1 <= d2 && day >= d1 && day <= d2) || (d1 > d2 && (day >= d1 || day <= d2))) ? T : F;
903                     
904                 case "dateRange": throw new JSExn("Ibex does not support dateRange() in PAC scripts");
905                 case "timeRange": throw new JSExn("Ibex does not support timeRange() in PAC scripts");
906                 //#end
907                 return super.callMethod(method, a0, a1, a2, rest, nargs);
908             }       
909             private static boolean match(String[] arr, String s, int index) {
910                 if (index >= arr.length) return true;
911                 for(int i=0; i<s.length(); i++) {
912                     String s2 = s.substring(i);
913                     if (s2.startsWith(arr[index]) && match(arr, s2.substring(arr[index].length()), index + 1)) return true;
914                 }
915                 return false;
916             }
917             public static String[] days = { "SUN", "MON", "TUE", "WED", "THU", "FRI", "SAT" };
918         }
919
920
921         /**
922          *  An implementation of Microsoft's proprietary NTLM authentication protocol.  This code was derived from Eric
923          *  Glass's work, and is copyright as follows:
924          *
925          *  Copyright (c) 2003 Eric Glass     (eglass1 at comcast.net). 
926          *
927          *  Permission to use, copy, modify, and distribute this document for any purpose and without any fee is hereby
928          *  granted, provided that the above copyright notice and this list of conditions appear in all copies.
929          *  The most current version of this document may be obtained from http://davenport.sourceforge.net/ntlm.html .
930          */ 
931         public static class NTLM {
932             
933             public static final byte[] type1 = new byte[] { 0x4e, 0x54, 0x4c, 0x4d, 0x53, 0x53, 0x50, 0x00, 0x01,
934                                                             0x00, 0x00, 0x00, 0x00, 0x02, 0x02, 0x00 };
935             
936             /**
937              * Calculates the NTLM Response for the given challenge, using the
938              * specified password.
939              *
940              * @param password The user's password.
941              * @param challenge The Type 2 challenge from the server.
942              *
943              * @return The NTLM Response.
944              */
945             public static byte[] getNTLMResponse(String password, byte[] challenge)
946                 throws UnsupportedEncodingException {
947                 byte[] ntlmHash = ntlmHash(password);
948                 return lmResponse(ntlmHash, challenge);
949             }
950
951             /**
952              * Calculates the LM Response for the given challenge, using the specified
953              * password.
954              *
955              * @param password The user's password.
956              * @param challenge The Type 2 challenge from the server.
957              *
958              * @return The LM Response.
959              */
960             public static byte[] getLMResponse(String password, byte[] challenge)
961                 {
962                 byte[] lmHash = lmHash(password);
963                 return lmResponse(lmHash, challenge);
964             }
965
966             /**
967              * Calculates the NTLMv2 Response for the given challenge, using the
968              * specified authentication target, username, password, target information
969              * block, and client challenge.
970              *
971              * @param target The authentication target (i.e., domain).
972              * @param user The username. 
973              * @param password The user's password.
974              * @param targetInformation The target information block from the Type 2
975              * message.
976              * @param challenge The Type 2 challenge from the server.
977              * @param clientChallenge The random 8-byte client challenge. 
978              *
979              * @return The NTLMv2 Response.
980              */
981             public static byte[] getNTLMv2Response(String target, String user,
982                                                    String password, byte[] targetInformation, byte[] challenge,
983                                                    byte[] clientChallenge) throws UnsupportedEncodingException {
984                 byte[] ntlmv2Hash = ntlmv2Hash(target, user, password);
985                 byte[] blob = createBlob(targetInformation, clientChallenge);
986                 return lmv2Response(ntlmv2Hash, blob, challenge);
987             }
988
989             /**
990              * Calculates the LMv2 Response for the given challenge, using the
991              * specified authentication target, username, password, and client
992              * challenge.
993              *
994              * @param target The authentication target (i.e., domain).
995              * @param user The username.
996              * @param password The user's password.
997              * @param challenge The Type 2 challenge from the server.
998              * @param clientChallenge The random 8-byte client challenge.
999              *
1000              * @return The LMv2 Response. 
1001              */
1002             public static byte[] getLMv2Response(String target, String user,
1003                                                  String password, byte[] challenge, byte[] clientChallenge)
1004                 throws UnsupportedEncodingException {
1005                 byte[] ntlmv2Hash = ntlmv2Hash(target, user, password);
1006                 return lmv2Response(ntlmv2Hash, clientChallenge, challenge);
1007             }
1008
1009             /**
1010              * Calculates the NTLM2 Session Response for the given challenge, using the
1011              * specified password and client challenge.
1012              *
1013              * @param password The user's password.
1014              * @param challenge The Type 2 challenge from the server.
1015              * @param clientChallenge The random 8-byte client challenge.
1016              *
1017              * @return The NTLM2 Session Response.  This is placed in the NTLM
1018              * response field of the Type 3 message; the LM response field contains
1019              * the client challenge, null-padded to 24 bytes.
1020              */
1021             public static byte[] getNTLM2SessionResponse(String password,
1022                                                          byte[] challenge, byte[] clientChallenge) throws UnsupportedEncodingException {
1023                 byte[] ntlmHash = ntlmHash(password);
1024                 MD5Digest md5 = new MD5Digest();
1025                 md5.update(challenge, 0, challenge.length);
1026                 md5.update(clientChallenge, 0, clientChallenge.length);
1027                 byte[] sessionHash = new byte[8];
1028                 byte[] md5_out = new byte[md5.getDigestSize()];
1029                 md5.doFinal(md5_out, 0);
1030                 System.arraycopy(md5_out, 0, sessionHash, 0, 8);
1031                 return lmResponse(ntlmHash, sessionHash);
1032             }
1033
1034             /**
1035              * Creates the LM Hash of the user's password.
1036              *
1037              * @param password The password.
1038              *
1039              * @return The LM Hash of the given password, used in the calculation
1040              * of the LM Response.
1041              */
1042             private static byte[] lmHash(String password) {
1043                 /*
1044                 byte[] oemPassword = password.toUpperCase().getBytes("UTF8");
1045                 int length = java.lang.Math.min(oemPassword.length, 14);
1046                 byte[] keyBytes = new byte[14];
1047                 System.arraycopy(oemPassword, 0, keyBytes, 0, length);
1048                 Key lowKey = createDESKey(keyBytes, 0);
1049                 Key highKey = createDESKey(keyBytes, 7);
1050                 byte[] magicConstant = "KGS!@#$%".getBytes("UTF8");
1051                 Cipher des = Cipher.getInstance("DES/ECB/NoPadding");
1052                 des.init(Cipher.ENCRYPT_MODE, lowKey);
1053                 byte[] lowHash = des.doFinal(magicConstant);
1054                 des.init(Cipher.ENCRYPT_MODE, highKey);
1055                 byte[] highHash = des.doFinal(magicConstant);
1056                 byte[] lmHash = new byte[16];
1057                 System.arraycopy(lowHash, 0, lmHash, 0, 8);
1058                 System.arraycopy(highHash, 0, lmHash, 8, 8);
1059                 return lmHash;
1060                 */
1061                 return null;
1062             }
1063
1064             /**
1065              * Creates the NTLM Hash of the user's password.
1066              *
1067              * @param password The password.
1068              *
1069              * @return The NTLM Hash of the given password, used in the calculation
1070              * of the NTLM Response and the NTLMv2 and LMv2 Hashes.
1071              */
1072             private static byte[] ntlmHash(String password) throws UnsupportedEncodingException {
1073                 byte[] unicodePassword = password.getBytes("UnicodeLittleUnmarked");
1074                 MD4Digest md4 = new MD4Digest();
1075                 md4.update(unicodePassword, 0, unicodePassword.length);
1076                 byte[] ret = new byte[md4.getDigestSize()];
1077                 return ret;
1078             }
1079
1080             /**
1081              * Creates the NTLMv2 Hash of the user's password.
1082              *
1083              * @param target The authentication target (i.e., domain).
1084              * @param user The username.
1085              * @param password The password.
1086              *
1087              * @return The NTLMv2 Hash, used in the calculation of the NTLMv2
1088              * and LMv2 Responses. 
1089              */
1090             private static byte[] ntlmv2Hash(String target, String user,
1091                                              String password) throws UnsupportedEncodingException {
1092                 byte[] ntlmHash = ntlmHash(password);
1093                 String identity = user.toUpperCase() + target.toUpperCase();
1094                 return hmacMD5(identity.getBytes("UnicodeLittleUnmarked"), ntlmHash);
1095             }
1096
1097             /**
1098              * Creates the LM Response from the given hash and Type 2 challenge.
1099              *
1100              * @param hash The LM or NTLM Hash.
1101              * @param challenge The server challenge from the Type 2 message.
1102              *
1103              * @return The response (either LM or NTLM, depending on the provided
1104              * hash).
1105              */
1106             private static byte[] lmResponse(byte[] hash, byte[] challenge)
1107                 {
1108                 /*
1109                 byte[] keyBytes = new byte[21];
1110                 System.arraycopy(hash, 0, keyBytes, 0, 16);
1111                 Key lowKey = createDESKey(keyBytes, 0);
1112                 Key middleKey = createDESKey(keyBytes, 7);
1113                 Key highKey = createDESKey(keyBytes, 14);
1114                 Cipher des = Cipher.getInstance("DES/ECB/NoPadding");
1115                 des.init(Cipher.ENCRYPT_MODE, lowKey);
1116                 byte[] lowResponse = des.doFinal(challenge);
1117                 des.init(Cipher.ENCRYPT_MODE, middleKey);
1118                 byte[] middleResponse = des.doFinal(challenge);
1119                 des.init(Cipher.ENCRYPT_MODE, highKey);
1120                 byte[] highResponse = des.doFinal(challenge);
1121                 byte[] lmResponse = new byte[24];
1122                 System.arraycopy(lowResponse, 0, lmResponse, 0, 8);
1123                 System.arraycopy(middleResponse, 0, lmResponse, 8, 8);
1124                 System.arraycopy(highResponse, 0, lmResponse, 16, 8);
1125                 return lmResponse;
1126                 */
1127                 return null;
1128             }
1129
1130             /**
1131              * Creates the LMv2 Response from the given hash, client data, and
1132              * Type 2 challenge.
1133              *
1134              * @param hash The NTLMv2 Hash.
1135              * @param clientData The client data (blob or client challenge).
1136              * @param challenge The server challenge from the Type 2 message.
1137              *
1138              * @return The response (either NTLMv2 or LMv2, depending on the
1139              * client data).
1140              */
1141             private static byte[] lmv2Response(byte[] hash, byte[] clientData,
1142                                                byte[] challenge) {
1143                 byte[] data = new byte[challenge.length + clientData.length];
1144                 System.arraycopy(challenge, 0, data, 0, challenge.length);
1145                 System.arraycopy(clientData, 0, data, challenge.length,
1146                                  clientData.length);
1147                 byte[] mac = hmacMD5(data, hash);
1148                 byte[] lmv2Response = new byte[mac.length + clientData.length];
1149                 System.arraycopy(mac, 0, lmv2Response, 0, mac.length);
1150                 System.arraycopy(clientData, 0, lmv2Response, mac.length,
1151                                  clientData.length);
1152                 return lmv2Response;
1153             }
1154
1155             /**
1156              * Creates the NTLMv2 blob from the given target information block and
1157              * client challenge.
1158              *
1159              * @param targetInformation The target information block from the Type 2
1160              * message.
1161              * @param clientChallenge The random 8-byte client challenge.
1162              *
1163              * @return The blob, used in the calculation of the NTLMv2 Response.
1164              */
1165             private static byte[] createBlob(byte[] targetInformation,
1166                                              byte[] clientChallenge) {
1167                 byte[] blobSignature = new byte[] {
1168                     (byte) 0x01, (byte) 0x01, (byte) 0x00, (byte) 0x00
1169                 };
1170                 byte[] reserved = new byte[] {
1171                     (byte) 0x00, (byte) 0x00, (byte) 0x00, (byte) 0x00
1172                 };
1173                 byte[] unknown1 = new byte[] {
1174                     (byte) 0x00, (byte) 0x00, (byte) 0x00, (byte) 0x00
1175                 };
1176                 byte[] unknown2 = new byte[] {
1177                     (byte) 0x00, (byte) 0x00, (byte) 0x00, (byte) 0x00
1178                 };
1179                 long time = System.currentTimeMillis();
1180                 time += 11644473600000l; // milliseconds from January 1, 1601 -> epoch.
1181                 time *= 10000; // tenths of a microsecond.
1182                 // convert to little-endian byte array.
1183                 byte[] timestamp = new byte[8];
1184                 for (int i = 0; i < 8; i++) {
1185                     timestamp[i] = (byte) time;
1186                     time >>>= 8;
1187                 }
1188                 byte[] blob = new byte[blobSignature.length + reserved.length +
1189                                        timestamp.length + clientChallenge.length +
1190                                        unknown1.length + targetInformation.length +
1191                                        unknown2.length];
1192                 int offset = 0;
1193                 System.arraycopy(blobSignature, 0, blob, offset, blobSignature.length);
1194                 offset += blobSignature.length;
1195                 System.arraycopy(reserved, 0, blob, offset, reserved.length);
1196                 offset += reserved.length;
1197                 System.arraycopy(timestamp, 0, blob, offset, timestamp.length);
1198                 offset += timestamp.length;
1199                 System.arraycopy(clientChallenge, 0, blob, offset,
1200                                  clientChallenge.length);
1201                 offset += clientChallenge.length;
1202                 System.arraycopy(unknown1, 0, blob, offset, unknown1.length);
1203                 offset += unknown1.length;
1204                 System.arraycopy(targetInformation, 0, blob, offset,
1205                                  targetInformation.length);
1206                 offset += targetInformation.length;
1207                 System.arraycopy(unknown2, 0, blob, offset, unknown2.length);
1208                 return blob;
1209             }
1210
1211             /**
1212              * Calculates the HMAC-MD5 hash of the given data using the specified
1213              * hashing key.
1214              *
1215              * @param data The data for which the hash will be calculated. 
1216              * @param key The hashing key.
1217              *
1218              * @return The HMAC-MD5 hash of the given data.
1219              */
1220             private static byte[] hmacMD5(byte[] data, byte[] key) {
1221                 byte[] ipad = new byte[64];
1222                 byte[] opad = new byte[64];
1223                 for (int i = 0; i < 64; i++) {
1224                     ipad[i] = (byte) 0x36;
1225                     opad[i] = (byte) 0x5c;
1226                 }
1227                 for (int i = key.length - 1; i >= 0; i--) {
1228                     ipad[i] ^= key[i];
1229                     opad[i] ^= key[i];
1230                 }
1231                 byte[] content = new byte[data.length + 64];
1232                 System.arraycopy(ipad, 0, content, 0, 64);
1233                 System.arraycopy(data, 0, content, 64, data.length);
1234                 MD5Digest md5 = new MD5Digest();
1235                 md5.update(content, 0, content.length);
1236                 data = new byte[md5.getDigestSize()];
1237                 md5.doFinal(data, 0);
1238                 content = new byte[data.length + 64];
1239                 System.arraycopy(opad, 0, content, 0, 64);
1240                 System.arraycopy(data, 0, content, 64, data.length);
1241                 md5 = new MD5Digest();
1242                 md5.update(content, 0, content.length);
1243                 byte[] ret = new byte[md5.getDigestSize()];
1244                 md5.doFinal(ret, 0);
1245                 return ret;
1246             }
1247
1248             /**
1249              * Creates a DES encryption key from the given key material.
1250              *
1251              * @param bytes A byte array containing the DES key material.
1252              * @param offset The offset in the given byte array at which
1253              * the 7-byte key material starts.
1254              *
1255              * @return A DES encryption key created from the key material
1256              * starting at the specified offset in the given byte array.
1257              */
1258                 /*
1259             private static Key createDESKey(byte[] bytes, int offset) {
1260                 byte[] keyBytes = new byte[7];
1261                 System.arraycopy(bytes, offset, keyBytes, 0, 7);
1262                 byte[] material = new byte[8];
1263                 material[0] = keyBytes[0];
1264                 material[1] = (byte) (keyBytes[0] << 7 | (keyBytes[1] & 0xff) >>> 1);
1265                 material[2] = (byte) (keyBytes[1] << 6 | (keyBytes[2] & 0xff) >>> 2);
1266                 material[3] = (byte) (keyBytes[2] << 5 | (keyBytes[3] & 0xff) >>> 3);
1267                 material[4] = (byte) (keyBytes[3] << 4 | (keyBytes[4] & 0xff) >>> 4);
1268                 material[5] = (byte) (keyBytes[4] << 3 | (keyBytes[5] & 0xff) >>> 5);
1269                 material[6] = (byte) (keyBytes[5] << 2 | (keyBytes[6] & 0xff) >>> 6);
1270                 material[7] = (byte) (keyBytes[6] << 1);
1271                 oddParity(material);
1272                 return new SecretKeySpec(material, "DES");
1273             }
1274                 */
1275
1276             /**
1277              * Applies odd parity to the given byte array.
1278              *
1279              * @param bytes The data whose parity bits are to be adjusted for
1280              * odd parity.
1281              */
1282             private static void oddParity(byte[] bytes) {
1283                 for (int i = 0; i < bytes.length; i++) {
1284                     byte b = bytes[i];
1285                     boolean needsParity = (((b >>> 7) ^ (b >>> 6) ^ (b >>> 5) ^
1286                                             (b >>> 4) ^ (b >>> 3) ^ (b >>> 2) ^
1287                                             (b >>> 1)) & 0x01) == 0;
1288                     if (needsParity) {
1289                         bytes[i] |= (byte) 0x01;
1290                     } else {
1291                         bytes[i] &= (byte) 0xfe;
1292                     }
1293                 }
1294             }
1295
1296         }
1297     }
1298 }