added '-l' for a limit on file size

added some error handling for files we don't want to check

README.md

@@ -6,16 +6,17 @@ A client for [LOLDrivers](https://github.com/magicsword-io/LOLDrivers) (Living O
 LOLDrivers-client.exe -m [MODE] [OPTIONS]
 
 Modes:
-    online    Download the newest driver set (default)
-    local     Use a local drivers.json file (requires '-f')
-    internal  Use the built-in driver set (can be outdated)
+  online    Download the newest driver set (default)
+  local     Use a local drivers.json file (requires '-f')
+  internal  Use the built-in driver set (can be outdated, fallback)
 
 Options:
-    -d Directory to scan for drivers (default: Windows Default)
-    -f File path to 'drivers.json' for mode 'local'
-    -t Number of threads to spawn (default: 20)
-    -v Print verbose messages (default: false)
-    -h Shows this text
+  -d        Directory to scan for drivers (default: Windows Default)
+  -l        Size limit for files to scan in MB (default: 10)
+  -f        File path to 'drivers.json' for mode 'local'
+  -t        Number of threads to spawn (default: 20)
+  -v        Print verbose messages (default: false)
+  -h        Shows this text
 ```
 **Warning:** This project is not affiliated with the [LOLDrivers](https://github.com/magicsword-io/LOLDrivers) repository. JSON structure changes in the LOLDrivers API may break this client. Since this clients gets compiled with a working data set, it will fall back to the internal data set, if the parsing of the online data or the local file was not successful.
 

cli/loldrivers-client/loldrivers-client.go

@@ -21,11 +21,13 @@ func main() {
 	// Setup & parse command line arguments
 	var flagMode string
 	var flagDir string
+	var flagFileLimit int64
 	var flagLocalFile string
 	var flagThreads int
 	var flagVerbose bool
 	flag.StringVar(&flagMode, "m", "online", "")
 	flag.StringVar(&flagDir, "d", "", "")
+	flag.Int64Var(&flagFileLimit, "l", 10, "")
 	flag.StringVar(&flagLocalFile, "f", "", "")
 	flag.IntVar(&flagThreads, "t", 20, "")
 	flag.BoolVar(&flagVerbose, "v", false, "")
@@ -34,16 +36,17 @@ func main() {
 LOLDrivers-client.exe -m [MODE] [OPTIONS]
 
 Modes:
-    online    Download the newest driver set (default)
-    local     Use a local drivers.json file (requires '-f')
-    internal  Use the built-in driver set (can be outdated)
+  online    Download the newest driver set (default)
+  local     Use a local drivers.json file (requires '-f')
+  internal  Use the built-in driver set (can be outdated, fallback)
 
 Options:
-    -d Directory to scan for drivers (default: Windows Default)
-    -f File path to 'drivers.json' for mode 'local'
-    -t Number of threads to spawn (default: 20)
-    -v Print verbose messages (default: false)
-    -h Shows this text
+  -d        Directory to scan for drivers (default: Windows Default)
+  -l        Size limit for files to scan in MB (default: 10)
+  -f        File path to 'drivers.json' for mode 'local'
+  -t        Number of threads to spawn (default: 20)
+  -v        Print verbose messages (default: false)
+  -h        Shows this text
 	`)
 	}
 	flag.Parse()
@@ -181,7 +184,7 @@ Options:
 
 		for _, path := range filesystem.DriverPaths {
 			// Get all files
-			files, err := filesystem.FilesInFolder(path)
+			files, err := filesystem.FilesInFolder(path, flagFileLimit)
 			if err != nil {
 				logger.CatchCrit(err)
 			}
@@ -194,7 +197,7 @@ Options:
 	} else {
 		// Scan the user specified folder for drivers
 		// Get all files
-		files, err := filesystem.FilesInFolder(flagDir)
+		files, err := filesystem.FilesInFolder(flagDir, flagFileLimit)
 		if err != nil {
 			logger.CatchCrit(err)
 		}

pkg/filesystem/filesystem.go

@@ -47,22 +47,45 @@ func FileRead(filepath string) (contentBytes []byte, err error) {
 	return contentBytes, nil
 }
 
-// FilesInFolder() will return a []string of files in a folder (recursively)
-func FilesInFolder(path string) (files []string, err error) {
+// FilesInFolder() will return a []string of filepaths from a given folder (recursively)
+//
+// sizeLimit in MB (ex: 5)
+func FilesInFolder(path string, sizeLimit int64) (files []string, err error) {
 	logger.Verbose(fmt.Sprintf("[*] Searching for files in %s", path))
 
 	// Walk over every file in a given folder
 	err = filepath.Walk(path, func(path string, info os.FileInfo, err error) error {
 		if err != nil {
+			// Ignore a file if we get "Access is denied" error
+			if os.IsPermission(err) {
+				return nil
+			}
+
+			// Ignore a file if we get "The system cannot find the file specified" error
+			if os.IsNotExist(err) {
+				return nil
+			}
+
 			return err
 		}
 
-		// Only append if it's not a directory
-		if !info.IsDir() {
-			// Append to slice
-			files = append(files, path)
+		// Skip directories and non regular files
+		if info.IsDir() || !info.Mode().IsRegular() {
+			return nil
+		}
+
+		// Skip files that can't be read
+		if info.Mode().Perm()&0400 == 0 {
+			return nil
+		}
+
+		// Skip files larger than the specified size limit
+		if info.Size() > sizeLimit*1024*1024 {
+			return nil
 		}
 
+		// Append to slice
+		files = append(files, path)
 		return nil
 	})