1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 /** 6 * Test whether the given domain is valid for a cookie. 7 * 8 * @param {string} domain Domain for a cookie. 9 * @return {boolean} True if the domain is valid, otherwise false. 10 */ 11 function isDomainValid(domain) { 12 var dummyCookie = 'ChromeDriverwjers908fljsdf37459fsdfgdfwru='; 13 14 document.cookie = dummyCookie + '; domain=' + domain; 15 if (document.cookie.indexOf(dummyCookie) != -1) { 16 // Expire the dummy cookie if it is added successfully. 17 document.cookie = dummyCookie + '; Max-Age=0'; 18 return true; 19 } 20 return false; 21 } 22 23 /** 24 * Add the given cookie to the current web page. 25 * 26 * If path is not specified, default to '/'. 27 * If domain is not specified, default to document.domain, otherwise remove its 28 * port number. 29 * 30 * Validate name, value, domain and path of the cookie in the same way as the 31 * method CanonicalCookie::Create in src/net/cookies/canonical_cookie.cc. Besides 32 * the following requirements, name, value, domain and path of the cookie should 33 * not start or end with ' ' or '\t', and should not contain '\n', '\r', or '\0'. 34 * <ul> 35 * <li>name: no ';' or '=' 36 * <li>value: no ';' 37 * <li>path: starts with '/', no ';' 38 * </ul> 39 * 40 * @param {!Object} cookie An object representing a Cookie JSON Object as 41 * specified in https://code.google.com/p/selenium/wiki/JsonWireProtocol. 42 */ 43 function addCookie(cookie) { 44 function isNameInvalid(value) { 45 return /(^[ \t])|([;=\n\r\0])|([ \t]$)/.test(value); 46 } 47 function isValueInvalid(value) { 48 return /(^[ \t])|([;\n\r\0])|([ \t]$)/.test(value); 49 } 50 function isPathInvalid(path) { 51 return path[0] != '/' || /([;\n\r\0])|([ \t]$)/.test(path); 52 } 53 54 var name = cookie['name']; 55 if (!name || isNameInvalid(name)) 56 throw new Error('name of cookie is missing or invalid:"' + name + '"'); 57 58 var value = cookie['value'] || ''; 59 if (isValueInvalid(value)) 60 throw new Error('value of cookie is invalid:"' + value + '"'); 61 62 var domain = cookie['domain']; 63 // Remove the port number from domain. 64 if (domain) { 65 var domain_parts = domain.split(':'); 66 if (domain_parts.length > 2) 67 throw new Error('domain of cookie has too many colons'); 68 else if (domain_parts.length == 2) 69 domain = domain_parts[0]; 70 } 71 // Validate domain. 72 if (domain && (isValueInvalid(domain) || !isDomainValid(domain))) { 73 var error = new Error(); 74 error.code = 24; // Error code for InvalidCookieDomain. 75 error.message = 'invalid domain:"' + domain + '"'; 76 throw error; 77 } 78 79 var path = cookie['path']; 80 if (path && isPathInvalid(path)) 81 throw new Error('path of cookie is invalid:"' + path + '"'); 82 83 var newCookie = name + '=' + value; 84 newCookie += '; path=' + (path || '/'); 85 newCookie += '; domain=' + (domain || document.domain); 86 if (cookie['expiry']) { 87 var expiredDate = new Date(cookie['expiry'] * 1000); 88 newCookie += '; expires=' + expiredDate.toUTCString(); 89 } 90 if (cookie['secure']) 91 newCookie += '; secure'; 92 93 document.cookie = newCookie; 94 } 95