feat(dumper): walk dir in async mode

Author: ayoubfaouzi

cmd/dump.go

@@ -14,6 +14,7 @@ import (
 	"path/filepath"
 	"reflect"
 	"strings"
+	"sync"
 	"text/tabwriter"
 	"time"
 	"unicode"
@@ -23,6 +24,50 @@ import (
 	"github.com/saferwall/pe/log"
 )
 
+var (
+	wg   sync.WaitGroup
+	jobs chan string = make(chan string)
+)
+
+func loopFilesWorker(cfg config) error {
+	for path := range jobs {
+		files, err := os.ReadDir(path)
+		if err != nil {
+			wg.Done()
+			return err
+		}
+
+		for _, file := range files {
+			if !file.IsDir() {
+				fullpath := filepath.Join(path, file.Name())
+				parse(fullpath, cfg)
+			}
+		}
+		wg.Done()
+	}
+	return nil
+}
+
+func LoopDirsFiles(path string) error {
+	files, err := os.ReadDir(path)
+	if err != nil {
+		return err
+	}
+	//Add this path as a job to the workers
+	//You must call it in a go routine, since if every worker is busy, then you have to wait for the channel to be free.
+	go func() {
+		wg.Add(1)
+		jobs <- path
+	}()
+	for _, file := range files {
+		if file.IsDir() {
+			//Recursively go further in the tree
+			LoopDirsFiles(filepath.Join(path, file.Name()))
+		}
+	}
+	return nil
+}
+
 func prettyPrint(buff []byte) string {
 	var prettyJSON bytes.Buffer
 	error := json.Indent(&prettyJSON, buff, "", "\t")
@@ -500,6 +545,12 @@ func parsePE(filename string, cfg config) {
 
 		fmt.Printf("\nRESOURCES\n**********\n")
 		printRsrcDir(pe.Resources)
+
+		r, err := pe.ParseVersionResources()
+		if err == nil {
+			fmt.Print(r)
+		}
+		fmt.Print()
 	}
 
 	if cfg.wantException && pe.FileInfo.HasException {

cmd/main.go

@@ -8,6 +8,7 @@ import (
 	"flag"
 	"fmt"
 	"os"
+	"runtime"
 )
 
 type config struct {
@@ -88,7 +89,14 @@ func main() {
 			wantCLR:         *dumpCLR,
 		}
 
-		parse(os.Args[2], cfg)
+		//Start as many workers you want, now 10 workers
+		numWorkers := runtime.GOMAXPROCS(runtime.NumCPU() - 1)
+		for w := 1; w <= numWorkers; w++ {
+			go loopFilesWorker(cfg)
+		}
+		//Start the recursion
+		LoopDirsFiles(os.Args[2])
+		wg.Wait()
 
 	case "version":
 		verCmd.Parse(os.Args[2:])
@@ -109,5 +117,6 @@ func showHelp() {
 	Brought to you by Saferwall (c) 2018 MIT
 `)
 	fmt.Println("\nAvailable sub-commands 'dump' or 'version' subcommands")
+
 	os.Exit(1)
 }