146 lines
3 KiB
Go
146 lines
3 KiB
Go
package shim
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"sync"
|
|
"syscall"
|
|
|
|
"github.com/crosbymichael/console"
|
|
runc "github.com/crosbymichael/go-runc"
|
|
"github.com/docker/containerd"
|
|
shimapi "github.com/docker/containerd/api/services/shim"
|
|
)
|
|
|
|
type initProcess struct {
|
|
sync.WaitGroup
|
|
|
|
id string
|
|
bundle string
|
|
console console.Console
|
|
io runc.IO
|
|
runc *runc.Runc
|
|
status int
|
|
pid int
|
|
}
|
|
|
|
func newInitProcess(context context.Context, r *shimapi.CreateRequest) (*initProcess, error) {
|
|
cwd, err := os.Getwd()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, rm := range r.Rootfs {
|
|
m := &containerd.Mount{
|
|
Type: rm.Type,
|
|
Source: rm.Source,
|
|
Options: rm.Options,
|
|
}
|
|
if err := m.Mount(filepath.Join(cwd, "rootfs")); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
runtime := &runc.Runc{
|
|
Command: r.Runtime,
|
|
Log: filepath.Join(cwd, "log.json"),
|
|
LogFormat: runc.JSON,
|
|
PdeathSignal: syscall.SIGKILL,
|
|
}
|
|
p := &initProcess{
|
|
id: r.ID,
|
|
bundle: r.Bundle,
|
|
runc: runtime,
|
|
}
|
|
var (
|
|
socket *runc.ConsoleSocket
|
|
io runc.IO
|
|
)
|
|
if r.Terminal {
|
|
if socket, err = runc.NewConsoleSocket(filepath.Join(cwd, "pty.sock")); err != nil {
|
|
return nil, err
|
|
}
|
|
defer os.Remove(socket.Path())
|
|
} else {
|
|
// TODO: get uid/gid
|
|
if io, err = runc.NewPipeIO(0, 0); err != nil {
|
|
return nil, err
|
|
}
|
|
p.io = io
|
|
}
|
|
opts := &runc.CreateOpts{
|
|
PidFile: filepath.Join(cwd, "init.pid"),
|
|
ConsoleSocket: socket,
|
|
IO: io,
|
|
NoPivot: r.NoPivot,
|
|
}
|
|
if err := p.runc.Create(context, r.ID, r.Bundle, opts); err != nil {
|
|
return nil, err
|
|
}
|
|
if socket != nil {
|
|
console, err := socket.ReceiveMaster()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
p.console = console
|
|
if err := copyConsole(context, console, r.Stdin, r.Stdout, r.Stderr, &p.WaitGroup); err != nil {
|
|
return nil, err
|
|
}
|
|
} else {
|
|
if err := copyPipes(context, io, r.Stdin, r.Stdout, r.Stderr, &p.WaitGroup); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
pid, err := runc.ReadPidFile(opts.PidFile)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
p.pid = pid
|
|
return p, nil
|
|
}
|
|
|
|
func (p *initProcess) Pid() int {
|
|
return p.pid
|
|
}
|
|
|
|
func (p *initProcess) Status() int {
|
|
return p.status
|
|
}
|
|
|
|
func (p *initProcess) Start(context context.Context) error {
|
|
return p.runc.Start(context, p.id)
|
|
}
|
|
|
|
func (p *initProcess) Exited(status int) {
|
|
p.status = status
|
|
}
|
|
|
|
func (p *initProcess) Delete(context context.Context) error {
|
|
p.killAll(context)
|
|
p.Wait()
|
|
err := p.runc.Delete(context, p.id)
|
|
if p.io != nil {
|
|
p.io.Close()
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (p *initProcess) Resize(ws console.WinSize) error {
|
|
if p.console == nil {
|
|
return nil
|
|
}
|
|
return p.console.Resize(ws)
|
|
}
|
|
|
|
func (p *initProcess) Pause(context context.Context) error {
|
|
return p.runc.Pause(context, p.id)
|
|
}
|
|
|
|
func (p *initProcess) Resume(context context.Context) error {
|
|
return p.runc.Resume(context, p.id)
|
|
}
|
|
|
|
func (p *initProcess) killAll(context context.Context) error {
|
|
return p.runc.Kill(context, p.id, int(syscall.SIGKILL), &runc.KillOpts{
|
|
All: true,
|
|
})
|
|
}
|