-
Notifications
You must be signed in to change notification settings - Fork 0
/
dial.go
79 lines (70 loc) · 1.92 KB
/
dial.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
package nmail
import (
"context"
"crypto/tls"
"fmt"
"net/smtp"
"strings"
)
func DefaultDialer(ctx context.Context, username, password, host, port string, tlsConfig *tls.Config) (*SMTPClient, error) {
var addr = fmt.Sprintf("%s:%s", host, port)
nClient, err := smtp.Dial(addr)
if err != nil {
return nil, err
}
if err = nClient.Hello("localhost"); err != nil {
nClient.Close()
return nil, err
}
if ok, _ := nClient.Extension("STARTTLS"); ok {
if err = nClient.StartTLS(tlsConfig); err != nil {
nClient.Close()
return nil, err
}
}
if ok, auth := nClient.Extension("AUTH"); ok {
var nAuth smtp.Auth
if strings.Contains(auth, "CRAM-MD5") {
nAuth = smtp.CRAMMD5Auth(username, password)
} else if strings.Contains(auth, "LOGIN") && !strings.Contains(auth, "PLAIN") {
nAuth = LoginAuth(username, password)
} else {
nAuth = smtp.PlainAuth("", username, password, host)
}
if err = nClient.Auth(nAuth); err != nil {
nClient.Close()
return nil, err
}
}
return &SMTPClient{Client: nClient}, nil
}
func TLSDialer(ctx context.Context, username, password, host, port string, tlsConfig *tls.Config) (*SMTPClient, error) {
var addr = fmt.Sprintf("%s:%s", host, port)
conn, err := tls.Dial("tcp", addr, tlsConfig)
if err != nil {
return nil, err
}
nClient, err := smtp.NewClient(conn, host)
if err != nil {
return nil, err
}
if err = nClient.Hello("localhost"); err != nil {
nClient.Close()
return nil, err
}
if ok, auth := nClient.Extension("AUTH"); ok {
var nAuth smtp.Auth
if strings.Contains(auth, "CRAM-MD5") {
nAuth = smtp.CRAMMD5Auth(username, password)
} else if strings.Contains(auth, "LOGIN") && !strings.Contains(auth, "PLAIN") {
nAuth = LoginAuth(username, password)
} else {
nAuth = smtp.PlainAuth("", username, password, host)
}
if err = nClient.Auth(nAuth); err != nil {
nClient.Close()
return nil, err
}
}
return &SMTPClient{Client: nClient}, nil
}