reduce memory usage in tls

transport/internet/tls/config_other.go

@@ -2,20 +2,62 @@
 
 package tls
 
-import "crypto/x509"
+import (
+	"crypto/x509"
+	"sync"
 
-func (c *Config) getCertPool() *x509.CertPool {
+	"v2ray.com/core/common/compare"
-	pool, err := x509.SystemCertPool()
+)
-	if err != nil {
+
-		newError("failed to get system cert pool.").Base(err).WriteToLog()
+type certPoolCache struct {
-		return nil
+	sync.Mutex
-	}
+	once       sync.Once
-	if pool != nil {
+	pool       *x509.CertPool
-		for _, cert := range c.Certificate {
+	extraCerts [][]byte
-			if cert.Usage == Certificate_AUTHORITY_VERIFY {
+}
-				pool.AppendCertsFromPEM(cert.Certificate)
+
-			}
+func (c *certPoolCache) hasCert(cert []byte) bool {
+	for _, xCert := range c.extraCerts {
+		if compare.BytesEqual(xCert, cert) {
+			return true
 		}
 	}
-	return pool
+	return false
+}
+
+func (c *certPoolCache) get(extraCerts []*Certificate) *x509.CertPool {
+	c.once.Do(func() {
+		pool, err := x509.SystemCertPool()
+		if err != nil {
+			newError("failed to get system cert pool.").Base(err).WriteToLog()
+			return
+		}
+		c.pool = pool
+	})
+
+	if c.pool == nil {
+		return nil
+	}
+
+	if len(extraCerts) == 0 {
+		return c.pool
+	}
+
+	c.Lock()
+	defer c.Unlock()
+
+	for _, cert := range extraCerts {
+		if !c.hasCert(cert.Certificate) {
+			c.pool.AppendCertsFromPEM(cert.Certificate)
+			c.extraCerts = append(c.extraCerts, cert.Certificate)
+		}
+	}
+
+	return c.pool
+}
+
+var combineCertPool certPoolCache
+
+func (c *Config) getCertPool() *x509.CertPool {
+	return combineCertPool.get(c.Certificate)
 }