mirror
/
redis
mirror of https://github.com/go-redis/redis.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

add test for tls connCheck #3025 (#3047) 

* add a check for TLS connections.
This commit is contained in:
naiqianz 2024-07-12 11:16:21 +08:00 committed by vladvildanov
parent 5c9c468bd9
commit 233ff45ac9
2 changed files with 23 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
internal/pool/conn_check.go
View File

@ -3,6 +3,7 @@
package pool package pool
import ( import (
"crypto/tls"
"errors" "errors"
"io" "io"
"net" "net"
@ -16,6 +17,10 @@ func connCheck(conn net.Conn) error {
// Reset previous timeout. // Reset previous timeout.
_ = conn.SetDeadline(time.Time{}) _ = conn.SetDeadline(time.Time{})
// Check if tls.Conn.
if c, ok := conn.(*tls.Conn); ok {
conn = c.NetConn()
}
sysConn, ok := conn.(syscall.Conn) sysConn, ok := conn.(syscall.Conn)
if !ok { if !ok {
return nil return nil

18
internal/pool/conn_check_test.go
View File

@ -3,6 +3,7 @@
package pool package pool
import ( import (
"crypto/tls"
"net" "net"
"net/http/httptest" "net/http/httptest"
"time" "time"
@ -14,12 +15,17 @@ import (
var _ = Describe("tests conn_check with real conns", func() { var _ = Describe("tests conn_check with real conns", func() {
var ts *httptest.Server var ts *httptest.Server
var conn net.Conn var conn net.Conn
var tlsConn *tls.Conn
var err error var err error
BeforeEach(func() { BeforeEach(func() {
ts = httptest.NewServer(nil) ts = httptest.NewServer(nil)
conn, err = net.DialTimeout(ts.Listener.Addr().Network(), ts.Listener.Addr().String(), time.Second) conn, err = net.DialTimeout(ts.Listener.Addr().Network(), ts.Listener.Addr().String(), time.Second)
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
tlsTestServer := httptest.NewUnstartedServer(nil)
tlsTestServer.StartTLS()
tlsConn, err = tls.DialWithDialer(&net.Dialer{Timeout: time.Second}, tlsTestServer.Listener.Addr().Network(), tlsTestServer.Listener.Addr().String(), &tls.Config{InsecureSkipVerify: true})
Expect(err).NotTo(HaveOccurred())
}) })
AfterEach(func() { AfterEach(func() {
@ -33,11 +39,23 @@ var _ = Describe("tests conn_check with real conns", func() {
Expect(connCheck(conn)).To(HaveOccurred()) Expect(connCheck(conn)).To(HaveOccurred())
}) })
It("good tls conn check", func() {
Expect(connCheck(tlsConn)).NotTo(HaveOccurred())
Expect(tlsConn.Close()).NotTo(HaveOccurred())
Expect(connCheck(tlsConn)).To(HaveOccurred())
})
It("bad conn check", func() { It("bad conn check", func() {
Expect(conn.Close()).NotTo(HaveOccurred()) Expect(conn.Close()).NotTo(HaveOccurred())
Expect(connCheck(conn)).To(HaveOccurred()) Expect(connCheck(conn)).To(HaveOccurred())
}) })
It("bad tls conn check", func() {
Expect(tlsConn.Close()).NotTo(HaveOccurred())
Expect(connCheck(tlsConn)).To(HaveOccurred())
})
It("check conn deadline", func() { It("check conn deadline", func() {
Expect(conn.SetDeadline(time.Now())).NotTo(HaveOccurred()) Expect(conn.SetDeadline(time.Now())).NotTo(HaveOccurred())
time.Sleep(time.Millisecond * 10) time.Sleep(time.Millisecond * 10)