From ead7c163a05bf64697e96460ed2dd44ca098dce3 Mon Sep 17 00:00:00 2001
From: Ritish Srivastava <121374890+Ritish134@users.noreply.github.com>
Date: Sun, 7 Sep 2025 14:58:01 +0530
Subject: [PATCH] validate mappings

Signed-off-by: Ritish Srivastava <121374890+Ritish134@users.noreply.github.com>
---
 main.go | 24 +++++++++++++++++++++---
 1 file changed, 21 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index 571e459..a6e2ba6 100644
--- a/main.go
+++ b/main.go
@@ -14,6 +14,7 @@ import (
 	"os"
 	"os/signal"
 	"path/filepath"
+	"strings"
 	"syscall"
 
 	"github.com/chainguard-dev/clog"
@@ -118,10 +119,27 @@ func cli() *cobra.Command {
 
 			// If custom mappings file is provided, load it as ExtraMappings
 			if mappingsFile != "" {
-				log.Info("Loading custom mappings file", "file", mappingsFile)
-				mappingsBytes, err := os.ReadFile(mappingsFile)
+				cleanedPath := filepath.Clean(mappingsFile)
+
+				// Reject empty path
+				if cleanedPath == "" || cleanedPath == "." {
+					return fmt.Errorf("mappings file path is empty or invalid: %q", cleanedPath)
+				}
+
+				// Prevents directory traversal
+				if strings.Contains(cleanedPath, "..") {
+					return fmt.Errorf("invalid mappings file path: %q", cleanedPath)
+				}
+
+				// Require .yaml or .yml extension
+				ext := filepath.Ext(cleanedPath)
+				if ext != ".yaml" && ext != ".yml" {
+					return fmt.Errorf("mappings file must be .yaml or .yml, got %q", ext)
+				}
+				log.Info("Loading custom mappings file", "file", cleanedPath)
+				mappingsBytes, err := os.ReadFile(cleanedPath)
 				if err != nil {
-					return fmt.Errorf("reading mappings file %s: %w", mappingsFile, err)
+					return fmt.Errorf("reading mappings file %s: %w", cleanedPath, err)
 				}
 
 				var extraMappings dfc.MappingsConfig