bench-forgejo/modules/mailer/mailer.go

283 lines
6.6 KiB
Go
Raw Normal View History

2014-03-19 17:57:27 +05:30
// Copyright 2014 The Gogs Authors. All rights reserved.
// Copyright 2017 The Gitea Authors. All rights reserved.
2014-03-19 17:57:27 +05:30
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package mailer
import (
"crypto/tls"
2014-03-19 17:57:27 +05:30
"fmt"
"io"
"net"
2014-03-19 17:57:27 +05:30
"net/smtp"
"os"
"os/exec"
2014-03-19 17:57:27 +05:30
"strings"
"time"
"code.gitea.io/gitea/modules/log"
"code.gitea.io/gitea/modules/setting"
"github.com/jaytaylor/html2text"
"gopkg.in/gomail.v2"
2014-03-19 17:57:27 +05:30
)
2016-11-25 07:14:04 +05:30
// Message mail body and log info
type Message struct {
Info string // Message information for log purpose.
*gomail.Message
}
// NewMessageFrom creates new mail message object with custom From header.
func NewMessageFrom(to []string, from, subject, body string) *Message {
log.Trace("NewMessageFrom (body):\n%s", body)
msg := gomail.NewMessage()
msg.SetHeader("From", from)
msg.SetHeader("To", to...)
msg.SetHeader("Subject", subject)
msg.SetDateHeader("Date", time.Now())
plainBody, err := html2text.FromString(body)
if err != nil || setting.MailService.SendAsPlainText {
if strings.Contains(body[:100], "<html>") {
log.Warn("Mail contains HTML but configured to send as plain text.")
}
msg.SetBody("text/plain", plainBody)
} else {
msg.SetBody("text/plain", plainBody)
msg.AddAlternative("text/html", body)
}
return &Message{
Message: msg,
}
}
// NewMessage creates new mail message object with default From header.
func NewMessage(to []string, subject, body string) *Message {
return NewMessageFrom(to, setting.MailService.From, subject, body)
}
type loginAuth struct {
2015-08-20 16:42:55 +05:30
username, password string
}
2016-11-25 07:14:04 +05:30
// LoginAuth SMTP AUTH LOGIN Auth Handler
func LoginAuth(username, password string) smtp.Auth {
return &loginAuth{username, password}
}
2016-11-25 07:14:04 +05:30
// Start start SMTP login auth
func (a *loginAuth) Start(server *smtp.ServerInfo) (string, []byte, error) {
return "LOGIN", []byte{}, nil
}
2016-11-25 07:14:04 +05:30
// Next next step of SMTP login auth
func (a *loginAuth) Next(fromServer []byte, more bool) ([]byte, error) {
if more {
switch string(fromServer) {
case "Username:":
return []byte(a.username), nil
case "Password:":
return []byte(a.password), nil
default:
2015-08-20 16:42:55 +05:30
return nil, fmt.Errorf("unknwon fromServer: %s", string(fromServer))
}
}
return nil, nil
}
// Sender SMTP mail sender
type smtpSender struct {
2014-03-19 17:57:27 +05:30
}
2016-11-25 07:14:04 +05:30
// Send send email
func (s *smtpSender) Send(from string, to []string, msg io.WriterTo) error {
opts := setting.MailService
2014-03-20 06:35:48 +05:30
host, port, err := net.SplitHostPort(opts.Host)
if err != nil {
return err
}
tlsconfig := &tls.Config{
InsecureSkipVerify: opts.SkipVerify,
ServerName: host,
}
if opts.UseCertificate {
cert, err := tls.LoadX509KeyPair(opts.CertFile, opts.KeyFile)
if err != nil {
return err
}
tlsconfig.Certificates = []tls.Certificate{cert}
}
2014-12-19 10:54:17 +05:30
conn, err := net.Dial("tcp", net.JoinHostPort(host, port))
if err != nil {
return err
}
defer conn.Close()
2014-12-19 10:54:17 +05:30
isSecureConn := false
// Start TLS directly if the port ends with 465 (SMTPS protocol)
if strings.HasSuffix(port, "465") {
conn = tls.Client(conn, tlsconfig)
2014-12-19 10:54:17 +05:30
isSecureConn = true
}
2014-12-19 10:54:17 +05:30
client, err := smtp.NewClient(conn, host)
if err != nil {
2015-12-19 13:14:34 +05:30
return fmt.Errorf("NewClient: %v", err)
}
if !opts.DisableHelo {
hostname := opts.HeloHostname
2015-07-03 11:38:18 +05:30
if len(hostname) == 0 {
hostname, err = os.Hostname()
if err != nil {
return err
}
}
2015-02-20 12:42:27 +05:30
2015-07-03 11:38:18 +05:30
if err = client.Hello(hostname); err != nil {
2015-12-19 13:14:34 +05:30
return fmt.Errorf("Hello: %v", err)
2015-07-03 11:38:18 +05:30
}
2015-02-20 12:42:27 +05:30
}
// If not using SMTPS, always use STARTTLS if available
2014-12-19 10:54:17 +05:30
hasStartTLS, _ := client.Extension("STARTTLS")
if !isSecureConn && hasStartTLS {
if err = client.StartTLS(tlsconfig); err != nil {
2015-12-19 13:14:34 +05:30
return fmt.Errorf("StartTLS: %v", err)
}
}
2014-12-19 10:54:17 +05:30
canAuth, options := client.Extension("AUTH")
if canAuth && len(opts.User) > 0 {
var auth smtp.Auth
if strings.Contains(options, "CRAM-MD5") {
auth = smtp.CRAMMD5Auth(opts.User, opts.Passwd)
} else if strings.Contains(options, "PLAIN") {
auth = smtp.PlainAuth("", opts.User, opts.Passwd, host)
} else if strings.Contains(options, "LOGIN") {
2015-08-20 16:42:55 +05:30
// Patch for AUTH LOGIN
auth = LoginAuth(opts.User, opts.Passwd)
}
if auth != nil {
if err = client.Auth(auth); err != nil {
2015-12-19 13:14:34 +05:30
return fmt.Errorf("Auth: %v", err)
}
}
}
if err = client.Mail(from); err != nil {
2015-12-19 13:14:34 +05:30
return fmt.Errorf("Mail: %v", err)
}
for _, rec := range to {
if err = client.Rcpt(rec); err != nil {
2015-12-19 13:14:34 +05:30
return fmt.Errorf("Rcpt: %v", err)
}
}
w, err := client.Data()
if err != nil {
2015-12-19 13:14:34 +05:30
return fmt.Errorf("Data: %v", err)
} else if _, err = msg.WriteTo(w); err != nil {
2015-12-19 13:14:34 +05:30
return fmt.Errorf("WriteTo: %v", err)
} else if err = w.Close(); err != nil {
2015-12-19 13:14:34 +05:30
return fmt.Errorf("Close: %v", err)
}
return client.Quit()
}
// Sender sendmail mail sender
type sendmailSender struct {
}
// Send send email
func (s *sendmailSender) Send(from string, to []string, msg io.WriterTo) error {
var err error
var closeError error
var waitError error
args := []string{"-F", from, "-i"}
args = append(args, to...)
log.Trace("Sending with: %s %v", setting.MailService.SendmailPath, args)
cmd := exec.Command(setting.MailService.SendmailPath, args...)
pipe, err := cmd.StdinPipe()
if err != nil {
return err
}
if err = cmd.Start(); err != nil {
return err
}
_, err = msg.WriteTo(pipe)
2014-03-19 17:57:27 +05:30
// we MUST close the pipe or sendmail will hang waiting for more of the message
// Also we should wait on our sendmail command even if something fails
closeError = pipe.Close()
waitError = cmd.Wait()
if err != nil {
return err
} else if closeError != nil {
return closeError
} else {
return waitError
}
}
func processMailQueue() {
for {
select {
case msg := <-mailQueue:
log.Trace("New e-mail sending request %s: %s", msg.GetHeader("To"), msg.Info)
if err := gomail.Send(Sender, msg.Message); err != nil {
log.Error(3, "Failed to send emails %s: %s - %v", msg.GetHeader("To"), msg.Info, err)
} else {
log.Trace("E-mails sent %s: %s", msg.GetHeader("To"), msg.Info)
2014-03-19 17:57:27 +05:30
}
}
}
}
2014-03-19 17:57:27 +05:30
var mailQueue chan *Message
// Sender sender for sending mail synchronously
var Sender gomail.Sender
2016-11-25 07:14:04 +05:30
// NewContext start mail queue service
func NewContext() {
// Need to check if mailQueue is nil because in during reinstall (user had installed
// before but swithed install lock off), this function will be called again
// while mail queue is already processing tasks, and produces a race condition.
if setting.MailService == nil || mailQueue != nil {
return
2014-03-19 17:57:27 +05:30
}
if setting.MailService.UseSendmail {
Sender = &sendmailSender{}
} else {
Sender = &smtpSender{}
}
mailQueue = make(chan *Message, setting.MailService.QueueLength)
go processMailQueue()
2014-03-19 17:57:27 +05:30
}
2016-11-25 07:14:04 +05:30
// SendAsync send mail asynchronous
2014-03-20 06:35:48 +05:30
func SendAsync(msg *Message) {
2014-03-19 17:57:27 +05:30
go func() {
2014-03-20 06:35:48 +05:30
mailQueue <- msg
2014-03-19 17:57:27 +05:30
}()
}