mirror of https://github.com/v2ray/v2ray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
811 B
41 lines
811 B
package tls
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"net"
|
|
|
|
"v2ray.com/core/common/buf"
|
|
)
|
|
|
|
//go:generate go run $GOPATH/src/v2ray.com/core/common/errors/errorgen/main.go -pkg tls -path Transport,Internet,TLS
|
|
|
|
var (
|
|
_ buf.Writer = (*conn)(nil)
|
|
)
|
|
|
|
type conn struct {
|
|
net.Conn
|
|
|
|
mergingWriter *buf.BufferedWriter
|
|
}
|
|
|
|
func (c *conn) WriteMultiBuffer(mb buf.MultiBuffer) error {
|
|
if c.mergingWriter == nil {
|
|
c.mergingWriter = buf.NewBufferedWriter(buf.NewWriter(c.Conn))
|
|
}
|
|
if err := c.mergingWriter.WriteMultiBuffer(mb); err != nil {
|
|
return err
|
|
}
|
|
return c.mergingWriter.Flush()
|
|
}
|
|
|
|
func Client(c net.Conn, config *tls.Config) net.Conn {
|
|
tlsConn := tls.Client(c, config)
|
|
return &conn{Conn: tlsConn}
|
|
}
|
|
|
|
func Server(c net.Conn, config *tls.Config) net.Conn {
|
|
tlsConn := tls.Server(c, config)
|
|
return &conn{Conn: tlsConn}
|
|
}
|