diff --git a/prowler/providers/aws/lib/s3/s3.py b/prowler/providers/aws/lib/s3/s3.py index 6ebb711f..f94d7c63 100644 --- a/prowler/providers/aws/lib/s3/s3.py +++ b/prowler/providers/aws/lib/s3/s3.py @@ -36,7 +36,14 @@ def send_to_s3_bucket( object_name = bucket_directory + "/" + output_mode + "/" + filename s3_client = audit_session.client("s3") - s3_client.upload_file(file_name, output_bucket_name, object_name) + + extra_args = {} + if filename.endswith('.html'): + extra_args['ContentType'] = 'text/html' + elif filename.endswith('.pdf'): + extra_args['ContentType'] = 'application/pdf' + + s3_client.upload_file(file_name, output_bucket_name, object_name, ExtraArgs=extra_args) except Exception as error: logger.error( @@ -49,4 +56,4 @@ def get_s3_object_path(output_directory: str) -> str: if "prowler/" in bucket_remote_dir: # Check if it is not a custom directory bucket_remote_dir = bucket_remote_dir.partition("prowler/")[-1] - return bucket_remote_dir + return bucket_remote_dir \ No newline at end of file