enhance cloud provider detection

appcelerator-archive · Jan 5, 2018 · b8e8243 · b8e8243
1 parent 6e961a3
commit b8e8243
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pkg/cloud/provider.go b/pkg/cloud/provider.go
@@ -25,6 +25,9 @@ const (
 func CloudProvider() (Provider, error) {
 	dataLen := 3
 	uuidFile, err := os.Open("/sys/hypervisor/uuid")
+	if err != nil {
+		uuidFile, err = os.Open("/sys/devices/virtual/dmi/id/product_uuid")
+	}
 	if err != nil {
 		// file does not exist, so we'll consider it's not a cloud deployment
 		return ProviderLocal, nil
@@ -40,7 +43,7 @@ func CloudProvider() (Provider, error) {
 		return ProviderLocal, nil
 	}
 	switch string(data) {
-	case "ec2":
+	case "ec2", "EC2":
 		return ProviderAWS, nil
 	default:
 		return ProviderUnknown, fmt.Errorf("found unexpected value in uuid file: %s", string(data))