mirror of https://github.com/k3s-io/k3s
420 lines
14 KiB
Go
420 lines
14 KiB
Go
// +build windows
|
|
|
|
/*
|
|
Copyright 2017 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 mount
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"path"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"k8s.io/klog"
|
|
"k8s.io/utils/keymutex"
|
|
|
|
utilpath "k8s.io/utils/path"
|
|
)
|
|
|
|
// Mounter provides the default implementation of mount.Interface
|
|
// for the windows platform. This implementation assumes that the
|
|
// kubelet is running in the host's root mount namespace.
|
|
type Mounter struct {
|
|
mounterPath string
|
|
}
|
|
|
|
// New returns a mount.Interface for the current system.
|
|
// It provides options to override the default mounter behavior.
|
|
// mounterPath allows using an alternative to `/bin/mount` for mounting.
|
|
func New(mounterPath string) Interface {
|
|
return &Mounter{
|
|
mounterPath: mounterPath,
|
|
}
|
|
}
|
|
|
|
// acquire lock for smb mount
|
|
var getSMBMountMutex = keymutex.NewHashed(0)
|
|
|
|
// Mount : mounts source to target with given options.
|
|
// currently only supports cifs(smb), bind mount(for disk)
|
|
func (mounter *Mounter) Mount(source string, target string, fstype string, options []string) error {
|
|
target = normalizeWindowsPath(target)
|
|
|
|
if source == "tmpfs" {
|
|
klog.V(3).Infof("mounting source (%q), target (%q), with options (%q)", source, target, options)
|
|
return os.MkdirAll(target, 0755)
|
|
}
|
|
|
|
parentDir := filepath.Dir(target)
|
|
if err := os.MkdirAll(parentDir, 0755); err != nil {
|
|
return err
|
|
}
|
|
|
|
klog.V(4).Infof("mount options(%q) source:%q, target:%q, fstype:%q, begin to mount",
|
|
options, source, target, fstype)
|
|
bindSource := source
|
|
|
|
// tell it's going to mount azure disk or azure file according to options
|
|
if bind, _, _ := isBind(options); bind {
|
|
// mount azure disk
|
|
bindSource = normalizeWindowsPath(source)
|
|
} else {
|
|
if len(options) < 2 {
|
|
klog.Warningf("mount options(%q) command number(%d) less than 2, source:%q, target:%q, skip mounting",
|
|
options, len(options), source, target)
|
|
return nil
|
|
}
|
|
|
|
// currently only cifs mount is supported
|
|
if strings.ToLower(fstype) != "cifs" {
|
|
return fmt.Errorf("only cifs mount is supported now, fstype: %q, mounting source (%q), target (%q), with options (%q)", fstype, source, target, options)
|
|
}
|
|
|
|
// lock smb mount for the same source
|
|
getSMBMountMutex.LockKey(source)
|
|
defer getSMBMountMutex.UnlockKey(source)
|
|
|
|
if output, err := newSMBMapping(options[0], options[1], source); err != nil {
|
|
if isSMBMappingExist(source) {
|
|
klog.V(2).Infof("SMB Mapping(%s) already exists, now begin to remove and remount", source)
|
|
if output, err := removeSMBMapping(source); err != nil {
|
|
return fmt.Errorf("Remove-SmbGlobalMapping failed: %v, output: %q", err, output)
|
|
}
|
|
if output, err := newSMBMapping(options[0], options[1], source); err != nil {
|
|
return fmt.Errorf("New-SmbGlobalMapping remount failed: %v, output: %q", err, output)
|
|
}
|
|
} else {
|
|
return fmt.Errorf("New-SmbGlobalMapping failed: %v, output: %q", err, output)
|
|
}
|
|
}
|
|
}
|
|
|
|
if output, err := exec.Command("cmd", "/c", "mklink", "/D", target, bindSource).CombinedOutput(); err != nil {
|
|
klog.Errorf("mklink failed: %v, source(%q) target(%q) output: %q", err, bindSource, target, string(output))
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// do the SMB mount with username, password, remotepath
|
|
// return (output, error)
|
|
func newSMBMapping(username, password, remotepath string) (string, error) {
|
|
if username == "" || password == "" || remotepath == "" {
|
|
return "", fmt.Errorf("invalid parameter(username: %s, password: %s, remoteapth: %s)", username, password, remotepath)
|
|
}
|
|
|
|
// use PowerShell Environment Variables to store user input string to prevent command line injection
|
|
// https://docs.microsoft.com/en-us/powershell/module/microsoft.powershell.core/about/about_environment_variables?view=powershell-5.1
|
|
cmdLine := `$PWord = ConvertTo-SecureString -String $Env:smbpassword -AsPlainText -Force` +
|
|
`;$Credential = New-Object -TypeName System.Management.Automation.PSCredential -ArgumentList $Env:smbuser, $PWord` +
|
|
`;New-SmbGlobalMapping -RemotePath $Env:smbremotepath -Credential $Credential`
|
|
cmd := exec.Command("powershell", "/c", cmdLine)
|
|
cmd.Env = append(os.Environ(),
|
|
fmt.Sprintf("smbuser=%s", username),
|
|
fmt.Sprintf("smbpassword=%s", password),
|
|
fmt.Sprintf("smbremotepath=%s", remotepath))
|
|
|
|
output, err := cmd.CombinedOutput()
|
|
return string(output), err
|
|
}
|
|
|
|
// check whether remotepath is already mounted
|
|
func isSMBMappingExist(remotepath string) bool {
|
|
cmd := exec.Command("powershell", "/c", `Get-SmbGlobalMapping -RemotePath $Env:smbremotepath`)
|
|
cmd.Env = append(os.Environ(), fmt.Sprintf("smbremotepath=%s", remotepath))
|
|
_, err := cmd.CombinedOutput()
|
|
return err == nil
|
|
}
|
|
|
|
// remove SMB mapping
|
|
func removeSMBMapping(remotepath string) (string, error) {
|
|
cmd := exec.Command("powershell", "/c", `Remove-SmbGlobalMapping -RemotePath $Env:smbremotepath -Force`)
|
|
cmd.Env = append(os.Environ(), fmt.Sprintf("smbremotepath=%s", remotepath))
|
|
output, err := cmd.CombinedOutput()
|
|
return string(output), err
|
|
}
|
|
|
|
// Unmount unmounts the target.
|
|
func (mounter *Mounter) Unmount(target string) error {
|
|
klog.V(4).Infof("azureMount: Unmount target (%q)", target)
|
|
target = normalizeWindowsPath(target)
|
|
if output, err := exec.Command("cmd", "/c", "rmdir", target).CombinedOutput(); err != nil {
|
|
klog.Errorf("rmdir failed: %v, output: %q", err, string(output))
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// List returns a list of all mounted filesystems. todo
|
|
func (mounter *Mounter) List() ([]MountPoint, error) {
|
|
return []MountPoint{}, nil
|
|
}
|
|
|
|
// IsMountPointMatch determines if the mountpoint matches the dir
|
|
func (mounter *Mounter) IsMountPointMatch(mp MountPoint, dir string) bool {
|
|
return mp.Path == dir
|
|
}
|
|
|
|
// IsNotMountPoint determines if a directory is a mountpoint.
|
|
func (mounter *Mounter) IsNotMountPoint(dir string) (bool, error) {
|
|
return isNotMountPoint(mounter, dir)
|
|
}
|
|
|
|
// IsLikelyNotMountPoint determines if a directory is not a mountpoint.
|
|
func (mounter *Mounter) IsLikelyNotMountPoint(file string) (bool, error) {
|
|
stat, err := os.Lstat(file)
|
|
if err != nil {
|
|
return true, err
|
|
}
|
|
// If current file is a symlink, then it is a mountpoint.
|
|
if stat.Mode()&os.ModeSymlink != 0 {
|
|
target, err := os.Readlink(file)
|
|
if err != nil {
|
|
return true, fmt.Errorf("readlink error: %v", err)
|
|
}
|
|
exists, err := mounter.ExistsPath(target)
|
|
if err != nil {
|
|
return true, err
|
|
}
|
|
return !exists, nil
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
// GetDeviceNameFromMount given a mnt point, find the device
|
|
func (mounter *Mounter) GetDeviceNameFromMount(mountPath, pluginDir string) (string, error) {
|
|
return getDeviceNameFromMount(mounter, mountPath, pluginDir)
|
|
}
|
|
|
|
// getDeviceNameFromMount find the device(drive) name in which
|
|
// the mount path reference should match the given plugin directory. In case no mount path reference
|
|
// matches, returns the volume name taken from its given mountPath
|
|
func getDeviceNameFromMount(mounter Interface, mountPath, pluginDir string) (string, error) {
|
|
refs, err := mounter.GetMountRefs(mountPath)
|
|
if err != nil {
|
|
klog.V(4).Infof("GetMountRefs failed for mount path %q: %v", mountPath, err)
|
|
return "", err
|
|
}
|
|
if len(refs) == 0 {
|
|
return "", fmt.Errorf("directory %s is not mounted", mountPath)
|
|
}
|
|
basemountPath := normalizeWindowsPath(path.Join(pluginDir, MountsInGlobalPDPath))
|
|
for _, ref := range refs {
|
|
if strings.Contains(ref, basemountPath) {
|
|
volumeID, err := filepath.Rel(normalizeWindowsPath(basemountPath), ref)
|
|
if err != nil {
|
|
klog.Errorf("Failed to get volume id from mount %s - %v", mountPath, err)
|
|
return "", err
|
|
}
|
|
return volumeID, nil
|
|
}
|
|
}
|
|
|
|
return path.Base(mountPath), nil
|
|
}
|
|
|
|
// DeviceOpened determines if the device is in use elsewhere
|
|
func (mounter *Mounter) DeviceOpened(pathname string) (bool, error) {
|
|
return false, nil
|
|
}
|
|
|
|
// PathIsDevice determines if a path is a device.
|
|
func (mounter *Mounter) PathIsDevice(pathname string) (bool, error) {
|
|
return false, nil
|
|
}
|
|
|
|
// MakeRShared checks that given path is on a mount with 'rshared' mount
|
|
// propagation. Empty implementation here.
|
|
func (mounter *Mounter) MakeRShared(path string) error {
|
|
return nil
|
|
}
|
|
|
|
// GetFileType checks for sockets/block/character devices
|
|
func (mounter *Mounter) GetFileType(pathname string) (FileType, error) {
|
|
return getFileType(pathname)
|
|
}
|
|
|
|
// MakeFile creates a new directory
|
|
func (mounter *Mounter) MakeDir(pathname string) error {
|
|
err := os.MkdirAll(pathname, os.FileMode(0755))
|
|
if err != nil {
|
|
if !os.IsExist(err) {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// MakeFile creates an empty file
|
|
func (mounter *Mounter) MakeFile(pathname string) error {
|
|
f, err := os.OpenFile(pathname, os.O_CREATE, os.FileMode(0644))
|
|
defer f.Close()
|
|
if err != nil {
|
|
if !os.IsExist(err) {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// ExistsPath checks whether the path exists
|
|
func (mounter *Mounter) ExistsPath(pathname string) (bool, error) {
|
|
return utilpath.Exists(utilpath.CheckFollowSymlink, pathname)
|
|
}
|
|
|
|
// EvalHostSymlinks returns the path name after evaluating symlinks
|
|
func (mounter *Mounter) EvalHostSymlinks(pathname string) (string, error) {
|
|
return filepath.EvalSymlinks(pathname)
|
|
}
|
|
|
|
func (mounter *SafeFormatAndMount) formatAndMount(source string, target string, fstype string, options []string) error {
|
|
// Try to mount the disk
|
|
klog.V(4).Infof("Attempting to formatAndMount disk: %s %s %s", fstype, source, target)
|
|
|
|
if err := ValidateDiskNumber(source); err != nil {
|
|
klog.Errorf("diskMount: formatAndMount failed, err: %v", err)
|
|
return err
|
|
}
|
|
|
|
if len(fstype) == 0 {
|
|
// Use 'NTFS' as the default
|
|
fstype = "NTFS"
|
|
}
|
|
|
|
// format disk if it is unformatted(raw)
|
|
cmd := fmt.Sprintf("Get-Disk -Number %s | Where partitionstyle -eq 'raw' | Initialize-Disk -PartitionStyle MBR -PassThru"+
|
|
" | New-Partition -AssignDriveLetter -UseMaximumSize | Format-Volume -FileSystem %s -Confirm:$false", source, fstype)
|
|
if output, err := mounter.Exec.Run("powershell", "/c", cmd); err != nil {
|
|
return fmt.Errorf("diskMount: format disk failed, error: %v, output: %q", err, string(output))
|
|
}
|
|
klog.V(4).Infof("diskMount: Disk successfully formatted, disk: %q, fstype: %q", source, fstype)
|
|
|
|
driveLetter, err := getDriveLetterByDiskNumber(source, mounter.Exec)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
driverPath := driveLetter + ":"
|
|
target = normalizeWindowsPath(target)
|
|
klog.V(4).Infof("Attempting to formatAndMount disk: %s %s %s", fstype, driverPath, target)
|
|
if output, err := mounter.Exec.Run("cmd", "/c", "mklink", "/D", target, driverPath); err != nil {
|
|
klog.Errorf("mklink failed: %v, output: %q", err, string(output))
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func normalizeWindowsPath(path string) string {
|
|
normalizedPath := strings.Replace(path, "/", "\\", -1)
|
|
if strings.HasPrefix(normalizedPath, "\\") {
|
|
normalizedPath = "c:" + normalizedPath
|
|
}
|
|
return normalizedPath
|
|
}
|
|
|
|
// ValidateDiskNumber : disk number should be a number in [0, 99]
|
|
func ValidateDiskNumber(disk string) error {
|
|
diskNum, err := strconv.Atoi(disk)
|
|
if err != nil {
|
|
return fmt.Errorf("wrong disk number format: %q, err:%v", disk, err)
|
|
}
|
|
|
|
if diskNum < 0 || diskNum > 99 {
|
|
return fmt.Errorf("disk number out of range: %q", disk)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Get drive letter according to windows disk number
|
|
func getDriveLetterByDiskNumber(diskNum string, exec Exec) (string, error) {
|
|
cmd := fmt.Sprintf("(Get-Partition -DiskNumber %s).DriveLetter", diskNum)
|
|
output, err := exec.Run("powershell", "/c", cmd)
|
|
if err != nil {
|
|
return "", fmt.Errorf("azureMount: Get Drive Letter failed: %v, output: %q", err, string(output))
|
|
}
|
|
if len(string(output)) < 1 {
|
|
return "", fmt.Errorf("azureMount: Get Drive Letter failed, output is empty")
|
|
}
|
|
return string(output)[:1], nil
|
|
}
|
|
|
|
// getAllParentLinks walks all symbolic links and return all the parent targets recursively
|
|
func getAllParentLinks(path string) ([]string, error) {
|
|
const maxIter = 255
|
|
links := []string{}
|
|
for {
|
|
links = append(links, path)
|
|
if len(links) > maxIter {
|
|
return links, fmt.Errorf("unexpected length of parent links: %v", links)
|
|
}
|
|
|
|
fi, err := os.Lstat(path)
|
|
if err != nil {
|
|
return links, fmt.Errorf("Lstat: %v", err)
|
|
}
|
|
if fi.Mode()&os.ModeSymlink == 0 {
|
|
break
|
|
}
|
|
|
|
path, err = os.Readlink(path)
|
|
if err != nil {
|
|
return links, fmt.Errorf("Readlink error: %v", err)
|
|
}
|
|
}
|
|
|
|
return links, nil
|
|
}
|
|
|
|
// GetMountRefs : empty implementation here since there is no place to query all mount points on Windows
|
|
func (mounter *Mounter) GetMountRefs(pathname string) ([]string, error) {
|
|
windowsPath := normalizeWindowsPath(pathname)
|
|
pathExists, pathErr := PathExists(windowsPath)
|
|
if !pathExists {
|
|
return []string{}, nil
|
|
} else if IsCorruptedMnt(pathErr) {
|
|
klog.Warningf("GetMountRefs found corrupted mount at %s, treating as unmounted path", windowsPath)
|
|
return []string{}, nil
|
|
} else if pathErr != nil {
|
|
return nil, fmt.Errorf("error checking path %s: %v", windowsPath, pathErr)
|
|
}
|
|
return []string{pathname}, nil
|
|
}
|
|
|
|
// Note that on windows, it always returns 0. We actually don't set FSGroup on
|
|
// windows platform, see SetVolumeOwnership implementation.
|
|
func (mounter *Mounter) GetFSGroup(pathname string) (int64, error) {
|
|
return 0, nil
|
|
}
|
|
|
|
func (mounter *Mounter) GetSELinuxSupport(pathname string) (bool, error) {
|
|
// Windows does not support SELinux.
|
|
return false, nil
|
|
}
|
|
|
|
func (mounter *Mounter) GetMode(pathname string) (os.FileMode, error) {
|
|
info, err := os.Stat(pathname)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return info.Mode(), nil
|
|
}
|