[golang]查询ssl证书剩余有效天数并邮件提醒

媒介

自从云厂商的免费ssl证书改成3个月，而且证书数量还是20个之后，自己网站的ssl证书就换成了别的免费方案。但是免费方案不会提醒证书过期，所以写个工具每天定时查询证书剩余有效天数，如果证书即将过期，就发送邮件提醒。
基本实现

最基本的代码功能就是检测网站ssl证书的有效天数，可以用下令行传参的方式指定网站域名。

1. package main
3. import (
4.         "crypto/tls"
5.         "flag"
6.         "fmt"
7.         "net"
8.         "os"
9.         "sync"
10.         "time"
11. )
13. var (
14.         port int
15.         wg   sync.WaitGroup
16. )
18. func checkssl(domain string, port int) {
19.         defer wg.Done()
20.         host := fmt.Sprintf("%s:%d", domain, port)
21.         conn, err := tls.DialWithDialer(&net.Dialer{
22.                 Timeout:  time.Second * 5,
23.                 Deadline: time.Now().Add(time.Second * 5),
24.         }, "tcp", host, &tls.Config{InsecureSkipVerify: true})
25.         if err != nil {
26.                 fmt.Println(err)
27.                 return
28.         }
29.         defer conn.Close()
31.         stats := conn.ConnectionState()
32.         certs := stats.PeerCertificates[0]
33.         localtz, _ := time.LoadLocation("Asia/Shanghai")
34.         issueTime := certs.NotBefore.In(localtz)
35.         expireTime := certs.NotAfter.In(localtz)
37.         today := time.Now().In(localtz)
38.         dayLeft := int(expireTime.Sub(today).Hours() / 24)
39.         fmt.Printf("%s, issue time: %v, expire time: %v, days left: %v\n", domain, issueTime, expireTime, dayLeft)
40. }
42. func main() {
43.         flag.IntVar(&port, "p", 443, "port, example: ./checkssl -p 1443 <domain name>")
44.         flag.Parse()
45.         positionArgs := flag.Args()
46.         if len(positionArgs) == 0 {
47.                 fmt.Println("Error: Missing domain name")
48.                 fmt.Println("Usage: ./checkssl <domain name>")
49.                 os.Exit(1)
50.         }
52.         wg.Add(len(positionArgs))
53.         for _, arg := range positionArgs {
54.                 go checkssl(arg, port)
55.         }
56.         wg.Wait()
57. }

复制代码

使用示例

1. # 1. 编译
2. go build
3. # 2. 命令行传参的方式指定域名
4. ./check-ssl baidu.com ithome.com qq.com
6. # 输出
7. baidu.com, issue time: 2024-01-30 08:00:00 +0800 CST, expire time: 2025-03-02 07:59:59 +0800 CST, days left: 187
8. ithome.com, issue time: 2024-01-22 08:00:00 +0800 CST, expire time: 2025-02-22 07:59:59 +0800 CST, days left: 179
9. qq.com, issue time: 2024-06-04 08:00:00 +0800 CST, expire time: 2025-06-11 07:59:59 +0800 CST, days left: 288

复制代码

完善功能

需要完善的功能紧张是发送邮件，这里使用SMTP协议来发送邮件。如果跟我一样用的是163邮箱，则需要先去获取一个SMTP的授权码。
因为需要配置SMTP的连接信息，所以改成了用文件来传入配置，也方便后期修改。配置文件config.yaml示例：

1. domains:
2.   - baidu.com
3.   - qq.com
5. email:
6.   smtp:
7.     host: "smtp.163.com"  # smtp服务器的地址
8.     port: 465  # 因为云服务器屏蔽了25端口, 只能使用tls加密的465端口
9.     from: ""   # 发送方邮箱
10.     token: ""  # 授权码
11.   sendto:
12.     - "qq@qq.com"  # 接收方的邮箱地址
13.   expire: 7  # 证书剩余有效天数, 小于7天时发送邮件提醒

复制代码

读取配置的代码文件config.go，使用viper来读取配置文件。

1. package main
3. import "github.com/spf13/viper"
5. var (
6.         v *viper.Viper
7. )
9. type SMTPServer struct {
10.         Host  string
11.         Port  int
12.         Token string
13.         From  string
14. }
16. func initViper() {
17.         v = viper.New()
18.         v.AddConfigPath(".")
19.         v.SetConfigType("yaml")
20.         v.SetConfigFile(configfile)
21.         err := v.ReadInConfig()
22.         if err != nil {
23.                 panic(err)
24.         }
25. }
27. type configer struct{}
29. func NewConfiger() configer {
30.         if v == nil {
31.                 initViper()
32.         }
33.         return configer{}
34. }
36. func (c configer) GetSMTPServer() SMTPServer {
37.         return SMTPServer{
38.                 Host:  v.GetString("email.smtp.host"),
39.                 Port:  v.GetInt("email.smtp.port"),
40.                 Token: v.GetString("email.smtp.token"),
41.                 From:  v.GetString("email.smtp.from"),
42.         }
43. }
45. func (c configer) GetDomains() []string {
46.         return v.GetStringSlice("domains")
47. }
49. func (c configer) GetSendTos() []string {
50.         return v.GetStringSlice("email.sendto")
51. }
53. func (c configer) GetExpiry() int {
54.         return v.GetInt("email.expire")
55. }

复制代码

发送邮件的相关代码文件：notify.go

1. package main
3. import (
4.         "crypto/tls"
5.         "fmt"
6.         "net/smtp"
8.         "github.com/jordan-wright/email"
9. )
11. type Postman struct {
12.         SmtpServer SMTPServer
13.         SendTos    []string
14. }
16. func (p Postman) SendEmail(domain string, dayleft int) {
17.         auth := smtp.PlainAuth("", p.SmtpServer.From, p.SmtpServer.Token, p.SmtpServer.Host)
18.         e := &email.Email{
19.                 To:      p.SendTos,
20.                 From:    fmt.Sprintf("YXHYW <%s>", p.SmtpServer.From),
21.                 Subject: fmt.Sprintf("域名 %s SSL证书过期提醒", domain),
22.                 Text:    []byte(fmt.Sprintf("域名 %s 的SSL证书即将过期, 剩余有效期 %d 天", domain, dayleft)),
23.         }
24.         // err := e.Send(fmt.Sprintf("%s:%d", p.SmtpServer.Host, p.SmtpServer.Port), auth)
25.         addr := fmt.Sprintf("%s:%d", p.SmtpServer.Host, p.SmtpServer.Port)
26.         fmt.Println("SMTP Server addr: ", addr)
27.         err := e.SendWithTLS(addr, auth, &tls.Config{
28.                 InsecureSkipVerify: false,
29.                 ServerName:         p.SmtpServer.Host,
30.         })
31.         if err != nil {
32.                 fmt.Printf("Send email failed, %v\n", err)
33.         }
34. }

复制代码

主体代码文件main.go，紧张修改地方：检测到证书即将过期后，调用发送邮件的相关方法。

1. package main
3. import (
4.         "crypto/tls"
5.         "flag"
6.         "fmt"
7.         "net"
8.         "sync"
9.         "time"
10. )
12. var (
13.         port       int
14.         configfile string
15.         wg         sync.WaitGroup
16.         c          configer = NewConfiger()
17. )
19. func checkssl(domain string, port int) {
20.         defer wg.Done()
21.         host := fmt.Sprintf("%s:%d", domain, port)
22.         conn, err := tls.DialWithDialer(&net.Dialer{
23.                 Timeout:  time.Second * 5,
24.                 Deadline: time.Now().Add(time.Second * 5),
25.         }, "tcp", host, &tls.Config{InsecureSkipVerify: true})
26.         if err != nil {
27.                 fmt.Println(err)
28.                 return
29.         }
30.         defer conn.Close()
32.         stats := conn.ConnectionState()
33.         certs := stats.PeerCertificates[0]
34.         localtz, _ := time.LoadLocation("Asia/Shanghai")
35.         issueTime := certs.NotBefore.In(localtz)
36.         expireTime := certs.NotAfter.In(localtz)
38.         today := time.Now().In(localtz)
39.         dayLeft := int(expireTime.Sub(today).Hours() / 24)
40.         fmt.Printf("%s, issue time: %v, expire time: %v, days left: %v\n", domain, issueTime, expireTime, dayLeft)
42.         // c := NewConfiger()
43.         if dayLeft < c.GetExpiry() {
44.                 p := Postman{SmtpServer: c.GetSMTPServer(), SendTos: c.GetSendTos()}
45.                 p.SendEmail(domain, dayLeft)
46.         }
47. }
49. func main() {
50.         flag.IntVar(&port, "p", 443, "port, example: ./check-ssl -p 1443 <domain name>")
51.         flag.StringVar(&configfile, "c", "config.yaml", "config file")
52.         flag.Parse()
54.         conf := NewConfiger()
55.         domains := conf.GetDomains()
57.         wg.Add(len(domains))
58.         for _, arg := range domains {
59.                 go checkssl(arg, port)
60.         }
61.         wg.Wait()
62. }

复制代码

当地测试通过后，可以配到服务器的crontab中每天执行。

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。