e9eaba3d53
Add new traces for better diagnosis, refrain binding to the wildcard address when possible. Reviewed-by: chegar, xuelei
384 lines
16 KiB
Java
384 lines
16 KiB
Java
/*
|
|
* Copyright (c) 2005, 2019, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
|
|
* or visit www.oracle.com if you need additional information or have any
|
|
* questions.
|
|
*/
|
|
|
|
/*
|
|
* @test
|
|
* @summary Unit test for java.net.CookieManager
|
|
* @bug 6244040 7150552 7051862
|
|
* @modules jdk.httpserver
|
|
* java.logging
|
|
* @run main/othervm -ea -esa CookieManagerTest
|
|
* @author Edward Wang
|
|
*/
|
|
|
|
import com.sun.net.httpserver.*;
|
|
import java.util.Collections;
|
|
import java.util.LinkedList;
|
|
import java.util.List;
|
|
import java.io.IOException;
|
|
import java.net.*;
|
|
import java.util.logging.Level;
|
|
import java.util.logging.Logger;
|
|
import static java.net.Proxy.NO_PROXY;
|
|
|
|
public class CookieManagerTest {
|
|
|
|
static CookieTransactionHandler httpTrans;
|
|
static HttpServer server;
|
|
|
|
static final String hostAddress = getAddr();
|
|
|
|
/** Returns an IP literal suitable for use by the test. */
|
|
static String getAddr() {
|
|
try {
|
|
InetAddress lh = InetAddress.getLocalHost();
|
|
System.out.println("Trying: " + lh);
|
|
if (lh.isReachable(5_000)) {
|
|
System.out.println("Using: " + lh);
|
|
return lh.getHostAddress();
|
|
}
|
|
} catch (IOException x) {
|
|
System.out.println("Debug: caught:" + x);
|
|
}
|
|
InetAddress loopback = InetAddress.getLoopbackAddress();
|
|
System.out.println("Using: \"" + loopback.getHostAddress() + "\"");
|
|
return loopback.getHostAddress();
|
|
}
|
|
|
|
public static void main(String[] args) throws Exception {
|
|
// logs everything...
|
|
Logger root = Logger.getLogger("");
|
|
root.setLevel(Level.ALL);
|
|
root.getHandlers()[0].setLevel(Level.ALL);
|
|
|
|
startHttpServer();
|
|
makeHttpCall();
|
|
|
|
if (httpTrans.badRequest) {
|
|
throw new RuntimeException("Test failed : bad cookie header");
|
|
}
|
|
checkCookiePolicy();
|
|
}
|
|
|
|
public static void startHttpServer() throws IOException {
|
|
httpTrans = new CookieTransactionHandler();
|
|
server = HttpServer.create(new InetSocketAddress(hostAddress, 0), 0);
|
|
server.createContext("/", httpTrans);
|
|
server.start();
|
|
}
|
|
|
|
/*
|
|
* Checks if CookiePolicy.ACCEPT_ORIGINAL_SERVER#shouldAccept()
|
|
* returns false for null arguments
|
|
*/
|
|
private static void checkCookiePolicy() throws Exception {
|
|
CookiePolicy cp = CookiePolicy.ACCEPT_ORIGINAL_SERVER;
|
|
boolean retVal;
|
|
retVal = cp.shouldAccept(null, null);
|
|
checkValue(retVal);
|
|
retVal = cp.shouldAccept(null, new HttpCookie("CookieName", "CookieVal"));
|
|
checkValue(retVal);
|
|
retVal = cp.shouldAccept((new URL("http", "localhost", 2345, "/")).toURI(),
|
|
null);
|
|
checkValue(retVal);
|
|
}
|
|
|
|
private static void checkValue(boolean val) {
|
|
if (val)
|
|
throw new RuntimeException("Return value is not false!");
|
|
}
|
|
|
|
public static void makeHttpCall() throws IOException {
|
|
try {
|
|
int port = server.getAddress().getPort();
|
|
System.out.println("http server listenining on: " + port);
|
|
|
|
// install CookieManager to use
|
|
CookieHandler.setDefault(new CookieManager());
|
|
|
|
for (int i = 0; i < CookieTransactionHandler.testCount; i++) {
|
|
System.out.println("====== CookieManager test " + (i+1)
|
|
+ " ======");
|
|
((CookieManager)CookieHandler.getDefault())
|
|
.setCookiePolicy(CookieTransactionHandler.testPolicies[i]);
|
|
((CookieManager)CookieHandler.getDefault())
|
|
.getCookieStore().removeAll();
|
|
URL url = new URL("http" ,
|
|
hostAddress,
|
|
server.getAddress().getPort(),
|
|
CookieTransactionHandler.testCases[i][0]
|
|
.serverPath);
|
|
System.out.println("Requesting " + url);
|
|
HttpURLConnection uc = (HttpURLConnection)url.openConnection(NO_PROXY);
|
|
uc.getResponseCode();
|
|
uc.disconnect();
|
|
}
|
|
} finally {
|
|
server.stop(0);
|
|
}
|
|
}
|
|
}
|
|
|
|
class CookieTransactionHandler implements HttpHandler {
|
|
|
|
private int testcaseDone = 0;
|
|
private int testDone = 0;
|
|
|
|
public static boolean badRequest = false;
|
|
// the main test control logic will also loop exactly this number
|
|
// to send http request
|
|
public static final int testCount = 6;
|
|
|
|
@Override
|
|
public void handle(HttpExchange exchange) throws IOException {
|
|
if (testDone < testCases[testcaseDone].length) {
|
|
// still have other tests to run,
|
|
// check the Cookie header and then redirect it
|
|
if (testDone > 0) checkRequest(exchange.getRequestHeaders());
|
|
exchange.getResponseHeaders().add("Location",
|
|
testCases[testcaseDone][testDone].serverPath);
|
|
exchange.getResponseHeaders()
|
|
.add(testCases[testcaseDone][testDone].headerToken,
|
|
testCases[testcaseDone][testDone].cookieToSend);
|
|
exchange.sendResponseHeaders(302, -1);
|
|
testDone++;
|
|
} else {
|
|
// the last test of this test case
|
|
if (testDone > 0) checkRequest(exchange.getRequestHeaders());
|
|
testcaseDone++;
|
|
testDone = 0;
|
|
exchange.sendResponseHeaders(200, -1);
|
|
}
|
|
exchange.close();
|
|
}
|
|
|
|
private static String trim(String s) {
|
|
StringBuilder sb = new StringBuilder();
|
|
for (int i=0; i<s.length(); i++) {
|
|
char c = s.charAt(i);
|
|
if (!Character.isWhitespace(c))
|
|
sb.append(c);
|
|
}
|
|
return sb.toString();
|
|
}
|
|
|
|
private static boolean cookieEquals(String s1, String s2) {
|
|
s1 = trim(s1);
|
|
s2 = trim(s2);
|
|
String[] s1a = s1.split(";");
|
|
String[] s2a = s2.split(";");
|
|
List<String> l1 = new LinkedList(List.of(s1a));
|
|
List<String> l2 = new LinkedList(List.of(s2a));
|
|
Collections.sort(l1);
|
|
Collections.sort(l2);
|
|
int i = 0;
|
|
for (String s : l1) {
|
|
if (!s.equals(l2.get(i++))) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
private void checkRequest(Headers hdrs) {
|
|
|
|
assert testDone > 0;
|
|
String cookieHeader = hdrs.getFirst("Cookie");
|
|
if (cookieHeader != null && cookieEquals(
|
|
cookieHeader, testCases[testcaseDone][testDone-1].cookieToRecv))
|
|
{
|
|
System.out.printf("%15s %s\n", "PASSED:", cookieHeader);
|
|
} else {
|
|
System.out.printf("%15s %s\n", "FAILED:", cookieHeader);
|
|
System.out.printf("%15s %s\n\n", "should be:",
|
|
testCases[testcaseDone][testDone-1].cookieToRecv);
|
|
badRequest = true;
|
|
}
|
|
}
|
|
|
|
// test cases
|
|
public static class CookieTestCase {
|
|
public String headerToken;
|
|
public String cookieToSend;
|
|
public String cookieToRecv;
|
|
public String serverPath;
|
|
|
|
public CookieTestCase(String h, String cts, String ctr, String sp) {
|
|
headerToken = h;
|
|
cookieToSend = cts;
|
|
cookieToRecv = ctr;
|
|
serverPath = sp;
|
|
}
|
|
};
|
|
|
|
/*
|
|
* these two must match each other,
|
|
* i.e. testCases.length == testPolicies.length
|
|
*/
|
|
|
|
// the test cases to run; each test case may contain multiple roundtrips
|
|
public static CookieTestCase[][] testCases = null;
|
|
// indicates what CookiePolicy to use with each test cases
|
|
public static CookiePolicy[] testPolicies = null;
|
|
|
|
CookieTransactionHandler() {
|
|
testCases = new CookieTestCase[testCount][];
|
|
testPolicies = new CookiePolicy[testCount];
|
|
|
|
String localHostAddr = CookieManagerTest.hostAddress;
|
|
|
|
int count = 0;
|
|
|
|
// an http session with Netscape cookies exchanged
|
|
testPolicies[count] = CookiePolicy.ACCEPT_ORIGINAL_SERVER;
|
|
testCases[count++] = new CookieTestCase[]{
|
|
new CookieTestCase("Set-Cookie",
|
|
"CUSTOMER=WILE:BOB; " +
|
|
"path=/; expires=Sat, 09-Nov-2030 23:12:40 GMT;" + "domain=." +
|
|
localHostAddr,
|
|
"CUSTOMER=WILE:BOB",
|
|
"/"
|
|
),
|
|
new CookieTestCase("Set-Cookie",
|
|
"PART_NUMBER=ROCKET_LAUNCHER_0001; path=/;" + "domain=." + localHostAddr,
|
|
"CUSTOMER=WILE:BOB; PART_NUMBER=ROCKET_LAUNCHER_0001",
|
|
"/"
|
|
),
|
|
new CookieTestCase("Set-Cookie",
|
|
"SHIPPING=FEDEX; path=/foo;" + "domain=." + localHostAddr,
|
|
"CUSTOMER=WILE:BOB; PART_NUMBER=ROCKET_LAUNCHER_0001",
|
|
"/"
|
|
),
|
|
new CookieTestCase("Set-Cookie",
|
|
"SHIPPING=FEDEX; path=/foo;" + "domain=." + localHostAddr,
|
|
"CUSTOMER=WILE:BOB; PART_NUMBER=ROCKET_LAUNCHER_0001; SHIPPING=FEDEX",
|
|
"/foo"
|
|
)
|
|
};
|
|
|
|
// check whether or not path rule is applied
|
|
testPolicies[count] = CookiePolicy.ACCEPT_ORIGINAL_SERVER;
|
|
testCases[count++] = new CookieTestCase[]{
|
|
new CookieTestCase("Set-Cookie",
|
|
"PART_NUMBER=ROCKET_LAUNCHER_0001; path=/;" + "domain=." + localHostAddr,
|
|
"PART_NUMBER=ROCKET_LAUNCHER_0001",
|
|
"/"
|
|
),
|
|
new CookieTestCase("Set-Cookie",
|
|
"PART_NUMBER=RIDING_ROCKET_0023; path=/ammo;" + "domain=." + localHostAddr,
|
|
"PART_NUMBER=RIDING_ROCKET_0023; PART_NUMBER=ROCKET_LAUNCHER_0001",
|
|
"/ammo"
|
|
)
|
|
};
|
|
|
|
// an http session with rfc2965 cookies exchanged
|
|
testPolicies[count] = CookiePolicy.ACCEPT_ORIGINAL_SERVER;
|
|
testCases[count++] = new CookieTestCase[]{
|
|
new CookieTestCase("Set-Cookie2",
|
|
"Customer=\"WILE_E_COYOTE\"; Version=\"1\"; Path=\"/acme\";" + "domain=." + localHostAddr,
|
|
"$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";$Domain=\"." + localHostAddr + "\"",
|
|
"/acme/login"
|
|
),
|
|
new CookieTestCase("Set-Cookie2",
|
|
"Part_Number=\"Rocket_Launcher_0001\"; Version=\"1\";Path=\"/acme\";" + "domain=." + localHostAddr,
|
|
"$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";" + "$Domain=\"." +
|
|
localHostAddr + "\"" + "; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";"
|
|
+ "$Domain=\"." + localHostAddr + "\"",
|
|
"/acme/pickitem"
|
|
),
|
|
new CookieTestCase("Set-Cookie2",
|
|
"Shipping=\"FedEx\"; Version=\"1\"; Path=\"/acme\";" + "domain=." + localHostAddr,
|
|
"$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";" + "$Domain=\"." + localHostAddr +
|
|
"\"" + "; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";" + "$Domain=\"."
|
|
+ localHostAddr + "\"" + "; Shipping=\"FedEx\";$Path=\"/acme\";" +
|
|
"$Domain=\"." + localHostAddr + "\"",
|
|
"/acme/shipping"
|
|
)
|
|
};
|
|
|
|
// check whether or not the path rule is applied
|
|
testPolicies[count] = CookiePolicy.ACCEPT_ORIGINAL_SERVER;
|
|
testCases[count++] = new CookieTestCase[]{
|
|
new CookieTestCase("Set-Cookie2",
|
|
"Part_Number=\"Rocket_Launcher_0001\"; Version=\"1\"; Path=\"/acme\";" + "domain=." + localHostAddr,
|
|
"$Version=\"1\"; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";$Domain=\"." + localHostAddr + "\"",
|
|
"/acme/ammo"
|
|
),
|
|
new CookieTestCase("Set-Cookie2",
|
|
"Part_Number=\"Riding_Rocket_0023\"; Version=\"1\"; Path=\"/acme/ammo\";" + "domain=."
|
|
+ localHostAddr,
|
|
"$Version=\"1\"; Part_Number=\"Riding_Rocket_0023\";$Path=\"/acme/ammo\";$Domain=\"."
|
|
+ localHostAddr + "\"" + "; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";"
|
|
+ "$Domain=\"." + localHostAddr + "\"",
|
|
"/acme/ammo"
|
|
),
|
|
new CookieTestCase("",
|
|
"",
|
|
"$Version=\"1\"; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";" + "$Domain=\"." + localHostAddr + "\"",
|
|
"/acme/parts"
|
|
)
|
|
};
|
|
|
|
// new cookie should overwrite old cookie
|
|
testPolicies[count] = CookiePolicy.ACCEPT_ORIGINAL_SERVER;
|
|
testCases[count++] = new CookieTestCase[]{
|
|
new CookieTestCase("Set-Cookie2",
|
|
"Part_Number=\"Rocket_Launcher_0001\"; Version=\"1\"; Path=\"/acme\";" + "domain=." + localHostAddr,
|
|
"$Version=\"1\"; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";$Domain=\"." + localHostAddr + "\"",
|
|
"/acme"
|
|
),
|
|
new CookieTestCase("Set-Cookie2",
|
|
"Part_Number=\"Rocket_Launcher_2000\"; Version=\"1\"; Path=\"/acme\";" + "domain=." + localHostAddr,
|
|
"$Version=\"1\"; Part_Number=\"Rocket_Launcher_2000\";$Path=\"/acme\";$Domain=\"." + localHostAddr + "\"",
|
|
"/acme"
|
|
)
|
|
};
|
|
|
|
// cookies without domain attributes
|
|
// RFC 2965 states that domain should default to host
|
|
testPolicies[count] = CookiePolicy.ACCEPT_ALL;
|
|
testCases[count++] = new CookieTestCase[]{
|
|
new CookieTestCase("Set-Cookie2",
|
|
"Customer=\"WILE_E_COYOTE\"; Version=\"1\"; Path=\"/acme\"",
|
|
"$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";$Domain=\""+localHostAddr+"\"",
|
|
"/acme/login"
|
|
),
|
|
new CookieTestCase("Set-Cookie2",
|
|
"Part_Number=\"Rocket_Launcher_0001\"; Version=\"1\";Path=\"/acme\"",
|
|
"$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";$Domain=\""+localHostAddr+"\"" +
|
|
"; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";$Domain=\""+localHostAddr+"\"",
|
|
"/acme/pickitem"
|
|
),
|
|
new CookieTestCase("Set-Cookie2",
|
|
"Shipping=\"FedEx\"; Version=\"1\"; Path=\"/acme\"",
|
|
"$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";$Domain=\""+localHostAddr+"\"" +
|
|
"; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";$Domain=\""+localHostAddr+"\"" +
|
|
"; Shipping=\"FedEx\";$Path=\"/acme\";$Domain=\""+localHostAddr+"\"",
|
|
"/acme/shipping"
|
|
)
|
|
};
|
|
|
|
assert count == testCount;
|
|
}
|
|
}
|