mirror of
https://github.com/gophish/gophish
synced 2024-11-14 16:27:23 +00:00
47f0049c30
* Refactoring servers to support custom workers and graceful shutdown. * Refactoring workers to support custom mailers. * Refactoring mailer to be an interface, with proper instances instead of a single global instance * Cleaning up a few things. Locking maillogs for campaigns set to launch immediately to prevent a race condition. * Cleaning up API middleware to be simpler * Moving template parameters to separate struct * Changed LoadConfig to return config object * Cleaned up some error handling, removing uninitialized global error in models package * Changed static file serving to use the unindexed package
112 lines
3.3 KiB
Go
112 lines
3.3 KiB
Go
package main
|
|
|
|
/*
|
|
gophish - Open-Source Phishing Framework
|
|
|
|
The MIT License (MIT)
|
|
|
|
Copyright (c) 2013 Jordan Wright
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in
|
|
all copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
THE SOFTWARE.
|
|
*/
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"os/signal"
|
|
|
|
"gopkg.in/alecthomas/kingpin.v2"
|
|
|
|
"github.com/gophish/gophish/auth"
|
|
"github.com/gophish/gophish/config"
|
|
"github.com/gophish/gophish/controllers"
|
|
log "github.com/gophish/gophish/logger"
|
|
"github.com/gophish/gophish/models"
|
|
)
|
|
|
|
var (
|
|
configPath = kingpin.Flag("config", "Location of config.json.").Default("./config.json").String()
|
|
disableMailer = kingpin.Flag("disable-mailer", "Disable the mailer (for use with multi-system deployments)").Bool()
|
|
)
|
|
|
|
func main() {
|
|
// Load the version
|
|
|
|
version, err := ioutil.ReadFile("./VERSION")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
kingpin.Version(string(version))
|
|
|
|
// Parse the CLI flags and load the config
|
|
kingpin.CommandLine.HelpFlag.Short('h')
|
|
kingpin.Parse()
|
|
|
|
// Load the config
|
|
conf, err := config.LoadConfig(*configPath)
|
|
// Just warn if a contact address hasn't been configured
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if conf.ContactAddress == "" {
|
|
log.Warnf("No contact address has been configured.")
|
|
log.Warnf("Please consider adding a contact_address entry in your config.json")
|
|
}
|
|
config.Version = string(version)
|
|
|
|
err = log.Setup(conf)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// Provide the option to disable the built-in mailer
|
|
// Setup the global variables and settings
|
|
err = models.Setup(conf)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
// Unlock any maillogs that may have been locked for processing
|
|
// when Gophish was last shutdown.
|
|
err = models.UnlockAllMailLogs()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// Create our servers
|
|
adminOptions := []controllers.AdminServerOption{}
|
|
if *disableMailer {
|
|
adminOptions = append(adminOptions, controllers.WithWorker(nil))
|
|
}
|
|
adminConfig := conf.AdminConf
|
|
adminServer := controllers.NewAdminServer(adminConfig, adminOptions...)
|
|
auth.Store.Options.Secure = adminConfig.UseTLS
|
|
|
|
phishConfig := conf.PhishConf
|
|
phishServer := controllers.NewPhishingServer(phishConfig)
|
|
|
|
go adminServer.Start()
|
|
go phishServer.Start()
|
|
|
|
// Handle graceful shutdown
|
|
c := make(chan os.Signal, 1)
|
|
signal.Notify(c, os.Interrupt)
|
|
<-c
|
|
log.Info("CTRL+C Received... Gracefully shutting down servers")
|
|
adminServer.Shutdown()
|
|
phishServer.Shutdown()
|
|
}
|