From 3c8280c08ecc008b27fa3ef738c49363baed2a43 Mon Sep 17 00:00:00 2001 From: evilsocket Date: Wed, 24 Jan 2018 11:37:13 +0100 Subject: [PATCH] balls --- main.go | 4 ---- session/session.go | 2 +- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/main.go b/main.go index 0dfec2d5..584eeb34 100644 --- a/main.go +++ b/main.go @@ -61,10 +61,6 @@ func main() { log.Fatal("%s", err) } - if line == "" || line[0] == '#' { - continue - } - for _, cmd := range session.ParseCommands(line) { if err = sess.Run(cmd); err != nil { log.Error("%s", err) diff --git a/session/session.go b/session/session.go index cc33b633..e311c528 100644 --- a/session/session.go +++ b/session/session.go @@ -50,7 +50,7 @@ func ParseCommands(buffer string) []string { cmds := make([]string, 0) for _, cmd := range strings.Split(buffer, ";") { cmd = strings.Trim(cmd, "\r\n\t ") - if cmd != "" { + if cmd != "" || (len(cmd) > 0 && cmd[0] != '#') { cmds = append(cmds, cmd) } }