42 lines
889 B
Go
42 lines
889 B
Go
// +build windows
|
|
|
|
package googlecompute
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"os/exec"
|
|
)
|
|
|
|
func NewTunnelDriver() TunnelDriver {
|
|
return &TunnelDriverWindows{}
|
|
}
|
|
|
|
type TunnelDriverWindows struct {
|
|
cmd *exec.Cmd
|
|
}
|
|
|
|
func (t *TunnelDriverWindows) StartTunnel(cancelCtx context.Context, tempScriptFileName string, timeout int) error {
|
|
args := []string{"/C", "call", tempScriptFileName}
|
|
cmd := exec.CommandContext(cancelCtx, "cmd", args...)
|
|
err := RunTunnelCommand(cmd, timeout)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// Store successful command on step so we can access it to cancel it
|
|
// later.
|
|
t.cmd = cmd
|
|
return nil
|
|
}
|
|
|
|
func (t *TunnelDriverWindows) StopTunnel() {
|
|
if t.cmd != nil && t.cmd.Process != nil {
|
|
err := t.cmd.Process.Kill()
|
|
if err != nil {
|
|
log.Printf("Issue stopping IAP tunnel: %s", err)
|
|
}
|
|
} else {
|
|
log.Printf("Couldn't find IAP tunnel process to kill. Continuing.")
|
|
}
|
|
}
|