Skip to content

hostirosti/trusted-execution-aggregation-service

 
 

Repository files navigation

Set up Aggregation Service for Aggregatable Reports

[NOTE] The latest aggregatable reports generated with Chrome version 104+ (currently available in Beta) are not yet compatible with the LocalTestingTool or the Aggregation Service running on AWS. We will update both shortly to support the new report format.

This repository contains instructions and scripts to set up and test the Aggregation Service for Aggregatable Reports locally and on Amazon Web Services Nitro Enclaves. If you want to learn more about the Privacy Sandbox Aggregation Service for the Attribution Reporting API, aggregatable, and summary reports click, read the Aggregation Service proposal.

Set up local testing

You can process aggregatable debug reports locally with the LocalTestingTool.jar into summary reports. Learn how to setup debug reports.

Disclaimer: encrypted reports can not be processed with the local testing tool!

Using the local testing tool

Download the local testing tool. You'll need Java JRE installed to use the tool.

The SHA256 of the LocalTestingTool_{version}.jar is 9d9ee93f0bf0750d549728deee30c5e9e353b49ddbae54ed6ac9c5e918bdeb4c obtained with openssl sha256 <jar>.

Follow the instructions on how to collect and batch aggregatable reports. Create an output domain file: output_domain.avro. For testing you can use our sample debug batch with the corresponding output domain avro.

To aggregate the resulting avro batch output_debug_reports.avro file into a summary report in the same directory where you run the tool, run the following command:

java -jar LocalTestingTool.jar \
--input_data_avro_file output_debug_reports.avro \
--domain_avro_file output_domain.avro \
--output_directory .

To see all supported flags for the local testing tool run java -jar LocalTestingTool.jar --help, e.g. you can adjust the noising epsilon with the --epsilon flag or disable noising all together with the --no_noising flag. See all flags and descriptions.

Test on AWS with support for encrypted reports

General Notes

Privacy Budget Enforcement

The no-duplicate rule is not enforced in the current test version but will be enforced in the future. We recommend users design their systems keeping the no-duplicate rule in consideration.

Prerequisites

To test the aggregation service with support for encrypted reports, you need the following:

Once you’ve submitted the onboarding form, we will contact you to verify your information. Then, we’ll send you the remaining instructions and information needed for this setup.
You won't be able to successfully setup your AWS system without registering for the origin trial and completing the onboarding process!

To set up aggregation service in AWS you'll use Terraform.

Clone the repository

Clone the repository into a local folder <repostory_root>:

git clone https://github.com/google/trusted-execution-aggregation-service;
cd trusted-execution-aggregation-service

Set up AWS client

Make sure you install and set up the latest AWS client.

Set up Terraform

Change into the <repository_root>/terraform/aws folder.

The setup scripts require terraform version 1.0.4. You can download Terraform version 1.0.4 from https://releases.hashicorp.com/terraform/1.0.4/ or at your own risk, you can install and use Terraform version manager instead.

If you have the Terraform version manager tfenv installed, run the following in your <repository_root> to set Terraform to version 1.0.4.

tfenv install 1.0.4;
tfenv use 1.0.4

We recommend you store the Terraform state in a cloud bucket. Create a S3 bucket via the console/cli, which we'll reference as tf_state_bucket_name.

Download dependencies

The Terraform scripts depend on 5 packaged jars for Lambda functions deployment. These jars are hosted on Google Cloud Storage (https://storage.googleapis.com/trusted-execution-aggregation-service-public-artifacts/{version}/{jar_file}) and can be downloaded with the <repository_root>/terraform/aws/download_dependencies.sh script. The downloaded jars will be stored in <repository_root>/terraform/aws/jars. License information of downloaded dependencies can be found in the DEPENDENCIES.md

Run the following script in the <repository_root>/terraform/aws folder.

sh ./download_dependencies.sh

For manual download into the <repository_root>/terraform/aws/jars folder you can download them from the links below. The sha256 was obtained with openssl sha256 <jar>.

jar download link sha256
AsgCapacityHandlerLambda_0.2.0.jar be5ed5bca082d9283a495c032d7a13a0df0204808636d4a5049f780ebdfc1cac
AwsChangeHandlerLambda_0.2.0.jar 58c31f1435c7c6dc1b2f418d4dbda195d2c5841afbd994a7a72af5320133e4f4
AwsFrontendCleanupLambda_0.2.0.jar de91b523bffd618465549186a5dfb3c06fd44ae4c3d09c1be09e26d60ce7e3f4
TerminatedInstanceHandlerLambda_0.2.0.jar 307beb1128ae61e5cc21b8a29a8a259f04eb29ae7cb653774e1e91fcf0f81035
aws_apigateway_frontend_0.2.0.jar 28e258193b947b227eafe9b6601f346466ee4e1c394e02e3be516c8a9c70efb9

Set up your deployment environment

We use the following folder structure <repository_root>/terraform/aws/environments/<environment_name> to separate deployment environments.

To set up your first environment (e.g dev), copy the demo environment. Run the following commands from the <repository_root>/terraform/aws/environments folder:

cp -R demo dev
cd dev

Make the following adjustments in the <repository_root>/terraform/aws/environments/dev folder:

  1. Add the tf_state_bucket_name to your main.tf by uncommenting and replacing the values using <...>:

    # backend "s3" {
    #   bucket = "<tf_state_bucket_name>"
    #   key    = "<environment_name>.tfstate"
    #   region = "us-east-1"
    # }
  2. Rename example.auto.tfvars to <environment>.auto.tfvars and adjust the values with <...> using the information you received in the onboarding email. Leave all other values as-is for the initial deployment.

    environment = "<environment_name>"
    ...
    
    assume_role_parameter = "<arn:aws:iam::example:role/example>"
    
    ...
    
    alarm_notification_email = "<[email protected]>"
    • environment: name of your environment
    • assume_role_parameter: IAM role given by us in the onboarding email
    • alarm_notification_email: Email to receive alarm notifications. Requires confirmation subscription through sign up email sent to this address.
  3. Once you’ve adjusted the configuration, run the following in the <repository_root>/terraform/aws/environments/dev folder

    Install all Terraform modules:

    terraform init

    Get an infrastructure setup plan:

    terraform plan

    If you see the following output on a fresh project:

    ...
    Plan: 128 to add, 0 to change, 0 to destroy.

    you can continue to apply the changes (needs confirmation after the planning step)

    terraform apply

    If your see the following output, your setup was successful:

    ...
    Apply complete! Resources: 127 added, 0 changed, 0 destroyed.
    
    Outputs:
    
    create_job_endpoint = "POST https://xyz.execute-api.us-east-1.amazonaws.com/stage/v1alpha/createJob"
    frontend_api_endpoint = "https://xyz.execute-api.us-east-1.amazonaws.com"
    frontend_api_id = "xyz"
    get_job_endpoint = "GET https://xyz.execute-api.us-east-1.amazonaws.com/stage/v1alpha/getJob"

    The output has the links to the createJob and getJob API endpoints. These are authenticated endpoints, refer to the Testing the System section to learn how to use them.

    If you run into any issues during deployment of your system, please consult the Troubleshooting and Support sections.

Testing the system

To test the system, you'll need encrypted aggregatable reports in avro batch format (follow the collecting and batching instructions) accessible by the aggregation service.

  1. Create an S3 bucket for your input and output data, we will refer to it as data_bucket. This bucket must be created in the same AWS account where you set up the aggregation service.

  2. Copy your reports.avro with batched encrypted aggregatable reports to <data_bucket>/input. To experiment with sample data, you can use our sample batch with the corresponding output domain avro.

  3. Create an aggregation job with the createJob API.

    POST https://<frontend_api_id>.execute-api.us-east-1.amazonaws.com/stage/v1alpha/createJob

    {
        "input_data_blob_prefix": "input/reports.avro",
        "input_data_bucket_name": "<data_bucket>",
        "output_data_blob_prefix": "output/summary_report.avro",
        "output_data_bucket_name": "<data_bucket>",
        "job_parameters": {
            "attribution_report_to": "<your_attribution_domain>",
            "output_domain_blob_prefix": "domain/domain.avro",
            "output_domain_bucket_name": "<data_bucket>"
        },
        "job_request_id": "test01"
    }

    Note: This API requires authentication. Follow the AWS instructions for sending an authenticated request.

  4. Check the status of your job with the getJob API, replace values in <...>

    GET https://<frontend_api_id>.execute-api.us-east-1.amazonaws.com/stage/v1alpha/getJob?job_request_id=test01 Note: This API requires authentication. Follow the AWS instructions for sending an authenticated request. Detailed API spec

Collect and batch aggregatable reports

Both the local testing tool and the aggregation service running on AWS Nitro Enclave expect aggregatable reports batched in the following Avro format.

{
  "type": "record",
  "name": "AggregatableReport",
  "fields": [
    {
      "name": "payload",
      "type": "bytes"
    },
    {
      "name": "key_id",
      "type": "string"
    },
    {
      "name": "shared_info",
      "type": "string"
    }
  ]
}

Additionally an output domain file is needed to declare all expected aggregation keys for aggregating the aggregatable reports (keys not listed in the domain file won't be aggregated)

{
 "type": "record",
 "name": "AggregationBucket",
 "fields": [
   {
     "name": "bucket",
     "type": "bytes"
     /* A single bucket that appears in
     the aggregation service output.
     128-bit integer encoded as a
     16-byte big-endian byte string. */
   }
 ]
}

Review code snippets which demonstrate how to collect and batch aggregatable reports.

Troubleshooting

The following error message points to a potential lack of instance availability. If you encounter this situation, run terraform destroy to remove your deployment and run terraform apply again.

Error: Error creating Auto Scaling Group: ValidationError: You must use a valid 
fully-formed launch template. Your requested instance type (m5.2xlarge) is not
supported in your requested Availability Zone (us-east-1e).
Please retry your request by not specifying an Availability Zone or choosing
us-east-1a, us-east-1b, us-east-1c, us-east-1d, us-east-1f.

Support

You can reach out to us for support through creating issues on this repository or sending us an email at aggregation-service-support<at>google.com. This address is monitored and only visible to selected support staff.

General security notes

  • The VPC subnet property map_public_ip_on_launch is currently set to true which assigns a public IP address to all instances in the subnet. This allows for easier console access, yet is considered a risk and will be addressed in a future release.
  • The worker VPC security group currently allows for inbound connections on port 22 from any source IP. This is considered a risk and will be addressed in a future release.

License

Apache 2.0 - See LICENSE for more information.

FAQ

Where should I post feedback/questions, this repo or the Attribution API repo?

This repo hosts an implementation of the Attribution Reporting API. For feedback/questions encountered during using this particular aggregation service implementation, please use the support channels provided by this repo. For feedback/requests related to the APIs in general, please initiate discussions in the Attribution Reporting API repo.

About

This repository contains instructions and scripts to set up and test the Privacy Sandbox Aggregation Service

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • HCL 99.3%
  • Shell 0.7%