jdk/test/sun/net/www/AuthHeaderTest.java
changeset 2 90ce3da70b43
child 5506 202f599c92aa
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/test/sun/net/www/AuthHeaderTest.java	Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,138 @@
+/*
+ * Copyright 2003 Sun Microsystems, Inc.  All Rights Reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
+ * CA 95054 USA or visit www.sun.com if you need additional information or
+ * have any questions.
+ */
+
+/**
+ * @test
+ * @bug 4804309
+ * @library ../../../sun/net/www/httptest/
+ * @build HttpCallback HttpServer ClosedChannelList HttpTransaction
+ * @run main AuthHeaderTest
+ * @summary AuthHeaderTest bug
+ */
+
+import java.io.*;
+import java.net.*;
+
+public class AuthHeaderTest implements HttpCallback {
+
+    static int count = 0;
+    static String authstring;
+
+    void errorReply (HttpTransaction req, String reply) throws IOException {
+        req.addResponseHeader ("Connection", "close");
+        req.addResponseHeader ("Www-authenticate", reply);
+        req.sendResponse (401, "Unauthorized");
+        req.orderlyClose();
+    }
+
+    void okReply (HttpTransaction req) throws IOException {
+        req.setResponseEntityBody ("Hello .");
+        req.sendResponse (200, "Ok");
+        req.orderlyClose();
+    }
+
+    public void request (HttpTransaction req) {
+        try {
+            authstring = req.getRequestHeader ("Authorization");
+            System.out.println (authstring);
+            switch (count) {
+            case 0:
+                errorReply (req, "Basic realm=\"wallyworld\"");
+                break;
+            case 1:
+                /* client stores a username/pw for wallyworld
+                 */
+                okReply (req);
+                break;
+            }
+            count ++;
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+    }
+
+    static void read (InputStream is) throws IOException {
+        int c;
+        System.out.println ("reading");
+        while ((c=is.read()) != -1) {
+            System.out.write (c);
+        }
+        System.out.println ("");
+        System.out.println ("finished reading");
+    }
+
+    static void client (String u) throws Exception {
+        URL url = new URL (u);
+        System.out.println ("client opening connection to: " + u);
+        URLConnection urlc = url.openConnection ();
+        InputStream is = urlc.getInputStream ();
+        read (is);
+        is.close();
+    }
+
+    static HttpServer server;
+
+    public static void main (String[] args) throws Exception {
+        MyAuthenticator auth = new MyAuthenticator ();
+        Authenticator.setDefault (auth);
+        try {
+            server = new HttpServer (new AuthHeaderTest(), 1, 10, 0);
+            System.out.println ("Server: listening on port: " + server.getLocalPort());
+            client ("http://localhost:"+server.getLocalPort()+"/d1/foo.html");
+        } catch (Exception e) {
+            if (server != null) {
+                server.terminate();
+            }
+            throw e;
+        }
+        int f = auth.getCount();
+        if (f != 1) {
+            except ("Authenticator was called "+f+" times. Should be 1");
+        }
+        server.terminate();
+    }
+
+    public static void except (String s) {
+        server.terminate();
+        throw new RuntimeException (s);
+    }
+
+    static class MyAuthenticator extends Authenticator {
+        MyAuthenticator () {
+            super ();
+        }
+
+        int count = 0;
+
+        public PasswordAuthentication getPasswordAuthentication () {
+            PasswordAuthentication pw;
+            pw = new PasswordAuthentication ("user", "pass2".toCharArray());
+            count ++;
+            return pw;
+        }
+
+        public int getCount () {
+            return (count);
+        }
+    }
+}