mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-19 21:30:24 -04:00
122 lines
3.1 KiB
Go
122 lines
3.1 KiB
Go
![]() |
// +build !windows
|
||
|
|
||
|
package main
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"net"
|
||
|
"os"
|
||
|
"os/signal"
|
||
|
"path/filepath"
|
||
|
"strconv"
|
||
|
|
||
|
"github.com/containerd/containerd/linux"
|
||
|
"github.com/docker/docker/cmd/dockerd/hack"
|
||
|
"github.com/docker/docker/daemon"
|
||
|
"github.com/docker/docker/libcontainerd"
|
||
|
"github.com/docker/libnetwork/portallocator"
|
||
|
"golang.org/x/sys/unix"
|
||
|
)
|
||
|
|
||
|
const defaultDaemonConfigFile = "/etc/docker/daemon.json"
|
||
|
|
||
|
// setDefaultUmask sets the umask to 0022 to avoid problems
|
||
|
// caused by custom umask
|
||
|
func setDefaultUmask() error {
|
||
|
desiredUmask := 0022
|
||
|
unix.Umask(desiredUmask)
|
||
|
if umask := unix.Umask(desiredUmask); umask != desiredUmask {
|
||
|
return fmt.Errorf("failed to set umask: expected %#o, got %#o", desiredUmask, umask)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func getDaemonConfDir(_ string) string {
|
||
|
return "/etc/docker"
|
||
|
}
|
||
|
|
||
|
func (cli *DaemonCli) getPlatformRemoteOptions() ([]libcontainerd.RemoteOption, error) {
|
||
|
opts := []libcontainerd.RemoteOption{
|
||
|
libcontainerd.WithOOMScore(cli.Config.OOMScoreAdjust),
|
||
|
libcontainerd.WithPlugin("linux", &linux.Config{
|
||
|
Shim: daemon.DefaultShimBinary,
|
||
|
Runtime: daemon.DefaultRuntimeBinary,
|
||
|
RuntimeRoot: filepath.Join(cli.Config.Root, "runc"),
|
||
|
ShimDebug: cli.Config.Debug,
|
||
|
}),
|
||
|
}
|
||
|
if cli.Config.Debug {
|
||
|
opts = append(opts, libcontainerd.WithLogLevel("debug"))
|
||
|
}
|
||
|
if cli.Config.ContainerdAddr != "" {
|
||
|
opts = append(opts, libcontainerd.WithRemoteAddr(cli.Config.ContainerdAddr))
|
||
|
} else {
|
||
|
opts = append(opts, libcontainerd.WithStartDaemon(true))
|
||
|
}
|
||
|
|
||
|
return opts, nil
|
||
|
}
|
||
|
|
||
|
// setupConfigReloadTrap configures the USR2 signal to reload the configuration.
|
||
|
func (cli *DaemonCli) setupConfigReloadTrap() {
|
||
|
c := make(chan os.Signal, 1)
|
||
|
signal.Notify(c, unix.SIGHUP)
|
||
|
go func() {
|
||
|
for range c {
|
||
|
cli.reloadConfig()
|
||
|
}
|
||
|
}()
|
||
|
}
|
||
|
|
||
|
// getSwarmRunRoot gets the root directory for swarm to store runtime state
|
||
|
// For example, the control socket
|
||
|
func (cli *DaemonCli) getSwarmRunRoot() string {
|
||
|
return filepath.Join(cli.Config.ExecRoot, "swarm")
|
||
|
}
|
||
|
|
||
|
// allocateDaemonPort ensures that there are no containers
|
||
|
// that try to use any port allocated for the docker server.
|
||
|
func allocateDaemonPort(addr string) error {
|
||
|
host, port, err := net.SplitHostPort(addr)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
intPort, err := strconv.Atoi(port)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
var hostIPs []net.IP
|
||
|
if parsedIP := net.ParseIP(host); parsedIP != nil {
|
||
|
hostIPs = append(hostIPs, parsedIP)
|
||
|
} else if hostIPs, err = net.LookupIP(host); err != nil {
|
||
|
return fmt.Errorf("failed to lookup %s address in host specification", host)
|
||
|
}
|
||
|
|
||
|
pa := portallocator.Get()
|
||
|
for _, hostIP := range hostIPs {
|
||
|
if _, err := pa.RequestPort(hostIP, "tcp", intPort); err != nil {
|
||
|
return fmt.Errorf("failed to allocate daemon listening port %d (err: %v)", intPort, err)
|
||
|
}
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// notifyShutdown is called after the daemon shuts down but before the process exits.
|
||
|
func notifyShutdown(err error) {
|
||
|
}
|
||
|
|
||
|
func wrapListeners(proto string, ls []net.Listener) []net.Listener {
|
||
|
switch proto {
|
||
|
case "unix":
|
||
|
ls[0] = &hack.MalformedHostHeaderOverride{ls[0]}
|
||
|
case "fd":
|
||
|
for i := range ls {
|
||
|
ls[i] = &hack.MalformedHostHeaderOverride{ls[i]}
|
||
|
}
|
||
|
}
|
||
|
return ls
|
||
|
}
|