forked from dan-v/awslambdaproxy
/
lambda.go
50 lines (42 loc) · 1.31 KB
/
lambda.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
package main
import (
"flag"
"log"
"os"
)
func lambdaInit(tunnelHost string, sshPort string, sshPrivateKeyFile string, sshUsername string) {
log.Println("Starting lambdaProxyServer")
lambdaProxyServer := startLambdaProxyServer()
log.Println("Establishing tunnel connection to", tunnelHost)
lambdaTunnelConnection, err := setupLambdaTunnelConnection(tunnelHost, sshPort, sshUsername, sshPrivateKeyFile)
if err != nil {
log.Fatal("Failed to establish connection to "+tunnelHost, err)
}
log.Println("Starting lambdaDataCopyManager")
dataCopyManager := newLambdaDataCopyManager(lambdaProxyServer, lambdaTunnelConnection)
dataCopyManager.run()
}
func main() {
addressPtr := flag.String("address", "localhost", "IP of server to connect to")
sshPortPtr := flag.String("ssh-port", "22", "SSH port")
sshUsernamePtr := flag.String("ssh-user", "ubuntu", "SSH username")
sshPrivateKeyFilePtr := flag.String("ssh-private-key", "/tmp/privatekey", "SSH private key file")
flag.Parse()
if *addressPtr == "" {
flag.PrintDefaults()
os.Exit(1)
}
if *sshPortPtr == "" {
flag.PrintDefaults()
os.Exit(1)
}
if *sshPrivateKeyFilePtr == "" {
flag.PrintDefaults()
os.Exit(1)
}
if *sshUsernamePtr == "" {
flag.PrintDefaults()
os.Exit(1)
}
lambdaInit(*addressPtr, *sshPortPtr, *sshPrivateKeyFilePtr, *sshUsernamePtr)
}