misc: small fix or general refactoring i did not bother commenting

This commit is contained in:
evilsocket 2018-02-11 18:56:53 +01:00
parent 47230567ba
commit 1847eb3b52

17
main.go
View file

@ -14,6 +14,13 @@ import (
var sess *session.Session var sess *session.Session
var err error var err error
// Some modules are enabled by default in order
// to make the interactive session useful.
var autoEnableList = []string{
"events.stream",
"net.recon",
}
func main() { func main() {
if sess, err = session.New(); err != nil { if sess, err = session.New(); err != nil {
fmt.Println(err) fmt.Println(err)
@ -24,7 +31,9 @@ func main() {
fmt.Printf("\n\nWARNING: This terminal does not support colors, view will be very limited.\n\n") fmt.Printf("\n\nWARNING: This terminal does not support colors, view will be very limited.\n\n")
} }
fmt.Printf(core.Bold("%s v%s\n\n"), core.Name, core.Version) appName := fmt.Sprintf("%s v%s", core.Name, core.Version)
fmt.Printf("%s (type '%s' for a list of commands)\n\n", core.Bold(appName), core.Bold("help"))
sess.Register(modules.NewEventsStream(sess)) sess.Register(modules.NewEventsStream(sess))
sess.Register(modules.NewTicker(sess)) sess.Register(modules.NewTicker(sess))
@ -45,12 +54,6 @@ func main() {
log.Fatal("%s", err) log.Fatal("%s", err)
} }
// Some modules are enabled by default in order
// to make the interactive session useful.
autoEnableList := []string{
"events.stream",
"net.recon",
}
for _, modName := range autoEnableList { for _, modName := range autoEnableList {
if err = sess.Run(modName + " on"); err != nil { if err = sess.Run(modName + " on"); err != nil {
log.Fatal("Error while starting module %s: %", modName, err) log.Fatal("Error while starting module %s: %", modName, err)