diff --git a/requirements.txt b/requirements.txt index 01c63d97..095bcf5e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,9 +1,9 @@ -certifi==2018.11.29 +certifi==2023.7.22 chardet==3.0.4 datadog==0.26.0 decorator==4.3 -idna==2.8 -requests==2.21 +idna==3.7 +requests==2.31.0 simplejson==3.16 -urllib3==1.24.2 +urllib3==1.26.18 pytz==2019.3 diff --git a/scan.py b/scan.py index 48545a06..e45dddf5 100644 --- a/scan.py +++ b/scan.py @@ -164,6 +164,16 @@ def sns_start_scan(sns_client, s3_object, scan_start_sns_arn, timestamp): TargetArn=scan_start_sns_arn, Message=json.dumps({"default": json.dumps(message)}), MessageStructure="json", + MessageAttributes={ + "bucket": { + "DataType": "String", + "StringValue": s3_object.bucket_name + }, + "key": { + "DataType": "String", + "StringValue": s3_object.key + } + } ) @@ -191,7 +201,18 @@ def sns_scan_results( Message=json.dumps({"default": json.dumps(message)}), MessageStructure="json", MessageAttributes={ - AV_STATUS_METADATA: {"DataType": "String", "StringValue": scan_result}, + "bucket": { + "DataType": "String", + "StringValue": s3_object.bucket_name + }, + "key": { + "DataType": "String", + "StringValue": s3_object.key + }, + AV_STATUS_METADATA: { + "DataType": "String", + "StringValue": scan_result + }, AV_SIGNATURE_METADATA: { "DataType": "String", "StringValue": scan_signature,