daf12cd194
AnonDialer will now return a "not found" error if the pipe is not found before the timeout is reached. If the pipe exists but the timeout is reached while attempting to connect, the timeout error will still be returned. This will allow the error handling logic to work properly when connecting to the shim log pipe. An error message is only logged if the error is not "not found", so now log noise from log pipes that were never intended to be created by the shim will be hidden. This change also cleans up the control flow for AnonDialer on Windows. The new code should be more easily readable, but the only semantic change is the error return value change. Signed-off-by: Kevin Parsons <kevpar@microsoft.com>
97 lines
2.6 KiB
Go
97 lines
2.6 KiB
Go
/*
|
|
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"
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
"syscall"
|
|
"time"
|
|
|
|
winio "github.com/Microsoft/go-winio"
|
|
"github.com/containerd/containerd/namespaces"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
const shimBinaryFormat = "containerd-shim-%s-%s.exe"
|
|
|
|
func getSysProcAttr() *syscall.SysProcAttr {
|
|
return nil
|
|
}
|
|
|
|
// SetScore sets the oom score for a process
|
|
func SetScore(pid int) error {
|
|
return nil
|
|
}
|
|
|
|
// AdjustOOMScore sets the OOM score for the process to the parents OOM score +1
|
|
// to ensure that they parent has a lower* score than the shim
|
|
func AdjustOOMScore(pid int) error {
|
|
return nil
|
|
}
|
|
|
|
// SocketAddress returns a npipe address
|
|
func SocketAddress(ctx context.Context, id string) (string, error) {
|
|
ns, err := namespaces.NamespaceRequired(ctx)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return fmt.Sprintf("\\\\.\\pipe\\containerd-shim-%s-%s-pipe", ns, id), nil
|
|
}
|
|
|
|
// AnonDialer returns a dialer for a npipe
|
|
func AnonDialer(address string, timeout time.Duration) (net.Conn, error) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), timeout)
|
|
defer cancel()
|
|
|
|
// If there is nobody serving the pipe we limit the timeout for this case to
|
|
// 5 seconds because any shim that would serve this endpoint should serve it
|
|
// within 5 seconds.
|
|
serveTimer := time.NewTimer(5 * time.Second)
|
|
defer serveTimer.Stop()
|
|
for {
|
|
c, err := winio.DialPipeContext(ctx, address)
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
select {
|
|
case <-serveTimer.C:
|
|
return nil, errors.Wrap(os.ErrNotExist, "pipe not found before timeout")
|
|
default:
|
|
// Wait 10ms for the shim to serve and try again.
|
|
time.Sleep(10 * time.Millisecond)
|
|
continue
|
|
}
|
|
} else if err == context.DeadlineExceeded {
|
|
return nil, errors.Wrapf(err, "timed out waiting for npipe %s", address)
|
|
}
|
|
return nil, err
|
|
}
|
|
return c, nil
|
|
}
|
|
}
|
|
|
|
// NewSocket returns a new npipe listener
|
|
func NewSocket(address string) (net.Listener, error) {
|
|
l, err := winio.ListenPipe(address, nil)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to listen to npipe %s", address)
|
|
}
|
|
return l, nil
|
|
}
|