mirror of
https://github.com/k3s-io/k3s.git
synced 2024-06-07 19:41:36 +00:00
135 lines
3.0 KiB
Go
135 lines
3.0 KiB
Go
package rootless
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"net"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/rootless-containers/rootlesskit/pkg/child"
|
|
"github.com/rootless-containers/rootlesskit/pkg/copyup/tmpfssymlink"
|
|
"github.com/rootless-containers/rootlesskit/pkg/network/slirp4netns"
|
|
"github.com/rootless-containers/rootlesskit/pkg/parent"
|
|
portbuiltin "github.com/rootless-containers/rootlesskit/pkg/port/builtin"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var (
|
|
pipeFD = "_K3S_ROOTLESS_FD"
|
|
childEnv = "_K3S_ROOTLESS_SOCK"
|
|
Sock = ""
|
|
)
|
|
|
|
func Rootless(stateDir string) error {
|
|
defer func() {
|
|
os.Unsetenv(pipeFD)
|
|
os.Unsetenv(childEnv)
|
|
}()
|
|
|
|
hasFD := os.Getenv(pipeFD) != ""
|
|
hasChildEnv := os.Getenv(childEnv) != ""
|
|
|
|
if hasFD {
|
|
logrus.Debug("Running rootless child")
|
|
childOpt, err := createChildOpt()
|
|
if err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
if err := child.Child(*childOpt); err != nil {
|
|
logrus.Fatal("child died", err)
|
|
}
|
|
}
|
|
|
|
if hasChildEnv {
|
|
Sock = os.Getenv(childEnv)
|
|
logrus.Debug("Running rootless process")
|
|
return setupMounts(stateDir)
|
|
}
|
|
|
|
logrus.Debug("Running rootless parent")
|
|
parentOpt, err := createParentOpt(filepath.Join(stateDir, "rootless"))
|
|
if err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
|
|
os.Setenv(childEnv, filepath.Join(parentOpt.StateDir, parent.StateFileAPISock))
|
|
if err := parent.Parent(*parentOpt); err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
os.Exit(0)
|
|
|
|
return nil
|
|
}
|
|
|
|
func parseCIDR(s string) (*net.IPNet, error) {
|
|
if s == "" {
|
|
return nil, nil
|
|
}
|
|
ip, ipnet, err := net.ParseCIDR(s)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if !ip.Equal(ipnet.IP) {
|
|
return nil, errors.Errorf("cidr must be like 10.0.2.0/24, not like 10.0.2.100/24")
|
|
}
|
|
return ipnet, nil
|
|
}
|
|
|
|
func createParentOpt(stateDir string) (*parent.Opt, error) {
|
|
if err := os.MkdirAll(stateDir, 0755); err != nil {
|
|
return nil, errors.Wrapf(err, "failed to mkdir %s", stateDir)
|
|
}
|
|
|
|
stateDir, err := ioutil.TempDir("", "rootless")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
opt := &parent.Opt{
|
|
StateDir: stateDir,
|
|
}
|
|
|
|
mtu := 0
|
|
ipnet, err := parseCIDR("10.41.0.0/16")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
disableHostLoopback := true
|
|
binary := "slirp4netns"
|
|
if _, err := exec.LookPath(binary); err != nil {
|
|
return nil, err
|
|
}
|
|
opt.NetworkDriver = slirp4netns.NewParentDriver(binary, mtu, ipnet, disableHostLoopback, "")
|
|
opt.PortDriver, err = portbuiltin.NewParentDriver(&logrusDebugWriter{}, stateDir)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
opt.PipeFDEnvKey = pipeFD
|
|
|
|
return opt, nil
|
|
}
|
|
|
|
type logrusDebugWriter struct {
|
|
}
|
|
|
|
func (w *logrusDebugWriter) Write(p []byte) (int, error) {
|
|
s := strings.TrimSuffix(string(p), "\n")
|
|
logrus.Debug(s)
|
|
return len(p), nil
|
|
}
|
|
|
|
func createChildOpt() (*child.Opt, error) {
|
|
opt := &child.Opt{}
|
|
opt.TargetCmd = os.Args
|
|
opt.PipeFDEnvKey = pipeFD
|
|
opt.NetworkDriver = slirp4netns.NewChildDriver()
|
|
opt.PortDriver = portbuiltin.NewChildDriver(&logrusDebugWriter{})
|
|
opt.CopyUpDirs = []string{"/etc", "/run", "/var/lib"}
|
|
opt.CopyUpDriver = tmpfssymlink.NewChildDriver()
|
|
return opt, nil
|
|
}
|