Files
mongo/jstests/ssl_linear/mongo_uri_secondaries.js
wbradmoore dc13be1be8 SERVER-87181: Move ssl_linear suite of tests to use trusted-ca.pem (#19388)
GitOrigin-RevId: cac2af321d8ebac7d7014c84e537b6fd85de6268
2024-03-04 16:33:45 +00:00

83 lines
2.9 KiB
JavaScript

// On OSX this test assumes that jstests/libs/trusted-ca.pem has been added as a trusted
// certificate to the login keychain of the evergreen user. See,
// https://github.com/10gen/buildslave-cookbooks/commit/af7cabe5b6e0885902ebd4902f7f974b64cc8961
// for details.
// To install trusted-ca.pem for local testing on OSX, invoke the following at a console:
// security add-trusted-cert -d jstests/libs/trusted-ca.pem
const HOST_TYPE = getBuildInfo().buildEnvironment.target_os;
if (HOST_TYPE == "windows") {
// OpenSSL backed imports Root CA and intermediate CA
runProgram("certutil.exe", "-addstore", "-user", "-f", "CA", "jstests\\libs\\trusted-ca.pem");
// SChannel backed follows Windows rules and only trusts the Root store in Local Machine and
// Current User.
runProgram("certutil.exe", "-addstore", "-f", "Root", "jstests\\libs\\trusted-ca.pem");
}
try {
const x509Options = {
tlsMode: 'requireTLS',
tlsCertificateKeyFile: 'jstests/libs/trusted-server.pem',
tlsCAFile: 'jstests/libs/trusted-ca.pem',
tlsAllowInvalidCertificates: '',
tlsWeakCertificateValidation: '',
};
const rst = new ReplSetTest({
nodes: 2,
name: "tlsSet",
useHostName: false,
nodeOptions: x509Options,
waitForKeys: false
});
rst.startSet();
rst.initiate();
const subShellCommand = function(hosts) {
var Ms = [];
for (let i = 0; i < 10; i++) {
Ms.push(new Mongo("mongodb://" + hosts[0] + "," + hosts[1] +
"/?ssl=true&replicaSet=tlsSet"));
}
for (let i = 0; i < 10; i++) {
var db = Ms[i].getDB("test");
db.setSecondaryOk();
db.col.find().readPref("secondary").toArray();
}
};
const subShellCommandFormatter = function(replSet) {
var hosts = [];
replSet.nodes.forEach((node) => {
hosts.push("localhost:" + node.port);
});
let command = `
(function () {
let command = ${subShellCommand.toString()};
let hosts = ${tojson(hosts)};
command(hosts);
}());`;
return command;
};
function runWithEnv(args, env) {
const pid = _startMongoProgram({args: args, env: env});
return waitProgram(pid);
}
const subShellArgs = ['mongo', '--nodb', '--eval', subShellCommandFormatter(rst)];
const retVal = runWithEnv(subShellArgs, {"SSL_CERT_FILE": "jstests/libs/trusted-ca.pem"});
assert.eq(retVal, 0, 'mongo shell did not succeed with exit code 0');
rst.stopSet();
} finally {
if (HOST_TYPE == "windows") {
const trusted_ca_thumbprint = cat('jstests/libs/trusted-ca.pem.digest.sha1');
runProgram("certutil.exe", "-delstore", "-f", "Root", trusted_ca_thumbprint);
runProgram("certutil.exe", "-delstore", "-user", "-f", "CA", trusted_ca_thumbprint);
}
}