Added a handler for SSH env.

Author: technosophos

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()