@@ -7,8 +7,10 @@ import (
77 "log"
88 "os"
99 "path/filepath"
10+ "runtime"
1011 "scan2html/internal/common"
1112 "scan2html/internal/epss"
13+ "scan2html/internal/exploit"
1214 "scan2html/internal/logger"
1315 "strings"
1416 "time"
@@ -25,10 +27,12 @@ func GenerateHtmlReport(pluginFlags common.Flags, version string) error {
2527
2628 reportName := pluginFlags ["--output" ]
2729 _ , withEpss := pluginFlags ["--with-epss" ]
30+ _ , withExploits := pluginFlags ["--with-exploits" ]
2831 reportTitle := pluginFlags ["--report-title" ]
2932 // Log input parameters for clarity
3033 logger .Logger .Infof ("Base Directory: %s\n " , baseDir )
3134 logger .Logger .Infof ("With EPSS: %t\n " , withEpss )
35+ logger .Logger .Infof ("With Exploits: %t\n " , withExploits )
3236 logger .Logger .Infof ("Report Title: %s\n " , reportTitle )
3337 logger .Logger .Infof ("Report Name: %s\n " , reportName )
3438
@@ -43,7 +47,7 @@ func GenerateHtmlReport(pluginFlags common.Flags, version string) error {
4347
4448 err = replaceTextByText (reportName , "TEMP_APP_VERSION" , version )
4549 if err != nil {
46- return fmt .Errorf ("failed to replace report title in %s: %v" , reportName , err )
50+ return fmt .Errorf ("failed to replace app version in %s: %v" , reportName , err )
4751 }
4852
4953 // Replace placeholders with actual content in the report file
@@ -57,26 +61,67 @@ func GenerateHtmlReport(pluginFlags common.Flags, version string) error {
5761 }
5862
5963 // Handle EPSS data if enabled
64+ // replaceTextByFile "$report_name" "\"TEMP_EPSS_DATA\"" "$epss_data"
65+ // Schedule deletion of the EPSS data file upon function exit
66+ shouldReturn , returnValue := handleEPSS (withEpss , reportName )
67+ if shouldReturn {
68+ return returnValue
69+ }
70+
71+ shouldReturn , returnValue = handleExploit (withExploits , reportName )
72+ if shouldReturn {
73+ return returnValue
74+ }
75+
76+ logger .Logger .Infof ("%s has been created successfully!\n " , reportName )
77+ return nil
78+ }
79+
80+ func handleEPSS (withEpss bool , reportName string ) (bool , error ) {
6081 if withEpss {
6182 logger .Logger .Infoln ("EPSS enabled!" )
6283 var epssDataFile , err = epss .PrepareEpssData ()
6384 if err != nil {
64- return fmt .Errorf ("failed to prepare EPSS data: %v" , err )
85+ return true , fmt .Errorf ("failed to prepare EPSS data: %v" , err )
6586 }
6687
67- // replaceTextByFile "$report_name" "\"TEMP_EPSS_DATA\"" "$epss_data"
6888 if err := replaceTextByFile (reportName , "\" TEMP_EPSS_DATA\" " , epssDataFile ); err != nil {
69- return fmt .Errorf ("failed to replace EPSS data in %s: %v" , reportName , err )
89+ return true , fmt .Errorf ("failed to replace EPSS data in %s: %v" , reportName , err )
7090 }
7191
7292 logger .Logger .Infoln ("EPSS data imported!" )
7393
74- // Schedule deletion of the EPSS data file upon function exit
7594 defer os .Remove (epssDataFile )
7695 }
96+ return false , nil
97+ }
7798
78- logger .Logger .Infof ("%s has been created successfully!\n " , reportName )
79- return nil
99+ func handleExploit (withExploits bool , reportName string ) (bool , error ) {
100+ if withExploits {
101+ logger .Logger .Infoln ("Exploits enabled!" )
102+ var exploitDataFile , err = exploit .PrepareExploitData ()
103+ if err != nil {
104+ return true , fmt .Errorf ("failed to prepare Exploits data: %v" , err )
105+ }
106+
107+ if err := replaceTextByFile (reportName , "{TEMP_EXPLOITS:0}" , exploitDataFile ); err != nil {
108+ return true , fmt .Errorf ("failed to replace Exploits data in %s: %v" , reportName , err )
109+ }
110+
111+ logger .Logger .Infoln ("Exploits data imported!" )
112+
113+ defer os .Remove (exploitDataFile )
114+ }
115+ return false , nil
116+ }
117+
118+ // replaceTextByFile replaces occurrences of search_text in the input file with content from replace_file.
119+ func replaceTextByFile (inputFile , searchText , replaceFile string ) error {
120+ replaceContent , err := os .ReadFile (replaceFile )
121+ if err != nil {
122+ return fmt .Errorf ("could not read file %s: %v" , replaceFile , err )
123+ }
124+ return replaceTextByText (inputFile , searchText , string (replaceContent ))
80125}
81126
82127// replaceTextByText replaces occurrences of search_text in the input file with replace_content.
@@ -87,11 +132,11 @@ func replaceTextByText(inputFile, searchText, replaceContent string) error {
87132 }
88133 defer file .Close ()
89134
90- tempFile , err := os .CreateTemp ("" , "modified_" )
135+ timestamp := time .Now ().Format ("2006_01_02_15_04_05_06" )
136+ tempFile , err := os .CreateTemp ("" , fmt .Sprintf ("modified_%s" , timestamp ))
91137 if err != nil {
92138 return fmt .Errorf ("could not create temp file: %v" , err )
93139 }
94- defer tempFile .Close ()
95140
96141 reader := bufio .NewReader (file )
97142 writer := bufio .NewWriter (tempFile )
@@ -119,45 +164,52 @@ func replaceTextByText(inputFile, searchText, replaceContent string) error {
119164 return fmt .Errorf ("error writing to temp file: %v" , err )
120165 }
121166
122-
123167 return copyAndRemove (tempFile .Name (), inputFile )
124168}
125169
126170func copyAndRemove (src , dst string ) error {
127- // Open the source file
128- sourceFile , err := os .Open (src )
129- if err != nil {
130- return err
131- }
132- defer sourceFile .Close ()
133-
134- // Create the destination file
135- destFile , err := os .Create (dst )
136- if err != nil {
137- return err
138- }
139- defer destFile .Close ()
140-
141- // Copy the contents
142- if _ , err := io .Copy (destFile , sourceFile ); err != nil {
143- return err
144- }
145-
146- // Close files before removal
147- sourceFile .Close ()
148- destFile .Close ()
149-
150- // Remove the source file
151- return os .Remove (src )
152- }
171+ // Open the source file
172+ sourceFile , err := os .Open (src )
173+ if err != nil {
174+ return fmt .Errorf ("failed to open source file %s: %v" , src , err )
175+ }
176+ defer sourceFile .Close ()
153177
154- // replaceTextByFile replaces occurrences of search_text in the input file with content from replace_file.
155- func replaceTextByFile (inputFile , searchText , replaceFile string ) error {
156- replaceContent , err := os .ReadFile (replaceFile )
178+ // Create the destination file
179+ destFile , err := os .Create (dst )
157180 if err != nil {
158- return fmt .Errorf ("could not read file %s: %v" , replaceFile , err )
181+ return fmt .Errorf ("failed to create destination file %s: %v" , dst , err )
159182 }
160- return replaceTextByText (inputFile , searchText , string (replaceContent ))
183+ defer destFile .Close ()
184+
185+ // Copy the contents
186+ if _ , err := io .Copy (destFile , sourceFile ); err != nil {
187+ return fmt .Errorf ("failed to copy contents from %s to %s: %v" , src , dst , err )
188+ }
189+
190+ // Ensure all data is written to the destination file
191+ if err := destFile .Sync (); err != nil {
192+ return fmt .Errorf ("failed to sync destination file %s: %v" , dst , err )
193+ }
194+
195+ // Close the destination file
196+ if err := destFile .Close (); err != nil {
197+ return fmt .Errorf ("failed to close destination file %s: %v" , dst , err )
198+ }
199+
200+ // Close the source file
201+ if err := sourceFile .Close (); err != nil {
202+ return fmt .Errorf ("failed to close source file %s: %v" , src , err )
203+ }
204+
205+ // Check if the operating system is Windows
206+ if runtime .GOOS == "windows" {
207+ // Remove the source file is failing on windows
208+ return nil
209+ }
210+
211+ defer os .Remove (src )
212+ return nil
161213}
162214
163215// generateReportName creates a unique report name based on timestamp if the file already exists.
0 commit comments