fix(key_errors): handle Key Errors in Lambda and EMR (#1871)

Co-authored-by: sergargar <sergargar@users.noreply.github.com>

prowler/providers/aws/services/awslambda/awslambda_function_no_secrets_in_code/awslambda_function_no_secrets_in_code.py

@@ -12,6 +12,7 @@ class awslambda_function_no_secrets_in_code(Check):
     def execute(self):
         findings = []
         for function in awslambda_client.functions.values():
+            if function.code:
                 report = Check_Report_AWS(self.metadata())
                 report.region = function.region
                 report.resource_id = function.name
@@ -21,13 +22,11 @@ class awslambda_function_no_secrets_in_code(Check):
                 report.status_extended = (
                     f"No secrets found in Lambda function {function.name} code"
                 )
-
                 with tempfile.TemporaryDirectory() as tmp_dir_name:
                     function.code.code_zip.extractall(tmp_dir_name)
                     # List all files
                     files_in_zip = next(os.walk(tmp_dir_name))[2]
                     for file in files_in_zip:
-
                         secrets = SecretsCollection()
                         with default_settings():
                             secrets.scan_file(f"{tmp_dir_name}/{file}")

prowler/providers/aws/services/awslambda/awslambda_service.py

@@ -87,6 +87,7 @@ class Lambda:
                     function_information = regional_client.get_function(
                         FunctionName=function.name
                     )
+                    if "Location" in function_information["Code"]:
                         code_location_uri = function_information["Code"]["Location"]
                         raw_code_zip = requests.get(code_location_uri).content
                         self.functions[function.name].code = LambdaCode(

prowler/providers/aws/services/emr/emr_service.py

@@ -97,6 +97,10 @@ class EMR:
                     slave_node_security_group = cluster_info["Cluster"][
                         "Ec2InstanceAttributes"
                     ]["EmrManagedSlaveSecurityGroup"]
+                    if (
+                        "AdditionalSlaveSecurityGroups"
+                        in cluster_info["Cluster"]["Ec2InstanceAttributes"]
+                    ):
                         slave_node_additional_security_groups = cluster_info["Cluster"][
                             "Ec2InstanceAttributes"
                         ]["AdditionalSlaveSecurityGroups"]