79 lines
4.0 KiB
Diff
79 lines
4.0 KiB
Diff
|
From 4bc84bccec88abcd9bd0b840532b23ed0934c0d4 Mon Sep 17 00:00:00 2001
|
||
|
|
From: Jason McCloskey <jamscclo@amazon.com>
|
||
|
|
Date: Mon, 10 Feb 2020 15:18:30 -0800
|
||
|
|
Subject: [PATCH] Allow user to specify role with which to query/update route
|
||
|
|
table
|
||
|
|
|
||
|
|
Allow user to specify role with which to query/update route table
|
||
|
|
|
||
|
|
Allow user to specify role with which to query/update route table
|
||
|
|
---
|
||
|
|
heartbeat/aws-vpc-move-ip | 40 +++++++++++++++++++++++++++++++++------
|
||
|
|
1 file changed, 34 insertions(+), 6 deletions(-)
|
||
|
|
|
||
|
|
diff --git a/heartbeat/aws-vpc-move-ip b/heartbeat/aws-vpc-move-ip
|
||
|
|
index b27d0bfcd..23eec18cc 100755
|
||
|
|
--- a/heartbeat/aws-vpc-move-ip
|
||
|
|
+++ b/heartbeat/aws-vpc-move-ip
|
||
|
|
@@ -145,6 +145,24 @@ Enable enhanced monitoring using AWS API calls to check route table entry
|
||
|
|
END
|
||
|
|
}
|
||
|
|
|
||
|
|
+
|
||
|
|
+execute_cmd_as_role(){
|
||
|
|
+ cmd=$1
|
||
|
|
+ role=$2
|
||
|
|
+ output="$(aws sts assume-role --role-arn $role --role-session-name AWSCLI-RouteTableUpdate --profile $OCF_RESKEY_profile --output=text)"
|
||
|
|
+ export AWS_ACCESS_KEY_ID="$(echo $output | awk -F" " '$4=="CREDENTIALS" {print $5}')"
|
||
|
|
+ export AWS_SECRET_ACCESS_KEY="$(echo $output | awk -F" " '$4=="CREDENTIALS" {print $7}')"
|
||
|
|
+ export AWS_SESSION_TOKEN="$(echo $output | awk -F" " '$4=="CREDENTIALS" {print $8}')"
|
||
|
|
+
|
||
|
|
+ #Execute command
|
||
|
|
+ ocf_log debug "Assumed Role ${role}"
|
||
|
|
+ ocf_log debug "$(aws sts get-caller-identity)"
|
||
|
|
+ ocf_log debug "executing command: $cmd"
|
||
|
|
+ response="$($cmd)"
|
||
|
|
+ unset output AWS_ACCESS_KEY_ID AWS_SECRET_ACCESS_KEY AWS_SESSION_TOKEN
|
||
|
|
+ echo $response
|
||
|
|
+}
|
||
|
|
+
|
||
|
|
ec2ip_set_address_param_compat(){
|
||
|
|
# Include backward compatibility for the deprecated address parameter
|
||
|
|
if [ -z "$OCF_RESKEY_ip" ] && [ -n "$OCF_RESKEY_address" ]; then
|
||
|
|
@@ -177,9 +195,14 @@ ec2ip_monitor() {
|
||
|
|
if ocf_is_true ${OCF_RESKEY_monapi} || [ "$__OCF_ACTION" = "start" ] || ocf_is_probe; then
|
||
|
|
for rtb in $(echo $OCF_RESKEY_routing_table | sed -e 's/,/ /g'); do
|
||
|
|
ocf_log info "monitor: check routing table (API call) - $rtb"
|
||
|
|
- cmd="$OCF_RESKEY_awscli --profile $OCF_RESKEY_profile --output text ec2 describe-route-tables --route-table-ids $rtb --query RouteTables[*].Routes[?DestinationCidrBlock=='$OCF_RESKEY_ip/32'].InstanceId"
|
||
|
|
- ocf_log debug "executing command: $cmd"
|
||
|
|
- ROUTE_TO_INSTANCE="$($cmd)"
|
||
|
|
+ if [[ -z "${OCF_RESKEY_routing_table_role}" ]]; then
|
||
|
|
+ cmd="$OCF_RESKEY_awscli --profile $OCF_RESKEY_profile --output text ec2 describe-route-tables --route-table-ids $rtb --query RouteTables[*].Routes[?DestinationCidrBlock=='$OCF_RESKEY_ip/32'].InstanceId"
|
||
|
|
+ ocf_log debug "executing command: $cmd"
|
||
|
|
+ ROUTE_TO_INSTANCE="$($cmd)"
|
||
|
|
+ else
|
||
|
|
+ cmd="$OCF_RESKEY_awscli --output text ec2 describe-route-tables --route-table-ids $rtb --query RouteTables[*].Routes[?DestinationCidrBlock=='$OCF_RESKEY_ip/32'].InstanceId"
|
||
|
|
+ ROUTE_TO_INSTANCE="$(execute_cmd_as_role "$cmd" $OCF_RESKEY_routing_table_role)"
|
||
|
|
+ fi
|
||
|
|
ocf_log debug "Overlay IP is currently routed to ${ROUTE_TO_INSTANCE}"
|
||
|
|
if [ -z "$ROUTE_TO_INSTANCE" ]; then
|
||
|
|
ROUTE_TO_INSTANCE="<unknown>"
|
||
|
|
@@ -253,9 +276,14 @@ ec2ip_get_and_configure() {
|
||
|
|
ocf_log debug "network interface id associated MAC address ${MAC_ADDR}: ${EC2_NETWORK_INTERFACE_ID}"
|
||
|
|
|
||
|
|
for rtb in $(echo $OCF_RESKEY_routing_table | sed -e 's/,/ /g'); do
|
||
|
|
- cmd="$OCF_RESKEY_awscli --profile $OCF_RESKEY_profile --output text ec2 replace-route --route-table-id $rtb --destination-cidr-block ${OCF_RESKEY_ip}/32 --network-interface-id $EC2_NETWORK_INTERFACE_ID"
|
||
|
|
- ocf_log debug "executing command: $cmd"
|
||
|
|
- $cmd
|
||
|
|
+ if [[ -z "${OCF_RESKEY_routing_table_role}" ]]; then
|
||
|
|
+ cmd="$OCF_RESKEY_awscli --profile $OCF_RESKEY_profile --output text ec2 replace-route --route-table-id $rtb --destination-cidr-block ${OCF_RESKEY_ip}/32 --network-interface-id $EC2_NETWORK_INTERFACE_ID"
|
||
|
|
+ ocf_log debug "executing command: $cmd"
|
||
|
|
+ $cmd
|
||
|
|
+ else
|
||
|
|
+ cmd="$OCF_RESKEY_awscli --output text ec2 replace-route --route-table-id $rtb --destination-cidr-block ${OCF_RESKEY_ip}/32 --network-interface-id $EC2_NETWORK_INTERFACE_ID"
|
||
|
|
+ update_response="$(execute_cmd_as_role "$cmd" $OCF_RESKEY_routing_table_role)"
|
||
|
|
+ fi
|
||
|
|
rc=$?
|
||
|
|
if [ "$rc" != 0 ]; then
|
||
|
|
ocf_log warn "command failed, rc: $rc"
|