103 lines
3.1 KiB
Go
103 lines
3.1 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 os
|
|
|
|
import (
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
|
|
"github.com/containerd/fifo"
|
|
"github.com/docker/docker/pkg/mount"
|
|
"golang.org/x/net/context"
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
// OS collects system level operations that need to be mocked out
|
|
// during tests.
|
|
type OS interface {
|
|
MkdirAll(path string, perm os.FileMode) error
|
|
RemoveAll(path string) error
|
|
OpenFifo(ctx context.Context, fn string, flag int, perm os.FileMode) (io.ReadWriteCloser, error)
|
|
Stat(name string) (os.FileInfo, error)
|
|
CopyFile(src, dest string, perm os.FileMode) error
|
|
WriteFile(filename string, data []byte, perm os.FileMode) error
|
|
Mount(source string, target string, fstype string, flags uintptr, data string) error
|
|
Unmount(target string, flags int) error
|
|
}
|
|
|
|
// RealOS is used to dispatch the real system level operations.
|
|
type RealOS struct{}
|
|
|
|
// MkdirAll will will call os.MkdirAll to create a directory.
|
|
func (RealOS) MkdirAll(path string, perm os.FileMode) error {
|
|
return os.MkdirAll(path, perm)
|
|
}
|
|
|
|
// RemoveAll will call os.RemoveAll to remove the path and its children.
|
|
func (RealOS) RemoveAll(path string) error {
|
|
return os.RemoveAll(path)
|
|
}
|
|
|
|
// OpenFifo will call fifo.OpenFifo to open a fifo.
|
|
func (RealOS) OpenFifo(ctx context.Context, fn string, flag int, perm os.FileMode) (io.ReadWriteCloser, error) {
|
|
return fifo.OpenFifo(ctx, fn, flag, perm)
|
|
}
|
|
|
|
// Stat will call os.Stat to get the status of the given file.
|
|
func (RealOS) Stat(name string) (os.FileInfo, error) {
|
|
return os.Stat(name)
|
|
}
|
|
|
|
// CopyFile copys src file to dest file
|
|
func (RealOS) CopyFile(src, dest string, perm os.FileMode) error {
|
|
in, err := os.Open(src)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer in.Close()
|
|
|
|
out, err := os.OpenFile(dest, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, perm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer out.Close()
|
|
|
|
_, err = io.Copy(out, in)
|
|
return err
|
|
}
|
|
|
|
// WriteFile will call ioutil.WriteFile to write data into a file.
|
|
func (RealOS) WriteFile(filename string, data []byte, perm os.FileMode) error {
|
|
return ioutil.WriteFile(filename, data, perm)
|
|
}
|
|
|
|
// Mount will call unix.Mount to mount the file.
|
|
func (RealOS) Mount(source string, target string, fstype string, flags uintptr, data string) error {
|
|
return unix.Mount(source, target, fstype, flags, data)
|
|
}
|
|
|
|
// Unmount will call unix.Unmount to unmount the file. The function doesn't
|
|
// return error if target is not mounted.
|
|
func (RealOS) Unmount(target string, flags int) error {
|
|
// TODO(random-liu): Follow symlink to make sure the result is correct.
|
|
if mounted, err := mount.Mounted(target); err != nil || !mounted {
|
|
return err
|
|
}
|
|
return unix.Unmount(target, flags)
|
|
}
|