feat: add feature file size limit

Signed-off-by: AhmedGrati <ahmedgrati1999@gmail.com>

source/local/local.go

@@ -45,6 +45,9 @@ const ExpiryTimeKey = "expiry-time"
 // DirectivePrefix defines the prefix of directives that should be parsed
 const DirectivePrefix = "# +"
 
+// MaxFeatureFileSize defines the maximum size of a feature file size
+const MaxFeatureFileSize = 64000
+
 // Config
 var (
 	featureFilesDir = "/etc/kubernetes/node-feature-discovery/features.d/"
@@ -321,6 +324,23 @@ func getFeaturesFromFiles() (map[string]string, error) {
 
 	for _, file := range files {
 		fileName := file.Name()
+
+		fileInfo, err := file.Info()
+		if err != nil {
+			klog.ErrorS(err, "failed to get file info", "fileName", fileName)
+			continue
+		}
+
+		fileSize := fileInfo.Size()
+		if fileSize > MaxFeatureFileSize {
+			klog.ErrorS(
+				fmt.Errorf("invalid file size: %d bytes > %d bytes", fileSize, MaxFeatureFileSize),
+				"skipping file with invalid size",
+				"fileName", fileName, "fileSize", fileSize,
+			)
+			continue
+		}
+
 		lines, err := getFileContent(fileName)
 		if err != nil {
 			klog.ErrorS(err, "failed to read file", "fileName", fileName)