mirror of https://github.com/k3s-io/k3s
69 lines
2.6 KiB
Go
69 lines
2.6 KiB
Go
|
// +build windows
|
||
|
|
|
||
|
|
/*
|
||
|
|
Copyright 2019 The Kubernetes Authors.
|
||
|
|
|
||
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
you may not use this file except in compliance with the License.
|
||
|
|
You may obtain a copy of the License at
|
||
|
|
|
||
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
|
||
|
|
Unless required by applicable law or agreed to in writing, software
|
||
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
See the License for the specific language governing permissions and
|
||
|
|
limitations under the License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
package awsebs
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"regexp"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// ebsnvme-id is present on AWS-provided Windows Server AMIs
|
||
|
|
// https://docs.aws.amazon.com/AWSEC2/latest/WindowsGuide/nvme-ebs-volumes.html#identify-nvme-ebs-device
|
||
|
|
const ebsnvmeID = `C:\ProgramData\Amazon\Tools\ebsnvme-id.exe`
|
||
|
|
|
||
|
|
func (attacher *awsElasticBlockStoreAttacher) getDevicePath(volumeID, partition, devicePath string) (string, error) {
|
||
|
|
return attacher.getDiskNumber(volumeID)
|
||
|
|
}
|
||
|
|
|
||
|
|
// getDiskNumber gets the Windows disk number for a given volume ID. The disk number is needed for mounting.
|
||
|
|
// TODO This only works for Nitro-based instances
|
||
|
|
// TODO fallback to Get-Disk
|
||
|
|
func (attacher *awsElasticBlockStoreAttacher) getDiskNumber(volumeID string) (string, error) {
|
||
|
|
// Split the ID from zone: aws://us-west-2b/vol-06d0909eb358b05f9
|
||
|
|
split := strings.Split(volumeID, "/")
|
||
|
|
volumeID = split[len(split)-1]
|
||
|
|
|
||
|
|
exec := attacher.host.GetExec(awsElasticBlockStorePluginName)
|
||
|
|
output, err := exec.Run(ebsnvmeID)
|
||
|
|
if err != nil {
|
||
|
|
return "", fmt.Errorf("error calling ebsnvme-id.exe: %v", err)
|
||
|
|
}
|
||
|
|
// ebsnvme-id.exe will output a list of disks in this format:
|
||
|
|
// ```
|
||
|
|
// Disk Number: 1
|
||
|
|
// Volume ID: vol-06d0909eb358b05f9
|
||
|
|
// Device Name: /dev/xvdch
|
||
|
|
// ```
|
||
|
|
// Don't try to match devicePath against "Device Name" not only because volume ID is sufficient,
|
||
|
|
// but because devicePath may change between Linux & Windows formats between WaitForAttach calls.
|
||
|
|
// The first attach and mount, WaitForAttach gets devicePath as the Linux format /dev/xvdch. Then
|
||
|
|
// WaitForAttach returns the disk number as the "right" devicePath and that is persisted to ASW.
|
||
|
|
// In subsequent mounts of the same disk, WaitForAttach gets devicePath as the Windows format it
|
||
|
|
// returned the first time.
|
||
|
|
diskRe := regexp.MustCompile(
|
||
|
|
`Disk Number: (\d+)\s*` +
|
||
|
|
`Volume ID: ` + volumeID + `\s*`)
|
||
|
|
matches := diskRe.FindStringSubmatch(string(output))
|
||
|
|
if len(matches) != 2 {
|
||
|
|
return "", fmt.Errorf("disk not found in ebsnvme-id.exe output: %q", string(output))
|
||
|
|
}
|
||
|
|
return matches[1], nil
|
||
|
|
}
|