cristiancmoises
/
gophish
mirror of https://github.com/gophish/gophish
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added error handling on startup of web servers. Fixes #70

pull/75/head
Jordan Wright 2016-01-14 22:46:43 -06:00
parent 0d57a78b74
commit 89e72dd78d
1 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
gophish.go
View File

@ -30,11 +30,12 @@ import (
"log" "log"
"net/http" "net/http"
"os" "os"
"sync"
"github.com/gorilla/handlers"
"github.com/gophish/gophish/config" "github.com/gophish/gophish/config"
"github.com/gophish/gophish/controllers" "github.com/gophish/gophish/controllers"
"github.com/gophish/gophish/models" "github.com/gophish/gophish/models"
"github.com/gorilla/handlers"
) )
var Logger = log.New(os.Stdout, " ", log.Ldate|log.Ltime|log.Lshortfile) var Logger = log.New(os.Stdout, " ", log.Ldate|log.Ltime|log.Lshortfile)
@ -45,9 +46,19 @@ func main() {
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
} }
wg := &sync.WaitGroup{}
wg.Add(1)
// Start the web servers // Start the web servers
Logger.Printf("Admin server started at http://%s\n", config.Conf.AdminURL) go func() {
go http.ListenAndServe(config.Conf.AdminURL, handlers.CombinedLoggingHandler(os.Stdout, controllers.CreateAdminRouter())) defer wg.Done()
Logger.Printf("Phishing server started at http://%s\n", config.Conf.PhishURL) Logger.Printf("Starting admin server at http://%s\n", config.Conf.AdminURL)
http.ListenAndServe(config.Conf.PhishURL, handlers.CombinedLoggingHandler(os.Stdout, controllers.CreatePhishingRouter())) Logger.Fatal(http.ListenAndServe(config.Conf.AdminURL, handlers.CombinedLoggingHandler(os.Stdout, controllers.CreateAdminRouter())))
}()
wg.Add(1)
go func() {
defer wg.Done()
Logger.Printf("Starting phishing server at http://%s\n", config.Conf.PhishURL)
Logger.Fatal(http.ListenAndServe(config.Conf.PhishURL, handlers.CombinedLoggingHandler(os.Stdout, controllers.CreatePhishingRouter())))
}()
wg.Wait()
} }