auth.go 1.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. package gomail
  2. import (
  3. "bytes"
  4. "errors"
  5. "fmt"
  6. "net/smtp"
  7. )
  8. // loginAuth is an smtp.Auth that implements the LOGIN authentication mechanism.
  9. type loginAuth struct {
  10. username string
  11. password string
  12. host string
  13. }
  14. func (a *loginAuth) Start(server *smtp.ServerInfo) (string, []byte, error) {
  15. if !server.TLS {
  16. advertised := false
  17. for _, mechanism := range server.Auth {
  18. if mechanism == "LOGIN" {
  19. advertised = true
  20. break
  21. }
  22. }
  23. if !advertised {
  24. return "", nil, errors.New("gomail: unencrypted connection")
  25. }
  26. }
  27. if server.Name != a.host {
  28. return "", nil, errors.New("gomail: wrong host name")
  29. }
  30. return "LOGIN", nil, nil
  31. }
  32. func (a *loginAuth) Next(fromServer []byte, more bool) ([]byte, error) {
  33. if !more {
  34. return nil, nil
  35. }
  36. switch {
  37. case bytes.Equal(fromServer, []byte("Username:")):
  38. return []byte(a.username), nil
  39. case bytes.Equal(fromServer, []byte("Password:")):
  40. return []byte(a.password), nil
  41. default:
  42. return nil, fmt.Errorf("gomail: unexpected server challenge: %s", fromServer)
  43. }
  44. }