AuthHeaderTest.java revision 6073:cea72c2bf071
1/* 2 * Copyright (c) 2003, 2012, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24/** 25 * @test 26 * @bug 4804309 27 * @library ../../../sun/net/www/httptest/ 28 * @build HttpCallback TestHttpServer ClosedChannelList HttpTransaction 29 * @run main AuthHeaderTest 30 * @summary AuthHeaderTest bug 31 */ 32 33import java.io.*; 34import java.net.*; 35 36public class AuthHeaderTest implements HttpCallback { 37 38 static int count = 0; 39 static String authstring; 40 41 void errorReply (HttpTransaction req, String reply) throws IOException { 42 req.addResponseHeader ("Connection", "close"); 43 req.addResponseHeader ("Www-authenticate", reply); 44 req.sendResponse (401, "Unauthorized"); 45 req.orderlyClose(); 46 } 47 48 void okReply (HttpTransaction req) throws IOException { 49 req.setResponseEntityBody ("Hello ."); 50 req.sendResponse (200, "Ok"); 51 req.orderlyClose(); 52 } 53 54 public void request (HttpTransaction req) { 55 try { 56 authstring = req.getRequestHeader ("Authorization"); 57 System.out.println (authstring); 58 switch (count) { 59 case 0: 60 errorReply (req, "Basic realm=\"wallyworld\""); 61 break; 62 case 1: 63 /* client stores a username/pw for wallyworld 64 */ 65 okReply (req); 66 break; 67 } 68 count ++; 69 } catch (IOException e) { 70 e.printStackTrace(); 71 } 72 } 73 74 static void read (InputStream is) throws IOException { 75 int c; 76 System.out.println ("reading"); 77 while ((c=is.read()) != -1) { 78 System.out.write (c); 79 } 80 System.out.println (""); 81 System.out.println ("finished reading"); 82 } 83 84 static void client (String u) throws Exception { 85 URL url = new URL (u); 86 System.out.println ("client opening connection to: " + u); 87 URLConnection urlc = url.openConnection (); 88 InputStream is = urlc.getInputStream (); 89 read (is); 90 is.close(); 91 } 92 93 static TestHttpServer server; 94 95 public static void main (String[] args) throws Exception { 96 MyAuthenticator auth = new MyAuthenticator (); 97 Authenticator.setDefault (auth); 98 try { 99 server = new TestHttpServer (new AuthHeaderTest(), 1, 10, 0); 100 System.out.println ("Server: listening on port: " + server.getLocalPort()); 101 client ("http://localhost:"+server.getLocalPort()+"/d1/foo.html"); 102 } catch (Exception e) { 103 if (server != null) { 104 server.terminate(); 105 } 106 throw e; 107 } 108 int f = auth.getCount(); 109 if (f != 1) { 110 except ("Authenticator was called "+f+" times. Should be 1"); 111 } 112 server.terminate(); 113 } 114 115 public static void except (String s) { 116 server.terminate(); 117 throw new RuntimeException (s); 118 } 119 120 static class MyAuthenticator extends Authenticator { 121 MyAuthenticator () { 122 super (); 123 } 124 125 int count = 0; 126 127 public PasswordAuthentication getPasswordAuthentication () { 128 PasswordAuthentication pw; 129 pw = new PasswordAuthentication ("user", "pass2".toCharArray()); 130 count ++; 131 return pw; 132 } 133 134 public int getCount () { 135 return (count); 136 } 137 } 138} 139