mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-11-02 09:17:55 -04:00
101 lines
2.5 KiB
Go
101 lines
2.5 KiB
Go
package tls_test
|
|
|
|
import (
|
|
gotls "crypto/tls"
|
|
"crypto/x509"
|
|
"testing"
|
|
"time"
|
|
|
|
"v2ray.com/core/common"
|
|
"v2ray.com/core/common/protocol/tls/cert"
|
|
. "v2ray.com/core/transport/internet/tls"
|
|
. "v2ray.com/ext/assert"
|
|
)
|
|
|
|
func TestCertificateIssuing(t *testing.T) {
|
|
assert := With(t)
|
|
|
|
certificate := ParseCertificate(cert.MustGenerate(nil, cert.Authority(true), cert.KeyUsage(x509.KeyUsageCertSign)))
|
|
certificate.Usage = Certificate_AUTHORITY_ISSUE
|
|
|
|
c := &Config{
|
|
Certificate: []*Certificate{
|
|
certificate,
|
|
},
|
|
}
|
|
|
|
tlsConfig := c.GetTLSConfig()
|
|
v2rayCert, err := tlsConfig.GetCertificate(&gotls.ClientHelloInfo{
|
|
ServerName: "www.v2ray.com",
|
|
})
|
|
common.Must(err)
|
|
|
|
x509Cert, err := x509.ParseCertificate(v2rayCert.Certificate[0])
|
|
common.Must(err)
|
|
assert(x509Cert.NotAfter.After(time.Now()), IsTrue)
|
|
}
|
|
|
|
func TestExpiredCertificate(t *testing.T) {
|
|
assert := With(t)
|
|
|
|
caCert := cert.MustGenerate(nil, cert.Authority(true), cert.KeyUsage(x509.KeyUsageCertSign))
|
|
expiredCert := cert.MustGenerate(caCert, cert.NotAfter(time.Now().Add(time.Minute*-2)), cert.CommonName("www.v2ray.com"), cert.DNSNames("www.v2ray.com"))
|
|
|
|
certificate := ParseCertificate(caCert)
|
|
certificate.Usage = Certificate_AUTHORITY_ISSUE
|
|
|
|
certificate2 := ParseCertificate(expiredCert)
|
|
|
|
c := &Config{
|
|
Certificate: []*Certificate{
|
|
certificate,
|
|
certificate2,
|
|
},
|
|
}
|
|
|
|
tlsConfig := c.GetTLSConfig()
|
|
v2rayCert, err := tlsConfig.GetCertificate(&gotls.ClientHelloInfo{
|
|
ServerName: "www.v2ray.com",
|
|
})
|
|
common.Must(err)
|
|
|
|
x509Cert, err := x509.ParseCertificate(v2rayCert.Certificate[0])
|
|
common.Must(err)
|
|
assert(x509Cert.NotAfter.After(time.Now()), IsTrue)
|
|
}
|
|
|
|
func TestInsecureCertificates(t *testing.T) {
|
|
c := &Config{
|
|
AllowInsecureCiphers: true,
|
|
}
|
|
|
|
tlsConfig := c.GetTLSConfig()
|
|
if len(tlsConfig.CipherSuites) > 0 {
|
|
t.Fatal("Unexpected tls cipher suites list: ", tlsConfig.CipherSuites)
|
|
}
|
|
}
|
|
|
|
func BenchmarkCertificateIssuing(b *testing.B) {
|
|
certificate := ParseCertificate(cert.MustGenerate(nil, cert.Authority(true), cert.KeyUsage(x509.KeyUsageCertSign)))
|
|
certificate.Usage = Certificate_AUTHORITY_ISSUE
|
|
|
|
c := &Config{
|
|
Certificate: []*Certificate{
|
|
certificate,
|
|
},
|
|
}
|
|
|
|
tlsConfig := c.GetTLSConfig()
|
|
lenCerts := len(tlsConfig.Certificates)
|
|
|
|
b.ResetTimer()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
_, _ = tlsConfig.GetCertificate(&gotls.ClientHelloInfo{
|
|
ServerName: "www.v2ray.com",
|
|
})
|
|
delete(tlsConfig.NameToCertificate, "www.v2ray.com")
|
|
tlsConfig.Certificates = tlsConfig.Certificates[:lenCerts]
|
|
}
|
|
}
|