mudhorn/n8n
1
0
Fork 0
mirror of https://github.com/n8n-io/n8n.git synced 2024-11-12 15:44:06 -08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix(SSH Node): Cleanup temporary binary files as soon as possible (#11305)

Browse source
This commit is contained in:
कारतोफ्फेलस्क्रिप्ट™ 2024-10-22 15:41:41 +02:00 committed by GitHub
parent 8404282046
commit 08a7b5b742
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 47 additions and 53 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

100
packages/nodes-base/nodes/Ssh/Ssh.node.ts
View file

@ -1,4 +1,4 @@
import { rm, writeFile } from 'fs/promises';
import { writeFile } from 'fs/promises';
import type { Readable } from 'stream';
import type {
ICredentialTestFunctions,
@ -285,8 +285,6 @@ export class Ssh implements INodeType {
): Promise<INodeCredentialTestResult> {
const credentials = credential.data as IDataObject;
const ssh = new NodeSSH();
const temporaryFiles: string[] = [];
try {
if (!credentials.privateKey) {
await ssh.connect({
@ -317,7 +315,6 @@ export class Ssh implements INodeType {
};
} finally {
ssh.dispose();
for (const tempFile of temporaryFiles) await rm(tempFile);
}
return {
status: 'OK',
@ -336,8 +333,6 @@ export class Ssh implements INodeType {
const operation = this.getNodeParameter('operation', 0);
const authentication = this.getNodeParameter('authentication', 0) as string;
const temporaryFiles: string[] = [];
const ssh = new NodeSSH();
try {
@ -396,33 +391,35 @@ export class Ssh implements INodeType {
i,
);
const { path } = await tmpFile({ prefix: 'n8n-ssh-' });
temporaryFiles.push(path);
const binaryFile = await tmpFile({ prefix: 'n8n-ssh-' });
try {
await ssh.getFile(binaryFile.path, parameterPath);
await ssh.getFile(path, parameterPath);
const newItem: INodeExecutionData = {
json: items[i].json,
binary: {},
pairedItem: {
item: i,
},
};
const newItem: INodeExecutionData = {
json: items[i].json,
binary: {},
pairedItem: {
item: i,
},
};
if (items[i].binary !== undefined && newItem.binary) {
// Create a shallow copy of the binary data so that the old
// data references which do not get changed still stay behind
// but the incoming data does not get changed.
Object.assign(newItem.binary, items[i].binary);
}
if (items[i].binary !== undefined && newItem.binary) {
// Create a shallow copy of the binary data so that the old
// data references which do not get changed still stay behind
// but the incoming data does not get changed.
Object.assign(newItem.binary, items[i].binary);
items[i] = newItem;
const fileName = this.getNodeParameter('options.fileName', i, '') as string;
items[i].binary![dataPropertyNameDownload] = await this.nodeHelpers.copyBinaryFile(
binaryFile.path,
fileName || parameterPath,
);
} finally {
await binaryFile.cleanup();
}
items[i] = newItem;
const fileName = this.getNodeParameter('options.fileName', i, '') as string;
items[i].binary![dataPropertyNameDownload] = await this.nodeHelpers.copyBinaryFile(
path,
fileName || parameterPath,
);
}
if (operation === 'upload') {
@ -444,25 +441,28 @@ export class Ssh implements INodeType {
uploadData = Buffer.from(binaryData.data, BINARY_ENCODING);
}
const { path } = await tmpFile({ prefix: 'n8n-ssh-' });
temporaryFiles.push(path);
await writeFile(path, uploadData);
const binaryFile = await tmpFile({ prefix: 'n8n-ssh-' });
try {
await writeFile(binaryFile.path, uploadData);
await ssh.putFile(
path,
`${parameterPath}${
parameterPath.charAt(parameterPath.length - 1) === '/' ? '' : '/'
}${fileName || binaryData.fileName}`,
);
await ssh.putFile(
binaryFile.path,
`${parameterPath}${
parameterPath.charAt(parameterPath.length - 1) === '/' ? '' : '/'
}${fileName || binaryData.fileName}`,
);
returnItems.push({
json: {
success: true,
},
pairedItem: {
item: i,
},
});
returnItems.push({
json: {
success: true,
},
pairedItem: {
item: i,
},
});
} finally {
await binaryFile.cleanup();
}
}
}
} catch (error) {
@ -488,16 +488,10 @@ export class Ssh implements INodeType {
throw error;
}
}
} catch (error) {
} finally {
ssh.dispose();
for (const tempFile of temporaryFiles) await rm(tempFile);
throw error;
}
for (const tempFile of temporaryFiles) await rm(tempFile);
ssh.dispose();
if (resource === 'file' && operation === 'download') {
// For file downloads the files get attached to the existing items
return [items];