mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
ead2f80073
Signed-off-by: John Howard <jhoward@microsoft.com>
107 lines
2.4 KiB
Go
107 lines
2.4 KiB
Go
// +build linux
|
|
|
|
package server
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/docker/docker/daemon"
|
|
"github.com/docker/docker/pkg/sockets"
|
|
"github.com/docker/docker/pkg/systemd"
|
|
"github.com/docker/libnetwork/portallocator"
|
|
)
|
|
|
|
// newServer sets up the required serverCloser and does protocol specific checking.
|
|
func (s *Server) newServer(proto, addr string) (serverCloser, error) {
|
|
var (
|
|
err error
|
|
l net.Listener
|
|
)
|
|
switch proto {
|
|
case "fd":
|
|
ls, err := systemd.ListenFD(addr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
chErrors := make(chan error, len(ls))
|
|
// We don't want to start serving on these sockets until the
|
|
// daemon is initialized and installed. Otherwise required handlers
|
|
// won't be ready.
|
|
<-s.start
|
|
// Since ListenFD will return one or more sockets we have
|
|
// to create a go func to spawn off multiple serves
|
|
for i := range ls {
|
|
listener := ls[i]
|
|
go func() {
|
|
httpSrv := http.Server{Handler: s.router}
|
|
chErrors <- httpSrv.Serve(listener)
|
|
}()
|
|
}
|
|
for i := 0; i < len(ls); i++ {
|
|
if err := <-chErrors; err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return nil, nil
|
|
case "tcp":
|
|
l, err = s.initTcpSocket(addr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
case "unix":
|
|
if l, err = sockets.NewUnixSocket(addr, s.cfg.SocketGroup, s.start); err != nil {
|
|
return nil, err
|
|
}
|
|
default:
|
|
return nil, fmt.Errorf("Invalid protocol format: %q", proto)
|
|
}
|
|
return &HttpServer{
|
|
&http.Server{
|
|
Addr: addr,
|
|
Handler: s.router,
|
|
},
|
|
l,
|
|
}, nil
|
|
}
|
|
|
|
func (s *Server) AcceptConnections(d *daemon.Daemon) {
|
|
// Tell the init daemon we are accepting requests
|
|
s.daemon = d
|
|
go systemd.SdNotify("READY=1")
|
|
// close the lock so the listeners start accepting connections
|
|
select {
|
|
case <-s.start:
|
|
default:
|
|
close(s.start)
|
|
}
|
|
}
|
|
|
|
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
|
|
}
|