diff --git a/src/buildx/history.ts b/src/buildx/history.ts index a8c1b6ea..44c2acc8 100644 --- a/src/buildx/history.ts +++ b/src/buildx/history.ts @@ -14,10 +14,11 @@ * limitations under the License. */ -import {spawn} from 'child_process'; +import {ChildProcessByStdio, spawn} from 'child_process'; import fs from 'fs'; import os from 'os'; import path from 'path'; +import {Readable, Writable} from 'stream'; import * as core from '@actions/core'; import {Buildx} from './buildx'; @@ -91,23 +92,32 @@ export class History { }); await Exec.exec('mkfifo', [buildxOutFifoPath]); - const buildxCmd = await this.buildx.getCommand(['--builder', builderName, 'dial-stdio']); - - core.info(`[command]${buildxCmd.command} ${buildxCmd.args.join(' ')}`); - const buildxDialStdioProc = spawn(buildxCmd.command, buildxCmd.args, { - stdio: ['pipe', 'pipe', 'inherit'], - detached: true - }); - fs.createReadStream(buildxInFifoPath).pipe(buildxDialStdioProc.stdin); - buildxDialStdioProc.stdout.pipe(fs.createWriteStream(buildxOutFifoPath)); - buildxDialStdioProc.on('exit', code => { - core.info(`Process "buildx dial-stdio" exited with code ${code}`); - }); - const tmpDockerbuildFilename = path.join(outDir, 'rec.dockerbuild'); const summaryFilename = path.join(outDir, 'summary.json'); + let buildxDialStdioProc: ChildProcessByStdio | undefined; + let buildxDialStdioKilled = false; + let dockerRunProc: ChildProcessByStdio | undefined; + let dockerRunProcKilled = false; + + const buildxDialStdioCmd = await this.buildx.getCommand(['--builder', builderName, 'dial-stdio']); await new Promise((resolve, reject) => { + core.info(`[command]${buildxDialStdioCmd.command} ${buildxDialStdioCmd.args.join(' ')}`); + buildxDialStdioProc = spawn(buildxDialStdioCmd.command, buildxDialStdioCmd.args, { + stdio: ['pipe', 'pipe', 'inherit'], + detached: true + }); + fs.createReadStream(buildxInFifoPath).pipe(buildxDialStdioProc.stdin); + buildxDialStdioProc.stdout.pipe(fs.createWriteStream(buildxOutFifoPath)); + buildxDialStdioProc.on('exit', (code, signal) => { + buildxDialStdioKilled = true; + if (signal) { + core.info(`Process "buildx dial-stdio" was killed with signal ${signal}`); + } else { + core.info(`Process "buildx dial-stdio" exited with code ${code}`); + } + }); + const ebargs: Array = ['--ref-state-dir=/buildx-refs', `--node=${builderName}/${nodeName}`]; for (const ref of refs) { ebargs.push(`--ref=${ref}`); @@ -127,7 +137,7 @@ export class History { ...ebargs ] core.info(`[command]docker ${dockerRunArgs.join(' ')}`); - const dockerRunProc = spawn('docker', dockerRunArgs, { + dockerRunProc = spawn('docker', dockerRunArgs, { stdio: ['pipe', 'pipe', 'inherit'] }); fs.createReadStream(buildxOutFifoPath).pipe(dockerRunProc.stdin); @@ -147,12 +157,28 @@ export class History { core.error(`Error executing "docker run": ${err}`); reject(err); }); - dockerRunProc.on('exit', code => { - core.info(`Process "docker run" exited with code ${code}`); + dockerRunProc.on('exit', (code, signal) => { + dockerRunProcKilled = true; + if (signal) { + core.info(`Process "docker run" was killed with signal ${signal}`); + } else { + core.info(`Process "docker run" exited with code ${code}`); + } + }); + }) + .catch(err => { + throw err; + }) + .finally(() => { + if (buildxDialStdioProc && !buildxDialStdioKilled) { + core.debug('Force terminating "buildx dial-stdio" process'); + buildxDialStdioProc.kill('SIGKILL'); + } + if (dockerRunProc && !dockerRunProcKilled) { + core.debug('Force terminating "docker run" process'); + dockerRunProc.kill('SIGKILL'); + } }); - }).catch(err => { - throw err; - }); let dockerbuildFilename = `${GitHub.context.repo.owner}~${GitHub.context.repo.repo}~${refs[0].substring(0, 6).toUpperCase()}`; if (refs.length > 1) {