minor: allow passing custom assets to start SB protected CM4

.gitignore

@@ -1,3 +1,5 @@
 build
 *~
 node_modules
+yalc.lock
+.yalc

examples/usbboot.ts

@@ -20,12 +20,47 @@ import { scanner, sourceDestination } from '../lib/';
 
 import { pipeSourceToDestinationsWithProgressBar } from './utils';
 
+// Parse command line arguments
+const args = process.argv.slice(2); // removes 'node' and the script name from the args
+const flags: any = {};
+
+args.forEach((arg: string, index: number) => {
+	// Check if the argument is a flag in the format --flag=value
+	if (arg.startsWith('--')) {
+		const key: string = arg.substring(2);
+		const value: string = args[index + 1];
+		flags[key] = value;
+	}
+});
+
+if (!flags.source) {
+	console.log("No source has been provided, won't try to flash anything");
+}
+
+if (flags.bootImageFolder !== '') {
+	console.log(`Using external folder ${flags['bootImageFolder']}`);
+}
+
+if (flags.help) {
+	console.log(
+		'Usage: ts-node usbboot.js --bootImageFolder <bootImageFolder> --source <image>',
+	);
+	console.log(
+		'Beware, `source` image will be flashed to all USBboot devices, so make sure you know what you are doing',
+	);
+	console.log(
+		'To unlock a secureboot CM4, set the bootImageFodler to the folder containing a signed boot-image and config.txt',
+	);
+	process.exit(0);
+}
+
 async function main() {
+	const bootImageFolder = flags.bootImageFolder;
 	const adapters: scanner.adapters.Adapter[] = [
 		new scanner.adapters.BlockDeviceAdapter({
 			includeSystemDrives: () => false,
 		}),
-		new scanner.adapters.UsbbootDeviceAdapter(),
+		new scanner.adapters.UsbbootDeviceAdapter(bootImageFolder),
 	];
 	const deviceScanner = new scanner.Scanner(adapters);
 	console.log('Waiting for one compute module');
@@ -66,7 +101,9 @@ async function main() {
 	});
 	progressBar.terminate();
 	computeModule.removeListener('progress', onProgress);
+
 	console.log('Waiting for compute module to reattach as a block device');
+
 	const dest = await new Promise(
 		(resolve: (drive: sourceDestination.BlockDevice) => void, reject) => {
 			function onAttach(drive: scanner.adapters.AdapterSourceDestination) {
@@ -84,11 +121,11 @@ async function main() {
 	);
 	deviceScanner.stop();
 
-	if (argv.length >= 3) {
+	if (flags.source) {
 		console.log(`Writing image ${argv[2]}`);
 		const source: sourceDestination.SourceDestination =
 			new sourceDestination.File({
-				path: argv[2],
+				path: flags.source,
 			});
 		void pipeSourceToDestinationsWithProgressBar({
 			source,

lib/scanner/adapters/usbboot.ts

@@ -27,11 +27,11 @@ export class UsbbootDeviceAdapter extends Adapter {
 	private drives: Map<UsbbootDevice, UsbbootDrive> = new Map();
 	private scanner?: UsbbootScannerType;
 
-	constructor() {
+	constructor(usbBootExtraFolder?: string | undefined) {
 		super();
 		const rpiUsbboot = getRaspberrypiUsbboot();
 		if (rpiUsbboot !== undefined) {
-			this.scanner = new rpiUsbboot.UsbbootScanner();
+			this.scanner = new rpiUsbboot.UsbbootScanner(usbBootExtraFolder);
 			this.scanner.on('attach', this.onAttach.bind(this));
 			this.scanner.on('detach', this.onDetach.bind(this));
 			this.scanner.on('ready', this.emit.bind(this, 'ready'));