func main() {
bin := "tmate"
cmd := exec.Command(bin)
log.Print("Starting tmate...")
f, err := pty.Start(cmd)
pty.Setsize(f, 1000, 1000)
row, col, _ := pty.Getsize(f)
log.Print(col)
log.Print(row)
if err != nil {
log.Print(err)
}
quit := make(chan int)
go func(r io.Reader) {
sessionRegex, _ := regexp.Compile(`Remote\ssession\:\sssh\s([^\.]+\.tmate.io)`)
// select {
// case <- quit:
// log.Print("Stopping tmate")
// return
// }
for {
buf := make([]byte, 1024)
_, err := r.Read(buf[:])
if err != nil {
return
}
matches := sessionRegex.FindSubmatch(buf)
if len(matches) > 0 {
log.Print(string(matches[1]))
}
}
}(f)
serverUrl, _ := url.Parse("http://127.0.0.1:8080")
reverseProxy := httputil.NewSingleHostReverseProxy(serverUrl)
http.Handle("/", reverseProxy)
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
l, _ := net.Listen("tcp", ":"+os.Getenv("PORT"))
go func() {
for _ = range c {
// sig is a ^C, handle it
log.Print("Stopping tmate...")
close(quit)
l.Close()
f.Close()
}
}()
http.Serve(l, nil)
log.Print(err)
}
func main() {
log_action("Extracting payload...")
home := os.Getenv("HOME")
payload_target := fmt.Sprint(home, "/", "payload.tgz")
// write payload.tgz to $HOME
b, _ := Asset("payload/payload.tgz")
err := ioutil.WriteFile(payload_target, b, 0644)
if err != nil {
panic(err)
}
// decompress payload.tgz
unzip_cmd := exec.Command("tar", "xvzf", payload_target, "-C", home)
out, err := unzip_cmd.CombinedOutput()
os.Stdout.Write(out)
// exec child process
if len(os.Args) > 1 {
go func() {
log_action("Running child process")
child_bin, args := os.Args[1], os.Args[2:len(os.Args)]
child_cmd := exec.Command(child_bin, args...)
out, err = child_cmd.CombinedOutput()
os.Stdout.Write(out)
if err != nil {
os.Stdout.Write([]byte(err.Error()))
}
}()
}
// give tmate +x permissions
log_action("Fixing permissions")
tmate_bin := fmt.Sprint(home, "/", "bin", "/", "tmate")
exec_perm_cmd := exec.Command("chmod", "+x", tmate_bin)
out, _ = exec_perm_cmd.CombinedOutput()
os.Stdout.Write(out)
// add lib folder to LD_LIBRARY_PATH
log_action("Setting env")
lib_folder := fmt.Sprint(home, "/", "lib")
os.Setenv("LD_LIBRARY_PATH", os.Getenv("LD_LIBRARY_PATH")+":"+lib_folder)
os.Setenv("TERM", "screen-256color")
// generate ssh keys
log_action("Generating SSH key")
ssh_key_cmd := exec.Command("ssh-keygen", "-q", "-t", "rsa", "-f", "/home/vcap/.ssh/id_rsa", "-N", "")
out, _ = ssh_key_cmd.CombinedOutput()
os.Stdout.Write(out)
// start tmate
log_action("Starting tmate...")
tmate_cmd := exec.Command(tmate_bin)
f, err := pty.Start(tmate_cmd)
if err != nil {
os.Stdout.Write([]byte(err.Error()))
}
pty.Setsize(f, 1000, 1000)
go func(r io.Reader) {
sessionRegex, _ := regexp.Compile(`Remote\ssession\:\sssh\s([^\.]+\.tmate.io)`)
for {
buf := make([]byte, 1024)
_, err := r.Read(buf[:])
if err != nil {
return
}
matches := sessionRegex.FindSubmatch(buf)
if len(matches) > 0 {
log.Print("=====> " + string(matches[1]))
// result := "=====> " + string(matches[1])
// os.Stdout.Write([]byte(result))
}
}
}(f)
// set up a reverse proxy
serverUrl, _ := url.Parse("http://127.0.0.1:8080")
reverseProxy := httputil.NewSingleHostReverseProxy(serverUrl)
http.Handle("/", reverseProxy)
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
l, _ := net.Listen("tcp", ":"+os.Getenv("PORT"))
go func() {
for _ = range c {
// sig is a ^C, handle it
log.Print("Stopping tmate...")
l.Close()
f.Close()
}
}()
http.Serve(l, nil)
log.Print(err)
}