extend `ncli` exception handling (#5905)

Make raised exceptions explicit in `ncli_common.nim`, and handle more of
them in `ncli_db.nim` to have better UX when directories cannot be read
or file names do not parse against the expected format.

ncli/ncli_common.nim

@@ -5,6 +5,8 @@
 #   * Apache v2 license (license terms in the root directory or at https://www.apache.org/licenses/LICENSE-2.0).
 # at your option. This file may not be copied, modified, or distributed except according to those terms.
 
+{.push raises: [].}
+
 import
   std/[os, strutils],
   stew/bitops2,
@@ -124,8 +126,9 @@ static:
       "15227487_86601706.echop"]: # Wrong extension
     doAssert not filename.matchFilenameAggregatedFiles
 
-proc getUnaggregatedFilesEpochRange*(dir: string):
+proc getUnaggregatedFilesEpochRange*(
-    tuple[firstEpoch, lastEpoch: Epoch] =
+    dir: string
+): tuple[firstEpoch, lastEpoch: Epoch] {.raises: [OSError, ValueError].} =
   var smallestEpochFileName =
     '9'.repeat(epochInfoFileNameDigitsCount) & epochFileNameExtension
   var largestEpochFileName =
@@ -141,10 +144,12 @@ proc getUnaggregatedFilesEpochRange*(dir: string):
   result.lastEpoch = parseUInt(
     largestEpochFileName[0 ..< epochInfoFileNameDigitsCount]).Epoch
 
-proc getUnaggregatedFilesLastEpoch*(dir: string): Epoch =
+proc getUnaggregatedFilesLastEpoch*(
+    dir: string): Epoch {.raises: [OSError, ValueError].} =
   dir.getUnaggregatedFilesEpochRange.lastEpoch
 
-proc getAggregatedFilesLastEpoch*(dir: string): Epoch =
+proc getAggregatedFilesLastEpoch*(
+    dir: string): Epoch {.raises: [OSError, ValueError].}=
   var largestEpochInFileName = 0'u
   for (_, fn) in walkDir(dir.string, relative = true):
     if fn.matchFilenameAggregatedFiles:
@@ -403,9 +408,15 @@ proc collectFromDeposits(
       var index = findValidatorIndex(forkyState.data, pubkey)
       if index.isNone:
         if pubkey in pubkeyToIndex:
+          try:
             index = Opt[ValidatorIndex].ok(pubkeyToIndex[pubkey])
+          except KeyError as e:
+            raiseAssert "pubkey was checked to exist: " & e.msg
       if index.isSome:
+        try:
           rewardsAndPenalties[index.get()].deposits += amount
+        except KeyError as e:
+          raiseAssert "rewardsAndPenalties lacks expected index " & $index.get()
       elif verify_deposit_signature(cfg, deposit.data):
         pubkeyToIndex[pubkey] = ValidatorIndex(rewardsAndPenalties.len)
         rewardsAndPenalties.add(

ncli/ncli_db.nim

@@ -5,8 +5,10 @@
 #   * Apache v2 license (license terms in the root directory or at https://www.apache.org/licenses/LICENSE-2.0).
 # at your option. This file may not be copied, modified, or distributed except according to those terms.
 
+{.push raises: [].}
+
 import
-  std/[os, stats, strformat, tables],
+  std/[os, stats, tables],
   snappy,
   chronicles, confutils, stew/[byteutils, io2], eth/db/kvstore_sqlite3,
   ../beacon_chain/networking/network_metadata,
@@ -240,7 +242,8 @@ proc cmdBench(conf: DbConf, cfg: RuntimeConfig) =
       seq[capella.TrustedSignedBeaconBlock],
       seq[deneb.TrustedSignedBeaconBlock])
 
-  echo &"Loaded head slot {dag.head.slot}, selected {blockRefs.len} blocks"
+  echo "Loaded head slot ", dag.head.slot,
+    " selected ", blockRefs.len, " blocks"
   doAssert blockRefs.len() > 0, "Must select at least one block"
 
   for b in 0 ..< blockRefs.len:
@@ -390,8 +393,18 @@ proc cmdPutState(conf: DbConf, cfg: RuntimeConfig) =
 
   for file in conf.stateFile:
     if shouldShutDown: quit QuitSuccess
-    let state = newClone(readSszForkedHashedBeaconState(
+
+    let state =
+      try:
+        newClone(readSszForkedHashedBeaconState(
           cfg, readAllBytes(file).tryGet()))
+      except ResultError[IoErrorCode] as e:
+        echo "Couldn't load ", file, ": ", e.msg
+        continue
+      except SerializationError as e:
+        echo "Malformed ", file, ": ", e.msg
+        continue
+
     withState(state[]):
       db.putState(forkyState)
 
@@ -427,8 +440,16 @@ proc cmdPutBlock(conf: DbConf, cfg: RuntimeConfig) =
   for file in conf.blckFile:
     if shouldShutDown: quit QuitSuccess
 
-    let blck = readSszForkedSignedBeaconBlock(
+    let blck =
+      try:
+        readSszForkedSignedBeaconBlock(
           cfg, readAllBytes(file).tryGet())
+      except ResultError[IoErrorCode] as e:
+        echo "Couldn't load ", file, ": ", e.msg
+        continue
+      except SerializationError as e:
+        echo "Malformed ", file, ": ", e.msg
+        continue
 
     withBlck(blck.asTrusted()):
       db.putBlock(forkyBlck)
@@ -576,7 +597,13 @@ proc cmdExportEra(conf: DbConf, cfg: RuntimeConfig) =
       try:
         moveFile(tmpName, name)
       except IOError as e:
-        warn "Failed to rename era file to its final name",
+        warn "Failed to rename era file to its final name (IOError)",
+          name, tmpName, error = e.msg
+      except OSError as e:
+        warn "Failed to rename era file to its final name (OSError)",
+          name, tmpName, error = e.msg
+      except Exception as e:
+        warn "Failed to rename era file to its final name (Exception)",
           name, tmpName, error = e.msg
     else:
       if (let e = io2.removeFile(name); e.isErr):
@@ -915,6 +942,7 @@ proc cmdValidatorDb(conf: DbConf, cfg: RuntimeConfig) =
       if conf.startEpoch.isSome:
         Epoch(conf.startEpoch.get)
       else:
+        try:
           let unaggregatedFilesNextEpoch = getUnaggregatedFilesLastEpoch(
             unaggregatedFilesOutputDir) + 1
           let aggregatedFilesNextEpoch = getAggregatedFilesLastEpoch(
@@ -927,6 +955,12 @@ proc cmdValidatorDb(conf: DbConf, cfg: RuntimeConfig) =
             aggregatedFilesNextEpoch
           else:
             min(unaggregatedFilesNextEpoch, aggregatedFilesNextEpoch)
+        except OSError as e:
+          fatal "Failed to iterate epoch files", e = e.msg
+          quit QuitFailure
+        except ValueError as e:
+          fatal "Failed to parse epoch file name", e = e.msg
+          quit QuitFailure
     endEpoch =
       if conf.endEpoch.isSome:
         Epoch(conf.endEpoch.get)
@@ -947,10 +981,28 @@ proc cmdValidatorDb(conf: DbConf, cfg: RuntimeConfig) =
     blockRefs = dag.getBlockRange(startSlot, endSlot)
 
   if not unaggregatedFilesOutputDir.dirExists:
-    unaggregatedFilesOutputDir.createDir
+    try:
+      unaggregatedFilesOutputDir.createDir()
+    except IOError as e:
+      fatal "Failed to create unaggregated files directory (IOError)",
+        dirName = unaggregatedFilesOutputDir, e = e.msg
+      quit QuitFailure
+    except OSError as e:
+      fatal "Failed to create unaggregated files directory (OSError)",
+        dirName = unaggregatedFilesOutputDir, e = e.msg
+      quit QuitFailure
 
   if not aggregatedFilesOutputDir.dirExists:
-    aggregatedFilesOutputDir.createDir
+    try:
+      aggregatedFilesOutputDir.createDir()
+    except IOError as e:
+      fatal "Failed to create aggregated files directory (IOError)",
+        dirName = aggregatedFilesOutputDir, e = e.msg
+      quit QuitFailure
+    except OSError as e:
+      fatal "Failed to create aggregated files directory (OSError)",
+        dirName = aggregatedFilesOutputDir, e = e.msg
+      quit QuitFailure
 
   let tmpState = newClone(dag.headState)
   var cache = StateCache()