mirror of https://github.com/gophish/gophish
491 lines
15 KiB
Go
491 lines
15 KiB
Go
package controllers
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"html/template"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/gophish/gophish/auth"
|
|
"github.com/gophish/gophish/config"
|
|
ctx "github.com/gophish/gophish/context"
|
|
mid "github.com/gophish/gophish/middleware"
|
|
"github.com/gophish/gophish/models"
|
|
"github.com/gorilla/csrf"
|
|
"github.com/gorilla/mux"
|
|
"github.com/gorilla/sessions"
|
|
)
|
|
|
|
// Logger is used to send logging messages to stdout.
|
|
var Logger = log.New(os.Stdout, " ", log.Ldate|log.Ltime|log.Lshortfile)
|
|
|
|
// CreateAdminRouter creates the routes for handling requests to the web interface.
|
|
// This function returns an http.Handler to be used in http.ListenAndServe().
|
|
func CreateAdminRouter() http.Handler {
|
|
router := mux.NewRouter()
|
|
// Base Front-end routes
|
|
router.HandleFunc("/", Use(Base, mid.RequireLogin))
|
|
router.HandleFunc("/login", Login)
|
|
router.HandleFunc("/logout", Use(Logout, mid.RequireLogin))
|
|
router.HandleFunc("/campaigns", Use(Campaigns, mid.RequireLogin))
|
|
router.HandleFunc("/campaigns/{id:[0-9]+}", Use(CampaignID, mid.RequireLogin))
|
|
router.HandleFunc("/templates", Use(Templates, mid.RequireLogin))
|
|
router.HandleFunc("/users", Use(Users, mid.RequireLogin))
|
|
router.HandleFunc("/landing_pages", Use(LandingPages, mid.RequireLogin))
|
|
router.HandleFunc("/sending_profiles", Use(SendingProfiles, mid.RequireLogin))
|
|
router.HandleFunc("/register", Use(Register, mid.RequireLogin))
|
|
router.HandleFunc("/settings", Use(Settings, mid.RequireLogin))
|
|
// Create the API routes
|
|
api := router.PathPrefix("/api").Subrouter()
|
|
api = api.StrictSlash(true)
|
|
api.HandleFunc("/", Use(API, mid.RequireLogin))
|
|
api.HandleFunc("/reset", Use(API_Reset, mid.RequireLogin))
|
|
api.HandleFunc("/campaigns/", Use(API_Campaigns, mid.RequireAPIKey))
|
|
api.HandleFunc("/campaigns/{id:[0-9]+}", Use(API_Campaigns_Id, mid.RequireAPIKey))
|
|
api.HandleFunc("/campaigns/{id:[0-9]+}/results", Use(API_Campaigns_Id_Results, mid.RequireAPIKey))
|
|
api.HandleFunc("/campaigns/{id:[0-9]+}/complete", Use(API_Campaigns_Id_Complete, mid.RequireAPIKey))
|
|
api.HandleFunc("/groups/", Use(API_Groups, mid.RequireAPIKey))
|
|
api.HandleFunc("/groups/{id:[0-9]+}", Use(API_Groups_Id, mid.RequireAPIKey))
|
|
api.HandleFunc("/templates/", Use(API_Templates, mid.RequireAPIKey))
|
|
api.HandleFunc("/templates/{id:[0-9]+}", Use(API_Templates_Id, mid.RequireAPIKey))
|
|
api.HandleFunc("/pages/", Use(API_Pages, mid.RequireAPIKey))
|
|
api.HandleFunc("/pages/{id:[0-9]+}", Use(API_Pages_Id, mid.RequireAPIKey))
|
|
api.HandleFunc("/smtp/", Use(API_SMTP, mid.RequireAPIKey))
|
|
api.HandleFunc("/smtp/{id:[0-9]+}", Use(API_SMTP_Id, mid.RequireAPIKey))
|
|
api.HandleFunc("/util/send_test_email", Use(API_Send_Test_Email, mid.RequireAPIKey))
|
|
api.HandleFunc("/import/group", API_Import_Group)
|
|
api.HandleFunc("/import/email", API_Import_Email)
|
|
api.HandleFunc("/import/site", API_Import_Site)
|
|
|
|
// Setup static file serving
|
|
router.PathPrefix("/").Handler(http.FileServer(http.Dir("./static/")))
|
|
|
|
// Setup CSRF Protection
|
|
csrfHandler := csrf.Protect([]byte(auth.GenerateSecureKey()),
|
|
csrf.FieldName("csrf_token"),
|
|
csrf.Secure(config.Conf.AdminConf.UseTLS))
|
|
csrfRouter := csrfHandler(router)
|
|
return Use(csrfRouter.ServeHTTP, mid.CSRFExceptions, mid.GetContext)
|
|
}
|
|
|
|
// CreatePhishingRouter creates the router that handles phishing connections.
|
|
func CreatePhishingRouter() http.Handler {
|
|
router := mux.NewRouter()
|
|
router.PathPrefix("/static/").Handler(http.StripPrefix("/static/", http.FileServer(http.Dir("./static/endpoint/"))))
|
|
router.HandleFunc("/track", PhishTracker)
|
|
router.HandleFunc("/{path:.*}", PhishHandler)
|
|
return router
|
|
}
|
|
|
|
// PhishTracker tracks emails as they are opened, updating the status for the given Result
|
|
func PhishTracker(w http.ResponseWriter, r *http.Request) {
|
|
r.ParseForm()
|
|
id := r.Form.Get("rid")
|
|
if id == "" {
|
|
Logger.Println("Missing Result ID")
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
rs, err := models.GetResult(id)
|
|
if err != nil {
|
|
Logger.Println("No Results found")
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
c, err := models.GetCampaign(rs.CampaignId, rs.UserId)
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
}
|
|
// Don't process events for completed campaigns
|
|
if c.Status == models.CAMPAIGN_COMPLETE {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
c.AddEvent(models.Event{Email: rs.Email, Message: models.EVENT_OPENED})
|
|
// Don't update the status if the user already clicked the link
|
|
// or submitted data to the campaign
|
|
if rs.Status == models.STATUS_SUCCESS {
|
|
http.ServeFile(w, r, "static/images/pixel.png")
|
|
return
|
|
}
|
|
err = rs.UpdateStatus(models.EVENT_OPENED)
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
}
|
|
ip, _, err := net.SplitHostPort(r.RemoteAddr)
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
return
|
|
}
|
|
// Respect X-Forwarded headers
|
|
if fips := r.Header.Get("X-Forwarded-For"); fips != "" {
|
|
ip = strings.Split(fips, ", ")[0]
|
|
}
|
|
// Handle post processing such as GeoIP
|
|
err = rs.UpdateGeo(ip)
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
}
|
|
http.ServeFile(w, r, "static/images/pixel.png")
|
|
}
|
|
|
|
// PhishHandler handles incoming client connections and registers the associated actions performed
|
|
// (such as clicked link, etc.)
|
|
func PhishHandler(w http.ResponseWriter, r *http.Request) {
|
|
err := r.ParseForm()
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
id := r.Form.Get("rid")
|
|
if id == "" {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
rs, err := models.GetResult(id)
|
|
if err != nil {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
c, err := models.GetCampaign(rs.CampaignId, rs.UserId)
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
}
|
|
// Don't process events for completed campaigns
|
|
if c.Status == models.CAMPAIGN_COMPLETE {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
rs.UpdateStatus(models.STATUS_SUCCESS)
|
|
p, err := models.GetPage(c.PageId, c.UserId)
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
}
|
|
d := struct {
|
|
Payload url.Values `json:"payload"`
|
|
Browser map[string]string `json:"browser"`
|
|
}{
|
|
Payload: r.Form,
|
|
Browser: make(map[string]string),
|
|
}
|
|
ip, _, err := net.SplitHostPort(r.RemoteAddr)
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
return
|
|
}
|
|
// Respect X-Forwarded headers
|
|
if fips := r.Header.Get("X-Forwarded-For"); fips != "" {
|
|
ip = strings.Split(fips, ", ")[0]
|
|
}
|
|
// Handle post processing such as GeoIP
|
|
err = rs.UpdateGeo(ip)
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
}
|
|
d.Browser["address"] = ip
|
|
d.Browser["user-agent"] = r.Header.Get("User-Agent")
|
|
rj, err := json.Marshal(d)
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
switch {
|
|
case r.Method == "GET":
|
|
err = c.AddEvent(models.Event{Email: rs.Email, Message: models.EVENT_CLICKED, Details: string(rj)})
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
}
|
|
case r.Method == "POST":
|
|
// If data was POST'ed, let's record it
|
|
// Store the data in an event
|
|
c.AddEvent(models.Event{Email: rs.Email, Message: models.EVENT_DATA_SUBMIT, Details: string(rj)})
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
}
|
|
// Redirect to the desired page
|
|
if p.RedirectURL != "" {
|
|
http.Redirect(w, r, p.RedirectURL, 302)
|
|
return
|
|
}
|
|
}
|
|
var htmlBuff bytes.Buffer
|
|
tmpl, err := template.New("html_template").Parse(p.HTML)
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
http.NotFound(w, r)
|
|
}
|
|
err = tmpl.Execute(&htmlBuff, rs)
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
http.NotFound(w, r)
|
|
}
|
|
w.Write(htmlBuff.Bytes())
|
|
}
|
|
|
|
// Use allows us to stack middleware to process the request
|
|
// Example taken from https://github.com/gorilla/mux/pull/36#issuecomment-25849172
|
|
func Use(handler http.HandlerFunc, mid ...func(http.Handler) http.HandlerFunc) http.HandlerFunc {
|
|
for _, m := range mid {
|
|
handler = m(handler)
|
|
}
|
|
return handler
|
|
}
|
|
|
|
// Register creates a new user
|
|
func Register(w http.ResponseWriter, r *http.Request) {
|
|
// If it is a post request, attempt to register the account
|
|
// Now that we are all registered, we can log the user in
|
|
params := struct {
|
|
Title string
|
|
Flashes []interface{}
|
|
User models.User
|
|
Token string
|
|
}{Title: "Register", Token: csrf.Token(r)}
|
|
session := ctx.Get(r, "session").(*sessions.Session)
|
|
switch {
|
|
case r.Method == "GET":
|
|
params.Flashes = session.Flashes()
|
|
session.Save(r, w)
|
|
templates := template.New("template")
|
|
_, err := templates.ParseFiles("templates/register.html", "templates/flashes.html")
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
}
|
|
template.Must(templates, err).ExecuteTemplate(w, "base", params)
|
|
case r.Method == "POST":
|
|
//Attempt to register
|
|
succ, err := auth.Register(r)
|
|
//If we've registered, redirect to the login page
|
|
if succ {
|
|
session.AddFlash(models.Flash{
|
|
Type: "success",
|
|
Message: "Registration successful!.",
|
|
})
|
|
session.Save(r, w)
|
|
http.Redirect(w, r, "/login", 302)
|
|
return
|
|
}
|
|
// Check the error
|
|
m := err.Error()
|
|
Logger.Println(err)
|
|
session.AddFlash(models.Flash{
|
|
Type: "danger",
|
|
Message: m,
|
|
})
|
|
session.Save(r, w)
|
|
http.Redirect(w, r, "/register", 302)
|
|
return
|
|
}
|
|
}
|
|
|
|
// Base handles the default path and template execution
|
|
func Base(w http.ResponseWriter, r *http.Request) {
|
|
params := struct {
|
|
User models.User
|
|
Title string
|
|
Flashes []interface{}
|
|
Token string
|
|
}{Title: "Dashboard", User: ctx.Get(r, "user").(models.User), Token: csrf.Token(r)}
|
|
getTemplate(w, "dashboard").ExecuteTemplate(w, "base", params)
|
|
}
|
|
|
|
// Campaigns handles the default path and template execution
|
|
func Campaigns(w http.ResponseWriter, r *http.Request) {
|
|
// Example of using session - will be removed.
|
|
params := struct {
|
|
User models.User
|
|
Title string
|
|
Flashes []interface{}
|
|
Token string
|
|
}{Title: "Campaigns", User: ctx.Get(r, "user").(models.User), Token: csrf.Token(r)}
|
|
getTemplate(w, "campaigns").ExecuteTemplate(w, "base", params)
|
|
}
|
|
|
|
// CampaignID handles the default path and template execution
|
|
func CampaignID(w http.ResponseWriter, r *http.Request) {
|
|
// Example of using session - will be removed.
|
|
params := struct {
|
|
User models.User
|
|
Title string
|
|
Flashes []interface{}
|
|
Token string
|
|
}{Title: "Campaign Results", User: ctx.Get(r, "user").(models.User), Token: csrf.Token(r)}
|
|
getTemplate(w, "campaign_results").ExecuteTemplate(w, "base", params)
|
|
}
|
|
|
|
// Templates handles the default path and template execution
|
|
func Templates(w http.ResponseWriter, r *http.Request) {
|
|
// Example of using session - will be removed.
|
|
params := struct {
|
|
User models.User
|
|
Title string
|
|
Flashes []interface{}
|
|
Token string
|
|
}{Title: "Email Templates", User: ctx.Get(r, "user").(models.User), Token: csrf.Token(r)}
|
|
getTemplate(w, "templates").ExecuteTemplate(w, "base", params)
|
|
}
|
|
|
|
// Users handles the default path and template execution
|
|
func Users(w http.ResponseWriter, r *http.Request) {
|
|
// Example of using session - will be removed.
|
|
params := struct {
|
|
User models.User
|
|
Title string
|
|
Flashes []interface{}
|
|
Token string
|
|
}{Title: "Users & Groups", User: ctx.Get(r, "user").(models.User), Token: csrf.Token(r)}
|
|
getTemplate(w, "users").ExecuteTemplate(w, "base", params)
|
|
}
|
|
|
|
// LandingPages handles the default path and template execution
|
|
func LandingPages(w http.ResponseWriter, r *http.Request) {
|
|
// Example of using session - will be removed.
|
|
params := struct {
|
|
User models.User
|
|
Title string
|
|
Flashes []interface{}
|
|
Token string
|
|
}{Title: "Landing Pages", User: ctx.Get(r, "user").(models.User), Token: csrf.Token(r)}
|
|
getTemplate(w, "landing_pages").ExecuteTemplate(w, "base", params)
|
|
}
|
|
|
|
// SendingProfiles handles the default path and template execution
|
|
func SendingProfiles(w http.ResponseWriter, r *http.Request) {
|
|
// Example of using session - will be removed.
|
|
params := struct {
|
|
User models.User
|
|
Title string
|
|
Flashes []interface{}
|
|
Token string
|
|
}{Title: "Sending Profiles", User: ctx.Get(r, "user").(models.User), Token: csrf.Token(r)}
|
|
getTemplate(w, "sending_profiles").ExecuteTemplate(w, "base", params)
|
|
}
|
|
|
|
// Settings handles the changing of settings
|
|
func Settings(w http.ResponseWriter, r *http.Request) {
|
|
switch {
|
|
case r.Method == "GET":
|
|
params := struct {
|
|
User models.User
|
|
Title string
|
|
Flashes []interface{}
|
|
Token string
|
|
Version string
|
|
}{Title: "Settings", Version: config.Version, User: ctx.Get(r, "user").(models.User), Token: csrf.Token(r)}
|
|
getTemplate(w, "settings").ExecuteTemplate(w, "base", params)
|
|
case r.Method == "POST":
|
|
err := auth.ChangePassword(r)
|
|
msg := models.Response{Success: true, Message: "Settings Updated Successfully"}
|
|
if err == auth.ErrInvalidPassword {
|
|
msg.Message = "Invalid Password"
|
|
msg.Success = false
|
|
JSONResponse(w, msg, http.StatusBadRequest)
|
|
return
|
|
}
|
|
if err != nil {
|
|
msg.Message = err.Error()
|
|
msg.Success = false
|
|
JSONResponse(w, msg, http.StatusBadRequest)
|
|
return
|
|
}
|
|
JSONResponse(w, msg, http.StatusOK)
|
|
}
|
|
}
|
|
|
|
// Login handles the authentication flow for a user. If credentials are valid,
|
|
// a session is created
|
|
func Login(w http.ResponseWriter, r *http.Request) {
|
|
params := struct {
|
|
User models.User
|
|
Title string
|
|
Flashes []interface{}
|
|
Token string
|
|
}{Title: "Login", Token: csrf.Token(r)}
|
|
session := ctx.Get(r, "session").(*sessions.Session)
|
|
switch {
|
|
case r.Method == "GET":
|
|
params.Flashes = session.Flashes()
|
|
session.Save(r, w)
|
|
templates := template.New("template")
|
|
_, err := templates.ParseFiles("templates/login.html", "templates/flashes.html")
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
}
|
|
template.Must(templates, err).ExecuteTemplate(w, "base", params)
|
|
case r.Method == "POST":
|
|
//Attempt to login
|
|
succ, u, err := auth.Login(r)
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
}
|
|
//If we've logged in, save the session and redirect to the dashboard
|
|
if succ {
|
|
session.Values["id"] = u.Id
|
|
session.Save(r, w)
|
|
http.Redirect(w, r, "/", 302)
|
|
} else {
|
|
Flash(w, r, "danger", "Invalid Username/Password")
|
|
http.Redirect(w, r, "/login", 302)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Logout destroys the current user session
|
|
func Logout(w http.ResponseWriter, r *http.Request) {
|
|
// If it is a post request, attempt to register the account
|
|
// Now that we are all registered, we can log the user in
|
|
session := ctx.Get(r, "session").(*sessions.Session)
|
|
delete(session.Values, "id")
|
|
Flash(w, r, "success", "You have successfully logged out")
|
|
http.Redirect(w, r, "/login", 302)
|
|
}
|
|
|
|
// Preview allows for the viewing of page html in a separate browser window
|
|
func Preview(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != "POST" {
|
|
http.Error(w, "Method not allowed", http.StatusBadRequest)
|
|
return
|
|
}
|
|
fmt.Fprintf(w, "%s", r.FormValue("html"))
|
|
}
|
|
|
|
// Clone takes a URL as a POST parameter and returns the site HTML
|
|
func Clone(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
if r.Method != "POST" {
|
|
http.Error(w, "Method not allowed", http.StatusBadRequest)
|
|
return
|
|
}
|
|
if url, ok := vars["url"]; ok {
|
|
Logger.Println(url)
|
|
}
|
|
http.Error(w, "No URL given.", http.StatusBadRequest)
|
|
}
|
|
|
|
func getTemplate(w http.ResponseWriter, tmpl string) *template.Template {
|
|
templates := template.New("template")
|
|
_, err := templates.ParseFiles("templates/base.html", "templates/"+tmpl+".html", "templates/flashes.html")
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
}
|
|
return template.Must(templates, err)
|
|
}
|
|
|
|
// Flash handles the rendering flash messages
|
|
func Flash(w http.ResponseWriter, r *http.Request, t string, m string) {
|
|
session := ctx.Get(r, "session").(*sessions.Session)
|
|
session.AddFlash(models.Flash{
|
|
Type: t,
|
|
Message: m,
|
|
})
|
|
session.Save(r, w)
|
|
}
|