91 lines
2.9 KiB
JavaScript
91 lines
2.9 KiB
JavaScript
Cu.import("resource://testing-common/httpd.js");
|
|
Cu.import("resource://gre/modules/NetUtil.jsm");
|
|
|
|
const sentCookieVal = "foo=bar";
|
|
const responseBody = "response body";
|
|
|
|
XPCOMUtils.defineLazyGetter(this, "baseURL", function() {
|
|
return "http://localhost:" + httpServer.identity.primaryPort;
|
|
});
|
|
|
|
const preRedirectPath = "/528292/pre-redirect";
|
|
|
|
XPCOMUtils.defineLazyGetter(this, "preRedirectURL", function() {
|
|
return baseURL + preRedirectPath;
|
|
});
|
|
|
|
const postRedirectPath = "/528292/post-redirect";
|
|
|
|
XPCOMUtils.defineLazyGetter(this, "postRedirectURL", function() {
|
|
return baseURL + postRedirectPath;
|
|
});
|
|
|
|
var httpServer = null;
|
|
var receivedCookieVal = null;
|
|
|
|
function preRedirectHandler(metadata, response)
|
|
{
|
|
response.setStatusLine(metadata.httpVersion, 302, "Found");
|
|
response.setHeader("Location", postRedirectURL, false);
|
|
return;
|
|
}
|
|
|
|
function postRedirectHandler(metadata, response)
|
|
{
|
|
receivedCookieVal = metadata.getHeader("Cookie");
|
|
response.setHeader("Content-Type", "text/plain");
|
|
response.bodyOutputStream.write(responseBody, responseBody.length);
|
|
}
|
|
|
|
function inChildProcess() {
|
|
return Cc["@mozilla.org/xre/app-info;1"]
|
|
.getService(Ci.nsIXULRuntime)
|
|
.processType != Ci.nsIXULRuntime.PROCESS_TYPE_DEFAULT;
|
|
}
|
|
|
|
function run_test()
|
|
{
|
|
// Start the HTTP server.
|
|
httpServer = new HttpServer();
|
|
httpServer.registerPathHandler(preRedirectPath, preRedirectHandler);
|
|
httpServer.registerPathHandler(postRedirectPath, postRedirectHandler);
|
|
httpServer.start(-1);
|
|
|
|
if (!inChildProcess()) {
|
|
// Disable third-party cookies in general.
|
|
Cc["@mozilla.org/preferences-service;1"].getService(Ci.nsIPrefBranch).
|
|
setIntPref("network.cookie.cookieBehavior", 1);
|
|
}
|
|
|
|
var ioService = Cc["@mozilla.org/network/io-service;1"].
|
|
getService(Ci.nsIIOService);
|
|
|
|
// Set up a channel with forceAllowThirdPartyCookie set to true. We'll use
|
|
// the channel both to set a cookie (since nsICookieService::setCookieString
|
|
// requires such a channel in order to successfully set a cookie) and then
|
|
// to load the pre-redirect URI.
|
|
var chan = NetUtil.newChannel({
|
|
uri: preRedirectURL,
|
|
loadUsingSystemPrincipal: true
|
|
}).QueryInterface(Ci.nsIHttpChannel)
|
|
.QueryInterface(Ci.nsIHttpChannelInternal);
|
|
chan.forceAllowThirdPartyCookie = true;
|
|
|
|
// Set a cookie on one of the URIs. It doesn't matter which one, since
|
|
// they're both from the same host, which is enough for the cookie service
|
|
// to send the cookie with both requests.
|
|
var postRedirectURI = ioService.newURI(postRedirectURL, "", null);
|
|
Cc["@mozilla.org/cookieService;1"].getService(Ci.nsICookieService).
|
|
setCookieString(postRedirectURI, null, sentCookieVal, chan);
|
|
|
|
// Load the pre-redirect URI.
|
|
chan.asyncOpen2(new ChannelListener(finish_test, null));
|
|
do_test_pending();
|
|
}
|
|
|
|
function finish_test(event)
|
|
{
|
|
do_check_eq(receivedCookieVal, sentCookieVal);
|
|
httpServer.stop(do_test_finished);
|
|
}
|