From 7d5a06c4bfcdec1c5a853f82a9800abb6f9dfe99 Mon Sep 17 00:00:00 2001 From: Tonis Tiigi Date: Fri, 27 Sep 2024 07:17:42 -0700 Subject: [PATCH] llbsolver: add input validation to policy recompute Similar error already exists in loadLLB but was missing in newer recomputeDigests code. Signed-off-by: Tonis Tiigi --- solver/llbsolver/vertex.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/solver/llbsolver/vertex.go b/solver/llbsolver/vertex.go index d4dae31467b7..d71239b63446 100644 --- a/solver/llbsolver/vertex.go +++ b/solver/llbsolver/vertex.go @@ -203,7 +203,10 @@ func recomputeDigests(ctx context.Context, all map[digest.Digest]*pb.Op, visited if dgst, ok := visited[dgst]; ok { return dgst, nil } - op := all[dgst] + op, ok := all[dgst] + if !ok { + return "", errors.Errorf("invalid missing input digest %s", dgst) + } var mutated bool for _, input := range op.Inputs {