Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Move client/unversioned/remotecommand to client-go
Module remotecommand originally part of kubernetes/pkg/client/unversioned was moved to client-go/tools, and will be used as authoritative in kubectl, e2e and other places. Module remotecommand relies on util/exec module which will be copied to client-go/pkg/util Kubernetes-commit: 2612e0c78ad18ac87bbd200d547100cf99f36089
- Loading branch information
1 parent
29b5bff
commit 20e59c6
Showing
13 changed files
with
1,423 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,14 @@ | ||
package(default_visibility = ["//visibility:public"]) | ||
|
||
licenses(["notice"]) | ||
|
||
load( | ||
"@io_bazel_rules_go//go:def.bzl", | ||
"go_library", | ||
) | ||
|
||
go_library( | ||
name = "go_default_library", | ||
srcs = ["exec.go"], | ||
tags = ["automanaged"], | ||
) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,188 @@ | ||
/* | ||
Copyright 2014 The Kubernetes 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 exec | ||
|
||
import ( | ||
"io" | ||
osexec "os/exec" | ||
"syscall" | ||
"time" | ||
) | ||
|
||
// ErrExecutableNotFound is returned if the executable is not found. | ||
var ErrExecutableNotFound = osexec.ErrNotFound | ||
|
||
// Interface is an interface that presents a subset of the os/exec API. Use this | ||
// when you want to inject fakeable/mockable exec behavior. | ||
type Interface interface { | ||
// Command returns a Cmd instance which can be used to run a single command. | ||
// This follows the pattern of package os/exec. | ||
Command(cmd string, args ...string) Cmd | ||
|
||
// LookPath wraps os/exec.LookPath | ||
LookPath(file string) (string, error) | ||
} | ||
|
||
// Cmd is an interface that presents an API that is very similar to Cmd from os/exec. | ||
// As more functionality is needed, this can grow. Since Cmd is a struct, we will have | ||
// to replace fields with get/set method pairs. | ||
type Cmd interface { | ||
// CombinedOutput runs the command and returns its combined standard output | ||
// and standard error. This follows the pattern of package os/exec. | ||
CombinedOutput() ([]byte, error) | ||
// Output runs the command and returns standard output, but not standard err | ||
Output() ([]byte, error) | ||
SetDir(dir string) | ||
SetStdin(in io.Reader) | ||
SetStdout(out io.Writer) | ||
// Stops the command by sending SIGTERM. It is not guaranteed the | ||
// process will stop before this function returns. If the process is not | ||
// responding, an internal timer function will send a SIGKILL to force | ||
// terminate after 10 seconds. | ||
Stop() | ||
} | ||
|
||
// ExitError is an interface that presents an API similar to os.ProcessState, which is | ||
// what ExitError from os/exec is. This is designed to make testing a bit easier and | ||
// probably loses some of the cross-platform properties of the underlying library. | ||
type ExitError interface { | ||
String() string | ||
Error() string | ||
Exited() bool | ||
ExitStatus() int | ||
} | ||
|
||
// Implements Interface in terms of really exec()ing. | ||
type executor struct{} | ||
|
||
// New returns a new Interface which will os/exec to run commands. | ||
func New() Interface { | ||
return &executor{} | ||
} | ||
|
||
// Command is part of the Interface interface. | ||
func (executor *executor) Command(cmd string, args ...string) Cmd { | ||
return (*cmdWrapper)(osexec.Command(cmd, args...)) | ||
} | ||
|
||
// LookPath is part of the Interface interface | ||
func (executor *executor) LookPath(file string) (string, error) { | ||
return osexec.LookPath(file) | ||
} | ||
|
||
// Wraps exec.Cmd so we can capture errors. | ||
type cmdWrapper osexec.Cmd | ||
|
||
func (cmd *cmdWrapper) SetDir(dir string) { | ||
cmd.Dir = dir | ||
} | ||
|
||
func (cmd *cmdWrapper) SetStdin(in io.Reader) { | ||
cmd.Stdin = in | ||
} | ||
|
||
func (cmd *cmdWrapper) SetStdout(out io.Writer) { | ||
cmd.Stdout = out | ||
} | ||
|
||
// CombinedOutput is part of the Cmd interface. | ||
func (cmd *cmdWrapper) CombinedOutput() ([]byte, error) { | ||
out, err := (*osexec.Cmd)(cmd).CombinedOutput() | ||
if err != nil { | ||
return out, handleError(err) | ||
} | ||
return out, nil | ||
} | ||
|
||
func (cmd *cmdWrapper) Output() ([]byte, error) { | ||
out, err := (*osexec.Cmd)(cmd).Output() | ||
if err != nil { | ||
return out, handleError(err) | ||
} | ||
return out, nil | ||
} | ||
|
||
// Stop is part of the Cmd interface. | ||
func (cmd *cmdWrapper) Stop() { | ||
c := (*osexec.Cmd)(cmd) | ||
if c.ProcessState.Exited() { | ||
return | ||
} | ||
c.Process.Signal(syscall.SIGTERM) | ||
time.AfterFunc(10*time.Second, func() { | ||
if c.ProcessState.Exited() { | ||
return | ||
} | ||
c.Process.Signal(syscall.SIGKILL) | ||
}) | ||
} | ||
|
||
func handleError(err error) error { | ||
if ee, ok := err.(*osexec.ExitError); ok { | ||
// Force a compile fail if exitErrorWrapper can't convert to ExitError. | ||
var x ExitError = &ExitErrorWrapper{ee} | ||
return x | ||
} | ||
if ee, ok := err.(*osexec.Error); ok { | ||
if ee.Err == osexec.ErrNotFound { | ||
return ErrExecutableNotFound | ||
} | ||
} | ||
return err | ||
} | ||
|
||
// ExitErrorWrapper is an implementation of ExitError in terms of os/exec ExitError. | ||
// Note: standard exec.ExitError is type *os.ProcessState, which already implements Exited(). | ||
type ExitErrorWrapper struct { | ||
*osexec.ExitError | ||
} | ||
|
||
var _ ExitError = ExitErrorWrapper{} | ||
|
||
// ExitStatus is part of the ExitError interface. | ||
func (eew ExitErrorWrapper) ExitStatus() int { | ||
ws, ok := eew.Sys().(syscall.WaitStatus) | ||
if !ok { | ||
panic("can't call ExitStatus() on a non-WaitStatus exitErrorWrapper") | ||
} | ||
return ws.ExitStatus() | ||
} | ||
|
||
// CodeExitError is an implementation of ExitError consisting of an error object | ||
// and an exit code (the upper bits of os.exec.ExitStatus). | ||
type CodeExitError struct { | ||
Err error | ||
Code int | ||
} | ||
|
||
var _ ExitError = CodeExitError{} | ||
|
||
func (e CodeExitError) Error() string { | ||
return e.Err.Error() | ||
} | ||
|
||
func (e CodeExitError) String() string { | ||
return e.Err.Error() | ||
} | ||
|
||
func (e CodeExitError) Exited() bool { | ||
return true | ||
} | ||
|
||
func (e CodeExitError) ExitStatus() int { | ||
return e.Code | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,51 @@ | ||
package(default_visibility = ["//visibility:public"]) | ||
|
||
licenses(["notice"]) | ||
|
||
load( | ||
"@io_bazel_rules_go//go:def.bzl", | ||
"go_library", | ||
"go_test", | ||
) | ||
|
||
go_test( | ||
name = "go_default_test", | ||
srcs = [ | ||
"v2_test.go", | ||
"v4_test.go", | ||
], | ||
library = ":go_default_library", | ||
tags = ["automanaged"], | ||
deps = [ | ||
"//vendor/k8s.io/apimachinery/pkg/util/httpstream:go_default_library", | ||
"//vendor/k8s.io/apimachinery/pkg/util/wait:go_default_library", | ||
"//vendor/k8s.io/client-go/pkg/api:go_default_library", | ||
], | ||
) | ||
|
||
go_library( | ||
name = "go_default_library", | ||
srcs = [ | ||
"doc.go", | ||
"errorstream.go", | ||
"remotecommand.go", | ||
"resize.go", | ||
"v1.go", | ||
"v2.go", | ||
"v3.go", | ||
"v4.go", | ||
], | ||
tags = ["automanaged"], | ||
deps = [ | ||
"//vendor/github.com/golang/glog:go_default_library", | ||
"//vendor/k8s.io/apimachinery/pkg/apis/meta/v1:go_default_library", | ||
"//vendor/k8s.io/apimachinery/pkg/util/httpstream:go_default_library", | ||
"//vendor/k8s.io/apimachinery/pkg/util/httpstream/spdy:go_default_library", | ||
"//vendor/k8s.io/apimachinery/pkg/util/remotecommand:go_default_library", | ||
"//vendor/k8s.io/apimachinery/pkg/util/runtime:go_default_library", | ||
"//vendor/k8s.io/client-go/pkg/api:go_default_library", | ||
"//vendor/k8s.io/client-go/pkg/util/exec:go_default_library", | ||
"//vendor/k8s.io/client-go/rest:go_default_library", | ||
"//vendor/k8s.io/client-go/transport:go_default_library", | ||
], | ||
) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,20 @@ | ||
/* | ||
Copyright 2015 The Kubernetes 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 remotecommand adds support for executing commands in containers, | ||
// with support for separate stdin, stdout, and stderr streams, as well as | ||
// TTY. | ||
package remotecommand // import "k8s.io/client-go/tools/remotecommand" |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,55 @@ | ||
/* | ||
Copyright 2016 The Kubernetes 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 remotecommand | ||
|
||
import ( | ||
"fmt" | ||
"io" | ||
"io/ioutil" | ||
|
||
"k8s.io/apimachinery/pkg/util/runtime" | ||
) | ||
|
||
// errorStreamDecoder interprets the data on the error channel and creates a go error object from it. | ||
type errorStreamDecoder interface { | ||
decode(message []byte) error | ||
} | ||
|
||
// watchErrorStream watches the errorStream for remote command error data, | ||
// decodes it with the given errorStreamDecoder, sends the decoded error (or nil if the remote | ||
// command exited successfully) to the returned error channel, and closes it. | ||
// This function returns immediately. | ||
func watchErrorStream(errorStream io.Reader, d errorStreamDecoder) chan error { | ||
errorChan := make(chan error) | ||
|
||
go func() { | ||
defer runtime.HandleCrash() | ||
|
||
message, err := ioutil.ReadAll(errorStream) | ||
switch { | ||
case err != nil && err != io.EOF: | ||
errorChan <- fmt.Errorf("error reading from error stream: %s", err) | ||
case len(message) > 0: | ||
errorChan <- d.decode(message) | ||
default: | ||
errorChan <- nil | ||
} | ||
close(errorChan) | ||
}() | ||
|
||
return errorChan | ||
} |
Oops, something went wrong.