d7a3a090df
From the 1.17 release notes (https://golang.org/doc/go1.17#gofmt): gofmt (and go fmt) now synchronizes //go:build lines with // +build lines. More info about this change can be found at https://golang.org/design/draft-gobuild
62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
//go:build linux
|
|
// +build linux
|
|
|
|
package tools
|
|
|
|
import (
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
// CheckCloneFileSupported runs explicit test of clone file on supplied directory.
|
|
// This function creates some (src and dst) file in the directory and remove after test finished.
|
|
//
|
|
// If check failed (e.g. directory is read-only), returns err.
|
|
func CheckCloneFileSupported(dir string) (supported bool, err error) {
|
|
src, err := ioutil.TempFile(dir, "src")
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
defer os.Remove(src.Name())
|
|
|
|
dst, err := ioutil.TempFile(dir, "dst")
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
defer os.Remove(dst.Name())
|
|
|
|
if ok, err := CloneFile(dst, src); err != nil {
|
|
return false, err
|
|
} else {
|
|
return ok, nil
|
|
}
|
|
}
|
|
|
|
func CloneFile(writer io.Writer, reader io.Reader) (bool, error) {
|
|
fdst, fdstFound := writer.(*os.File)
|
|
fsrc, fsrcFound := reader.(*os.File)
|
|
if fdstFound && fsrcFound {
|
|
if err := unix.IoctlFileClone(int(fdst.Fd()), int(fsrc.Fd())); err != nil {
|
|
return false, err
|
|
}
|
|
return true, nil
|
|
}
|
|
return false, nil
|
|
}
|
|
|
|
func CloneFileByPath(dst, src string) (bool, error) {
|
|
srcFile, err := os.Open(src)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
dstFile, err := os.Create(dst) //truncating, it if it already exists.
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
return CloneFile(dstFile, srcFile)
|
|
}
|