mirror of
https://github.com/AikidoSec/safe-chain.git
synced 2026-05-26 12:10:49 +00:00
Simplify setting certificates
This commit is contained in:
parent
b886bb1cfe
commit
70dc89c3e8
4 changed files with 77 additions and 16 deletions
|
|
@ -1,13 +1,20 @@
|
|||
import { ui } from "../../environment/userInteraction.js";
|
||||
import { safeSpawn } from "../../utils/safeSpawn.js";
|
||||
import { mergeSafeChainProxyEnvironmentVariables } from "../../registryProxy/registryProxy.js";
|
||||
|
||||
import { getCaCertPath } from "../../registryProxy/certUtils.js";
|
||||
|
||||
export async function runPip(command, args) {
|
||||
try {
|
||||
const result = await safeSpawn(command, args, {
|
||||
const env = mergeSafeChainProxyEnvironmentVariables(process.env);
|
||||
|
||||
// Pass --cert with our CA to pip so it trusts our MITM for known registries.
|
||||
// pip will append this to its default CA bundle, so it still validates
|
||||
// non-registry HTTPS (GitHub, custom mirrors) against system CAs.
|
||||
const finalArgs = [...args, "--cert", getCaCertPath()];
|
||||
|
||||
const result = await safeSpawn(command, finalArgs, {
|
||||
stdio: "inherit",
|
||||
env: mergeSafeChainProxyEnvironmentVariables(process.env),
|
||||
env,
|
||||
});
|
||||
return { status: result.status };
|
||||
} catch (error) {
|
||||
|
|
|
|||
|
|
@ -0,0 +1,63 @@
|
|||
import { describe, it, beforeEach, afterEach, mock } from "node:test";
|
||||
import assert from "node:assert";
|
||||
|
||||
describe("runPipCommand --cert handling", () => {
|
||||
let runPip;
|
||||
let capturedArgs = null;
|
||||
|
||||
beforeEach(async () => {
|
||||
capturedArgs = null;
|
||||
|
||||
// Mock safeSpawn to capture args
|
||||
mock.module("../../utils/safeSpawn.js", {
|
||||
namedExports: {
|
||||
safeSpawn: async (command, args, options) => {
|
||||
capturedArgs = { command, args, options };
|
||||
return { status: 0 };
|
||||
},
|
||||
},
|
||||
});
|
||||
|
||||
// Mock proxy env merge
|
||||
mock.module("../../registryProxy/registryProxy.js", {
|
||||
namedExports: {
|
||||
mergeSafeChainProxyEnvironmentVariables: (env) => ({
|
||||
...env,
|
||||
HTTPS_PROXY: "http://localhost:8080",
|
||||
}),
|
||||
},
|
||||
});
|
||||
|
||||
// Mock certUtils to point to a test CA path
|
||||
mock.module("../../registryProxy/certUtils.js", {
|
||||
namedExports: {
|
||||
getCaCertPath: () => "/tmp/test-ca.pem",
|
||||
},
|
||||
});
|
||||
|
||||
const mod = await import("./runPipCommand.js");
|
||||
runPip = mod.runPip;
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
mock.reset();
|
||||
});
|
||||
|
||||
it("should append --cert with our CA path to pip args", async () => {
|
||||
const res = await runPip("pip3", ["install", "requests"]);
|
||||
assert.strictEqual(res.status, 0);
|
||||
|
||||
// safeSpawn should be called with --cert flag
|
||||
assert.ok(capturedArgs, "safeSpawn should have been called");
|
||||
|
||||
const idx = capturedArgs.args.indexOf("--cert");
|
||||
assert.ok(idx >= 0, "--cert flag should be present in pip args");
|
||||
|
||||
const certPath = capturedArgs.args[idx + 1];
|
||||
assert.strictEqual(certPath, "/tmp/test-ca.pem", "CA path should match getCaCertPath()");
|
||||
|
||||
// Original args should be preserved before --cert
|
||||
assert.strictEqual(capturedArgs.args[0], "install");
|
||||
assert.strictEqual(capturedArgs.args[1], "requests");
|
||||
});
|
||||
});
|
||||
|
|
@ -34,14 +34,6 @@ function getSafeChainProxyEnvironmentVariables() {
|
|||
HTTPS_PROXY: `http://localhost:${state.port}`,
|
||||
GLOBAL_AGENT_HTTP_PROXY: `http://localhost:${state.port}`,
|
||||
NODE_EXTRA_CA_CERTS: getCaCertPath(),
|
||||
|
||||
// Following env vars point pip and Python's requests/urllib at a CA Cert file.
|
||||
// pip checks PIP_CERT first
|
||||
// If pip uses requests library internally, it needs REQUESTS_CA_BUNDLE
|
||||
// Other Python packages or pip's fallback SSL code may use SSL_CERT_FILE
|
||||
PIP_CERT: getCaCertPath(),
|
||||
REQUESTS_CA_BUNDLE: getCaCertPath(),
|
||||
SSL_CERT_FILE: getCaCertPath(),
|
||||
};
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -145,12 +145,11 @@ describe("registryProxy.mitm", () => {
|
|||
});
|
||||
|
||||
// --- Pip registry MITM and env var tests ---
|
||||
it("should set pip CA trust environment variables", () => {
|
||||
it("should NOT set global Python CA environment variables", () => {
|
||||
const envVars = mergeSafeChainProxyEnvironmentVariables([]);
|
||||
const caPath = getCaCertPath();
|
||||
assert.strictEqual(envVars.PIP_CERT, caPath);
|
||||
assert.strictEqual(envVars.REQUESTS_CA_BUNDLE, caPath);
|
||||
assert.strictEqual(envVars.SSL_CERT_FILE, caPath);
|
||||
assert.strictEqual(envVars.PIP_CERT, undefined);
|
||||
assert.strictEqual(envVars.REQUESTS_CA_BUNDLE, undefined);
|
||||
assert.strictEqual(envVars.SSL_CERT_FILE, undefined);
|
||||
});
|
||||
|
||||
it("should intercept HTTPS requests to pypi.org for pip package", async () => {
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue