764afa0d18
Use full name including extension for shim binary format on Windows in order to match any stat path faster without a fallback. Signed-off-by: Justin Terry (VM) <juterry@microsoft.com>
93 lines
2.4 KiB
Go
93 lines
2.4 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
|
|
}
|
|
|
|
// 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) {
|
|
var c net.Conn
|
|
var lastError error
|
|
timedOutError := errors.Errorf("timed out waiting for npipe %s", address)
|
|
start := time.Now()
|
|
for {
|
|
remaining := timeout - time.Since(start)
|
|
if remaining <= 0 {
|
|
lastError = timedOutError
|
|
break
|
|
}
|
|
c, lastError = winio.DialPipe(address, &remaining)
|
|
if lastError == nil {
|
|
break
|
|
}
|
|
if !os.IsNotExist(lastError) {
|
|
break
|
|
}
|
|
// 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. We use the passed in timeout for the
|
|
// `DialPipe` timeout if the pipe exists however to give the pipe time
|
|
// to `Accept` the connection.
|
|
if time.Since(start) >= 5*time.Second {
|
|
lastError = timedOutError
|
|
break
|
|
}
|
|
time.Sleep(10 * time.Millisecond)
|
|
}
|
|
return c, lastError
|
|
}
|
|
|
|
// 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
|
|
}
|