manufacturer_windows.go 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. /*
  2. *
  3. * Copyright 2022 gRPC authors.
  4. *
  5. * Licensed under the Apache License, Version 2.0 (the "License");
  6. * you may not use this file except in compliance with the License.
  7. * You may obtain a copy of the License at
  8. *
  9. * http://www.apache.org/licenses/LICENSE-2.0
  10. *
  11. * Unless required by applicable law or agreed to in writing, software
  12. * distributed under the License is distributed on an "AS IS" BASIS,
  13. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14. * See the License for the specific language governing permissions and
  15. * limitations under the License.
  16. *
  17. */
  18. package googlecloud
  19. import (
  20. "errors"
  21. "os/exec"
  22. "regexp"
  23. "strings"
  24. )
  25. const (
  26. windowsCheckCommand = "powershell.exe"
  27. windowsCheckCommandArgs = "Get-WmiObject -Class Win32_BIOS"
  28. powershellOutputFilter = "Manufacturer"
  29. windowsManufacturerRegex = ":(.*)"
  30. )
  31. func manufacturer() ([]byte, error) {
  32. cmd := exec.Command(windowsCheckCommand, windowsCheckCommandArgs)
  33. out, err := cmd.Output()
  34. if err != nil {
  35. return nil, err
  36. }
  37. for _, line := range strings.Split(strings.TrimSuffix(string(out), "\n"), "\n") {
  38. if strings.HasPrefix(line, powershellOutputFilter) {
  39. re := regexp.MustCompile(windowsManufacturerRegex)
  40. name := re.FindString(line)
  41. name = strings.TrimLeft(name, ":")
  42. return []byte(name), nil
  43. }
  44. }
  45. return nil, errors.New("cannot determine the machine's manufacturer")
  46. }