mirrors/ntfy
1
0
Fork 0
mirror of https://github.com/binwiederhier/ntfy.git synced 2025-04-04 21:57:40 +03:00
Code Issues Projects Releases Packages Wiki Activity

Docs

Browse source
This commit is contained in:
binwiederhier 2024-09-29 13:12:51 -04:00
parent fc3624cd50
commit d38c149263
7 changed files with 55 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

21
server/smtp_server.go
View file

@ -70,19 +70,18 @@ func (b *smtpBackend) Counts() (total int64, success int64, failure int64) {
// smtpSession is returned after EHLO.
type smtpSession struct {
backend *smtpBackend
conn *smtp.Conn
topic string
token string
mu sync.Mutex
basic_auth string
backend *smtpBackend
conn *smtp.Conn
topic string
token string // If email address contains token, e.g. topic+token@domain
basicAuth string // If SMTP AUTH PLAIN was used
mu sync.Mutex
}
func (s *smtpSession) AuthPlain(username, password string) error {
logem(s.conn).Field("smtp_username", username).Debug("AUTH PLAIN (with username %s)", username)
basic_auth := base64.StdEncoding.EncodeToString([]byte(fmt.Sprintf("%s:%s", username, password)))
s.mu.Lock()
s.basic_auth = basic_auth
s.basicAuth = base64.StdEncoding.EncodeToString([]byte(fmt.Sprintf("%s:%s", username, password)))
s.mu.Unlock()
return nil
}
@ -203,8 +202,8 @@ func (s *smtpSession) publishMessage(m *message) error {
}
if s.token != "" {
req.Header.Add("Authorization", "Bearer "+s.token)
} else if s.basic_auth != "" {
req.Header.Add("Authorization", "Basic "+s.basic_auth)
} else if s.basicAuth != "" {
req.Header.Add("Authorization", "Basic "+s.basicAuth)
}
rr := httptest.NewRecorder()
s.backend.handler(rr, req)
@ -222,7 +221,7 @@ func (s *smtpSession) Reset() {
func (s *smtpSession) Logout() error {
s.mu.Lock()
s.basic_auth = ""
s.basicAuth = ""
s.mu.Unlock()
return nil
}