Merge pull request #11 from gliderlabs/master

v0.2.1

Author: progrium

Makefile

@@ -2,15 +2,15 @@ NAME=sshfront
 OWNER=gliderlabs
 ARCH=$(shell uname -m)
 RMFLAG=--rm
-VERSION=0.2.0
+VERSION=0.2.1
 
 build:
 	mkdir -p build/Linux && GOOS=linux CGO_ENABLED=0 go build -a \
-		-ldflags "-X main.Version $(VERSION)" \
+		-ldflags "-X main.Version=$(VERSION)" \
 		-installsuffix cgo \
 		-o build/Linux/$(NAME)
 	mkdir -p build/Darwin && GOOS=darwin CGO_ENABLED=0 go build -a \
-		-ldflags "-X main.Version $(VERSION)" \
+		-ldflags "-X main.Version=$(VERSION)" \
 		-installsuffix cgo \
 		-o build/Darwin/$(NAME)
 

README.md

@@ -1,5 +1,8 @@
 # sshfront
 
+[![CircleCI](https://img.shields.io/circleci/project/gliderlabs/sshfront/release.svg)](https://circleci.com/gh/gliderlabs/sshfront)
+[![IRC Channel](https://img.shields.io/badge/irc-%23gliderlabs-blue.svg)](https://kiwiirc.com/client/irc.freenode.net/#gliderlabs)
+
 A lightweight SSH server frontend where authentication and connections
 are controlled with command handlers / shell scripts.
 
@@ -74,3 +77,4 @@ This project was made possible thanks to [Deis](http://deis.io) and [DigitalOcea
 ## License
 
 MIT
+<img src="https://ga-beacon.appspot.com/UA-58928488-2/sshfront/readme?pixel" />

example/stdinecho

@@ -0,0 +1,3 @@
+#!/bin/bash
+echo "Stdin:"
+cat

handlers.go

@@ -163,10 +163,16 @@ func (h *sshHandler) Exit(err error) error {
 	return err
 }
 
+type EnvVar struct {
+	Name, Value string
+}
+
 func (h *sshHandler) Request(req *ssh.Request) {
 	switch req.Type {
 	case "exec":
 		h.handleExec(req)
+	case "env":
+		h.handleEnv(req)
 	case "pty-req":
 		h.handlePty(req)
 	case "window-change":
@@ -178,6 +184,14 @@ func (h *sshHandler) Request(req *ssh.Request) {
 	}
 }
 
+func (h *sshHandler) handleEnv(req *ssh.Request) {
+	var pair EnvVar
+	ssh.Unmarshal(req.Payload, &pair)
+	envvar := fmt.Sprintf("%s=%s", pair.Name, pair.Value)
+	h.Env = append(h.Env, envvar)
+	req.Reply(true, nil)
+}
+
 func (h *sshHandler) handleExec(req *ssh.Request) {
 	h.Lock()
 	defer h.Unlock()
@@ -207,7 +221,10 @@ func (h *sshHandler) handleExec(req *ssh.Request) {
 		h.channel.Close()
 		return
 	}
-	go io.Copy(stdinPipe, h.channel)
+	go func() {
+		defer stdinPipe.Close()
+		io.Copy(stdinPipe, h.channel)
+	}()
 
 	if req.WantReply {
 		req.Reply(true, nil)