]> source.dussan.org Git - gitea.git/commitdiff
Allow HOST has no port (#22280) (#22409)
authorLunny Xiao <xiaolunwen@gmail.com>
Thu, 12 Jan 2023 01:57:03 +0000 (09:57 +0800)
committerGitHub <noreply@github.com>
Thu, 12 Jan 2023 01:57:03 +0000 (09:57 +0800)
Fix #22274
Backport #22280

This PR will allow `HOST` without port. Then a default port will be
given in future steps.

modules/setting/mailer.go
modules/setting/mailer_test.go [new file with mode: 0644]
modules/setting/setting.go

index e21d8c83cd3061c1162dd8bc259a8a36ec18bb4d..4dc16bebafad1c9f2efd06776f578b5836b67cc4 100644 (file)
@@ -13,6 +13,7 @@ import (
        "code.gitea.io/gitea/modules/log"
 
        shellquote "github.com/kballard/go-shellquote"
+       ini "gopkg.in/ini.v1"
 )
 
 // Mailer represents mail service.
@@ -50,8 +51,8 @@ type Mailer struct {
 // MailService the global mailer
 var MailService *Mailer
 
-func newMailService() {
-       sec := Cfg.Section("mailer")
+func parseMailerConfig(rootCfg *ini.File) {
+       sec := rootCfg.Section("mailer")
        // Check mailer setting.
        if !sec.Key("ENABLED").MustBool() {
                return
@@ -71,9 +72,14 @@ func newMailService() {
        if sec.HasKey("HOST") && !sec.HasKey("SMTP_ADDR") {
                givenHost := sec.Key("HOST").String()
                addr, port, err := net.SplitHostPort(givenHost)
-               if err != nil {
+               if err != nil && strings.Contains(err.Error(), "missing port in address") {
+                       addr = givenHost
+               } else if err != nil {
                        log.Fatal("Invalid mailer.HOST (%s): %v", givenHost, err)
                }
+               if addr == "" {
+                       addr = "127.0.0.1"
+               }
                sec.Key("SMTP_ADDR").MustString(addr)
                sec.Key("SMTP_PORT").MustString(port)
        }
@@ -173,6 +179,9 @@ func newMailService() {
                        default:
                                log.Error("unable to infer unspecified mailer.PROTOCOL from mailer.SMTP_PORT = %q, assume using smtps", MailService.SMTPPort)
                                MailService.Protocol = "smtps"
+                               if MailService.SMTPPort == "" {
+                                       MailService.SMTPPort = "465"
+                               }
                        }
                }
        }
diff --git a/modules/setting/mailer_test.go b/modules/setting/mailer_test.go
new file mode 100644 (file)
index 0000000..0fc9b0e
--- /dev/null
@@ -0,0 +1,43 @@
+// Copyright 2022 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package setting
+
+import (
+       "testing"
+
+       "github.com/stretchr/testify/assert"
+       ini "gopkg.in/ini.v1"
+)
+
+func TestParseMailerConfig(t *testing.T) {
+       iniFile := ini.Empty()
+       kases := map[string]*Mailer{
+               "smtp.mydomain.com": {
+                       SMTPAddr: "smtp.mydomain.com",
+                       SMTPPort: "465",
+               },
+               "smtp.mydomain.com:123": {
+                       SMTPAddr: "smtp.mydomain.com",
+                       SMTPPort: "123",
+               },
+               ":123": {
+                       SMTPAddr: "127.0.0.1",
+                       SMTPPort: "123",
+               },
+       }
+       for host, kase := range kases {
+               t.Run(host, func(t *testing.T) {
+                       iniFile.DeleteSection("mailer")
+                       sec := iniFile.Section("mailer")
+                       sec.NewKey("ENABLED", "true")
+                       sec.NewKey("HOST", host)
+
+                       // Check mailer setting
+                       parseMailerConfig(iniFile)
+
+                       assert.EqualValues(t, kase.SMTPAddr, MailService.SMTPAddr)
+                       assert.EqualValues(t, kase.SMTPPort, MailService.SMTPPort)
+               })
+       }
+}
index 3897b9c56bd0959faa52abd0066b8cf0fb33b10c..1eefa188c2e700d4c2495f5e4d92fe758dc738d1 100644 (file)
@@ -1334,7 +1334,7 @@ func NewServices() {
        newCacheService()
        newSessionService()
        newCORSService()
-       newMailService()
+       parseMailerConfig(Cfg)
        newRegisterMailService()
        newNotifyMailService()
        newProxyService()
@@ -1351,5 +1351,5 @@ func NewServices() {
 // NewServicesForInstall initializes the services for install
 func NewServicesForInstall() {
        newService()
-       newMailService()
+       parseMailerConfig(Cfg)
 }