websocket-cookies-helper.sub.js (1939B)
1 // Set up global variables. 2 (_ => { 3 var HOST = '{{host}}'; 4 var CROSS_ORIGIN_HOST = '{{hosts[alt][]}}'; 5 var WSS_PORT = ':{{ports[wss][0]}}'; 6 var HTTPS_PORT = ':{{ports[https][0]}}'; 7 8 window.WSS_ORIGIN = 'wss://' + HOST + WSS_PORT; 9 window.WSS_CROSS_SITE_ORIGIN = 'wss://' + CROSS_ORIGIN_HOST + WSS_PORT; 10 window.HTTPS_ORIGIN = 'https://' + HOST + HTTPS_PORT; 11 window.HTTPS_CROSS_SITE_ORIGIN = 'https://' + CROSS_ORIGIN_HOST + HTTPS_PORT; 12 })(); 13 14 // Sets a cookie with each SameSite option. 15 function setSameSiteCookies(origin, value) { 16 return new Promise(resolve => { 17 const ws = new WebSocket(origin + '/set-cookies-samesite?value=' + value); 18 ws.onopen = () => { 19 ws.close(); 20 }; 21 ws.onclose = resolve; 22 }); 23 } 24 25 // Clears cookies set by setSameSiteCookies(). 26 function clearSameSiteCookies(origin) { 27 return new Promise(resolve => { 28 const ws = new WebSocket(origin + '/set-cookies-samesite?clear'); 29 ws.onopen = () => ws.close(); 30 ws.onclose = resolve; 31 }); 32 } 33 34 // Gets value of Cookie header sent in request. 35 function connectAndGetRequestCookiesFrom(origin) { 36 return new Promise((resolve, reject) => { 37 var ws = new WebSocket(origin + '/echo-cookie'); 38 ws.onmessage = evt => { 39 var cookies = evt.data 40 resolve(cookies); 41 ws.onerror = undefined; 42 ws.onclose = undefined; 43 }; 44 ws.onerror = () => reject('Unexpected error event'); 45 ws.onclose = evt => reject('Unexpected close event: ' + JSON.stringify(evt)); 46 }); 47 } 48 49 // Assert that a given cookie is or is not present in the string |cookies|. 50 function assertCookie(cookies, name, value, present) { 51 var assertion = present ? assert_true : assert_false; 52 var description = name + '=' + value + ' cookie is' + 53 (present ? ' ' : ' not ') + 'present.'; 54 var re = new RegExp('(?:^|; )' + name + '=' + value + '(?:$|;)'); 55 assertion(re.test(cookies), description); 56 }