153 lines
4.4 KiB
Go
153 lines
4.4 KiB
Go
/*
|
|
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 nfs
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/container-storage-interface/spec/lib/go/csi"
|
|
"k8s.io/klog/v2"
|
|
mount "k8s.io/mount-utils"
|
|
)
|
|
|
|
type Driver struct {
|
|
name string
|
|
nodeID string
|
|
version string
|
|
|
|
endpoint string
|
|
|
|
perm *uint32
|
|
|
|
//ids *identityServer
|
|
ns *NodeServer
|
|
cap map[csi.VolumeCapability_AccessMode_Mode]bool
|
|
cscap []*csi.ControllerServiceCapability
|
|
nscap []*csi.NodeServiceCapability
|
|
volumeLocks *VolumeLocks
|
|
}
|
|
|
|
const (
|
|
DefaultDriverName = "nfs.csi.k8s.io"
|
|
// Address of the NFS server
|
|
paramServer = "server"
|
|
// Base directory of the NFS server to create volumes under.
|
|
// The base directory must be a direct child of the root directory.
|
|
// The root directory is omitted from the string, for example:
|
|
// "base" instead of "/base"
|
|
paramShare = "share"
|
|
)
|
|
|
|
func NewDriver(nodeID, driverName, endpoint string, perm *uint32) *Driver {
|
|
klog.V(2).Infof("Driver: %v version: %v", driverName, driverVersion)
|
|
|
|
n := &Driver{
|
|
name: driverName,
|
|
version: driverVersion,
|
|
nodeID: nodeID,
|
|
endpoint: endpoint,
|
|
cap: map[csi.VolumeCapability_AccessMode_Mode]bool{},
|
|
perm: perm,
|
|
}
|
|
|
|
vcam := []csi.VolumeCapability_AccessMode_Mode{
|
|
csi.VolumeCapability_AccessMode_SINGLE_NODE_WRITER,
|
|
csi.VolumeCapability_AccessMode_SINGLE_NODE_READER_ONLY,
|
|
csi.VolumeCapability_AccessMode_SINGLE_NODE_SINGLE_WRITER,
|
|
csi.VolumeCapability_AccessMode_SINGLE_NODE_MULTI_WRITER,
|
|
csi.VolumeCapability_AccessMode_MULTI_NODE_READER_ONLY,
|
|
csi.VolumeCapability_AccessMode_MULTI_NODE_SINGLE_WRITER,
|
|
csi.VolumeCapability_AccessMode_MULTI_NODE_MULTI_WRITER,
|
|
}
|
|
n.AddVolumeCapabilityAccessModes(vcam)
|
|
|
|
n.AddControllerServiceCapabilities([]csi.ControllerServiceCapability_RPC_Type{
|
|
csi.ControllerServiceCapability_RPC_CREATE_DELETE_VOLUME,
|
|
csi.ControllerServiceCapability_RPC_SINGLE_NODE_MULTI_WRITER,
|
|
})
|
|
|
|
n.AddNodeServiceCapabilities([]csi.NodeServiceCapability_RPC_Type{
|
|
csi.NodeServiceCapability_RPC_GET_VOLUME_STATS,
|
|
csi.NodeServiceCapability_RPC_SINGLE_NODE_MULTI_WRITER,
|
|
csi.NodeServiceCapability_RPC_UNKNOWN,
|
|
})
|
|
n.volumeLocks = NewVolumeLocks()
|
|
return n
|
|
}
|
|
|
|
func NewNodeServer(n *Driver, mounter mount.Interface) *NodeServer {
|
|
return &NodeServer{
|
|
Driver: n,
|
|
mounter: mounter,
|
|
}
|
|
}
|
|
|
|
func (n *Driver) Run(testMode bool) {
|
|
versionMeta, err := GetVersionYAML(n.name)
|
|
if err != nil {
|
|
klog.Fatalf("%v", err)
|
|
}
|
|
klog.Infof("\nDRIVER INFORMATION:\n-------------------\n%s\n\nStreaming logs below:", versionMeta)
|
|
|
|
n.ns = NewNodeServer(n, mount.New(""))
|
|
s := NewNonBlockingGRPCServer()
|
|
s.Start(n.endpoint,
|
|
NewDefaultIdentityServer(n),
|
|
// NFS plugin has not implemented ControllerServer
|
|
// using default controllerserver.
|
|
NewControllerServer(n),
|
|
n.ns,
|
|
testMode)
|
|
s.Wait()
|
|
}
|
|
|
|
func (n *Driver) AddVolumeCapabilityAccessModes(vc []csi.VolumeCapability_AccessMode_Mode) []*csi.VolumeCapability_AccessMode {
|
|
var vca []*csi.VolumeCapability_AccessMode
|
|
for _, c := range vc {
|
|
klog.Infof("Enabling volume access mode: %v", c.String())
|
|
vca = append(vca, &csi.VolumeCapability_AccessMode{Mode: c})
|
|
n.cap[c] = true
|
|
}
|
|
return vca
|
|
}
|
|
|
|
func (n *Driver) AddControllerServiceCapabilities(cl []csi.ControllerServiceCapability_RPC_Type) {
|
|
var csc []*csi.ControllerServiceCapability
|
|
|
|
for _, c := range cl {
|
|
klog.Infof("Enabling controller service capability: %v", c.String())
|
|
csc = append(csc, NewControllerServiceCapability(c))
|
|
}
|
|
|
|
n.cscap = csc
|
|
}
|
|
|
|
func (n *Driver) AddNodeServiceCapabilities(nl []csi.NodeServiceCapability_RPC_Type) {
|
|
var nsc []*csi.NodeServiceCapability
|
|
for _, n := range nl {
|
|
klog.Infof("Enabling node service capability: %v", n.String())
|
|
nsc = append(nsc, NewNodeServiceCapability(n))
|
|
}
|
|
n.nscap = nsc
|
|
}
|
|
|
|
func IsCorruptedDir(dir string) bool {
|
|
_, pathErr := mount.PathExists(dir)
|
|
fmt.Printf("IsCorruptedDir(%s) returned with error: %v", dir, pathErr)
|
|
return pathErr != nil && mount.IsCorruptedMnt(pathErr)
|
|
}
|