Falls back to SPDY for gorilla/websocket https proxy error

Kubernetes-commit: 9d560540c5268e0e2aebf5306907494cf522c260
kubernetes · Jul 19, 2024 · 2c588bc · 2c588bc
1 parent b4d17b8
commit 2c588bc
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/pkg/cmd/attach/attach.go b/pkg/cmd/attach/attach.go
@@ -184,7 +184,9 @@ func createExecutor(url *url.URL, config *restclient.Config) (remotecommand.Exec
  if err != nil {
  return nil, err
  }
- exec, err = remotecommand.NewFallbackExecutor(websocketExec, exec, httpstream.IsUpgradeFailure)
+ exec, err = remotecommand.NewFallbackExecutor(websocketExec, exec, func(err error) bool {
+ return httpstream.IsUpgradeFailure(err) || httpstream.IsHTTPSProxyError(err)
+ })
  if err != nil {
  return nil, err
  }

diff --git a/pkg/cmd/exec/exec.go b/pkg/cmd/exec/exec.go
@@ -147,7 +147,9 @@ func createExecutor(url *url.URL, config *restclient.Config) (remotecommand.Exec
  if err != nil {
  return nil, err
  }
- exec, err = remotecommand.NewFallbackExecutor(websocketExec, exec, httpstream.IsUpgradeFailure)
+ exec, err = remotecommand.NewFallbackExecutor(websocketExec, exec, func(err error) bool {
+ return httpstream.IsUpgradeFailure(err) || httpstream.IsHTTPSProxyError(err)
+ })
  if err != nil {
  return nil, err
  }

diff --git a/pkg/cmd/portforward/portforward.go b/pkg/cmd/portforward/portforward.go
@@ -148,7 +148,9 @@ func createDialer(method string, url *url.URL, opts PortForwardOptions) (httpstr
  return nil, err
  }
  // First attempt tunneling (websocket) dialer, then fallback to spdy dialer.
- dialer = portforward.NewFallbackDialer(tunnelingDialer, dialer, httpstream.IsUpgradeFailure)
+ dialer = portforward.NewFallbackDialer(tunnelingDialer, dialer, func(err error) bool {
+ return httpstream.IsUpgradeFailure(err) || httpstream.IsHTTPSProxyError(err)
+ })
  }
  return dialer, nil
 }