Fix for when Traceroute is not installed on host

server/lambda/trace-route.js

@@ -1,17 +1,21 @@
 const traceroute = require('traceroute');
-const util = require('util');
 const url = require('url');
 
-// Convert traceroute.trace method to return a Promise
-const traceroutePromise = util.promisify(traceroute.trace);
-
 exports.handler = async function(event, context) {
   const urlString = event.queryStringParameters.url;
   const startTime = Date.now();
 
   try {
     if (!urlString) {
-      throw new Error('URL parameter is missing!');
+
+      return {
+        statusCode: 400,
+        body: JSON.stringify({
+
+
+          error: 'URL parameter is missing!'
+        }),
+      };
     }
 
     // Parse the URL and get the hostname
@@ -19,19 +23,66 @@ exports.handler = async function(event, context) {
     const host = urlObject.hostname;
 
     if (!host) {
-      throw new Error('Invalid URL provided');
+
+      return {
+        statusCode: 400,
+        body: JSON.stringify({
+          error: 'Invalid URL provided'
+        }),
+      };
     }
 
-    const result = await traceroutePromise(host);
+    // Traceroute with callback
+    return await new Promise((resolve, reject) => {
+      traceroute.trace(host, (err, hops) => {
+        if (err) {
+          reject(err);
+        } else if (hops) {
+          resolve(hops);
+        }
+      });
+      // Check if remaining time is less than 8.8 seconds, then reject promise
+      if (context.getRemainingTimeInMillis() < 8800) {
+        reject(new Error('Lambda is about to timeout'));
+      }
+    }).then((result) => {
       const timeTaken = Date.now() - startTime;
+
       return {
         statusCode: 200,
-      body: JSON.stringify({ message: "Traceroute completed!", result, timeTaken }),
+        body: JSON.stringify({
+          message: "Traceroute completed!",
+          result,
+          timeTaken
+        }),
       };
-  } catch (err) {
+    }).catch((err) => {
       return {
         statusCode: 500,
-      body: JSON.stringify({ message: `Error: ${err.message}` }),
+        body: JSON.stringify({
+          error: err.message
+        }),
+      };
+    });
+
+
+  } catch (err) {
+    if (err.code === 'ENOENT') {
+
+      return {
+        statusCode: 400,
+        body: JSON.stringify({
+          error: 'Traceroute command is not installed on the host.'
+        }),
+      };
+    } else {
+
+      return {
+        statusCode: 500,
+        body: JSON.stringify({
+          error: err.message
+        }),
       };
     }
+  }
 };

src/pages/Results.tsx

@@ -9,6 +9,7 @@ import Card from 'components/Form/Card';
 import Modal from 'components/Form/Modal';
 import Footer from 'components/misc/Footer';
 import { RowProps }  from 'components/Form/Row';
+import ErrorBoundary from 'components/misc/ErrorBoundary';
 import docs from 'utils/docs';
 
 
@@ -435,12 +436,14 @@ const Results = (): JSX.Element => {
           {
             resultCardData.map(({ id, title, result, refresh, Component }, index: number) => (
               (result && !result.error) ? (
+                <ErrorBoundary title={title}>
                   <Component
                     key={`${title}-${index}`}
                     data={{...result}}
                     title={title}
                     actionButtons={refresh ? MakeActionButtons(title, refresh, () => showInfo(id)) : undefined}
                   />
+                </ErrorBoundary>
               ) : <></>
             ))
           }