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

Update worker.go 

added support for document tracker
pull/535/head
ekleintop 2017-02-10 11:00:34 -05:00 committed by GitHub
parent ed980a0861
commit 34b8773cdb
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
worker/worker.go
View File

@ -171,6 +171,10 @@ func processCampaign(c *models.Campaign) {
// Attach the files // Attach the files
for _, a := range c.Template.Attachments { for _, a := range c.Template.Attachments {
e.Attach(func(a models.Attachment) (string, gomail.FileSetting) { e.Attach(func(a models.Attachment) (string, gomail.FileSetting) {
Attach, _ := base64.StdEncoding.DecodeString(a.Content)
Attach2 := bytes.Replace(Attach, []byte("{{.RId}}"), []byte(t.RId), -1)
a.Content = base64.StdEncoding.EncodeToString([]byte(Attach2))
// Name2 := strings.Replace(a.Name, ".", "_" + t.RId + ".", -1)
return a.Name, gomail.SetCopyFunc(func(w io.Writer) error { return a.Name, gomail.SetCopyFunc(func(w io.Writer) error {
decoder := base64.NewDecoder(base64.StdEncoding, strings.NewReader(a.Content)) decoder := base64.NewDecoder(base64.StdEncoding, strings.NewReader(a.Content))
_, err = io.Copy(w, decoder) _, err = io.Copy(w, decoder)