c80ca4f4a2
Because tasks may be deleted while listing containers, we need to ignore errors from state requests that are due to a closed error. All of these get mapped to ErrNotFound, which can be used to filter the entries. There may be a better fix that does a better job of keeping track of the intended state of a backend task. The current condition of assuming that a closed client is a shutdown task may be too naive. Signed-off-by: Stephen J Day <stephen.day@docker.com>
136 lines
3.3 KiB
Go
136 lines
3.3 KiB
Go
// +build linux
|
|
|
|
package linux
|
|
|
|
import (
|
|
"context"
|
|
|
|
eventstypes "github.com/containerd/containerd/api/events"
|
|
"github.com/containerd/containerd/api/types/task"
|
|
"github.com/containerd/containerd/errdefs"
|
|
shim "github.com/containerd/containerd/linux/shim/v1"
|
|
"github.com/containerd/containerd/runtime"
|
|
"github.com/pkg/errors"
|
|
"github.com/stevvooe/ttrpc"
|
|
)
|
|
|
|
// Process implements a linux process
|
|
type Process struct {
|
|
id string
|
|
t *Task
|
|
}
|
|
|
|
// ID of the process
|
|
func (p *Process) ID() string {
|
|
return p.id
|
|
}
|
|
|
|
// Kill sends the provided signal to the underlying process
|
|
//
|
|
// Unable to kill all processes in the task using this method on a process
|
|
func (p *Process) Kill(ctx context.Context, signal uint32, _ bool) error {
|
|
_, err := p.t.shim.Kill(ctx, &shim.KillRequest{
|
|
Signal: signal,
|
|
ID: p.id,
|
|
})
|
|
if err != nil {
|
|
return errdefs.FromGRPC(err)
|
|
}
|
|
return err
|
|
}
|
|
|
|
// State of process
|
|
func (p *Process) State(ctx context.Context) (runtime.State, error) {
|
|
// use the container status for the status of the process
|
|
response, err := p.t.shim.State(ctx, &shim.StateRequest{
|
|
ID: p.id,
|
|
})
|
|
if err != nil {
|
|
if errors.Cause(err) != ttrpc.ErrClosed {
|
|
return runtime.State{}, errdefs.FromGRPC(err)
|
|
}
|
|
|
|
// We treat ttrpc.ErrClosed as the shim being closed, but really this
|
|
// likely means that the process no longer exists. We'll have to plumb
|
|
// the connection differently if this causes problems.
|
|
return runtime.State{}, errdefs.ErrNotFound
|
|
}
|
|
var status runtime.Status
|
|
switch response.Status {
|
|
case task.StatusCreated:
|
|
status = runtime.CreatedStatus
|
|
case task.StatusRunning:
|
|
status = runtime.RunningStatus
|
|
case task.StatusStopped:
|
|
status = runtime.StoppedStatus
|
|
case task.StatusPaused:
|
|
status = runtime.PausedStatus
|
|
case task.StatusPausing:
|
|
status = runtime.PausingStatus
|
|
}
|
|
return runtime.State{
|
|
Pid: response.Pid,
|
|
Status: status,
|
|
Stdin: response.Stdin,
|
|
Stdout: response.Stdout,
|
|
Stderr: response.Stderr,
|
|
Terminal: response.Terminal,
|
|
ExitStatus: response.ExitStatus,
|
|
}, nil
|
|
}
|
|
|
|
// ResizePty changes the side of the process's PTY to the provided width and height
|
|
func (p *Process) ResizePty(ctx context.Context, size runtime.ConsoleSize) error {
|
|
_, err := p.t.shim.ResizePty(ctx, &shim.ResizePtyRequest{
|
|
ID: p.id,
|
|
Width: size.Width,
|
|
Height: size.Height,
|
|
})
|
|
if err != nil {
|
|
err = errdefs.FromGRPC(err)
|
|
}
|
|
return err
|
|
}
|
|
|
|
// CloseIO closes the provided IO pipe for the process
|
|
func (p *Process) CloseIO(ctx context.Context) error {
|
|
_, err := p.t.shim.CloseIO(ctx, &shim.CloseIORequest{
|
|
ID: p.id,
|
|
Stdin: true,
|
|
})
|
|
if err != nil {
|
|
return errdefs.FromGRPC(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Start the process
|
|
func (p *Process) Start(ctx context.Context) error {
|
|
r, err := p.t.shim.Start(ctx, &shim.StartRequest{
|
|
ID: p.id,
|
|
})
|
|
if err != nil {
|
|
return errdefs.FromGRPC(err)
|
|
}
|
|
p.t.events.Publish(ctx, runtime.TaskExecStartedEventTopic, &eventstypes.TaskExecStarted{
|
|
ContainerID: p.t.id,
|
|
Pid: r.Pid,
|
|
ExecID: p.id,
|
|
})
|
|
return nil
|
|
}
|
|
|
|
// Wait on the process to exit and return the exit status and timestamp
|
|
func (p *Process) Wait(ctx context.Context) (*runtime.Exit, error) {
|
|
r, err := p.t.shim.Wait(ctx, &shim.WaitRequest{
|
|
ID: p.id,
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &runtime.Exit{
|
|
Timestamp: r.ExitedAt,
|
|
Status: r.ExitStatus,
|
|
}, nil
|
|
}
|