1 /* 2 * Copyright (c) 2005, 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 * @summary Unit test for java.net.CookieManager 27 * @bug 6244040 28 * @library ../../../sun/net/www/httptest/ 29 * @build HttpCallback TestHttpServer ClosedChannelList HttpTransaction 30 * @run main/othervm -ea CookieManagerTest 31 * @author Edward Wang 32 */ 33 34 import java.net.*; 35 import java.util.*; 36 import java.io.*; 37 import sun.net.www.MessageHeader; 38 39 public class CookieManagerTest { 40 static CookieHttpTransaction httpTrans; 41 static TestHttpServer server; 42 43 public static void main(String[] args) throws Exception { 44 startHttpServer(); 45 makeHttpCall(); 46 47 if (httpTrans.badRequest) { 48 throw new RuntimeException("Test failed : bad cookie header"); 49 } 50 } 51 52 public static void startHttpServer() { 53 try { 54 httpTrans = new CookieHttpTransaction(); 55 server = new TestHttpServer(httpTrans, 1, 1, 0); 56 } catch (IOException e) { 57 e.printStackTrace(); 58 } 59 } 60 61 public static void makeHttpCall() { 62 try { 63 System.out.println("http server listen on: " + server.getLocalPort()); 64 65 // install CookieManager to use 66 CookieHandler.setDefault(new CookieManager()); 67 68 for (int i = 0; i < CookieHttpTransaction.testCount; i++) { 69 System.out.println("====== CookieManager test " + (i+1) + " ======"); 70 ((CookieManager)CookieHandler.getDefault()).setCookiePolicy(CookieHttpTransaction.testPolicies[i]); 71 ((CookieManager)CookieHandler.getDefault()).getCookieStore().removeAll(); 72 URL url = new URL("http" , InetAddress.getLocalHost().getHostAddress(), 73 server.getLocalPort(), CookieHttpTransaction.testCases[i][0].serverPath); 74 HttpURLConnection uc = (HttpURLConnection)url.openConnection(); 75 uc.getResponseCode(); 76 uc.disconnect(); 77 } 78 } catch (IOException e) { 79 e.printStackTrace(); 80 } finally { 81 server.terminate(); 82 } 83 } 84 } 85 86 class CookieHttpTransaction implements HttpCallback { 87 public static boolean badRequest = false; 88 // the main test control logic will also loop exactly this number 89 // to send http request 90 public static final int testCount = 6; 91 92 private String localHostAddr = "127.0.0.1"; 93 94 // test cases 95 public static class CookieTestCase { 96 public String headerToken; 97 public String cookieToSend; 98 public String cookieToRecv; 99 public String serverPath; 100 101 public CookieTestCase(String h, String cts, String ctr, String sp) { 102 headerToken = h; 103 cookieToSend = cts; 104 cookieToRecv = ctr; 105 serverPath = sp; 106 } 107 }; 108 109 // 110 // these two must match each other, i.e. testCases.length == testPolicies.length 111 // 112 public static CookieTestCase[][] testCases = null; // the test cases to run; each test case may contain multiple roundtrips 113 public static CookiePolicy[] testPolicies = null; // indicates what CookiePolicy to use with each test cases 114 115 CookieHttpTransaction() { 116 testCases = new CookieTestCase[testCount][]; 117 testPolicies = new CookiePolicy[testCount]; 118 119 try { 120 localHostAddr = InetAddress.getLocalHost().getHostAddress(); 121 } catch (Exception ignored) { 122 }; 123 int count = 0; 124 125 // an http session with Netscape cookies exchanged 126 testPolicies[count] = CookiePolicy.ACCEPT_ORIGINAL_SERVER; 127 testCases[count++] = new CookieTestCase[]{ 128 new CookieTestCase("Set-Cookie", 129 "CUSTOMER=WILE:BOB; path=/; expires=Wednesday, 09-Nov-2030 23:12:40 GMT;" + "domain=." + localHostAddr, 130 "CUSTOMER=WILE:BOB", 131 "/" 132 ), 133 new CookieTestCase("Set-Cookie", 134 "PART_NUMBER=ROCKET_LAUNCHER_0001; path=/;" + "domain=." + localHostAddr, 135 "CUSTOMER=WILE:BOB; PART_NUMBER=ROCKET_LAUNCHER_0001", 136 "/" 137 ), 138 new CookieTestCase("Set-Cookie", 139 "SHIPPING=FEDEX; path=/foo;" + "domain=." + localHostAddr, 140 "CUSTOMER=WILE:BOB; PART_NUMBER=ROCKET_LAUNCHER_0001", 141 "/" 142 ), 143 new CookieTestCase("Set-Cookie", 144 "SHIPPING=FEDEX; path=/foo;" + "domain=." + localHostAddr, 145 "CUSTOMER=WILE:BOB; PART_NUMBER=ROCKET_LAUNCHER_0001; SHIPPING=FEDEX", 146 "/foo" 147 ) 148 }; 149 150 // check whether or not path rule is applied 151 testPolicies[count] = CookiePolicy.ACCEPT_ORIGINAL_SERVER; 152 testCases[count++] = new CookieTestCase[]{ 153 new CookieTestCase("Set-Cookie", 154 "PART_NUMBER=ROCKET_LAUNCHER_0001; path=/;" + "domain=." + localHostAddr, 155 "PART_NUMBER=ROCKET_LAUNCHER_0001", 156 "/" 157 ), 158 new CookieTestCase("Set-Cookie", 159 "PART_NUMBER=RIDING_ROCKET_0023; path=/ammo;" + "domain=." + localHostAddr, 160 "PART_NUMBER=RIDING_ROCKET_0023; PART_NUMBER=ROCKET_LAUNCHER_0001", 161 "/ammo" 162 ) 163 }; 164 165 // an http session with rfc2965 cookies exchanged 166 testPolicies[count] = CookiePolicy.ACCEPT_ORIGINAL_SERVER; 167 testCases[count++] = new CookieTestCase[]{ 168 new CookieTestCase("Set-Cookie2", 169 "Customer=\"WILE_E_COYOTE\"; Version=\"1\"; Path=\"/acme\";" + "domain=." + localHostAddr, 170 "$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";$Domain=\"." + localHostAddr + "\"", 171 "/acme/login" 172 ), 173 new CookieTestCase("Set-Cookie2", 174 "Part_Number=\"Rocket_Launcher_0001\"; Version=\"1\";Path=\"/acme\";" + "domain=." + localHostAddr, 175 "$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";" + "$Domain=\"." + localHostAddr + "\"" + "; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";" + "$Domain=\"." + localHostAddr + "\"", 176 "/acme/pickitem" 177 ), 178 new CookieTestCase("Set-Cookie2", 179 "Shipping=\"FedEx\"; Version=\"1\"; Path=\"/acme\";" + "domain=." + localHostAddr, 180 "$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 + "\"", 181 "/acme/shipping" 182 ) 183 }; 184 185 // check whether or not the path rule is applied 186 testPolicies[count] = CookiePolicy.ACCEPT_ORIGINAL_SERVER; 187 testCases[count++] = new CookieTestCase[]{ 188 new CookieTestCase("Set-Cookie2", 189 "Part_Number=\"Rocket_Launcher_0001\"; Version=\"1\"; Path=\"/acme\";" + "domain=." + localHostAddr, 190 "$Version=\"1\"; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";$Domain=\"." + localHostAddr + "\"", 191 "/acme/ammo" 192 ), 193 new CookieTestCase("Set-Cookie2", 194 "Part_Number=\"Riding_Rocket_0023\"; Version=\"1\"; Path=\"/acme/ammo\";" + "domain=." + localHostAddr, 195 "$Version=\"1\"; Part_Number=\"Riding_Rocket_0023\";$Path=\"/acme/ammo\";$Domain=\"." + localHostAddr + "\"" + "; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";" + "$Domain=\"." + localHostAddr + "\"", 196 "/acme/ammo" 197 ), 198 new CookieTestCase("", 199 "", 200 "$Version=\"1\"; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";" + "$Domain=\"." + localHostAddr + "\"", 201 "/acme/parts" 202 ) 203 }; 204 205 // new cookie should overwrite old cookie 206 testPolicies[count] = CookiePolicy.ACCEPT_ORIGINAL_SERVER; 207 testCases[count++] = new CookieTestCase[]{ 208 new CookieTestCase("Set-Cookie2", 209 "Part_Number=\"Rocket_Launcher_0001\"; Version=\"1\"; Path=\"/acme\";" + "domain=." + localHostAddr, 210 "$Version=\"1\"; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";$Domain=\"." + localHostAddr + "\"", 211 "/acme" 212 ), 213 new CookieTestCase("Set-Cookie2", 214 "Part_Number=\"Rocket_Launcher_2000\"; Version=\"1\"; Path=\"/acme\";" + "domain=." + localHostAddr, 215 "$Version=\"1\"; Part_Number=\"Rocket_Launcher_2000\";$Path=\"/acme\";$Domain=\"." + localHostAddr + "\"", 216 "/acme" 217 ) 218 }; 219 220 // cookies without domain attributes 221 // RFC 2965 states that domain should default to host 222 testPolicies[count] = CookiePolicy.ACCEPT_ALL; 223 testCases[count++] = new CookieTestCase[]{ 224 new CookieTestCase("Set-Cookie2", 225 "Customer=\"WILE_E_COYOTE\"; Version=\"1\"; Path=\"/acme\"", 226 "$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";$Domain=\""+localHostAddr+"\"", 227 "/acme/login" 228 ), 229 new CookieTestCase("Set-Cookie2", 230 "Part_Number=\"Rocket_Launcher_0001\"; Version=\"1\";Path=\"/acme\"", 231 "$Version=\"1\"; Customer=\"WILE_E_COYOTE\";$Path=\"/acme\";$Domain=\""+localHostAddr+"\"" + "; Part_Number=\"Rocket_Launcher_0001\";$Path=\"/acme\";$Domain=\""+localHostAddr+"\"", 232 "/acme/pickitem" 233 ), 234 new CookieTestCase("Set-Cookie2", 235 "Shipping=\"FedEx\"; Version=\"1\"; Path=\"/acme\"", 236 "$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+"\"", 237 "/acme/shipping" 238 ) 239 }; 240 241 assert count == testCount; 242 } 243 244 private int testcaseDone = 0; 245 private int testDone = 0; 246 /* 247 * Our http server which is conducted by testCases array 248 */ 249 public void request(HttpTransaction trans) { 250 try { 251 if (testDone < testCases[testcaseDone].length) { 252 // still have other tests to run, 253 // check the Cookie header and then redirect it 254 if (testDone > 0) checkResquest(trans); 255 trans.addResponseHeader("Location", testCases[testcaseDone][testDone].serverPath); 256 trans.addResponseHeader(testCases[testcaseDone][testDone].headerToken, 257 testCases[testcaseDone][testDone].cookieToSend); 258 testDone++; 259 trans.sendResponse(302, "Moved Temporarily"); 260 } else { 261 // the last test of this test case 262 if (testDone > 0) checkResquest(trans); 263 testcaseDone++; 264 testDone = 0; 265 trans.sendResponse(200, "OK"); 266 } 267 } catch (Exception e) { 268 e.printStackTrace(); 269 } 270 } 271 272 private void checkResquest(HttpTransaction trans) { 273 String cookieHeader = null; 274 275 assert testDone > 0; 276 cookieHeader = trans.getRequestHeader("Cookie"); 277 if (cookieHeader != null && 278 cookieHeader.equalsIgnoreCase(testCases[testcaseDone][testDone-1].cookieToRecv)) 279 { 280 System.out.printf("%15s %s\n", "PASSED:", cookieHeader); 281 } else { 282 System.out.printf("%15s %s\n", "FAILED:", cookieHeader); 283 System.out.printf("%15s %s\n\n", "should be:", testCases[testcaseDone][testDone-1].cookieToRecv); 284 badRequest = true; 285 } 286 } 287 }