Introduce containerd-shim-runhcs-v1 on Windows
Implements the containerd-shim-runhcs-v1 shim on Windows for the runtime v2 shim API. Signed-off-by: Justin Terry (VM) <juterry@microsoft.com>
This commit is contained in:
178
runtime/v2/runhcs/process.go
Normal file
178
runtime/v2/runhcs/process.go
Normal file
@@ -0,0 +1,178 @@
|
||||
// +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 runhcs
|
||||
|
||||
import (
|
||||
"context"
|
||||
"os"
|
||||
"os/exec"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
eventstypes "github.com/containerd/containerd/api/events"
|
||||
"github.com/containerd/containerd/runtime"
|
||||
)
|
||||
|
||||
func newProcess(ctx context.Context, s *service, id string, pid uint32, pr *pipeRelay, bundle, stdin, stdout, stderr string, terminal bool) (*process, error) {
|
||||
p, err := os.FindProcess(int(pid))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
process := &process{
|
||||
cid: id,
|
||||
id: id,
|
||||
pid: pid,
|
||||
bundle: bundle,
|
||||
stdin: stdin,
|
||||
stdout: stdout,
|
||||
stderr: stderr,
|
||||
terminal: terminal,
|
||||
relay: pr,
|
||||
waitBlock: make(chan struct{}),
|
||||
}
|
||||
// Store the default non-exited value for calls to stat
|
||||
process.exit.Store(&processExit{
|
||||
pid: pid,
|
||||
exitStatus: 255,
|
||||
exitedAt: time.Time{},
|
||||
exitErr: nil,
|
||||
})
|
||||
go waitForProcess(ctx, process, p, s)
|
||||
return process, nil
|
||||
}
|
||||
|
||||
func waitForProcess(ctx context.Context, process *process, p *os.Process, s *service) {
|
||||
var status int
|
||||
_, eerr := p.Wait()
|
||||
if eerr != nil {
|
||||
status = 255
|
||||
if exitErr, ok := eerr.(*exec.ExitError); ok {
|
||||
if ws, ok := exitErr.Sys().(syscall.WaitStatus); ok {
|
||||
status = ws.ExitStatus()
|
||||
}
|
||||
}
|
||||
}
|
||||
now := time.Now()
|
||||
process.exit.Store(&processExit{
|
||||
pid: process.pid,
|
||||
exitStatus: uint32(status),
|
||||
exitedAt: now,
|
||||
exitErr: eerr,
|
||||
})
|
||||
|
||||
// Wait for the relay
|
||||
process.relay.wait()
|
||||
|
||||
// close the client io, and free upstream waiters
|
||||
process.close()
|
||||
|
||||
s.publisher.Publish(
|
||||
ctx,
|
||||
runtime.TaskExitEventTopic,
|
||||
&eventstypes.TaskExit{
|
||||
ContainerID: process.cid,
|
||||
ID: process.id,
|
||||
Pid: process.pid,
|
||||
ExitStatus: uint32(status),
|
||||
ExitedAt: now,
|
||||
})
|
||||
}
|
||||
|
||||
func newExecProcess(ctx context.Context, s *service, cid, id string, pr *pipeRelay, bundle, stdin, stdout, stderr string, terminal bool) (*process, error) {
|
||||
process := &process{
|
||||
cid: cid,
|
||||
id: id,
|
||||
bundle: bundle,
|
||||
stdin: stdin,
|
||||
stdout: stdout,
|
||||
stderr: stderr,
|
||||
terminal: terminal,
|
||||
relay: pr,
|
||||
waitBlock: make(chan struct{}),
|
||||
}
|
||||
// Store the default non-exited value for calls to stat
|
||||
process.exit.Store(&processExit{
|
||||
exitStatus: 255,
|
||||
exitedAt: time.Time{},
|
||||
exitErr: nil,
|
||||
})
|
||||
return process, nil
|
||||
}
|
||||
|
||||
type process struct {
|
||||
sync.Mutex
|
||||
|
||||
cid string
|
||||
id string
|
||||
pid uint32
|
||||
|
||||
bundle string
|
||||
stdin string
|
||||
stdout string
|
||||
stderr string
|
||||
terminal bool
|
||||
relay *pipeRelay
|
||||
|
||||
// started track if the process has ever been started and will not be reset
|
||||
// for the lifetime of the process object.
|
||||
started bool
|
||||
|
||||
waitBlock chan struct{}
|
||||
// exit holds the exit value for all calls to `stat`. By default a
|
||||
// non-exited value is stored of status: 255, at: time 0.
|
||||
exit atomic.Value
|
||||
|
||||
// closeOnce is responsible for closing waitBlock and any io.
|
||||
closeOnce sync.Once
|
||||
}
|
||||
|
||||
// closeIO closes the stdin of the executing process to unblock any waiters
|
||||
func (p *process) closeIO() {
|
||||
p.Lock()
|
||||
defer p.Unlock()
|
||||
|
||||
p.relay.closeIO()
|
||||
}
|
||||
|
||||
// close closes all stdio and frees any waiters. This is safe to call multiple
|
||||
// times.
|
||||
func (p *process) close() {
|
||||
p.closeOnce.Do(func() {
|
||||
p.relay.close()
|
||||
|
||||
// Free any waiters
|
||||
close(p.waitBlock)
|
||||
})
|
||||
}
|
||||
|
||||
// stat is a non-blocking query of the current process state.
|
||||
func (p *process) stat() *processExit {
|
||||
er := p.exit.Load()
|
||||
return er.(*processExit)
|
||||
}
|
||||
|
||||
// wait waits for the container process to exit and returns the exit status. If
|
||||
// the process failed post start the processExit will contain the exitErr. This
|
||||
// is safe to call previous to calling start().
|
||||
func (p *process) wait() *processExit {
|
||||
<-p.waitBlock
|
||||
return p.stat()
|
||||
}
|
||||
Reference in New Issue
Block a user