82 lines
1.9 KiB
Go
82 lines
1.9 KiB
Go
package allocdir
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"syscall"
|
|
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
const (
|
|
// secretDirTmpfsSize is the size of the tmpfs per task in MBs
|
|
secretDirTmpfsSize = 1
|
|
|
|
// secretMarker is the filename of the marker created so Nomad doesn't
|
|
// try to mount the secrets tmpfs more than once
|
|
secretMarker = ".nomad-mount"
|
|
)
|
|
|
|
// linkDir bind mounts src to dst as Linux doesn't support hardlinking
|
|
// directories.
|
|
func linkDir(src, dst string) error {
|
|
if err := os.MkdirAll(dst, 0777); err != nil {
|
|
return err
|
|
}
|
|
|
|
return syscall.Mount(src, dst, "", syscall.MS_BIND, "")
|
|
}
|
|
|
|
// unlinkDir unmounts a bind mounted directory as Linux doesn't support
|
|
// hardlinking directories.
|
|
func unlinkDir(dir string) error {
|
|
return syscall.Unmount(dir, 0)
|
|
}
|
|
|
|
// createSecretDir creates the secrets dir folder at the given path using a
|
|
// tmpfs
|
|
func createSecretDir(dir string) error {
|
|
// Only mount the tmpfs if we are root
|
|
if unix.Geteuid() == 0 {
|
|
if err := os.MkdirAll(dir, 0777); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Check for marker file and skip mounting if it exists
|
|
marker := filepath.Join(dir, secretMarker)
|
|
if _, err := os.Stat(marker); err == nil {
|
|
return nil
|
|
}
|
|
|
|
var flags uintptr
|
|
flags = syscall.MS_NOEXEC
|
|
options := fmt.Sprintf("size=%dm", secretDirTmpfsSize)
|
|
if err := syscall.Mount("tmpfs", dir, "tmpfs", flags, options); err != nil {
|
|
return os.NewSyscallError("mount", err)
|
|
}
|
|
|
|
// Create the marker file so we don't try to mount more than once
|
|
f, err := os.OpenFile(marker, os.O_RDWR|os.O_CREATE, 0666)
|
|
if err != nil {
|
|
// Hard fail since if this fails something is really wrong
|
|
return err
|
|
}
|
|
f.Close()
|
|
return nil
|
|
}
|
|
|
|
return os.MkdirAll(dir, 0777)
|
|
}
|
|
|
|
// createSecretDir removes the secrets dir folder
|
|
func removeSecretDir(dir string) error {
|
|
if unix.Geteuid() == 0 {
|
|
if err := syscall.Unmount(dir, 0); err != nil {
|
|
return os.NewSyscallError("unmount", err)
|
|
}
|
|
}
|
|
|
|
return os.RemoveAll(dir)
|
|
}
|