-
Notifications
You must be signed in to change notification settings - Fork 9
/
portforward.go
89 lines (71 loc) · 1.99 KB
/
portforward.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
// Copyright 2022 Namespace Labs Inc; All rights reserved.
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
package cluster
import (
"context"
"errors"
"fmt"
"io"
"net"
"os"
"github.com/spf13/cobra"
"namespacelabs.dev/foundation/internal/cli/fncobra"
"namespacelabs.dev/foundation/internal/console"
"namespacelabs.dev/foundation/internal/fnerrors"
"namespacelabs.dev/foundation/internal/fnnet"
"namespacelabs.dev/foundation/internal/providers/nscloud/api"
)
func newPortForwardCmd() *cobra.Command {
cmd := &cobra.Command{
Use: "port-forward [cluster-id]",
Short: "Opens a local port which connects to the cluster.",
Args: cobra.MaximumNArgs(1),
}
port := cmd.Flags().Int("target_port", 0, "Which port to forward to.")
cmd.RunE = fncobra.RunE(func(ctx context.Context, args []string) error {
if *port == 0 {
return fnerrors.New("--target_port is required")
}
cluster, _, err := SelectRunningCluster(ctx, args)
if err != nil {
if errors.Is(err, ErrEmptyClusterList) {
PrintCreateClusterMsg(ctx)
return nil
}
return err
}
if cluster == nil {
return nil
}
return portForward(ctx, cluster, *port)
})
return cmd
}
func portForward(ctx context.Context, cluster *api.KubernetesCluster, targetPort int) error {
lst, err := fnnet.ListenPort(ctx, "127.0.0.1", 0, targetPort)
if err != nil {
return err
}
localPort := lst.Addr().(*net.TCPAddr).Port
fmt.Fprintf(console.Stdout(ctx), "Listening on 127.0.0.1:%d\n", localPort)
for {
conn, err := lst.Accept()
if err != nil {
return err
}
fmt.Fprintf(console.Stdout(ctx), "New connection from %v\n", conn.RemoteAddr())
go func() {
defer conn.Close()
proxyConn, err := api.DialPort(ctx, cluster, targetPort)
if err != nil {
fmt.Fprintf(os.Stderr, "Failed to connect: %v\n", err)
return
}
go func() {
_, _ = io.Copy(conn, proxyConn)
}()
_, _ = io.Copy(proxyConn, conn)
}()
}
}