mirror of https://github.com/gophish/gophish
65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
package worker
|
|
|
|
import (
|
|
"log"
|
|
"net/smtp"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/jordan-wright/email"
|
|
"github.com/jordan-wright/gophish/models"
|
|
)
|
|
|
|
var Logger = log.New(os.Stdout, " ", log.Ldate|log.Ltime|log.Lshortfile)
|
|
|
|
type Worker struct {
|
|
Queue chan *models.Campaign
|
|
}
|
|
|
|
func New() *Worker {
|
|
return &Worker{
|
|
Queue: make(chan *models.Campaign),
|
|
}
|
|
}
|
|
|
|
// Start launches the worker to monitor the database for any jobs.
|
|
// If a job is found, it launches the job
|
|
func (w *Worker) Start() {
|
|
Logger.Println("Background Worker Started Successfully - Waiting for Campaigns")
|
|
for {
|
|
processCampaign(<-w.Queue)
|
|
}
|
|
}
|
|
|
|
func processCampaign(c *models.Campaign) {
|
|
Logger.Printf("Worker received: %s", c.Name)
|
|
err := models.UpdateCampaignStatus(c, models.IN_PROGRESS)
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
}
|
|
e := email.Email{
|
|
Subject: c.Template.Subject,
|
|
From: c.SMTP.FromAddress,
|
|
Text: []byte(c.Template.Text),
|
|
HTML: []byte(c.Template.HTML),
|
|
}
|
|
Logger.Println(c.SMTP.Username)
|
|
Logger.Println(c.SMTP.Password)
|
|
Logger.Println(c.SMTP.FromAddress)
|
|
var auth smtp.Auth
|
|
if c.SMTP.Username != "" && c.SMTP.Password != "" {
|
|
auth = smtp.PlainAuth("", c.SMTP.Username, c.SMTP.Password, strings.Split(c.SMTP.Host, ":")[0])
|
|
}
|
|
for _, t := range c.Results {
|
|
Logger.Println("Creating email using template")
|
|
e.To = []string{t.Email}
|
|
Logger.Println(e.To)
|
|
Logger.Println(e.From)
|
|
err := e.Send(c.SMTP.Host, auth)
|
|
if err != nil {
|
|
Logger.Println(err)
|
|
}
|
|
Logger.Printf("Sending Email to %s\n", t.Email)
|
|
}
|
|
}
|