From cb8d892e744200e3383b6ce3e2c8a196ab232d45 Mon Sep 17 00:00:00 2001
From: Sudhir Verma <sudhirvvermaa@gmail.com>
Date: Tue, 2 Jan 2024 18:23:37 +0530
Subject: [PATCH] Added Two download buttons in Defence and vulnerability
 report (#505)

---
 .../AssessmentReport/index.tsx                | 213 +++++++++++-------
 src/pages/FuzzingModel/DefenceData.tsx        | 173 ++++++++++----
 .../FuzzingModel/sections/DefenseSummary.tsx  |   8 +-
 src/utils/fatch_json_report.ts                |   7 +
 4 files changed, 271 insertions(+), 130 deletions(-)

diff --git a/src/pages/BesVersionHistory/AssessmentReport/index.tsx b/src/pages/BesVersionHistory/AssessmentReport/index.tsx
index 97633f33..d6233099 100644
--- a/src/pages/BesVersionHistory/AssessmentReport/index.tsx
+++ b/src/pages/BesVersionHistory/AssessmentReport/index.tsx
@@ -18,15 +18,26 @@ export const verifyLink = async (link: any, setLinkStatus: any) => {
     const response = await fetchJsonReport(link);
     try {
       let data = JSON.parse(response);
-      //console.log(data);
-      setLinkStatus(data);
+      if (link.toLocaleLowerCase().endsWith(".pdf")) {
+        setLinkStatus(true);
+      } else {
+        setLinkStatus(data);
+      }
       return true;
     } catch (err) {
-      setLinkStatus({});
+      if (link.toLocaleLowerCase().endsWith(".pdf")) {
+        setLinkStatus(false);
+      } else {
+        setLinkStatus({});
+      }
       return false;
     }
   } catch (error) {
-    setLinkStatus({});
+    if (link.toLocaleLowerCase().endsWith(".pdf")) {
+      setLinkStatus(false);
+    } else {
+      setLinkStatus({});
+    }
     return false;
   }
 };
@@ -34,15 +45,15 @@ export const verifyLink = async (link: any, setLinkStatus: any) => {
 const CheckLink = ({ version, name, report }: any) => {
   const [linkStatus, setLinkStatus]: any = React.useState({});
   let reportNameMap = "";
-  if(report === "Criticality Score"){
+  if (report === "Criticality Score") {
     reportNameMap = "Criticality Score";
-  }else if(report === "Vulnerabilities"){
+  } else if (report === "Vulnerabilities") {
     reportNameMap = "Codeql";
-  }else if(report === "License Compliance"){
+  } else if (report === "License Compliance") {
     reportNameMap = "Fossology";
-  }else if(report === "Dependencies"){
+  } else if (report === "Dependencies") {
     reportNameMap = "SBOM";
-  }else if(report === "ScoreCard"){
+  } else if (report === "ScoreCard") {
     reportNameMap = "Scorecard";
   }
 
@@ -53,7 +64,7 @@ const CheckLink = ({ version, name, report }: any) => {
     }
   }, [version]);
   let linkStatusLength: number = Object.values(linkStatus).length;
-  if (report === "Criticality Score" && linkStatusLength !== 0){
+  if (report === "Criticality Score" && linkStatusLength !== 0) {
     return (
       <Typography variant="subtitle1" color="inherit">
         {linkStatus.criticality_score}
@@ -64,7 +75,7 @@ const CheckLink = ({ version, name, report }: any) => {
   const myObject = { pathname: pathName, state: linkStatus } as {
     pathname: string;
   };
-  
+
   if (report === "ScoreCard" && linkStatusLength !== 0) {
     return <Link to={myObject}>{linkStatus.score}</Link>;
   }
@@ -73,26 +84,28 @@ const CheckLink = ({ version, name, report }: any) => {
   }
   if (report === "License Compliance" && linkStatusLength !== 0) {
     let uniqueLicenses: any = [];
-    for(let i=0; i<linkStatus.length; i++){
-      let flag =0;
-      for(let j=0; j<uniqueLicenses.length; j++){
-          if(linkStatus[i].LicenseConcluded === uniqueLicenses[j] ||
-            linkStatus[i].LicenseConcluded === "NOASSERTION"){
-             flag=1;
-             break;
-          }
+    for (let i = 0; i < linkStatus.length; i++) {
+      let flag = 0;
+      for (let j = 0; j < uniqueLicenses.length; j++) {
+        if (
+          linkStatus[i].LicenseConcluded === uniqueLicenses[j] ||
+          linkStatus[i].LicenseConcluded === "NOASSERTION"
+        ) {
+          flag = 1;
+          break;
+        }
       }
-      if(flag===0){
+      if (flag === 0) {
         uniqueLicenses.push(linkStatus[i].LicenseConcluded);
       }
     }
-    console.log("UniqueLicenses Are="+uniqueLicenses);
+    console.log("UniqueLicenses Are=" + uniqueLicenses);
     return <Link to={myObject}>{uniqueLicenses.length}</Link>;
   }
   if (report === "Dependencies" && linkStatusLength !== 0) {
     return <Link to={myObject}>{linkStatus.packages.length}</Link>;
   }
-  
+
   return (
     <Typography variant="subtitle1" color="inherit">
       --
@@ -101,43 +114,75 @@ const CheckLink = ({ version, name, report }: any) => {
 };
 const GetHeadings = ({ receivedValue }: any) => {
   //const [fieldInfo, setfieldInfo]: any = React.useState({});
-    if(receivedValue === "License Compliance"){
-       return(<> Unique Licenses
-                  <Icon title="Number of unique licenses used in the OSS" sx={{fontSize: '0.7rem !important'}}>
-                    info
-                  </Icon>
-              </>);
-    }else if(receivedValue === "Dependencies"){
-      return(<> Dependencies(SBOM)
-                <Icon title="Software Bill Of Material" sx={{fontSize: '0.7rem !important'}}>
-                  info
-                </Icon>
-              </>);
-      
-    }else if(receivedValue === "ScoreCard"){
-      return(<> {receivedValue}
-                <Icon title="Overall Security Score of the project" sx={{fontSize: '0.7rem !important'}}>
-                  info
-                </Icon>
-              </>);
-     
-    }else if(receivedValue === "Criticality Score"){
-      return(<> {receivedValue}
-                <Icon title="Score to tell how critical the OSS project" sx={{fontSize: '0.7rem !important'}}>
-                  info
-                </Icon>
-              </>);
-     
-    }else if(receivedValue === "Vulnerabilities"){
-      return(<> SAST Risks
-              <Icon title="Provides Static Code Analysis (SAST) report by CodeQL / SonarQube" sx={{fontSize: '0.7rem !important'}}>
-                info
-              </Icon>
-              </>);
-    }else{
-      return(receivedValue);
-    }
-}
+  if (receivedValue === "License Compliance") {
+    return (
+      <>
+        {" "}
+        Unique Licenses
+        <Icon
+          title="Number of unique licenses used in the OSS"
+          sx={{ fontSize: "0.7rem !important" }}
+        >
+          info
+        </Icon>
+      </>
+    );
+  } else if (receivedValue === "Dependencies") {
+    return (
+      <>
+        {" "}
+        Dependencies(SBOM)
+        <Icon
+          title="Software Bill Of Material"
+          sx={{ fontSize: "0.7rem !important" }}
+        >
+          info
+        </Icon>
+      </>
+    );
+  } else if (receivedValue === "ScoreCard") {
+    return (
+      <>
+        {" "}
+        {receivedValue}
+        <Icon
+          title="Overall Security Score of the project"
+          sx={{ fontSize: "0.7rem !important" }}
+        >
+          info
+        </Icon>
+      </>
+    );
+  } else if (receivedValue === "Criticality Score") {
+    return (
+      <>
+        {" "}
+        {receivedValue}
+        <Icon
+          title="Score to tell how critical the OSS project"
+          sx={{ fontSize: "0.7rem !important" }}
+        >
+          info
+        </Icon>
+      </>
+    );
+  } else if (receivedValue === "Vulnerabilities") {
+    return (
+      <>
+        {" "}
+        SAST Risks
+        <Icon
+          title="Provides Static Code Analysis (SAST) report by CodeQL / SonarQube"
+          sx={{ fontSize: "0.7rem !important" }}
+        >
+          info
+        </Icon>
+      </>
+    );
+  } else {
+    return receivedValue;
+  }
+};
 function AssessmentReport({ title, name, version, ...other }: any) {
   const report: string[] = [
     "ScoreCard",
@@ -147,41 +192,49 @@ function AssessmentReport({ title, name, version, ...other }: any) {
     "Dependencies"
   ];
   return (
-    
-    <Card sx={{ height: "100%" }} >
-      <Grid container p={2} justifyContent="center" >
-        
+    <Card sx={{ height: "100%" }}>
+      <Grid container p={2} justifyContent="center">
         {report.map((value, index) => {
-            return (
-              <>
+          return (
+            <>
               <Grid item xs={2.4}>
-              <MKBox p={1.5} borderRadius="lg">
-              <Grid p={1} justifyContent="center" style={{backgroundColor: "#f3f6f4", borderRadius: 10}} >
-                  
-                    <Grid container justifyContent="center" alignItems="center" >
+                <MKBox p={1.5} borderRadius="lg">
+                  <Grid
+                    p={1}
+                    justifyContent="center"
+                    style={{ backgroundColor: "#f3f6f4", borderRadius: 10 }}
+                  >
+                    <Grid container justifyContent="center" alignItems="center">
                       <Grid item justifyContent="center">
                         <Typography variant="h6" color="black">
-                          <GetHeadings receivedValue={value}/> 
-                          </Typography>
+                          <GetHeadings receivedValue={value} />
+                        </Typography>
                       </Grid>
                     </Grid>
                     <Grid>
                       <Grid item>
-                        <Grid container justifyContent="center" alignItems="center">
+                        <Grid
+                          container
+                          justifyContent="center"
+                          alignItems="center"
+                        >
                           <Grid item>
-                            <CheckLink version={version} name={name} report={value} />
+                            <CheckLink
+                              version={version}
+                              name={name}
+                              report={value}
+                            />
                           </Grid>
                         </Grid>
                       </Grid>
                     </Grid>
-                </Grid>
-              </MKBox>
+                  </Grid>
+                </MKBox>
               </Grid>
-              </>
-            );
-          })}
-        
-        </Grid>
+            </>
+          );
+        })}
+      </Grid>
     </Card>
   );
 }
diff --git a/src/pages/FuzzingModel/DefenceData.tsx b/src/pages/FuzzingModel/DefenceData.tsx
index 3b07ae52..c2b5f0f9 100644
--- a/src/pages/FuzzingModel/DefenceData.tsx
+++ b/src/pages/FuzzingModel/DefenceData.tsx
@@ -1,11 +1,18 @@
 import React, { useState } from "react";
 
-import { Card, CircularProgress } from "@mui/material";
-import MKBox from "../../components/MKBox";
+import { Card, CircularProgress, Grid } from "@mui/material";
 import MKTypography from "../../components/MKTypography";
 import DefenceReport from "./DefenceReport";
+import MKButton from "../../components/MKButton";
+import { verifyLink } from "../BesVersionHistory/AssessmentReport";
+import { besecureMlAssessmentDataStore } from "../../dataStore";
+import { NavLink, useLocation } from "react-router-dom";
 
-function reportDisplay(loading: any, report: any) {
+function reportDisplay(report: any) {
+  return <DefenceReport report={report} />;
+}
+
+function loadingCircular(loading: any) {
   if (loading) {
     return (
       <CircularProgress
@@ -19,63 +26,137 @@ function reportDisplay(loading: any, report: any) {
         thickness={4}
       />
     );
-  } else {
-    return <DefenceReport report={report} />;
   }
 }
 
-export default function DefenceData({ report }: any) {
+export default function DefenceData({ report, reportName }: any) {
   let [loading, setLoading] = useState(true);
+  const location = useLocation();
+  const selectedFuzz: any = location.state.selectedFuzz;
+
+  const [defenceReport, setDefenceReport]: any = React.useState(false);
+  const [vulnerabilityReport, setvulnerabilityReport]: any =
+    React.useState(false);
+
+  let defenceReportLink = `${besecureMlAssessmentDataStore}/${selectedFuzz.name}/fuzz-test/${reportName}/DefenceReport.pdf`;
+  let vulnerabilityReportLink = `${besecureMlAssessmentDataStore}/${selectedFuzz.name}/fuzz-test/${reportName}/VulnerabilityReport.pdf`;
   React.useEffect(() => {
+    verifyLink(defenceReportLink, setDefenceReport);
+    verifyLink(vulnerabilityReportLink, setvulnerabilityReport);
     try {
       setTimeout(setLoading, 6000);
-    } catch(e) {
+    } catch (e) {
       // ignore
     }
   }, []);
   const reportLength = Object.values(report).length;
   return (
-    <MKBox m={1}>
-      <Card
-        style={
-          loading
-            ? {
-              display: "flex",
-              alignItems: "center",
-              justifyContent: "center",
-              backgroundColor: "#f0f2f5",
-              padding: "0.4rem",
-              width: "100%"
-            }
-            : {
+    <>
+      {loading ? (
+        <Card
+          style={{
+            display: "flex",
+            alignItems: "center",
+            justifyContent: "center",
+            backgroundColor: "#f0f2f5",
+            padding: "0.4rem",
+            width: "100%"
+          }}
+        >
+          {" "}
+          {loadingCircular(loading)}
+        </Card>
+      ) : (
+        <Grid container>
+          <Grid item width={"65%"}>
+            <Card
+              style={{
                 backgroundColor: "#f0f2f5",
                 padding: "0.4rem",
-                width: "100%"
-              }
-        }
-      >
-        <MKTypography
-          variant="body2"
-          component="p"
-          color="text"
-          textAlign="left"
-        >
-          {reportLength > 0 ? (
-            reportDisplay(loading, report)
-          ) : (
-            <MKTypography
-              color="black"
-              textAlign="center"
-              variant="h6"
-              sx={{ margin: "auto" }}
-              pt={8}
-              pb={8}
+                width: "100%",
+                margin: "1%"
+              }}
             >
-              Not Available
-            </MKTypography>
-          )}
-        </MKTypography>
-      </Card>
-    </MKBox>
+              <MKTypography
+                variant="body2"
+                component="p"
+                color="text"
+                textAlign="left"
+              >
+                {reportLength > 0 ? (
+                  reportDisplay(report)
+                ) : (
+                  <MKTypography
+                    color="black"
+                    textAlign="center"
+                    variant="h6"
+                    sx={{ margin: "auto" }}
+                    pt={8}
+                    pb={8}
+                  >
+                    Not Available
+                  </MKTypography>
+                )}
+              </MKTypography>
+            </Card>
+          </Grid>
+          <Grid item width={"35%"} pt={3}>
+            {defenceReport ? (
+              <NavLink
+                to={defenceReportLink}
+                rel="noopener noreferrer"
+                style={{ color: "#587f2f", cursor: "pointer" }}
+              >
+                <MKButton
+                  variant={"gradient"}
+                  color={"info"}
+                  size="small"
+                  sx={{ width: "80%" }}
+                >
+                  Defense Report
+                </MKButton>
+              </NavLink>
+            ) : (
+              <MKButton
+                variant={"gradient"}
+                color={"info"}
+                size="small"
+                sx={{ width: "80%" }}
+                disabled
+              >
+                Defense Report
+              </MKButton>
+            )}
+
+            {vulnerabilityReport ? (
+              <NavLink
+                to={vulnerabilityReportLink}
+                rel="noopener noreferrer"
+                style={{ color: "#587f2f", cursor: "pointer" }}
+              >
+                <MKButton
+                  variant={"gradient"}
+                  color={"info"}
+                  size="small"
+                  sx={{ width: "80%", marginTop: "34%" }}
+                >
+                  Vulnerability Report
+                </MKButton>
+              </NavLink>
+            ) : (
+              <MKButton
+                variant={"gradient"}
+                color={"info"}
+                size="small"
+                sx={{ width: "80%", marginTop: "34%" }}
+                disabled
+              >
+                Vulnerability Report
+              </MKButton>
+            )}
+          </Grid>
+        </Grid>
+      )}
+    </>
   );
 }
diff --git a/src/pages/FuzzingModel/sections/DefenseSummary.tsx b/src/pages/FuzzingModel/sections/DefenseSummary.tsx
index 1c85ae97..f8dcd02c 100644
--- a/src/pages/FuzzingModel/sections/DefenseSummary.tsx
+++ b/src/pages/FuzzingModel/sections/DefenseSummary.tsx
@@ -130,16 +130,16 @@ function DefenseSummary() {
                 </Tabs>
               </Box>
               <CustomTabPanel value={value} index={0}>
-                <DefenceData report={evasion} />
+                <DefenceData report={evasion} reportName="evasion" />
               </CustomTabPanel>
               <CustomTabPanel value={value} index={1}>
-                <DefenceData report={inference} />
+                <DefenceData report={inference} reportName="inference" />
               </CustomTabPanel>
               <CustomTabPanel value={value} index={2}>
-                <DefenceData report={extraction} />
+                <DefenceData report={extraction} reportName="extraction" />
               </CustomTabPanel>
               <CustomTabPanel value={value} index={3}>
-                <DefenceData report={dataPoisoning} />
+                <DefenceData report={dataPoisoning} reportName="dataPoisoning"  />
               </CustomTabPanel>
             </Box>
           </MKBox>
diff --git a/src/utils/fatch_json_report.ts b/src/utils/fatch_json_report.ts
index 88146fbe..85cda189 100644
--- a/src/utils/fatch_json_report.ts
+++ b/src/utils/fatch_json_report.ts
@@ -11,6 +11,13 @@ export function fetchJsonReport(url: string): Promise<any> {
         });
 
         response.on("end", () => {
+          if (url.toLocaleLowerCase().endsWith(".pdf")) {
+            let regex = /404: Not Found/gm;
+            if (regex.test(data)) {
+              resolve(data);
+            }
+            resolve('{"pdfFile":true}');
+          }
           resolve(data);
         });
       })