Architecture Related
Below are some of the most common questions related to Architecture
No. The Antivirus for Amazon S3 solution will fit into your existing workflow. You do not have to make any changes to your current workflow.
Yes, Antivirus for Amazon S3 supports
cross-account scanning. This means you can centrally install the console and scanning agents to protect not only the account you are deployed within, but also any other AWS account where you can install a cross-account role.Yes, it is very easy to setup a two bucket system and we have many customers using this approach. With a combination of Event Based scanning and Proactive Notifications you can easily implement this approach.
- 1.Create
staging bucketor utilize existing bucket - 2.
- 3.Create a Python (latest version) Lambda Function with sample code provided below
- 4.Make adjustments to the Lambda settings with the information below
- 5.Subscribe Lambda to SNS Notifications Topic
- 6.Add IAM permissions to Lambda with provided permission blocks below
- 7.Modify Topic Subscription to filter down to
cleanobjects with provided filter block below - 8.Test clean and "not clean" files to ensure behavior is as expected
3. Sample Copy Lambda
4. Make Adjustments to Lambda Settings
6. Permissions to add to Lambda Role
7. Subscription Filter for Clean Results
Sample Copy Lambda
The code below is a starting point and does work out of the box, but more can be done with it and to it. Feel free to do so.
import json
import boto3
import os
from botocore.exceptions import ClientError, ParamValidationError
import random
from urllib import parse
def lambda_handler(event, context):
try:
print(json.dumps(event))
SOURCE_BUCKET = os.getenv("SOURCE_BUCKET", 'any')
DESTINATION_BUCKET = os.getenv("DESTINATION_BUCKET", '<some failover bucket>')
DELETE_STAGING = os.getenv("DELETE_STAGING", 'no')
#print("Source bucket is:" + SOURCE_BUCKET)
#print("Destination bucket is:" + DESTINATION_BUCKET)
record = event['Records'][0]
messageBucket = record['Sns']['MessageAttributes']['bucket']
print("The messageBucket value is:")
print(messageBucket['Value'])
if (messageBucket['Value'] == SOURCE_BUCKET or SOURCE_BUCKET == 'any'):
message = json.loads(record['Sns']['Message'])
#print("The message content is:" + str(message))
#print("The message key is: " + message['key'])
s3 = boto3.resource('s3')
copy_source = {
'Bucket': messageBucket['Value'],
'Key': message['key']
}
if 'PartsCount' in s3.meta.client.head_object(Bucket=messageBucket['Value'], Key=message['key'], PartNumber=1):
# get the tags, then copy with tags specified
#print("doing a multipart copy with tags")
try:
tagging = s3.meta.client.get_object_tagging(Bucket=messageBucket['Value'], Key=message['key'])
#print("get object tagging = " + str(tagging))
s3.meta.client.copy(copy_source, DESTINATION_BUCKET, message['key'], ExtraArgs={'Tagging': parse.urlencode({tag['Key']: tag['Value'] for tag in tagging['TagSet']})})
except Exception as e:
print(e)
raise(e)
else:
# copy as normal
#print("doing a normal copy")
try:
s3.meta.client.copy_object(CopySource=copy_source, Bucket=DESTINATION_BUCKET, Key=message['key'])
except Exception as e:
print(e)
raise(e)
print("Copied: " + message['key'] + " to production bucket: " + DESTINATION_BUCKET)
#print("Delete files: " + DELETE_STAGING)
if (DELETE_STAGING == 'yes'):
try:
s3.meta.client.delete_object(Bucket=SOURCE_BUCKET, Key=message['key'])
print("Deleted: " + message['key'] + " from source bucket: " + SOURCE_BUCKET)
except Exception as e:
print(e)
raise(e)
return {
'statusCode': 200,
'body': json.dumps('Non-infected object moved to production bucket')
}
return {
'statusCode': 200,
'body': json.dumps('Not from the Staging bucket')
}
except ClientError as e:
return {
'statusCode': 400,
'body': "Unexpected error: %s" % e}
except ParamValidationError as e:
return {
'statusCode': 400,
'body': "Parameter validation error: %s" % e}
Make Adjustments to Lambda Settings
There are some adjustments to the Lambda you'll probably need to make:
- Set the environment variables for the Staging Bucket and Production Bucket names as seen here:
Field | Description |
|---|---|
DELETE_STAGING | yes or no value based on whether you want the original file to be deleted after the copy has occurred |
DESTINATION_BUCKET | Clean or Production bucket name identifying where to copy clean files to |
SOURCE_BUCKET | Dirty or Staging bucket name identifying the originating bucket
Note: you can leverage SNS Topic Filtering to eliminate the need for this value and the if check inside the code |
- Change the Time Out under General Configuration to a value that will work for the typical file sizes you deal with. The larger the file size, the longer you may want to make it so the lambda doesn't time out before the copy finishes.
Permissions to add to Lambda Role
Add an Inline Policy to the Lambda Role that was created when you created the Lambda. Paste the below into the JSON screen and change the sections that have
<> to match your staging bucket name and production destination buckets{
"Version": "2012-10-17",
"Statement": [
{
"Sid": "VisualEditor0",
"Effect": "Allow",
"Action": [
"s3:DeleteObjectTagging",
"s3:GetObject",
"s3:GetObjectTagging",
"s3:ListBucket",
"s3:DeleteObject"
],
"Resource": [
"arn:aws:s3:::<staging bucket name>",
"arn:aws:s3:::<staging bucket name>/*"
]
},
{
"Sid": "VisualEditor1",
"Effect": "Allow",
"Action": [
"s3:PutObject",
"s3:ListBucket",
"s3:PutObjectTagging"
],
"Resource": [
"arn:aws:s3:::<production destination bucket name>/*",
"arn:aws:s3:::<production destination bucket name>"
]
}
]
}
Subscription Filter for Clean Results
Go to the SNS Topic itself and find the subscription created for the Lambda. Edit the filter settings by pasting the below value in. You can modify the filter further to include more scan results and even filter down by bucket. Bucket filtering is a good idea if you have event based scanning setup for any other bucket and you do not want the lambda to copy those clean files over as well.
{
"notificationType": [
"ScanResult"
],
"scanResult": [
"Clean"
]
}
Alternatively, with bucket filtering:
{
"notificationType": [
"ScanResult"
],
"scanResult": [
"Clean"
],
"bucket": [
"<staging bucket name>"
]
}
How the 2 Bucket System Flows:
Port 443 for:
- Outbound for Lambda calls
- Outbound Console and Agent access to Elastic Container Repository (ECR)
- Inbound access to Console for public access
- Public access is not required as long as you have access via private IP
Port 80, 53 and high range (1024:65535) ports for:
- AV signature updates
You can now setup local signature updates rather than reach out over the internet. This will allow you to setup an Amazon S3 bucket for the solution to look at.
You can get a more detailed view and additional options for routing on the Deployment Details page. In either the standard deployment or the VPC Endpoints deployment, with local signature updates you can remove all non-AWS calls from the application run space. With VPC Endpoint you can remove almost all public calls as well.
Yes. The Console Settings page gives you the option to modify the inboud Security Group rules, the VPC and Subnets and the specs of the task (vCPU and Memory). The Agent Settings page allows you to change the VPC and Subnets the agents run in, the specs of the task (vCPU and Memory) as well as all the scaling configuration aspects.
Neither. Antivirus for Amazon S3 infrastructure is built around AWS Fargate containers. We wanted to be serverless like Lambda and faster and more flexible than EC2s. Fargate containers give you a persistance and other benefits that Lambdas aren't prepared to give you yet. We explored Lambda and do see some advantages there, but not enough to win out over AWS Fargate containers.
We do leverage two lambdas for the subdomain registration, but not for any of the workload at this time. If you are interested in a lambda-driven solution, please Contact Us to let us know. We are always exploring the best way to build and run our solution.
A landing zone is a well-architected, multi-account AWS environment that's based on security and compliance best practices. AWS Control Tower automates the setup of a new landing zone using best-practices blueprints for identity, federated access, and account structure.
Antivirus for Amazon S3 is not currently tightly integrated with AWS Control Tower, but is designed to work within the landing zone context. Antivirus for Amazon S3 can be centrally deployed in a
Security Services account while leveraging Linked Accounts to scan all other accounts.Yes you can leverage SSO with our solution. Antivirus for Amazon S3 utilizes Amazon Cognito for user management. Amazon Cognito allows SAML integrations. Leveraging this capability we can utilize various providers as part of SSO into our solution, both from the SSO Dashboard as well as from within the application itself.
It is a fairly simple task to setup the connection and AWS has been kind enough to document how to do it. Follow the instructions found here: How do I set up Okta as a SAML identity provider in an Amazon Cognito user pool?. The sample is for Okta, but performing the corresponding steps for whichever SSO you use should also work.
Okta Examples below:
Upon customer request, we have documented the steps to get GSuite working as your SSO provider. Please leverage the document below. We have had other customers leverage these steps (along with the Okta write up) to setup other providers as well such as Keycloak.
GSuite Setup Instructions
Okta identities will be auto-created within Amazon Cognito (and therefore Antivirus for Amazon S3) as simple
Users and not Admins. They are also not assigned to a particular Group. This state enables them to login, but manage nothing. One-time only you will need to assign the user to a group and the admin role. After this initial assignment, each subsequent login will allow for proper management.SSO users will also standout as their username will be created from the SSO sign in process.
Last modified 25d ago