Fixed the Cannot read property 'host' of undefined issue when only on…

…e site is available
kasunkv · May 12, 2018 · 65e7ebe · 65e7ebe
1 parent ef0c11c
commit 65e7ebe
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/.taskkey b/.taskkey
@@ -1 +1 @@
-cca8c7a9-9e4f-4e4b-8397-bae8755b7cf2
+8eb12ee0-acf7-4d11-a6dc-be4d079a03fa
diff --git a/BuildTask/OwaspZapScan/classes/Helper.ts b/BuildTask/OwaspZapScan/classes/Helper.ts
@@ -28,7 +28,8 @@ export class Helper {
             }
 
             const reportJson: ScanReport = res;
-            const sites: Site[] = reportJson.OWASPZAPReport.site;
+            const siteCollection: any = reportJson.OWASPZAPReport.site;
+            const sites: Site[] = Object.keys(siteCollection)[0] === '0' ? siteCollection : [siteCollection as Site];
 
             for (const idx in sites) {
                 if (targetUrl.includes(sites[idx].$.host)) {

diff --git a/BuildTask/OwaspZapScan/interfaces/types/ZapReport.ts b/BuildTask/OwaspZapScan/interfaces/types/ZapReport.ts
@@ -4,7 +4,7 @@ export interface ScanReport {
 }
 
 export interface OwaspZapReport {
-    site: Array<Site>;
+    site: Site[];
     $: { version: string, generated: string };
 }