diff --git a/events/README.md b/events/README.md index 5c3ccdc9..1b43082e 100644 --- a/events/README.md +++ b/events/README.md @@ -12,6 +12,8 @@ This package provides input types for Lambda functions that process AWS events. [AppSync](README_AppSync.md) +[ClientVPN Connection Handler](README_ClientVPN.md) + [CloudFormation Events](../cfn/README.md) [CloudWatch Events](README_CloudWatch_Events.md) diff --git a/events/README_ClientVPN.md b/events/README_ClientVPN.md new file mode 100644 index 00000000..d375bb10 --- /dev/null +++ b/events/README_ClientVPN.md @@ -0,0 +1,59 @@ +# Sample Function + +The following is a sample Lambda function that receives a Client VPN connection handler request as an input and then validates the IP address input and checks whether the connection source IP is on the allowed list defined as a map inside the function. If the source IP matches an allowed IP address it allows the access, otherwise an error message is presented to the user. Debug logs are generated to CloudWatch Logs. (Note that by default anything written to Console will be logged as CloudWatch Logs events.) + +```go +import ( + "fmt" + "log" + "net" + + "encoding/json" + + "github.com/aws/aws-lambda-go/events" + "github.com/aws/aws-lambda-go/lambda" +) + +var ( + AllowedIPs = map[string]bool{ + "10.11.12.13": true, + } +) + +func handler(request events.ClientVPNConnectionHandlerRequest) (events.ClientVPNConnectionHandlerResponse, error) { + requestJson, _ := json.MarshalIndent(request, "", " ") + log.Printf("REQUEST: %s", requestJson) + + sourceIP := request.PublicIP + if net.ParseIP(sourceIP) == nil { + return events.ClientVPNConnectionHandlerResponse{}, fmt.Errorf("Invalid parameter PublicIP passed in request: '%s'", sourceIP) + } + + log.Printf("SOURCE IP: %s", sourceIP) + + postureCompliance := []string{} + + allowed, ok := AllowedIPs[sourceIP] + if !ok { + allowed = false + postureCompliance = []string{"BlockedSourceIP"} + } + + if allowed { + log.Printf("Allowing access from: %s", sourceIP) + } else { + log.Printf("Blocking access from: %s", sourceIP) + } + + return events.ClientVPNConnectionHandlerResponse{ + Allow: allowed, + ErrorMsgOnFailedPostureCompliance: "You're accessing from a blocked IP range.", + PostureComplianceStatuses: postureCompliance, + SchemaVersion: "v1", + }, nil +} + +func main() { + lambda.Start(handler) +} +```