94 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			94 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// +build !windows
 | 
						|
 | 
						|
/*
 | 
						|
   Copyright The containerd Authors.
 | 
						|
 | 
						|
   Licensed under the Apache License, Version 2.0 (the "License");
 | 
						|
   you may not use this file except in compliance with the License.
 | 
						|
   You may obtain a copy of the License at
 | 
						|
 | 
						|
       http://www.apache.org/licenses/LICENSE-2.0
 | 
						|
 | 
						|
   Unless required by applicable law or agreed to in writing, software
 | 
						|
   distributed under the License is distributed on an "AS IS" BASIS,
 | 
						|
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
						|
   See the License for the specific language governing permissions and
 | 
						|
   limitations under the License.
 | 
						|
*/
 | 
						|
 | 
						|
package shim
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
	"io"
 | 
						|
	"net"
 | 
						|
	"os"
 | 
						|
	"os/signal"
 | 
						|
	"syscall"
 | 
						|
 | 
						|
	"github.com/containerd/fifo"
 | 
						|
	"github.com/pkg/errors"
 | 
						|
	"github.com/sirupsen/logrus"
 | 
						|
	"golang.org/x/sys/unix"
 | 
						|
)
 | 
						|
 | 
						|
// setupSignals creates a new signal handler for all signals and sets the shim as a
 | 
						|
// sub-reaper so that the container processes are reparented
 | 
						|
func setupSignals(config Config) (chan os.Signal, error) {
 | 
						|
	signals := make(chan os.Signal, 32)
 | 
						|
	smp := []os.Signal{unix.SIGTERM, unix.SIGINT, unix.SIGPIPE}
 | 
						|
	if !config.NoReaper {
 | 
						|
		smp = append(smp, unix.SIGCHLD)
 | 
						|
	}
 | 
						|
	signal.Notify(signals, smp...)
 | 
						|
	return signals, nil
 | 
						|
}
 | 
						|
 | 
						|
func setupDumpStacks(dump chan<- os.Signal) {
 | 
						|
	signal.Notify(dump, syscall.SIGUSR1)
 | 
						|
}
 | 
						|
 | 
						|
func serveListener(path string) (net.Listener, error) {
 | 
						|
	var (
 | 
						|
		l   net.Listener
 | 
						|
		err error
 | 
						|
	)
 | 
						|
	if path == "" {
 | 
						|
		l, err = net.FileListener(os.NewFile(3, "socket"))
 | 
						|
		path = "[inherited from parent]"
 | 
						|
	} else {
 | 
						|
		if len(path) > 106 {
 | 
						|
			return nil, errors.Errorf("%q: unix socket path too long (> 106)", path)
 | 
						|
		}
 | 
						|
		l, err = net.Listen("unix", "\x00"+path)
 | 
						|
	}
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	logrus.WithField("socket", path).Debug("serving api on abstract socket")
 | 
						|
	return l, nil
 | 
						|
}
 | 
						|
 | 
						|
func handleSignals(ctx context.Context, logger *logrus.Entry, signals chan os.Signal) error {
 | 
						|
	logger.Info("starting signal loop")
 | 
						|
 | 
						|
	for {
 | 
						|
		select {
 | 
						|
		case <-ctx.Done():
 | 
						|
			return ctx.Err()
 | 
						|
		case s := <-signals:
 | 
						|
			switch s {
 | 
						|
			case unix.SIGCHLD:
 | 
						|
				if err := Reap(); err != nil {
 | 
						|
					logger.WithError(err).Error("reap exit status")
 | 
						|
				}
 | 
						|
			case unix.SIGPIPE:
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func openLog(ctx context.Context, _ string) (io.Writer, error) {
 | 
						|
	return fifo.OpenFifo(ctx, "log", unix.O_WRONLY, 0700)
 | 
						|
}
 |