refactor: Simplify resolveIcon checks and add tests (#11165)

packages/cli/src/__tests__/load-nodes-and-credentials.test.ts

@@ -0,0 +1,37 @@
+import { mock } from 'jest-mock-extended';
+import type { DirectoryLoader } from 'n8n-core';
+
+import { LoadNodesAndCredentials } from '../load-nodes-and-credentials';
+
+describe('LoadNodesAndCredentials', () => {
+	describe('resolveIcon', () => {
+		let instance: LoadNodesAndCredentials;
+
+		beforeEach(() => {
+			instance = new LoadNodesAndCredentials(mock(), mock(), mock());
+			instance.loaders.package1 = mock<DirectoryLoader>({
+				directory: '/icons/package1',
+			});
+		});
+
+		it('should return undefined if the loader for the package is not found', () => {
+			const result = instance.resolveIcon('unknownPackage', '/icons/unknownPackage/icon.png');
+			expect(result).toBeUndefined();
+		});
+
+		it('should return undefined if the resolved file path is outside the loader directory', () => {
+			const result = instance.resolveIcon('package1', '/some/other/path/icon.png');
+			expect(result).toBeUndefined();
+		});
+
+		it('should return the file path if the file is within the loader directory', () => {
+			const result = instance.resolveIcon('package1', '/icons/package1/icon.png');
+			expect(result).toBe('/icons/package1/icon.png');
+		});
+
+		it('should return undefined if the URL is outside the package directory', () => {
+			const result = instance.resolveIcon('package1', '/icons/package1/../../../etc/passwd');
+			expect(result).toBeUndefined();
+		});
+	});
+});

packages/cli/src/load-nodes-and-credentials.ts

@@ -29,6 +29,7 @@ import {
 	inE2ETests,
 } from '@/constants';
 import { Logger } from '@/logging/logger.service';
+import { isContainedWithin } from '@/utils/path-util';
 
 interface LoadedNodesAndCredentials {
 	nodes: INodeTypeData;
@@ -155,14 +156,13 @@ export class LoadNodesAndCredentials {
 
 	resolveIcon(packageName: string, url: string): string | undefined {
 		const loader = this.loaders[packageName];
-		if (loader) {
+		if (!loader) {
-			const pathPrefix = `/icons/${packageName}/`;
+			return undefined;
-			const filePath = path.resolve(loader.directory, url.substring(pathPrefix.length));
-			if (!path.relative(loader.directory, filePath).includes('..')) {
-				return filePath;
-			}
 		}
-		return undefined;
+		const pathPrefix = `/icons/${packageName}/`;
+		const filePath = path.resolve(loader.directory, url.substring(pathPrefix.length));
+
+		return isContainedWithin(loader.directory, filePath) ? filePath : undefined;
 	}
 
 	getCustomDirectories(): string[] {