Ignore low severity malwares

pkg/output/output.go

@@ -3,19 +3,18 @@ package output
 import (
 	"encoding/json"
 	"fmt"
+	"os"
 	"path/filepath"
+	"strings"
+	"time"
+	"unicode/utf8"
 
 	"github.com/deepfence/YaraHunter/utils"
 	pb "github.com/deepfence/agent-plugins-grpc/srcgo"
 	log "github.com/sirupsen/logrus"
 
 	// "github.com/fatih/color"
 
-	"os"
-	"strings"
-	"time"
-	"unicode/utf8"
-
 	tw "github.com/olekukonko/tablewriter"
 )
 

pkg/scan/process_image.go

@@ -3,6 +3,7 @@ package scan
 import (
 	"bytes"
 	"errors"
+	"fmt"
 	"io"
 	"math"
 	"os/exec"
@@ -11,8 +12,6 @@ import (
 	"syscall"
 	"unsafe"
 
-	"fmt"
-
 	"github.com/gabriel-vasile/mimetype"
 
 	"github.com/deepfence/YaraHunter/pkg/output"
@@ -34,13 +33,6 @@ type manifestItem struct {
 	LayerIds []string `json:",omitempty"`
 }
 
-type fileMatches struct {
-	fileName        string
-	iocs            []output.IOCFound
-	updatedScore    float64
-	updatedSeverity string
-}
-
 func calculateSeverity(lenMatch int, severity string, severityScore float64) (string, float64) {
 
 	updatedSeverity := "low"
@@ -214,12 +206,12 @@ func ScanFile(s *Scanner, fileName string, f io.ReadSeeker, fsize int, iocs *[]o
 			Matches:          matches,
 		})
 	}
-	var fileMat fileMatches
-	fileMat.fileName = fileName
-	fileMat.iocs = iocsFound
 	updatedSeverity, updatedScore := calculateSeverity(totalMatches, "low", 0)
-	fileMat.updatedSeverity = updatedSeverity
-	fileMat.updatedScore = updatedScore
+	//if updatedSeverity == "low" {
+	//	// Ignore low severity malwares
+	//	return nil
+	//}
+	logrus.Infof("Updated severity: %s", updatedSeverity)
 	if len(matches) > 0 {
 		for _, m := range iocsFound {
 			m.FileSeverity = updatedSeverity