From 1c4a2b71e8dd34a302064fed541d40262b75fa0c Mon Sep 17 00:00:00 2001
From: Alexandru <alexandru@balena.io>
Date: Tue, 9 Jul 2024 11:11:32 +0300
Subject: [PATCH] flashing/index: Power off Jetson DUT on container flash
 failure

This should allow the DUT to cool off while not in use, because in rcm-boot mode the fan is not spinning.

Also, with this commit we check if the following sporadic error observed on J3010 disappears:

  ECID is
  Board ID() version() sku() revision()
  Chip SKU(00:00:00:D3) ramcode() fuselevel(fuselevel_production)   board_FAB()
  emc_opt_disable_fuse:(0)
  Error: Unrecognized module SKU

Change-type: patch
Signed-off-by: Alexandru Costache <alexandru@balena.io
---
 lib/flashing/index.ts | 31 +++++++++++++++++++++----------
 1 file changed, 21 insertions(+), 10 deletions(-)

diff --git a/lib/flashing/index.ts b/lib/flashing/index.ts
index 08db410..9fffb35 100644
--- a/lib/flashing/index.ts
+++ b/lib/flashing/index.ts
@@ -429,11 +429,13 @@ async function flashJetson(filename: string, autoKit: Autokit, deviceType: strin
                     if (code === 0) {
                         resolve();
                     } else {
-                        reject(new Error(`Docker build exit code was: ${code}`));
+                        autoKit.power.off();
+                        reject(new Error(`Docker build exit code was: ${code}. Powered off DUT.`));
                     }
                 });
                 build.on('error', (err) => {
-                    reject(new Error(`Error with docker build: ${err.message}`));
+                    autoKit.power.off();
+                    reject(new Error(`Error with docker build: ${err.message}. Powered off DUT.`));
                 });
             });
         } else {
@@ -454,19 +456,24 @@ async function flashJetson(filename: string, autoKit: Autokit, deviceType: strin
                     if (code === 0) {
                         resolve();
                     } else {
-                        console.log(`Got exit code ${code}.. error`);
+                        autoKit.power.off();
+                        console.log(`Got exit code ${code}.. error. Powered off DUT.`);
                         reject(new Error(`Docker build exit code was: ${code}`));
                     }
                 });
                 buildAndRun.on('error', (err) => {
-                    reject(new Error(`Error with docker build: ${err.message}`));
+                    autoKit.power.off();
+                    reject(new Error(`Error with docker build: ${err.message}. Powered off DUT.`));
                 });
             });
         }
     } catch (err: any){
-        throw new Error(`Failed during jetson-flash container build: ${err.message}`);
+        await autoKit.power.off();
+        throw new Error(`Failed during jetson-flash container build: ${err.message}. Powered off DUT`);
     }
 
+    await autoKit.power.on();
+    await delay(5 * 1000);
     // run flash container
     console.log(`File path: ${filename}`)
     // Then run container
@@ -498,16 +505,19 @@ async function flashJetson(filename: string, autoKit: Autokit, deviceType: strin
                 if (code === 0) {
                     resolve();
                 } else {
-                    console.log(`Got exit code ${code}.. error`);
+                    autoKit.power.off();
+                    console.log(`Got exit code ${code}.. error. Powered off DUT.`);
                     reject(new Error(`Docker run exit code was: ${code}`))
                 }
             });
             flash.on('error', (err) => {
-                reject(new Error(`Error with docker run: ${err.message}`));
+                autoKit.power.off();
+                reject(new Error(`Error with docker run: ${err.message}. Powered off DUT.`));
             });
         });
     } catch(err:any){
-        throw new Error(`Failed during jetson-flash container run: ${err.message}`)
+        await autoKit.power.off();
+        throw new Error(`Failed during jetson-flash container run: ${err.message}. Powered off DUT.`)
     }
 
 
@@ -544,7 +554,8 @@ async function flashJetson(filename: string, autoKit: Autokit, deviceType: strin
             }
             attempt += 1;
             if (attempt === TIMEOUT_COUNT){
-                throw new Error(`Timed out while trying to flash internal storage!!`)
+                await autoKit.power.off();
+                throw new Error(`Timed out while trying to flash internal storage!!. Powered off DUT.`)
             }
         }
         console.log('Internally flashed - powering off DUT');
@@ -669,4 +680,4 @@ async function flash(filename: string, deviceType: string, autoKit: Autokit, por
     }
 }
 
-export { flash }
\ No newline at end of file
+export { flash }