1/* 2 * Copyright (c) 2005, 2006, 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 6270015 27 * @summary Light weight HTTP server 28 */ 29 30import com.sun.net.httpserver.*; 31 32import java.util.*; 33import java.util.concurrent.*; 34import java.io.*; 35import java.net.*; 36import java.security.*; 37import javax.security.auth.callback.*; 38import javax.net.ssl.*; 39 40/** 41 * Test authentication 42 */ 43 44public class Test2 extends Test { 45 46 public static void main (String[] args) throws Exception { 47 Handler handler = new Handler(); 48 InetSocketAddress addr = new InetSocketAddress (0); 49 HttpServer server = HttpServer.create (addr, 0); 50 HttpContext ctx = server.createContext ("/test", handler); 51 BasicAuthenticator a = new BasicAuthenticator ("foobar@test.realm") { 52 public boolean checkCredentials (String username, String pw) { 53 return "fred".equals(username) && pw.charAt(0) == 'x'; 54 } 55 }; 56 57 ctx.setAuthenticator (a); 58 ExecutorService executor = Executors.newCachedThreadPool(); 59 server.setExecutor (executor); 60 server.start (); 61 java.net.Authenticator.setDefault (new MyAuthenticator()); 62 63 URL url = new URL ("http://localhost:"+server.getAddress().getPort()+"/test/foo.html"); 64 System.out.print ("Test2: " ); 65 HttpURLConnection urlc = (HttpURLConnection)url.openConnection (); 66 InputStream is = urlc.getInputStream(); 67 int c = 0; 68 while (is.read()!= -1) { 69 c ++; 70 } 71 server.stop(2); 72 executor.shutdown(); 73 if (error ) { 74 throw new RuntimeException ("test failed error"); 75 } 76 if (c != 0) { 77 throw new RuntimeException ("test failed c"); 78 } 79 if (count != 2) { 80 throw new RuntimeException ("test failed count = " + count); 81 } 82 System.out.println ("OK"); 83 84 } 85 86 public static boolean error = false; 87 public static int count = 0; 88 89 static class MyAuthenticator extends java.net.Authenticator { 90 public PasswordAuthentication getPasswordAuthentication () { 91 PasswordAuthentication pw; 92 if (!getRequestingPrompt().equals ("foobar@test.realm")) { 93 Test2.error = true; 94 } 95 if (count == 0) { 96 pw = new PasswordAuthentication ("bad", "wrong".toCharArray()); 97 } else { 98 pw = new PasswordAuthentication ("fred", "xyz".toCharArray()); 99 } 100 count ++; 101 return pw; 102 } 103 } 104 105 static class Handler implements HttpHandler { 106 int invocation = 1; 107 public void handle (HttpExchange t) 108 throws IOException 109 { 110 InputStream is = t.getRequestBody(); 111 Headers map = t.getRequestHeaders(); 112 Headers rmap = t.getResponseHeaders(); 113 while (is.read () != -1) ; 114 is.close(); 115 t.sendResponseHeaders (200, -1); 116 HttpPrincipal p = t.getPrincipal (); 117 if (!p.getUsername().equals("fred")) { 118 error = true; 119 } 120 if (!p.getRealm().equals("foobar@test.realm")) { 121 error = true; 122 } 123 t.close(); 124 } 125 } 126} 127