support GCP metadata

Author: complexsplit

hardware/hardware.sh

@@ -130,8 +130,20 @@ else
 fi
 echo
 
-echo "Instance type from IMDS (if available):"
-curl -s --connect-timeout 1 'http://169.254.169.254/latest/meta-data/instance-type' | tee instance.txt
+echo "Instance type from Metadata service (if available):"
+AWS_HTTP_CODE=$(curl -s -o /dev/null -w "%{http_code}" --connect-timeout 1 'http://169.254.169.254/latest/meta-data/')
+if [[ "$AWS_HTTP_CODE" == "200" ]]; then
+    AWS_INSTANCE_TYPE=$(curl -s --connect-timeout 1 'http://169.254.169.254/latest/meta-data/instance-type')
+    echo "$AWS_INSTANCE_TYPE" > instance.txt
+    exit 0
+fi
+
+GCP_HTTP_CODE=$(curl -s -o /dev/null -w "%{http_code}" --connect-timeout 1 -H "Metadata-Flavor: Google" 'http://metadata.google.internal/computeMetadata/v1/instance/')
+if [[ "$GCP_HTTP_CODE" == "200" ]]; then
+    GCP_INSTANCE_TYPE=$(curl -s --connect-timeout 1 -H "Metadata-Flavor: Google" 'http://metadata.google.internal/computeMetadata/v1/instance/machine-type' | awk -F/ '{print $NF}')
+    echo "$GCP_INSTANCE_TYPE" > instance.txt
+    exit 0
+fi
 echo
 
 echo "Uploading the results (if possible)"