fix for crash in lambda function cft yaml if code key is not present (#…

…1644)
tenable · Nov 27, 2023 · 7a3779a · 7a3779a
1 parent 2029321
commit 7a3779a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/mapper/iac-providers/cft/config/lambda-function.go b/pkg/mapper/iac-providers/cft/config/lambda-function.go
@@ -195,10 +195,10 @@ func getLambdaConfig(lf *lambda.Function) []AWSResourceConfig {
 func setLambdaCodePackage(cf LambdaFunctionConfig, f *lambda.Function) LambdaFunctionConfig {
 	if f.Code != nil {
 		cf.ImageURI = functions.GetVal(f.Code.ImageUri)
+		cf.S3Bucket = functions.GetVal(f.Code.S3Bucket)
+		cf.S3Key = functions.GetVal(f.Code.S3Key)
+		cf.S3ObjectVersion = functions.GetVal(f.Code.S3ObjectVersion)
 		return cf
 	}
-	cf.S3Bucket = functions.GetVal(f.Code.S3Bucket)
-	cf.S3Key = functions.GetVal(f.Code.S3Key)
-	cf.S3ObjectVersion = functions.GetVal(f.Code.S3ObjectVersion)
 	return cf
 }